AstrBot 4.12.4__py3-none-any.whl → 4.13.1__py3-none-any.whl

astrbot/core/computer/computer_client.py

@@ -0,0 +1,102 @@
+import os
+import shutil
+import uuid
+from pathlib import Path
+
+from astrbot.api import logger
+from astrbot.core.skills.skill_manager import SANDBOX_SKILLS_ROOT
+from astrbot.core.star.context import Context
+from astrbot.core.utils.astrbot_path import (
+    get_astrbot_skills_path,
+    get_astrbot_temp_path,
+)
+
+from .booters.base import ComputerBooter
+from .booters.local import LocalBooter
+
+session_booter: dict[str, ComputerBooter] = {}
+local_booter: ComputerBooter | None = None
+
+
+async def _sync_skills_to_sandbox(booter: ComputerBooter) -> None:
+    skills_root = get_astrbot_skills_path()
+    if not os.path.isdir(skills_root):
+        return
+    if not any(Path(skills_root).iterdir()):
+        return
+
+    temp_dir = get_astrbot_temp_path()
+    os.makedirs(temp_dir, exist_ok=True)
+    zip_base = os.path.join(temp_dir, "skills_bundle")
+    zip_path = f"{zip_base}.zip"
+
+    try:
+        if os.path.exists(zip_path):
+            os.remove(zip_path)
+        shutil.make_archive(zip_base, "zip", skills_root)
+        remote_zip = Path(SANDBOX_SKILLS_ROOT) / "skills.zip"
+        await booter.shell.exec(f"mkdir -p {SANDBOX_SKILLS_ROOT}")
+        upload_result = await booter.upload_file(zip_path, str(remote_zip))
+        if not upload_result.get("success", False):
+            raise RuntimeError("Failed to upload skills bundle to sandbox.")
+        await booter.shell.exec(
+            f"unzip -o {remote_zip} -d {SANDBOX_SKILLS_ROOT} && rm -f {remote_zip}"
+        )
+    finally:
+        if os.path.exists(zip_path):
+            try:
+                os.remove(zip_path)
+            except Exception:
+                logger.warning(f"Failed to remove temp skills zip: {zip_path}")
+
+
+async def get_booter(
+    context: Context,
+    session_id: str,
+) -> ComputerBooter:
+    config = context.get_config(umo=session_id)
+
+    sandbox_cfg = config.get("provider_settings", {}).get("sandbox", {})
+    booter_type = sandbox_cfg.get("booter", "shipyard")
+
+    if session_id in session_booter:
+        booter = session_booter[session_id]
+        if not await booter.available():
+            # rebuild
+            session_booter.pop(session_id, None)
+    if session_id not in session_booter:
+        uuid_str = uuid.uuid5(uuid.NAMESPACE_DNS, session_id).hex
+        if booter_type == "shipyard":
+            from .booters.shipyard import ShipyardBooter
+
+            ep = sandbox_cfg.get("shipyard_endpoint", "")
+            token = sandbox_cfg.get("shipyard_access_token", "")
+            ttl = sandbox_cfg.get("shipyard_ttl", 3600)
+            max_sessions = sandbox_cfg.get("shipyard_max_sessions", 10)
+
+            client = ShipyardBooter(
+                endpoint_url=ep, access_token=token, ttl=ttl, session_num=max_sessions
+            )
+        elif booter_type == "boxlite":
+            from .booters.boxlite import BoxliteBooter
+
+            client = BoxliteBooter()
+        else:
+            raise ValueError(f"Unknown booter type: {booter_type}")
+
+        try:
+            await client.boot(uuid_str)
+            await _sync_skills_to_sandbox(client)
+        except Exception as e:
+            logger.error(f"Error booting sandbox for session {session_id}: {e}")
+            raise e
+
+        session_booter[session_id] = client
+    return session_booter[session_id]
+
+
+def get_local_booter() -> ComputerBooter:
+    global local_booter
+    if local_booter is None:
+        local_booter = LocalBooter()
+    return local_booter

astrbot/core/{sandbox → computer}/tools/__init__.py

@@ -1,10 +1,11 @@
 from .fs import FileDownloadTool, FileUploadTool
-from .python import PythonTool
+from .python import LocalPythonTool, PythonTool
 from .shell import ExecuteShellTool
 
 __all__ = [
     "FileUploadTool",
     "PythonTool",
+    "LocalPythonTool",
     "ExecuteShellTool",
     "FileDownloadTool",
 ]

