feat：新的人格结构，完成20%

2025-04-06 11:00:59 +08:00
parent 851afc3bbf
commit 594c30ab74
20 changed files with 629 additions and 614 deletions
--- a/src/plugins/chat_module/reasoning_chat/reasoning_prompt_builder.py
+++ b/src/plugins/chat_module/reasoning_chat/reasoning_prompt_builder.py
@@ -1,14 +1,13 @@
 import random
 import time
 from typing import Optional, Union
-import re
-import jieba
 import numpy as np

 from ....common.database import db
 from ...chat.utils import get_embedding, get_recent_group_detailed_plain_text, get_recent_group_speaker
 from ...chat.chat_stream import chat_manager
 from ...moods.moods import MoodManager
+from ....individuality.individuality import Individuality
 from ...memory_system.Hippocampus import HippocampusManager
 from ...schedule.schedule_generator import bot_schedule
 from ...config.config import global_config
@@ -28,7 +27,23 @@ class PromptBuilder:
    ) -> tuple[str, str]:
    
        # 开始构建prompt
-
+        prompt_personality = "你"
+        #person
+        individuality = Individuality.get_instance()
+        
+        personality_core = individuality.personality.personality_core
+        prompt_personality += personality_core
+        
+        personality_sides = individuality.personality.personality_sides
+        random.shuffle(personality_sides)
+        prompt_personality += f",{personality_sides[0]}"
+        
+        identity_detail = individuality.identity.identity_detail
+        random.shuffle(identity_detail)
+        prompt_personality += f",{identity_detail[0]}"
+        
+        
+            
        # 关系
        who_chat_in_group = [(chat_stream.user_info.platform, 
                              chat_stream.user_info.user_id, 
@@ -105,20 +120,6 @@ class PromptBuilder:
                    )
                    keywords_reaction_prompt += rule.get("reaction", "") + "，"

-        # 人格选择
-        personality = global_config.PROMPT_PERSONALITY
-        probability_1 = global_config.PERSONALITY_1
-        probability_2 = global_config.PERSONALITY_2
-
-        personality_choice = random.random()
-
-        if personality_choice < probability_1:  # 第一种风格
-            prompt_personality = personality[0]
-        elif personality_choice < probability_1 + probability_2:  # 第二种风格
-            prompt_personality = personality[1]
-        else:  # 第三种人格
-            prompt_personality = personality[2]
-
        # 中文高手(新加的好玩功能)
        prompt_ger = ""
        if random.random() < 0.04:
--- 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
@@ -10,7 +10,7 @@ from ...chat.utils import get_recent_group_detailed_plain_text, get_recent_group
 from ...chat.chat_stream import chat_manager
 from src.common.logger import get_module_logger
 from ...person_info.relationship_manager import relationship_manager
-
+from ....individuality.individuality import Individuality
 from src.heart_flow.heartflow import heartflow

 logger = get_module_logger("prompt")
@@ -28,6 +28,20 @@ class PromptBuilder:
        current_mind_info = heartflow.get_subheartflow(stream_id).current_mind

        # 开始构建prompt
+        prompt_personality = "你"
+        #person
+        individuality = Individuality.get_instance()
+        
+        personality_core = individuality.personality.personality_core
+        prompt_personality += personality_core
+        
+        personality_sides = individuality.personality.personality_sides
+        random.shuffle(personality_sides)
+        prompt_personality += f",{personality_sides[0]}"
+        
+        identity_detail = individuality.identity.identity_detail
+        random.shuffle(identity_detail)
+        prompt_personality += f",{identity_detail[0]}"

        # 关系
        who_chat_in_group = [(chat_stream.user_info.platform, 
@@ -90,20 +104,6 @@ class PromptBuilder:
                    )
                    keywords_reaction_prompt += rule.get("reaction", "") + "，"

-        # 人格选择
-        personality = global_config.PROMPT_PERSONALITY
-        probability_1 = global_config.PERSONALITY_1
-        probability_2 = global_config.PERSONALITY_2
-
-        personality_choice = random.random()
-
-        if personality_choice < probability_1:  # 第一种风格
-            prompt_personality = personality[0]
-        elif personality_choice < probability_1 + probability_2:  # 第二种风格
-            prompt_personality = personality[1]
-        else:  # 第三种人格
-            prompt_personality = personality[2]
-
        # 中文高手(新加的好玩功能)
        prompt_ger = ""
        if random.random() < 0.04:
@@ -133,73 +133,5 @@ class PromptBuilder:
        
        return prompt

-    def _build_initiative_prompt_select(self, group_id, probability_1=0.8, probability_2=0.1):
-        current_date = time.strftime("%Y-%m-%d", time.localtime())
-        current_time = time.strftime("%H:%M:%S", time.localtime())
-        bot_schedule_now_time, bot_schedule_now_activity = bot_schedule.get_current_task()
-        prompt_date = f"""今天是{current_date}，现在是{current_time}，你今天的日程是：
-{bot_schedule.today_schedule}
-你现在正在{bot_schedule_now_activity}
-"""
-
-        chat_talking_prompt = ""
-        if group_id:
-            chat_talking_prompt = get_recent_group_detailed_plain_text(
-                group_id, limit=global_config.MAX_CONTEXT_SIZE, combine=True
-            )
-
-        chat_talking_prompt = f"以下是群里正在聊天的内容：\n{chat_talking_prompt}"
-        # print(f"\033[1;34m[调试]\033[0m 已从数据库获取群 {group_id} 的消息记录:{chat_talking_prompt}")
-
-        # 获取主动发言的话题
-        all_nodes = HippocampusManager.get_instance().memory_graph.dots
-        all_nodes = filter(lambda dot: len(dot[1]["memory_items"]) > 3, all_nodes)
-        nodes_for_select = random.sample(all_nodes, 5)
-        topics = [info[0] for info in nodes_for_select]
-
-        # 激活prompt构建
-        activate_prompt = ""
-        activate_prompt = "以上是群里正在进行的聊天。"
-        personality = global_config.PROMPT_PERSONALITY
-        prompt_personality = ""
-        personality_choice = random.random()
-        if personality_choice < probability_1:  # 第一种人格
-            prompt_personality = f"""{activate_prompt}你的网名叫{global_config.BOT_NICKNAME}，{personality[0]}"""
-        elif personality_choice < probability_1 + probability_2:  # 第二种人格
-            prompt_personality = f"""{activate_prompt}你的网名叫{global_config.BOT_NICKNAME}，{personality[1]}"""
-        else:  # 第三种人格
-            prompt_personality = f"""{activate_prompt}你的网名叫{global_config.BOT_NICKNAME}，{personality[2]}"""
-
-        topics_str = ",".join(f'"{topics}"')
-        prompt_for_select = (
-            f"你现在想在群里发言，回忆了一下，想到几个话题，分别是{topics_str}，综合当前状态以及群内气氛，"
-            f"请你在其中选择一个合适的话题，注意只需要输出话题，除了话题什么也不要输出(双引号也不要输出)"
-        )
-
-        prompt_initiative_select = f"{prompt_date}\n{prompt_personality}\n{prompt_for_select}"
-        prompt_regular = f"{prompt_date}\n{prompt_personality}"
-
-        return prompt_initiative_select, nodes_for_select, prompt_regular
-
-    def _build_initiative_prompt_check(self, selected_node, prompt_regular):
-        memory = random.sample(selected_node["memory_items"], 3)
-        memory = "\n".join(memory)
-        prompt_for_check = (
-            f"{prompt_regular}你现在想在群里发言，回忆了一下，想到一个话题,是{selected_node['concept']}，"
-            f"关于这个话题的记忆有\n{memory}\n，以这个作为主题发言合适吗？请在把握群里的聊天内容的基础上，"
-            f"综合群内的氛围，如果认为应该发言请输出yes，否则输出no，请注意是决定是否需要发言，而不是编写回复内容，"
-            f"除了yes和no不要输出任何回复内容。"
-        )
-        return prompt_for_check, memory
-
-    def _build_initiative_prompt(self, selected_node, prompt_regular, memory):
-        prompt_for_initiative = (
-            f"{prompt_regular}你现在想在群里发言，回忆了一下，想到一个话题,是{selected_node['concept']}，"
-            f"关于这个话题的记忆有\n{memory}\n，请在把握群里的聊天内容的基础上，综合群内的氛围，"
-            f"以日常且口语化的口吻，简短且随意一点进行发言，不要说的太有条理，可以有个性。"
-            f"记住不要输出多余内容(包括前后缀，冒号和引号，括号，表情,@等)"
-        )
-        return prompt_for_initiative
-

 prompt_builder = PromptBuilder()