Source code for agentscope_runtime.sandbox.box.base.base_sandbox

# -*- coding: utf-8 -*-
from typing import Optional

from ...constant import IMAGE_TAG
from ...registry import SandboxRegistry
from ...enums import SandboxType
from ...box.sandbox import Sandbox




@SandboxRegistry.register(
    f"agentscope/runtime-sandbox-base:{IMAGE_TAG}",
    sandbox_type=SandboxType.BASE,
    security_level="medium",
    timeout=30,
    description="Base Sandbox",
)
class BaseSandbox(Sandbox):


    def __init__(
        self,
        sandbox_id: Optional[str] = None,
        timeout: int = 3000,
        base_url: Optional[str] = None,
        bearer_token: Optional[str] = None,
    ):
        super().__init__(
            sandbox_id,
            timeout,
            base_url,
            bearer_token,
            SandboxType.BASE,
        )




    def run_ipython_cell(self, code: str):
        return self.call_tool("run_ipython_cell", {"code": code})




    def run_shell_command(self, command: str):
        return self.call_tool("run_shell_command", {"command": command})