refactor(chat): 优化上下文管理器,添加启动与停止功能,改进消息用户ID获取逻辑

This commit is contained in:
Windpicker-owo
2025-09-27 16:31:23 +08:00
parent c49b3f3ac4
commit 9a70c7a93d
3 changed files with 43 additions and 5 deletions

View File

@@ -79,6 +79,17 @@ class MessageManager:
def add_message(self, stream_id: str, message: DatabaseMessages):
"""添加消息到指定聊天流"""
# 检查流上下文是否存在,不存在则创建
context = self.context_manager.get_stream_context(stream_id)
if not context:
# 创建新的流上下文
from src.common.data_models.message_manager_data_model import StreamContext
new_context = StreamContext(stream_id=stream_id)
success = self.context_manager.add_stream_context(stream_id, new_context)
if not success:
logger.error(f"无法为流 {stream_id} 创建上下文")
return
# 使用 context_manager 添加消息
success = self.context_manager.add_message_to_context(stream_id, message)