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

scalebox/test/test_sandbox_async_comprehensive.py

@@ -274,7 +274,9 @@ class AsyncSandboxValidator:
         await self.sandbox.files.write("/tmp/old_name.txt", "content to rename")
 
         # 重命名
-        result = await self.sandbox.files.rename("/tmp/old_name.txt", "/tmp/new_name.txt")
+        result = await self.sandbox.files.rename(
+            "/tmp/old_name.txt", "/tmp/new_name.txt"
+        )
         print(result)
         assert isinstance(result, EntryInfo)
         assert result.name == "new_name.txt"
@@ -290,7 +292,7 @@ class AsyncSandboxValidator:
         # 删除文件
         await self.sandbox.files.remove("/tmp/new_name.txt")
         assert await self.sandbox.files.exists("/tmp/new_name.txt") == False
-        
+
         # 删除目录
         await self.sandbox.files.remove("/tmp/test_dir")
         assert await self.sandbox.files.exists("/tmp/test_dir") == False

scalebox/test/test_sandbox_object_storage_example.py

@@ -38,7 +38,9 @@ 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")
+logging.basicConfig(
+    level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
+)
 
 # CONFIG_ENV = "SCALEBOX_OBJECT_STORAGE_CONFIG"
 # TEMPLATE_ENV = "SCALEBOX_OBJECT_STORAGE_TEMPLATE"
@@ -104,12 +106,14 @@ def create_sandbox_with_object_storage():
             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",},
+            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)
@@ -120,7 +124,9 @@ def create_sandbox_with_object_storage():
         print(info)
         print(f"get_info.object_storage: {info.object_storage}")
 
-        listed = Sandbox.list(query=SandboxQuery(metadata={"example": "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)
@@ -143,4 +149,3 @@ def create_sandbox_with_object_storage():
 
 if __name__ == "__main__":
     create_sandbox_with_object_storage()
-

scalebox/test/test_sandbox_object_storage_example_async.py

@@ -44,7 +44,9 @@ 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")
+logging.basicConfig(
+    level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
+)
 
 # CONFIG_ENV = "SCALEBOX_OBJECT_STORAGE_CONFIG"
 # TEMPLATE_ENV = "SCALEBOX_OBJECT_STORAGE_TEMPLATE"
@@ -140,7 +142,9 @@ async def create_sandbox_with_object_storage_async():
             )
 
         # 在沙箱中运行简单命令，确认可正常访问
-        result = await sandbox.commands.run("echo 'async object storage sandbox is ready'")
+        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)
@@ -153,4 +157,3 @@ async def create_sandbox_with_object_storage_async():
 
 if __name__ == "__main__":
     asyncio.run(create_sandbox_with_object_storage_async())
-

scalebox/test/test_sandbox_sync_comprehensive.py

@@ -292,7 +292,7 @@ class SandboxValidator:
         # 删除文件
         self.sandbox.files.remove("/tmp/new_name.txt")
         assert self.sandbox.files.exists("/tmp/new_name.txt") == False
-        
+
         # 删除目录
         self.sandbox.files.remove("/tmp/test_dir")
         assert self.sandbox.files.exists("/tmp/test_dir") == False

scalebox/test/test_sf.py

@@ -15,7 +15,7 @@ _chat = ChatOpenAI(
     model="Qwen/Qwen2.5-72B-Instruct",
     api_key="sk-nbxsvrydquzfkfrrnnuutnfchpghzeuxilgawvsjqzjhumkh",
     base_url="https://api.siliconflow.cn/v1",
-    temperature=0.1
+    temperature=0.1,
 )
 
 
@@ -23,8 +23,10 @@ llm = ChatOpenAI(
     model="deepseek-ai/DeepSeek-R1",
     api_key="sk-nbxsvrydquzfkfrrnnuutnfchpghzeuxilgawvsjqzjhumkh",
     base_url="https://api.siliconflow.cn/v1",
-    temperature=0.7
+    temperature=0.7,
 )
+
+
 # 2. 包一层模型，补 .provider + 转发所有必要方法
 class SiliconFlowModel:
     provider = "siliconflow"
@@ -50,8 +52,10 @@ class SiliconFlowModel:
     def bind(self, *args, **kwargs):
         return self.chat.bind(*args, **kwargs)
 
+
 # llm = SiliconFlowModel(_chat)
 
+
 # 3. 创建 Agent
 async def main():
     # chrome_path = subprocess.check_output(
@@ -77,13 +81,12 @@ async def main():
         browser = await pw.chromium.launch(
             proxy={
                 "server": "http://net-proxy:80",
-                "username":"sbx-jz9kgqolumvmfc7ds",
-                "password": "npt-Vd4CpGtjBOQEI8wdBv3wSOd8P4Bup1NgfsM"
+                "username": "sbx-jz9kgqolumvmfc7ds",
+                "password": "npt-Vd4CpGtjBOQEI8wdBv3wSOd8P4Bup1NgfsM",
             },
             headless=False,
             args=[
                 # '--no-sandbox',
-
                 # '--disable-dev-shm-usage',
                 # '--disable-gpu',
                 # '--disable-extensions',
@@ -92,7 +95,7 @@ async def main():
                 # '--disable-features=TranslateUI',
                 "--disable-blink-features=AutomationControlled",  # 禁用自动化控制标志
                 "--start-maximized",  # 最大化窗口
-            ]
+            ],
         )
         # task = """
         #     1. 打开淘宝官网 (taobao.com)
