re-style: 格式化代码

This commit is contained in:
John Richard
2025-10-02 20:26:01 +08:00
parent ecb02cae31
commit 7923eafef3
263 changed files with 3103 additions and 3123 deletions

View File

@@ -1,14 +1,13 @@
from typing import List, Tuple, Type
from src.common.logger import get_logger
from src.plugin_system.base.base_plugin import BasePlugin
from src.plugin_system import (
BaseEventHandler,
BasePlugin,
ConfigField,
register_plugin,
EventHandlerInfo,
BaseEventHandler,
register_plugin,
)
from src.plugin_system.base.base_plugin import BasePlugin
from .proacive_thinker_event import ProactiveThinkerEventHandler
logger = get_logger(__name__)
@@ -33,9 +32,9 @@ class ProactiveThinkerPlugin(BasePlugin):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
def get_plugin_components(self) -> List[Tuple[EventHandlerInfo, Type[BaseEventHandler]]]:
def get_plugin_components(self) -> list[tuple[EventHandlerInfo, type[BaseEventHandler]]]:
"""返回插件的EventHandler组件"""
components: List[Tuple[EventHandlerInfo, Type[BaseEventHandler]]] = [
components: list[tuple[EventHandlerInfo, type[BaseEventHandler]]] = [
(ProactiveThinkerEventHandler.get_handler_info(), ProactiveThinkerEventHandler)
]
return components

View File

@@ -2,17 +2,17 @@ import asyncio
import random
import time
from datetime import datetime
from typing import List, Union
from maim_message import UserInfo
from src.chat.message_receive.chat_stream import get_chat_manager
from src.common.logger import get_logger
from src.config.config import global_config
from src.manager.async_task_manager import async_task_manager, AsyncTask
from src.plugin_system import EventType, BaseEventHandler
from src.manager.async_task_manager import AsyncTask, async_task_manager
from src.plugin_system import BaseEventHandler, EventType
from src.plugin_system.apis import chat_api, person_api
from src.plugin_system.base.base_event import HandlerResult
from .proactive_thinker_executor import ProactiveThinkerExecutor
logger = get_logger(__name__)
@@ -199,7 +199,7 @@ class ProactiveThinkerEventHandler(BaseEventHandler):
handler_name: str = "proactive_thinker_on_start"
handler_description: str = "主动思考插件的启动事件处理器"
init_subscribe: List[Union[EventType, str]] = [EventType.ON_START]
init_subscribe: list[EventType | str] = [EventType.ON_START]
async def execute(self, kwargs: dict | None) -> "HandlerResult":
"""在机器人启动时执行,根据配置决定是否启动后台任务。"""

View File

@@ -1,20 +1,21 @@
import orjson
from typing import Optional, Dict, Any
from datetime import datetime
from typing import Any
import orjson
from src.common.logger import get_logger
from src.config.config import global_config, model_config
from src.person_info.person_info import get_person_info_manager
from src.plugin_system.apis import (
chat_api,
database_api,
generator_api,
llm_api,
message_api,
person_api,
schedule_api,
send_api,
llm_api,
message_api,
generator_api,
database_api,
)
from src.config.config import global_config, model_config
from src.person_info.person_info import get_person_info_manager
logger = get_logger(__name__)
@@ -101,7 +102,7 @@ class ProactiveThinkerExecutor:
logger.error(f"解析 stream_id ({stream_id}) 或获取 stream 失败: {e}")
return None
async def _gather_context(self, stream_id: str) -> Optional[Dict[str, Any]]:
async def _gather_context(self, stream_id: str) -> dict[str, Any] | None:
"""
收集构建提示词所需的所有上下文信息
"""
@@ -165,7 +166,7 @@ class ProactiveThinkerExecutor:
"current_time": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
}
async def _make_decision(self, context: Dict[str, Any], start_mode: str) -> Optional[Dict[str, Any]]:
async def _make_decision(self, context: dict[str, Any], start_mode: str) -> dict[str, Any] | None:
"""
决策模块:判断是否应该主动发起对话,以及聊什么话题
"""
@@ -234,7 +235,7 @@ class ProactiveThinkerExecutor:
logger.error(f"决策LLM返回的JSON格式无效: {response}")
return {"should_reply": False, "reason": "决策模型返回格式错误"}
def _build_plan_prompt(self, context: Dict[str, Any], start_mode: str, topic: str, reason: str) -> str:
def _build_plan_prompt(self, context: dict[str, Any], start_mode: str, topic: str, reason: str) -> str:
"""
根据启动模式和决策话题,构建最终的规划提示词
"""