From 2a5184ba46a0cb9b50f9ca63487d344956c151a8 Mon Sep 17 00:00:00 2001 From: 114514 <2514624910@qq.com> Date: Sat, 26 Apr 2025 14:24:48 +0800 Subject: [PATCH] =?UTF-8?q?=E8=AF=95=E5=9B=BE=E4=BF=AE=E5=A4=8DPFC?= =?UTF-8?q?=E5=BC=80=E5=A7=8B=E5=AF=B9=E8=AF=9D=E5=AE=9E=E4=BE=8B=E6=97=B6?= =?UTF-8?q?=E8=BD=BD=E5=85=A5=E8=81=8A=E5=A4=A9=E8=AE=B0=E5=BD=95=E7=BC=BA?= =?UTF-8?q?=E5=A4=B1=E7=9A=84bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 开始对话示例时出现神秘bug导致聊天记录缺失(只有对方的没有bot的),经过高人指点将提取聊天记录方式改用chat_message_builder中的函数而不是storage --- src/plugins/PFC/conversation.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/src/plugins/PFC/conversation.py b/src/plugins/PFC/conversation.py index 39ebccc1d..c290008b8 100644 --- a/src/plugins/PFC/conversation.py +++ b/src/plugins/PFC/conversation.py @@ -1,7 +1,8 @@ import time import asyncio import datetime -from .message_storage import MongoDBMessageStorage +# from .message_storage import MongoDBMessageStorage +from src.plugins.utils.chat_message_builder import get_raw_msg_before_timestamp_with_chat from ...config.config import global_config from typing import Dict, Any from ..chat.message import Message @@ -75,13 +76,10 @@ class Conversation: raise try: logger.info(f"为 {self.stream_id} 加载初始聊天记录...") - storage = MongoDBMessageStorage() # 创建存储实例 - # 获取当前时间点之前最多 N 条消息 (比如 30 条) - # get_messages_before 返回的是按时间正序排列的列表 - initial_messages = await storage.get_messages_before( + initial_messages = await get_raw_msg_before_timestamp_with_chat( # chat_id=self.stream_id, - time_point=time.time(), - limit=30, # 加载最近20条作为初始上下文,可以调整 + timestamp=time.time(), + limit=30, # 加载最近30条作为初始上下文,可以调整 ) if initial_messages: # 将加载的消息填充到 ObservationInfo 的 chat_history