From 13c47d5d1282b7843ad3871a2a71b8bd36ac908e Mon Sep 17 00:00:00 2001 From: SengokuCola <1026294844@qq.com> Date: Wed, 2 Apr 2025 00:05:33 +0800 Subject: [PATCH] =?UTF-8?q?fix:=E4=B8=80=E4=BA=9B=E5=B0=8F=E4=BF=AE?= =?UTF-8?q?=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/heart_flow/observation.py | 7 +++++-- src/plugins/P.F.C/pfc.py | 3 +++ src/plugins/chat_module/reasoning_chat/reasoning_chat.py | 6 ++++-- src/plugins/chat_module/think_flow_chat/think_flow_chat.py | 7 ++++--- 4 files changed, 16 insertions(+), 7 deletions(-) create mode 100644 src/plugins/P.F.C/pfc.py diff --git a/src/heart_flow/observation.py b/src/heart_flow/observation.py index b2ad3ce6f..1a907229f 100644 --- a/src/heart_flow/observation.py +++ b/src/heart_flow/observation.py @@ -23,6 +23,8 @@ class ChattingObservation(Observation): self.talking_message = [] self.talking_message_str = "" + + self.personality_info = " ".join(global_config.PROMPT_PERSONALITY) self.observe_times = 0 @@ -112,10 +114,11 @@ class ChattingObservation(Observation): # 基于已经有的talking_summary，和新的talking_message，生成一个summary # print(f"更新聊天总结：{self.talking_summary}") prompt = "" - prompt = f"你正在参与一个qq群聊的讨论，你记得这个群之前在聊的内容是：{self.observe_info}\n" + prompt += f"你{self.personality_info}，请注意识别你自己的聊天发言" + prompt += f"你正在参与一个qq群聊的讨论，你记得这个群之前在聊的内容是：{self.observe_info}\n" prompt += f"现在群里的群友们产生了新的讨论，有了新的发言，具体内容如下：{new_messages_str}\n" prompt += """以上是群里在进行的聊天，请你对这个聊天内容进行总结，总结内容要包含聊天的大致内容， - 以及聊天中的一些重要信息，记得不要分点，不要太长，精简的概括成一段文本\n""" + 以及聊天中的一些重要信息，注意识别你自己的发言，记得不要分点，不要太长，精简的概括成一段文本\n""" prompt += "总结概括：" self.observe_info, reasoning_content = await self.llm_summary.generate_response_async(prompt) print(f"prompt：{prompt}") diff --git a/src/plugins/P.F.C/pfc.py b/src/plugins/P.F.C/pfc.py new file mode 100644 index 000000000..9b83bce40 --- /dev/null +++ b/src/plugins/P.F.C/pfc.py @@ -0,0 +1,3 @@ +#Programmable Friendly Conversationalist +#Prefrontal cortex + diff --git a/src/plugins/chat_module/reasoning_chat/reasoning_chat.py b/src/plugins/chat_module/reasoning_chat/reasoning_chat.py index ed7db2a2a..6ad043804 100644 --- a/src/plugins/chat_module/reasoning_chat/reasoning_chat.py +++ b/src/plugins/chat_module/reasoning_chat/reasoning_chat.py @@ -134,8 +134,10 @@ class ReasoningChat: messageinfo = message.message_info - if groupinfo.group_id not in global_config.talk_allowed_groups: - return + if groupinfo == None and global_config.enable_friend_chat:#如果是私聊 + pass + elif groupinfo.group_id not in global_config.talk_allowed_groups: + return # logger.info("使用推理聊天模式") diff --git a/src/plugins/chat_module/think_flow_chat/think_flow_chat.py b/src/plugins/chat_module/think_flow_chat/think_flow_chat.py index e2a96b985..f665d90fd 100644 --- a/src/plugins/chat_module/think_flow_chat/think_flow_chat.py +++ b/src/plugins/chat_module/think_flow_chat/think_flow_chat.py @@ -145,9 +145,10 @@ class ThinkFlowChat: userinfo = message.message_info.user_info messageinfo = message.message_info - if groupinfo.group_id not in global_config.talk_allowed_groups: - return - # logger.info("使用思维流聊天模式") + if groupinfo == None and global_config.enable_friend_chat:#如果是私聊 + pass + elif groupinfo.group_id not in global_config.talk_allowed_groups: + return # 创建聊天流 chat = await chat_manager.get_or_create_stream(