@@ -118,13 +121,14 @@ async def main():
             llm=llm,
             # browser_profile=browser_profile,
             browser=browser,
-            use_vision=False
+            use_vision=False,
         )
         result = await agent.run()
         print("---------- 最终结果 ----------")
         print(result)
         await browser.close()
 
+
 # if __name__ == "__main__":
 #     asyncio.run(main())
 loop = asyncio.get_event_loop()
@@ -134,4 +138,4 @@ if loop.is_closed():
 try:
     loop.run_until_complete(main())
 finally:
-    loop.close()
+    loop.close()

scalebox/test/test_watch_dir_async.py

@@ -14,12 +14,16 @@ 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")
+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:
+    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)
 
@@ -52,5 +56,3 @@ async def main():
 
 if __name__ == "__main__":
     asyncio.run(main())
-
-

scalebox/test/testagetinfo.py

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

scalebox/test/testsandbox_api.py

@@ -3,11 +3,13 @@ import time
 from scalebox.sandbox.commands.command_handle import PtySize
 from scalebox.sandbox_sync.main import Sandbox
 
+
 def output_handler(output):
     """处理 输出的回调函数"""
     print(f"PTY 输出: {output}")
 
-sandboxes=Sandbox.list()
+
+sandboxes = Sandbox.list()
 print(sandboxes)
-box=Sandbox.connect(sandbox_id="sbx-0b3ccqezz67asn3ax")
-print(box.files.list("/"))
+box = Sandbox.connect(sandbox_id="sbx-0b3ccqezz67asn3ax")
+print(box.files.list("/"))

scalebox/test/testsandbox_async.py

@@ -13,7 +13,7 @@ async def pty_output_handler(output):
 
 
 async def main():
-    sandbox = await (AsyncSandbox.create(timeout=3600))
+    sandbox = await AsyncSandbox.create(timeout=3600)
     proc = await sandbox.commands.run("echo hello from async")
     print("exit_code =", proc.exit_code)
     print("stdout   =", proc.stdout)
@@ -24,99 +24,69 @@ async def main():
         on_data=pty_output_handler,
         cwd="/root",
         envs={"CUSTOM_ENV": "value"},
