From b998d7a05b8f5425ec8053e1faf4bf609e42e19a Mon Sep 17 00:00:00 2001
From: 114514 <2514624910@qq.com>
Date: Fri, 9 May 2025 17:52:40 +0800
Subject: [PATCH] fix

---
 src/plugins/PFC/pfc_utils.py       | 9 +++++++--
 src/plugins/PFC/reply_generator.py | 6 ++++++
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/plugins/PFC/pfc_utils.py b/src/plugins/PFC/pfc_utils.py
index 2441941f..5e89ee6e 100644
--- a/src/plugins/PFC/pfc_utils.py
+++ b/src/plugins/PFC/pfc_utils.py
@@ -2,6 +2,7 @@ import traceback
 import json
 import re
 import asyncio # 确保导入 asyncio
+import time
 from typing import Dict, Any, Optional, Tuple, List, Union # 确保导入这些类型
 
 from src.common.logger_manager import get_logger
@@ -29,7 +30,8 @@ logger = get_logger("pfc_utils")
 async def find_most_relevant_historical_message(
     chat_id: str,
     query_text: str,
-    similarity_threshold: float = 0.3 # 相似度阈值，可以根据效果调整
+    similarity_threshold: float = 0.3, # 相似度阈值，可以根据效果调整
+    exclude_recent_seconds: int = 300 # 新增参数：排除最近多少秒内的消息（例如5分钟）
 ) -> Optional[Dict[str, Any]]:
     """
     根据查询文本，在指定 chat_id 的历史消息中查找最相关的消息。
@@ -47,6 +49,8 @@ async def find_most_relevant_historical_message(
         logger.warning(f"[{chat_id}] (私聊历史)未能为查询文本 '{query_text[:50]}...' 生成嵌入向量。")
         return None
 
+    current_timestamp = time.time() # 获取当前时间戳
+
     pipeline = [
         {
             "$match": {
@@ -235,7 +239,8 @@ async def retrieve_contextual_info(
             most_relevant_message_doc = await find_most_relevant_historical_message(
                 chat_id=chat_id,
                 query_text=query_for_historical_chat,
-                similarity_threshold=0.5 # 你可以根据需要调整这个阈值
+                similarity_threshold=0.5, # 你可以根据需要调整这个阈值
+                exclude_recent_seconds=300
             )
             if most_relevant_message_doc:
                 anchor_id = most_relevant_message_doc.get("message_id")
diff --git a/src/plugins/PFC/reply_generator.py b/src/plugins/PFC/reply_generator.py
index 0a82bad4..2a0cda6d 100644
--- a/src/plugins/PFC/reply_generator.py
+++ b/src/plugins/PFC/reply_generator.py
@@ -60,6 +60,9 @@ PROMPT_DIRECT_REPLY = """
 {retrieved_knowledge_str}
 请你**记住上面的知识**，在回复中有可能会用到。
 
+你有以下记忆可供参考：
+{retrieved_global_memory_str}
+
 你还想到了一些你们之前的聊天记录：
 {retrieved_historical_chat_str}
 
@@ -99,6 +102,9 @@ PROMPT_SEND_NEW_MESSAGE = """
 {retrieved_knowledge_str}
 请你**记住上面的知识**，在发消息时有可能会用到。
 
+你有以下记忆可供参考：
+{retrieved_global_memory_str}
+
 你还想到了一些你们之前的聊天记录：
 {retrieved_historical_chat_str}