chore: code lint

Author: lss233

.pre-commit-config.yaml

@@ -0,0 +1,19 @@
+repos:
+  - repo: https://github.com/PyCQA/isort
+    rev: 6.0.0
+    hooks:
+      - id: isort
+        name: isort (python3)
+        language_version: python3
+        args: ["--atomic"]
+
+  - repo: https://github.com/myint/autoflake
+    rev: v2.3.0
+    hooks:
+      - id: autoflake
+        args:
+          [
+            "--remove-all-unused-imports",
+            "--in-place",
+            "--recursive",
+          ]

framework/config/config_loader.py

@@ -1,14 +1,14 @@
+import os
+import shutil
 from functools import wraps
-import sys
-import warnings
-from ruamel.yaml import YAML
-from pydantic import BaseModel, ValidationError
 from typing import Type
-import shutil
-import os
+
+from pydantic import BaseModel, ValidationError
+from ruamel.yaml import YAML
 
 from framework.logger import get_logger
 
+
 class ConfigLoader:
     """
     配置文件加载器，支持加载和保存 YAML 文件，并保留注释。
@@ -42,7 +42,7 @@ def save_config(config_path: str, config_object: BaseModel):
         """
         with open(config_path, "w", encoding="utf-8") as f:
             ConfigLoader.yaml.dump(config_object.model_dump(), f)
-    
+
     @staticmethod
     def save_config_with_backup(config_path: str, config_object: BaseModel):
         """
@@ -55,9 +55,11 @@ def save_config_with_backup(config_path: str, config_object: BaseModel):
             backup_path = f"{config_path}.bak"
             shutil.copy2(config_path, backup_path)
         ConfigLoader.save_config(config_path, config_object)
-        
+
+
 def pydantic_validation_wrapper(func):
     logger = get_logger("ConfigLoader")
+
     @wraps(func)
     def wrapper(*args, **kwargs):
         try:
@@ -66,9 +68,12 @@ def wrapper(*args, **kwargs):
             # 使用 loguru 输出错误信息
             logger.error(f"Pydantic 验证错误: '{e.title}':")
             for error in e.errors():
-                logger.error(f"字段: {error['loc'][0]}, 错误类型: {error['type']}, 错误信息: {error['msg']}")
+                logger.error(
+                    f"字段: {error['loc'][0]}, 错误类型: {error['type']}, 错误信息: {error['msg']}"
+                )
             # 记录堆栈跟踪
 
             logger.opt(exception=True).error("堆栈跟踪如下：")
             raise  # 可以选择重新抛出异常，或者处理异常后返回一个默认值
-    return wrapper
+
+    return wrapper

framework/config/global_config.py

@@ -5,57 +5,70 @@
 
 class IMConfig(BaseModel):
     """IM配置"""
+
     name: str = Field(default="", description="IM标识名称")
     enable: bool = Field(default=True, description="是否启用IM")
     adapter: str = Field(default="dummy", description="IM适配器类型")
     config: Dict[str, Any] = Field(default={}, description="IM的配置")
 
+
 class LLMBackendConfig(BaseModel):
     """LLM后端配置"""
+
     name: str = Field(description="后端标识名称")
     adapter: str = Field(description="LLM适配器类型")
     config: Dict[str, Any] = Field(default={}, description="后端配置")
     enable: bool = Field(default=True, description="是否启用")
     models: List[str] = Field(default=[], description="支持的模型列表")
 
+
 class LLMConfig(BaseModel):
-    api_backends: List[LLMBackendConfig] = Field(default=[], description="LLM API后端列表")
+    api_backends: List[LLMBackendConfig] = Field(
+        default=[], description="LLM API后端列表"
+    )
+
 
 class DefaultConfig(BaseModel):
-    llm_model: str = Field(default="gemini-1.5-flash", description="默认使用的 LLM 模型名称")
+    llm_model: str = Field(
+        default="gemini-1.5-flash", description="默认使用的 LLM 模型名称"
+    )
+
 
 class MemoryPersistenceConfig(BaseModel):
     type: str = Field(default="file", description="持久化类型: file/redis")
     file: Dict[str, Any] = Field(
-        default={
-            "storage_dir": "./data/memory"
-        },
-        description="文件持久化配置"
+        default={"storage_dir": "./data/memory"}, description="文件持久化配置"
     )
     redis: Dict[str, Any] = Field(
-        default={
-            "host": "localhost",
-            "port": 6379,
-            "db": 0
-        },
-        description="Redis持久化配置"
+        default={"host": "localhost", "port": 6379, "db": 0},
+        description="Redis持久化配置",
     )
 
+
 class MemoryConfig(BaseModel):
     persistence: MemoryPersistenceConfig = MemoryPersistenceConfig()
     max_entries: int = Field(default=100, description="每个作用域最大记忆条目数")
     default_scope: str = Field(default="member", description="默认作用域类型")
-    
+
+
 class WebConfig(BaseModel):
     host: str = Field(default="127.0.0.1", description="Web服务绑定的IP地址")
     port: int = Field(default=8080, description="Web服务端口号")
     secret_key: str = Field(default="", description="Web服务的密钥，用于JWT等加密")
-    password_file: str = Field(default="./data/web/password.hash", description="密码哈希存储路径")
+    password_file: str = Field(
+        default="./data/web/password.hash", description="密码哈希存储路径"
+    )
+
 
 class PluginConfig(BaseModel):
     """插件配置"""
+
     enable: List[str] = Field(default=[], description="启用的外部插件列表")
-    market_base_url: str = Field(default="https://kirara-plugin.app.lss233.com/api/v1", description="插件市场基础URL")
+    market_base_url: str = Field(
+        default="https://kirara-plugin.app.lss233.com/api/v1",
+        description="插件市场基础URL",
+    )
+
 
 class GlobalConfig(BaseModel):
     ims: List[IMConfig] = Field(default=[], description="IM配置列表")

framework/events/event.py

@@ -25,4 +25,4 @@ def decorator(func: Callable):
 
         return func
 
-    return decorator
+    return decorator

framework/events/event_bus.py

@@ -1,5 +1,6 @@
 import inspect
-from typing import Callable, Type, Dict, List
+from typing import Callable, Dict, List, Type
+
 
 class EventBus:
     def __init__(self):
@@ -20,24 +21,26 @@ def post(self, event):
             for listener in self._listeners[event_type]:
                 listener(event)
 
+
 def Event(event_bus: EventBus):
     def decorator(func: Callable):
         # 获取函数的参数签名
         signature = inspect.signature(func)
         params = list(signature.parameters.values())
-        
+
         # 假设第一个参数是事件类型
         if len(params) == 0:
             raise ValueError("Listener function must have at least one parameter")
-        
+
         event_type = params[0].annotation
-        
+
         # 如果没有指定类型注解，抛出异常
         if event_type == inspect.Parameter.empty:
             raise ValueError("Listener function must have an annotated first parameter")
-        
+
         # 注册监听器
         event_bus.register(event_type, func)
-        
+
         return func
-    return decorator
+
+    return decorator

framework/im/adapter.py

@@ -1,41 +1,50 @@
 from abc import ABC, abstractmethod
 from typing import Any, Protocol
+
 from typing_extensions import runtime_checkable
+
 from framework.im.message import IMMessage
 from framework.im.sender import ChatSender
 from framework.llm.llm_manager import LLMManager
+
 from .profile import UserProfile
 
+
 @runtime_checkable
 class EditStateAdapter(Protocol):
     """
     编辑状态适配器接口，定义了如何设置或取消对话的编辑状态
     """
-    async def set_chat_editing_state(self, chat_sender: ChatSender, is_editing: bool = True):
+
+    async def set_chat_editing_state(
+        self, chat_sender: ChatSender, is_editing: bool = True
+    ):
         """
         设置或取消对话的编辑状态
         :param chat_sender: 对话的发送者
         :param is_editing: True 表示正在编辑，False 表示取消编辑状态
         """
-        pass
+
 
 @runtime_checkable
 class UserProfileAdapter(Protocol):
     """
     用户资料查询适配器接口，定义了如何获取用户资料
     """
+
     async def query_user_profile(self, chat_sender: ChatSender) -> UserProfile:
         """
         查询用户资料
         :param chat_sender: 用户的聊天发送者信息
         :return: 用户资料
         """
-        pass
+
 
 class IMAdapter(ABC):
     """
     通用的 IM 适配器接口，定义了如何将不同平台的原始消息转换为 Message 对象。
     """
+
     llm_manager: LLMManager
 
     @abstractmethod
@@ -45,7 +54,6 @@ def convert_to_message(self, raw_message: Any) -> IMMessage:
         :param raw_message: 平台的原始消息对象。
         :return: 转换后的 Message 对象。
         """
