检查是否为冲突导致的bug

This commit is contained in:
meng_xi_pan
2025-04-02 04:20:53 +08:00
parent 5a83edb783
commit 59043abefc
9 changed files with 33 additions and 8 deletions

View File

@@ -17,6 +17,7 @@ from ...message import UserInfo, Seg
from src.heart_flow.heartflow import heartflow
from src.common.logger import get_module_logger, CHAT_STYLE_CONFIG, LogConfig
from ...chat.chat_stream import chat_manager
from ...person_info.relationship_manager import relationship_manager
# 定义日志配置
chat_config = LogConfig(
@@ -135,6 +136,14 @@ class ThinkFlowChat:
await heartflow.get_subheartflow(stream_id).do_thinking_after_reply(response_set, chat_talking_prompt)
async def _update_relationship(self, message, response_set):
"""更新关系"""
ori_response = ",".join(response_set)
stance, emotion = await self.gpt._get_emotion_tags(ori_response, message.processed_plain_text)
await relationship_manager.calculate_update_relationship_value(
chat_stream=message.chat_stream, label=emotion, stance=stance
)
async def process_message(self, message_data: str) -> None:
"""处理消息并生成回复"""
timing_results = {}
@@ -267,6 +276,12 @@ class ThinkFlowChat:
timer2 = time.time()
timing_results["更新心流"] = timer2 - timer1
# # 更新关系
# timer1 = time.time()
# await self._update_relationship(message, response_set)
# timer2 = time.time()
# timing_results["更新关系"] = timer2 - timer1
# 输出性能计时结果
if do_reply:
timing_str = " | ".join([f"{step}: {duration:.2f}" for step, duration in timing_results.items()])

View File

@@ -147,6 +147,8 @@ class ResponseGenerator:
- 严格基于文字直接表达的对立关系判断
"""
logger.info(prompt)
# 调用模型生成结果
result, _, _ = await self.model_sum.generate_response(prompt)
result = result.strip()

View File

@@ -6,10 +6,10 @@ from ...memory_system.Hippocampus import HippocampusManager
from ...moods.moods import MoodManager
from ...schedule.schedule_generator import bot_schedule
from ...config.config import global_config
from ...chat.utils import get_recent_group_detailed_plain_text
from ...chat.utils import get_recent_group_detailed_plain_text, get_recent_group_speaker
from ...chat.chat_stream import chat_manager
from src.common.logger import get_module_logger
from .relationship_manager import relationship_manager
from ...person_info.relationship_manager import relationship_manager
from src.heart_flow.heartflow import heartflow