gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

重构ChatStream和StreamContext：移除context_manager引用

Browse Source 
- 在ChatStream及相关类中，将所有context_manager的实例替换为直接上下文访问。
- 更新方法，利用新的上下文结构来管理聊天状态和消息。
- 增强的StreamContext，增加了用于消息处理、统计和历史管理的方法。
- 在重构过程中改进了错误处理和日志记录。
This commit is contained in:
Windpicker-owo
2025-11-25 12:01:26 +08:00
parent d30b0544b5
commit 1ebdc37b22
16 changed files with 487 additions and 753 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
src/chat/replyer/default_generator.py
View File

@@ -396,7 +396,7 @@ class DefaultReplyer:
try:
# 设置正在回复的状态
self.chat_stream.context_manager.context.is_replying = True
self.chat_stream.context.is_replying = True
content, reasoning_content, model_name, tool_call = await self.llm_generate_content(prompt)
logger.debug(f"replyer生成内容: {content}")
llm_response = {
@@ -413,7 +413,7 @@ class DefaultReplyer:
return False, None, prompt # LLM 调用失败则无法生成回复
finally:
# 重置正在回复的状态
self.chat_stream.context_manager.context.is_replying = False
self.chat_stream.context.is_replying = False
# 触发 AFTER_LLM 事件
if not from_plugin:
@@ -910,7 +910,7 @@ class DefaultReplyer:
chat_manager = get_chat_manager()
chat_stream = await chat_manager.get_stream(chat_id)
if chat_stream:
stream_context = chat_stream.context_manager
stream_context = chat_stream.context
# 确保历史消息已从数据库加载
await stream_context.ensure_history_initialized()
@@ -1140,7 +1140,7 @@ class DefaultReplyer:
chat_stream_obj = await chat_manager.get_stream(chat_id)
if chat_stream_obj:
unread_messages = chat_stream_obj.context_manager.get_unread_messages()
unread_messages = chat_stream_obj.context.get_unread_messages()
if unread_messages:
# 使用最后一条未读消息作为参考
last_msg = unread_messages[-1]
@@ -1262,12 +1262,12 @@ class DefaultReplyer:
if chat_stream_obj:
# 确保历史消息已初始化
await chat_stream_obj.context_manager.ensure_history_initialized()
await chat_stream_obj.context.ensure_history_initialized()
# 获取所有消息（历史+未读）
all_messages = (
chat_stream_obj.context_manager.context.history_messages +
chat_stream_obj.context_manager.get_unread_messages()
chat_stream_obj.context.history_messages +
chat_stream_obj.context.get_unread_messages()
)
# 转换为字典格式
@@ -1639,12 +1639,12 @@ class DefaultReplyer:
if chat_stream_obj:
# 确保历史消息已初始化
await chat_stream_obj.context_manager.ensure_history_initialized()
await chat_stream_obj.context.ensure_history_initialized()
# 获取所有消息（历史+未读）
all_messages = (
chat_stream_obj.context_manager.context.history_messages +
chat_stream_obj.context_manager.get_unread_messages()
chat_stream_obj.context.history_messages +
chat_stream_obj.context.get_unread_messages()
)
# 转换为字典格式，限制数量
@@ -2071,12 +2071,12 @@ class DefaultReplyer:
if chat_stream_obj:
# 确保历史消息已初始化
await chat_stream_obj.context_manager.ensure_history_initialized()
await chat_stream_obj.context.ensure_history_initialized()
# 获取所有消息（历史+未读）
all_messages = (
chat_stream_obj.context_manager.context.history_messages +
chat_stream_obj.context_manager.get_unread_messages()
chat_stream_obj.context.history_messages +
chat_stream_obj.context.get_unread_messages()
)
# 转换为字典格式，限制数量