Merge branch 'dev' of https://github.com/MaiM-with-u/MaiBot into dev

2025-04-29 20:38:14 +08:00
parent dec2e4f442 738728bb91
commit bc801fff54
4 changed files with 166 additions and 27 deletions
--- 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,90 @@ 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)}"
--- 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}]不想再理你了...")
--- a/src/plugins/PFC/reply_generator.py
+++ b/src/plugins/PFC/reply_generator.py
@@ -57,6 +57,24 @@ 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,7 +153,10 @@ 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 (首次/非连续回复生成)")

--- 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))
            )

            # 将表情包信息转换为可读的字符串