Ruff Format

src/memory_system/memory_retrieval.py

@@ -2,7 +2,7 @@ import time
 import json
 import asyncio
 from datetime import datetime
-from typing import List, Dict, Any, Optional, Tuple, Callable, cast
+from typing import List, Dict, Any, Optional, Tuple, Callable
 from src.common.logger import get_logger
 from src.config.config import global_config, model_config
 from src.prompt.prompt_manager import prompt_manager

src/memory_system/retrieval_tools/query_chat_history.py

@@ -105,25 +105,27 @@ async def search_chat_history(
         # 检查参数
         if not keyword and not participant and not start_time and not end_time:
             return "未指定查询参数（需要提供keyword、participant、start_time或end_time之一）"
-        
+
         # 解析时间参数
         start_timestamp = None
         end_timestamp = None
-        
+
         if start_time:
             try:
                 from src.memory_system.memory_utils import parse_datetime_to_timestamp
+
                 start_timestamp = parse_datetime_to_timestamp(start_time)
             except ValueError as e:
                 return f"开始时间格式错误: {str(e)}，支持格式如：'2025-01-01' 或 '2025-01-01 12:00:00' 或 '2025/01/01'"
-        
+
         if end_time:
             try:
                 from src.memory_system.memory_utils import parse_datetime_to_timestamp
+
                 end_timestamp = parse_datetime_to_timestamp(end_time)
             except ValueError as e:
                 return f"结束时间格式错误: {str(e)}，支持格式如：'2025-01-01' 或 '2025-01-01 12:00:00' 或 '2025/01/01'"
-        
+
         # 验证时间范围
         if start_timestamp and end_timestamp and start_timestamp > end_timestamp:
             return "开始时间不能晚于结束时间"
@@ -158,23 +160,20 @@ async def search_chat_history(
                     f"search_chat_history 当前聊天流在黑名单中，强制使用本地查询，chat_id={chat_id}, keyword={keyword}, participant={participant}"
                 )
             query = ChatHistory.select().where(ChatHistory.chat_id == chat_id)
-        
+
         # 添加时间过滤条件
         if start_timestamp is not None and end_timestamp is not None:
             # 查询指定时间段内的记录（记录的时间范围与查询时间段有交集）
             # 记录的开始时间在查询时间段内，或记录的结束时间在查询时间段内，或记录完全包含查询时间段
             query = query.where(
                 (
-                    (ChatHistory.start_time >= start_timestamp)
-                    & (ChatHistory.start_time <= end_timestamp)
+                    (ChatHistory.start_time >= start_timestamp) & (ChatHistory.start_time <= end_timestamp)
                 )  # 记录开始时间在查询时间段内
                 | (
-                    (ChatHistory.end_time >= start_timestamp)
-                    & (ChatHistory.end_time <= end_timestamp)
+                    (ChatHistory.end_time >= start_timestamp) & (ChatHistory.end_time <= end_timestamp)
                 )  # 记录结束时间在查询时间段内
                 | (
-                    (ChatHistory.start_time <= start_timestamp)
-                    & (ChatHistory.end_time >= end_timestamp)
+                    (ChatHistory.start_time <= start_timestamp) & (ChatHistory.end_time >= end_timestamp)
                 )  # 记录完全包含查询时间段
             )
             logger.debug(
@@ -302,7 +301,7 @@ async def search_chat_history(
                     time_desc = f"时间<='{end_str}'"
                 if time_desc:
                     conditions.append(time_desc)
-            
+
             if conditions:
                 conditions_str = "且".join(conditions)
                 return f"未找到满足条件（{conditions_str}）的聊天记录"

src/memory_system/retrieval_tools/query_words.py

@@ -30,7 +30,7 @@ async def query_words(chat_id: str, words: str) -> str:
             if separator in words:
                 words_list = [w.strip() for w in words.split(separator) if w.strip()]
                 break
-        
+
         # 如果没有找到分隔符，整个字符串作为一个词语
         if not words_list:
             words_list = [words.strip()]
@@ -76,4 +76,3 @@ def register_tool():
         ],
         execute_func=query_words,
     )
-