scalebox-sdk 0.1.25__py3-none-any.whl → 1.0.1__py3-none-any.whl

scalebox/test/test_download_url.py

@@ -0,0 +1,49 @@
+from scalebox.code_interpreter import Sandbox
+import json
+
+# 1. 准备要上传的文件内容
+print("正在准备文件内容...")
+results_data = {
+    "experiment_id": "exp_001",
+    "status": "completed",
+    "metrics": {
+        "accuracy": 0.952,
+        "precision": 0.94,
+        "recall": 0.96
+    },
+    "timestamp": "2024-01-01T12:00:00Z",
+    "details": [
+        {"epoch": 1, "loss": 0.45},
+        {"epoch": 2, "loss": 0.32},
+        {"epoch": 3, "loss": 0.18}
+    ]
+}
+
+# 转换为 JSON 字符串
+json_content = json.dumps(results_data, indent=2)
+
+# 2. 创建沙箱并写入文件
+print("\n正在创建沙箱并写入文件...")
+sandbox = Sandbox.create(timeout=1800)
+remote_path = "/home/user/results.json"
+
+# 关键修改：使用 sandbox.write() 写入文件
+sandbox.files.write(remote_path, json_content)
+print(f"✅ 文件已写入沙箱: {remote_path}")
+
+# 3. 获取预签名下载 URL
+print("\n正在生成下载链接...")
+download_url = sandbox.download_url(
+    path=remote_path,
+    use_signature_expiration=180  # 180秒有效期
+)
+
+print(f"\n📥 下载 URL: {download_url}")
+print(f"⏰ 链接将在 3 分钟后过期")
+
+upload_url = sandbox.upload_url(
+    path="/home/user",
+    use_signature_expiration=360
+)
+print(f"\n📥 上传 URL: {upload_url}")
+print(f"⏰ 链接将在 6 分钟后过期")

scalebox/test/test_sandbox_async_comprehensive.py

@@ -78,7 +78,7 @@ class AsyncSandboxValidator:
             template="base",
             debug=False,
             timeout=300,
-            api_key="sk-Wk4IgtUYOqnttxGaxZmELEV4p2FXh15Evt0FIcSa",
+            api_key="sk-Wk4Ig",
             metadata={"test": "async_validation"},
             envs={"TEST_ENV": "async_test"},
         )

scalebox/test/test_sandbox_object_storage_example.py