-        pass
 
     @abstractmethod
     async def send_message(self, message: IMMessage, recipient: Any):
@@ -54,12 +62,11 @@ async def send_message(self, message: IMMessage, recipient: Any):
         :param message: 要发送的消息对象。
         :param recipient: 接收消息的目标对象，可以是用户ID、用户对象、群组ID等，具体由各平台实现决定。
         """
-        pass
 
     @abstractmethod
     async def start(self):
         pass
-    
+
     @abstractmethod
     async def stop(self):
-        pass
+        pass

framework/im/im_registry.py

@@ -1,8 +1,10 @@
 from typing import Dict, Type
+
 from pydantic import BaseModel
 
 from framework.im.adapter import IMAdapter
 
+
 class IMRegistry:
     """
     适配器注册表，用于动态注册和管理 adapter。
@@ -11,7 +13,9 @@ class IMRegistry:
     _registry: Dict[str, Type[IMAdapter]] = {}
     _config_registry: Dict[str, Type[BaseModel]] = {}
 
-    def register(self, name: str, adapter_class: Type[IMAdapter], config_class: Type[BaseModel]):
+    def register(
+        self, name: str, adapter_class: Type[IMAdapter], config_class: Type[BaseModel]
+    ):
         """
         注册一个新的 adapter 及其配置类。
         :param name: adapter 的名称。
@@ -20,7 +24,7 @@ def register(self, name: str, adapter_class: Type[IMAdapter], config_class: Type
         """
         self._registry[name] = adapter_class
         self._config_registry[name] = config_class
-        
+
     def unregister(self, name: str):
         """
         注销一个 adapter。
@@ -48,11 +52,10 @@ def get_config_class(self, name: str) -> Type[BaseModel]:
         if name not in self._config_registry:
             raise ValueError(f"Config class for adapter '{name}' is not registered.")
         return self._config_registry[name]
-    
+
     def get_all_adapters(self) -> Dict[str, Type[IMAdapter]]:
         """
         获取所有已注册的 adapter。
         :return: 所有已注册的 adapter。
         """
         return self._registry
-