feat: 支持外部message重载prompt template

2025-04-12 23:27:31 +08:00
parent 265d9a06f4
commit 990a7a3c05
7 changed files with 234 additions and 110 deletions
--- a/src/heart_flow/heartflow.py
+++ b/src/heart_flow/heartflow.py
@@ -155,7 +155,7 @@ class Heartflow:
        # prompt += f"你现在{mood_info}。"
        # prompt += "现在你接下去继续思考，产生新的想法，但是要基于原有的主要想法，不要分点输出，"
        # prompt += "输出连贯的内心独白，不要太长，但是记得结合上述的消息，关注新内容:"
-        prompt = global_prompt_manager.get_prompt("thinking_prompt").format(
+        prompt = (await global_prompt_manager.get_prompt_async("thinking_prompt")).format(
            schedule_info, personality_info, related_memory_info, current_thinking_info, sub_flows_info, mood_info
        )

@@ -212,7 +212,7 @@ class Heartflow:
        # prompt += f"你现在{mood_info}\n"
        # prompt += """现在请你总结这些聊天内容，注意关注聊天内容对原有的想法的影响，输出连贯的内心独白
        # 不要太长，但是记得结合上述的消息，要记得你的人设，关注新内容:"""
-        prompt = global_prompt_manager.get_prompt("mind_summary_prompt").format(
+        prompt = (await global_prompt_manager.get_prompt_async("mind_summary_prompt")).format(
            personality_info, global_config.BOT_NICKNAME, self.current_mind, minds_str, mood_info
        )

--- a/src/heart_flow/sub_heartflow.py
+++ b/src/heart_flow/sub_heartflow.py
@@ -16,7 +16,8 @@ import random
 from src.plugins.chat.chat_stream import ChatStream
 from src.plugins.person_info.relationship_manager import relationship_manager
 from src.plugins.chat.utils import get_recent_group_speaker
-from src.do_tool.tool_use import ToolUser   
+from src.do_tool.tool_use import ToolUser
+from ..plugins.utils.prompt_builder import Prompt,global_prompt_manager   

 subheartflow_config = LogConfig(
    # 使用海马体专用样式
@@ -25,6 +26,35 @@ subheartflow_config = LogConfig(
 )
 logger = get_module_logger("subheartflow", config=subheartflow_config)

+def init_prompt():
+    prompt = ""
+    # prompt += f"麦麦的总体想法是：{self.main_heartflow_info}\n\n"
+    prompt += "{collected_info}\n"
+    prompt += "{relation_prompt_all}\n"
+    prompt += "{prompt_personality}\n"
+    prompt += "刚刚你的想法是{current_thinking_info}。如果有新的内容，记得转换话题\n"
+    prompt += "-----------------------------------\n"
+    prompt += "现在你正在上网，和qq群里的网友们聊天，群里正在聊的话题是：{chat_observe_info}\n"
+    prompt += "你现在{mood_info}\n"
+    prompt += "你注意到{sender_name}刚刚说：{message_txt}\n"
+    prompt += "现在你接下去继续思考，产生新的想法，不要分点输出，输出连贯的内心独白"
+    prompt += "思考时可以想想如何对群聊内容进行回复。回复的要求是：平淡一些，简短一些，说中文，尽量不要说你说过的话\n"
+    prompt += "请注意不要输出多余内容(包括前后缀，冒号和引号，括号， 表情，等)，不要带有括号和动作描写"
+    prompt += "记得结合上述的消息，生成内心想法，文字不要浮夸，注意你就是{bot_name}，{bot_name}指的就是你。"
+    Prompt(prompt,"sub_heartflow_prompt_before")
+    prompt = ""
+    # prompt += f"你现在正在做的事情是：{schedule_info}\n"
+    prompt += "{prompt_personality}\n"
+    prompt += "现在你正在上网，和qq群里的网友们聊天，群里正在聊的话题是：{chat_observe_info}\n"
+    prompt += "刚刚你的想法是{current_thinking_info}。"
+    prompt += "你现在看到了网友们发的新消息:{message_new_info}\n"
+    prompt += "你刚刚回复了群友们:{reply_info}"
+    prompt += "你现在{mood_info}"
+    prompt += "现在你接下去继续思考，产生新的想法，记得保留你刚刚的想法，不要分点输出，输出连贯的内心独白"
+    prompt += "不要太长，但是记得结合上述的消息，要记得你的人设，关注聊天和新内容，关注你回复的内容，不要思考太多:"
+    Prompt(prompt,'sub_heartflow_prompt_after')
+    
+

 class CurrentState:
    def __init__(self):
@@ -170,26 +200,34 @@ class SubHeartflow:
        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}的关系和态度进行回复，明确你的立场和情感。"
-        )
+        # relation_prompt_all = (
+        #     f"{relation_prompt}关系等级越大，关系越好，请分析聊天记录，"
+        #     f"根据你和说话者{sender_name}的关系和态度进行回复，明确你的立场和情感。"
+        # )
+        relation_prompt_all = (await global_prompt_manager.get_prompt_async('relationship_prompt')).format(
+            relation_prompt,sender_name
+            )

-        prompt = ""
-        # prompt += f"麦麦的总体想法是：{self.main_heartflow_info}\n\n"
-        if tool_result.get("used_tools", False):
-            prompt += f"{collected_info}\n"
-        prompt += f"{relation_prompt_all}\n"
-        prompt += f"{prompt_personality}\n"
-        prompt += f"刚刚你的想法是{current_thinking_info}。如果有新的内容，记得转换话题\n"
-        prompt += "-----------------------------------\n"
-        prompt += f"现在你正在上网，和qq群里的网友们聊天，群里正在聊的话题是：{chat_observe_info}\n"
-        prompt += f"你现在{mood_info}\n"
-        prompt += f"你注意到{sender_name}刚刚说：{message_txt}\n"
-        prompt += "现在你接下去继续思考，产生新的想法，不要分点输出，输出连贯的内心独白"
-        prompt += "思考时可以想想如何对群聊内容进行回复。回复的要求是：平淡一些，简短一些，说中文，尽量不要说你说过的话\n"
-        prompt += "请注意不要输出多余内容(包括前后缀，冒号和引号，括号， 表情，等)，不要带有括号和动作描写"
-        prompt += f"记得结合上述的消息，生成内心想法，文字不要浮夸，注意你就是{self.bot_name}，{self.bot_name}指的就是你。"
+        # prompt = ""
+        # # prompt += f"麦麦的总体想法是：{self.main_heartflow_info}\n\n"
+        # if tool_result.get("used_tools", False):
+        #     prompt += f"{collected_info}\n"
+        # prompt += f"{relation_prompt_all}\n"
+        # prompt += f"{prompt_personality}\n"
+        # prompt += f"刚刚你的想法是{current_thinking_info}。如果有新的内容，记得转换话题\n"
+        # prompt += "-----------------------------------\n"
+        # prompt += f"现在你正在上网，和qq群里的网友们聊天，群里正在聊的话题是：{chat_observe_info}\n"
+        # prompt += f"你现在{mood_info}\n"
+        # prompt += f"你注意到{sender_name}刚刚说：{message_txt}\n"
+        # prompt += "现在你接下去继续思考，产生新的想法，不要分点输出，输出连贯的内心独白"
+        # prompt += "思考时可以想想如何对群聊内容进行回复。回复的要求是：平淡一些，简短一些，说中文，尽量不要说你说过的话\n"
+        # prompt += "请注意不要输出多余内容(包括前后缀，冒号和引号，括号， 表情，等)，不要带有括号和动作描写"
+        # prompt += f"记得结合上述的消息，生成内心想法，文字不要浮夸，注意你就是{self.bot_name}，{self.bot_name}指的就是你。"
+
+        prompt= (await global_prompt_manager.get_prompt_async("sub_heartflow_prompt_before")).format(
+            collected_info,relation_prompt_all,prompt_personality,current_thinking_info,chat_observe_info,mood_info,sender_name,
+            message_txt,self.bot_name
+        )

        try:
            response, reasoning_content = await self.llm_model.generate_response_async(prompt)
@@ -233,16 +271,20 @@ class SubHeartflow:
        reply_info = reply_content
        # schedule_info = bot_schedule.get_current_num_task(num=1, time_info=False)

-        prompt = ""
-        # prompt += f"你现在正在做的事情是：{schedule_info}\n"
-        prompt += f"{prompt_personality}\n"
-        prompt += f"现在你正在上网，和qq群里的网友们聊天，群里正在聊的话题是：{chat_observe_info}\n"
-        prompt += f"刚刚你的想法是{current_thinking_info}。"
-        prompt += f"你现在看到了网友们发的新消息:{message_new_info}\n"
-        prompt += f"你刚刚回复了群友们:{reply_info}"
-        prompt += f"你现在{mood_info}"
-        prompt += "现在你接下去继续思考，产生新的想法，记得保留你刚刚的想法，不要分点输出，输出连贯的内心独白"
-        prompt += "不要太长，但是记得结合上述的消息，要记得你的人设，关注聊天和新内容，关注你回复的内容，不要思考太多:"
+        # prompt = ""
+        # # prompt += f"你现在正在做的事情是：{schedule_info}\n"
+        # prompt += f"{prompt_personality}\n"
+        # prompt += f"现在你正在上网，和qq群里的网友们聊天，群里正在聊的话题是：{chat_observe_info}\n"
+        # prompt += f"刚刚你的想法是{current_thinking_info}。"
+        # prompt += f"你现在看到了网友们发的新消息:{message_new_info}\n"
+        # prompt += f"你刚刚回复了群友们:{reply_info}"
+        # prompt += f"你现在{mood_info}"
+        # prompt += "现在你接下去继续思考，产生新的想法，记得保留你刚刚的想法，不要分点输出，输出连贯的内心独白"
+        # prompt += "不要太长，但是记得结合上述的消息，要记得你的人设，关注聊天和新内容，关注你回复的内容，不要思考太多:"
+        prompt=(await global_prompt_manager.get_prompt_async('sub_heartflow_prompt_after')).format(
+            prompt_personality,chat_observe_info,current_thinking_info,message_new_info,reply_info,mood_info
+        )
+        
        try:
            response, reasoning_content = await self.llm_model.generate_response_async(prompt)
        except Exception as e:
@@ -301,5 +343,5 @@ class SubHeartflow:
        self.past_mind.append(self.current_mind)
        self.current_mind = response

-
+init_prompt()
 # subheartflow = SubHeartflow()