🤖 自动格式化代码 [skip ci]

src/experimental/PFC/message_storage.py

@@ -1,5 +1,6 @@
 from abc import ABC, abstractmethod
 from typing import List, Dict, Any
+
 # from src.common.database.database import db  # Peewee db 导入
 from src.common.database.database_model import Messages  # Peewee Messages 模型导入
 from playhouse.shortcuts import model_to_dict  # 用于将模型实例转换为字典
@@ -53,20 +54,23 @@ class PeeweeMessageStorage(MessageStorage):
     """Peewee消息存储实现"""
 
     async def get_messages_after(self, chat_id: str, message_time: float) -> List[Dict[str, Any]]:
-        query = Messages.select().where(
-            (Messages.chat_id == chat_id) &
-            (Messages.time > message_time)
-        ).order_by(Messages.time.asc())
+        query = (
+            Messages.select()
+            .where((Messages.chat_id == chat_id) & (Messages.time > message_time))
+            .order_by(Messages.time.asc())
+        )
 
         # print(f"storage_check_message: {message_time}")
         messages_models = list(query)
         return [model_to_dict(msg) for msg in messages_models]
 
     async def get_messages_before(self, chat_id: str, time_point: float, limit: int = 5) -> List[Dict[str, Any]]:
-        query = Messages.select().where(
-            (Messages.chat_id == chat_id) &
-            (Messages.time < time_point)
-        ).order_by(Messages.time.desc()).limit(limit)
+        query = (
+            Messages.select()
+            .where((Messages.chat_id == chat_id) & (Messages.time < time_point))
+            .order_by(Messages.time.desc())
+            .limit(limit)
+        )
 
         messages_models = list(query)
         # 将消息按时间正序排列
@@ -74,10 +78,7 @@ class PeeweeMessageStorage(MessageStorage):
         return [model_to_dict(msg) for msg in messages_models]
 
     async def has_new_messages(self, chat_id: str, after_time: float) -> bool:
-        return Messages.select().where(
-            (Messages.chat_id == chat_id) &
-            (Messages.time > after_time)
-        ).exists()
+        return Messages.select().where((Messages.chat_id == chat_id) & (Messages.time > after_time)).exists()
 
 
 # # 创建一个内存消息存储实现，用于测试