astrbot/core/{sandbox → computer}/tools/fs.py

@@ -9,7 +9,7 @@ from astrbot.core.astr_agent_context import AstrAgentContext
 from astrbot.core.message.components import File
 from astrbot.core.utils.astrbot_path import get_astrbot_temp_path
 
-from ..sandbox_client import get_booter
+from ..computer_client import get_booter
 
 # @dataclass
 # class CreateFileTool(FunctionTool):

astrbot/core/computer/tools/python.py

@@ -0,0 +1,94 @@
+from dataclasses import dataclass, field
+
+import mcp
+
+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 astrbot.core.computer.computer_client import get_booter, get_local_booter
+
+param_schema = {
+    "type": "object",
+    "properties": {
+        "code": {
+            "type": "string",
+            "description": "The Python code to execute.",
+        },
+        "silent": {
+            "type": "boolean",
+            "description": "Whether to suppress the output of the code execution.",
+            "default": False,
+        },
+    },
+    "required": ["code"],
+}
+
+
+def handle_result(result: dict) -> ToolExecResult:
+    data = result.get("data", {})
+    output = data.get("output", {})
+    error = data.get("error", "")
+    images: list[dict] = output.get("images", [])
+    text: str = output.get("text", "")
+
+    resp = mcp.types.CallToolResult(content=[])
+
+    if error:
+        resp.content.append(mcp.types.TextContent(type="text", text=f"error: {error}"))
+
+    if images:
+        for img in images:
+            resp.content.append(
+                mcp.types.ImageContent(
+                    type="image", data=img["image/png"], mimeType="image/png"
+                )
+            )
+    if text:
+        resp.content.append(mcp.types.TextContent(type="text", text=text))
+
+    if not resp.content:
+        resp.content.append(mcp.types.TextContent(type="text", text="No output."))
+
+    return resp
+
+
+@dataclass
+class PythonTool(FunctionTool):
+    name: str = "astrbot_execute_ipython"
+    description: str = "Run codes in an IPython shell."
+    parameters: dict = field(default_factory=lambda: param_schema)
+
+    async def call(
+        self, context: ContextWrapper[AstrAgentContext], code: str, silent: bool = False
+    ) -> ToolExecResult:
+        sb = await get_booter(
+            context.context.context,
+            context.context.event.unified_msg_origin,
+        )
+        try:
+            result = await sb.python.exec(code, silent=silent)
+            return handle_result(result)
+        except Exception as e:
+            return f"Error executing code: {str(e)}"
+
+
+@dataclass
+class LocalPythonTool(FunctionTool):
+    name: str = "astrbot_execute_python"
+    description: str = "Execute codes in a Python environment."
+
+    parameters: dict = field(default_factory=lambda: param_schema)
+
+    async def call(
+        self, context: ContextWrapper[AstrAgentContext], code: str, silent: bool = False
+    ) -> ToolExecResult:
+        if context.context.event.role != "admin":
+            return "error: Permission denied. Local Python execution is only allowed for admin users. Tell user to set admins in AstrBot WebUI."
+
+        sb = get_local_booter()
+        try:
+            result = await sb.python.exec(code, silent=silent)
+            return handle_result(result)
+        except Exception as e:
+            return f"Error executing code: {str(e)}"

astrbot/core/{sandbox → computer}/tools/shell.py

@@ -6,7 +6,7 @@ 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
+from ..computer_client import get_booter, get_local_booter
 
 
 @dataclass
@@ -37,6 +37,8 @@ class ExecuteShellTool(FunctionTool):
         }
     )
 
+    is_local: bool = False
+
     async def call(
         self,
         context: ContextWrapper[AstrAgentContext],
@@ -44,10 +46,16 @@ class ExecuteShellTool(FunctionTool):
         background: bool = False,
         env: dict = {},
     ) -> ToolExecResult:
-        sb = await get_booter(
-            context.context.context,
-            context.context.event.unified_msg_origin,
-        )
+        if context.context.event.role != "admin":
+            return "error: Permission denied. Shell execution is only allowed for admin users. Tell user to Set admins in AstrBot WebUI."
+
+        if self.is_local:
+            sb = get_local_booter()
+        else:
+            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)

astrbot/core/config/default.py

@@ -5,7 +5,7 @@ from typing import Any, TypedDict
 
 from astrbot.core.utils.astrbot_path import get_astrbot_data_path
 
