Merge pull request #44 from tcmofashi/feature

fix: 解决反斜杠，toml加入人格

config/bot_config_template.toml

@@ -2,6 +2,13 @@
 qq = 123
 nickname = "麦麦"
 
+[personality]
+prompt_personality = [
+        "曾经是一个学习地质的女大学生，现在学习心理学和脑科学，你会刷贴吧",    # 贴吧人格 
+        "是一个女大学生，你有黑色头发，你会刷小红书"    # 小红书人格
+    ]
+prompt_schedule = "一个曾经学习地质,现在学习心理学和脑科学的女大学生，喜欢刷qq，贴吧，知乎和小红书"
+
 [message]
 min_text_length = 2 # 与麦麦聊天时麦麦只会回答文本大于等于此数的消息
 max_context_size = 15 # 麦麦获得的上下文数量，超出数量后自动丢弃

src/plugins/chat/config.py

@@ -76,6 +76,13 @@ class BotConfig:
         if os.path.exists(config_path):
             with open(config_path, "rb") as f:
                 toml_dict = tomli.load(f)
+            
+            if 'personality' in toml_dict:
+                personality_config=toml_dict['personality']
+                personality=personality_config.get('prompt_personality')
+                if len(personality) >= 2:
+                    config.PROMPT_PERSONALITY=personality_config.get('prompt_personality')
+                config.PROMPT_SCHEDULE_GEN=personality_config.get('prompt_schedule')
 
             if "emoji" in toml_dict:
                 emoji_config = toml_dict["emoji"]

src/plugins/chat/prompt_builder.py

@@ -244,14 +244,12 @@ class PromptBuilder:
     
     def _build_initiative_prompt_check(self,selected_node,prompt_regular):
         memory=random.sample(selected_node['memory_items'],3)
-        prompt_for_check=f"{prompt_regular}你现在想在群里发言，回忆了一下，想到一个话题,是\"{selected_node['concept']}\"，关于这个话题的记忆有\n{'\n'.join(memory)}\n，以这个作为主题发言合适吗？\
-            请在把握群里的聊天内容的基础上，综合群内的氛围，如果认为应该发言请输出yes，否则输出no，请注意是决定是否需要发言，而不是编写回复内容，除了yes和no不要输出任何回复内容。"
+        memory='\n'.join(memory)
+        prompt_for_check=f"{prompt_regular}你现在想在群里发言，回忆了一下，想到一个话题,是{selected_node['concept']}，关于这个话题的记忆有\n{memory}\n，以这个作为主题发言合适吗？请在把握群里的聊天内容的基础上，综合群内的氛围，如果认为应该发言请输出yes，否则输出no，请注意是决定是否需要发言，而不是编写回复内容，除了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']}\"，关于这个话题的记忆有\n{'\n'.join(memory)}\n，\
-        请在把握群里的聊天内容的基础上，综合群内的氛围，以日常且口语化的口吻，简短且随意一点进行发言，不要说的太有条理，可以有个性。记住不要输出多余内容(包括前后缀，冒号和引号，\
-            括号，表情等)"
+        prompt_for_initiative=f"{prompt_regular}你现在想在群里发言，回忆了一下，想到一个话题,是{selected_node['concept']}，关于这个话题的记忆有\n{memory}\n，请在把握群里的聊天内容的基础上，综合群内的氛围，以日常且口语化的口吻，简短且随意一点进行发言，不要说的太有条理，可以有个性。记住不要输出多余内容(包括前后缀，冒号和引号，括号，表情等)"
         return prompt_for_initiative