feat: 新增统一命名功能，自动替换prompt内唯一标识符，优化prompt效果

2025-04-16 01:13:18 +08:00
parent 42ccb3dccb
commit 9fc74cb066
13 changed files with 443 additions and 129 deletions
--- a/src/plugins/chat_module/think_flow_chat/think_flow_chat.py
+++ b/src/plugins/chat_module/think_flow_chat/think_flow_chat.py
@@ -235,6 +235,7 @@ class ThinkFlowChat:
        do_reply = False
        if random() < reply_probability:
            try:
+                
                do_reply = True

                # 回复前处理
@@ -258,7 +259,7 @@ class ThinkFlowChat:
                        await heartflow.get_subheartflow(chat.stream_id).do_observe()
                except Exception as e:
                    logger.error(f"心流观察失败: {e}")
-                    traceback.print_exc()
+                    logger.error(traceback.format_exc())

                info_catcher.catch_after_observe(timing_results["观察"])

@@ -329,13 +330,17 @@ class ThinkFlowChat:
                            chat.stream_id
                        ).do_thinking_before_reply(
                            message_txt=message.processed_plain_text,
-                            sender_name=message.message_info.user_info.user_nickname,
+                            sender_info=message.message_info.user_info,
                            chat_stream=chat,
                            obs_id=get_mid_memory_id,
                            extra_info=tool_result_info,
                        )
                except Exception as e:
                    logger.error(f"心流思考前脑内状态失败: {e}")
+                    logger.error(traceback.format_exc())
+                    # 确保变量被定义，即使在错误情况下
+                    current_mind = ""
+                    past_mind = ""

                info_catcher.catch_afer_shf_step(timing_results["思考前脑内状态"], past_mind, current_mind)

@@ -373,6 +378,7 @@ class ThinkFlowChat:
                except Exception as e:
                    logger.error(f"心流处理表情包失败: {e}")

+                # 思考后脑内状态更新
                try:
                    with Timer("思考后脑内状态更新", timing_results):
                        stream_id = message.chat_stream.stream_id
@@ -387,9 +393,43 @@ class ThinkFlowChat:
                        )
                except Exception as e:
                    logger.error(f"心流思考后脑内状态更新失败: {e}")
+                    logger.error(traceback.format_exc())

                # 回复后处理
                await willing_manager.after_generate_reply_handle(message.message_info.message_id)
+                
+                # 处理认识关系
+                try:
+                    is_known = await relationship_manager.is_known_some_one(
+                        message.message_info.platform, 
+                        message.message_info.user_info.user_id
+                    )
+                    if not is_known:
+                        logger.info(f"首次认识用户: {message.message_info.user_info.user_nickname}")
+                        await relationship_manager.first_knowing_some_one(
+                            message.message_info.platform,
+                            message.message_info.user_info.user_id,
+                            message.message_info.user_info.user_nickname,
+                            message.message_info.user_info.user_cardname or message.message_info.user_info.user_nickname,
+                            ""
+                        )
+                    else:
+                        logger.debug(f"已认识用户: {message.message_info.user_info.user_nickname}")
+                        if not await relationship_manager.is_qved_name(
+                            message.message_info.platform, 
+                            message.message_info.user_info.user_id
+                        ):
+                            logger.info(f"更新已认识但未取名的用户: {message.message_info.user_info.user_nickname}")
+                            await relationship_manager.first_knowing_some_one(
+                                message.message_info.platform,
+                                message.message_info.user_info.user_id,
+                                message.message_info.user_info.user_nickname,
+                                message.message_info.user_info.user_cardname or message.message_info.user_info.user_nickname,
+                                ""
+                            )
+                except Exception as e:
+                    logger.error(f"处理认识关系失败: {e}")
+                    logger.error(traceback.format_exc())

            except Exception as e:
                logger.error(f"心流处理消息失败: {e}")
--- a/src/plugins/chat_module/think_flow_chat/think_flow_generator.py
+++ b/src/plugins/chat_module/think_flow_chat/think_flow_generator.py
@@ -100,15 +100,17 @@ class ResponseGenerator:

        info_catcher = info_catcher_manager.get_info_catcher(thinking_id)