@@ -0,0 +1,146 @@
+#!/usr/bin/env python3
+"""
+示例：如何在创建同步 Sandbox 时携带 object_storage 参数。
+
+运行前准备：
+1. 设置环境变量 SCALEBOX_OBJECT_STORAGE_CONFIG 为 JSON 字符串或 JSON 文件路径，
+   该内容会被原样传递给后端。例如：
+   {
+       "provider": "s3",
+       "bucket": "my-sbx-bucket",
+       "region": "us-east-1",
+       "access_key_id": "AKIA...",
+       "secret_access_key": "******",
+       "endpoint": "https://s3.amazonaws.com"
+   }
+2. （可选）设置 SCALEBOX_OBJECT_STORAGE_TEMPLATE / SCALEBOX_OBJECT_STORAGE_TIMEOUT
+   调整模板和沙箱超时时间。
+
+注意：目前只有同步版本支持 object_storage，本示例不会自动运行，
+      需要在提供有效配置后手动执行：
+
+      PYTHONPATH=/home/ubuntu/git_home/scalebox:$PYTHONPATH \\
+      SCALEBOX_OBJECT_STORAGE_CONFIG=/path/to/config.json \\
+      python test/test_sandbox_object_storage_example.py
+"""
+
+from __future__ import annotations
+
+import json
+import logging
+import os
+import sys
+from pathlib import Path
+from typing import Any, Dict, Optional
+
+from scalebox.exceptions import SandboxException
+from scalebox.sandbox_sync.main import Sandbox
+from scalebox.sandbox.sandbox_api import SandboxQuery
+
+logger = logging.getLogger(__name__)
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
+
+# CONFIG_ENV = "SCALEBOX_OBJECT_STORAGE_CONFIG"
+# TEMPLATE_ENV = "SCALEBOX_OBJECT_STORAGE_TEMPLATE"
+# TIMEOUT_ENV = "SCALEBOX_OBJECT_STORAGE_TIMEOUT"
+#
+#
+# def _load_object_storage_payload() -> Optional[Dict[str, Any]]:
+#     raw_value = os.environ.get(CONFIG_ENV)
+#     if not raw_value:
+#         logger.warning(
+#             "未检测到 %s 环境变量，跳过 object_storage 示例。请设置 JSON 配置后重新运行。",
+#             CONFIG_ENV,
+#         )
+#         return None
+#
+#     config_text: Optional[str] = None
+#     potential_path = Path(raw_value)
+#     if potential_path.is_file():
+#         config_text = potential_path.read_text(encoding="utf-8")
+#     else:
+#         config_text = raw_value
+#
+#     try:
+#         payload = json.loads(config_text)
+#     except json.JSONDecodeError as exc:
+#         logger.error("无法解析 object_storage 配置（需合法 JSON）：%s", exc)
+#         raise SystemExit(1) from exc
+#
+#     if not isinstance(payload, dict):
+#         logger.error("object_storage 配置必须是 JSON 对象，示例详见文件头部注释。")
+#         raise SystemExit(1)
+#
+#     return payload
+
+
+# def _mask_value(value: str) -> str:
+#     if len(value) <= 4:
+#         return "*" * len(value)
+#     return f"{value[:2]}***{value[-2:]}"
+
+
+# def _display_object_storage_keys(config: Dict[str, Any]) -> Dict[str, Any]:
+#     masked = {}
+#     for key, value in config.items():
+#         if isinstance(value, str):
+#             masked[key] = _mask_value(value)
+#         else:
+#             masked[key] = value
+#     return masked
+
+
+def create_sandbox_with_object_storage():
+    # payload = _load_object_storage_payload()
+    # if payload is None:
+    #     return
+    #
+    # template = os.environ.get(TEMPLATE_ENV, "base")
+    # timeout = int(os.environ.get(TIMEOUT_ENV, "600"))
+
+    sandbox: Optional[Sandbox] = None
+    try:
+        sandbox = Sandbox.create(
+            template="base",
+            timeout=3600,
+            metadata={"example": "object_storage"},
+            object_storage={"uri": "s3://bgd-test/test/",
+                            "mount_point": "/mnt/oss",
+                            # "access_key": "",
+                            # "secret_key": "",
+                            "region": "us-east-2",
+                            "endpoint": "https://s3.us-east-2.amazonaws.com",},
+        )
+        print(sandbox.sandbox_id)
+        print(sandbox.sandbox_domain)
+        if sandbox.object_storage:
+            print(sandbox.object_storage)
+
+        info = sandbox.get_info()
+        print(info)
+        print(f"get_info.object_storage: {info.object_storage}")
+
+        listed = Sandbox.list(query=SandboxQuery(metadata={"example": "object_storage"}))
+        print(f"List 根据 metadata 命中的沙箱数量: {len(listed)}")
+        for listed_box in listed:
+            detailed = Sandbox.get_info(sandbox_id=listed_box.sandbox_id)
+            print(detailed)
+            print(
+                f"list -> sandbox {listed_box.sandbox_id} object_storage: {detailed.object_storage}"
+            )
+
+        # 在沙箱中运行简单命令，确认可正常访问
+        result = sandbox.commands.run("echo 'object storage sandbox is ready'")
+        print(f"命令输出: {result.stdout.strip()}")
+    except SandboxException as exc:
+        logger.error("创建或使用沙箱失败：%s", exc)
+        raise
+    finally:
+        if sandbox is not None:
+            # sandbox.kill()
+            logger.info("沙箱已销毁")
+
+
+if __name__ == "__main__":
+    create_sandbox_with_object_storage()
+

scalebox/test/test_sandbox_object_storage_example_async.py

