scalebox-sdk 0.1.0__py3-none-any.whl

scalebox/sandbox_sync/sandbox_api.py

@@ -0,0 +1,356 @@
+import datetime
+import urllib.parse
+from typing import Dict, List, Optional
+
+from packaging.version import Version
+
+from ..api import ApiClient, SandboxCreateResponse, handle_api_exception
+from ..api.client.api.sandboxes import (
+    delete_sandboxes_sandbox_id,
+    get_sandboxes,
+    get_sandboxes_sandbox_id,
+    get_sandboxes_sandbox_id_metrics,
+    post_sandboxes,
+    post_sandboxes_sandbox_id_timeout,
+)
+from ..api.client.models import Error, NewSandbox, PostSandboxesSandboxIDTimeoutBody
+from ..connection_config import ConnectionConfig, ProxyTypes
+from ..exceptions import SandboxException, TemplateException
+from ..sandbox.sandbox_api import (
+    ListedSandbox,
+    SandboxApiBase,
+    SandboxInfo,
+    SandboxMetrics,
+    SandboxQuery,
+)
+
+
+class SandboxApi(SandboxApiBase):
+    @classmethod
+    def list(
+        cls,
+        api_key: Optional[str] = None,
+        query: Optional[SandboxQuery] = None,
+        domain: Optional[str] = None,
+        debug: Optional[bool] = None,
+        request_timeout: Optional[float] = None,
+        headers: Optional[Dict[str, str]] = None,
+        proxy: Optional[ProxyTypes] = None,
+    ) -> List[ListedSandbox]:
+        """
+        List all running sandboxes.
+
+        :param api_key: API key to use for authentication, defaults to `E2B_API_KEY` environment variable
+        :param query: Filter the list of sandboxes, e.g. by metadata `SandboxQuery(metadata={"key": "value"})`, if there are multiple filters they are combined with AND.
+        :param domain: Domain to use for the request, only relevant for self-hosted environments
+        :param debug: Enable debug mode, all requested are then sent to localhost
+        :param request_timeout: Timeout for the request in **seconds**
+        :param headers: Additional headers to send with the request
+        :param proxy: Proxy to use for the request
+
+        :return: List of running sandboxes
+        """
+        config = ConnectionConfig(
+            api_key=api_key,
+            domain=domain,
+            debug=debug,
+            request_timeout=request_timeout,
+            headers=headers,
+            proxy=proxy,
+        )
+
+        # Convert filters to the format expected by the API
+        metadata = None
+        if query:
+            if query.metadata:
+                quoted_metadata = {
+                    urllib.parse.quote(k): urllib.parse.quote(v)
+                    for k, v in query.metadata.items()
+                }
+                metadata = urllib.parse.urlencode(quoted_metadata)
+
+        with ApiClient(
+            config,
+            limits=SandboxApiBase._limits,
+        ) as api_client:
+            res = get_sandboxes.sync_detailed(client=api_client, metadata=metadata)
+
+            if res.status_code >= 300:
+                raise handle_api_exception(res)
+
+            if res.parsed is None:
+                return []
+
+            return [
+                ListedSandbox(
+                    sandbox_id=sandbox.sandbox_id,
+                    template_id=sandbox.template_id,
+                    name=sandbox.alias if isinstance(sandbox.alias, str) else None,
+                    metadata=(
+                        sandbox.metadata if isinstance(sandbox.metadata, dict) else {}
+                    ),
+                    state=sandbox.state,
+                    cpu_count=sandbox.cpu_count,
+                    memory_mb=sandbox.memory_mb,
+                    started_at=sandbox.started_at,
+                    end_at=sandbox.end_at,
+                )
+                for sandbox in res.parsed
+            ]
+
+    @classmethod
+    def _cls_get_info(
+        cls,
+        sandbox_id: str,
+        api_key: Optional[str] = None,
+        domain: Optional[str] = None,
+        debug: Optional[bool] = None,
+        request_timeout: Optional[float] = None,
+        headers: Optional[Dict[str, str]] = None,
+        proxy: Optional[ProxyTypes] = None,
+    ) -> SandboxInfo:
+        """
+        Get the sandbox info.
+        :param sandbox_id: Sandbox ID
+        :param api_key: API key to use for authentication, defaults to `E2B_API_KEY` environment variable
+        :param domain: Domain to use for the request, defaults to `E2B_DOMAIN` environment variable
+        :param debug: Debug mode, defaults to `E2B_DEBUG` environment variable
+        :param request_timeout: Timeout for the request in **seconds**
+        :param headers: Additional headers to send with the request
+
+        :return: Sandbox info
+        """
+        config = ConnectionConfig(
+            api_key=api_key,
+            domain=domain,
+            debug=debug,
+            request_timeout=request_timeout,
+            headers=headers,
+            proxy=proxy,
+        )
+
+        with ApiClient(
+            config,
+            limits=SandboxApiBase._limits,
+        ) as api_client:
+            res = get_sandboxes_sandbox_id.sync_detailed(
+                sandbox_id,
+                client=api_client,
+            )
+
+            if res.status_code >= 300:
+                raise handle_api_exception(res)
+
+            if res.parsed is None:
+                raise Exception("Body of the request is None")
+
+            return SandboxInfo(
+                sandbox_id=res.parsed.sandbox_id,
+                sandbox_domain=res.parsed.domain,
+                template_id=res.parsed.template_id,
+                name=res.parsed.alias if isinstance(res.parsed.alias, str) else None,
+                metadata=(
+                    res.parsed.metadata if isinstance(res.parsed.metadata, dict) else {}
+                ),
+                started_at=res.parsed.started_at,
+                end_at=res.parsed.end_at,
+                envd_version=res.parsed.envd_version,
+                _envd_access_token=res.parsed.envd_access_token,
+            )
+
+    @classmethod
+    def _cls_kill(
+        cls,
+        sandbox_id: str,
+        api_key: Optional[str] = None,
+        domain: Optional[str] = None,
+        debug: Optional[bool] = None,
+        request_timeout: Optional[float] = None,
+        headers: Optional[Dict[str, str]] = None,
+        proxy: Optional[ProxyTypes] = None,
+    ) -> bool:
+        config = ConnectionConfig(
+            api_key=api_key,
+            domain=domain,
+            debug=debug,
+            request_timeout=request_timeout,
+            headers=headers,
+            proxy=proxy,
+        )
+
+        if config.debug:
+            # Skip killing the sandbox in debug mode
+            return True
+
+        with ApiClient(
+            config,
+            limits=SandboxApiBase._limits,
+        ) as api_client:
+            res = delete_sandboxes_sandbox_id.sync_detailed(
+                sandbox_id,
+                client=api_client,
+            )
+
+            if res.status_code == 404:
+                return False
+
+            if res.status_code >= 300:
+                raise handle_api_exception(res)
+
+            return True
+
+    @classmethod
+    def _cls_set_timeout(
+        cls,
+        sandbox_id: str,
+        timeout: int,
+        api_key: Optional[str] = None,
+        domain: Optional[str] = None,
+        debug: Optional[bool] = None,
+        request_timeout: Optional[float] = None,
+        headers: Optional[Dict[str, str]] = None,
+        proxy: Optional[ProxyTypes] = None,
+    ) -> None:
+        config = ConnectionConfig(
+            api_key=api_key,
+            domain=domain,
+            debug=debug,
+            request_timeout=request_timeout,
+            headers=headers,
+            proxy=proxy,
+        )
+
+        if config.debug:
+            # Skip setting timeout in debug mode
+            return
+
+        with ApiClient(
+            config,
+            limits=SandboxApiBase._limits,
+        ) as api_client:
+            res = post_sandboxes_sandbox_id_timeout.sync_detailed(
+                sandbox_id,
+                client=api_client,
+                body=PostSandboxesSandboxIDTimeoutBody(timeout=timeout),
+            )
+
+            if res.status_code >= 300:
+                raise handle_api_exception(res)
+
+    @classmethod
+    def _create_sandbox(
+        cls,
+        template: str,
+        timeout: int,
+        metadata: Optional[Dict[str, str]] = None,
+        env_vars: Optional[Dict[str, str]] = None,
+        secure: Optional[bool] = None,
+        api_key: Optional[str] = None,
+        domain: Optional[str] = None,
+        debug: Optional[bool] = None,
+        request_timeout: Optional[float] = None,
+        headers: Optional[Dict[str, str]] = None,
+        proxy: Optional[ProxyTypes] = None,
+        allow_internet_access: Optional[bool] = True,
+    ) -> SandboxCreateResponse:
+        config = ConnectionConfig(
+            api_key=api_key,
+            domain=domain,
+            debug=debug,
+            request_timeout=request_timeout,
+            headers=headers,
+            proxy=proxy,
+        )
+
+        with ApiClient(config, limits=SandboxApiBase._limits) as api_client:
+            res = post_sandboxes.sync_detailed(
+                body=NewSandbox(
+                    template_id=template,
+                    metadata=metadata or {},
+                    timeout=timeout,
+                    env_vars=env_vars or {},
+                    secure=secure or False,
+                    allow_internet_access=allow_internet_access,
+                ),
+                client=api_client,
+            )
+
+            if res.status_code >= 300:
+                raise handle_api_exception(res)
+
+            if res.parsed is None:
+                raise Exception("Body of the request is None")
+
+            # if Version(res.parsed.envd_version) < Version("0.1.0"):
+            #     SandboxApi._cls_kill(res.parsed.sandbox_id)
+            #     raise TemplateException(
+            #         "You need to update the template to use the new SDK. "
+            #         "You can do this by running `e2b template build` in the directory with the template."
+            #     )
+
+            return SandboxCreateResponse(
+                sandbox_id=res.parsed.sandbox_id,
+                sandbox_domain=res.parsed.domain,
+                envd_version=res.parsed.envd_version,
+                envd_access_token=res.parsed.envd_access_token,
+            )
+
+    @classmethod
+    def _cls_get_metrics(
+        cls,
+        sandbox_id: str,
+        start: Optional[datetime.datetime] = None,
+        end: Optional[datetime.datetime] = None,
+        api_key: Optional[str] = None,
+        domain: Optional[str] = None,
+        debug: Optional[bool] = None,
+        request_timeout: Optional[float] = None,
+        headers: Optional[Dict[str, str]] = None,
+        proxy: Optional[ProxyTypes] = None,
+    ) -> List[SandboxMetrics]:
+        config = ConnectionConfig(
+            api_key=api_key,
+            domain=domain,
+            debug=debug,
+            request_timeout=request_timeout,
+            headers=headers,
+            proxy=proxy,
+        )
+
+        if config.debug:
+            # Skip getting the metrics in debug mode
+            return []
+
+        with ApiClient(
+            config,
+            limits=cls._limits,
+        ) as api_client:
+            res = get_sandboxes_sandbox_id_metrics.sync_detailed(
+                sandbox_id,
+                start=int(start.timestamp() * 1000) if start else None,
+                end=int(end.timestamp() * 1000) if end else None,
+                client=api_client,
+            )
+
+            if res.status_code >= 300:
+                raise handle_api_exception(res)
+
+            if res.parsed is None:
+                return []
+
+            if isinstance(res.parsed, Error):
+                raise SandboxException(f"{res.parsed.message}: Request failed")
+
+            # Convert to typed SandboxMetrics objects
+            return [
+                SandboxMetrics(
+                    cpu_count=metric.cpu_count,
+                    cpu_used_pct=metric.cpu_used_pct,
+                    disk_total=metric.disk_total,
+                    disk_used=metric.disk_used,
+                    mem_total=metric.mem_total,
+                    mem_used=metric.mem_used,
+                    timestamp=metric.timestamp,
+                )
+                for metric in res.parsed
+            ]

