chore: 补全类型声明

astrbot/builtin_stars/astrbot/long_term_memory.py

@@ -17,13 +17,13 @@
 
 
 class LongTermMemory:
-    def __init__(self, acm: AstrBotConfigManager, context: star.Context):
+    def __init__(self, acm: AstrBotConfigManager, context: star.Context) -> None:
         self.acm = acm
         self.context = context
         self.session_chats = defaultdict(list)
         """记录群成员的群聊记录"""
 
-    def cfg(self, event: AstrMessageEvent):
+    def cfg(self, event: AstrMessageEvent) -> dict:
         cfg = self.context.get_config(umo=event.unified_msg_origin)
         try:
             max_cnt = int(cfg["provider_ltm_settings"]["group_message_max_cnt"])
@@ -111,7 +111,7 @@ async def need_active_reply(self, event: AstrMessageEvent) -> bool:
 
         return False
 
-    async def handle_message(self, event: AstrMessageEvent):
+    async def handle_message(self, event: AstrMessageEvent) -> None:
         """仅支持群聊"""
         if event.get_message_type() == MessageType.GROUP_MESSAGE:
             datetime_str = datetime.datetime.now().strftime("%H:%M:%S")
@@ -148,7 +148,7 @@ async def handle_message(self, event: AstrMessageEvent):
             if len(self.session_chats[event.unified_msg_origin]) > cfg["max_cnt"]:
                 self.session_chats[event.unified_msg_origin].pop(0)
 
-    async def on_req_llm(self, event: AstrMessageEvent, req: ProviderRequest):
+    async def on_req_llm(self, event: AstrMessageEvent, req: ProviderRequest) -> None:
         """当触发 LLM 请求前，调用此方法修改 req"""
         if event.unified_msg_origin not in self.session_chats:
             return
@@ -171,7 +171,9 @@ async def on_req_llm(self, event: AstrMessageEvent, req: ProviderRequest):
             )
             req.system_prompt += chats_str
 
-    async def after_req_llm(self, event: AstrMessageEvent, llm_resp: LLMResponse):
+    async def after_req_llm(
+        self, event: AstrMessageEvent, llm_resp: LLMResponse
+    ) -> None:
         if event.unified_msg_origin not in self.session_chats:
             return
 

astrbot/builtin_stars/astrbot/main.py

@@ -1,4 +1,5 @@
 import traceback
+from collections.abc import AsyncGenerator
 
 from astrbot.api import star
 from astrbot.api.event import AstrMessageEvent, filter
@@ -21,14 +22,16 @@ def __init__(self, context: star.Context) -> None:
 
         self.proc_llm_req = ProcessLLMRequest(self.context)
 
-    def ltm_enabled(self, event: AstrMessageEvent):
+    def ltm_enabled(self, event: AstrMessageEvent) -> bool:
         ltmse = self.context.get_config(umo=event.unified_msg_origin)[
             "provider_ltm_settings"
         ]
         return ltmse["group_icl_enable"] or ltmse["active_reply"]["enable"]
 
     @filter.platform_adapter_type(filter.PlatformAdapterType.ALL)
-    async def on_message(self, event: AstrMessageEvent):
+    async def on_message(
+        self, event: AstrMessageEvent
+    ) -> AsyncGenerator[ProviderRequest, None]:
         """群聊记忆增强"""
         has_image_or_plain = False
         for comp in event.message_obj.message:
@@ -89,7 +92,9 @@ async def on_message(self, event: AstrMessageEvent):
                     logger.error(f"主动回复失败: {e}")
 
     @filter.on_llm_request()
-    async def decorate_llm_req(self, event: AstrMessageEvent, req: ProviderRequest):
+    async def decorate_llm_req(
+        self, event: AstrMessageEvent, req: ProviderRequest
+    ) -> None:
         """在请求 LLM 前注入人格信息、Identifier、时间、回复内容等 System Prompt"""
         await self.proc_llm_req.process_llm_request(event, req)
 
