89 lines
2.3 KiB
Python
89 lines
2.3 KiB
Python
"""
|
||
Redis 协调运行时：进程内持有客户端与常用队列句柄。
|
||
|
||
本文件职责：懒创建 / 关闭；供 Worker、llm、outbox 使用。
|
||
禁止：在 HTTP 回调里用本运行时同步等 LLM 结果。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
import threading
|
||
from dataclasses import dataclass, field
|
||
from typing import Optional
|
||
|
||
from agent.config import Settings, get_settings
|
||
from agent.redis_coord.client import RedisClient, create_redis_client
|
||
from agent.redis_coord.stream_queue import StreamQueue, llm_queue, recognition_queue
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
@dataclass
|
||
class RedisRuntime:
|
||
"""
|
||
Redis 协调门面。
|
||
|
||
生命周期：HTTP/Worker 启动时 get_redis_runtime()；退出 shutdown。
|
||
"""
|
||
|
||
client: RedisClient
|
||
llm_stream: StreamQueue
|
||
recognition_stream: StreamQueue
|
||
settings: Settings
|
||
_closed: bool = field(default=False)
|
||
|
||
@classmethod
|
||
def create(cls, settings: Optional[Settings] = None) -> "RedisRuntime":
|
||
"""
|
||
建连并准备 LLM/识别 Stream 消费组（骨架）。
|
||
|
||
副作用：ensure_group；memory 后端无外部 IO。
|
||
"""
|
||
cfg = settings or get_settings()
|
||
client = create_redis_client(cfg)
|
||
llm = llm_queue(client)
|
||
recog = recognition_queue(client)
|
||
llm.ensure_group()
|
||
recog.ensure_group()
|
||
logger.info(
|
||
"RedisRuntime 就绪 backend=%s prefix=%s",
|
||
client.backend,
|
||
client.key_prefix,
|
||
)
|
||
return cls(
|
||
client=client,
|
||
llm_stream=llm,
|
||
recognition_stream=recog,
|
||
settings=cfg,
|
||
)
|
||
|
||
def close(self) -> None:
|
||
"""关闭底层连接。"""
|
||
if self._closed:
|
||
return
|
||
self.client.close()
|
||
self._closed = True
|
||
|
||
|
||
_runtime: Optional[RedisRuntime] = None
|
||
_guard = threading.Lock()
|
||
|
||
|
||
def get_redis_runtime(settings: Optional[Settings] = None) -> RedisRuntime:
|
||
"""进程内单例；首次创建时可能连 Redis。"""
|
||
global _runtime
|
||
with _guard:
|
||
if _runtime is None:
|
||
_runtime = RedisRuntime.create(settings)
|
||
return _runtime
|
||
|
||
|
||
def shutdown_redis_runtime() -> None:
|
||
"""进程退出时关闭。"""
|
||
global _runtime
|
||
with _guard:
|
||
if _runtime is not None:
|
||
_runtime.close()
|
||
_runtime = None
|