@@ -0,0 +1,156 @@
+#!/usr/bin/env python3
+# pyright: reportMissingImports=false
+"""
+示例：如何在创建异步 Sandbox 时携带 object_storage 参数。
+
+运行前准备：
+1. 设置环境变量 SCALEBOX_OBJECT_STORAGE_CONFIG 为 JSON 字符串或 JSON 文件路径，
+   该内容会被原样传递给后端。例如：
+   {
+       "provider": "s3",
+       "bucket": "my-sbx-bucket",
+       "region": "us-east-1",
+       "access_key_id": "AKIA...",
+       "secret_access_key": "******",
+       "endpoint": "https://s3.amazonaws.com"
+   }
+2. （可选）设置 SCALEBOX_OBJECT_STORAGE_TEMPLATE / SCALEBOX_OBJECT_STORAGE_TIMEOUT
+   调整模板和沙箱超时时间。
+
+注意：object_storage 功能仍主要在同步版本可用，异步示例取决于后端实际支持状况。
+      需要在提供有效配置后手动执行：
+
+      PYTHONPATH=/home/ubuntu/git_home/scalebox:$PYTHONPATH \\
+      SCALEBOX_OBJECT_STORAGE_CONFIG=/path/to/config.json \\
+      python test/test_sandbox_object_storage_example_async.py
+"""
+
+from __future__ import annotations
+
+import asyncio
+import json
+import logging
+import os
+import sys
+from pathlib import Path
+from typing import Any, Dict, Optional
+
+WORKSPACE_ROOT = Path(__file__).resolve().parents[1]
+if str(WORKSPACE_ROOT) not in sys.path:
+    sys.path.insert(0, str(WORKSPACE_ROOT))
+
+from scalebox.exceptions import SandboxException
+from scalebox.sandbox_async.main import AsyncSandbox
+from scalebox.sandbox.sandbox_api import SandboxQuery
+
+logger = logging.getLogger(__name__)
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
+
+# CONFIG_ENV = "SCALEBOX_OBJECT_STORAGE_CONFIG"
+# TEMPLATE_ENV = "SCALEBOX_OBJECT_STORAGE_TEMPLATE"
+# TIMEOUT_ENV = "SCALEBOX_OBJECT_STORAGE_TIMEOUT"
+#
+#
+# def _load_object_storage_payload() -> Optional[Dict[str, Any]]:
+#     raw_value = os.environ.get(CONFIG_ENV)
+#     if not raw_value:
+#         logger.warning(
+#             "未检测到 %s 环境变量，跳过 object_storage 示例。请设置 JSON 配置后重新运行。",
+#             CONFIG_ENV,
+#         )
+#         return None
+#
+#     config_text: Optional[str] = None
+#     potential_path = Path(raw_value)
+#     if potential_path.is_file():
+#         config_text = potential_path.read_text(encoding="utf-8")
+#     else:
+#         config_text = raw_value
+#
+#     try:
+#         payload = json.loads(config_text)
+#     except json.JSONDecodeError as exc:
+#         logger.error("无法解析 object_storage 配置（需合法 JSON）：%s", exc)
+#         raise SystemExit(1) from exc
+#
+#     if not isinstance(payload, dict):
+#         logger.error("object_storage 配置必须是 JSON 对象，示例详见文件头部注释。")
+#         raise SystemExit(1)
+#
+#     return payload
+#
+#
+# def _mask_value(value: str) -> str:
+#     if len(value) <= 4:
+#         return "*" * len(value)
+#     return f"{value[:2]}***{value[-2:]}"
+#
+#
+# def _display_object_storage_keys(config: Dict[str, Any]) -> Dict[str, Any]:
+#     masked = {}
+#     for key, value in config.items():
+#         if isinstance(value, str):
+#             masked[key] = _mask_value(value)
+#         else:
+#             masked[key] = value
+#     return masked
+
+
+async def create_sandbox_with_object_storage_async():
+    # payload = _load_object_storage_payload()
+    # if payload is None:
+    #     return
+    #
+    # template = os.environ.get(TEMPLATE_ENV, "base")
+    # timeout = int(os.environ.get(TIMEOUT_ENV, "600"))
+
+    sandbox: Optional[AsyncSandbox] = None
+    try:
+        sandbox = await AsyncSandbox.create(
+            template="base",
+            timeout=3600,
+            metadata={"example": "object_storage_async"},
+            object_storage={
+                "uri": "s3://bgd-test/test/",
+                "mount_point": "/mnt/oss",
+                "access_key": "",
+                "secret_key": "",
+                "region": "us-east-2",
+                "endpoint": "https://s3.us-east-2.amazonaws.com",
+            },
+        )
+        print(sandbox.sandbox_id)
+        print(sandbox.sandbox_domain)
+        if sandbox.object_storage:
+            print(sandbox.object_storage)
+
+        info = await sandbox.get_info()
+        print(info)
+        print(f"get_info.object_storage: {info.object_storage}")
+
+        listed = await AsyncSandbox.list(
+            query=SandboxQuery(metadata={"example": "object_storage_async"})
+        )
+        print(f"Async list 根据 metadata 命中的沙箱数量: {len(listed)}")
+        for listed_box in listed:
+            detailed = await AsyncSandbox.get_info(sandbox_id=listed_box.sandbox_id)
+            print(detailed)
+            print(
+                f"list -> sandbox {listed_box.sandbox_id} object_storage: {detailed.object_storage}"
+            )
+
+        # 在沙箱中运行简单命令，确认可正常访问
+        result = await sandbox.commands.run("echo 'async object storage sandbox is ready'")
+        print(f"命令输出: {result.stdout.strip()}")
+    except SandboxException as exc:
+        logger.error("创建或使用异步沙箱失败：%s", exc)
+        raise
+    finally:
+        if sandbox is not None:
+            # await sandbox.kill()
+            logger.info("异步沙箱已销毁")
+
+
+if __name__ == "__main__":
+    asyncio.run(create_sandbox_with_object_storage_async())
+