@@ -100,7 +105,9 @@ async def decorate_llm_req(self, event: AstrMessageEvent, req: ProviderRequest):
                 logger.error(f"ltm: {e}")
 
     @filter.on_llm_response()
-    async def record_llm_resp_to_ltm(self, event: AstrMessageEvent, resp: LLMResponse):
+    async def record_llm_resp_to_ltm(
+        self, event: AstrMessageEvent, resp: LLMResponse
+    ) -> None:
         """在 LLM 响应后记录对话"""
         if self.ltm and self.ltm_enabled(event):
             try:
@@ -109,7 +116,7 @@ async def record_llm_resp_to_ltm(self, event: AstrMessageEvent, resp: LLMRespons
                 logger.error(f"ltm: {e}")
 
     @filter.after_message_sent()
-    async def after_message_sent(self, event: AstrMessageEvent):
+    async def after_message_sent(self, event: AstrMessageEvent) -> None:
         """消息发送后处理"""
         if self.ltm and self.ltm_enabled(event):
             try:

astrbot/builtin_stars/astrbot/process_llm_request.py

@@ -12,7 +12,7 @@
 
 
 class ProcessLLMRequest:
-    def __init__(self, context: star.Context):
+    def __init__(self, context: star.Context) -> None:
         self.ctx = context
         cfg = context.get_config()
         self.timezone = cfg.get("timezone")
@@ -22,7 +22,7 @@ def __init__(self, context: star.Context):
         else:
             logger.info(f"Timezone set to: {self.timezone}")
 
-    async def _ensure_persona(self, req: ProviderRequest, cfg: dict, umo: str):
+    async def _ensure_persona(self, req: ProviderRequest, cfg: dict, umo: str) -> None:
         """确保用户人格已加载"""
         if not req.conversation:
             return
@@ -78,7 +78,7 @@ async def _ensure_img_caption(
         req: ProviderRequest,
         cfg: dict,
         img_cap_prov_id: str,
-    ):
+    ) -> None:
         try:
             caption = await self._request_img_caption(
                 img_cap_prov_id,
@@ -118,7 +118,9 @@ async def _request_img_caption(
             f"Cannot get image caption because provider `{provider_id}` is not exist.",
         )
 
-    async def process_llm_request(self, event: AstrMessageEvent, req: ProviderRequest):
+    async def process_llm_request(
+        self, event: AstrMessageEvent, req: ProviderRequest
+    ) -> None:
         """在请求 LLM 前注入人格信息、Identifier、时间、回复内容等 System Prompt"""
         cfg: dict = self.ctx.get_config(umo=event.unified_msg_origin)[
             "provider_settings"

astrbot/builtin_stars/builtin_commands/commands/admin.py

@@ -5,10 +5,10 @@
 
 
 class AdminCommands:
-    def __init__(self, context: star.Context):
+    def __init__(self, context: star.Context) -> None:
         self.context = context
 
-    async def op(self, event: AstrMessageEvent, admin_id: str = ""):
+    async def op(self, event: AstrMessageEvent, admin_id: str = "") -> None:
         """授权管理员。op <admin_id>"""
         if not admin_id:
             event.set_result(
@@ -21,7 +21,7 @@ async def op(self, event: AstrMessageEvent, admin_id: str = ""):
         self.context.get_config().save_config()
         event.set_result(MessageEventResult().message("授权成功。"))
 
-    async def deop(self, event: AstrMessageEvent, admin_id: str = ""):
+    async def deop(self, event: AstrMessageEvent, admin_id: str = "") -> None:
         """取消授权管理员。deop <admin_id>"""
         if not admin_id:
             event.set_result(
@@ -39,7 +39,7 @@ async def deop(self, event: AstrMessageEvent, admin_id: str = ""):
                 MessageEventResult().message("此用户 ID 不在管理员名单内。"),
             )
 
-    async def wl(self, event: AstrMessageEvent, sid: str = ""):
+    async def wl(self, event: AstrMessageEvent, sid: str = "") -> None:
         """添加白名单。wl <sid>"""
         if not sid:
             event.set_result(
@@ -53,7 +53,7 @@ async def wl(self, event: AstrMessageEvent, sid: str = ""):
         cfg.save_config()
         event.set_result(MessageEventResult().message("添加白名单成功。"))
 
-    async def dwl(self, event: AstrMessageEvent, sid: str = ""):
+    async def dwl(self, event: AstrMessageEvent, sid: str = "") -> None:
         """删除白名单。dwl <sid>"""
         if not sid:
             event.set_result(
@@ -70,7 +70,7 @@ async def dwl(self, event: AstrMessageEvent, sid: str = ""):
         except ValueError:
             event.set_result(MessageEventResult().message("此 SID 不在白名单内。"))
 
-    async def update_dashboard(self, event: AstrMessageEvent):
+    async def update_dashboard(self, event: AstrMessageEvent) -> None:
         """更新管理面板"""
         await event.send(MessageChain().message("正在尝试更新管理面板..."))
         await download_dashboard(version=f"v{VERSION}", latest=False)

astrbot/builtin_stars/builtin_commands/commands/alter_cmd.py

@@ -11,10 +11,10 @@
 
 
 class AlterCmdCommands(CommandParserMixin):
-    def __init__(self, context: star.Context):
+    def __init__(self, context: star.Context) -> None:
         self.context = context
 
-    async def update_reset_permission(self, scene_key: str, perm_type: str):
+    async def update_reset_permission(self, scene_key: str, perm_type: str) -> None:
         """更新reset命令在特定场景下的权限设置"""
         from astrbot.api import sp
 
@@ -26,7 +26,7 @@ async def update_reset_permission(self, scene_key: str, perm_type: str):
         alter_cmd_cfg["astrbot"] = plugin_cfg
         await sp.global_put("alter_cmd", alter_cmd_cfg)
 
-    async def alter_cmd(self, event: AstrMessageEvent):
+    async def alter_cmd(self, event: AstrMessageEvent) -> None:
         token = self.parse_commands(event.message_str)
         if token.len < 3:
             await event.send(

astrbot/builtin_stars/builtin_commands/commands/conversation.py

@@ -16,10 +16,10 @@
 
 
 class ConversationCommands:
-    def __init__(self, context: star.Context):
+    def __init__(self, context: star.Context) -> None:
         self.context = context
 
-    async def _get_current_persona_id(self, session_id):
+    async def _get_current_persona_id(self, session_id: str) -> str | None:
         curr = await self.context.conversation_manager.get_curr_conversation_id(
             session_id,
         )
@@ -33,7 +33,7 @@ async def _get_current_persona_id(self, session_id):
             return None
         return conv.persona_id
 
-    async def reset(self, message: AstrMessageEvent):
+    async def reset(self, message: AstrMessageEvent) -> None:
         """重置 LLM 会话"""
         umo = message.unified_msg_origin
         cfg = self.context.get_config(umo=message.unified_msg_origin)
@@ -98,7 +98,7 @@ async def reset(self, message: AstrMessageEvent):
 
         message.set_result(MessageEventResult().message(ret))
 
-    async def his(self, message: AstrMessageEvent, page: int = 1):
+    async def his(self, message: AstrMessageEvent, page: int = 1) -> None:
         """查看对话记录"""
         if not self.context.get_using_provider(message.unified_msg_origin):
             message.set_result(
@@ -141,7 +141,7 @@ async def his(self, message: AstrMessageEvent, page: int = 1):
 
         message.set_result(MessageEventResult().message(ret).use_t2i(False))
 
-    async def convs(self, message: AstrMessageEvent, page: int = 1):
+    async def convs(self, message: AstrMessageEvent, page: int = 1) -> None:
         """查看对话列表"""
         cfg = self.context.get_config(umo=message.unified_msg_origin)
         agent_runner_type = cfg["provider_settings"]["agent_runner_type"]
@@ -216,7 +216,7 @@ async def convs(self, message: AstrMessageEvent, page: int = 1):
         message.set_result(MessageEventResult().message(ret).use_t2i(False))
         return
 
-    async def new_conv(self, message: AstrMessageEvent):
+    async def new_conv(self, message: AstrMessageEvent) -> None:
         """创建新对话"""
         cfg = self.context.get_config(umo=message.unified_msg_origin)
         agent_runner_type = cfg["provider_settings"]["agent_runner_type"]
@@ -242,7 +242,7 @@ async def new_conv(self, message: AstrMessageEvent):
             MessageEventResult().message(f"切换到新对话: 新对话({cid[:4]})。"),
         )
 
-    async def groupnew_conv(self, message: AstrMessageEvent, sid: str = ""):
+    async def groupnew_conv(self, message: AstrMessageEvent, sid: str = "") -> None:
         """创建新群聊对话"""
         if sid:
             session = str(
@@ -273,7 +273,7 @@ async def switch_conv(
         self,
         message: AstrMessageEvent,
         index: int | None = None,
-    ):
+    ) -> None:
         """通过 /ls 前面的序号切换对话"""
         if not isinstance(index, int):
             message.set_result(
@@ -308,7 +308,7 @@ async def switch_conv(
                 ),
             )
 
-    async def rename_conv(self, message: AstrMessageEvent, new_name: str = ""):
+    async def rename_conv(self, message: AstrMessageEvent, new_name: str = "") -> None:
         """重命名对话"""
         if not new_name:
             message.set_result(MessageEventResult().message("请输入新的对话名称。"))
@@ -319,7 +319,7 @@ async def rename_conv(self, message: AstrMessageEvent, new_name: str = ""):
         )
         message.set_result(MessageEventResult().message("重命名对话成功。"))
 
-    async def del_conv(self, message: AstrMessageEvent):
+    async def del_conv(self, message: AstrMessageEvent) -> None:
         """删除当前对话"""
         cfg = self.context.get_config(umo=message.unified_msg_origin)
         is_unique_session = cfg["platform_settings"]["unique_session"]

astrbot/builtin_stars/builtin_commands/commands/help.py

@@ -8,10 +8,10 @@
 
 
 class HelpCommand:
-    def __init__(self, context: star.Context):
+    def __init__(self, context: star.Context) -> None:
         self.context = context
 
-    async def _query_astrbot_notice(self):
+    async def _query_astrbot_notice(self) -> str:
         try:
             async with aiohttp.ClientSession(trust_env=True) as session:
                 async with session.get(
@@ -34,7 +34,7 @@ async def _build_reserved_command_lines(self) -> list[str]:
         lines: list[str] = []
         hidden_commands = {"set", "unset", "websearch"}
 
-        def walk(items: list[dict], indent: int = 0):
+        def walk(items: list[dict], indent: int = 0) -> None:
             for item in items:
                 if not item.get("reserved") or not item.get("enabled"):
                     continue
@@ -62,7 +62,7 @@ def walk(items: list[dict], indent: int = 0):
         walk(commands)
         return lines
 
-    async def help(self, event: AstrMessageEvent):
+    async def help(self, event: AstrMessageEvent) -> None:
         """查看帮助"""
         notice = ""
         try:

astrbot/builtin_stars/builtin_commands/commands/llm.py

@@ -3,10 +3,10 @@
 
 
 class LLMCommands:
-    def __init__(self, context: star.Context):
+    def __init__(self, context: star.Context) -> None:
         self.context = context
 
-    async def llm(self, event: AstrMessageEvent):
+    async def llm(self, event: AstrMessageEvent) -> None:
         """开启/关闭 LLM"""
         cfg = self.context.get_config(umo=event.unified_msg_origin)
         enable = cfg["provider_settings"].get("enable", True)

astrbot/builtin_stars/builtin_commands/commands/persona.py

@@ -5,10 +5,10 @@
 
 
 class PersonaCommands:
-    def __init__(self, context: star.Context):
+    def __init__(self, context: star.Context) -> None:
         self.context = context
 
-    async def persona(self, message: AstrMessageEvent):
+    async def persona(self, message: AstrMessageEvent) -> None:
         l = message.message_str.split(" ")  # noqa: E741
         umo = message.unified_msg_origin