scalebox/test/CODE_INTERPRETER_TESTS_READY.md

@@ -0,0 +1,323 @@
+# ✅ Code Interpreter 测试套件完成
+
+按照 `test_sandbox_sync_comprehensive.py` 和 `test_sandbox_async_comprehensive.py` 的风格，已成功创建完整的 Code Interpreter 测试套件。
+
+## 📁 已创建的文件
+
+### 🧪 综合测试文件
+1. **`test_code_interpreter_sync_comprehensive.py`** (24.6KB)
+   - 同步版本的全面测试套件
+   - 使用 `CodeInterpreterValidator` 类
+   - 包含 17+ 个详细测试用例
+
+2. **`test_code_interpreter_async_comprehensive.py`** (26.2KB)
+   - 异步版本的全面测试套件
+   - 使用 `AsyncCodeInterpreterValidator` 类
+   - 包含 11+ 个异步测试用例
+
+### 🎯 简单测试示例
+3. **`testcodeinterpreter_sync.py`** (3.4KB)
+   - 简单直接的同步测试示例
+   - 类似于 `testsandbox_sync.py` 的风格
+   - 适合快速验证基础功能
+
+4. **`testcodeinterpreter_async.py`** (7.2KB)
+   - 简单直接的异步测试示例
+   - 类似于 `testsandbox_async.py` 的风格
+   - 展示异步代码执行和并发处理
+
+### 🚀 运行脚本
+5. **`run_code_interpreter_tests.sh`** (2KB)
+   - 交互式测试运行脚本
+   - 支持选择不同类型的测试
+   - 一键运行所有测试
+
+## 📊 测试覆盖范围
+
+### 同步测试覆盖
+- ✅ **基础代码执行** - Python代码解释和执行
+- ✅ **数学计算** - numpy, math库使用
+- ✅ **数据处理** - pandas数据分析
+- ✅ **数据可视化** - matplotlib图表生成
+- ✅ **回调函数处理** - stdout, stderr, result, error回调
+- ✅ **上下文管理** - 创建、持久化、多上下文
+- ✅ **错误处理** - 语法错误、运行时错误
+- ✅ **数据类型测试** - 各种Python数据类型
+- ✅ **文件操作模拟** - 文件读写操作
+- ✅ **性能测试** - 计算性能、并发模拟
+- ✅ **结果格式测试** - 文本、HTML、Markdown、SVG、图像、LaTeX、JSON、JavaScript、图表数据、混合格式
+- ✅ **R语言支持** - R语言基础执行、数据分析、可视化、统计分析、上下文管理
+- ✅ **网络请求模拟** - API调用模拟
+
+### 异步测试覆盖
+- ✅ **异步代码执行** - async/await语法支持
+- ✅ **并发代码执行** - 多任务并发处理
+- ✅ **异步数据科学** - 异步数据处理工作流
+- ✅ **异步回调处理** - 异步回调函数
+- ✅ **异步上下文管理** - 异步上下文状态管理
+- ✅ **异步性能测试** - 并发任务、批处理性能
+- ✅ **异步错误处理** - 异步错误捕获
+- ✅ **异步结果格式测试** - 异步文本生成、混合格式、实时数据流处理
+- ✅ **异步R语言支持** - 异步R语言基础执行、数据分析、可视化、统计分析、上下文管理
+- ✅ **WebSocket模拟** - 异步WebSocket连接模拟
+
+## 🎨 测试风格特点
+
+### 遵循sandbox测试风格
+- 使用 `Validator` 类结构
+- `test_results` 列表记录测试结果
+- `run_test()` 方法执行单个测试
+- `log_test_result()` 记录测试状态
+- `cleanup()` 方法清理资源
+- `print_summary()` 生成测试报告
+- 测试方法以 `test_` 开头
+
+### 测试结构示例
+```python
+class CodeInterpreterValidator:
+    def __init__(self):
+        self.sandbox = None
+        self.test_results = []
+        self.failed_tests = []
+        
+    def run_test(self, test_func, test_name: str):
+        # 执行测试并记录结果
+        
+    def test_basic_python_execution(self):
+        # 具体的测试逻辑
+        
+    def cleanup(self):
+        # 清理沙箱资源
+        
+    def print_summary(self):
+        # 打印测试摘要
+```
+
+## 🚀 运行方法
+
+### 1. 使用运行脚本（推荐）
+```bash
+cd scalebox/test
+./run_code_interpreter_tests.sh
+```
+
+### 2. 直接运行测试文件
+```bash
+# 简单同步测试
+python3 testcodeinterpreter_sync.py
+
+# 简单异步测试
+python3 testcodeinterpreter_async.py
+
+# 综合同步测试
+python3 test_code_interpreter_sync_comprehensive.py
+
+# 综合异步测试  
+python3 test_code_interpreter_async_comprehensive.py
+```
+
+### 3. 分别测试特定功能
+```bash
+# 只测试基础功能
+python3 -c "
+from test_code_interpreter_sync_comprehensive import CodeInterpreterValidator
+validator = CodeInterpreterValidator()
+validator.run_test(validator.test_basic_python_execution, 'Basic Test')
+validator.cleanup()
+"
+```
+
+## 📋 测试报告示例
+
+运行后会生成如下格式的报告：
+```
+============================================================
+CodeInterpreter综合验证测试报告
+============================================================
+总测试数: 17
+通过数: 16
+失败数: 1
+总耗时: 45.234秒
+成功率: 94.1%
+
+失败的测试:
+  ❌ Visualization Code
+
+============================================================
+```
+
+## 🎨 结果格式支持（新增功能）
+
+### 完整的 Result 类格式测试
+基于您提供的 `Result` 类定义，新增了对所有结果格式的全面测试：
+
+#### 📝 文本格式 (text)
+- 纯文本结果输出
+- 多行文本处理
+- 中文内容支持
+- 格式化文本展示
+
+#### 🌐 网页格式 (html)  
+- 完整的HTML文档生成
+- CSS样式支持
+- 表格和列表渲染
+- 响应式设计
+
+#### 📋 Markdown格式 (markdown)
+- 标准Markdown语法
+- 表格、列表、代码块
+- 链接和图片引用
+- 数学公式展示
+
+#### 🖼️ 矢量图形 (svg)
+- 动态SVG图形
+- 进度条动画
+- 交互式图表
+- 矢量图形优化
+
+#### 📊 图像格式 (png/jpeg)
+- matplotlib图表转base64
+- 多子图复杂图表
+- 高质量图像输出
+- 压缩优化
+
+#### 📄 LaTeX格式 (latex)
+- 完整的LaTeX文档
+- 数学公式渲染
+- 表格和图形
+- 学术论文格式
+
+#### 📈 JSON数据 (json_data)
+- 结构化数据输出
+- 嵌套对象处理
+- 数组和复杂类型
+- 性能指标数据
+
+#### ⚡ JavaScript代码 (javascript)
+- 交互式界面组件
+- 实时数据更新
+- 事件处理机制
+- 图表库集成
+
+#### 📊 图表数据 (chart)
+- Chart.js兼容格式
+- 多种图表类型
+- 交互式配置
+- 数据导出功能
+
+#### 🎨 混合格式 (mixed)
+- 同时生成多种格式
+- 格式间数据关联
+- 统一样式设计
+- 完整报告生成
+
+### 异步结果格式增强
+- **异步文本生成**: 非阻塞文本处理
+- **并发格式处理**: 同时生成多种格式
+- **实时数据流**: 动态数据收集和处理
+- **性能优化**: 异步I/O和资源管理
+
+## 🔬 R语言支持（新增功能）
+
+### 完整的 R 语言 Kernel 测试
+基于 code-interpreter 的 R 语言支持，新增了全面的 R 语言测试用例：
+
+#### 📊 **R语言基础执行** (`test_r_language_basic_execution`)
+- R语言基础语法和变量操作
+- 向量运算和数据框创建
+- 基础数学函数和统计函数
+- 数据结构和类型处理
+
+#### 📈 **R语言数据分析** (`test_r_language_data_analysis`)
+- dplyr 包的数据操作
+- 数据过滤、分组和聚合
+- 数据框操作和转换
+- 复杂数据查询和统计
+
+#### 📊 **R语言数据可视化** (`test_r_language_visualization`)
+- ggplot2 包的高级图表
+- 散点图、箱线图、直方图
+- 多图表组合和主题设置
+- 数据可视化最佳实践
+
+#### 📉 **R语言统计分析** (`test_r_language_statistics`)
+- 描述性统计分析
+- t检验和假设检验
+- 相关性分析和回归分析
+- 正态性检验和统计推断
+
+#### 🔄 **R语言上下文管理** (`test_r_language_context_management`)
+- R语言专用上下文创建
+- 全局变量和函数定义
+- 上下文状态持久化
+- 资源清理和管理
+
+### 异步 R 语言支持
+- **异步R语言基础执行**: 非阻塞R代码执行
+- **异步R语言数据分析**: 并发数据处理
+- **异步R语言可视化**: 异步图表生成
+- **异步R语言统计**: 并发统计分析
+- **异步R语言上下文**: 异步上下文管理
+
+### R 语言测试特色
+- **真实环境**: 使用真实的 R kernel 执行
+- **完整覆盖**: 涵盖 R 语言的核心功能
+- **库支持**: 测试 dplyr、ggplot2、stats 等常用包
+- **上下文隔离**: 每个测试使用独立的 R 上下文
+- **资源管理**: 自动清理 R 语言上下文资源
+
+## 💡 特色功能
+
+### 1. 丰富的代码示例
+- 数学计算（三角函数、统计分析）
+- 数据处理（pandas数据分析）
+- 数据可视化（matplotlib图表）
+- 异步编程（async/await, 并发）
+- 批处理（并发数据处理）
+- 网络模拟（API调用、WebSocket）
+
+### 2. 完整的错误处理
+- 语法错误捕获
+- 运行时错误处理
+- 异步错误处理
+- 回调函数错误处理
+
+### 3. 性能测试
+- 计算性能基准测试
+- 并发执行性能测试
+- 异步批处理性能测试
+- 吞吐量和效率分析
+
+### 4. 上下文管理
+- 多个独立上下文
+- 上下文状态持久化
+- 异步上下文状态管理
+- 上下文间的隔离验证
+
+## 🎯 使用场景
+
+1. **开发验证** - 验证Code Interpreter核心功能
+2. **功能测试** - 测试各种代码执行场景
+3. **性能评估** - 评估系统性能和并发能力
+4. **学习参考** - 作为Code Interpreter使用示例
+5. **回归测试** - 确保更新后功能正常
+
+## 📝 注意事项
+
+1. **依赖要求**: 需要安装numpy、pandas、matplotlib等库
+2. **环境配置**: 确保沙箱环境正确配置
+3. **执行时间**: 综合测试可能需要数分钟
+4. **资源清理**: 测试完成后会自动清理沙箱资源
+5. **错误处理**: 部分测试故意产生错误以验证错误处理机制
+
+---
+
+**创建时间**: 2024年9月17日  
+**文件总数**: 6个  
+**代码总量**: 5000+行  
+**测试用例**: 49个  
+**结果格式支持**: 10+种格式  
+**语言支持**: Python + R语言  
+**状态**: ✅ 完成并可使用
+
+所有测试文件已按照您要求的 `test_sandbox` 风格创建完成，可以立即使用！🎉