scalebox/test/test_sf.py

@@ -0,0 +1,137 @@
+from browser_use import Agent, BrowserProfile
+from playwright.async_api import async_playwright
+from dotenv import load_dotenv
+import os, asyncio
+from browser_use.llm import ChatOpenAI
+import subprocess, pathlib
+
+load_dotenv()
+os.environ["PLAYWRIGHT_CHROMIUM_EXTRA_ARGS"] = (
+    "--no-sandbox --disable-dev-shm-usage --disable-gpu --disable-software-rasterizer"
+)
+
+# 1. 正常实例化 ChatOpenAI
+_chat = ChatOpenAI(
+    model="Qwen/Qwen2.5-72B-Instruct",
+    api_key="sk-nbxsvrydquzfkfrrnnuutnfchpghzeuxilgawvsjqzjhumkh",
+    base_url="https://api.siliconflow.cn/v1",
+    temperature=0.1
+)
+
+
+llm = ChatOpenAI(
+    model="deepseek-ai/DeepSeek-R1",
+    api_key="sk-nbxsvrydquzfkfrrnnuutnfchpghzeuxilgawvsjqzjhumkh",
+    base_url="https://api.siliconflow.cn/v1",
+    temperature=0.7
+)
+# 2. 包一层模型，补 .provider + 转发所有必要方法
+class SiliconFlowModel:
+    provider = "siliconflow"
+
+    def __init__(self, chat: ChatOpenAI):
+        self.chat = chat
+        # 把 browser-use 可能用到的字段全部代理
+        self.model_name = chat.model_name  # ← 新增
+        self.model = chat.model_name  # ← 新增（防它再换字段）
+        self.ainvoke = chat.ainvoke
+        self.invoke = chat.invoke
+        self.bind = chat.bind
+
+    # === 同步 ===
+    def invoke(self, *args, **kwargs):
+        return self.chat.invoke(*args, **kwargs)
+
+    # === 异步 ===
+    async def ainvoke(self, *args, **kwargs):
+        return await self.chat.ainvoke(*args, **kwargs)
+
+    # 如有其他缺失方法，继续代理即可
+    def bind(self, *args, **kwargs):
+        return self.chat.bind(*args, **kwargs)
+
+# llm = SiliconFlowModel(_chat)
+
+# 3. 创建 Agent
+async def main():
+    # chrome_path = subprocess.check_output(
+    #     ["which", "google-chrome-stable"], text=True
+    # ).strip()
+    # browser_profile = BrowserProfile(
+    #     headless=True,  # 先跑通再改 False
+    #     # executable_path="/usr/bin/google-chrome-stable",
+    #     extra_args=[
+    #         "--no-sandbox",
+    #         "--disable-dev-shm-usage",
+    #         "--disable-gpu",
+    #         "--disable-software-rasterizer",
+    #         "--remote-debugging-port=9222",
+    #         "--remote-debugging-address=0.0.0.0",  # 允许所有地址连接
+    #         "--user-data-dir=/tmp/chrome-manual-test"  # 指定用户数据目录
+    #     ],
+    # )
+    # "--disable-dev-shm-usage",
+    # "--disable-gpu",
+    # "--disable-software-rasterizer",
+    async with async_playwright() as pw:
+        browser = await pw.chromium.launch(
+            proxy={
+                "server": "http://net-proxy:80",
+                "username":"sbx-jz9kgqolumvmfc7ds",
+                "password": "npt-Vd4CpGtjBOQEI8wdBv3wSOd8P4Bup1NgfsM"
+            },
+            headless=False,
+            args=[
+                # '--no-sandbox',
+
+                # '--disable-dev-shm-usage',
+                # '--disable-gpu',
+                # '--disable-extensions',
+                # '--disable-background-timer-throttling',
+                # '--disable-renderer-backgrounding',
+                # '--disable-features=TranslateUI',
+                "--disable-blink-features=AutomationControlled",  # 禁用自动化控制标志
+                "--start-maximized",  # 最大化窗口
+            ]
+        )
+        # task = """
+        #     1. 打开淘宝官网 (taobao.com)
+        #     2. 搜索"格力空调"
+        #     3. 记录下第一款商品的价格和标题
+        #     4. 打开京东官网 (jd.com)
+        #     5. 同样搜索"格力空调"
+        #     6. 记录下第一款商品的价格和标题
+        #     7. 对比在两个平台上看到的第一个商品的价格
+        #     """
+        task = """
+            1. 打开google搜索
+            2、输入今日新闻
+            3、输出第一条内容
+            """
+        # task = """
+        #     1、打开新浪网(http://sina.cn/)
+        #     2. 搜索"科技"
+        #     3. 记录前10条记录
+        #     """
+        agent = Agent(
+            task=task,
+            llm=llm,
+            # browser_profile=browser_profile,
+            browser=browser,
+            use_vision=False
+        )
+        result = await agent.run()
+        print("---------- 最终结果 ----------")
+        print(result)
+        await browser.close()
+
+# if __name__ == "__main__":
+#     asyncio.run(main())
+loop = asyncio.get_event_loop()
+if loop.is_closed():
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+try:
+    loop.run_until_complete(main())
+finally:
+    loop.close()

