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

fix(replyer): 移除历史记录的硬编码上限

Browse Source 
修复了在构建聊天记录上下文时，已读历史消息被硬编码截断为50条的问题。

现在，`build_s4u_chat_history_prompts` 及其回退方法将完全遵循 `max_context_size` 配置，确保模型能获取到完整的上下文信息。
This commit is contained in:
tt-P607
2025-11-04 08:20:41 +08:00
parent 6b91b570b9
commit 445b90560a
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -982,7 +982,7 @@ class DefaultReplyer:
# 按时间排序并限制数量 # 按时间排序并限制数量
sorted_messages = sorted(read_messages_dicts, key=lambda x: x.get("time", 0)) sorted_messages = sorted(read_messages_dicts, key=lambda x: x.get("time", 0))
final_history = sorted_messages[-50:] # 限制最多50条 final_history = sorted_messages[-global_config.chat.max_context_size:] # 使用配置的上下文长度
read_content = await build_readable_messages( read_content = await build_readable_messages(
final_history, final_history,
@@ -1087,7 +1087,7 @@ class DefaultReplyer:
read_history_prompt = "" read_history_prompt = ""
if read_messages: if read_messages:
read_content = await build_readable_messages( read_content = await build_readable_messages(
read_messages[-50:], read_messages[-global_config.chat.max_context_size:],
replace_bot_name=True, replace_bot_name=True,
timestamp_mode="normal_no_YMD", timestamp_mode="normal_no_YMD",
truncate=True, truncate=True,
@@ -1293,7 +1293,7 @@ class DefaultReplyer:
# 转换为字典格式 # 转换为字典格式
message_list_before_now_long = [msg.flatten() for msg in all_messages[-(global_config.chat.max_context_size * 2):]] message_list_before_now_long = [msg.flatten() for msg in all_messages[-(global_config.chat.max_context_size * 2):]]
message_list_before_short = [msg.flatten() for msg in all_messages[-int(global_config.chat.max_context_size * 0.33):]] message_list_before_short = [msg.flatten() for msg in all_messages[-int(global_config.chat.max_context_size):]]
logger.debug(f"使用内存中的消息: long={len(message_list_before_now_long)}, short={len(message_list_before_short)}") logger.debug(f"使用内存中的消息: long={len(message_list_before_now_long)}, short={len(message_list_before_short)}")
else: else:
@@ -1307,7 +1307,7 @@ class DefaultReplyer:
message_list_before_short = await get_raw_msg_before_timestamp_with_chat( message_list_before_short = await get_raw_msg_before_timestamp_with_chat(
chat_id=chat_id, chat_id=chat_id,
timestamp=time.time(), timestamp=time.time(),
limit=int(global_config.chat.max_context_size * 0.33), limit=int(global_config.chat.max_context_size),
) )
chat_talking_prompt_short = await build_readable_messages( chat_talking_prompt_short = await build_readable_messages(
@@ -1655,7 +1655,7 @@ class DefaultReplyer:
) )
# 转换为字典格式，限制数量 # 转换为字典格式，限制数量
limit = min(int(global_config.chat.max_context_size * 0.33), 15) limit = int(global_config.chat.max_context_size)
message_list_before_now_half = [msg.flatten() for msg in all_messages[-limit:]] message_list_before_now_half = [msg.flatten() for msg in all_messages[-limit:]]
logger.debug(f"Rewrite使用内存中的 {len(message_list_before_now_half)} 条消息") logger.debug(f"Rewrite使用内存中的 {len(message_list_before_now_half)} 条消息")
@@ -1665,7 +1665,7 @@ class DefaultReplyer:
message_list_before_now_half = await get_raw_msg_before_timestamp_with_chat( message_list_before_now_half = await get_raw_msg_before_timestamp_with_chat(
chat_id=chat_id, chat_id=chat_id,
timestamp=time.time(), timestamp=time.time(),
limit=min(int(global_config.chat.max_context_size * 0.33), 15), limit=int(global_config.chat.max_context_size),
) )
chat_talking_prompt_half = await build_readable_messages( chat_talking_prompt_half = await build_readable_messages(
@@ -2103,7 +2103,7 @@ class DefaultReplyer:
) )
# 转换为字典格式，限制数量 # 转换为字典格式，限制数量
limit = int(global_config.chat.max_context_size * 0.33) limit = int(global_config.chat.max_context_size)
message_list_before_short = [msg.flatten() for msg in all_messages[-limit:]] message_list_before_short = [msg.flatten() for msg in all_messages[-limit:]]
logger.debug(f"记忆存储使用内存中的 {len(message_list_before_short)} 条消息") logger.debug(f"记忆存储使用内存中的 {len(message_list_before_short)} 条消息")
@@ -2113,7 +2113,7 @@ class DefaultReplyer:
message_list_before_short = await get_raw_msg_before_timestamp_with_chat( message_list_before_short = await get_raw_msg_before_timestamp_with_chat(
chat_id=stream.stream_id, chat_id=stream.stream_id,
timestamp=time.time(), timestamp=time.time(),
limit=int(global_config.chat.max_context_size * 0.33), limit=int(global_config.chat.max_context_size),
) )
chat_history = await build_readable_messages( chat_history = await build_readable_messages(
message_list_before_short, message_list_before_short,