-        request_timeout=40
+        request_timeout=40,
     )
     print(f"PTY 已创建，PID: {pty_handle.pid}")
 
     # 5. 向 PTY 发送输入
     await sandbox.pty.send_stdin(
-        pid=pty_handle.pid,
-        data=b"echo 'Hello from PTY 0s'\n",
-        request_timeout=10
+        pid=pty_handle.pid, data=b"echo 'Hello from PTY 0s'\n", request_timeout=10
     )
     await asyncio.sleep(30)
     await sandbox.pty.send_stdin(
-        pid=pty_handle.pid,
-        data=b"echo 'Hello from PTY 30s'\n",
-        request_timeout=10
+        pid=pty_handle.pid, data=b"echo 'Hello from PTY 30s'\n", request_timeout=10
     )
     await asyncio.sleep(30)
     await sandbox.pty.send_stdin(
-        pid=pty_handle.pid,
-        data=b"echo 'Hello from PTY 60s'\n",
-        request_timeout=10
+        pid=pty_handle.pid, data=b"echo 'Hello from PTY 60s'\n", request_timeout=10
     )
     await asyncio.sleep(30)
     await sandbox.pty.send_stdin(
-        pid=pty_handle.pid,
-        data=b"echo 'Hello from PTY 90s'\n",
-        request_timeout=10
+        pid=pty_handle.pid, data=b"echo 'Hello from PTY 90s'\n", request_timeout=10
     )
     await asyncio.sleep(30)
     await sandbox.pty.send_stdin(
-        pid=pty_handle.pid,
-        data=b"echo 'Hello from PTY 120s'\n",
-        request_timeout=10
+        pid=pty_handle.pid, data=b"echo 'Hello from PTY 120s'\n", request_timeout=10
     )
     await asyncio.sleep(30)
     await sandbox.pty.send_stdin(
-        pid=pty_handle.pid,
-        data=b"echo 'Hello from PTY 150s'\n",
-        request_timeout=10
+        pid=pty_handle.pid, data=b"echo 'Hello from PTY 150s'\n", request_timeout=10
     )
     await asyncio.sleep(30)
     await sandbox.pty.send_stdin(
-        pid=pty_handle.pid,
-        data=b"echo 'Hello from PTY 180s'\n",
-        request_timeout=10
+        pid=pty_handle.pid, data=b"echo 'Hello from PTY 180s'\n", request_timeout=10
     )
     await asyncio.sleep(30)
     await sandbox.pty.send_stdin(
-        pid=pty_handle.pid,
-        data=b"echo 'Hello from PTY 210s'\n",
-        request_timeout=10
+        pid=pty_handle.pid, data=b"echo 'Hello from PTY 210s'\n", request_timeout=10
     )
     await asyncio.sleep(30)
     await sandbox.pty.send_stdin(
-        pid=pty_handle.pid,
-        data=b"echo 'Hello from PTY 240s'\n",
-        request_timeout=10
+        pid=pty_handle.pid, data=b"echo 'Hello from PTY 240s'\n", request_timeout=10
     )
     await asyncio.sleep(30)
     await sandbox.pty.send_stdin(
-        pid=pty_handle.pid,
-        data=b"echo 'Hello from PTY 270s'\n",
-        request_timeout=10
+        pid=pty_handle.pid, data=b"echo 'Hello from PTY 270s'\n", request_timeout=10
     )
     await asyncio.sleep(30)
     await sandbox.pty.send_stdin(
-        pid=pty_handle.pid,
-        data=b"echo 'Hello from PTY 300s'\n",
-        request_timeout=10
+        pid=pty_handle.pid, data=b"echo 'Hello from PTY 300s'\n", request_timeout=10
     )
     await asyncio.sleep(30)
     await sandbox.pty.send_stdin(
-        pid=pty_handle.pid,
-        data=b"echo 'Hello from PTY 330s'\n",
-        request_timeout=10
+        pid=pty_handle.pid, data=b"echo 'Hello from PTY 330s'\n", request_timeout=10
     )
     await asyncio.sleep(30)
     await sandbox.pty.send_stdin(
-        pid=pty_handle.pid,
-        data=b"echo 'Hello from PTY 360s'\n",
-        request_timeout=10
+        pid=pty_handle.pid, data=b"echo 'Hello from PTY 360s'\n", request_timeout=10
     )
     await asyncio.sleep(30)
     await sandbox.pty.send_stdin(
-        pid=pty_handle.pid,
-        data=b"echo 'Hello from PTY 390s'\n",
-        request_timeout=10
+        pid=pty_handle.pid, data=b"echo 'Hello from PTY 390s'\n", request_timeout=10
     )
     await asyncio.sleep(30)
     await sandbox.pty.send_stdin(
-        pid=pty_handle.pid,
-        data=b"echo 'Hello from PTY 420s'\n",
-        request_timeout=10
+        pid=pty_handle.pid, data=b"echo 'Hello from PTY 420s'\n", request_timeout=10
     )
 
     # 6. 调整 PTY 大小

