Jiayi Zhang
d0cf79b6f4
Pre-commit checks / pre-commit-check (push) Failing after 1s
Add E2B sandbox for code execution, and basic Sandbox for computer use.
37 lines
1.6 KiB
Python
37 lines
1.6 KiB
Python
from typing import Optional
|
|
from app.tool.base import BaseTool, ToolResult
|
|
from app.config import config
|
|
|
|
class LinuxSandboxTool(BaseTool):
|
|
"""A tool to start an e2b DesktopSandbox and return the VNC URL, using config.cloud_sandbox for credentials."""
|
|
|
|
name: str = "linux_sandbox"
|
|
description: str = (
|
|
"Starts an e2b DesktopSandbox and returns the VNC URL. "
|
|
"API key and domain are read from config.cloud_sandbox."
|
|
)
|
|
parameters: dict = {"type": "object", "properties": {}, "required": []}
|
|
|
|
async def execute(self, **kwargs) -> ToolResult:
|
|
"""
|
|
Start an e2b DesktopSandbox and return the VNC URL, using config.cloud_sandbox.
|
|
"""
|
|
try:
|
|
from e2b_desktop import Sandbox as DesktopSandbox
|
|
except ImportError:
|
|
return ToolResult(error="e2b_desktop is not installed. Please install it via pip.")
|
|
sandbox_config = getattr(config, "cloud_sandbox", None)
|
|
if not sandbox_config or not sandbox_config.get("api_key") or not sandbox_config.get("domain"):
|
|
return ToolResult(error="cloud_sandbox config missing or incomplete. Please set api_key and domain in config.cloud_sandbox.")
|
|
try:
|
|
desktop = DesktopSandbox(api_key=sandbox_config["api_key"], domain=sandbox_config["domain"])
|
|
desktop.stream.start()
|
|
url = desktop.stream.get_url()
|
|
return ToolResult(output={
|
|
"status": "started",
|
|
"vnc_url": url,
|
|
"domain": sandbox_config["domain"],
|
|
})
|
|
except Exception as e:
|
|
return ToolResult(error=f"Failed to start e2b DesktopSandbox: {e}")
|