AstrBot 4.5.0__py3-none-any.whl → 4.5.2__py3-none-any.whl

astrbot/core/star/session_llm_manager.py

@@ -1,6 +1,4 @@
-"""
-会话服务管理器 - 负责管理每个会话的LLM、TTS等服务的启停状态
-"""
+"""会话服务管理器 - 负责管理每个会话的LLM、TTS等服务的启停状态"""
 
 from astrbot.core import logger, sp
 from astrbot.core.platform.astr_message_event import AstrMessageEvent
@@ -22,10 +20,14 @@ class SessionServiceManager:
 
         Returns:
             bool: True表示启用，False表示禁用
+
         """
         # 获取会话服务配置
         session_services = sp.get(
-            "session_service_config", {}, scope="umo", scope_id=session_id
+            "session_service_config",
+            {},
+            scope="umo",
+            scope_id=session_id,
         )
 
         # 如果配置了该会话的LLM状态，返回该状态
@@ -43,13 +45,17 @@ class SessionServiceManager:
         Args:
             session_id: 会话ID (unified_msg_origin)
             enabled: True表示启用，False表示禁用
+
         """
         session_config = (
             sp.get("session_service_config", {}, scope="umo", scope_id=session_id) or {}
         )
         session_config["llm_enabled"] = enabled
         sp.put(
-            "session_service_config", session_config, scope="umo", scope_id=session_id
+            "session_service_config",
+            session_config,
+            scope="umo",
+            scope_id=session_id,
         )
 
     @staticmethod
@@ -61,6 +67,7 @@ class SessionServiceManager:
 
         Returns:
             bool: True表示应该处理，False表示跳过
+
         """
         session_id = event.unified_msg_origin
         return SessionServiceManager.is_llm_enabled_for_session(session_id)
@@ -78,10 +85,14 @@ class SessionServiceManager:
 
         Returns:
             bool: True表示启用，False表示禁用
+
         """
         # 获取会话服务配置
         session_services = sp.get(
-            "session_service_config", {}, scope="umo", scope_id=session_id
+            "session_service_config",
+            {},
+            scope="umo",
+            scope_id=session_id,
         )
 
         # 如果配置了该会话的TTS状态，返回该状态
@@ -99,17 +110,21 @@ class SessionServiceManager:
         Args:
             session_id: 会话ID (unified_msg_origin)
             enabled: True表示启用，False表示禁用
+
         """
         session_config = (
             sp.get("session_service_config", {}, scope="umo", scope_id=session_id) or {}
         )
         session_config["tts_enabled"] = enabled
         sp.put(
-            "session_service_config", session_config, scope="umo", scope_id=session_id
+            "session_service_config",
+            session_config,
+            scope="umo",
+            scope_id=session_id,
         )
 
         logger.info(
-            f"会话 {session_id} 的TTS状态已更新为: {'启用' if enabled else '禁用'}"
+            f"会话 {session_id} 的TTS状态已更新为: {'启用' if enabled else '禁用'}",
         )
 
     @staticmethod
@@ -121,6 +136,7 @@ class SessionServiceManager:
 
         Returns:
             bool: True表示应该处理，False表示跳过
+
         """
         session_id = event.unified_msg_origin
         return SessionServiceManager.is_tts_enabled_for_session(session_id)
@@ -138,10 +154,14 @@ class SessionServiceManager:
 
         Returns:
             bool: True表示启用，False表示禁用
+
         """
         # 获取会话服务配置
         session_services = sp.get(
-            "session_service_config", {}, scope="umo", scope_id=session_id
+            "session_service_config",
+            {},
+            scope="umo",
+            scope_id=session_id,
         )
 
         # 如果配置了该会话的整体状态，返回该状态
@@ -159,17 +179,21 @@ class SessionServiceManager:
         Args:
             session_id: 会话ID (unified_msg_origin)
             enabled: True表示启用，False表示禁用
+
         """
         session_config = (
             sp.get("session_service_config", {}, scope="umo", scope_id=session_id) or {}
         )
         session_config["session_enabled"] = enabled
         sp.put(
-            "session_service_config", session_config, scope="umo", scope_id=session_id
+            "session_service_config",
+            session_config,
+            scope="umo",
+            scope_id=session_id,
         )
 
         logger.info(
-            f"会话 {session_id} 的整体状态已更新为: {'启用' if enabled else '禁用'}"
+            f"会话 {session_id} 的整体状态已更新为: {'启用' if enabled else '禁用'}",
         )
 
     @staticmethod
@@ -181,6 +205,7 @@ class SessionServiceManager:
 
         Returns:
             bool: True表示应该处理，False表示跳过
+
         """
         session_id = event.unified_msg_origin
         return SessionServiceManager.is_session_enabled(session_id)
@@ -198,9 +223,13 @@ class SessionServiceManager:
 
         Returns:
             str: 自定义名称，如果没有设置则返回None
+
         """
         session_services = sp.get(
-            "session_service_config", {}, scope="umo", scope_id=session_id
+            "session_service_config",
+            {},
+            scope="umo",
+            scope_id=session_id,
         )
         return session_services.get("custom_name")
 
@@ -211,6 +240,7 @@ class SessionServiceManager:
         Args:
             session_id: 会话ID (unified_msg_origin)
             custom_name: 自定义名称，可以为空字符串来清除名称
+
         """
         session_config = (
             sp.get("session_service_config", {}, scope="umo", scope_id=session_id) or {}
@@ -221,11 +251,14 @@ class SessionServiceManager:
             # 如果传入空名称，则删除自定义名称
             session_config.pop("custom_name", None)
         sp.put(
-            "session_service_config", session_config, scope="umo", scope_id=session_id
+            "session_service_config",
+            session_config,
+            scope="umo",
+            scope_id=session_id,
         )
 
         logger.info(
-            f"会话 {session_id} 的自定义名称已更新为: {custom_name.strip() if custom_name and custom_name.strip() else '已清除'}"
+            f"会话 {session_id} 的自定义名称已更新为: {custom_name.strip() if custom_name and custom_name.strip() else '已清除'}",
         )
 
     @staticmethod
@@ -237,6 +270,7 @@ class SessionServiceManager:
 
         Returns:
             str: 显示名称
+
         """
         custom_name = SessionServiceManager.get_session_custom_name(session_id)
         if custom_name:

astrbot/core/star/session_plugin_manager.py

@@ -1,9 +1,6 @@
-"""
-会话插件管理器 - 负责管理每个会话的插件启停状态
-"""
+"""会话插件管理器 - 负责管理每个会话的插件启停状态"""
 
-from astrbot.core import sp, logger
-from typing import Dict, List
+from astrbot.core import logger, sp
 from astrbot.core.platform.astr_message_event import AstrMessageEvent
 
 
@@ -20,10 +17,14 @@ class SessionPluginManager:
 
         Returns:
             bool: True表示启用，False表示禁用
+
         """
         # 获取会话插件配置
         session_plugin_config = sp.get(
-            "session_plugin_config", {}, scope="umo", scope_id=session_id
+            "session_plugin_config",
+            {},
+            scope="umo",
+            scope_id=session_id,
         )
         session_config = session_plugin_config.get(session_id, {})
 
@@ -43,7 +44,9 @@ class SessionPluginManager:
 
     @staticmethod
     def set_plugin_status_for_session(
-        session_id: str, plugin_name: str, enabled: bool
+        session_id: str,
+        plugin_name: str,
+        enabled: bool,
     ) -> None:
         """设置插件在指定会话中的启停状态
 
@@ -51,10 +54,14 @@ class SessionPluginManager:
             session_id: 会话ID (unified_msg_origin)
             plugin_name: 插件名称
             enabled: True表示启用，False表示禁用
+
         """
         # 获取当前配置
         session_plugin_config = sp.get(
-            "session_plugin_config", {}, scope="umo", scope_id=session_id
+            "session_plugin_config",
+            {},
+            scope="umo",
+            scope_id=session_id,
         )
         if session_id not in session_plugin_config:
             session_plugin_config[session_id] = {
@@ -91,11 +98,11 @@ class SessionPluginManager:
         )
 
         logger.info(
-            f"会话 {session_id} 的插件 {plugin_name} 状态已更新为: {'启用' if enabled else '禁用'}"
+            f"会话 {session_id} 的插件 {plugin_name} 状态已更新为: {'启用' if enabled else '禁用'}",
         )
 
     @staticmethod
-    def get_session_plugin_config(session_id: str) -> Dict[str, List[str]]:
+    def get_session_plugin_config(session_id: str) -> dict[str, list[str]]:
         """获取指定会话的插件配置
 
         Args:
@@ -103,16 +110,21 @@ class SessionPluginManager:
 
         Returns:
             Dict[str, List[str]]: 包含enabled_plugins和disabled_plugins的字典
+
         """
         session_plugin_config = sp.get(
-            "session_plugin_config", {}, scope="umo", scope_id=session_id
+            "session_plugin_config",
+            {},
+            scope="umo",
+            scope_id=session_id,
         )
         return session_plugin_config.get(
-            session_id, {"enabled_plugins": [], "disabled_plugins": []}
+            session_id,
+            {"enabled_plugins": [], "disabled_plugins": []},
         )
 
     @staticmethod
-    def filter_handlers_by_session(event: AstrMessageEvent, handlers: List) -> List:
+    def filter_handlers_by_session(event: AstrMessageEvent, handlers: list) -> list:
         """根据会话配置过滤处理器列表
 
         Args:
@@ -121,6 +133,7 @@ class SessionPluginManager:
 
         Returns:
             List: 过滤后的处理器列表
+
         """
         from astrbot.core.star.star import star_map
 
@@ -145,12 +158,13 @@ class SessionPluginManager:
 
             # 检查插件是否在当前会话中启用
             if SessionPluginManager.is_plugin_enabled_for_session(
-                session_id, plugin.name
+                session_id,
+                plugin.name,
             ):
                 filtered_handlers.append(handler)
             else:
                 logger.debug(
-                    f"插件 {plugin.name} 在会话 {session_id} 中被禁用，跳过处理器 {handler.handler_name}"
+                    f"插件 {plugin.name} 在会话 {session_id} 中被禁用，跳过处理器 {handler.handler_name}",
                 )
 
         return filtered_handlers

astrbot/core/star/star.py

@@ -16,8 +16,7 @@ if TYPE_CHECKING:
 
 @dataclass
 class StarMetadata:
-    """
-    插件的元数据。
+    """插件的元数据。
 
     当 activated 为 False 时，star_cls 可能为 None，请不要在插件未激活时调用 star_cls 的方法。
     """

astrbot/core/star/star_handler.py

@@ -1,7 +1,10 @@
 from __future__ import annotations
+
 import enum
+from collections.abc import Awaitable, Callable
 from dataclasses import dataclass, field
-from typing import Callable, Awaitable, Any, List, Dict, TypeVar, Generic
+from typing import Any, Generic, TypeVar
+
 from .filter import HandlerFilter
 from .star import star_map
 
@@ -10,8 +13,8 @@ T = TypeVar("T", bound="StarHandlerMetadata")
 
 class StarHandlerRegistry(Generic[T]):
     def __init__(self):
-        self.star_handlers_map: Dict[str, StarHandlerMetadata] = {}
-        self._handlers: List[StarHandlerMetadata] = []
+        self.star_handlers_map: dict[str, StarHandlerMetadata] = {}
+        self._handlers: list[StarHandlerMetadata] = []
 
     def append(self, handler: StarHandlerMetadata):
         """添加一个 Handler，并保持按优先级有序"""
@@ -31,7 +34,7 @@ class StarHandlerRegistry(Generic[T]):
         event_type: EventType,
         only_activated=True,
         plugins_name: list[str] | None = None,
-    ) -> List[StarHandlerMetadata]:
+    ) -> list[StarHandlerMetadata]:
         handlers = []
         for handler in self._handlers:
             # 过滤事件类型
@@ -64,8 +67,9 @@ class StarHandlerRegistry(Generic[T]):
         return self.star_handlers_map.get(full_name, None)
 
     def get_handlers_by_module_name(
-        self, module_name: str
-    ) -> List[StarHandlerMetadata]:
+        self,
+        module_name: str,
+    ) -> list[StarHandlerMetadata]:
         return [
             handler
             for handler in self._handlers
@@ -126,7 +130,7 @@ class StarHandlerMetadata:
     handler: Callable[..., Awaitable[Any]]
     """Handler 的函数对象，应当是一个异步函数"""
 
-    event_filters: List[HandlerFilter]
+    event_filters: list[HandlerFilter]
     """一个适配器消息事件过滤器，用于描述这个 Handler 能够处理、应该处理的适配器消息事件"""
 
     desc: str = ""
@@ -138,5 +142,6 @@ class StarHandlerMetadata:
     def __lt__(self, other: StarHandlerMetadata):
         """定义小于运算符以支持优先队列"""
         return self.extras_configs.get("priority", 0) < other.extras_configs.get(
-            "priority", 0
+            "priority",
+            0,
         )