scalebox/test/testsandbox_sync.py

@@ -7,6 +7,8 @@ from scalebox.sandbox_sync.main import Sandbox
 def output_handler(output):
     """处理 输出的回调函数"""
     print(f"PTY 输出: {output}")
+
+
 sandbox = Sandbox.create(api_key=f"sk-Wk4Ig")
 # print(sandbox.files.list("/root",2))
 # proc = sandbox.commands._start(
@@ -36,30 +38,32 @@ result = sandbox.commands.run(
 )
 print(result.exit_code)
 print(result.error)
-pty=sandbox.pty.create(size=PtySize(1024, 768),user="root",cwd="/root/",request_timeout=3600)
-sandbox.pty.send_stdin(pid=pty.pid,data=b"echo \"hello\"")
-sandbox.pty.send_stdin(pid=pty.pid,data=b"echo \"world!\"")
+pty = sandbox.pty.create(
+    size=PtySize(1024, 768), user="root", cwd="/root/", request_timeout=3600
+)
+sandbox.pty.send_stdin(pid=pty.pid, data=b'echo "hello"')
+sandbox.pty.send_stdin(pid=pty.pid, data=b'echo "world!"')
 time.sleep(30)
-sandbox.pty.send_stdin(pid=pty.pid,data=b"echo \"world!\" 30s")
+sandbox.pty.send_stdin(pid=pty.pid, data=b'echo "world!" 30s')
 time.sleep(30)
-sandbox.pty.send_stdin(pid=pty.pid,data=b"echo \"world!\" 60s")
+sandbox.pty.send_stdin(pid=pty.pid, data=b'echo "world!" 60s')
 time.sleep(30)
-sandbox.pty.send_stdin(pid=pty.pid,data=b"echo \"world!\" 90s")
+sandbox.pty.send_stdin(pid=pty.pid, data=b'echo "world!" 90s')
 time.sleep(30)
-sandbox.pty.send_stdin(pid=pty.pid,data=b"echo \"world!\" 120s")
+sandbox.pty.send_stdin(pid=pty.pid, data=b'echo "world!" 120s')
 time.sleep(30)
-sandbox.pty.send_stdin(pid=pty.pid,data=b"echo \"world!\" 150s")
+sandbox.pty.send_stdin(pid=pty.pid, data=b'echo "world!" 150s')
 time.sleep(30)
-sandbox.pty.send_stdin(pid=pty.pid,data=b"echo \"world!\" 180s")
+sandbox.pty.send_stdin(pid=pty.pid, data=b'echo "world!" 180s')
 time.sleep(30)
-sandbox.pty.send_stdin(pid=pty.pid,data=b"echo \"world!\" 210s")
+sandbox.pty.send_stdin(pid=pty.pid, data=b'echo "world!" 210s')
 time.sleep(30)
-sandbox.pty.send_stdin(pid=pty.pid,data=b"echo \"world!\" 240s")
+sandbox.pty.send_stdin(pid=pty.pid, data=b'echo "world!" 240s')
 time.sleep(30)
-sandbox.pty.send_stdin(pid=pty.pid,data=b"echo \"world!\" 270s")
+sandbox.pty.send_stdin(pid=pty.pid, data=b'echo "world!" 270s')
 time.sleep(30)
-sandbox.pty.send_stdin(pid=pty.pid,data=b"echo \"world!\" 300s")
-result=pty.wait(
+sandbox.pty.send_stdin(pid=pty.pid, data=b'echo "world!" 300s')
+result = pty.wait(
     on_pty=lambda data: print("[STDOUT]", data, end=""),
 )
 print("exit_code =", result.exit_code)