o Ղi @sUdZddlmZddlmZddlZddlZddlmZee Z dZ dZ Gdd d e eZGd d d Zdaeeed <d efddZdS)zZ Redis-based LLM service status manager. Manages inference/training mode switching state. )Enum)OptionalN) get_loggerzllm:service:statuszllm:training:infoc@seZdZdZdZdZdZdS) ServiceMode inferencetrainingmedia_processing switchingN)__name__ __module__ __qualname__ INFERENCETRAININGMEDIA_PROCESSING SWITCHINGrrO/lsinfo/ai/hellotax_ai/base_platform/app/services/llm/service_status_manager.pyrs rc@sfeZdZdejfddZdefddZdded ee ddfd d Z dee fd d Z de fddZ dS)ServiceStatusManager redis_clientcCs ||_dSN)_redis)selfrrrr__init__s zServiceStatusManager.__init__returnc Cs~z|jt}|durtjWSt|tr|n|}t|WStj t fy>}zt d|tjWYd}~Sd}~ww)zIReturn current service mode from Redis. Defaults to INFERENCE if not set.Nz*Failed to read service status from Redis: ) rgetREDIS_KEY_STATUSrr isinstancebytesdecoderedis RedisError ValueErrorloggerwarningrvaluedecodederrrget_current_modes  z%ServiceStatusManager.get_current_modeNmode training_infoc Csz1|jt|j|tjkr|r|jtt|n |tj kr&|j tt d|jWdSt jyG}z t d|d}~ww)z&Set the current service mode in Redis.zService mode set to: z%Failed to set service mode in Redis: N)rsetrr%rrREDIS_KEY_TRAINING_INFOjsondumpsr deleter"inforr error)rr)r*r'rrrset_mode(s  zServiceStatusManager.set_modec Cs~z|jt}|durWdSt|tr|n|}t|WStj tj t fy>}zt d|WYd}~dSd}~ww)z7Return training task metadata stored in Redis, or None.Nz)Failed to read training info from Redis: )rrr,rrrr-loadsrr JSONDecodeErrorr!r"r#r$rrrget_training_info5s  z&ServiceStatusManager.get_training_infocCs|tjkS)z(Return True only when mode is INFERENCE.)r(rr )rrrris_inference_availableAsz+ServiceStatusManager.is_inference_availabler)r r r rRedisrrr(rdictr2r5boolr6rrrrrs    r_status_managerrcCsXtdur*ddlm}|jrtj|jddd}ntj|j|j|j |j ddd}t |atS)z7Return the module-level ServiceStatusManager singleton.Nr)settingsF)decode_responsessocket_connect_timeout)hostportdbpasswordr=r>) r: app.configr; REDIS_URLrfrom_urlr7 REDIS_HOST REDIS_PORTREDIS_DBREDIS_PASSWORDr)r;clientrrrget_status_managerJs rK)__doc__enumrtypingrr-rapp.core.loggingrr r"rr,strrrr:__annotations__rKrrrrs   /