-VERSION = "4.12.4"
+VERSION = "4.13.1"
 DB_PATH = os.path.join(get_astrbot_data_path(), "data_v4.db")
 
 WEBHOOK_SUPPORTED_PLATFORMS = [
@@ -106,6 +106,7 @@ DEFAULT_CONFIG = {
         "reachability_check": False,
         "max_agent_step": 30,
         "tool_call_timeout": 60,
+        "tool_schema_mode": "full",
         "llm_safety_mode": True,
         "safety_mode_strategy": "system_prompt",  # TODO: llm judge
         "file_extract": {
@@ -121,6 +122,7 @@ DEFAULT_CONFIG = {
             "shipyard_ttl": 3600,
             "shipyard_max_sessions": 10,
         },
+        "skills": {"runtime": "sandbox"},
     },
     "provider_stt_settings": {
         "enable": False,
@@ -166,6 +168,7 @@ DEFAULT_CONFIG = {
         "jwt_secret": "",
         "host": "0.0.0.0",
         "port": 6185,
+        "disable_access_log": True,
     },
     "platform": [],
     "platform_specific": {
@@ -773,27 +776,21 @@ CONFIG_METADATA_2 = {
                             "interval_method": {
                                 "type": "string",
                                 "options": ["random", "log"],
-                                "hint": "分段回复的间隔时间计算方法。random 为随机时间，log 为根据消息长度计算，$y=log_<log_base>(x)$，x为字数，y的单位为秒。",
                             },
                             "interval": {
                                 "type": "string",
-                                "hint": "`random` 方法用。每一段回复的间隔时间，格式为 `最小时间,最大时间`。如 `0.75,2.5`",
                             },
                             "log_base": {
                                 "type": "float",
-                                "hint": "`log` 方法用。对数函数的底数。默认为 2.6",
                             },
                             "words_count_threshold": {
                                 "type": "int",
-                                "hint": "分段回复的字数上限。只有字数小于此值的消息才会被分段，超过此值的长消息将直接发送（不分段）。默认为 150",
                             },
                             "regex": {
                                 "type": "string",
-                                "hint": "用于分隔一段消息。默认情况下会根据句号、问号等标点符号分隔。re.findall(r'<regex>', text)",
                             },
                             "content_cleanup_rule": {
                                 "type": "string",
-                                "hint": "移除分段后的内容中的指定的内容。支持正则表达式。如填写 `[。？！]` 将移除所有的句号、问号、感叹号。re.sub(r'<regex>', '', text)",
                             },
                         },
                     },
@@ -2187,6 +2184,9 @@ CONFIG_METADATA_2 = {
                     "tool_call_timeout": {
                         "type": "int",
                     },
+                    "tool_schema_mode": {
+                        "type": "string",
+                    },
                     "file_extract": {
                         "type": "object",
                         "items": {
@@ -2201,6 +2201,17 @@ CONFIG_METADATA_2 = {
                             },
                         },
                     },
+                    "skills": {
+                        "type": "object",
+                        "items": {
+                            "enable": {
+                                "type": "bool",
+                            },
+                            "runtime": {
+                                "type": "string",
+                            },
+                        },
+                    },
                 },
             },
             "provider_stt_settings": {
@@ -2578,6 +2589,7 @@ CONFIG_METADATA_3 = {
             # },
             "sandbox": {
                 "description": "Agent 沙箱环境",
+                "hint": "",
                 "type": "object",
                 "items": {
                     "provider_settings.sandbox.enable": {
@@ -2589,6 +2601,7 @@ CONFIG_METADATA_3 = {
                         "description": "沙箱环境驱动器",
                         "type": "string",
                         "options": ["shipyard"],
+                        "labels": ["Shipyard"],
                         "condition": {
                             "provider_settings.sandbox.enable": True,
                         },
@@ -2631,6 +2644,27 @@ CONFIG_METADATA_3 = {
                         },
                     },
                 },
+                "condition": {
+                    "provider_settings.agent_runner_type": "local",
+                    "provider_settings.enable": True,
+                },
+            },
+            "skills": {
+                "description": "Skills",
+                "type": "object",
+                "items": {
+                    "provider_settings.skills.runtime": {
+                        "description": "Skill Runtime",
+                        "type": "string",
+                        "options": ["local", "sandbox"],
+                        "labels": ["本地", "沙箱"],
+                        "hint": "选择 Skills 运行环境。使用沙箱时需先启用沙箱环境。",
+                    },
+                },
+                "condition": {
+                    "provider_settings.agent_runner_type": "local",
+                    "provider_settings.enable": True,
+                },
             },
             "truncate_and_compress": {
                 "description": "上下文管理策略",
@@ -2691,6 +2725,10 @@ CONFIG_METADATA_3 = {
                         },
                     },
                 },
+                "condition": {
+                    "provider_settings.agent_runner_type": "local",
+                    "provider_settings.enable": True,
+                },
             },
             "others": {
                 "description": "其他配置",
@@ -2778,6 +2816,16 @@ CONFIG_METADATA_3 = {
                             "provider_settings.agent_runner_type": "local",
                         },
                     },
+                    "provider_settings.tool_schema_mode": {
+                        "description": "工具调用模式",
+                        "type": "string",
+                        "options": ["skills_like", "full"],
+                        "labels": ["Skills-like（两阶段）", "Full（完整参数）"],
+                        "hint": "skills-like 先下发工具名称与描述，再下发参数；full 一次性下发完整参数。",
+                        "condition": {
+                            "provider_settings.agent_runner_type": "local",
+                        },
+                    },
                     "provider_settings.wake_prefix": {
                         "description": "LLM 聊天额外唤醒前缀 ",
                         "type": "string",
@@ -3045,7 +3093,8 @@ CONFIG_METADATA_3 = {
                         "type": "bool",
                     },
                     "platform_settings.segmented_reply.interval_method": {
-                        "description": "间隔方法",
+                        "description": "间隔方法。",
+                        "hint": "random 为随机时间，log 为根据消息长度计算，$y=log_<log_base>(x)$，x为字数，y的单位为秒。",
                         "type": "string",
                         "options": ["random", "log"],
                     },
@@ -3060,13 +3109,14 @@ CONFIG_METADATA_3 = {
                     "platform_settings.segmented_reply.log_base": {
                         "description": "对数底数",
                         "type": "float",
-                        "hint": "对数间隔的底数，默认为 2.0。取值范围为 1.0-10.0。",
+                        "hint": "对数间隔的底数，默认为 2.6。取值范围为 1.0-10.0。",
                         "condition": {
                             "platform_settings.segmented_reply.interval_method": "log",
                         },
                     },
                     "platform_settings.segmented_reply.words_count_threshold": {
                         "description": "分段回复字数阈值",
+                        "hint": "分段回复的字数上限。只有字数小于此值的消息才会被分段，超过此值的长消息将直接发送（不分段）。默认为 150",
                         "type": "int",
                     },
                     "platform_settings.segmented_reply.split_mode": {
@@ -3077,6 +3127,7 @@ CONFIG_METADATA_3 = {
                     },
                     "platform_settings.segmented_reply.regex": {
                         "description": "分段正则表达式",
+                        "hint": "用于分隔一段消息。默认情况下会根据句号、问号等标点符号分隔。如填写 `[。？！]` 将移除所有的句号、问号、感叹号。re.findall(r'<regex>', text)",
                         "type": "string",
                         "condition": {
                             "platform_settings.segmented_reply.split_mode": "regex",
@@ -3246,6 +3297,7 @@ DEFAULT_VALUE_MAP = {
     "string": "",
     "text": "",
     "list": [],
+    "file": [],
     "object": {},
     "template_list": [],
 }

astrbot/core/db/__init__.py

@@ -254,6 +254,7 @@ class BaseDatabase(abc.ABC):
         system_prompt: str,
         begin_dialogs: list[str] | None = None,
         tools: list[str] | None = None,
+        skills: list[str] | None = None,
         folder_id: str | None = None,
         sort_order: int = 0,
     ) -> Persona:
@@ -264,6 +265,7 @@ class BaseDatabase(abc.ABC):
             system_prompt: System prompt for the persona
             begin_dialogs: Optional list of initial dialog strings
             tools: Optional list of tool names (None means all tools, [] means no tools)
+            skills: Optional list of skill names (None means all skills, [] means no skills)
             folder_id: Optional folder ID to place the persona in (None means root)
             sort_order: Sort order within the folder (default 0)
         """
@@ -286,6 +288,7 @@ class BaseDatabase(abc.ABC):
         system_prompt: str | None = None,
         begin_dialogs: list[str] | None = None,
         tools: list[str] | None = None,
+        skills: list[str] | None = None,
     ) -> Persona | None:
         """Update a persona's system prompt or begin dialogs."""
         ...