97dc145b76
Changed get_replyer and related calls to async across multiple modules for proper coroutine handling. Added format_memories_bracket_style utility for memory formatting. Improved video analysis caching logic and type annotations. Updated error logging for message processing.
58 lines
2.1 KiB
Python
58 lines
2.1 KiB
Python
from src.chat.message_receive.chat_stream import ChatStream, get_chat_manager
|
||
from src.chat.replyer.default_generator import DefaultReplyer
|
||
from src.common.logger import get_logger
|
||
|
||
logger = get_logger("ReplyerManager")
|
||
|
||
|
||
class ReplyerManager:
|
||
def __init__(self):
|
||
self._repliers: dict[str, DefaultReplyer] = {}
|
||
|
||
async def get_replyer(
|
||
self,
|
||
chat_stream: ChatStream | None = None,
|
||
chat_id: str | None = None,
|
||
request_type: str = "replyer",
|
||
) -> DefaultReplyer | None:
|
||
"""
|
||
获取或创建回复器实例。
|
||
|
||
model_configs 仅在首次为某个 chat_id/stream_id 创建实例时有效。
|
||
后续调用将返回已缓存的实例，忽略 model_configs 参数。
|
||
"""
|
||
stream_id = chat_stream.stream_id if chat_stream else chat_id
|
||
if not stream_id:
|
||
logger.warning("[ReplyerManager] 缺少 stream_id，无法获取回复器。")
|
||
return None
|
||
|
||
# 如果已有缓存实例，直接返回
|
||
if stream_id in self._repliers:
|
||
logger.debug(f"[ReplyerManager] 为 stream_id '{stream_id}' 返回已存在的回复器实例。")
|
||
return self._repliers[stream_id]
|
||
|
||
# 如果没有缓存，则创建新实例（首次初始化）
|
||
logger.debug(f"[ReplyerManager] 为 stream_id '{stream_id}' 创建新的回复器实例并缓存。")
|
||
|
||
target_stream = chat_stream
|
||
if not target_stream:
|
||
if chat_manager := get_chat_manager():
|
||
# get_stream 为异步，需要等待
|
||
target_stream = await chat_manager.get_stream(stream_id)
|
||
|
||
if not target_stream:
|
||
logger.warning(f"[ReplyerManager] 未找到 stream_id='{stream_id}' 的聊天流，无法创建回复器。")
|
||
return None
|
||
|
||
# model_configs 只在此时（初始化时）生效
|
||
replyer = DefaultReplyer(
|
||
chat_stream=target_stream,
|
||
request_type=request_type,
|
||
)
|
||
self._repliers[stream_id] = replyer
|
||
return replyer
|
||
|
||
|
||
# 创建一个全局实例
|
||
replyer_manager = ReplyerManager()
|