From 502f509630e4f073ff6b8d3db582d34cc2cd3ba3 Mon Sep 17 00:00:00 2001
From: SengokuCola <1026294844@qq.com>
Date: Fri, 2 May 2025 19:15:15 +0800
Subject: [PATCH] =?UTF-8?q?fix=EF=BC=9A=E4=BF=AE=E5=A4=8Drelationship?=
 =?UTF-8?q?=E5=8A=A0=E9=94=99=E4=BA=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/plugins/chat/message_sender.py              |  6 +++---
 src/plugins/heartFC_chat/normal_chat.py         |  5 ++++-
 src/plugins/person_info/relationship_manager.py | 15 +++++++--------
 3 files changed, 14 insertions(+), 12 deletions(-)

diff --git a/src/plugins/chat/message_sender.py b/src/plugins/chat/message_sender.py
index 61e2dd49c..c50c7aad3 100644
--- a/src/plugins/chat/message_sender.py
+++ b/src/plugins/chat/message_sender.py
@@ -216,9 +216,9 @@ class MessageManager:
             # print(f"message.reply:{message.reply}")
 
             # --- 条件应用 set_reply 逻辑 ---
-            logger.debug(
-                f"[message.apply_set_reply_logic:{message.apply_set_reply_logic},message.is_head:{message.is_head},thinking_messages_count:{thinking_messages_count},thinking_messages_length:{thinking_messages_length},message.is_private_message():{message.is_private_message()}]"
-            )
+            # logger.debug(
+            #     f"[message.apply_set_reply_logic:{message.apply_set_reply_logic},message.is_head:{message.is_head},thinking_messages_count:{thinking_messages_count},thinking_messages_length:{thinking_messages_length},message.is_private_message():{message.is_private_message()}]"
+            # )
             if (
                 message.apply_set_reply_logic  # 检查标记
                 and message.is_head
diff --git a/src/plugins/heartFC_chat/normal_chat.py b/src/plugins/heartFC_chat/normal_chat.py
index 70568f835..1c1372c52 100644
--- a/src/plugins/heartFC_chat/normal_chat.py
+++ b/src/plugins/heartFC_chat/normal_chat.py
@@ -178,8 +178,11 @@ class NormalChat:
         """更新关系情绪"""
         ori_response = ",".join(response_set)
         stance, emotion = await self.gpt._get_emotion_tags(ori_response, message.processed_plain_text)
+        user_info = message.message_info.user_info
+        platform = user_info.platform
         await relationship_manager.calculate_update_relationship_value(
-            chat_stream=self.chat_stream,
+            user_info,
+            platform,
             label=emotion,
             stance=stance,  # 使用 self.chat_stream
         )
diff --git a/src/plugins/person_info/relationship_manager.py b/src/plugins/person_info/relationship_manager.py
index fc8cf548e..3c264b054 100644
--- a/src/plugins/person_info/relationship_manager.py
+++ b/src/plugins/person_info/relationship_manager.py
@@ -5,6 +5,7 @@ from bson.decimal128 import Decimal128
 from .person_info import person_info_manager
 import time
 import random
+from maim_message import UserInfo, Seg
 # import re
 # import traceback
 
@@ -102,7 +103,7 @@ class RelationshipManager:
         # await person_info_manager.update_one_field(person_id, "user_avatar", user_avatar)
         await person_info_manager.qv_person_name(person_id, user_nickname, user_cardname, user_avatar)
 
-    async def calculate_update_relationship_value(self, chat_stream: ChatStream, label: str, stance: str) -> tuple:
+    async def calculate_update_relationship_value(self, user_info: UserInfo, platform: str, label: str, stance: str):
         """计算并变更关系值
         新的关系值变更计算方式：
             将关系值限定在-1000到1000
@@ -134,11 +135,11 @@ class RelationshipManager:
             "困惑": 0.5,
         }
 
-        person_id = person_info_manager.get_person_id(chat_stream.user_info.platform, chat_stream.user_info.user_id)
+        person_id = person_info_manager.get_person_id(platform, user_info.user_id)
         data = {
-            "platform": chat_stream.user_info.platform,
-            "user_id": chat_stream.user_info.user_id,
-            "nickname": chat_stream.user_info.user_nickname,
+            "platform": platform,
+            "user_id": user_info.user_id,
+            "nickname": user_info.user_nickname,
             "konw_time": int(time.time()),
         }
         old_value = await person_info_manager.get_value(person_id, "relationship_value")
@@ -178,7 +179,7 @@ class RelationshipManager:
         level_num = self.calculate_level_num(old_value + value)
         relationship_level = ["厌恶", "冷漠", "一般", "友好", "喜欢", "暧昧"]
         logger.info(
-            f"用户: {chat_stream.user_info.user_nickname}"
+            f"用户: {user_info.user_nickname}"
             f"当前关系: {relationship_level[level_num]}, "
             f"关系值: {old_value:.2f}, "
             f"当前立场情感: {stance}-{label}, "
@@ -187,8 +188,6 @@ class RelationshipManager:
 
         await person_info_manager.update_one_field(person_id, "relationship_value", old_value + value, data)
 
-        return chat_stream.user_info.user_nickname, value, relationship_level[level_num]
-
     async def calculate_update_relationship_value_with_reason(
         self, chat_stream: ChatStream, label: str, stance: str, reason: str
     ) -> tuple: