AstrBot 4.11.4__py3-none-any.whl → 4.12.1__py3-none-any.whl

astrbot/core/sandbox/tools/shell.py

@@ -0,0 +1,55 @@
+import json
+from dataclasses import dataclass, field
+
+from astrbot.api import FunctionTool
+from astrbot.core.agent.run_context import ContextWrapper
+from astrbot.core.agent.tool import ToolExecResult
+from astrbot.core.astr_agent_context import AstrAgentContext
+
+from ..sandbox_client import get_booter
+
+
+@dataclass
+class ExecuteShellTool(FunctionTool):
+    name: str = "astrbot_execute_shell"
+    description: str = "Execute a command in the shell."
+    parameters: dict = field(
+        default_factory=lambda: {
+            "type": "object",
+            "properties": {
+                "command": {
+                    "type": "string",
+                    "description": "The bash command to execute. Equal to 'cd {working_dir} && {your_command}'.",
+                },
+                "background": {
+                    "type": "boolean",
+                    "description": "Whether to run the command in the background.",
+                    "default": False,
+                },
+                "env": {
+                    "type": "object",
+                    "description": "Optional environment variables to set for the file creation process.",
+                    "additionalProperties": {"type": "string"},
+                    "default": {},
+                },
+            },
+            "required": ["command"],
+        }
+    )
+
+    async def call(
+        self,
+        context: ContextWrapper[AstrAgentContext],
+        command: str,
+        background: bool = False,
+        env: dict = {},
+    ) -> ToolExecResult:
+        sb = await get_booter(
+            context.context.context,
+            context.context.event.unified_msg_origin,
+        )
+        try:
+            result = await sb.shell.exec(command, background=background, env=env)
+            return json.dumps(result)
+        except Exception as e:
+            return f"Error executing command: {str(e)}"

astrbot/core/star/context.py

@@ -49,7 +49,7 @@ class Context:
 
     registered_web_apis: list = []
 
-    # back compatibility
+    # 向后兼容的变量
     _register_tasks: list[Awaitable] = []
     _star_manager = None
 
@@ -73,12 +73,19 @@ class Context:
         self._db = db
         """AstrBot 数据库"""
         self.provider_manager = provider_manager
+        """模型提供商管理器"""
         self.platform_manager = platform_manager
+        """平台适配器管理器"""
         self.conversation_manager = conversation_manager
+        """会话管理器"""
         self.message_history_manager = message_history_manager
+        """平台消息历史管理器"""
         self.persona_manager = persona_manager
+        """人格角色设定管理器"""
         self.astrbot_config_mgr = astrbot_config_mgr
+        """配置文件管理器(非webui)"""
         self.kb_manager = knowledge_base_manager
+        """知识库管理器"""
 
     async def llm_generate(
         self,
@@ -226,14 +233,16 @@ class Context:
         return llm_resp
 
     async def get_current_chat_provider_id(self, umo: str) -> str:
-        """Get the ID of the currently used chat provider.
+        """获取当前使用的聊天模型 Provider ID。
 
         Args:
-            umo(str): unified_message_origin value, if provided and user has enabled provider session isolation, the provider preferred by that session will be used.
+            umo: unified_message_origin。消息会话来源 ID。
 
-        Raises:
-            ProviderNotFoundError: If the specified chat provider is not found
+        Returns:
+            指定消息会话来源当前使用的聊天模型 Provider ID。
 
+        Raises:
+            ProviderNotFoundError: 未找到。
         """
         prov = self.get_using_provider(umo)
         if not prov:
@@ -255,20 +264,27 @@ class Context:
         return self.provider_manager.llm_tools
 
     def activate_llm_tool(self, name: str) -> bool:
-        """激活一个已经注册的函数调用工具。注册的工具默认是激活状态。
+        """激活一个已经注册的函数调用工具。
+
+        Args:
+            name: 工具名称。
 
         Returns:
-            如果没找到，会返回 False
+            如果成功激活返回 True，如果没找到工具返回 False。
 
+        Note:
+            注册的工具默认是激活状态。
         """
         return self.provider_manager.llm_tools.activate_llm_tool(name, star_map)
 
     def deactivate_llm_tool(self, name: str) -> bool:
         """停用一个已经注册的函数调用工具。
 
-        Returns:
-            如果没找到，会返回 False
+        Args:
+            name: 工具名称。
 
+        Returns:
+            如果成功停用返回 True，如果没找到工具返回 False。
         """
         return self.provider_manager.llm_tools.deactivate_llm_tool(name)
 
@@ -278,7 +294,17 @@ class Context:
     ) -> (
         Provider | TTSProvider | STTProvider | EmbeddingProvider | RerankProvider | None
     ):
-        """通过 ID 获取对应的 LLM Provider。"""
+        """通过 ID 获取对应的 LLM Provider。
+
+        Args:
+            provider_id: 提供者 ID。
+
+        Returns:
+            提供者实例，如果未找到则返回 None。
+
+        Note:
+            如果提供者 ID 存在但未找到提供者，会记录警告日志。
+        """
         prov = self.provider_manager.inst_map.get(provider_id)
         if provider_id and not prov:
             logger.warning(
@@ -303,11 +329,20 @@ class Context:
         return self.provider_manager.embedding_provider_insts
 
     def get_using_provider(self, umo: str | None = None) -> Provider:
-        """获取当前使用的用于文本生成任务的 LLM Provider(Chat_Completion 类型)。通过 /provider 指令切换。
+        """获取当前使用的用于文本生成任务的 LLM Provider(Chat_Completion 类型)。
 
         Args:
-            umo(str): unified_message_origin 值，如果传入并且用户启用了提供商会话隔离，则使用该会话偏好的提供商。
+            umo: unified_message_origin 值，如果传入并且用户启用了提供商会话隔离，
+                 则使用该会话偏好的提供商。
+
+        Returns:
+            当前使用的文本生成提供者。
 
+        Raises:
+            ValueError: 返回的提供者不是 Provider 类型。
+
+        Note:
+            通过 /provider 指令可以切换提供者。
         """
         prov = self.provider_manager.get_using_provider(
             provider_type=ProviderType.CHAT_COMPLETION,
@@ -321,8 +356,13 @@ class Context:
         """获取当前使用的用于 TTS 任务的 Provider。
 
         Args:
-            umo(str): unified_message_origin 值，如果传入，则使用该会话偏好的提供商。
+            umo: unified_message_origin 值，如果传入，则使用该会话偏好的提供商。
+
+        Returns:
+            当前使用的 TTS 提供者，如果未设置则返回 None。
 
+        Raises:
+            ValueError: 返回的提供者不是 TTSProvider 类型。
         """
         prov = self.provider_manager.get_using_provider(
             provider_type=ProviderType.TEXT_TO_SPEECH,
@@ -336,8 +376,13 @@ class Context:
         """获取当前使用的用于 STT 任务的 Provider。
 
         Args:
-            umo(str): unified_message_origin 值，如果传入，则使用该会话偏好的提供商。
+            umo: unified_message_origin 值，如果传入，则使用该会话偏好的提供商。
+
+        Returns:
+            当前使用的 STT 提供者，如果未设置则返回 None。
 
+        Raises:
+            ValueError: 返回的提供者不是 STTProvider 类型。
         """
         prov = self.provider_manager.get_using_provider(
             provider_type=ProviderType.SPEECH_TO_TEXT,
@@ -348,9 +393,19 @@ class Context:
         return prov
 
     def get_config(self, umo: str | None = None) -> AstrBotConfig:
-        """获取 AstrBot 的配置。"""
+        """获取 AstrBot 的配置。
+
+        Args:
+            umo: unified_message_origin 值，用于获取特定会话的配置。
+
+        Returns:
+            AstrBot 配置对象。
+
+        Note:
+            如果不提供 umo 参数，将返回默认配置。
+        """
         if not umo:
-            # using default config
+            # 使用默认配置
             return self._config
         return self.astrbot_config_mgr.get_conf(umo)
 
@@ -361,14 +416,19 @@ class Context:
     ) -> bool:
         """根据 session(unified_msg_origin) 主动发送消息。
 
-        @param session: 消息会话。通过 event.session 或者 event.unified_msg_origin 获取。
-        @param message_chain: 消息链。
+        Args:
+            session: 消息会话。通过 event.session 或者 event.unified_msg_origin 获取。
+            message_chain: 消息链。
 
-        @return: 是否找到匹配的平台。
+        Returns:
+            是否找到匹配的平台。
 
-        当 session 为字符串时，会尝试解析为 MessageSesion 对象，如果解析失败，会抛出 ValueError 异常。
+        Raises:
+            ValueError: session 字符串不合法时抛出。
 
-        NOTE: qq_official(QQ 官方 API 平台) 不支持此方法
+        Note:
+            当 session 为字符串时，会尝试解析为 MessageSession 对象。(类名为MessageSesion是因为历史遗留拼写错误)
+            qq_official(QQ 官方 API 平台) 不支持此方法。
         """
         if isinstance(session, str):
             try:
@@ -383,7 +443,14 @@ class Context:
         return False
 
     def add_llm_tools(self, *tools: FunctionTool) -> None:
-        """添加 LLM 工具。"""
+        """添加 LLM 工具。
+
+        Args:
+            *tools: 要添加的函数工具对象。
+
+        Note:
+            如果工具已存在，会替换已存在的工具。
+        """
         tool_name = {tool.name for tool in self.provider_manager.llm_tools.func_list}
         module_path = ""
         for tool in tools:
@@ -416,6 +483,17 @@ class Context:
         methods: list,
         desc: str,
     ):
+        """注册 Web API。
+
+        Args:
+            route: API 路由路径。
+            view_handler: 异步视图处理函数。
+            methods: HTTP 方法列表。
+            desc: API 描述。
+
+        Note:
+            如果相同路由和方法已注册，会替换现有的 API。
+        """
         for idx, api in enumerate(self.registered_web_apis):
             if api[0] == route and methods == api[2]:
                 self.registered_web_apis[idx] = (route, view_handler, methods, desc)
@@ -434,7 +512,14 @@ class Context:
     def get_platform(self, platform_type: PlatformAdapterType | str) -> Platform | None:
         """获取指定类型的平台适配器。
 
-        该方法已经过时，请使用 get_platform_inst 方法。(>= AstrBot v4.0.0)
+        Args:
+            platform_type: 平台类型或平台名称。
+
+        Returns:
+            平台适配器实例，如果未找到则返回 None。
+
+        Note:
+            该方法已经过时，请使用 get_platform_inst 方法。(>= AstrBot v4.0.0)
         """
         for platform in self.platform_manager.platform_insts:
             name = platform.meta().name
@@ -451,22 +536,32 @@ class Context:
         """获取指定 ID 的平台适配器实例。
 
         Args:
-            platform_id (str): 平台适配器的唯一标识符。你可以通过 event.get_platform_id() 获取。
+            platform_id: 平台适配器的唯一标识符。
 
         Returns:
-            Platform: 平台适配器实例，如果未找到则返回 None。
+            平台适配器实例，如果未找到则返回 None。
 
+        Note:
+            可以通过 event.get_platform_id() 获取平台 ID。
         """
         for platform in self.platform_manager.platform_insts:
             if platform.meta().id == platform_id:
                 return platform
 
     def get_db(self) -> BaseDatabase:
-        """获取 AstrBot 数据库。"""
+        """获取 AstrBot 数据库。
+
+        Returns:
+            数据库实例。
+        """
         return self._db
 
     def register_provider(self, provider: Provider):
-        """注册一个 LLM Provider(Chat_Completion 类型)。"""
+        """注册一个 LLM Provider(Chat_Completion 类型)。
+
+        Args:
+            provider: 提供者实例。
+        """
         self.provider_manager.provider_insts.append(provider)
 
     def register_llm_tool(
@@ -478,12 +573,16 @@ class Context:
     ) -> None:
         """[DEPRECATED]为函数调用（function-calling / tools-use）添加工具。
 
-        @param name: 函数名
-        @param func_args: 函数参数列表，格式为 [{"type": "string", "name": "arg_name", "description": "arg_description"}, ...]
-        @param desc: 函数描述
-        @param func_obj: 异步处理函数。
-
-        异步处理函数会接收到额外的的关键词参数：event: AstrMessageEvent, context: Context。
+        Args:
+            name: 函数名。
+            func_args: 函数参数列表，格式为
+                [{"type": "string", "name": "arg_name", "description": "arg_description"}, ...]。
+            desc: 函数描述。
+            func_obj: 异步处理函数。
+
+        Note:
+            异步处理函数会接收到额外的关键词参数：event: AstrMessageEvent, context: Context。
+            该方法已弃用，请使用新的注册方式。
         """
         md = StarHandlerMetadata(
             event_type=EventType.OnLLMRequestEvent,
@@ -498,7 +597,15 @@ class Context:
         self.provider_manager.llm_tools.add_func(name, func_args, desc, func_obj)
 
     def unregister_llm_tool(self, name: str) -> None:
-        """[DEPRECATED]删除一个函数调用工具。如果再要启用，需要重新注册。"""
+        """[DEPRECATED]删除一个函数调用工具。
+
+        Args:
+            name: 工具名称。
+
+        Note:
+            如果再要启用，需要重新注册。
+            该方法已弃用。
+        """
         self.provider_manager.llm_tools.remove_func(name)
 
     def register_commands(
@@ -511,16 +618,19 @@ class Context:
         use_regex=False,
         ignore_prefix=False,
     ):
-        """注册一个命令。
-
-        [Deprecated] 推荐使用装饰器注册指令。该方法将在未来的版本中被移除。
-
-        @param star_name: 插件（Star）名称。
-        @param command_name: 命令名称。
-        @param desc: 命令描述。
-        @param priority: 优先级。1-10。
-        @param awaitable: 异步处理函数。
+        """[DEPRECATED]注册一个命令。
 
+        Args:
+            star_name: 插件（Star）名称。
+            command_name: 命令名称。
+            desc: 命令描述。
+            priority: 优先级。1-10。
+            awaitable: 异步处理函数。
+            use_regex: 是否使用正则表达式匹配命令。
+            ignore_prefix: 是否忽略命令前缀。
+
+        Note:
+            推荐使用装饰器注册指令。该方法将在未来的版本中被移除。
         """
         md = StarHandlerMetadata(
             event_type=EventType.AdapterMessageEvent,
@@ -540,5 +650,13 @@ class Context:
         star_handlers_registry.append(md)
 
     def register_task(self, task: Awaitable, desc: str):
-        """[DEPRECATED]注册一个异步任务。"""
+        """[DEPRECATED]注册一个异步任务。
+
+        Args:
+            task: 异步任务。
+            desc: 任务描述。
+
+        Note:
+            该方法已弃用。
+        """
         self._register_tasks.append(task)

astrbot/dashboard/routes/__init__.py

@@ -1,6 +1,7 @@
 from .auth import AuthRoute
 from .backup import BackupRoute
 from .chat import ChatRoute
+from .chatui_project import ChatUIProjectRoute
 from .command import CommandRoute
 from .config import ConfigRoute
 from .conversation import ConversationRoute
@@ -20,6 +21,7 @@ __all__ = [
     "AuthRoute",
     "BackupRoute",
     "ChatRoute",
+    "ChatUIProjectRoute",
     "CommandRoute",
     "ConfigRoute",
     "ConversationRoute",

astrbot/dashboard/routes/chat.py

@@ -296,6 +296,8 @@ class ChatRoute(Route):
         # 构建用户消息段（包含 path 用于传递给 adapter）
         message_parts = await self._build_user_message_parts(message)
 
+        message_id = str(uuid.uuid4())
+
         async def stream():
             client_disconnected = False
             accumulated_parts = []
@@ -319,6 +321,13 @@ class ChatRoute(Route):
                         if not result:
                             continue
 
+                        if (
+                            "message_id" in result
+                            and result["message_id"] != message_id
+                        ):
+                            logger.warning("webchat stream message_id mismatch")
+                            continue
+
                         result_text = result["data"]
                         msg_type = result.get("type")
                         streaming = result.get("streaming", False)
@@ -456,6 +465,7 @@ class ChatRoute(Route):
                     "selected_provider": selected_provider,
                     "selected_model": selected_model,
                     "enable_streaming": enable_streaming,
+                    "message_id": message_id,
                 },
             ),
         )
@@ -618,9 +628,17 @@ class ChatRoute(Route):
             page_size=100,  # 暂时返回前100个
         )
 
-        # 转换为字典格式，并添加额外信息
+        # 转换为字典格式，并添加项目信息
+        # get_platform_sessions_by_creator 现在返回 list[dict] 包含 session 和项目字段
         sessions_data = []
-        for session in sessions:
+        for item in sessions:
+            session = item["session"]
+            project_id = item["project_id"]
+
+            # 跳过属于项目的会话（在侧边栏对话列表中不显示）
+            if project_id is not None:
+                continue
+
             sessions_data.append(
                 {
                     "session_id": session.session_id,
@@ -645,6 +663,12 @@ class ChatRoute(Route):
         session = await self.db.get_platform_session_by_id(session_id)
         platform_id = session.platform_id if session else "webchat"
 
+        # 获取项目信息（如果会话属于某个项目）
+        username = g.get("username", "guest")
+        project_info = await self.db.get_project_by_session(
+            session_id=session_id, creator=username
+        )
+
         # Get platform message history using session_id
         history_ls = await self.platform_history_mgr.get(
             platform_id=platform_id,
@@ -655,16 +679,20 @@ class ChatRoute(Route):
 
         history_res = [history.model_dump() for history in history_ls]
 
-        return (
-            Response()
-            .ok(
-                data={
-                    "history": history_res,
-                    "is_running": self.running_convs.get(session_id, False),
-                },
-            )
-            .__dict__
-        )
+        response_data = {
+            "history": history_res,
+            "is_running": self.running_convs.get(session_id, False),
+        }
+
+        # 如果会话属于项目，添加项目信息
+        if project_info:
+            response_data["project"] = {
+                "project_id": project_info.project_id,
+                "title": project_info.title,
+                "emoji": project_info.emoji,
+            }
+
+        return Response().ok(data=response_data).__dict__
 
     async def update_session_display_name(self):
         """Update a Platform session's display name."""