-        if message.chat_stream.user_info.user_cardname and message.chat_stream.user_info.user_nickname:
-            sender_name = (
-                f"[({message.chat_stream.user_info.user_id}){message.chat_stream.user_info.user_nickname}]"
-                f"{message.chat_stream.user_info.user_cardname}"
-            )
-        elif message.chat_stream.user_info.user_nickname:
-            sender_name = f"({message.chat_stream.user_info.user_id}){message.chat_stream.user_info.user_nickname}"
-        else:
-            sender_name = f"用户({message.chat_stream.user_info.user_id})"
+        # if message.chat_stream.user_info.user_cardname and message.chat_stream.user_info.user_nickname:
+        #     sender_name = (
+        #         f"[({message.chat_stream.user_info.user_id}){message.chat_stream.user_info.user_nickname}]"
+        #         f"{message.chat_stream.user_info.user_cardname}"
+        #     )
+        # elif message.chat_stream.user_info.user_nickname:
+        #     sender_name = f"({message.chat_stream.user_info.user_id}){message.chat_stream.user_info.user_nickname}"
+        # else:
+        #     sender_name = f"用户({message.chat_stream.user_info.user_id})"
+            
+        sender_name = f"<{message.chat_stream.user_info.platform}:{message.chat_stream.user_info.user_id}:{message.chat_stream.user_info.user_nickname}:{message.chat_stream.user_info.user_cardname}>"

        # 构建prompt
        with Timer() as t_build_prompt:
@@ -119,14 +121,7 @@ class ResponseGenerator:
                    sender_name=sender_name,
                    stream_id=message.chat_stream.stream_id,
                )
-            elif mode == "simple":
-                prompt = await prompt_builder._build_prompt_simple(
-                    message.chat_stream,
-                    message_txt=message.processed_plain_text,
-                    sender_name=sender_name,
-                    stream_id=message.chat_stream.stream_id,
-                )
-        logger.info(f"构建{mode}prompt时间: {t_build_prompt.human_readable}")
+        logger.info(f"构建prompt时间: {t_build_prompt.human_readable}")

        try:
            content, reasoning_content, self.current_model_name = await model.generate_response(prompt)
@@ -141,49 +136,6 @@ class ResponseGenerator:

        return content

-    async def _check_response_with_model(
-        self, message: MessageRecv, content: str, model: LLM_request, thinking_id: str
-    ) -> str:
-        _info_catcher = info_catcher_manager.get_info_catcher(thinking_id)
-
-        sender_name = ""
-        if message.chat_stream.user_info.user_cardname and message.chat_stream.user_info.user_nickname:
-            sender_name = (
-                f"[({message.chat_stream.user_info.user_id}){message.chat_stream.user_info.user_nickname}]"
-                f"{message.chat_stream.user_info.user_cardname}"
-            )
-        elif message.chat_stream.user_info.user_nickname:
-            sender_name = f"({message.chat_stream.user_info.user_id}){message.chat_stream.user_info.user_nickname}"
-        else:
-            sender_name = f"用户({message.chat_stream.user_info.user_id})"
-
-        # 构建prompt
-        with Timer() as t_build_prompt_check:
-            prompt = await prompt_builder._build_prompt_check_response(
-                message.chat_stream,
-                message_txt=message.processed_plain_text,
-                sender_name=sender_name,
-                stream_id=message.chat_stream.stream_id,
-                content=content,
-            )
-        logger.info(f"构建check_prompt: {prompt}")
-        logger.info(f"构建check_prompt时间: {t_build_prompt_check.human_readable}")
-
-        try:
-            checked_content, reasoning_content, self.current_model_name = await model.generate_response(prompt)
-
-            # info_catcher.catch_after_llm_generated(
-            #     prompt=prompt,
-            #     response=content,
-            #     reasoning_content=reasoning_content,
-            #     model_name=self.current_model_name)
-
-        except Exception:
-            logger.exception("检查回复时出错")
-            return None
-
-        return checked_content
-
    async def _get_emotion_tags(self, content: str, processed_plain_text: str):
        """提取情感标签，结合立场和情绪"""
        try:
--- a/src/plugins/chat_module/think_flow_chat/think_flow_prompt_builder.py
+++ b/src/plugins/chat_module/think_flow_chat/think_flow_prompt_builder.py
@@ -8,7 +8,8 @@ from src.common.logger import get_module_logger
 from ....individuality.individuality import Individuality
 from src.heart_flow.heartflow import heartflow
 from src.plugins.utils.prompt_builder import Prompt, global_prompt_manager
-
+from src.plugins.person_info.relationship_manager import relationship_manager
+from src.plugins.chat.utils import parse_text_timestamps
 logger = get_module_logger("prompt")


@@ -160,7 +161,10 @@ class PromptBuilder:
            prompt_ger=prompt_ger,
            moderation_prompt=await global_prompt_manager.get_prompt_async("moderation_prompt"),
        )
-
+        
+        prompt = await relationship_manager.convert_all_person_sign_to_person_name(prompt)
+        prompt = parse_text_timestamps(prompt, mode="lite")
+        
        return prompt

    async def _build_prompt_simple(