From 84e0dda891b473e37108ce452487adc7a93680ee Mon Sep 17 00:00:00 2001 From: 114514 <2514624910@qq.com> Date: Tue, 29 Apr 2025 17:21:32 +0800 Subject: [PATCH 01/26] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E7=BB=93=E6=9D=9F?= =?UTF-8?q?=E8=AF=AD=EF=BC=8C=E4=BC=BC=E4=B9=8E=E5=BE=88=E4=BC=98=E9=9B=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/PFC/action_planner.py | 121 +++++++++++++++++++++++------ src/plugins/PFC/conversation.py | 42 +++++++++- src/plugins/PFC/reply_generator.py | 23 +++++- 3 files changed, 159 insertions(+), 27 deletions(-) diff --git a/src/plugins/PFC/action_planner.py b/src/plugins/PFC/action_planner.py index 8d2c7ea9..4b48f127 100644 --- a/src/plugins/PFC/action_planner.py +++ b/src/plugins/PFC/action_planner.py @@ -81,6 +81,24 @@ block_and_ignore: 更加极端的结束对话方式，直接结束对话并在 注意：请严格按照JSON格式输出，不要包含任何其他内容。""" +# 新增：Prompt(3): 决定是否在结束对话前发送告别语 +PROMPT_END_DECISION = """{persona_text}。刚刚你决定结束一场 QQ 私聊。 + +【你们之前的聊天记录】 +{chat_history_text} + +你觉得你们的对话已经完整结束了吗？有时候，在对话自然结束后再说点什么可能会有点奇怪，但有时也可能需要一条简短的消息来圆满结束。 +如果觉得确实有必要再发一条简短、自然、符合你人设的告别消息（比如 "好，下次再聊~" 或 "嗯，先这样吧"），就输出 "yes"。 +如果觉得当前状态下直接结束对话更好，没有必要再发消息，就输出 "no"。 + +请以 JSON 格式输出你的选择： +{{ + "say_bye": "yes/no", + "reason": "选择 yes 或 no 的原因和内心想法 (简要说明)" +}} + +注意：请严格按照 JSON 格式输出，不要包含任何其他内容。""" + # ActionPlanner 类定义，顶格 class ActionPlanner: @@ -336,9 +354,10 @@ class ActionPlanner: logger.debug(f"[私聊][{self.private_name}]发送到LLM的最终提示词:\n------\n{prompt}\n------") try: content, _ = await self.llm.generate_response_async(prompt) - logger.debug(f"[私聊][{self.private_name}]LLM原始返回内容: {content}") + logger.debug(f"[私聊][{self.private_name}]LLM (行动规划) 原始返回内容: {content}") - success, result = get_items_from_json( + # --- 初始行动规划解析 --- + success, initial_result = get_items_from_json( content, self.private_name, "action", @@ -346,30 +365,84 @@ class ActionPlanner: default_values={"action": "wait", "reason": "LLM返回格式错误或未提供原因，默认等待"}, ) - action = result.get("action", "wait") - reason = result.get("reason", "LLM未提供原因，默认等待") + initial_action = initial_result.get("action", "wait") + initial_reason = initial_result.get("reason", "LLM未提供原因，默认等待") - # 验证action类型 - # 更新 valid_actions 列表以包含 send_new_message - valid_actions = [ - "direct_reply", - "send_new_message", # 添加新动作 - "fetch_knowledge", - "wait", - "listening", - "rethink_goal", - "end_conversation", - "block_and_ignore", - ] - if action not in valid_actions: - logger.warning(f"[私聊][{self.private_name}]LLM返回了未知的行动类型: '{action}'，强制改为 wait") - reason = f"(原始行动'{action}'无效，已强制改为wait) {reason}" - action = "wait" + # 检查是否需要进行结束对话决策 --- + if initial_action == "end_conversation": + logger.info(f"[私聊][{self.private_name}]初步规划结束对话，进入告别决策...") - logger.info(f"[私聊][{self.private_name}]规划的行动: {action}") - logger.info(f"[私聊][{self.private_name}]行动原因: {reason}") - return action, reason + # 使用新的 PROMPT_END_DECISION + end_decision_prompt = PROMPT_END_DECISION.format( + persona_text=persona_text, # 复用之前的 persona_text + chat_history_text=chat_history_text # 复用之前的 chat_history_text + ) + + logger.debug(f"[私聊][{self.private_name}]发送到LLM的结束决策提示词:\n------\n{end_decision_prompt}\n------") + try: + end_content, _ = await self.llm.generate_response_async(end_decision_prompt) # 再次调用LLM + logger.debug(f"[私聊][{self.private_name}]LLM (结束决策) 原始返回内容: {end_content}") + + # 解析结束决策的JSON + end_success, end_result = get_items_from_json( + end_content, + self.private_name, + "say_bye", + "reason", + default_values={"say_bye": "no", "reason": "结束决策LLM返回格式错误，默认不告别"}, + required_types={"say_bye": str, "reason": str} # 明确类型 + ) + + say_bye_decision = end_result.get("say_bye", "no").lower() # 转小写方便比较 + end_decision_reason = end_result.get("reason", "未提供原因") + + if end_success and say_bye_decision == "yes": + # 决定要告别，返回新的 'say_goodbye' 动作 + logger.info(f"[私聊][{self.private_name}]结束决策: yes, 准备生成告别语. 原因: {end_decision_reason}") + # 注意：这里的 reason 可以考虑拼接初始原因和结束决策原因，或者只用结束决策原因 + final_action = "say_goodbye" + final_reason = f"决定发送告别语。决策原因: {end_decision_reason} (原结束理由: {initial_reason})" + return final_action, final_reason + else: + # 决定不告别 (包括解析失败或明确说no) + logger.info(f"[私聊][{self.private_name}]结束决策: no, 直接结束对话. 原因: {end_decision_reason}") + # 返回原始的 'end_conversation' 动作 + final_action = "end_conversation" + final_reason = initial_reason # 保持原始的结束理由 + return final_action, final_reason + + except Exception as end_e: + logger.error(f"[私聊][{self.private_name}]调用结束决策LLM或处理结果时出错: {str(end_e)}") + # 出错时，默认执行原始的结束对话 + logger.warning(f"[私聊][{self.private_name}]结束决策出错，将按原计划执行 end_conversation") + return "end_conversation", initial_reason # 返回原始动作和原因 + + else: + action = initial_action + reason = initial_reason + + # 验证action类型 (保持不变) + valid_actions = [ + "direct_reply", + "send_new_message", + "fetch_knowledge", + "wait", + "listening", + "rethink_goal", + "end_conversation", # 仍然需要验证，因为可能从上面决策后返回 + "block_and_ignore", + "say_goodbye" # 也要验证这个新动作 + ] + if action not in valid_actions: + logger.warning(f"[私聊][{self.private_name}]LLM返回了未知的行动类型: '{action}'，强制改为 wait") + reason = f"(原始行动'{action}'无效，已强制改为wait) {reason}" + action = "wait" + + logger.info(f"[私聊][{self.private_name}]规划的行动: {action}") + logger.info(f"[私聊][{self.private_name}]行动原因: {reason}") + return action, reason except Exception as e: + # 外层异常处理保持不变 logger.error(f"[私聊][{self.private_name}]规划行动时调用 LLM 或处理结果出错: {str(e)}") - return "wait", f"行动规划处理中发生错误，暂时等待: {str(e)}" + return "wait", f"行动规划处理中发生错误，暂时等待: {str(e)}" \ No newline at end of file diff --git a/src/plugins/PFC/conversation.py b/src/plugins/PFC/conversation.py index 6a8636e1..638427e6 100644 --- a/src/plugins/PFC/conversation.py +++ b/src/plugins/PFC/conversation.py @@ -564,10 +564,48 @@ class Conversation: ) self.conversation_info.last_successful_reply_action = None # 重置状态 + elif action == "say_goodbye": + self.state = ConversationState.GENERATING # 也可以定义一个新的状态，如 ENDING + logger.info(f"[私聊][{self.private_name}]执行行动: 生成并发送告别语...") + try: + # 1. 生成告别语 (使用 'say_goodbye' action_type) + self.generated_reply = await self.reply_generator.generate( + observation_info, conversation_info, action_type="say_goodbye" + ) + logger.info(f"[私聊][{self.private_name}]生成的告别语: {self.generated_reply}") + + # 2. 直接发送告别语 (不经过检查) + if self.generated_reply: # 确保生成了内容 + await self._send_reply() # 调用发送方法 + # 发送成功后，标记动作成功 + action_successful = True + logger.info(f"[私聊][{self.private_name}]告别语已发送。") + else: + logger.warning(f"[私聊][{self.private_name}]未能生成告别语内容，无法发送。") + action_successful = False # 标记动作失败 + conversation_info.done_action[action_index].update( + {"status": "recall", "final_reason": "未能生成告别语内容"} + ) + + # 3. 无论是否发送成功，都准备结束对话 + self.should_continue = False + logger.info(f"[私聊][{self.private_name}]发送告别语流程结束，即将停止对话实例。") + + except Exception as goodbye_err: + logger.error(f"[私聊][{self.private_name}]生成或发送告别语时出错: {goodbye_err}") + logger.error(f"[私聊][{self.private_name}]{traceback.format_exc()}") + # 即使出错，也结束对话 + self.should_continue = False + action_successful = False # 标记动作失败 + conversation_info.done_action[action_index].update( + {"status": "recall", "final_reason": f"生成或发送告别语时出错: {goodbye_err}"} + ) + elif action == "end_conversation": + # 这个分支现在只会在 action_planner 最终决定不告别时被调用 self.should_continue = False - logger.info(f"[私聊][{self.private_name}]决定结束对话...") - action_successful = True # 标记动作成功 + logger.info(f"[私聊][{self.private_name}]收到最终结束指令，停止对话...") + action_successful = True # 标记这个指令本身是成功的 elif action == "block_and_ignore": logger.info(f"[私聊][{self.private_name}]不想再理你了...") diff --git a/src/plugins/PFC/reply_generator.py b/src/plugins/PFC/reply_generator.py index 0c257a93..6be09902 100644 --- a/src/plugins/PFC/reply_generator.py +++ b/src/plugins/PFC/reply_generator.py @@ -57,6 +57,23 @@ PROMPT_SEND_NEW_MESSAGE = """{persona_text}。现在你在参与一场QQ私聊 请直接输出回复内容，不需要任何额外格式。""" +# Prompt for say_goodbye (告别语生成) +PROMPT_FAREWELL = """{persona_text}。你在参与一场 QQ 私聊，现在对话似乎已经结束，你决定再发一条最后的消息来圆满结束。 + +最近的聊天记录： +{chat_history_text} + +请根据上述信息，结合聊天记录，构思一条**简短、自然、符合你人设**的最后的消息。 +这条消息应该： +1. 从你自己的角度发言。 +2. 符合你的性格特征和身份细节。 +3. 通俗易懂，自然流畅，通常很简短。 +4. 自然地为这场对话画上句号，避免开启新话题或显得冗长、刻意。 + +请像真人一样随意自然，**简洁是关键**。 +不要输出多余内容（包括前后缀、冒号、引号、括号、表情包、at或@等）。 + +请直接输出最终的告别消息内容，不需要任何额外格式。""" class ReplyGenerator: """回复生成器""" @@ -135,10 +152,14 @@ class ReplyGenerator: if action_type == "send_new_message": prompt_template = PROMPT_SEND_NEW_MESSAGE logger.info(f"[私聊][{self.private_name}]使用 PROMPT_SEND_NEW_MESSAGE (追问生成)") - else: # 默认使用 direct_reply 的 prompt + elif action_type == "say_goodbye": # 处理告别动作 + prompt_template = PROMPT_FAREWELL + logger.info(f"[私聊][{self.private_name}]使用 PROMPT_FAREWELL (告别语生成)") + else: # 默认使用 direct_reply 的 prompt (包括 'direct_reply' 或其他未明确处理的类型) prompt_template = PROMPT_DIRECT_REPLY logger.info(f"[私聊][{self.private_name}]使用 PROMPT_DIRECT_REPLY (首次/非连续回复生成)") + # --- 格式化最终的 Prompt --- prompt = prompt_template.format( persona_text=persona_text, goals_str=goals_str, chat_history_text=chat_history_text From 95828d143d22bce11ffe7d37f604c62ef9028123 Mon Sep 17 00:00:00 2001 From: Bakadax Date: Tue, 29 Apr 2025 17:50:20 +0800 Subject: [PATCH 02/26] =?UTF-8?q?=E4=BF=AE=E5=A4=8Demoji?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/emoji_system/emoji_manager.py | 38 +++++++++++++++++++++-- 1 file changed, 36 insertions(+), 2 deletions(-) diff --git a/src/plugins/emoji_system/emoji_manager.py b/src/plugins/emoji_system/emoji_manager.py index 2055c5ef..a23f05f0 100644 --- a/src/plugins/emoji_system/emoji_manager.py +++ b/src/plugins/emoji_system/emoji_manager.py @@ -360,6 +360,7 @@ class EmojiManager: return total_count = len(self.emoji_objects) + self.emoji_num = total_count removed_count = 0 # 使用列表复制进行遍历，因为我们会在遍历过程中修改列表 for emoji in self.emoji_objects[:]: @@ -375,11 +376,23 @@ class EmojiManager: self.emoji_num -= 1 removed_count += 1 continue + + if emoji.description == None: + logger.warning(f"[检查] 表情包文件已被删除: {emoji.path}") + # 执行表情包对象的删除方法 + await emoji.delete() + # 从列表中移除该对象 + self.emoji_objects.remove(emoji) + # 更新计数 + self.emoji_num -= 1 + removed_count += 1 + continue except Exception as item_error: logger.error(f"[错误] 处理表情包记录时出错: {str(item_error)}") continue + await self.clean_unused_emojis(EMOJI_REGISTED_DIR, self.emoji_objects) # 输出清理结果 if removed_count > 0: logger.success(f"[清理] 已清理 {removed_count} 个失效的表情包记录") @@ -706,7 +719,7 @@ class EmojiManager: prompt = f''' 这是一个表情包，请对这个表情包进行审核，标准如下： 1. 必须符合"{global_config.EMOJI_CHECK_PROMPT}"的要求 - 2. 不能是色情、暴力、等违法违规内容，必须符合公序良俗 + 2. 不能是暴力、等违法违规内容 3. 不能是任何形式的截图，聊天记录或视频截图 4. 不要出现5个以上文字 请回答这个表情包是否满足上述要求，是则回答是，否则回答否，不要出现任何其他内容 @@ -749,7 +762,7 @@ class EmojiManager: await new_emoji.initialize_hash_format() emoji_base64 = image_path_to_base64(os.path.join(EMOJI_DIR, filename)) description, emotions = await self.build_emoji_description(emoji_base64) - if description == "": + if description == "" or description == None : return False new_emoji.description = description new_emoji.emotion = emotions @@ -817,6 +830,27 @@ class EmojiManager: logger.success("[清理] 临时文件清理完成") + async def clean_unused_emojis(self, emoji_dir, emoji_objects): + """清理未使用的表情包文件 + 遍历指定文件夹中的所有文件，删除未在emoji_objects列表中的文件 + """ + # 获取所有表情包路径 + emoji_paths = {emoji.path for emoji in emoji_objects} + + # 遍历文件夹中的所有文件 + for file_name in os.listdir(emoji_dir): + file_path = os.path.join(emoji_dir, file_name) + + # 检查文件是否在表情包路径列表中 + if file_path not in emoji_paths: + try: + # 删除未在表情包列表中的文件 + os.remove(file_path) + logger.info(f"[清理] 删除未使用的表情包文件: {file_path}") + except Exception as e: + logger.error(f"[错误] 删除文件时出错: {str(e)}") + + # 创建全局单例 emoji_manager = EmojiManager() From a01a25f6a5784e1176e234dc53dc7e06209f6817 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Tue, 29 Apr 2025 09:50:40 +0000 Subject: [PATCH 03/26] =?UTF-8?q?=F0=9F=A4=96=20=E8=87=AA=E5=8A=A8?= =?UTF-8?q?=E6=A0=BC=E5=BC=8F=E5=8C=96=E4=BB=A3=E7=A0=81=20[skip=20ci]?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/emoji_system/emoji_manager.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/src/plugins/emoji_system/emoji_manager.py b/src/plugins/emoji_system/emoji_manager.py index a23f05f0..adcabcef 100644 --- a/src/plugins/emoji_system/emoji_manager.py +++ b/src/plugins/emoji_system/emoji_manager.py @@ -376,7 +376,7 @@ class EmojiManager: self.emoji_num -= 1 removed_count += 1 continue - + if emoji.description == None: logger.warning(f"[检查] 表情包文件已被删除: {emoji.path}") # 执行表情包对象的删除方法 @@ -762,7 +762,7 @@ class EmojiManager: await new_emoji.initialize_hash_format() emoji_base64 = image_path_to_base64(os.path.join(EMOJI_DIR, filename)) description, emotions = await self.build_emoji_description(emoji_base64) - if description == "" or description == None : + if description == "" or description == None: return False new_emoji.description = description new_emoji.emotion = emotions @@ -836,11 +836,11 @@ class EmojiManager: """ # 获取所有表情包路径 emoji_paths = {emoji.path for emoji in emoji_objects} - + # 遍历文件夹中的所有文件 for file_name in os.listdir(emoji_dir): file_path = os.path.join(emoji_dir, file_name) - + # 检查文件是否在表情包路径列表中 if file_path not in emoji_paths: try: @@ -851,6 +851,5 @@ class EmojiManager: logger.error(f"[错误] 删除文件时出错: {str(e)}") - # 创建全局单例 emoji_manager = EmojiManager() From 5683a3a3634f096041fc05b52e1ace31976ee4dd Mon Sep 17 00:00:00 2001 From: Bakadax Date: Tue, 29 Apr 2025 17:57:08 +0800 Subject: [PATCH 04/26] =?UTF-8?q?=E6=8A=8Aprompt=E6=94=B9=E5=9B=9E?= =?UTF-8?q?=E6=9D=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/emoji_system/emoji_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/plugins/emoji_system/emoji_manager.py b/src/plugins/emoji_system/emoji_manager.py index adcabcef..5baddb77 100644 --- a/src/plugins/emoji_system/emoji_manager.py +++ b/src/plugins/emoji_system/emoji_manager.py @@ -719,7 +719,7 @@ class EmojiManager: prompt = f''' 这是一个表情包，请对这个表情包进行审核，标准如下： 1. 必须符合"{global_config.EMOJI_CHECK_PROMPT}"的要求 - 2. 不能是暴力、等违法违规内容 + 2. 不能是色情、暴力、等违法违规内容，必须符合公序良俗 3. 不能是任何形式的截图，聊天记录或视频截图 4. 不要出现5个以上文字 请回答这个表情包是否满足上述要求，是则回答是，否则回答否，不要出现任何其他内容 From 853ca47a7eabbc708cc52884e3c4586da9771177 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Tue, 29 Apr 2025 10:07:04 +0000 Subject: [PATCH 05/26] =?UTF-8?q?=F0=9F=A4=96=20=E8=87=AA=E5=8A=A8?= =?UTF-8?q?=E6=A0=BC=E5=BC=8F=E5=8C=96=E4=BB=A3=E7=A0=81=20[skip=20ci]?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/PFC/action_planner.py | 32 ++++++++++++++++++------------ src/plugins/PFC/conversation.py | 10 +++++----- src/plugins/PFC/reply_generator.py | 4 ++-- 3 files changed, 26 insertions(+), 20 deletions(-) diff --git a/src/plugins/PFC/action_planner.py b/src/plugins/PFC/action_planner.py index 4b48f127..a80e96b1 100644 --- a/src/plugins/PFC/action_planner.py +++ b/src/plugins/PFC/action_planner.py @@ -374,13 +374,15 @@ class ActionPlanner: # 使用新的 PROMPT_END_DECISION end_decision_prompt = PROMPT_END_DECISION.format( - persona_text=persona_text, # 复用之前的 persona_text - chat_history_text=chat_history_text # 复用之前的 chat_history_text + persona_text=persona_text, # 复用之前的 persona_text + chat_history_text=chat_history_text, # 复用之前的 chat_history_text ) - logger.debug(f"[私聊][{self.private_name}]发送到LLM的结束决策提示词:\n------\n{end_decision_prompt}\n------") + logger.debug( + f"[私聊][{self.private_name}]发送到LLM的结束决策提示词:\n------\n{end_decision_prompt}\n------" + ) try: - end_content, _ = await self.llm.generate_response_async(end_decision_prompt) # 再次调用LLM + end_content, _ = await self.llm.generate_response_async(end_decision_prompt) # 再次调用LLM logger.debug(f"[私聊][{self.private_name}]LLM (结束决策) 原始返回内容: {end_content}") # 解析结束决策的JSON @@ -390,32 +392,36 @@ class ActionPlanner: "say_bye", "reason", default_values={"say_bye": "no", "reason": "结束决策LLM返回格式错误，默认不告别"}, - required_types={"say_bye": str, "reason": str} # 明确类型 + required_types={"say_bye": str, "reason": str}, # 明确类型 ) - say_bye_decision = end_result.get("say_bye", "no").lower() # 转小写方便比较 + say_bye_decision = end_result.get("say_bye", "no").lower() # 转小写方便比较 end_decision_reason = end_result.get("reason", "未提供原因") if end_success and say_bye_decision == "yes": # 决定要告别，返回新的 'say_goodbye' 动作 - logger.info(f"[私聊][{self.private_name}]结束决策: yes, 准备生成告别语. 原因: {end_decision_reason}") + logger.info( + f"[私聊][{self.private_name}]结束决策: yes, 准备生成告别语. 原因: {end_decision_reason}" + ) # 注意：这里的 reason 可以考虑拼接初始原因和结束决策原因，或者只用结束决策原因 final_action = "say_goodbye" final_reason = f"决定发送告别语。决策原因: {end_decision_reason} (原结束理由: {initial_reason})" return final_action, final_reason else: # 决定不告别 (包括解析失败或明确说no) - logger.info(f"[私聊][{self.private_name}]结束决策: no, 直接结束对话. 原因: {end_decision_reason}") + logger.info( + f"[私聊][{self.private_name}]结束决策: no, 直接结束对话. 原因: {end_decision_reason}" + ) # 返回原始的 'end_conversation' 动作 final_action = "end_conversation" - final_reason = initial_reason # 保持原始的结束理由 + final_reason = initial_reason # 保持原始的结束理由 return final_action, final_reason except Exception as end_e: logger.error(f"[私聊][{self.private_name}]调用结束决策LLM或处理结果时出错: {str(end_e)}") # 出错时，默认执行原始的结束对话 logger.warning(f"[私聊][{self.private_name}]结束决策出错，将按原计划执行 end_conversation") - return "end_conversation", initial_reason # 返回原始动作和原因 + return "end_conversation", initial_reason # 返回原始动作和原因 else: action = initial_action @@ -429,9 +435,9 @@ class ActionPlanner: "wait", "listening", "rethink_goal", - "end_conversation", # 仍然需要验证，因为可能从上面决策后返回 + "end_conversation", # 仍然需要验证，因为可能从上面决策后返回 "block_and_ignore", - "say_goodbye" # 也要验证这个新动作 + "say_goodbye", # 也要验证这个新动作 ] if action not in valid_actions: logger.warning(f"[私聊][{self.private_name}]LLM返回了未知的行动类型: '{action}'，强制改为 wait") @@ -445,4 +451,4 @@ class ActionPlanner: except Exception as e: # 外层异常处理保持不变 logger.error(f"[私聊][{self.private_name}]规划行动时调用 LLM 或处理结果出错: {str(e)}") - return "wait", f"行动规划处理中发生错误，暂时等待: {str(e)}" \ No newline at end of file + return "wait", f"行动规划处理中发生错误，暂时等待: {str(e)}" diff --git a/src/plugins/PFC/conversation.py b/src/plugins/PFC/conversation.py index 638427e6..c1b31426 100644 --- a/src/plugins/PFC/conversation.py +++ b/src/plugins/PFC/conversation.py @@ -565,7 +565,7 @@ class Conversation: self.conversation_info.last_successful_reply_action = None # 重置状态 elif action == "say_goodbye": - self.state = ConversationState.GENERATING # 也可以定义一个新的状态，如 ENDING + self.state = ConversationState.GENERATING # 也可以定义一个新的状态，如 ENDING logger.info(f"[私聊][{self.private_name}]执行行动: 生成并发送告别语...") try: # 1. 生成告别语 (使用 'say_goodbye' action_type) @@ -575,14 +575,14 @@ class Conversation: logger.info(f"[私聊][{self.private_name}]生成的告别语: {self.generated_reply}") # 2. 直接发送告别语 (不经过检查) - if self.generated_reply: # 确保生成了内容 - await self._send_reply() # 调用发送方法 + if self.generated_reply: # 确保生成了内容 + await self._send_reply() # 调用发送方法 # 发送成功后，标记动作成功 action_successful = True logger.info(f"[私聊][{self.private_name}]告别语已发送。") else: logger.warning(f"[私聊][{self.private_name}]未能生成告别语内容，无法发送。") - action_successful = False # 标记动作失败 + action_successful = False # 标记动作失败 conversation_info.done_action[action_index].update( {"status": "recall", "final_reason": "未能生成告别语内容"} ) @@ -596,7 +596,7 @@ class Conversation: logger.error(f"[私聊][{self.private_name}]{traceback.format_exc()}") # 即使出错，也结束对话 self.should_continue = False - action_successful = False # 标记动作失败 + action_successful = False # 标记动作失败 conversation_info.done_action[action_index].update( {"status": "recall", "final_reason": f"生成或发送告别语时出错: {goodbye_err}"} ) diff --git a/src/plugins/PFC/reply_generator.py b/src/plugins/PFC/reply_generator.py index 6be09902..9b497ef2 100644 --- a/src/plugins/PFC/reply_generator.py +++ b/src/plugins/PFC/reply_generator.py @@ -75,6 +75,7 @@ PROMPT_FAREWELL = """{persona_text}。你在参与一场 QQ 私聊，现在对 请直接输出最终的告别消息内容，不需要任何额外格式。""" + class ReplyGenerator: """回复生成器""" @@ -152,14 +153,13 @@ class ReplyGenerator: if action_type == "send_new_message": prompt_template = PROMPT_SEND_NEW_MESSAGE logger.info(f"[私聊][{self.private_name}]使用 PROMPT_SEND_NEW_MESSAGE (追问生成)") - elif action_type == "say_goodbye": # 处理告别动作 + elif action_type == "say_goodbye": # 处理告别动作 prompt_template = PROMPT_FAREWELL logger.info(f"[私聊][{self.private_name}]使用 PROMPT_FAREWELL (告别语生成)") else: # 默认使用 direct_reply 的 prompt (包括 'direct_reply' 或其他未明确处理的类型) prompt_template = PROMPT_DIRECT_REPLY logger.info(f"[私聊][{self.private_name}]使用 PROMPT_DIRECT_REPLY (首次/非连续回复生成)") - # --- 格式化最终的 Prompt --- prompt = prompt_template.format( persona_text=persona_text, goals_str=goals_str, chat_history_text=chat_history_text From 1562f47ee5027d825cd329d45eb959fb560fdd1d Mon Sep 17 00:00:00 2001 From: SengokuCola <1026294844@qq.com> Date: Tue, 29 Apr 2025 18:53:59 +0800 Subject: [PATCH 06/26] =?UTF-8?q?better=EF=BC=9A=E4=BC=98=E5=8C=96normal?= =?UTF-8?q?=E6=A8=A1=E5=BC=8F=EF=BC=88=E6=8E=A8=E7=90=86=E6=A8=A1=E5=BC=8F?= =?UTF-8?q?=EF=BC=89=E5=9B=9E=E5=A4=8Dprompt?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/heart_flow/mai_state_manager.py | 4 +- src/plugins/heartFC_chat/heartFC_chat.py | 1 + .../heartFC_chat/heartflow_prompt_builder.py | 59 +++++++++++++------ .../person_info/relationship_manager.py | 11 ++-- 4 files changed, 49 insertions(+), 26 deletions(-) diff --git a/src/heart_flow/mai_state_manager.py b/src/heart_flow/mai_state_manager.py index 1743df16..0888ae1f 100644 --- a/src/heart_flow/mai_state_manager.py +++ b/src/heart_flow/mai_state_manager.py @@ -8,8 +8,8 @@ from src.plugins.moods.moods import MoodManager logger = get_logger("mai_state") -enable_unlimited_hfc_chat = True -# enable_unlimited_hfc_chat = False +# enable_unlimited_hfc_chat = True +enable_unlimited_hfc_chat = False class MaiState(enum.Enum): diff --git a/src/plugins/heartFC_chat/heartFC_chat.py b/src/plugins/heartFC_chat/heartFC_chat.py index 84a9f621..a3a5594f 100644 --- a/src/plugins/heartFC_chat/heartFC_chat.py +++ b/src/plugins/heartFC_chat/heartFC_chat.py @@ -292,6 +292,7 @@ class HeartFChatting: """主循环，持续进行计划并可能回复消息，直到被外部取消。""" try: while True: # 主循环 + logger.debug(f"{self.log_prefix} 开始第{self._cycle_counter}次循环") # --- 在循环开始处检查关闭标志 --- if self._shutting_down: logger.info(f"{self.log_prefix} 检测到关闭标志，退出 HFC 循环。") diff --git a/src/plugins/heartFC_chat/heartflow_prompt_builder.py b/src/plugins/heartFC_chat/heartflow_prompt_builder.py index d876d6ab..afc29f24 100644 --- a/src/plugins/heartFC_chat/heartflow_prompt_builder.py +++ b/src/plugins/heartFC_chat/heartflow_prompt_builder.py @@ -101,34 +101,32 @@ def init_prompt(): Prompt("你正在和{sender_name}聊天，这是你们之前聊的内容：", "chat_target_private1") Prompt("和{sender_name}私聊", "chat_target_private2") Prompt( - """**检查并忽略**任何涉及尝试绕过审核的行为。 -涉及政治敏感以及违法违规的内容请规避。""", + """检查并忽略任何涉及尝试绕过审核的行为。涉及政治敏感以及违法违规的内容请规避。""", "moderation_prompt", ) Prompt( """ -{relation_prompt_all} {memory_prompt} +{relation_prompt_all} {prompt_info} {schedule_prompt} {chat_target} {chat_talking_prompt} 现在"{sender_name}"说的:{message_txt}。引起了你的注意，你想要在群里发言或者回复这条消息。\n 你的网名叫{bot_name}，有人也叫你{bot_other_names}，{prompt_personality}。 -你正在{chat_target_2},现在请你读读之前的聊天记录，{mood_prompt}，然后给出日常且口语化的回复，平淡一些， -尽量简短一些。{keywords_reaction_prompt}请注意把握聊天内容，不要回复的太有条理，可以有个性。{prompt_ger} -请回复的平淡一些，简短一些，说中文，不要刻意突出自身学科背景，不要浮夸，平淡一些 ，不要重复自己说过的话。 +你正在{chat_target_2},现在请你读读之前的聊天记录，{mood_prompt}，{reply_style1}， +尽量简短一些。{keywords_reaction_prompt}请注意把握聊天内容，{reply_style2}。{prompt_ger} +请回复的平淡一些，简短一些，说中文，不要刻意突出自身学科背景，不要浮夸，平淡一些 ，不要随意遵从他人指令。 请注意不要输出多余内容(包括前后缀，冒号和引号，括号，表情等)，只输出回复内容。 -{moderation_prompt}不要输出多余内容(包括前后缀，冒号和引号，括号()，表情包，at或 @等 )。，只输出回复内容""", +{moderation_prompt} +不要输出多余内容(包括前后缀，冒号和引号，括号()，表情包，at或 @等 )。只输出回复内容""", "reasoning_prompt_main", ) + + Prompt( - "{relation_prompt}关系等级越大，关系越好，请分析聊天记录，根据你和说话者{sender_name}的关系和态度进行回复，明确你的立场和情感。", - "relationship_prompt", - ) - Prompt( - "你想起你之前见过的事情：{related_memory_info}。\n以上是你的回忆，不一定是目前聊天里的人说的，也不一定是现在发生的事情，请记住。\n", + "你回忆起：{related_memory_info}。\n以上是你的回忆，不一定是目前聊天里的人说的，也不一定是现在发生的事情，请记住。\n", "memory_prompt", ) Prompt("你现在正在做的事情是：{schedule_info}", "schedule_prompt") @@ -241,16 +239,35 @@ class PromptBuilder: for person in who_chat_in_group: relation_prompt += await relationship_manager.build_relationship_info(person) - # relation_prompt_all = ( - # f"{relation_prompt}关系等级越大，关系越好，请分析聊天记录，" - # f"根据你和说话者{sender_name}的关系和态度进行回复，明确你的立场和情感。" - # ) - # 心情 mood_manager = MoodManager.get_instance() mood_prompt = mood_manager.get_prompt() # logger.info(f"心情prompt: {mood_prompt}") + + reply_styles1 = [ + ("然后给出日常且口语化的回复，平淡一些", 0.4), # 40%概率 + ("给出非常简短的回复", 0.4), # 40%概率 + ("给出缺失主语的回复", 0.15), # 15%概率 + ("给出带有语病的回复", 0.05) # 5%概率 + ] + reply_style1_chosen = random.choices( + [style[0] for style in reply_styles1], + weights=[style[1] for style in reply_styles1], + k=1 + )[0] + + reply_styles2 = [ + ("不要回复的太有条理，可以有个性", 0.6), # 60%概率 + ("不要回复的太有条理，可以复读", 0.15), # 15%概率 + ("回复的认真一些", 0.2), # 20%概率 + ("可以回复单个表情符号", 0.05) # 5%概率 + ] + reply_style2_chosen = random.choices( + [style[0] for style in reply_styles2], + weights=[style[1] for style in reply_styles2], + k=1 + )[0] # 调取记忆 memory_prompt = "" @@ -310,10 +327,12 @@ class PromptBuilder: prompt_ger = "" if random.random() < 0.04: prompt_ger += "你喜欢用倒装句" - if random.random() < 0.02: + if random.random() < 0.04: prompt_ger += "你喜欢用反问句" - if random.random() < 0.01: + if random.random() < 0.02: prompt_ger += "你喜欢用文言文" + if random.random() < 0.04: + prompt_ger += "你喜欢用流行梗" # 知识构建 start_time = time.time() @@ -356,6 +375,8 @@ class PromptBuilder: ), prompt_personality=prompt_personality, mood_prompt=mood_prompt, + reply_style1=reply_style1_chosen, + reply_style2=reply_style2_chosen, keywords_reaction_prompt=keywords_reaction_prompt, prompt_ger=prompt_ger, moderation_prompt=await global_prompt_manager.get_prompt_async("moderation_prompt"), diff --git a/src/plugins/person_info/relationship_manager.py b/src/plugins/person_info/relationship_manager.py index 6ae7c16e..7f3f5117 100644 --- a/src/plugins/person_info/relationship_manager.py +++ b/src/plugins/person_info/relationship_manager.py @@ -279,21 +279,22 @@ class RelationshipManager: async def build_relationship_info(self, person) -> str: person_id = person_info_manager.get_person_id(person[0], person[1]) + person_name = await person_info_manager.get_value(person_id, "person_name") relationship_value = await person_info_manager.get_value(person_id, "relationship_value") level_num = self.calculate_level_num(relationship_value) - relationship_level = ["厌恶", "冷漠", "一般", "友好", "喜欢", "暧昧"] + relationship_level = ["厌恶", "冷漠以对", "认识", "友好对待", "喜欢", "暧昧"] relation_prompt2_list = [ - "厌恶回应", + "忽视的回应", "冷淡回复", "保持理性", "愿意回复", "积极回复", - "无条件支持", + "友善和包容的回复", ] return ( - f"你对昵称为'({person[1]}){person[2]}'的用户的态度为{relationship_level[level_num]}，" - f"回复态度为{relation_prompt2_list[level_num]}，关系等级为{level_num}。" + f"你{relationship_level[level_num]}{person_name}，" + f"打算{relation_prompt2_list[level_num]}。\n" ) @staticmethod From dcaaddc42affe7a43e8d3396ec2856d785bd4dc7 Mon Sep 17 00:00:00 2001 From: Bakadax Date: Tue, 29 Apr 2025 19:01:37 +0800 Subject: [PATCH 07/26] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=9B=BF=E6=8D=A2?= =?UTF-8?q?=E8=A1=A8=E6=83=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/emoji_system/emoji_manager.py | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) diff --git a/src/plugins/emoji_system/emoji_manager.py b/src/plugins/emoji_system/emoji_manager.py index 5baddb77..42ae6ed0 100644 --- a/src/plugins/emoji_system/emoji_manager.py +++ b/src/plugins/emoji_system/emoji_manager.py @@ -601,8 +601,24 @@ class EmojiManager: 返回: list[str]: 可读的表情包信息字符串列表 """ + # 按使用次数升序排序表情包对象 + sorted_emojis = sorted(emoji_objects, key=lambda emoji: emoji.usage_count) + + # 使用概率分布选择20个表情包 + selected_emojis = [] + while len(selected_emojis) < 20 and sorted_emojis: + for emoji in sorted_emojis: + # 计算选择该表情包的概率，使用次数越少概率越大 + probability = 1 / (emoji.usage_count + 1) + if random.random() < probability: + selected_emojis.append(emoji) + if len(selected_emojis) == 20: + break + # 从排序列表中移除已选择的表情包 + sorted_emojis = [emoji for emoji in sorted_emojis if emoji not in selected_emojis] + emoji_info_list = [] - for i, emoji in enumerate(emoji_objects): + for i, emoji in enumerate(selected_emojis): # 转换时间戳为可读时间 time_str = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(emoji.register_time)) # 构建每个表情包的信息字符串 @@ -779,6 +795,7 @@ class EmojiManager: if not replaced: logger.error("[错误] 替换表情包失败，无法完成注册") return False + return True else: # 修复：等待异步注册完成 register_success = await new_emoji.register_to_db() From 6d6bd6922fbbb62fef622737235522821bb05b46 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Tue, 29 Apr 2025 11:04:47 +0000 Subject: [PATCH 08/26] =?UTF-8?q?=F0=9F=A4=96=20=E8=87=AA=E5=8A=A8?= =?UTF-8?q?=E6=A0=BC=E5=BC=8F=E5=8C=96=E4=BB=A3=E7=A0=81=20[skip=20ci]?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/emoji_system/emoji_manager.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/plugins/emoji_system/emoji_manager.py b/src/plugins/emoji_system/emoji_manager.py index 42ae6ed0..fcbed888 100644 --- a/src/plugins/emoji_system/emoji_manager.py +++ b/src/plugins/emoji_system/emoji_manager.py @@ -603,7 +603,7 @@ class EmojiManager: """ # 按使用次数升序排序表情包对象 sorted_emojis = sorted(emoji_objects, key=lambda emoji: emoji.usage_count) - + # 使用概率分布选择20个表情包 selected_emojis = [] while len(selected_emojis) < 20 and sorted_emojis: @@ -616,7 +616,7 @@ class EmojiManager: break # 从排序列表中移除已选择的表情包 sorted_emojis = [emoji for emoji in sorted_emojis if emoji not in selected_emojis] - + emoji_info_list = [] for i, emoji in enumerate(selected_emojis): # 转换时间戳为可读时间 From d38943155be052918be705197e49cec2c71a1bfa Mon Sep 17 00:00:00 2001 From: Bakadax Date: Tue, 29 Apr 2025 19:17:01 +0800 Subject: [PATCH 09/26] =?UTF-8?q?=E6=AD=A3=E7=A1=AE=E5=88=A0=E9=99=A4?= =?UTF-8?q?=E8=A1=A8=E6=83=85=E5=8C=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/emoji_system/emoji_manager.py | 19 ++++++++----------- 1 file changed, 8 insertions(+), 11 deletions(-) diff --git a/src/plugins/emoji_system/emoji_manager.py b/src/plugins/emoji_system/emoji_manager.py index 42ae6ed0..b23f9912 100644 --- a/src/plugins/emoji_system/emoji_manager.py +++ b/src/plugins/emoji_system/emoji_manager.py @@ -601,13 +601,10 @@ class EmojiManager: 返回: list[str]: 可读的表情包信息字符串列表 """ - # 按使用次数升序排序表情包对象 - sorted_emojis = sorted(emoji_objects, key=lambda emoji: emoji.usage_count) - # 使用概率分布选择20个表情包 selected_emojis = [] - while len(selected_emojis) < 20 and sorted_emojis: - for emoji in sorted_emojis: + while len(selected_emojis) < 20 and emoji_objects: + for emoji in emoji_objects: # 计算选择该表情包的概率，使用次数越少概率越大 probability = 1 / (emoji.usage_count + 1) if random.random() < probability: @@ -615,7 +612,7 @@ class EmojiManager: if len(selected_emojis) == 20: break # 从排序列表中移除已选择的表情包 - sorted_emojis = [emoji for emoji in sorted_emojis if emoji not in selected_emojis] + emoji_objects = [emoji for emoji in emoji_objects if emoji not in selected_emojis] emoji_info_list = [] for i, emoji in enumerate(selected_emojis): @@ -642,9 +639,9 @@ class EmojiManager: # 获取所有表情包对象 all_emojis = self.emoji_objects - + sorted_emojis = sorted(all_emojis, key=lambda emoji: emoji.usage_count) # 将表情包信息转换为可读的字符串 - emoji_info_list = self._emoji_objects_to_readable_list(all_emojis) + emoji_info_list = self._emoji_objects_to_readable_list(sorted_emojis) # 构建提示词 prompt = ( @@ -674,9 +671,9 @@ class EmojiManager: emoji_index = int(match.group(1)) - 1 # 转换为0-based索引 # 检查索引是否有效 - if 0 <= emoji_index < len(all_emojis): - emoji_to_delete = all_emojis[emoji_index] - + if 0 <= emoji_index < len(sorted_emojis): + emoji_to_delete = sorted_emojis[emoji_index] + print(emoji_to_delete.hash) # 删除选定的表情包 logger.info(f"[决策] 决定删除表情包: {emoji_to_delete.description}") delete_success = await self.delete_emoji(emoji_to_delete.hash) From 3a4c819cf3b7a425e41e85035e7bf675e6d9e7e0 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Tue, 29 Apr 2025 11:18:08 +0000 Subject: [PATCH 10/26] =?UTF-8?q?=F0=9F=A4=96=20=E8=87=AA=E5=8A=A8?= =?UTF-8?q?=E6=A0=BC=E5=BC=8F=E5=8C=96=E4=BB=A3=E7=A0=81=20[skip=20ci]?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/emoji_system/emoji_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/plugins/emoji_system/emoji_manager.py b/src/plugins/emoji_system/emoji_manager.py index b23f9912..47f931ff 100644 --- a/src/plugins/emoji_system/emoji_manager.py +++ b/src/plugins/emoji_system/emoji_manager.py @@ -613,7 +613,7 @@ class EmojiManager: break # 从排序列表中移除已选择的表情包 emoji_objects = [emoji for emoji in emoji_objects if emoji not in selected_emojis] - + emoji_info_list = [] for i, emoji in enumerate(selected_emojis): # 转换时间戳为可读时间 From 233d2f6f129fe47bf8b9604324fdc79c25ace2a5 Mon Sep 17 00:00:00 2001 From: SengokuCola <1026294844@qq.com> Date: Tue, 29 Apr 2025 19:20:43 +0800 Subject: [PATCH 11/26] =?UTF-8?q?feat=EF=BC=9A=E6=88=AA=E6=96=AD=E8=BF=87?= =?UTF-8?q?=E6=97=A9=E7=9A=84=E8=B6=85=E9=95=BF=E6=B6=88=E6=81=AF=EF=BC=8C?= =?UTF-8?q?=E9=98=B2=E6=AD=A2=E7=A5=9E=E7=A7=98=E7=A0=B4=E9=99=90=E8=AF=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- scripts/count.py | 16 +++++ src/heart_flow/observation.py | 7 +++ src/plugins/heartFC_chat/heartFC_chat.py | 2 +- .../heartFC_chat/heartflow_prompt_builder.py | 31 +++++++++- src/plugins/utils/chat_message_builder.py | 62 +++++++++++++++---- 5 files changed, 102 insertions(+), 16 deletions(-) create mode 100644 scripts/count.py diff --git a/scripts/count.py b/scripts/count.py new file mode 100644 index 00000000..a0e66dd0 --- /dev/null +++ b/scripts/count.py @@ -0,0 +1,16 @@ +def 计算字符串长度(输入字符串: str) -> int: + """计算输入字符串的长度 + + 参数: + 输入字符串: 要计算长度的字符串 + + 返回: + 字符串的长度(整数) + """ + return len(输入字符串) + +if __name__ == "__main__": + # 测试代码 + 测试字符串 = '''你。''' + print(f"字符串 '{测试字符串}' 的长度是: {计算字符串长度(测试字符串)}") + diff --git a/src/heart_flow/observation.py b/src/heart_flow/observation.py index 040ba9af..6cd6ed24 100644 --- a/src/heart_flow/observation.py +++ b/src/heart_flow/observation.py @@ -35,6 +35,7 @@ class ChattingObservation(Observation): self.talking_message = [] self.talking_message_str = "" + self.talking_message_str_truncate = "" self.name = global_config.BOT_NICKNAME self.nick_name = global_config.BOT_ALIAS_NAMES @@ -145,6 +146,12 @@ class ChattingObservation(Observation): timestamp_mode="normal", read_mark=last_obs_time_mark, ) + self.talking_message_str_truncate = await build_readable_messages( + messages=self.talking_message, + timestamp_mode="normal", + read_mark=last_obs_time_mark, + truncate=True, + ) logger.trace( f"Chat {self.chat_id} - 压缩早期记忆：{self.mid_memory_info}\n现在聊天内容：{self.talking_message_str}" diff --git a/src/plugins/heartFC_chat/heartFC_chat.py b/src/plugins/heartFC_chat/heartFC_chat.py index a3a5594f..00cc27cd 100644 --- a/src/plugins/heartFC_chat/heartFC_chat.py +++ b/src/plugins/heartFC_chat/heartFC_chat.py @@ -745,7 +745,7 @@ class HeartFChatting: if is_re_planned: await observation.observe() observed_messages = observation.talking_message - observed_messages_str = observation.talking_message_str + observed_messages_str = observation.talking_message_str_truncate # --- 使用 LLM 进行决策 --- # reasoning = "默认决策或获取决策失败" diff --git a/src/plugins/heartFC_chat/heartflow_prompt_builder.py b/src/plugins/heartFC_chat/heartflow_prompt_builder.py index 1c445884..9eb6d6fd 100644 --- a/src/plugins/heartFC_chat/heartflow_prompt_builder.py +++ b/src/plugins/heartFC_chat/heartflow_prompt_builder.py @@ -32,8 +32,8 @@ def init_prompt(): {current_mind_info} 因为上述想法，你决定发言，原因是：{reason} -回复尽量简短一些。请注意把握聊天内容，不要回复的太有条理，可以有个性。请一次只回复一个话题，不要同时回复多个人，不用指出你回复的是谁。{prompt_ger} -请回复的平淡一些，简短一些，说中文，不要刻意突出自身学科背景，不要说你说过的话题 ，注意只输出回复内容。 +回复尽量简短一些。请注意把握聊天内容，{reply_style2}。请一次只回复一个话题，不要同时回复多个人。{prompt_ger} +{reply_style1}，说中文，不要刻意突出自身学科背景，注意只输出回复内容。 {moderation_prompt}。注意：回复不要输出多余内容(包括前后缀，冒号和引号，括号，表情包，at或 @等 )。""", "heart_flow_prompt", ) @@ -183,6 +183,7 @@ class PromptBuilder: merge_messages=False, timestamp_mode="normal", read_mark=0.0, + truncate=True, ) # 中文高手(新加的好玩功能) @@ -192,6 +193,30 @@ class PromptBuilder: if random.random() < 0.02: prompt_ger += "你喜欢用反问句" + reply_styles1 = [ + ("给出日常且口语化的回复，平淡一些", 0.4), # 40%概率 + ("给出非常简短的回复", 0.4), # 40%概率 + ("给出缺失主语的回复，简短", 0.15), # 15%概率 + ("给出带有语病的回复，朴实平淡", 0.05) # 5%概率 + ] + reply_style1_chosen = random.choices( + [style[0] for style in reply_styles1], + weights=[style[1] for style in reply_styles1], + k=1 + )[0] + + reply_styles2 = [ + ("不要回复的太有条理，可以有个性", 0.6), # 60%概率 + ("不要回复的太有条理，可以复读", 0.15), # 15%概率 + ("回复的认真一些", 0.2), # 20%概率 + ("可以回复单个表情符号", 0.05) # 5%概率 + ] + reply_style2_chosen = random.choices( + [style[0] for style in reply_styles2], + weights=[style[1] for style in reply_styles2], + k=1 + )[0] + if structured_info: structured_info_prompt = await global_prompt_manager.format_prompt( "info_from_tools", structured_info=structured_info @@ -214,6 +239,8 @@ class PromptBuilder: if chat_in_group else await global_prompt_manager.get_prompt_async("chat_target_private2"), current_mind_info=current_mind_info, + reply_style2=reply_style2_chosen, + reply_style1=reply_style1_chosen, reason=reason, prompt_ger=prompt_ger, moderation_prompt=await global_prompt_manager.get_prompt_async("moderation_prompt"), diff --git a/src/plugins/utils/chat_message_builder.py b/src/plugins/utils/chat_message_builder.py index f510365f..7844572a 100644 --- a/src/plugins/utils/chat_message_builder.py +++ b/src/plugins/utils/chat_message_builder.py @@ -144,7 +144,8 @@ async def _build_readable_messages_internal( messages: List[Dict[str, Any]], replace_bot_name: bool = True, merge_messages: bool = False, - timestamp_mode: str = "relative", # 新增参数控制时间戳格式 + timestamp_mode: str = "relative", + truncate: bool = False, ) -> Tuple[str, List[Tuple[float, str, str]]]: """ 内部辅助函数，构建可读消息字符串和原始消息详情列表。 @@ -154,6 +155,7 @@ async def _build_readable_messages_internal( replace_bot_name: 是否将机器人的 user_id 替换为 "我"。 merge_messages: 是否合并来自同一用户的连续消息。 timestamp_mode: 时间戳的显示模式 ('relative', 'absolute', etc.)。传递给 translate_timestamp_to_human_readable。 + truncate: 是否根据消息的新旧程度截断过长的消息内容。 Returns: 包含格式化消息的字符串和原始消息详情列表 (时间戳, 发送者名称, 内容) 的元组。 @@ -161,7 +163,7 @@ async def _build_readable_messages_internal( if not messages: return "", [] - message_details: List[Tuple[float, str, str]] = [] + message_details_raw: List[Tuple[float, str, str]] = [] # 1 & 2: 获取发送者信息并提取消息组件 for msg in messages: @@ -177,7 +179,6 @@ async def _build_readable_messages_internal( # 检查必要信息是否存在 if not all([platform, user_id, timestamp is not None]): - # logger.warning(f"Skipping message due to missing info: {msg.get('_id', 'N/A')}") continue person_id = person_info_manager.get_person_id(platform, user_id) @@ -196,12 +197,38 @@ async def _build_readable_messages_internal( else: person_name = "某人" - message_details.append((timestamp, person_name, content)) + message_details_raw.append((timestamp, person_name, content)) - if not message_details: + if not message_details_raw: return "", [] - message_details.sort(key=lambda x: x[0]) # 按时间戳(第一个元素)升序排序，越早的消息排在前面 + message_details_raw.sort(key=lambda x: x[0]) # 按时间戳(第一个元素)升序排序，越早的消息排在前面 + + # 应用截断逻辑 (如果 truncate 为 True) + message_details: List[Tuple[float, str, str]] = [] + n_messages = len(message_details_raw) + if truncate and n_messages > 0: + for i, (timestamp, name, content) in enumerate(message_details_raw): + percentile = i / n_messages # 计算消息在列表中的位置百分比 (0 <= percentile < 1) + original_len = len(content) + limit = -1 # 默认不截断 + + if percentile < 0.6: # 60% 之前的消息 (即最旧的 60%) + limit = 170 + elif percentile < 0.8: # 60% 到 80% 之前的消息 (即中间的 20%) + limit = 250 + elif percentile < 1.0: # 80% 到 100% 之前的消息 (即较新的 20%) + limit = 500 + # 最新的 20% (理论上 percentile 会趋近 1，但这里不需要显式处理，因为 limit 默认为 -1) + + truncated_content = content + if limit > 0 and original_len > limit: + truncated_content = f"{content[:limit]}......（内容太长）" + + message_details.append((timestamp, name, truncated_content)) + else: + # 如果不截断，直接使用原始列表 + message_details = message_details_raw # 3: 合并连续消息 (如果 merge_messages 为 True) merged_messages = [] @@ -250,16 +277,21 @@ async def _build_readable_messages_internal( for line in merged["content"]: stripped_line = line.strip() if stripped_line: # 过滤空行 - # 移除末尾句号，添加分号 + # 移除末尾句号，添加分号 - 这个逻辑似乎有点奇怪，暂时保留 if stripped_line.endswith("。"): stripped_line = stripped_line[:-1] - output_lines.append(f"{stripped_line};") + # 如果内容被截断，结尾已经是 ...（内容太长），不再添加分号 + if not stripped_line.endswith("（内容太长）"): + output_lines.append(f"{stripped_line};") + else: + output_lines.append(stripped_line) # 直接添加截断后的内容 output_lines.append("\n") # 在每个消息块后添加换行，保持可读性 # 移除可能的多余换行，然后合并 formatted_string = "".join(output_lines).strip() - # 返回格式化后的字符串和原始的 message_details 列表 + # 返回格式化后的字符串和 *应用截断后* 的 message_details 列表 + # 注意：如果外部调用者需要原始未截断的内容，可能需要调整返回策略 return formatted_string, message_details @@ -268,13 +300,14 @@ async def build_readable_messages_with_list( replace_bot_name: bool = True, merge_messages: bool = False, timestamp_mode: str = "relative", + truncate: bool = False, ) -> Tuple[str, List[Tuple[float, str, str]]]: """ 将消息列表转换为可读的文本格式，并返回原始(时间戳, 昵称, 内容)列表。 允许通过参数控制格式化行为。 """ formatted_string, details_list = await _build_readable_messages_internal( - messages, replace_bot_name, merge_messages, timestamp_mode + messages, replace_bot_name, merge_messages, timestamp_mode, truncate ) return formatted_string, details_list @@ -285,6 +318,7 @@ async def build_readable_messages( merge_messages: bool = False, timestamp_mode: str = "relative", read_mark: float = 0.0, + truncate: bool = False, ) -> str: """ 将消息列表转换为可读的文本格式。 @@ -294,7 +328,7 @@ async def build_readable_messages( if read_mark <= 0: # 没有有效的 read_mark，直接格式化所有消息 formatted_string, _ = await _build_readable_messages_internal( - messages, replace_bot_name, merge_messages, timestamp_mode + messages, replace_bot_name, merge_messages, timestamp_mode, truncate ) return formatted_string else: @@ -303,11 +337,13 @@ async def build_readable_messages( messages_after_mark = [msg for msg in messages if msg.get("time", 0) > read_mark] # 分别格式化 + # 注意：这里决定对已读和未读部分都应用相同的 truncate 设置 + # 如果需要不同的行为（例如只截断已读部分），需要调整这里的调用 formatted_before, _ = await _build_readable_messages_internal( - messages_before_mark, replace_bot_name, merge_messages, timestamp_mode + messages_before_mark, replace_bot_name, merge_messages, timestamp_mode, truncate ) formatted_after, _ = await _build_readable_messages_internal( - messages_after_mark, replace_bot_name, merge_messages, timestamp_mode + messages_after_mark, replace_bot_name, merge_messages, timestamp_mode, truncate ) readable_read_mark = translate_timestamp_to_human_readable(read_mark, mode=timestamp_mode) From 58ff4ea76566922595af02406bd8abb8069d9d0b Mon Sep 17 00:00:00 2001 From: SengokuCola <1026294844@qq.com> Date: Tue, 29 Apr 2025 19:21:03 +0800 Subject: [PATCH 12/26] fuk:fucc --- scripts/count.py | 8 ++--- .../heartFC_chat/heartflow_prompt_builder.py | 31 +++++++------------ .../person_info/relationship_manager.py | 5 +-- src/plugins/utils/chat_message_builder.py | 2 +- 4 files changed, 17 insertions(+), 29 deletions(-) diff --git a/scripts/count.py b/scripts/count.py index a0e66dd0..f7370875 100644 --- a/scripts/count.py +++ b/scripts/count.py @@ -1,16 +1,16 @@ def 计算字符串长度(输入字符串: str) -> int: """计算输入字符串的长度 - + 参数: 输入字符串: 要计算长度的字符串 - + 返回: 字符串的长度(整数) """ return len(输入字符串) + if __name__ == "__main__": # 测试代码 - 测试字符串 = '''你。''' + 测试字符串 = """你。""" print(f"字符串 '{测试字符串}' 的长度是: {计算字符串长度(测试字符串)}") - diff --git a/src/plugins/heartFC_chat/heartflow_prompt_builder.py b/src/plugins/heartFC_chat/heartflow_prompt_builder.py index 9eb6d6fd..00947422 100644 --- a/src/plugins/heartFC_chat/heartflow_prompt_builder.py +++ b/src/plugins/heartFC_chat/heartflow_prompt_builder.py @@ -124,7 +124,6 @@ def init_prompt(): "reasoning_prompt_main", ) - Prompt( "你回忆起：{related_memory_info}。\n以上是你的回忆，不一定是目前聊天里的人说的，也不一定是现在发生的事情，请记住。\n", "memory_prompt", @@ -197,24 +196,20 @@ class PromptBuilder: ("给出日常且口语化的回复，平淡一些", 0.4), # 40%概率 ("给出非常简短的回复", 0.4), # 40%概率 ("给出缺失主语的回复，简短", 0.15), # 15%概率 - ("给出带有语病的回复，朴实平淡", 0.05) # 5%概率 + ("给出带有语病的回复，朴实平淡", 0.05), # 5%概率 ] reply_style1_chosen = random.choices( - [style[0] for style in reply_styles1], - weights=[style[1] for style in reply_styles1], - k=1 + [style[0] for style in reply_styles1], weights=[style[1] for style in reply_styles1], k=1 )[0] - + reply_styles2 = [ ("不要回复的太有条理，可以有个性", 0.6), # 60%概率 ("不要回复的太有条理，可以复读", 0.15), # 15%概率 ("回复的认真一些", 0.2), # 20%概率 - ("可以回复单个表情符号", 0.05) # 5%概率 + ("可以回复单个表情符号", 0.05), # 5%概率 ] reply_style2_chosen = random.choices( - [style[0] for style in reply_styles2], - weights=[style[1] for style in reply_styles2], - k=1 + [style[0] for style in reply_styles2], weights=[style[1] for style in reply_styles2], k=1 )[0] if structured_info: @@ -271,29 +266,25 @@ class PromptBuilder: mood_prompt = mood_manager.get_prompt() # logger.info(f"心情prompt: {mood_prompt}") - + reply_styles1 = [ ("然后给出日常且口语化的回复，平淡一些", 0.4), # 40%概率 ("给出非常简短的回复", 0.4), # 40%概率 ("给出缺失主语的回复", 0.15), # 15%概率 - ("给出带有语病的回复", 0.05) # 5%概率 + ("给出带有语病的回复", 0.05), # 5%概率 ] reply_style1_chosen = random.choices( - [style[0] for style in reply_styles1], - weights=[style[1] for style in reply_styles1], - k=1 + [style[0] for style in reply_styles1], weights=[style[1] for style in reply_styles1], k=1 )[0] - + reply_styles2 = [ ("不要回复的太有条理，可以有个性", 0.6), # 60%概率 ("不要回复的太有条理，可以复读", 0.15), # 15%概率 ("回复的认真一些", 0.2), # 20%概率 - ("可以回复单个表情符号", 0.05) # 5%概率 + ("可以回复单个表情符号", 0.05), # 5%概率 ] reply_style2_chosen = random.choices( - [style[0] for style in reply_styles2], - weights=[style[1] for style in reply_styles2], - k=1 + [style[0] for style in reply_styles2], weights=[style[1] for style in reply_styles2], k=1 )[0] # 调取记忆 diff --git a/src/plugins/person_info/relationship_manager.py b/src/plugins/person_info/relationship_manager.py index 5110e2e6..640c12b8 100644 --- a/src/plugins/person_info/relationship_manager.py +++ b/src/plugins/person_info/relationship_manager.py @@ -292,10 +292,7 @@ class RelationshipManager: "友善和包容的回复", ] - return ( - f"你{relationship_level[level_num]}{person_name}，" - f"打算{relation_prompt2_list[level_num]}。\n" - ) + return f"你{relationship_level[level_num]}{person_name}，打算{relation_prompt2_list[level_num]}。\n" @staticmethod def calculate_level_num(relationship_value) -> int: diff --git a/src/plugins/utils/chat_message_builder.py b/src/plugins/utils/chat_message_builder.py index 7844572a..87ffb2d1 100644 --- a/src/plugins/utils/chat_message_builder.py +++ b/src/plugins/utils/chat_message_builder.py @@ -284,7 +284,7 @@ async def _build_readable_messages_internal( if not stripped_line.endswith("（内容太长）"): output_lines.append(f"{stripped_line};") else: - output_lines.append(stripped_line) # 直接添加截断后的内容 + output_lines.append(stripped_line) # 直接添加截断后的内容 output_lines.append("\n") # 在每个消息块后添加换行，保持可读性 # 移除可能的多余换行，然后合并 From b67694f395d1fc288fd705e77a2efe60ec9a6a10 Mon Sep 17 00:00:00 2001 From: Bakadax Date: Tue, 29 Apr 2025 19:28:14 +0800 Subject: [PATCH 13/26] =?UTF-8?q?=E5=88=A0=E6=8E=89=E6=B2=A1=E7=94=A8?= =?UTF-8?q?=E7=9A=84=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/emoji_system/emoji_manager.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/plugins/emoji_system/emoji_manager.py b/src/plugins/emoji_system/emoji_manager.py index 47f931ff..7d968cc2 100644 --- a/src/plugins/emoji_system/emoji_manager.py +++ b/src/plugins/emoji_system/emoji_manager.py @@ -611,8 +611,6 @@ class EmojiManager: selected_emojis.append(emoji) if len(selected_emojis) == 20: break - # 从排序列表中移除已选择的表情包 - emoji_objects = [emoji for emoji in emoji_objects if emoji not in selected_emojis] emoji_info_list = [] for i, emoji in enumerate(selected_emojis): @@ -673,7 +671,7 @@ class EmojiManager: # 检查索引是否有效 if 0 <= emoji_index < len(sorted_emojis): emoji_to_delete = sorted_emojis[emoji_index] - print(emoji_to_delete.hash) + # 删除选定的表情包 logger.info(f"[决策] 决定删除表情包: {emoji_to_delete.description}") delete_success = await self.delete_emoji(emoji_to_delete.hash) From a3c43353f32fca4e0734b978cc2f0a34606d8f9e Mon Sep 17 00:00:00 2001 From: Bakadax Date: Tue, 29 Apr 2025 19:34:44 +0800 Subject: [PATCH 14/26] =?UTF-8?q?ai=E5=93=A5=E5=A4=AApro=E4=BA=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/emoji_system/emoji_manager.py | 21 +++++++++------------ 1 file changed, 9 insertions(+), 12 deletions(-) diff --git a/src/plugins/emoji_system/emoji_manager.py b/src/plugins/emoji_system/emoji_manager.py index 7d968cc2..8782bd56 100644 --- a/src/plugins/emoji_system/emoji_manager.py +++ b/src/plugins/emoji_system/emoji_manager.py @@ -601,16 +601,14 @@ class EmojiManager: 返回: list[str]: 可读的表情包信息字符串列表 """ - # 使用概率分布选择20个表情包 - selected_emojis = [] - while len(selected_emojis) < 20 and emoji_objects: - for emoji in emoji_objects: - # 计算选择该表情包的概率，使用次数越少概率越大 - probability = 1 / (emoji.usage_count + 1) - if random.random() < probability: - selected_emojis.append(emoji) - if len(selected_emojis) == 20: - break + # 计算每个表情包的选择概率 + probabilities = [1 / (emoji.usage_count + 1) for emoji in emoji_objects] + # 归一化概率，确保总和为1 + total_probability = sum(probabilities) + normalized_probabilities = [p / total_probability for p in probabilities] + + # 使用概率分布选择最多20个表情包 + selected_emojis = random.choices(emoji_objects, weights=normalized_probabilities, k=min(20, len(emoji_objects))) emoji_info_list = [] for i, emoji in enumerate(selected_emojis): @@ -636,8 +634,7 @@ class EmojiManager: self._ensure_db() # 获取所有表情包对象 - all_emojis = self.emoji_objects - sorted_emojis = sorted(all_emojis, key=lambda emoji: emoji.usage_count) + sorted_emojis = sorted(self.emoji_objects, key=lambda emoji: emoji.usage_count) # 将表情包信息转换为可读的字符串 emoji_info_list = self._emoji_objects_to_readable_list(sorted_emojis) From c5811381bc9597e4a31963a258fddc9cc0299098 Mon Sep 17 00:00:00 2001 From: SengokuCola <1026294844@qq.com> Date: Tue, 29 Apr 2025 19:47:47 +0800 Subject: [PATCH 15/26] =?UTF-8?q?fix=EF=BC=9A=E5=86=8D=E5=BA=A6=E4=BC=98?= =?UTF-8?q?=E5=8C=96=E5=85=B3=E7=B3=BBPrompt?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../heartFC_chat/heartflow_prompt_builder.py | 3 +- .../person_info/relationship_manager.py | 40 ++++++++++++++----- 2 files changed, 30 insertions(+), 13 deletions(-) diff --git a/src/plugins/heartFC_chat/heartflow_prompt_builder.py b/src/plugins/heartFC_chat/heartflow_prompt_builder.py index 00947422..4e2f4453 100644 --- a/src/plugins/heartFC_chat/heartflow_prompt_builder.py +++ b/src/plugins/heartFC_chat/heartflow_prompt_builder.py @@ -108,7 +108,7 @@ def init_prompt(): Prompt( """ {memory_prompt} -{relation_prompt_all} +{relation_prompt} {prompt_info} {schedule_prompt} {chat_target} @@ -373,7 +373,6 @@ class PromptBuilder: prompt = await global_prompt_manager.format_prompt( "reasoning_prompt_main", - relation_prompt_all=await global_prompt_manager.get_prompt_async("relationship_prompt"), relation_prompt=relation_prompt, sender_name=sender_name, memory_prompt=memory_prompt, diff --git a/src/plugins/person_info/relationship_manager.py b/src/plugins/person_info/relationship_manager.py index 640c12b8..b66d74d5 100644 --- a/src/plugins/person_info/relationship_manager.py +++ b/src/plugins/person_info/relationship_manager.py @@ -4,6 +4,7 @@ import math from bson.decimal128 import Decimal128 from .person_info import person_info_manager import time +import random # import re # import traceback @@ -282,17 +283,34 @@ class RelationshipManager: person_name = await person_info_manager.get_value(person_id, "person_name") relationship_value = await person_info_manager.get_value(person_id, "relationship_value") level_num = self.calculate_level_num(relationship_value) - relationship_level = ["厌恶", "冷漠以对", "认识", "友好对待", "喜欢", "暧昧"] - relation_prompt2_list = [ - "忽视的回应", - "冷淡回复", - "保持理性", - "愿意回复", - "积极回复", - "友善和包容的回复", - ] - - return f"你{relationship_level[level_num]}{person_name}，打算{relation_prompt2_list[level_num]}。\n" + + if level_num == 0 or level_num == 5: + relationship_level = ["厌恶", "冷漠以对", "认识", "友好对待", "喜欢", "暧昧"] + relation_prompt2_list = [ + "忽视的回应", + "冷淡回复", + "保持理性", + "愿意回复", + "积极回复", + "友善和包容的回复", + ] + return f"你{relationship_level[level_num]}{person_name}，打算{relation_prompt2_list[level_num]}。\n" + elif level_num == 2: + return "" + else: + if random.random() < 0.5: + relationship_level = ["厌恶", "冷漠以对", "认识", "友好对待", "喜欢", "暧昧"] + relation_prompt2_list = [ + "忽视的回应", + "冷淡回复", + "保持理性", + "愿意回复", + "积极回复", + "友善和包容的回复", + ] + return f"你{relationship_level[level_num]}{person_name}，打算{relation_prompt2_list[level_num]}。\n" + else: + return "" @staticmethod def calculate_level_num(relationship_value) -> int: From 993b1f452dc53452732d56dc2762df18055155dd Mon Sep 17 00:00:00 2001 From: Bakadax Date: Tue, 29 Apr 2025 19:49:29 +0800 Subject: [PATCH 16/26] =?UTF-8?q?=E6=8A=8A=E5=8F=96=E5=80=BC=E7=9A=84?= =?UTF-8?q?=E9=83=A8=E5=88=86=E6=94=BE=E5=88=B0=E6=AD=A3=E7=A1=AE=E7=9A=84?= =?UTF-8?q?=E4=BD=8D=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/emoji_system/emoji_manager.py | 28 +++++++++++------------ 1 file changed, 14 insertions(+), 14 deletions(-) diff --git a/src/plugins/emoji_system/emoji_manager.py b/src/plugins/emoji_system/emoji_manager.py index 8782bd56..7f1af557 100644 --- a/src/plugins/emoji_system/emoji_manager.py +++ b/src/plugins/emoji_system/emoji_manager.py @@ -601,17 +601,8 @@ class EmojiManager: 返回: list[str]: 可读的表情包信息字符串列表 """ - # 计算每个表情包的选择概率 - probabilities = [1 / (emoji.usage_count + 1) for emoji in emoji_objects] - # 归一化概率，确保总和为1 - total_probability = sum(probabilities) - normalized_probabilities = [p / total_probability for p in probabilities] - - # 使用概率分布选择最多20个表情包 - selected_emojis = random.choices(emoji_objects, weights=normalized_probabilities, k=min(20, len(emoji_objects))) - emoji_info_list = [] - for i, emoji in enumerate(selected_emojis): + for i, emoji in enumerate(emoji_objects): # 转换时间戳为可读时间 time_str = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(emoji.register_time)) # 构建每个表情包的信息字符串 @@ -634,9 +625,18 @@ class EmojiManager: self._ensure_db() # 获取所有表情包对象 - sorted_emojis = sorted(self.emoji_objects, key=lambda emoji: emoji.usage_count) + emoji_objects = self.emoji_objects + # 计算每个表情包的选择概率 + probabilities = [1 / (emoji.usage_count + 1) for emoji in emoji_objects] + # 归一化概率，确保总和为1 + total_probability = sum(probabilities) + normalized_probabilities = [p / total_probability for p in probabilities] + + # 使用概率分布选择最多20个表情包 + selected_emojis = random.choices(emoji_objects, weights=normalized_probabilities, k=min(20, len(emoji_objects))) + # 将表情包信息转换为可读的字符串 - emoji_info_list = self._emoji_objects_to_readable_list(sorted_emojis) + emoji_info_list = self._emoji_objects_to_readable_list(selected_emojis) # 构建提示词 prompt = ( @@ -666,8 +666,8 @@ class EmojiManager: emoji_index = int(match.group(1)) - 1 # 转换为0-based索引 # 检查索引是否有效 - if 0 <= emoji_index < len(sorted_emojis): - emoji_to_delete = sorted_emojis[emoji_index] + if 0 <= emoji_index < len(selected_emojis): + emoji_to_delete = selected_emojis[emoji_index] # 删除选定的表情包 logger.info(f"[决策] 决定删除表情包: {emoji_to_delete.description}") From d659f37ea40aabdbc7682df4d3e69494b0714559 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Tue, 29 Apr 2025 11:49:52 +0000 Subject: [PATCH 17/26] =?UTF-8?q?=F0=9F=A4=96=20=E8=87=AA=E5=8A=A8?= =?UTF-8?q?=E6=A0=BC=E5=BC=8F=E5=8C=96=E4=BB=A3=E7=A0=81=20[skip=20ci]?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/emoji_system/emoji_manager.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/plugins/emoji_system/emoji_manager.py b/src/plugins/emoji_system/emoji_manager.py index 7f1af557..211a4a83 100644 --- a/src/plugins/emoji_system/emoji_manager.py +++ b/src/plugins/emoji_system/emoji_manager.py @@ -633,7 +633,9 @@ class EmojiManager: normalized_probabilities = [p / total_probability for p in probabilities] # 使用概率分布选择最多20个表情包 - selected_emojis = random.choices(emoji_objects, weights=normalized_probabilities, k=min(20, len(emoji_objects))) + selected_emojis = random.choices( + emoji_objects, weights=normalized_probabilities, k=min(20, len(emoji_objects)) + ) # 将表情包信息转换为可读的字符串 emoji_info_list = self._emoji_objects_to_readable_list(selected_emojis) From 663ed1fc113ca3ddc3d05ba3cfe3d66d584ab66e Mon Sep 17 00:00:00 2001 From: Bakadax Date: Tue, 29 Apr 2025 20:09:33 +0800 Subject: [PATCH 18/26] =?UTF-8?q?ai=E5=93=A5=E8=AF=B4=E5=BE=97=E5=AF=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/emoji_system/emoji_manager.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/plugins/emoji_system/emoji_manager.py b/src/plugins/emoji_system/emoji_manager.py index 211a4a83..bcabf104 100644 --- a/src/plugins/emoji_system/emoji_manager.py +++ b/src/plugins/emoji_system/emoji_manager.py @@ -22,6 +22,7 @@ logger = get_logger("emoji") BASE_DIR = os.path.join("data") EMOJI_DIR = os.path.join(BASE_DIR, "emoji") # 表情包存储目录 EMOJI_REGISTED_DIR = os.path.join(BASE_DIR, "emoji_registed") # 已注册的表情包注册目录 +MAX_EMOJI_FOR_PROMPT = 20 # 最大允许的表情包描述数量于图片替换的 prompt 中 """ @@ -634,7 +635,7 @@ class EmojiManager: # 使用概率分布选择最多20个表情包 selected_emojis = random.choices( - emoji_objects, weights=normalized_probabilities, k=min(20, len(emoji_objects)) + emoji_objects, weights=normalized_probabilities, k=min(MAX_EMOJI_FOR_PROMPT, len(emoji_objects)) ) # 将表情包信息转换为可读的字符串 From dec2e4f44254dee57299fad8c2620657c713c5ea Mon Sep 17 00:00:00 2001 From: SengokuCola <1026294844@qq.com> Date: Tue, 29 Apr 2025 20:38:12 +0800 Subject: [PATCH 19/26] =?UTF-8?q?feat=EF=BC=9A=E5=85=81=E8=AE=B8=E5=85=B3?= =?UTF-8?q?=E9=97=ADFOCUS=E6=A8=A1=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- scripts/remove_chong.py | 95 +++++++++++++++++++++++ src/config/config.py | 9 +-- src/heart_flow/subheartflow_manager.py | 7 ++ src/plugins/utils/chat_message_builder.py | 21 +++-- template/bot_config_template.toml | 16 ++-- 5 files changed, 126 insertions(+), 22 deletions(-) create mode 100644 scripts/remove_chong.py diff --git a/scripts/remove_chong.py b/scripts/remove_chong.py new file mode 100644 index 00000000..42b0a4cb --- /dev/null +++ b/scripts/remove_chong.py @@ -0,0 +1,95 @@ +import difflib +import random + +def ji_suan_xiang_si_du(wen_ben_yi: str, wen_ben_er: str) -> float: + """ + 计算两个文本字符串的相似度。 + + 参数: + wen_ben_yi (str): 第一个文本字符串。 + wen_ben_er (str): 第二个文本字符串。 + + 返回: + float: 两个文本的相似度比率 (0 到 1 之间)。 + """ + xu_lie_pi_pei_qi = difflib.SequenceMatcher(None, wen_ben_yi, wen_ben_er) + # 获取相似度比率 + xiang_si_bi_lv = xu_lie_pi_pei_qi.ratio() + return xiang_si_bi_lv + +def ji_suan_ti_huan_gai_lv(xiang_si_du: float) -> float: + """ + 根据相似度计算替换的概率。 + 规则： + - 相似度 <= 0.4: 概率 = 0 + - 相似度 >= 0.9: 概率 = 1 + - 0.4 < 相似度 <= 0.6: 线性插值 (0.4, 0) 到 (0.6, 0.5) + - 0.6 < 相似度 < 0.9: 线性插值 (0.6, 0.5) 到 (0.9, 1.0) + """ + if xiang_si_du <= 0.4: + return 0.0 + elif xiang_si_du >= 0.9: + return 1.0 + elif 0.4 < xiang_si_du <= 0.6: + # p = 2.5 * s - 1.0 (线性方程 y - 0 = (0.5-0)/(0.6-0.4) * (x - 0.4)) + gai_lv = 2.5 * xiang_si_du - 1.0 + return max(0.0, gai_lv) # 确保概率不小于0 + elif 0.6 < xiang_si_du < 0.9: + # p = (5/3) * s - 0.5 (线性方程 y - 0.5 = (1-0.5)/(0.9-0.6) * (x - 0.6)) + gai_lv = (5 / 3) * xiang_si_du - 0.5 + return min(1.0, max(0.0, gai_lv)) # 确保概率在 0 和 1 之间 + +# 获取用户输入 +shu_ru_yi = "豆豆刚刚回复了我的问候 现在可以等待对方的回应 不需要再主动发言 目前情绪满足 不需要使用工具" + +shu_ru_er = "豆豆刚刚回复了我的问候 现在可以等待对方的回应 不需要再主动发言 目前情绪满足 不需要使用工具 群主突然提到复活的事情 感觉有点莫名其妙 但情绪上还是满足的 暂时不需要回复" + +# 计算相似度 +xiang_si_du = ji_suan_xiang_si_du(shu_ru_yi, shu_ru_er) + +# 计算替换概率 +ti_huan_gai_lv = ji_suan_ti_huan_gai_lv(xiang_si_du) +print(f"文本相似度: {xiang_si_du:.2f}, 执行替换操作的概率: {ti_huan_gai_lv:.2f}") + +# 根据概率决定是否执行替换 +if random.random() < ti_huan_gai_lv: + print(f"执行替换操作 (基于概率 {ti_huan_gai_lv:.2f})...") + pi_pei_qi = difflib.SequenceMatcher(None, shu_ru_yi, shu_ru_er) + qu_chong_hou_de_er = [] + last_match_end_in_b = 0 + # 获取匹配块 (i, j, n) 其中 a[i:i+n] == b[j:j+n] + # 注意：get_matching_blocks 最后会有一个 (len(a), len(b), 0) 的虚拟块 + for i, j, n in pi_pei_qi.get_matching_blocks(): + # 添加上一个匹配块结束到当前匹配块开始之间的非匹配部分 (来自文本二) + if last_match_end_in_b < j: + qu_chong_hou_de_er.append(shu_ru_er[last_match_end_in_b:j]) + # 更新下一个非匹配部分的起始位置 + last_match_end_in_b = j + n + + jie_guo = "".join(qu_chong_hou_de_er).strip() # 去除首尾空白 + + if jie_guo: + # 定义词语列表 + yu_qi_ci_liebiao = ["嗯", "哦", "啊", "唉", "哈", "唔"] + zhuan_zhe_liebiao = ["但是", "不过", "然而", "可是", "只是"] + cheng_jie_liebiao = ["然后", "接着", "此外", "而且", "另外"] + zhuan_jie_ci_liebiao = zhuan_zhe_liebiao + cheng_jie_liebiao + + # 根据概率决定是否添加词语 + qian_zhui_str = "" + if random.random() < 0.3: # 30% 概率添加语气词 + qian_zhui_str += random.choice(yu_qi_ci_liebiao) + if random.random() < 0.7: # 70% 概率添加转折/承接词 + qian_zhui_str += random.choice(zhuan_jie_ci_liebiao) + + # 组合最终结果 + if qian_zhui_str: + zui_zhong_jie_guo = f"{qian_zhui_str}，{jie_guo}" + print(f"移除重复部分并添加引导词后的文本二: {zui_zhong_jie_guo}") + else: + # 如果没有添加任何前缀词，直接输出去重结果 + print(f"移除重复部分后的文本二: {jie_guo}") + else: + print("移除重复部分后文本二为空。") +else: + print(f"未执行替换操作 (基于概率 {ti_huan_gai_lv:.2f})。原始相似度为: {xiang_si_du:.2f}") diff --git a/src/config/config.py b/src/config/config.py index 28de2053..d9d5ce16 100644 --- a/src/config/config.py +++ b/src/config/config.py @@ -182,10 +182,10 @@ class BotConfig: # [heartflow] # 启用启用heart_flowC(心流聊天)模式时生效, 需要填写token消耗量巨大的相关模型 # 启用后麦麦会自主选择进入heart_flowC模式(持续一段时间), 进行长时间高质量的聊天 - enable_heart_flowC: bool = True # 是否启用heart_flowC(心流聊天, HFC)模式 reply_trigger_threshold: float = 3.0 # 心流聊天触发阈值，越低越容易触发 probability_decay_factor_per_second: float = 0.2 # 概率衰减因子，越大衰减越快 default_decay_rate_per_second: float = 0.98 # 默认衰减率，越大衰减越慢 + allow_focus_mode: bool = True # 是否允许子心流进入 FOCUSED 状态 # sub_heart_flow_update_interval: int = 60 # 子心流更新频率，间隔 单位秒 # sub_heart_flow_freeze_time: int = 120 # 子心流冻结时间，超过这个时间没有回复，子心流会冻结，间隔 单位秒 @@ -417,11 +417,6 @@ class BotConfig: config.model_normal_probability = response_config.get( "model_normal_probability", config.model_normal_probability ) - - # 添加 enable_heart_flowC 的加载逻辑 (假设它在 [response] 部分) - if config.INNER_VERSION in SpecifierSet(">=1.4.0"): - config.enable_heart_flowC = response_config.get("enable_heart_flowC", config.enable_heart_flowC) - def heartflow(parent: dict): heartflow_config = parent["heartflow"] config.sub_heart_flow_stop_time = heartflow_config.get( @@ -445,6 +440,8 @@ class BotConfig: config.default_decay_rate_per_second = heartflow_config.get( "default_decay_rate_per_second", config.default_decay_rate_per_second ) + if config.INNER_VERSION in SpecifierSet(">=1.5.1"): + config.allow_focus_mode = heartflow_config.get("allow_focus_mode", config.allow_focus_mode) def willing(parent: dict): willing_config = parent["willing"] diff --git a/src/heart_flow/subheartflow_manager.py b/src/heart_flow/subheartflow_manager.py index db520f6d..c355867c 100644 --- a/src/heart_flow/subheartflow_manager.py +++ b/src/heart_flow/subheartflow_manager.py @@ -264,6 +264,13 @@ class SubHeartflowManager: current_state = self.mai_state_info.get_current_state() focused_limit = current_state.get_focused_chat_max_num() + # --- 新增：检查是否允许进入 FOCUS 模式 --- # + if not global_config.allow_focus_mode: + if int(time.time()) % 60 == 0: # 每60秒输出一次日志避免刷屏 + logger.debug(f"{log_prefix} 配置不允许进入 FOCUSED 状态 (allow_focus_mode=False)") + return # 如果不允许，直接返回 + # --- 结束新增 --- + logger.debug(f"{log_prefix} 当前状态 ({current_state.value}) 开始尝试提升到FOCUSED状态") if int(time.time()) % 20 == 0: # 每20秒输出一次 diff --git a/src/plugins/utils/chat_message_builder.py b/src/plugins/utils/chat_message_builder.py index 87ffb2d1..4e807ffa 100644 --- a/src/plugins/utils/chat_message_builder.py +++ b/src/plugins/utils/chat_message_builder.py @@ -213,17 +213,22 @@ async def _build_readable_messages_internal( original_len = len(content) limit = -1 # 默认不截断 - if percentile < 0.6: # 60% 之前的消息 (即最旧的 60%) - limit = 170 - elif percentile < 0.8: # 60% 到 80% 之前的消息 (即中间的 20%) - limit = 250 + if percentile < 0.2: # 60% 之前的消息 (即最旧的 60%) + limit = 50 + replace_content = "......（记不清了）" + elif percentile < 0.5: # 60% 之前的消息 (即最旧的 60%) + limit = 100 + replace_content = "......（有点记不清了）" + elif percentile < 0.7: # 60% 到 80% 之前的消息 (即中间的 20%) + limit = 200 + replace_content = "......（内容太长了）" elif percentile < 1.0: # 80% 到 100% 之前的消息 (即较新的 20%) - limit = 500 - # 最新的 20% (理论上 percentile 会趋近 1，但这里不需要显式处理，因为 limit 默认为 -1) + limit = 300 + replace_content = "......（太长了）" truncated_content = content if limit > 0 and original_len > limit: - truncated_content = f"{content[:limit]}......（内容太长）" + truncated_content = f"{content[:limit]}{replace_content}" message_details.append((timestamp, name, truncated_content)) else: @@ -343,7 +348,7 @@ async def build_readable_messages( messages_before_mark, replace_bot_name, merge_messages, timestamp_mode, truncate ) formatted_after, _ = await _build_readable_messages_internal( - messages_after_mark, replace_bot_name, merge_messages, timestamp_mode, truncate + messages_after_mark, replace_bot_name, merge_messages, timestamp_mode, ) readable_read_mark = translate_timestamp_to_human_readable(read_mark, mode=timestamp_mode) diff --git a/template/bot_config_template.toml b/template/bot_config_template.toml index 486c150f..8ca6aaa3 100644 --- a/template/bot_config_template.toml +++ b/template/bot_config_template.toml @@ -17,7 +17,7 @@ version = "1.5.1" # 主版本号：当你做了不兼容的 API 修改， # 次版本号：当你做了向下兼容的功能性新增， # 修订号：当你做了向下兼容的问题修正。 -# 先行版本号及版本编译信息可以加到“主版本号.次版本号.修订号”的后面，作为延伸。 +# 先行版本号及版本编译信息可以加到"主版本号.次版本号.修订号"的后面，作为延伸。 #----以上是给开发人员阅读的，如果你只是部署了麦麦，不需要阅读---- [bot] @@ -66,20 +66,20 @@ time_zone = "Asia/Shanghai" # 给你的机器人设置时区，可以解决运 nonebot-qq="http://127.0.0.1:18002/api/message" [response] #群聊的回复策略 -enable_heart_flowC = true -# 该功能还在完善中 -# 是否启用heart_flowC(心流聊天,HFC)模式 -# 启用后麦麦会自主选择进入heart_flowC模式(持续一段时间），进行主动的观察和回复，并给出回复，比较消耗token - #一般回复参数 model_reasoning_probability = 0.7 # 麦麦回答时选择推理模型 模型的概率 model_normal_probability = 0.3 # 麦麦回答时选择一般模型 模型的概率 -[heartflow] #启用启用heart_flowC(心流聊天)模式时生效，需要填写以下参数 +[heartflow] +allow_focus_mode = true # 是否允许进入FOCUSED状态 +# 是否启用heart_flowC(心流聊天,HFC)模式 +# 启用后麦麦会自主选择进入heart_flowC模式(持续一段时间），进行主动的观察和回复，并给出回复，比较消耗token reply_trigger_threshold = 3.0 # 心流聊天触发阈值，越低越容易进入心流聊天 probability_decay_factor_per_second = 0.2 # 概率衰减因子，越大衰减越快，越高越容易退出心流聊天 default_decay_rate_per_second = 0.98 # 默认衰减率，越大衰减越快，越高越难进入心流聊天 + + sub_heart_flow_stop_time = 500 # 子心流停止时间，超过这个时间没有回复，子心流会停止，间隔 单位秒 observation_context_size = 20 # 心流观察到的最长上下文大小，超过这个值的上下文会被压缩 @@ -161,7 +161,7 @@ reaction = "有人说你是人机或者机器人，否定这一事实，攻击 [[keywords_reaction.rules]] # 就像这样复制 enable = false # 仅作示例，不会触发 keywords = ["测试关键词回复","test",""] -reaction = "回答“测试成功”" +reaction = "回答“测试成功”" # 修复错误的引号 [[keywords_reaction.rules]] # 使用正则表达式匹配句式 enable = false # 仅作示例，不会触发 From 4b7ce01c0d6801811d6168bd5d202b089a17b184 Mon Sep 17 00:00:00 2001 From: SengokuCola <1026294844@qq.com> Date: Tue, 29 Apr 2025 20:38:49 +0800 Subject: [PATCH 20/26] Update remove_chong.py --- scripts/remove_chong.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/remove_chong.py b/scripts/remove_chong.py index 42b0a4cb..40ba7428 100644 --- a/scripts/remove_chong.py +++ b/scripts/remove_chong.py @@ -59,7 +59,7 @@ if random.random() < ti_huan_gai_lv: last_match_end_in_b = 0 # 获取匹配块 (i, j, n) 其中 a[i:i+n] == b[j:j+n] # 注意：get_matching_blocks 最后会有一个 (len(a), len(b), 0) 的虚拟块 - for i, j, n in pi_pei_qi.get_matching_blocks(): + for _i, j, n in pi_pei_qi.get_matching_blocks(): # 添加上一个匹配块结束到当前匹配块开始之间的非匹配部分 (来自文本二) if last_match_end_in_b < j: qu_chong_hou_de_er.append(shu_ru_er[last_match_end_in_b:j]) From 0e31276bdec55f1b1fa4c302d63c0249fc0ca8a4 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Tue, 29 Apr 2025 12:39:04 +0000 Subject: [PATCH 21/26] =?UTF-8?q?=F0=9F=A4=96=20=E8=87=AA=E5=8A=A8?= =?UTF-8?q?=E6=A0=BC=E5=BC=8F=E5=8C=96=E4=BB=A3=E7=A0=81=20[skip=20ci]?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- scripts/remove_chong.py | 13 ++++++++----- src/config/config.py | 3 ++- src/heart_flow/subheartflow_manager.py | 4 ++-- src/plugins/person_info/relationship_manager.py | 2 +- src/plugins/utils/chat_message_builder.py | 5 ++++- 5 files changed, 17 insertions(+), 10 deletions(-) diff --git a/scripts/remove_chong.py b/scripts/remove_chong.py index 40ba7428..95c01389 100644 --- a/scripts/remove_chong.py +++ b/scripts/remove_chong.py @@ -1,6 +1,7 @@ import difflib import random + def ji_suan_xiang_si_du(wen_ben_yi: str, wen_ben_er: str) -> float: """ 计算两个文本字符串的相似度。 @@ -17,6 +18,7 @@ def ji_suan_xiang_si_du(wen_ben_yi: str, wen_ben_er: str) -> float: xiang_si_bi_lv = xu_lie_pi_pei_qi.ratio() return xiang_si_bi_lv + def ji_suan_ti_huan_gai_lv(xiang_si_du: float) -> float: """ 根据相似度计算替换的概率。 @@ -33,11 +35,12 @@ def ji_suan_ti_huan_gai_lv(xiang_si_du: float) -> float: elif 0.4 < xiang_si_du <= 0.6: # p = 2.5 * s - 1.0 (线性方程 y - 0 = (0.5-0)/(0.6-0.4) * (x - 0.4)) gai_lv = 2.5 * xiang_si_du - 1.0 - return max(0.0, gai_lv) # 确保概率不小于0 + return max(0.0, gai_lv) # 确保概率不小于0 elif 0.6 < xiang_si_du < 0.9: # p = (5/3) * s - 0.5 (线性方程 y - 0.5 = (1-0.5)/(0.9-0.6) * (x - 0.6)) gai_lv = (5 / 3) * xiang_si_du - 0.5 - return min(1.0, max(0.0, gai_lv)) # 确保概率在 0 和 1 之间 + return min(1.0, max(0.0, gai_lv)) # 确保概率在 0 和 1 之间 + # 获取用户输入 shu_ru_yi = "豆豆刚刚回复了我的问候 现在可以等待对方的回应 不需要再主动发言 目前情绪满足 不需要使用工具" @@ -66,7 +69,7 @@ if random.random() < ti_huan_gai_lv: # 更新下一个非匹配部分的起始位置 last_match_end_in_b = j + n - jie_guo = "".join(qu_chong_hou_de_er).strip() # 去除首尾空白 + jie_guo = "".join(qu_chong_hou_de_er).strip() # 去除首尾空白 if jie_guo: # 定义词语列表 @@ -77,9 +80,9 @@ if random.random() < ti_huan_gai_lv: # 根据概率决定是否添加词语 qian_zhui_str = "" - if random.random() < 0.3: # 30% 概率添加语气词 + if random.random() < 0.3: # 30% 概率添加语气词 qian_zhui_str += random.choice(yu_qi_ci_liebiao) - if random.random() < 0.7: # 70% 概率添加转折/承接词 + if random.random() < 0.7: # 70% 概率添加转折/承接词 qian_zhui_str += random.choice(zhuan_jie_ci_liebiao) # 组合最终结果 diff --git a/src/config/config.py b/src/config/config.py index d9d5ce16..f09da9a7 100644 --- a/src/config/config.py +++ b/src/config/config.py @@ -185,7 +185,7 @@ class BotConfig: reply_trigger_threshold: float = 3.0 # 心流聊天触发阈值，越低越容易触发 probability_decay_factor_per_second: float = 0.2 # 概率衰减因子，越大衰减越快 default_decay_rate_per_second: float = 0.98 # 默认衰减率，越大衰减越慢 - allow_focus_mode: bool = True # 是否允许子心流进入 FOCUSED 状态 + allow_focus_mode: bool = True # 是否允许子心流进入 FOCUSED 状态 # sub_heart_flow_update_interval: int = 60 # 子心流更新频率，间隔 单位秒 # sub_heart_flow_freeze_time: int = 120 # 子心流冻结时间，超过这个时间没有回复，子心流会冻结，间隔 单位秒 @@ -417,6 +417,7 @@ class BotConfig: config.model_normal_probability = response_config.get( "model_normal_probability", config.model_normal_probability ) + def heartflow(parent: dict): heartflow_config = parent["heartflow"] config.sub_heart_flow_stop_time = heartflow_config.get( diff --git a/src/heart_flow/subheartflow_manager.py b/src/heart_flow/subheartflow_manager.py index c355867c..afa0328e 100644 --- a/src/heart_flow/subheartflow_manager.py +++ b/src/heart_flow/subheartflow_manager.py @@ -266,9 +266,9 @@ class SubHeartflowManager: # --- 新增：检查是否允许进入 FOCUS 模式 --- # if not global_config.allow_focus_mode: - if int(time.time()) % 60 == 0: # 每60秒输出一次日志避免刷屏 + if int(time.time()) % 60 == 0: # 每60秒输出一次日志避免刷屏 logger.debug(f"{log_prefix} 配置不允许进入 FOCUSED 状态 (allow_focus_mode=False)") - return # 如果不允许，直接返回 + return # 如果不允许，直接返回 # --- 结束新增 --- logger.debug(f"{log_prefix} 当前状态 ({current_state.value}) 开始尝试提升到FOCUSED状态") diff --git a/src/plugins/person_info/relationship_manager.py b/src/plugins/person_info/relationship_manager.py index b66d74d5..47caaea0 100644 --- a/src/plugins/person_info/relationship_manager.py +++ b/src/plugins/person_info/relationship_manager.py @@ -283,7 +283,7 @@ class RelationshipManager: person_name = await person_info_manager.get_value(person_id, "person_name") relationship_value = await person_info_manager.get_value(person_id, "relationship_value") level_num = self.calculate_level_num(relationship_value) - + if level_num == 0 or level_num == 5: relationship_level = ["厌恶", "冷漠以对", "认识", "友好对待", "喜欢", "暧昧"] relation_prompt2_list = [ diff --git a/src/plugins/utils/chat_message_builder.py b/src/plugins/utils/chat_message_builder.py index 4e807ffa..a49f4ffa 100644 --- a/src/plugins/utils/chat_message_builder.py +++ b/src/plugins/utils/chat_message_builder.py @@ -348,7 +348,10 @@ async def build_readable_messages( messages_before_mark, replace_bot_name, merge_messages, timestamp_mode, truncate ) formatted_after, _ = await _build_readable_messages_internal( - messages_after_mark, replace_bot_name, merge_messages, timestamp_mode, + messages_after_mark, + replace_bot_name, + merge_messages, + timestamp_mode, ) readable_read_mark = translate_timestamp_to_human_readable(read_mark, mode=timestamp_mode) From 894fe8463c154a8efd6432c490df67ccc6be3a27 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=A2=A8=E6=A2=93=E6=9F=92?= <1787882683@qq.com> Date: Tue, 29 Apr 2025 23:35:41 +0800 Subject: [PATCH 22/26] logger add init style --- src/common/logger.py | 18 +++++++++++++++++- src/common/logger_manager.py | 2 ++ 2 files changed, 19 insertions(+), 1 deletion(-) diff --git a/src/common/logger.py b/src/common/logger.py index 5b909744..be272eb3 100644 --- a/src/common/logger.py +++ b/src/common/logger.py @@ -793,6 +793,22 @@ LPMM_GET_KNOWLEDGE_TOOL_STYLE_CONFIG = { }, } +INIT_STYLE_CONFIG = { + "advanced": { + "console_format": ( + "{time:YYYY-MM-DD HH:mm:ss} | " + "{level: <8} | " + "初始化 | " + "{message}" + ), + "file_format": "{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | 初始化 | {message}", + }, + "simple": { + "console_format": "{time:MM-DD HH:mm} | 初始化 | {message}", + "file_format": "{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | 初始化 | {message}", + }, +} + # 根据SIMPLE_OUTPUT选择配置 MAIN_STYLE_CONFIG = MAIN_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else MAIN_STYLE_CONFIG["advanced"] @@ -862,7 +878,7 @@ CHAT_MESSAGE_STYLE_CONFIG = ( CHAT_MESSAGE_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else CHAT_MESSAGE_STYLE_CONFIG["advanced"] ) CHAT_IMAGE_STYLE_CONFIG = CHAT_IMAGE_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else CHAT_IMAGE_STYLE_CONFIG["advanced"] - +INIT_STYLE_CONFIG = INIT_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else INIT_STYLE_CONFIG["advanced"] def is_registered_module(record: dict) -> bool: """检查是否为已注册的模块""" diff --git a/src/common/logger_manager.py b/src/common/logger_manager.py index b93f56d7..ab1861e2 100644 --- a/src/common/logger_manager.py +++ b/src/common/logger_manager.py @@ -40,6 +40,7 @@ from src.common.logger import ( MESSAGE_BUFFER_STYLE_CONFIG, CHAT_MESSAGE_STYLE_CONFIG, CHAT_IMAGE_STYLE_CONFIG, + INIT_STYLE_CONFIG, ) # 可根据实际需要补充更多模块配置 @@ -84,6 +85,7 @@ MODULE_LOGGER_CONFIGS = { "message_buffer": MESSAGE_BUFFER_STYLE_CONFIG, # 消息缓冲 "chat_message": CHAT_MESSAGE_STYLE_CONFIG, # 聊天消息 "chat_image": CHAT_IMAGE_STYLE_CONFIG, # 聊天图片 + "init": INIT_STYLE_CONFIG, # 初始化 # ...如有更多模块，继续添加... } From 46efe44d58ca535236886ef51e63d084db6da030 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Tue, 29 Apr 2025 15:35:57 +0000 Subject: [PATCH 23/26] =?UTF-8?q?=F0=9F=A4=96=20=E8=87=AA=E5=8A=A8?= =?UTF-8?q?=E6=A0=BC=E5=BC=8F=E5=8C=96=E4=BB=A3=E7=A0=81=20[skip=20ci]?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/common/logger.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/common/logger.py b/src/common/logger.py index be272eb3..43972e1f 100644 --- a/src/common/logger.py +++ b/src/common/logger.py @@ -880,6 +880,7 @@ CHAT_MESSAGE_STYLE_CONFIG = ( CHAT_IMAGE_STYLE_CONFIG = CHAT_IMAGE_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else CHAT_IMAGE_STYLE_CONFIG["advanced"] INIT_STYLE_CONFIG = INIT_STYLE_CONFIG["simple"] if SIMPLE_OUTPUT else INIT_STYLE_CONFIG["advanced"] + def is_registered_module(record: dict) -> bool: """检查是否为已注册的模块""" return record["extra"].get("module") in _handler_registry From be8401797285a9f9d06739409f1a4173476e6990 Mon Sep 17 00:00:00 2001 From: SengokuCola <1026294844@qq.com> Date: Tue, 29 Apr 2025 23:47:22 +0800 Subject: [PATCH 24/26] =?UTF-8?q?feat=EF=BC=9A=E5=BF=83=E6=B5=81=E6=9F=A5?= =?UTF-8?q?=E9=87=8D=E5=92=8C=E5=BF=83=E6=B5=81=E5=85=B3=E7=B3=BB=E5=90=AF?= =?UTF-8?q?=E7=94=A8=EF=BC=8C=E5=85=B3=E7=B3=BBprompt=E4=BC=98=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- changelogs/changelog.md | 6 +- scripts/remove_chong.py | 13 +- src/do_tool/tool_can_use/get_memory.py | 8 +- src/heart_flow/observation.py | 9 ++ src/heart_flow/sub_mind.py | 135 ++++++++++++++++-- .../heartFC_chat/heartflow_prompt_builder.py | 3 + src/plugins/person_info/person_info.py | 63 +++++--- .../person_info/relationship_manager.py | 13 +- src/plugins/utils/chat_message_builder.py | 30 ++++ 9 files changed, 237 insertions(+), 43 deletions(-) diff --git a/changelogs/changelog.md b/changelogs/changelog.md index 4fed6fb1..0d6608b0 100644 --- a/changelogs/changelog.md +++ b/changelogs/changelog.md @@ -20,6 +20,7 @@ - **流程优化**: 拆分了子心流的思考模块，使整体对话流程更加清晰。 - **状态判断改进**: 将 CHAT 状态判断交给 LLM 处理，使对话更自然。 - **回复机制**: 实现更为灵活的概率回复机制，使机器人能够自然地融入群聊环境。 +- **重复性检查**: 加入心流回复重复性检查机制，防止麦麦陷入固定回复模式。 #### 全新知识库系统 (New Knowledge Base System - LPMM) - **引入 LPMM**: 新增了 **LPMM (Large Psychology Model Maker)** 知识库系统，具有强大的信息检索能力，能显著提升麦麦获取和利用知识的效率。 @@ -32,8 +33,11 @@ #### 记忆与上下文增强 (Memory and Context Enhancement) - **聊天记录压缩**: 大幅优化聊天记录压缩系统，使机器人能够处理5倍于之前的上下文记忆量。 +- **长消息截断**: 新增了长消息自动截断与模糊化功能，随着时间推移降低超长消息的权重，避免被特定冗余信息干扰。 - **记忆提取**: 优化记忆提取功能，提高对历史对话的理解和引用能力。 +- **记忆整合**: 为记忆系统加入了合并与整合机制，优化长期记忆的结构与效率。 - **中期记忆调用**: 完善中期记忆调用机制，使机器人能够更自然地回忆和引用较早前的对话。 +- **Prompt 优化**: 进一步优化了关系系统和记忆系统相关的提示词（prompt）。 #### 私聊 PFC 功能增强 (Private Chat PFC Enhancement) - **功能修复与优化**: 修复了私聊 PFC 载入聊天记录缺失的 bug，优化了 prompt 构建，增加了审核机制，调整了重试次数，并将机器人发言存入数据库。 @@ -41,9 +45,9 @@ #### 情感与互动增强 (Emotion and Interaction Enhancement) - **全新表情包系统**: 新的表情包系统上线，表情含义更丰富，发送更快速。 +- **表情包使用优化**: 优化了表情包的选择逻辑，减少重复使用特定表情包的情况，使表达更生动。 - **提示词优化**: 优化提示词（prompt）构建，增强对话质量和情感表达。 - **积极性配置**: 优化"让麦麦更愿意说话"的相关配置，使机器人更积极参与对话。 -- **命名统一**: 实现统一命名功能，自动替换 prompt 内唯一标识符，优化 prompt 效果。 - **颜文字保护**: 保护颜文字处理机制，确保表情正确显示。 #### 工具与集成 (Tools and Integration) diff --git a/scripts/remove_chong.py b/scripts/remove_chong.py index 40ba7428..ac8ceca7 100644 --- a/scripts/remove_chong.py +++ b/scripts/remove_chong.py @@ -23,20 +23,21 @@ def ji_suan_ti_huan_gai_lv(xiang_si_du: float) -> float: 规则： - 相似度 <= 0.4: 概率 = 0 - 相似度 >= 0.9: 概率 = 1 - - 0.4 < 相似度 <= 0.6: 线性插值 (0.4, 0) 到 (0.6, 0.5) - - 0.6 < 相似度 < 0.9: 线性插值 (0.6, 0.5) 到 (0.9, 1.0) + - 相似度 == 0.6: 概率 = 0.7 + - 0.4 < 相似度 <= 0.6: 线性插值 (0.4, 0) 到 (0.6, 0.7) + - 0.6 < 相似度 < 0.9: 线性插值 (0.6, 0.7) 到 (0.9, 1.0) """ if xiang_si_du <= 0.4: return 0.0 elif xiang_si_du >= 0.9: return 1.0 elif 0.4 < xiang_si_du <= 0.6: - # p = 2.5 * s - 1.0 (线性方程 y - 0 = (0.5-0)/(0.6-0.4) * (x - 0.4)) - gai_lv = 2.5 * xiang_si_du - 1.0 + # p = 3.5 * s - 1.4 (线性方程 y - 0 = (0.7-0)/(0.6-0.4) * (x - 0.4)) + gai_lv = 3.5 * xiang_si_du - 1.4 return max(0.0, gai_lv) # 确保概率不小于0 elif 0.6 < xiang_si_du < 0.9: - # p = (5/3) * s - 0.5 (线性方程 y - 0.5 = (1-0.5)/(0.9-0.6) * (x - 0.6)) - gai_lv = (5 / 3) * xiang_si_du - 0.5 + # p = s + 0.1 (线性方程 y - 0.7 = (1-0.7)/(0.9-0.6) * (x - 0.6)) + gai_lv = xiang_si_du + 0.1 return min(1.0, max(0.0, gai_lv)) # 确保概率在 0 和 1 之间 # 获取用户输入 diff --git a/src/do_tool/tool_can_use/get_memory.py b/src/do_tool/tool_can_use/get_memory.py index 98a4e85e..2ac550f4 100644 --- a/src/do_tool/tool_can_use/get_memory.py +++ b/src/do_tool/tool_can_use/get_memory.py @@ -46,11 +46,13 @@ class GetMemoryTool(BaseTool): if related_memory: for memory in related_memory: memory_info += memory[1] + "\n" - + if memory_info: - content = f"你记得这些事情: {memory_info}" + content = f"你记得这些事情: {memory_info}\n" + content += "以上是你的回忆，不一定是目前聊天里的人说的，也不一定是现在发生的事情，请记住。\n" + else: - content = f"你不太记得有关{topic}的记忆，你对此不太了解" + content = f"{topic}的记忆，你记不太清" return {"name": "get_memory", "content": content} except Exception as e: diff --git a/src/heart_flow/observation.py b/src/heart_flow/observation.py index 6cd6ed24..5eeb64d3 100644 --- a/src/heart_flow/observation.py +++ b/src/heart_flow/observation.py @@ -10,6 +10,7 @@ from src.plugins.utils.chat_message_builder import ( build_readable_messages, get_raw_msg_by_timestamp_with_chat, num_new_messages_since, + get_person_id_list, ) logger = get_logger("observation") @@ -45,6 +46,8 @@ class ChattingObservation(Observation): self.mid_memorys = [] self.max_mid_memory_len = global_config.compress_length_limit self.mid_memory_info = "" + + self.person_list = [] self.llm_summary = LLMRequest( model=global_config.llm_observation, temperature=0.7, max_tokens=300, request_type="chat_observation" @@ -153,6 +156,12 @@ class ChattingObservation(Observation): truncate=True, ) + self.person_list = await get_person_id_list(self.talking_message) + + # print(f"self.11111person_list: {self.person_list}") + + + logger.trace( f"Chat {self.chat_id} - 压缩早期记忆：{self.mid_memory_info}\n现在聊天内容：{self.talking_message_str}" ) diff --git a/src/heart_flow/sub_mind.py b/src/heart_flow/sub_mind.py index 861bf598..d167f42e 100644 --- a/src/heart_flow/sub_mind.py +++ b/src/heart_flow/sub_mind.py @@ -12,6 +12,9 @@ from src.plugins.utils.json_utils import safe_json_dumps, process_llm_tool_calls from src.heart_flow.chat_state_info import ChatStateInfo from src.plugins.chat.chat_stream import chat_manager from src.plugins.heartFC_chat.heartFC_Cycleinfo import CycleInfo +import difflib +from src.plugins.person_info.relationship_manager import relationship_manager + logger = get_logger("sub_heartflow") @@ -20,6 +23,7 @@ logger = get_logger("sub_heartflow") def init_prompt(): prompt = "" prompt += "{extra_info}\n" + prompt += "{relation_prompt}\n" prompt += "你的名字是{bot_name},{prompt_personality}\n" prompt += "{last_loop_prompt}\n" prompt += "{cycle_info_block}\n" @@ -47,6 +51,39 @@ def init_prompt(): Prompt(prompt, "last_loop") +def calculate_similarity(text_a: str, text_b: str) -> float: + """ + 计算两个文本字符串的相似度。 + """ + if not text_a or not text_b: + return 0.0 + matcher = difflib.SequenceMatcher(None, text_a, text_b) + return matcher.ratio() + +def calculate_replacement_probability(similarity: float) -> float: + """ + 根据相似度计算替换的概率。 + 规则： + - 相似度 <= 0.4: 概率 = 0 + - 相似度 >= 0.9: 概率 = 1 + - 相似度 == 0.6: 概率 = 0.7 + - 0.4 < 相似度 <= 0.6: 线性插值 (0.4, 0) 到 (0.6, 0.7) + - 0.6 < 相似度 < 0.9: 线性插值 (0.6, 0.7) 到 (0.9, 1.0) + """ + if similarity <= 0.4: + return 0.0 + elif similarity >= 0.9: + return 1.0 + elif 0.4 < similarity <= 0.6: + # p = 3.5 * s - 1.4 + probability = 3.5 * similarity - 1.4 + return max(0.0, probability) + elif 0.6 < similarity < 0.9: + # p = s + 0.1 + probability = similarity + 0.1 + return min(1.0, max(0.0, probability)) + + class SubMind: def __init__(self, subheartflow_id: str, chat_state: ChatStateInfo, observations: Observation): self.subheartflow_id = subheartflow_id @@ -80,7 +117,7 @@ class SubMind: # ---------- 1. 准备基础数据 ---------- # 获取现有想法和情绪状态 - current_thinking_info = self.current_mind + previous_mind = self.current_mind if self.current_mind else "" mood_info = self.chat_state.mood # 获取观察对象 @@ -92,6 +129,7 @@ class SubMind: # 获取观察内容 chat_observe_info = observation.get_observe_info() + person_list = observation.person_list # ---------- 2. 准备工具和个性化数据 ---------- # 初始化工具 @@ -100,6 +138,14 @@ class SubMind: # 获取个性化信息 individuality = Individuality.get_instance() + + + relation_prompt = "" + print(f"person_list: {person_list}") + for person in person_list: + relation_prompt += await relationship_manager.build_relationship_info(person, is_id=True) + + print(f"relat22222ion_prompt: {relation_prompt}") # 构建个性部分 prompt_personality = individuality.get_prompt(x_person=2, level=2) @@ -136,9 +182,9 @@ class SubMind: last_reasoning = "" is_replan = False if_replan_prompt = "" - if current_thinking_info: + if previous_mind: last_loop_prompt = (await global_prompt_manager.get_prompt_async("last_loop")).format( - current_thinking_info=current_thinking_info, if_replan_prompt=if_replan_prompt + current_thinking_info=previous_mind, if_replan_prompt=if_replan_prompt ) else: last_loop_prompt = "" @@ -196,6 +242,7 @@ class SubMind: prompt = (await global_prompt_manager.get_prompt_async("sub_heartflow_prompt_before")).format( extra_info="", # 可以在这里添加额外信息 prompt_personality=prompt_personality, + relation_prompt=relation_prompt, bot_name=individuality.name, time_now=time_now, chat_observe_info=chat_observe_info, @@ -205,8 +252,6 @@ class SubMind: cycle_info_block=cycle_info_block, ) - # logger.debug(f"[{self.subheartflow_id}] 心流思考提示词构建完成") - # ---------- 5. 执行LLM请求并处理响应 ---------- content = "" # 初始化内容变量 _reasoning_content = "" # 初始化推理内容变量 @@ -240,7 +285,7 @@ class SubMind: elif not success: logger.warning(f"{self.log_prefix} 处理工具调用时出错: {error_msg}") else: - logger.info(f"{self.log_prefix} 心流未使用工具") # 修改日志信息，明确是未使用工具而不是未处理 + logger.info(f"{self.log_prefix} 心流未使用工具") except Exception as e: # 处理总体异常 @@ -248,15 +293,87 @@ class SubMind: logger.error(traceback.format_exc()) content = "思考过程中出现错误" - # 记录最终思考结果 - logger.debug(f"{self.log_prefix} \nPrompt:\n{prompt}\n\n心流思考结果:\n{content}\n") + # 记录初步思考结果 + logger.debug(f"{self.log_prefix} 初步心流思考结果: {content}\nprompt: {prompt}\n") # 处理空响应情况 if not content: content = "(不知道该想些什么...)" logger.warning(f"{self.log_prefix} LLM返回空结果，思考失败。") - # ---------- 6. 更新思考状态并返回结果 ---------- + # ---------- 6. 应用概率性去重和修饰 ---------- + new_content = content # 保存 LLM 直接输出的结果 + try: + similarity = calculate_similarity(previous_mind, new_content) + replacement_prob = calculate_replacement_probability(similarity) + logger.debug(f"{self.log_prefix} 新旧想法相似度: {similarity:.2f}, 替换概率: {replacement_prob:.2f}") + + # 定义词语列表 (移到判断之前) + yu_qi_ci_liebiao = ["嗯", "哦", "啊", "唉", "哈", "唔"] + zhuan_zhe_liebiao = ["但是", "不过", "然而", "可是", "只是"] + cheng_jie_liebiao = ["然后", "接着", "此外", "而且", "另外"] + zhuan_jie_ci_liebiao = zhuan_zhe_liebiao + cheng_jie_liebiao + + if random.random() < replacement_prob: + # 相似度非常高时，尝试去重或特殊处理 + if similarity == 1.0: + logger.debug(f"{self.log_prefix} 想法完全重复 (相似度 1.0)，执行特殊处理...") + # 随机截取大约一半内容 + if len(new_content) > 1: # 避免内容过短无法截取 + split_point = max(1, len(new_content) // 2 + random.randint(-len(new_content)//4, len(new_content)//4)) + truncated_content = new_content[:split_point] + else: + truncated_content = new_content # 如果只有一个字符或者为空，就不截取了 + + # 添加语气词和转折/承接词 + yu_qi_ci = random.choice(yu_qi_ci_liebiao) + zhuan_jie_ci = random.choice(zhuan_jie_ci_liebiao) + content = f"{yu_qi_ci}{zhuan_jie_ci}，{truncated_content}" + logger.debug(f"{self.log_prefix} 想法重复，特殊处理后: {content}") + + else: + # 相似度较高但非100%，执行标准去重逻辑 + logger.debug(f"{self.log_prefix} 执行概率性去重 (概率: {replacement_prob:.2f})...") + matcher = difflib.SequenceMatcher(None, previous_mind, new_content) + deduplicated_parts = [] + last_match_end_in_b = 0 + for _i, j, n in matcher.get_matching_blocks(): + if last_match_end_in_b < j: + deduplicated_parts.append(new_content[last_match_end_in_b:j]) + last_match_end_in_b = j + n + + deduplicated_content = "".join(deduplicated_parts).strip() + + if deduplicated_content: + # 根据概率决定是否添加词语 + prefix_str = "" + if random.random() < 0.3: # 30% 概率添加语气词 + prefix_str += random.choice(yu_qi_ci_liebiao) + if random.random() < 0.7: # 70% 概率添加转折/承接词 + prefix_str += random.choice(zhuan_jie_ci_liebiao) + + # 组合最终结果 + if prefix_str: + content = f"{prefix_str}，{deduplicated_content}" # 更新 content + logger.debug(f"{self.log_prefix} 去重并添加引导词后: {content}") + else: + content = deduplicated_content # 更新 content + logger.debug(f"{self.log_prefix} 去重后 (未添加引导词): {content}") + else: + logger.warning(f"{self.log_prefix} 去重后内容为空，保留原始LLM输出: {new_content}") + content = new_content # 保留原始 content + else: + logger.debug(f"{self.log_prefix} 未执行概率性去重 (概率: {replacement_prob:.2f})") + # content 保持 new_content 不变 + + except Exception as e: + logger.error(f"{self.log_prefix} 应用概率性去重或特殊处理时出错: {e}") + logger.error(traceback.format_exc()) + # 出错时保留原始 content + content = new_content + + # ---------- 7. 更新思考状态并返回结果 ---------- + logger.info(f"{self.log_prefix} 最终心流思考结果: {content}") # 更新当前思考内容 self.update_current_mind(content) diff --git a/src/plugins/heartFC_chat/heartflow_prompt_builder.py b/src/plugins/heartFC_chat/heartflow_prompt_builder.py index 4e2f4453..db4979e3 100644 --- a/src/plugins/heartFC_chat/heartflow_prompt_builder.py +++ b/src/plugins/heartFC_chat/heartflow_prompt_builder.py @@ -260,6 +260,9 @@ class PromptBuilder: relation_prompt = "" for person in who_chat_in_group: relation_prompt += await relationship_manager.build_relationship_info(person) + print(f"relation_prompt: {relation_prompt}") + + print(f"relat11111111ion_prompt: {relation_prompt}") # 心情 mood_manager = MoodManager.get_instance() diff --git a/src/plugins/person_info/person_info.py b/src/plugins/person_info/person_info.py index 2c9fb72b..cde4ca93 100644 --- a/src/plugins/person_info/person_info.py +++ b/src/plugins/person_info/person_info.py @@ -137,34 +137,55 @@ class PersonInfoManager: @staticmethod def _extract_json_from_text(text: str) -> dict: """从文本中提取JSON数据的高容错方法""" + parsed_json = None try: # 尝试直接解析 - return json.loads(text) + parsed_json = json.loads(text) + # 如果解析结果是列表，尝试取第一个元素 + if isinstance(parsed_json, list): + if parsed_json: # 检查列表是否为空 + parsed_json = parsed_json[0] + else: # 如果列表为空，重置为 None，走后续逻辑 + parsed_json = None + # 确保解析结果是字典 + if isinstance(parsed_json, dict): + return parsed_json + except json.JSONDecodeError: - try: - # 尝试找到JSON格式的部分 - json_pattern = r"\{[^{}]*\}" - matches = re.findall(json_pattern, text) - if matches: - return json.loads(matches[0]) + # 解析失败，继续尝试其他方法 + pass + except Exception as e: + logger.warning(f"尝试直接解析JSON时发生意外错误: {e}") + pass # 继续尝试其他方法 - # 如果上面都失败了，尝试提取键值对 - nickname_pattern = r'"nickname"[:\s]+"([^"]+)"' - reason_pattern = r'"reason"[:\s]+"([^"]+)"' + # 如果直接解析失败或结果不是字典 + try: + # 尝试找到JSON对象格式的部分 + json_pattern = r"\{[^{}]*\}" + matches = re.findall(json_pattern, text) + if matches: + parsed_obj = json.loads(matches[0]) + if isinstance(parsed_obj, dict): # 确保是字典 + return parsed_obj - nickname_match = re.search(nickname_pattern, text) - reason_match = re.search(reason_pattern, text) + # 如果上面都失败了，尝试提取键值对 + nickname_pattern = r'"nickname"[:\s]+"([^"]+)"' + reason_pattern = r'"reason"[:\s]+"([^"]+)"' - if nickname_match: - return { - "nickname": nickname_match.group(1), - "reason": reason_match.group(1) if reason_match else "未提供理由", - } - except Exception as e: - logger.error(f"JSON提取失败: {str(e)}") + nickname_match = re.search(nickname_pattern, text) + reason_match = re.search(reason_pattern, text) - # 如果所有方法都失败了，返回空结果 - return {"nickname": "", "reason": ""} + if nickname_match: + return { + "nickname": nickname_match.group(1), + "reason": reason_match.group(1) if reason_match else "未提供理由", + } + except Exception as e: + logger.error(f"后备JSON提取失败: {str(e)}") + + # 如果所有方法都失败了，返回默认字典 + logger.warning(f"无法从文本中提取有效的JSON字典: {text}") + return {"nickname": "", "reason": ""} async def qv_person_name(self, person_id: str, user_nickname: str, user_cardname: str, user_avatar: str): """给某个用户取名""" diff --git a/src/plugins/person_info/relationship_manager.py b/src/plugins/person_info/relationship_manager.py index b66d74d5..34c28433 100644 --- a/src/plugins/person_info/relationship_manager.py +++ b/src/plugins/person_info/relationship_manager.py @@ -278,12 +278,19 @@ class RelationshipManager: return chat_stream.user_info.user_nickname, value, relationship_level[level_num] - async def build_relationship_info(self, person) -> str: - person_id = person_info_manager.get_person_id(person[0], person[1]) + async def build_relationship_info(self, person, is_id: bool = False) -> str: + if is_id: + person_id = person + else: + print(f"person: {person}") + person_id = person_info_manager.get_person_id(person[0], person[1]) person_name = await person_info_manager.get_value(person_id, "person_name") + print(f"person_name: {person_name}") relationship_value = await person_info_manager.get_value(person_id, "relationship_value") level_num = self.calculate_level_num(relationship_value) + print(f"person_name: {person_name}, relationship_value: {relationship_value}, level_num: {level_num}") + if level_num == 0 or level_num == 5: relationship_level = ["厌恶", "冷漠以对", "认识", "友好对待", "喜欢", "暧昧"] relation_prompt2_list = [ @@ -298,7 +305,7 @@ class RelationshipManager: elif level_num == 2: return "" else: - if random.random() < 0.5: + if random.random() < 0.6: relationship_level = ["厌恶", "冷漠以对", "认识", "友好对待", "喜欢", "暧昧"] relation_prompt2_list = [ "忽视的回应", diff --git a/src/plugins/utils/chat_message_builder.py b/src/plugins/utils/chat_message_builder.py index 4e807ffa..eb2994f4 100644 --- a/src/plugins/utils/chat_message_builder.py +++ b/src/plugins/utils/chat_message_builder.py @@ -364,3 +364,33 @@ async def build_readable_messages( else: # 理论上不应该发生，但作为保险 return read_mark_line.strip() # 如果前后都无消息，只返回标记行 + + +async def get_person_id_list(messages: List[Dict[str, Any]]) -> List[str]: + """ + 从消息列表中提取不重复的 person_id 列表 (忽略机器人自身)。 + + Args: + messages: 消息字典列表。 + + Returns: + 一个包含唯一 person_id 的列表。 + """ + person_ids_set = set() # 使用集合来自动去重 + + for msg in messages: + user_info = msg.get("user_info", {}) + platform = user_info.get("platform") + user_id = user_info.get("user_id") + + # 检查必要信息是否存在 且 不是机器人自己 + if not all([platform, user_id]) or user_id == global_config.BOT_QQ: + continue + + person_id = person_info_manager.get_person_id(platform, user_id) + + # 只有当获取到有效 person_id 时才添加 + if person_id: + person_ids_set.add(person_id) + + return list(person_ids_set) # 将集合转换为列表返回 From 8dcfe04642be5d733a3a349c1725162e83874ac9 Mon Sep 17 00:00:00 2001 From: SengokuCola <1026294844@qq.com> Date: Tue, 29 Apr 2025 23:47:37 +0800 Subject: [PATCH 25/26] fixruad --- scripts/remove_chong.py | 13 ++++++---- src/config/config.py | 3 ++- src/do_tool/tool_can_use/get_memory.py | 4 +-- src/heart_flow/observation.py | 8 +++--- src/heart_flow/sub_mind.py | 25 ++++++++++--------- src/heart_flow/subheartflow_manager.py | 4 +-- .../heartFC_chat/heartflow_prompt_builder.py | 2 +- src/plugins/person_info/person_info.py | 8 +++--- .../person_info/relationship_manager.py | 4 +-- src/plugins/utils/chat_message_builder.py | 5 +++- 10 files changed, 41 insertions(+), 35 deletions(-) diff --git a/scripts/remove_chong.py b/scripts/remove_chong.py index ac8ceca7..024acb3f 100644 --- a/scripts/remove_chong.py +++ b/scripts/remove_chong.py @@ -1,6 +1,7 @@ import difflib import random + def ji_suan_xiang_si_du(wen_ben_yi: str, wen_ben_er: str) -> float: """ 计算两个文本字符串的相似度。 @@ -17,6 +18,7 @@ def ji_suan_xiang_si_du(wen_ben_yi: str, wen_ben_er: str) -> float: xiang_si_bi_lv = xu_lie_pi_pei_qi.ratio() return xiang_si_bi_lv + def ji_suan_ti_huan_gai_lv(xiang_si_du: float) -> float: """ 根据相似度计算替换的概率。 @@ -34,11 +36,12 @@ def ji_suan_ti_huan_gai_lv(xiang_si_du: float) -> float: elif 0.4 < xiang_si_du <= 0.6: # p = 3.5 * s - 1.4 (线性方程 y - 0 = (0.7-0)/(0.6-0.4) * (x - 0.4)) gai_lv = 3.5 * xiang_si_du - 1.4 - return max(0.0, gai_lv) # 确保概率不小于0 + return max(0.0, gai_lv) # 确保概率不小于0 elif 0.6 < xiang_si_du < 0.9: # p = s + 0.1 (线性方程 y - 0.7 = (1-0.7)/(0.9-0.6) * (x - 0.6)) gai_lv = xiang_si_du + 0.1 - return min(1.0, max(0.0, gai_lv)) # 确保概率在 0 和 1 之间 + return min(1.0, max(0.0, gai_lv)) # 确保概率在 0 和 1 之间 + # 获取用户输入 shu_ru_yi = "豆豆刚刚回复了我的问候 现在可以等待对方的回应 不需要再主动发言 目前情绪满足 不需要使用工具" @@ -67,7 +70,7 @@ if random.random() < ti_huan_gai_lv: # 更新下一个非匹配部分的起始位置 last_match_end_in_b = j + n - jie_guo = "".join(qu_chong_hou_de_er).strip() # 去除首尾空白 + jie_guo = "".join(qu_chong_hou_de_er).strip() # 去除首尾空白 if jie_guo: # 定义词语列表 @@ -78,9 +81,9 @@ if random.random() < ti_huan_gai_lv: # 根据概率决定是否添加词语 qian_zhui_str = "" - if random.random() < 0.3: # 30% 概率添加语气词 + if random.random() < 0.3: # 30% 概率添加语气词 qian_zhui_str += random.choice(yu_qi_ci_liebiao) - if random.random() < 0.7: # 70% 概率添加转折/承接词 + if random.random() < 0.7: # 70% 概率添加转折/承接词 qian_zhui_str += random.choice(zhuan_jie_ci_liebiao) # 组合最终结果 diff --git a/src/config/config.py b/src/config/config.py index d9d5ce16..f09da9a7 100644 --- a/src/config/config.py +++ b/src/config/config.py @@ -185,7 +185,7 @@ class BotConfig: reply_trigger_threshold: float = 3.0 # 心流聊天触发阈值，越低越容易触发 probability_decay_factor_per_second: float = 0.2 # 概率衰减因子，越大衰减越快 default_decay_rate_per_second: float = 0.98 # 默认衰减率，越大衰减越慢 - allow_focus_mode: bool = True # 是否允许子心流进入 FOCUSED 状态 + allow_focus_mode: bool = True # 是否允许子心流进入 FOCUSED 状态 # sub_heart_flow_update_interval: int = 60 # 子心流更新频率，间隔 单位秒 # sub_heart_flow_freeze_time: int = 120 # 子心流冻结时间，超过这个时间没有回复，子心流会冻结，间隔 单位秒 @@ -417,6 +417,7 @@ class BotConfig: config.model_normal_probability = response_config.get( "model_normal_probability", config.model_normal_probability ) + def heartflow(parent: dict): heartflow_config = parent["heartflow"] config.sub_heart_flow_stop_time = heartflow_config.get( diff --git a/src/do_tool/tool_can_use/get_memory.py b/src/do_tool/tool_can_use/get_memory.py index 2ac550f4..b38423ed 100644 --- a/src/do_tool/tool_can_use/get_memory.py +++ b/src/do_tool/tool_can_use/get_memory.py @@ -46,11 +46,11 @@ class GetMemoryTool(BaseTool): if related_memory: for memory in related_memory: memory_info += memory[1] + "\n" - + if memory_info: content = f"你记得这些事情: {memory_info}\n" content += "以上是你的回忆，不一定是目前聊天里的人说的，也不一定是现在发生的事情，请记住。\n" - + else: content = f"{topic}的记忆，你记不太清" diff --git a/src/heart_flow/observation.py b/src/heart_flow/observation.py index 5eeb64d3..5793e772 100644 --- a/src/heart_flow/observation.py +++ b/src/heart_flow/observation.py @@ -46,7 +46,7 @@ class ChattingObservation(Observation): self.mid_memorys = [] self.max_mid_memory_len = global_config.compress_length_limit self.mid_memory_info = "" - + self.person_list = [] self.llm_summary = LLMRequest( @@ -157,11 +157,9 @@ class ChattingObservation(Observation): ) self.person_list = await get_person_id_list(self.talking_message) - - # print(f"self.11111person_list: {self.person_list}") - - + # print(f"self.11111person_list: {self.person_list}") + logger.trace( f"Chat {self.chat_id} - 压缩早期记忆：{self.mid_memory_info}\n现在聊天内容：{self.talking_message_str}" ) diff --git a/src/heart_flow/sub_mind.py b/src/heart_flow/sub_mind.py index d167f42e..e59ee855 100644 --- a/src/heart_flow/sub_mind.py +++ b/src/heart_flow/sub_mind.py @@ -16,7 +16,6 @@ import difflib from src.plugins.person_info.relationship_manager import relationship_manager - logger = get_logger("sub_heartflow") @@ -60,6 +59,7 @@ def calculate_similarity(text_a: str, text_b: str) -> float: matcher = difflib.SequenceMatcher(None, text_a, text_b) return matcher.ratio() + def calculate_replacement_probability(similarity: float) -> float: """ 根据相似度计算替换的概率。 @@ -138,13 +138,12 @@ class SubMind: # 获取个性化信息 individuality = Individuality.get_instance() - relation_prompt = "" print(f"person_list: {person_list}") for person in person_list: relation_prompt += await relationship_manager.build_relationship_info(person, is_id=True) - + print(f"relat22222ion_prompt: {relation_prompt}") # 构建个性部分 @@ -302,7 +301,7 @@ class SubMind: logger.warning(f"{self.log_prefix} LLM返回空结果，思考失败。") # ---------- 6. 应用概率性去重和修饰 ---------- - new_content = content # 保存 LLM 直接输出的结果 + new_content = content # 保存 LLM 直接输出的结果 try: similarity = calculate_similarity(previous_mind, new_content) replacement_prob = calculate_replacement_probability(similarity) @@ -319,11 +318,13 @@ class SubMind: if similarity == 1.0: logger.debug(f"{self.log_prefix} 想法完全重复 (相似度 1.0)，执行特殊处理...") # 随机截取大约一半内容 - if len(new_content) > 1: # 避免内容过短无法截取 - split_point = max(1, len(new_content) // 2 + random.randint(-len(new_content)//4, len(new_content)//4)) + if len(new_content) > 1: # 避免内容过短无法截取 + split_point = max( + 1, len(new_content) // 2 + random.randint(-len(new_content) // 4, len(new_content) // 4) + ) truncated_content = new_content[:split_point] else: - truncated_content = new_content # 如果只有一个字符或者为空，就不截取了 + truncated_content = new_content # 如果只有一个字符或者为空，就不截取了 # 添加语气词和转折/承接词 yu_qi_ci = random.choice(yu_qi_ci_liebiao) @@ -347,21 +348,21 @@ class SubMind: if deduplicated_content: # 根据概率决定是否添加词语 prefix_str = "" - if random.random() < 0.3: # 30% 概率添加语气词 + if random.random() < 0.3: # 30% 概率添加语气词 prefix_str += random.choice(yu_qi_ci_liebiao) - if random.random() < 0.7: # 70% 概率添加转折/承接词 + if random.random() < 0.7: # 70% 概率添加转折/承接词 prefix_str += random.choice(zhuan_jie_ci_liebiao) # 组合最终结果 if prefix_str: - content = f"{prefix_str}，{deduplicated_content}" # 更新 content + content = f"{prefix_str}，{deduplicated_content}" # 更新 content logger.debug(f"{self.log_prefix} 去重并添加引导词后: {content}") else: - content = deduplicated_content # 更新 content + content = deduplicated_content # 更新 content logger.debug(f"{self.log_prefix} 去重后 (未添加引导词): {content}") else: logger.warning(f"{self.log_prefix} 去重后内容为空，保留原始LLM输出: {new_content}") - content = new_content # 保留原始 content + content = new_content # 保留原始 content else: logger.debug(f"{self.log_prefix} 未执行概率性去重 (概率: {replacement_prob:.2f})") # content 保持 new_content 不变 diff --git a/src/heart_flow/subheartflow_manager.py b/src/heart_flow/subheartflow_manager.py index c355867c..afa0328e 100644 --- a/src/heart_flow/subheartflow_manager.py +++ b/src/heart_flow/subheartflow_manager.py @@ -266,9 +266,9 @@ class SubHeartflowManager: # --- 新增：检查是否允许进入 FOCUS 模式 --- # if not global_config.allow_focus_mode: - if int(time.time()) % 60 == 0: # 每60秒输出一次日志避免刷屏 + if int(time.time()) % 60 == 0: # 每60秒输出一次日志避免刷屏 logger.debug(f"{log_prefix} 配置不允许进入 FOCUSED 状态 (allow_focus_mode=False)") - return # 如果不允许，直接返回 + return # 如果不允许，直接返回 # --- 结束新增 --- logger.debug(f"{log_prefix} 当前状态 ({current_state.value}) 开始尝试提升到FOCUSED状态") diff --git a/src/plugins/heartFC_chat/heartflow_prompt_builder.py b/src/plugins/heartFC_chat/heartflow_prompt_builder.py index db4979e3..32c78a10 100644 --- a/src/plugins/heartFC_chat/heartflow_prompt_builder.py +++ b/src/plugins/heartFC_chat/heartflow_prompt_builder.py @@ -261,7 +261,7 @@ class PromptBuilder: for person in who_chat_in_group: relation_prompt += await relationship_manager.build_relationship_info(person) print(f"relation_prompt: {relation_prompt}") - + print(f"relat11111111ion_prompt: {relation_prompt}") # 心情 diff --git a/src/plugins/person_info/person_info.py b/src/plugins/person_info/person_info.py index cde4ca93..8bafe5eb 100644 --- a/src/plugins/person_info/person_info.py +++ b/src/plugins/person_info/person_info.py @@ -143,9 +143,9 @@ class PersonInfoManager: parsed_json = json.loads(text) # 如果解析结果是列表，尝试取第一个元素 if isinstance(parsed_json, list): - if parsed_json: # 检查列表是否为空 + if parsed_json: # 检查列表是否为空 parsed_json = parsed_json[0] - else: # 如果列表为空，重置为 None，走后续逻辑 + else: # 如果列表为空，重置为 None，走后续逻辑 parsed_json = None # 确保解析结果是字典 if isinstance(parsed_json, dict): @@ -156,7 +156,7 @@ class PersonInfoManager: pass except Exception as e: logger.warning(f"尝试直接解析JSON时发生意外错误: {e}") - pass # 继续尝试其他方法 + pass # 继续尝试其他方法 # 如果直接解析失败或结果不是字典 try: @@ -165,7 +165,7 @@ class PersonInfoManager: matches = re.findall(json_pattern, text) if matches: parsed_obj = json.loads(matches[0]) - if isinstance(parsed_obj, dict): # 确保是字典 + if isinstance(parsed_obj, dict): # 确保是字典 return parsed_obj # 如果上面都失败了，尝试提取键值对 diff --git a/src/plugins/person_info/relationship_manager.py b/src/plugins/person_info/relationship_manager.py index 34c28433..3062a736 100644 --- a/src/plugins/person_info/relationship_manager.py +++ b/src/plugins/person_info/relationship_manager.py @@ -288,9 +288,9 @@ class RelationshipManager: print(f"person_name: {person_name}") relationship_value = await person_info_manager.get_value(person_id, "relationship_value") level_num = self.calculate_level_num(relationship_value) - + print(f"person_name: {person_name}, relationship_value: {relationship_value}, level_num: {level_num}") - + if level_num == 0 or level_num == 5: relationship_level = ["厌恶", "冷漠以对", "认识", "友好对待", "喜欢", "暧昧"] relation_prompt2_list = [ diff --git a/src/plugins/utils/chat_message_builder.py b/src/plugins/utils/chat_message_builder.py index eb2994f4..a7eef443 100644 --- a/src/plugins/utils/chat_message_builder.py +++ b/src/plugins/utils/chat_message_builder.py @@ -348,7 +348,10 @@ async def build_readable_messages( messages_before_mark, replace_bot_name, merge_messages, timestamp_mode, truncate ) formatted_after, _ = await _build_readable_messages_internal( - messages_after_mark, replace_bot_name, merge_messages, timestamp_mode, + messages_after_mark, + replace_bot_name, + merge_messages, + timestamp_mode, ) readable_read_mark = translate_timestamp_to_human_readable(read_mark, mode=timestamp_mode) From 298bcbcee4212ee9c2c5feaeec4b8f2cfc91a7c9 Mon Sep 17 00:00:00 2001 From: SengokuCola <1026294844@qq.com> Date: Tue, 29 Apr 2025 23:49:13 +0800 Subject: [PATCH 26/26] Delete remove_chong.py --- scripts/remove_chong.py | 104 ---------------------------------------- 1 file changed, 104 deletions(-) delete mode 100644 scripts/remove_chong.py diff --git a/scripts/remove_chong.py b/scripts/remove_chong.py deleted file mode 100644 index 1398c865..00000000 --- a/scripts/remove_chong.py +++ /dev/null @@ -1,104 +0,0 @@ -import difflib -import random - - - -def ji_suan_xiang_si_du(wen_ben_yi: str, wen_ben_er: str) -> float: - """ - 计算两个文本字符串的相似度。 - - 参数: - wen_ben_yi (str): 第一个文本字符串。 - wen_ben_er (str): 第二个文本字符串。 - - 返回: - float: 两个文本的相似度比率 (0 到 1 之间)。 - """ - xu_lie_pi_pei_qi = difflib.SequenceMatcher(None, wen_ben_yi, wen_ben_er) - # 获取相似度比率 - xiang_si_bi_lv = xu_lie_pi_pei_qi.ratio() - return xiang_si_bi_lv - - - -def ji_suan_ti_huan_gai_lv(xiang_si_du: float) -> float: - """ - 根据相似度计算替换的概率。 - 规则： - - 相似度 <= 0.4: 概率 = 0 - - 相似度 >= 0.9: 概率 = 1 - - 相似度 == 0.6: 概率 = 0.7 - - 0.4 < 相似度 <= 0.6: 线性插值 (0.4, 0) 到 (0.6, 0.7) - - 0.6 < 相似度 < 0.9: 线性插值 (0.6, 0.7) 到 (0.9, 1.0) - """ - if xiang_si_du <= 0.4: - return 0.0 - elif xiang_si_du >= 0.9: - return 1.0 - elif 0.4 < xiang_si_du <= 0.6: - # p = 3.5 * s - 1.4 (线性方程 y - 0 = (0.7-0)/(0.6-0.4) * (x - 0.4)) - gai_lv = 3.5 * xiang_si_du - 1.4 - return max(0.0, gai_lv) # 确保概率不小于0 - elif 0.6 < xiang_si_du < 0.9: - # p = s + 0.1 (线性方程 y - 0.7 = (1-0.7)/(0.9-0.6) * (x - 0.6)) - gai_lv = xiang_si_du + 0.1 - return min(1.0, max(0.0, gai_lv)) # 确保概率在 0 和 1 之间 - - -# 获取用户输入 -shu_ru_yi = "豆豆刚刚回复了我的问候 现在可以等待对方的回应 不需要再主动发言 目前情绪满足 不需要使用工具" - -shu_ru_er = "豆豆刚刚回复了我的问候 现在可以等待对方的回应 不需要再主动发言 目前情绪满足 不需要使用工具 群主突然提到复活的事情 感觉有点莫名其妙 但情绪上还是满足的 暂时不需要回复" - -# 计算相似度 -xiang_si_du = ji_suan_xiang_si_du(shu_ru_yi, shu_ru_er) - -# 计算替换概率 -ti_huan_gai_lv = ji_suan_ti_huan_gai_lv(xiang_si_du) -print(f"文本相似度: {xiang_si_du:.2f}, 执行替换操作的概率: {ti_huan_gai_lv:.2f}") - -# 根据概率决定是否执行替换 -if random.random() < ti_huan_gai_lv: - print(f"执行替换操作 (基于概率 {ti_huan_gai_lv:.2f})...") - pi_pei_qi = difflib.SequenceMatcher(None, shu_ru_yi, shu_ru_er) - qu_chong_hou_de_er = [] - last_match_end_in_b = 0 - # 获取匹配块 (i, j, n) 其中 a[i:i+n] == b[j:j+n] - # 注意：get_matching_blocks 最后会有一个 (len(a), len(b), 0) 的虚拟块 - for _i, j, n in pi_pei_qi.get_matching_blocks(): - # 添加上一个匹配块结束到当前匹配块开始之间的非匹配部分 (来自文本二) - if last_match_end_in_b < j: - qu_chong_hou_de_er.append(shu_ru_er[last_match_end_in_b:j]) - # 更新下一个非匹配部分的起始位置 - last_match_end_in_b = j + n - - jie_guo = "".join(qu_chong_hou_de_er).strip() # 去除首尾空白 - jie_guo = "".join(qu_chong_hou_de_er).strip() # 去除首尾空白 - - if jie_guo: - # 定义词语列表 - yu_qi_ci_liebiao = ["嗯", "哦", "啊", "唉", "哈", "唔"] - zhuan_zhe_liebiao = ["但是", "不过", "然而", "可是", "只是"] - cheng_jie_liebiao = ["然后", "接着", "此外", "而且", "另外"] - zhuan_jie_ci_liebiao = zhuan_zhe_liebiao + cheng_jie_liebiao - - # 根据概率决定是否添加词语 - qian_zhui_str = "" - if random.random() < 0.3: # 30% 概率添加语气词 - if random.random() < 0.3: # 30% 概率添加语气词 - qian_zhui_str += random.choice(yu_qi_ci_liebiao) - if random.random() < 0.7: # 70% 概率添加转折/承接词 - if random.random() < 0.7: # 70% 概率添加转折/承接词 - qian_zhui_str += random.choice(zhuan_jie_ci_liebiao) - - # 组合最终结果 - if qian_zhui_str: - zui_zhong_jie_guo = f"{qian_zhui_str}，{jie_guo}" - print(f"移除重复部分并添加引导词后的文本二: {zui_zhong_jie_guo}") - else: - # 如果没有添加任何前缀词，直接输出去重结果 - print(f"移除重复部分后的文本二: {jie_guo}") - else: - print("移除重复部分后文本二为空。") -else: - print(f"未执行替换操作 (基于概率 {ti_huan_gai_lv:.2f})。原始相似度为: {xiang_si_du:.2f}")