Losita/mai-bot
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat：优化记忆查询，添加时间信息

Browse Source
This commit is contained in:
SengokuCola
2025-12-31 14:06:14 +08:00
parent ce9e17df25
commit b296f0683f
5 changed files with 1071 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/chat/heart_flow/heartFC_chat.py
View File

@@ -531,11 +531,15 @@ class HeartFChatting:
quote_message: Optional[bool] = None,
) -> str:
# 根据 llm_quote 配置决定是否使用 quote_message 参数
if global_config.chat.llm_quote and quote_message is not None:
if global_config.chat.llm_quote:
# 如果配置为 true，使用 llm_quote 参数决定是否引用回复
need_reply = quote_message
if need_reply:
logger.info(f"{self.log_prefix} LLM 决定使用引用回复")
if quote_message is None:
logger.warning(f"{self.log_prefix} quote_message 参数为空，不引用")
need_reply = False
else:
need_reply = quote_message
if need_reply:
logger.info(f"{self.log_prefix} LLM 决定使用引用回复")
else:
# 如果配置为 false，使用原来的模式
new_message_count = message_api.count_new_messages(
@@ -663,7 +667,7 @@ class HeartFChatting:
unknown_words = cleaned_uw
# 从 Planner 的 action_data 中提取 quote_message 参数
qm = action_planner_info.action_data.get("quote_message")
qm = action_planner_info.action_data.get("quote")
if qm is not None:
# 支持多种格式：true/false, "true"/"false", 1/0
if isinstance(qm, bool):
@@ -672,6 +676,8 @@ class HeartFChatting:
quote_message = qm.lower() in ("true", "1", "yes")
elif isinstance(qm, (int, float)):
quote_message = bool(qm)
logger.info(f"{self.log_prefix} {qm}引用回复设置: {quote_message}")
success, llm_response = await generator_api.generate_reply(
chat_stream=self.chat_stream,

4
src/chat/planner_actions/planner.py
View File

@@ -531,7 +531,7 @@ class ActionPlanner:
'"question":"需要查询的问题"'
)
if global_config.chat.llm_quote:
reply_action_example += ', "quote_message":"如果需要引用该message，设置为true"'
reply_action_example += ', "quote":"如果需要引用该message，设置为true"'
reply_action_example += "}"
else:
reply_action_example = (
@@ -546,7 +546,7 @@ class ActionPlanner:
'"question":"需要查询的问题"'
)
if global_config.chat.llm_quote:
reply_action_example += ', "quote_message":"如果需要引用该message，设置为true"'
reply_action_example += ', "quote":"如果需要引用该message，设置为true"'
reply_action_example += "}"
planner_prompt_template = await global_prompt_manager.get_prompt_async("planner_prompt")