mirror of https://github.com/Mai-with-u/MaiBot.git
再次邪恶直推
114514
parent
0ed2e22360
commit
34323b65a2
|
|
@ -101,6 +101,20 @@ class ChatObserver:
|
|||
message: 消息数据
|
||||
"""
|
||||
try:
|
||||
if isinstance(message, dict): # 确保是字典才添加
|
||||
self.message_cache.append(message)
|
||||
# 可选：限制 message_cache 的大小，例如只保留最近 N 条
|
||||
# 你可以根据你的需求调整 MAX_CACHE_SIZE
|
||||
# 对于情绪判断，可能不需要太长的历史，例如 5-10 条可能就足够了
|
||||
# 但 ChatObserver 的 get_cached_messages 也可能被其他地方用于获取更长的历史
|
||||
# 所以这里的 MAX_CACHE_SIZE 需要权衡，或者让调用者自己决定 limit
|
||||
MAX_CACHE_SIZE = 30 # 例如，保留最近30条作为通用缓存
|
||||
if len(self.message_cache) > MAX_CACHE_SIZE:
|
||||
self.message_cache = self.message_cache[-MAX_CACHE_SIZE:]
|
||||
logger.debug(f"[私聊][{self.private_name}] 消息已添加到 ChatObserver 缓存，当前缓存大小: {len(self.message_cache)}")
|
||||
else:
|
||||
logger.warning(f"[私聊][{self.private_name}] 尝试向 message_cache 添加非字典类型消息: {type(message)}")
|
||||
|
||||
# 发送新消息通知
|
||||
notification = create_new_message_notification(
|
||||
sender="chat_observer", target="observation_info", message=message
|
||||
|
|
|
|||
|
|
@ -26,7 +26,7 @@ from ...config.config import global_config
|
|||
# 导入用户信息
|
||||
from ..person_info.person_info import person_info_manager
|
||||
# 导入关系
|
||||
from ..person_info.relationship_manager import relationship_manager
|
||||
from .pfc_relationship_translator import translate_relationship_value_to_text
|
||||
# 导入情绪
|
||||
from ..moods.moods import MoodManager
|
||||
|
||||
|
|
@ -93,7 +93,7 @@ class Conversation:
|
|||
|
||||
# --- 新增：初始化管理器实例 ---
|
||||
self.person_info_mng = person_info_manager
|
||||
self.relationship_mng = relationship_manager
|
||||
self.relationship_mng = translate_relationship_value_to_text
|
||||
self.mood_mng = MoodManager.get_instance() # MoodManager 是单例
|
||||
|
||||
# 初始化所有核心组件为 None，将在 _initialize 中创建
|
||||
|
|
@ -281,6 +281,27 @@ class Conversation:
|
|||
else:
|
||||
logger.warning(f"[私聊][{self.private_name}] MoodManager 未能启动，相关功能可能受限。")
|
||||
|
||||
# --- 在初始化完成前，尝试加载一次关系和情绪信息 ---
|
||||
if self.conversation_info and self.conversation_info.person_id and self.relationship_mng:
|
||||
try:
|
||||
self.conversation_info.relationship_text = await self.relationship_mng.build_relationship_info(
|
||||
self.conversation_info.person_id,
|
||||
is_id=True
|
||||
)
|
||||
logger.info(f"[私聊][{self.private_name}] 初始化时加载关系文本: {self.conversation_info.relationship_text}")
|
||||
except Exception as e_init_rel:
|
||||
logger.error(f"[私聊][{self.private_name}] 初始化时加载关系文本出错: {e_init_rel}")
|
||||
# 保留默认值
|
||||
|
||||
if self.conversation_info and self.mood_mng:
|
||||
try:
|
||||
self.conversation_info.current_emotion_text = self.mood_mng.get_prompt()
|
||||
logger.info(f"[私聊][{self.private_name}] 初始化时加载情绪文本: {self.conversation_info.current_emotion_text}")
|
||||
except Exception as e_init_emo:
|
||||
logger.error(f"[私聊][{self.private_name}] 初始化时加载情绪文本出错: {e_init_emo}")
|
||||
# 保留默认值
|
||||
|
||||
|
||||
# 6. 标记初始化成功并设置运行状态
|
||||
self._initialized = True
|
||||
self.should_continue = True # 初始化成功，标记可以继续运行循环
|
||||
|
|
@ -428,6 +449,22 @@ class Conversation:
|
|||
logger.info(f"[私聊][{self.private_name}] 正在停止对话实例: {self.stream_id}")
|
||||
self.should_continue = False # 设置标志，让主循环退出
|
||||
|
||||
# --- 新增：在对话结束时调用最终关系更新 ---
|
||||
if self._initialized and self.relationship_updater and self.conversation_info and self.observation_info and self.chat_observer:
|
||||
try:
|
||||
logger.info(f"[私聊][{self.private_name}] 准备执行最终关系评估...")
|
||||
await self.relationship_updater.update_relationship_final(
|
||||
conversation_info=self.conversation_info,
|
||||
observation_info=self.observation_info,
|
||||
chat_observer_for_history=self.chat_observer
|
||||
)
|
||||
logger.info(f"[私聊][{self.private_name}] 最终关系评估已调用。")
|
||||
except Exception as e_final_rel:
|
||||
logger.error(f"[私聊][{self.private_name}] 调用最终关系评估时出错: {e_final_rel}")
|
||||
logger.error(traceback.format_exc())
|
||||
else:
|
||||
logger.warning(f"[私聊][{self.private_name}] 跳过最终关系评估，因为实例未完全初始化或缺少必要组件。")
|
||||
|
||||
# 停止空闲对话检测器
|
||||
if self.idle_conversation_starter:
|
||||
self.idle_conversation_starter.stop()
|
||||
|
|
@ -515,46 +552,72 @@ class Conversation:
|
|||
try:
|
||||
if self.conversation_info and self._initialized: # 确保 conversation_info 和实例已初始化
|
||||
# 更新关系文本
|
||||
if self.conversation_info.person_id and self.observation_info and self.observation_info.sender_platform and self.observation_info.sender_user_id:
|
||||
if self.conversation_info.person_id: # 确保 person_id 已获取
|
||||
try:
|
||||
# relationship_manager.build_relationship_info 需要一个 (platform, user_id, user_nickname) 格式的元组
|
||||
# 注意 user_id 应为 RelationshipManager 所期望的类型（通常是原始的，可能是字符串或数字，检查其内部实现）
|
||||
# person_info.py 中 get_person_id 接收 int，但 build_relationship_info 的 person 参数更灵活
|
||||
# 这里我们假设 observation_info.sender_user_id 是字符串形式的 user_id
|
||||
# 如果 RelationshipManager 内部的 get_person_id 需要 int，这里可能需要转换
|
||||
# 但 RelationshipManager 的 build_relationship_info 方法第一个参数 person 可以直接是 person_id (is_id=True)
|
||||
# 或者是一个 (platform, user_id, nickname) 元组 (is_id=False)
|
||||
# 为了安全，我们用 person_id
|
||||
self.conversation_info.relationship_text = await self.relationship_mng.build_relationship_info(
|
||||
# 1.直接从 person_info_manager 获取数值型的 relationship_value
|
||||
numeric_relationship_value = await self.person_info_mng.get_value(
|
||||
self.conversation_info.person_id,
|
||||
is_id=True # 明确告知是 person_id
|
||||
"relationship_value"
|
||||
)
|
||||
logger.debug(f"[私聊][{self.private_name}] 更新关系文本: {self.conversation_info.relationship_text}")
|
||||
|
||||
# 确保 relationship_value 是浮点数 (可以复用 relationship_manager 中的 ensure_float，或者在这里简单处理)
|
||||
if not isinstance(numeric_relationship_value, (int, float)):
|
||||
# 尝试从 Decimal128 转换 (如果你的数据库用的是这个)
|
||||
from bson.decimal128 import Decimal128
|
||||
if isinstance(numeric_relationship_value, Decimal128):
|
||||
numeric_relationship_value = float(numeric_relationship_value.to_decimal())
|
||||
else: # 其他类型，或转换失败，给默认值
|
||||
logger.warning(f"[私聊][{self.private_name}] 获取的 relationship_value 类型未知 ({type(numeric_relationship_value)}) 或转换失败，默认为0.0")
|
||||
numeric_relationship_value = 0.0
|
||||
|
||||
logger.debug(f"[私聊][{self.private_name}] 获取到数值型关系值: {numeric_relationship_value}")
|
||||
|
||||
# 2. 使用PFC内部的翻译函数将其转换为文本描述
|
||||
simplified_relationship_text = translate_relationship_value_to_text(numeric_relationship_value)
|
||||
self.conversation_info.relationship_text = simplified_relationship_text
|
||||
|
||||
logger.debug(f"[私聊][{self.private_name}] 更新后关系文本 (PFC内部翻译): {self.conversation_info.relationship_text}")
|
||||
|
||||
except Exception as e_rel:
|
||||
logger.error(f"[私聊][{self.private_name}] 更新关系文本时出错: {e_rel}")
|
||||
self.conversation_info.relationship_text = "我们之间的关系有点微妙。" # 出错时的默认值
|
||||
elif not self.conversation_info.person_id and self.observation_info and self.observation_info.sender_user_id and self.observation_info.sender_platform:
|
||||
# 如果 person_id 之前没获取到，在这里尝试再次获取
|
||||
logger.error(f"[私聊][{self.private_name}] 更新关系文本(PFC内部翻译)时出错: {e_rel}")
|
||||
self.conversation_info.relationship_text = "你们的关系是：普通。" # 出错时的默认值
|
||||
elif self.observation_info and self.observation_info.sender_user_id and self.observation_info.sender_platform:
|
||||
# 如果 person_id 之前没获取到，在这里尝试再次获取 (这部分逻辑保持，因为 person_id 是必须的)
|
||||
try:
|
||||
private_user_id_int = int(self.observation_info.sender_user_id)
|
||||
self.conversation_info.person_id = self.person_info_mng.get_person_id(
|
||||
self.observation_info.sender_platform,
|
||||
private_user_id_int
|
||||
)
|
||||
await self.person_info_mng.get_or_create_person(
|
||||
await self.person_info_mng.get_or_create_person( #确保用户存在
|
||||
platform=self.observation_info.sender_platform,
|
||||
user_id=private_user_id_int,
|
||||
nickname=self.observation_info.sender_name if self.observation_info.sender_name else "未知用户",
|
||||
)
|
||||
if self.conversation_info.person_id: # 再次尝试更新关系文本
|
||||
self.conversation_info.relationship_text = await self.relationship_mng.build_relationship_info(
|
||||
self.conversation_info.person_id, is_id=True
|
||||
)
|
||||
if self.conversation_info.person_id: # 如果成功获取 person_id，则再次尝试更新关系文本
|
||||
numeric_relationship_value = await self.person_info_mng.get_value(
|
||||
self.conversation_info.person_id,
|
||||
"relationship_value"
|
||||
)
|
||||
if not isinstance(numeric_relationship_value, (int, float)):
|
||||
from bson.decimal128 import Decimal128
|
||||
if isinstance(numeric_relationship_value, Decimal128):
|
||||
numeric_relationship_value = float(numeric_relationship_value.to_decimal())
|
||||
else:
|
||||
numeric_relationship_value = 0.0
|
||||
self.conversation_info.relationship_text = translate_relationship_value_to_text(numeric_relationship_value)
|
||||
logger.debug(f"[私聊][{self.private_name}] (备用逻辑)更新后关系文本: {self.conversation_info.relationship_text}")
|
||||
|
||||
except ValueError:
|
||||
logger.error(f"[私聊][{self.private_name}] 循环中无法将 sender_user_id ('{self.observation_info.sender_user_id}') 转换为整数。")
|
||||
self.conversation_info.relationship_text = "你们的关系是：普通。"
|
||||
except Exception as e_pid_loop:
|
||||
logger.error(f"[私聊][{self.private_name}] 循环中获取 person_id 时出错: {e_pid_loop}")
|
||||
|
||||
logger.error(f"[私聊][{self.private_name}] 循环中获取 person_id 并更新关系时出错: {e_pid_loop}")
|
||||
self.conversation_info.relationship_text = "你们的关系是：普通。"
|
||||
else:
|
||||
# 如果 person_id 仍无法确定
|
||||
self.conversation_info.relationship_text = "你们的关系是：普通。" # 或 "你们的关系是：未知。"
|
||||
|
||||
# 更新情绪文本
|
||||
if self.mood_mng:
|
||||
self.conversation_info.current_emotion_text = self.mood_mng.get_prompt()
|
||||
|
|
|
|||
|
|
@ -0,0 +1,60 @@
|
|||
# 直接完整导入群聊的relationship_manager.py可能不可取，因为对于PFC的planner来说，其暗示了选择回复，所以新建代码文件来适配PFC的决策层面
|
||||
from src.common.logger_manager import get_logger
|
||||
|
||||
logger = get_logger("pfc_relationship_translator")
|
||||
|
||||
# 这个函数复用 relationship_manager.py 中的 calculate_level_num 的逻辑
|
||||
def _calculate_relationship_level_num(relationship_value: float) -> int:
|
||||
"""
|
||||
根据关系值计算关系等级编号 (0-5)。
|
||||
这里的阈值应与 relationship_manager.py 中的保持一致
|
||||
"""
|
||||
if not isinstance(relationship_value, (int, float)):
|
||||
logger.warning(f"传入的 relationship_value '{relationship_value}' 不是有效的数值类型，默认为0。")
|
||||
relationship_value = 0.0
|
||||
|
||||
if -1000 <= relationship_value < -227:
|
||||
level_num = 0 # 厌恶
|
||||
elif -227 <= relationship_value < -73:
|
||||
level_num = 1 # 冷漠
|
||||
elif -73 <= relationship_value < 227:
|
||||
level_num = 2 # 普通/认识
|
||||
elif 227 <= relationship_value < 587:
|
||||
level_num = 3 # 友好
|
||||
elif 587 <= relationship_value < 900:
|
||||
level_num = 4 # 喜欢
|
||||
elif 900 <= relationship_value <= 1000:
|
||||
level_num = 5 # 暧昧
|
||||
else:
|
||||
# 超出范围的值处理
|
||||
if relationship_value > 1000:
|
||||
level_num = 5
|
||||
elif relationship_value < -1000:
|
||||
level_num = 0
|
||||
else: # 理论上不会到这里，除非前面的条件逻辑有误
|
||||
logger.warning(f"关系值 {relationship_value} 未落入任何预设范围，默认为普通。")
|
||||
level_num = 2
|
||||
return level_num
|
||||
|
||||
def translate_relationship_value_to_text(relationship_value: float) -> str:
|
||||
"""
|
||||
将数值型的关系值转换为PFC私聊场景下简洁的关系描述文本。
|
||||
"""
|
||||
level_num = _calculate_relationship_level_num(relationship_value)
|
||||
|
||||
relationship_descriptions = [
|
||||
"厌恶", # level_num 0
|
||||
"冷漠", # level_num 1
|
||||
"初识", # level_num 2
|
||||
"友好", # level_num 3
|
||||
"喜欢", # level_num 4
|
||||
"暧昧" # level_num 5
|
||||
]
|
||||
|
||||
if 0 <= level_num < len(relationship_descriptions):
|
||||
description = relationship_descriptions[level_num]
|
||||
else:
|
||||
description = "普通" # 默认或错误情况
|
||||
logger.warning(f"计算出的 level_num ({level_num}) 无效，关系描述默认为 '普通'")
|
||||
|
||||
return f"你们的关系是：{description}。"
|
||||
Loading…
Reference in New Issue