scalebox/test/test_watch_dir_async.py

@@ -0,0 +1,56 @@
+#!/usr/bin/env python3
+"""
+Minimal async test for sandbox_async.filesystem.watch_dir.
+Creates an AsyncSandbox, watches a temp directory, writes a file to trigger
+an event, asserts events were received, and cleans up.
+"""
+
+import asyncio
+import time
+import logging
+from typing import List
+
+from scalebox.sandbox_async.main import AsyncSandbox
+from scalebox.sandbox.filesystem.watch_handle import FilesystemEvent
+
+
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
+logger = logging.getLogger(__name__)
+
+
+async def main():
+    async with await AsyncSandbox.create(template="base", timeout=300, metadata={"test": "watch_dir_async"}) as sandbox:
+        watch_path = "/tmp/watch_async_test"
+        await sandbox.files.make_dir(watch_path)
+
+        events: List[FilesystemEvent] = []
+
+        async def on_event(ev: FilesystemEvent):
+            print(ev)
+            events.append(ev)
+
+        # Start watching directory
+        handle = await sandbox.files.watch_dir(
+            path=watch_path,
+            on_event=on_event,
+            timeout=10,
+            recursive=False,
+        )
+
+        # Trigger a filesystem event
+        await sandbox.files.write(f"{watch_path}/new_file.txt", "content")
+
+        # Wait a bit for event propagation
+        await asyncio.sleep(2)
+
+        # Stop watching
+        await handle.stop()
+
+        logger.info(f"Received {len(events)} filesystem events")
+        assert len(events) >= 1
+
+
+if __name__ == "__main__":
+    asyncio.run(main())
+
+

scalebox/test/testacreate.py

@@ -7,7 +7,7 @@ from scalebox.sandbox_async import AsyncSandbox
 async def main():
     sandbox = await AsyncSandbox.create(
         template="base",  # 或自定义模板名
-        api_key="sk-Wk4IgtUYOqnttxGaxZmELEV4p2FXh15Evt0FIcSa",
+        api_key="sk-Wk4IgtUYOqnttx",
         timeout=3600,
     )
     print("✅ 沙箱已启动，sandbox_domain:", sandbox.sandbox_domain)

scalebox/test/testagetinfo.py

@@ -6,7 +6,7 @@ from sandbox_async import AsyncSandbox
 
 async def main():
     sandbox = await AsyncSandbox.get_info(
-        sandbox_id="", api_key="sk-Wk4IgtUYOqnttxGaxZmELEV4p2FXh15Evt0FIcSa"
+        sandbox_id="", api_key="sk-Wk4Ig"
     )
     print("✅ 沙箱已启动，sandbox_domain:", sandbox.sandbox_domain)
     print("✅ 沙箱已启动，ID:", sandbox.sandbox_id)