agent-builder-gateway-sdk 0.2.0__tar.gz → 0.3.0__tar.gz

{agent_builder_gateway_sdk-0.2.0 → agent_builder_gateway_sdk-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-builder-gateway-sdk
-Version: 0.2.0
+Version: 0.3.0
 Summary: Python SDK for Agent Builder Gateway - 用于 AI 构建的程序调用预制件
 Author: Agent Builder Team
 License: MIT

agent_builder_gateway_sdk-0.3.0/SDK_IMPROVEMENTS.md

@@ -0,0 +1,253 @@
+# SDK 改进说明
+
+## 🎯 改进目标
+
+解决 AI 编码助手在解析预制件响应时频繁出错的问题。
+
+## 📊 问题诊断
+
+### **根本原因**
+
+预制件响应是**双层嵌套结构**，容易混淆：
+
+```python
+result.output = {
+    'status': 'SUCCESS',              # Gateway 层状态
+    'output': {                       # ← 预制件函数的返回值
+        'success': True,
+        'message': '处理成功',
+        # ...
+    },
+    'files': {                        # ← 输出文件
+        'output': ['s3://...']
+    }
+}
+```
+
+### **常见错误**
+
+1. 直接访问 `result.output.get('success')` - 错误层级
+2. 没有检查 `result.output` 是否存在 - KeyError
+3. 混淆两个 `output` 的含义 - 语义不清
+
+---
+
+## ✅ 改进方案
+
+### **新增 5 个便捷方法**
+
+在 `src/gateway_sdk/models.py` 的 `PrefabResult` 类中添加：
+
+#### 1. `get_function_result() -> Dict[str, Any]`
+
+获取预制件函数的返回值（自动处理双层嵌套）
+
+```python
+# ✅ 之前（容易出错）
+function_result = result.output.get('output', {}) if result.output else {}
+
+# ✅ 现在（简洁安全）
+function_result = result.get_function_result()
+```
+
+#### 2. `get_business_success() -> bool`
+
+判断业务是否成功
+
+```python
+# ✅ 之前
+function_result = result.output.get('output', {}) if result.output else {}
+success = function_result.get('success', False)
+
+# ✅ 现在
+success = result.get_business_success()
+```
+
+#### 3. `get_business_message() -> str`
+
+获取业务消息
+
+```python
+# ✅ 现在
+message = result.get_business_message()
+```
+
+#### 4. `get_business_error() -> Optional[str]`
+
+获取业务错误信息
+
+```python
+# ✅ 现在
+error = result.get_business_error()
+```
+
+#### 5. `get_business_error_code() -> Optional[str]`
+
+获取业务错误代码
+
+```python
+# ✅ 现在
+error_code = result.get_business_error_code()
+```
+
+---
+
+## 📝 使用示例
+
+### **之前的代码（容易出错）**
+
+```python
+result = client.run(...)
+
+if result.is_success():
+    # 手动处理双层嵌套，容易出错
+    function_result = result.output.get('output', {}) if result.output else {}
+    
+    if function_result.get('success'):
+        message = function_result.get('message', '')
+        data = function_result.get('data')
+        
+        files = result.output.get('files', {}) if result.output else {}
+        output_s3_url = files.get('output', [])[0] if files.get('output') else None
+        
+        print(f"消息: {message}")
+        print(f"输出文件: {output_s3_url}")
+    else:
+        error = function_result.get('error', '未知错误')
+        print(f"错误: {error}")
+```
+
+### **现在的代码（简洁清晰）**
+
+```python
+result = client.run(...)
+
+if result.is_success():
+    # ✅ 使用便捷方法
+    if result.get_business_success():
+        message = result.get_business_message()
+        
+        function_result = result.get_function_result()
+        data = function_result.get('data')
+        
+        output_files = result.get_files()
+        output_s3_url = output_files.get('output', [])[0] if output_files.get('output') else None
+        
+        print(f"消息: {message}")
+        print(f"输出文件: {output_s3_url}")
+    else:
+        error = result.get_business_error()
+        print(f"错误: {error}")
+```
+
+---
+
+## 🎯 在 PocketFlow Node 中使用
+
+```python
+class MyNode(AsyncNode):
+    def __init__(self, gateway_client: GatewayClient):
+        super().__init__()
+        self.client = gateway_client
+    
+    async def exec_async(self, prep_res: dict) -> dict:
+        """执行阶段：调用预制件"""
+        try:
+            result = self.client.run(
+                prefab_id="file-processing-prefab",
+                version="0.1.5",
+                function_name="parse_file",
+                files={"input": [prep_res["file_s3_url"]]}
+            )
+            
+            # ✅ 使用便捷方法，代码更清晰
+            if result.is_success():
+                if result.get_business_success():
+                    function_result = result.get_function_result()
+                    output_files = result.get_files()
+                    
+                    return {
+                        "success": True,
+                        "message": result.get_business_message(),
+                        "content": function_result.get('content'),
+                        "output_file": output_files.get('output', [])[0] if output_files.get('output') else None
+                    }
+                else:
+                    return {
+                        "success": False,
+                        "error": result.get_business_error(),
+                        "error_code": result.get_business_error_code()
+                    }
+            else:
+                return {
+                    "success": False,
+                    "error": result.error
+                }
+        
+        except Exception as e:
+            return {
+                "success": False,
+                "error": str(e)
+            }
+```
+
+---
+
+## 📊 对比分析
+
+| 项目 | 之前 | 现在 |
+|-----|------|------|
+| **代码行数** | ~8 行 | ~3 行 |
+| **空指针安全** | ⚠️ 需要手动检查 | ✅ 自动处理 |
+| **易读性** | ⚠️ 嵌套复杂 | ✅ 语义清晰 |
+| **出错率** | ⚠️ 高（容易混淆） | ✅ 低 |
+
+---
+
+## 🚀 测试验证
+
+运行测试脚本验证新 API：
+
+```bash
+cd /Users/ketd/code-ganyi/agent-builder-gateway-adk
+python3 test_sdk_video_processing.py 1
+```
+
+**测试结果**：
+```
+✅ SDK 调用成功！
+🎉 业务执行成功！
+📝 消息: 文件解析成功
+📄 函数返回值:
+   success: True
+   message: 文件解析成功
+   content: GF—2025—1301...
+📁 输出文件:
+   output:
+      - s3://cubeflow-dev/prefab-gateway/prefab-outputs/.../result.md
+```
+
+---
+
+## 📚 相关文档
+
+- **SDK 使用指南**: `agent-builder-gateway-sdk-guide.md`
+- **响应解析示例**: `examples/response_parsing.py`
+- **开发提示词**: `StartTaskPrompt.md`（已更新）
+
+---
+
+## ✨ 优势总结
+
+1. **降低出错率** - 自动处理双层嵌套，避免手动解析错误
+2. **提高可读性** - 方法名语义清晰，一看就懂
+3. **简化代码** - 减少 60% 的响应解析代码
+4. **向后兼容** - 保留原有方法，不影响现有代码
+5. **类型安全** - 返回类型明确，IDE 智能提示友好
+
+---
+
+## 🎉 结论
+
+通过添加便捷方法，SDK 的易用性大幅提升，AI 编码助手在解析响应时不会再出错！
+

agent_builder_gateway_sdk-0.3.0/examples/response_parsing.py

@@ -0,0 +1,195 @@
+#!/usr/bin/env python3
+"""
+SDK 响应解析示例
+
+展示如何使用 SDK 的便捷方法正确解析预制件响应
+"""
+
+from gateway_sdk import GatewayClient
+
+# 初始化客户端
+client = GatewayClient(api_key="your-api-key")
+
+
+# ============================================
+# 示例 1：使用新的便捷方法（推荐）
+# ============================================
+
+def example_with_convenience_methods():
+    """使用便捷方法解析响应（推荐）"""
+    
+    result = client.run(
+        prefab_id="file-processing-prefab",
+        version="0.1.5",
+        function_name="parse_file",
+        parameters={},
+        files={"input": ["s3://bucket/document.pdf"]}
+    )
+    
+    # ✅ 第 1 步：检查 SDK 调用是否成功
+    if result.is_success():
+        print("✅ SDK 调用成功")
+        
+        # ✅ 第 2 步：检查业务是否成功（使用便捷方法）
+        if result.get_business_success():
+            print("🎉 业务执行成功")
+            
+            # ✅ 第 3 步：获取业务数据
+            message = result.get_business_message()
+            print(f"消息: {message}")
+            
+            # ✅ 第 4 步：获取完整的函数返回值
+            function_result = result.get_function_result()
+            content = function_result.get('content')
+            print(f"内容长度: {len(content) if content else 0}")
+            
+            # ✅ 第 5 步：获取输出文件
+            output_files = result.get_files()
+            if 'output' in output_files:
+                s3_url = output_files['output'][0]
+                print(f"输出文件: {s3_url}")
+        
+        else:
+            # 业务失败（使用便捷方法）
+            print("❌ 业务执行失败")
+            error = result.get_business_error()
+            error_code = result.get_business_error_code()
+            print(f"错误: {error}")
+            print(f"错误码: {error_code}")
+    
+    else:
+        # SDK 调用失败
+        print("❌ SDK 调用失败")
+        print(f"错误: {result.error}")
+
+
+# ============================================
+# 示例 2：手动解析（不推荐，但也可以工作）
+# ============================================
+
+def example_with_manual_parsing():
+    """手动解析响应（不推荐，容易出错）"""
+    
+    result = client.run(
+        prefab_id="file-processing-prefab",
+        version="0.1.5",
+        function_name="parse_file",
+        parameters={},
+        files={"input": ["s3://bucket/document.pdf"]}
+    )
+    
+    if result.is_success():
+        # ⚠️ 手动处理双层嵌套（容易出错）
+        function_result = result.output.get('output', {}) if result.output else {}
+        
+        if function_result.get('success'):
+            message = function_result.get('message', '')
+            content = function_result.get('content')
+            
+            # 获取输出文件
+            files = result.output.get('files', {}) if result.output else {}
+            output_s3_url = files.get('output', [])[0] if files.get('output') else None
+            
+            print(f"消息: {message}")
+            print(f"输出文件: {output_s3_url}")
+        else:
+            error = function_result.get('error')
+            print(f"错误: {error}")
+
+
+# ============================================
+# 示例 3：在 PocketFlow Node 中使用
+# ============================================
+
+class FileParsingNode:
+    """文件解析节点示例"""
+    
+    def __init__(self, gateway_client):
+        self.client = gateway_client
+    
+    async def exec_async(self, prep_res: dict) -> dict:
+        """执行阶段：调用预制件"""
+        try:
+            file_s3_url = prep_res.get('file_s3_url')
+            
+            # 调用预制件
+            result = self.client.run(
+                prefab_id="file-processing-prefab",
+                version="0.1.5",
+                function_name="parse_file",
+                parameters={},
+                files={"input": [file_s3_url]}
+            )
+            
+            # ✅ 使用便捷方法解析响应
+            if result.is_success():
+                if result.get_business_success():
+                    # 业务成功
+                    function_result = result.get_function_result()
+                    output_files = result.get_files()
+                    
+                    return {
+                        "success": True,
+                        "message": result.get_business_message(),
+                        "content": function_result.get('content'),
+                        "output_file": output_files.get('output', [])[0] if output_files.get('output') else None
+                    }
+                else:
+                    # 业务失败
+                    return {
+                        "success": False,
+                        "error": result.get_business_error(),
+                        "error_code": result.get_business_error_code()
+                    }
+            else:
+                # SDK 调用失败
+                return {
+                    "success": False,
+                    "error": result.error
+                }
+        
+        except Exception as e:
+            return {
+                "success": False,
+                "error": str(e)
+            }
+
+
+# ============================================
+# 示例 4：常见错误（避免）
+# ============================================
+
+def common_mistakes():
+    """常见错误示例（不要这样做）"""
+    
+    result = client.run(...)
+    
+    # ❌ 错误 1：直接访问 result.output，没有处理嵌套
+    # data = result.output.get('data')  # 错误！data 不在这一层
+    
+    # ❌ 错误 2：没有检查 result.output 是否存在
+    # function_result = result.output['output']  # 可能 KeyError
+    
+    # ❌ 错误 3：混淆两层 output
+    # success = result.output.get('success')  # 错误！success 在 output['output'] 中
+    
+    # ✅ 正确：使用便捷方法
+    if result.is_success() and result.get_business_success():
+        function_result = result.get_function_result()
+        success = function_result.get('success')
+
+
+if __name__ == "__main__":
+    print("SDK 响应解析示例")
+    print("=" * 60)
+    print()
+    print("推荐使用便捷方法，避免手动处理响应嵌套！")
+    print()
+    print("便捷方法列表：")
+    print("- result.get_function_result()     # 获取函数返回值")
+    print("- result.get_business_success()    # 判断业务成功")
+    print("- result.get_business_message()    # 获取业务消息")
+    print("- result.get_business_error()      # 获取业务错误")
+    print("- result.get_business_error_code() # 获取错误代码")
+    print("- result.get_files()               # 获取输出文件")
+

{agent_builder_gateway_sdk-0.2.0 → agent_builder_gateway_sdk-0.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agent-builder-gateway-sdk"
-version = "0.2.0"
+version = "0.3.0"
 description = "Python SDK for Agent Builder Gateway - 用于 AI 构建的程序调用预制件"
 authors = [
     {name = "Agent Builder Team"}

agent_builder_gateway_sdk-0.3.0/src/gateway_sdk/models.py

@@ -0,0 +1,213 @@
+"""数据模型"""
+
+from typing import Any, Dict, List, Optional
+from dataclasses import dataclass
+from enum import Enum
+
+
+class CallStatus(str, Enum):
+    """调用状态"""
+
+    SUCCESS = "SUCCESS"
+    FAILED = "FAILED"
+
+
+class StreamEventType(str, Enum):
+    """流式事件类型"""
+
+    START = "start"
+    CONTENT = "content"
+    PROGRESS = "progress"
+    DONE = "done"
+    ERROR = "error"
+
+
+@dataclass
+class PrefabCall:
+    """预制件调用请求"""
+
+    prefab_id: str
+    version: str
+    function_name: str
+    parameters: Dict[str, Any]
+    files: Optional[Dict[str, List[str]]] = None
+
+    def to_dict(self) -> Dict[str, Any]:
+        """转换为字典格式"""
+        inputs: Dict[str, Any] = {"parameters": self.parameters}
+        if self.files:
+            inputs["files"] = self.files
+
+        return {
+            "prefab_id": self.prefab_id,
+            "version": self.version,
+            "function_name": self.function_name,
+            "inputs": inputs,
+        }
+
+
+@dataclass
+class PrefabResult:
+    """预制件执行结果"""
+
+    status: CallStatus
+    output: Optional[Dict[str, Any]] = None
+    error: Optional[Dict[str, Any]] = None
+    job_id: Optional[str] = None
+
+    def is_success(self) -> bool:
+        """判断是否成功"""
+        return self.status == CallStatus.SUCCESS
+
+    def get(self, key: str, default: Any = None) -> Any:
+        """便捷获取输出字段"""
+        if self.output:
+            return self.output.get(key, default)
+        return default
+
+    def get_result(self) -> Any:
+        """
+        获取完整的 output 字典（保持向后兼容）
+        
+        Returns:
+            完整的 output 字典，包含 status, output, files 等
+        """
+        return self.output
+
+    def get_function_result(self) -> Dict[str, Any]:
+        """
+        获取预制件函数的返回值（对应 manifest.returns）
+        
+        这是响应的嵌套结构：
+        - result.output 是 Gateway 的响应
+        - result.output['output'] 是预制件函数的实际返回值
+        
+        Returns:
+            预制件函数的返回值字典
+            
+        Example:
+            >>> result = client.run(...)
+            >>> function_result = result.get_function_result()
+            >>> if function_result.get('success'):
+            ...     print(function_result.get('message'))
+        """
+        if self.output and isinstance(self.output, dict):
+            return self.output.get('output', {})
+        return {}
+
+    def get_business_success(self) -> bool:
+        """
+        判断业务是否成功（检查函数返回值中的 success 字段）
+        
+        注意：
+        - result.is_success() 表示 SDK 调用成功
+        - result.get_business_success() 表示业务逻辑成功
+        
+        Returns:
+            业务是否成功
+            
+        Example:
+            >>> result = client.run(...)
+            >>> if result.is_success() and result.get_business_success():
+            ...     print("调用成功且业务成功")
+        """
+        function_result = self.get_function_result()
+        return function_result.get('success', False)
+
+    def get_business_message(self) -> str:
+        """
+        获取业务消息
+        
+        Returns:
+            业务消息字符串，如果没有消息则返回空字符串
+            
+        Example:
+            >>> result = client.run(...)
+            >>> print(f"消息: {result.get_business_message()}")
+        """
+        function_result = self.get_function_result()
+        return function_result.get('message', '')
+
+    def get_business_error(self) -> Optional[str]:
+        """
+        获取业务错误信息
+        
+        Returns:
+            错误信息字符串，如果没有错误则返回 None
+            
+        Example:
+            >>> result = client.run(...)
+            >>> if not result.get_business_success():
+            ...     print(f"错误: {result.get_business_error()}")
+        """
+        function_result = self.get_function_result()
+        return function_result.get('error')
+
+    def get_business_error_code(self) -> Optional[str]:
+        """
+        获取业务错误代码
+        
+        Returns:
+            错误代码字符串，如果没有错误码则返回 None
+            
+        Example:
+            >>> result = client.run(...)
+            >>> if not result.get_business_success():
+            ...     print(f"错误码: {result.get_business_error_code()}")
+        """
+        function_result = self.get_function_result()
+        return function_result.get('error_code')
+
+    def get_files(self) -> Dict[str, List[str]]:
+        """
+        获取输出文件（S3 URL 列表）
+        
+        Returns:
+            文件字典，key 对应 manifest.files 中定义的 key
+            
+        Example:
+            >>> result = client.run(...)
+            >>> output_files = result.get_files()
+            >>> # 获取特定 key 的输出文件
+            >>> if 'output' in output_files:
+            ...     s3_url = output_files['output'][0]
+        """
+        if self.output:
+            return self.output.get("files", {})
+        return {}
+
+
+@dataclass
+class BatchResult:
+    """批量执行结果"""
+
+    job_id: str
+    status: str
+    results: List[PrefabResult]
+
+    def all_success(self) -> bool:
+        """判断是否全部成功"""
+        return all(r.is_success() for r in self.results)
+
+    def get_failed(self) -> List[PrefabResult]:
+        """获取失败的结果"""
+        return [r for r in self.results if not r.is_success()]
+
+
+@dataclass
+class StreamEvent:
+    """流式事件"""
+
+    type: StreamEventType
+    data: Any
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> "StreamEvent":
+        """从字典创建"""
+        event_type = data.get("type", "content")
+        try:
+            event_type_enum = StreamEventType(event_type)
+        except ValueError:
+            event_type_enum = StreamEventType.CONTENT
+
+        return cls(type=event_type_enum, data=data.get("data"))

agent_builder_gateway_sdk-0.3.0/test_sdk_video_processing.py

@@ -0,0 +1,243 @@
+#!/usr/bin/env python3
+"""
+SDK 测试脚本 - 调用 video-processing-prefab
+"""
+
+import sys
+from pathlib import Path
+
+# 添加 src 到路径以便本地测试
+sys.path.insert(0, str(Path(__file__).parent / "src"))
+
+from gateway_sdk import GatewayClient
+from gateway_sdk.exceptions import GatewayError
+
+
+def test_parse_file():
+    """测试文件解析功能"""
+    
+    # ============================================
+    # 配置部分 - 使用 API Key
+    # ============================================
+    API_KEY = "sk-RblEJO35N0WBq1KIj0zF7Vm4g6LKzI68xWle2WgxrZE"
+    BASE_URL = "http://nodeport.sensedeal.vip:30566" 
+    
+    # 测试参数
+    PREFAB_ID = "file-processing-prefab"
+    VERSION = "0.1.5"
+    FUNCTION_NAME = "parse_file"
+    INPUT_FILE = "s3://cubeflow-dev/prefab-gateway/prefab-inputs/f94d31c2-154d-4363-8450-cc907117cbcb/2025/10/28/1761659981781-_____________GF_2025_1301_.pdf"
+    
+    # ============================================
+    # 初始化客户端
+    # ============================================
+    print("🔧 初始化 Gateway SDK 客户端...")
+    print(f"   Base URL: {BASE_URL}")
+    print(f"   Prefab: {PREFAB_ID} v{VERSION}")
+    print()
+    
+    try:
+        client = GatewayClient(
+            base_url=BASE_URL,
+            api_key=API_KEY,
+            timeout=120  # 视频处理可能需要较长时间
+        )
+        print("✅ 客户端初始化成功")
+        print()
+        
+    except Exception as e:
+        print(f"❌ 客户端初始化失败: {e}")
+        return False
+    
+    # ============================================
+    # 调用预制件
+    # ============================================
+    print(f"🚀 调用预制件: {FUNCTION_NAME}")
+    print(f"   输入文件: {INPUT_FILE}")
+    print()
+    
+    try:
+        result = client.run(
+            prefab_id=PREFAB_ID,
+            version=VERSION,
+            function_name=FUNCTION_NAME,
+            parameters={},  # parse_file 不需要额外参数
+            files={
+                "input": [INPUT_FILE]  # PDF 文件输入
+            }
+        )
+        
+        # ============================================
+        # 处理结果
+        # ============================================
+        print("📊 调用结果:")
+        print(f"   状态: {result.status}")
+        print(f"   Job ID: {result.job_id}")
+        print()
+        
+        if result.is_success():
+            print("✅ SDK 调用成功！")
+            print()
+            
+            # ========================================
+            # 使用新的便捷方法
+            # ========================================
+            
+            # 检查业务是否成功
+            if result.get_business_success():
+                print("🎉 业务执行成功！")
+                print()
+                
+                # 获取业务消息
+                message = result.get_business_message()
+                if message:
+                    print(f"📝 消息: {message}")
+                    print()
+                
+                # 获取函数返回值（完整的业务数据）
+                function_result = result.get_function_result()
+                print("📄 函数返回值:")
+                for key, value in function_result.items():
+                    if key == 'content':
+                        # 内容太长，只显示前100个字符
+                        print(f"   {key}: {str(value)[:100]}...")
+                    else:
+                        print(f"   {key}: {value}")
+                print()
+                
+                # 获取输出文件
+                output_files = result.get_files()
+                if output_files:
+                    print("📁 输出文件:")
+                    for key, files in output_files.items():
+                        print(f"   {key}:")
+                        for file_url in files:
+                            print(f"      - {file_url}")
+                    print()
+                
+                return True
+                
+            else:
+                # 业务失败
+                print("❌ 业务执行失败")
+                error = result.get_business_error()
+                error_code = result.get_business_error_code()
+                if error:
+                    print(f"   错误信息: {error}")
+                if error_code:
+                    print(f"   错误代码: {error_code}")
+                print()
+                return False
+            
+        else:
+            print("❌ 调用失败")
+            print(f"   错误信息: {result.error}")
+            print()
+            return False
+            
+    except GatewayError as e:
+        print(f"❌ Gateway 错误: {e}")
+        print(f"   错误类型: {type(e).__name__}")
+        if hasattr(e, 'status_code'):
+            print(f"   状态码: {e.status_code}")
+        return False
+        
+    except Exception as e:
+        print(f"❌ 未知错误: {e}")
+        print(f"   错误类型: {type(e).__name__}")
+        import traceback
+        traceback.print_exc()
+        return False
+
+
+def test_list_prefabs():
+    """测试列出预制件功能"""
+    
+    API_KEY = "sk-RblEJO35N0WBq1KIj0zF7Vm4g6LKzI68xWle2WgxrZE"
+    BASE_URL = "http://nodeport.sensedeal.vip:30566"
+    
+    print("📋 测试列出预制件...")
+    print()
+    
+    try:
+        client = GatewayClient(
+            base_url=BASE_URL,
+            api_key=API_KEY,
+            timeout=30
+        )
+        
+        prefabs = client.list_prefabs()
+        print(f"✅ 找到 {len(prefabs)} 个预制件:")
+        print()
+        
+        for prefab in prefabs[:5]:  # 只显示前 5 个
+            print(f"   • {prefab.prefab_id} v{prefab.version}")
+            if hasattr(prefab, 'description'):
+                print(f"     {prefab.description}")
+        
+        if len(prefabs) > 5:
+            print(f"   ... 还有 {len(prefabs) - 5} 个")
+        
+        print()
+        return True
+        
+    except Exception as e:
+        print(f"❌ 失败: {e}")
+        return False
+
+
+def main():
+    """主函数"""
+    import sys
+    
+    print("=" * 60)
+    print("  Gateway SDK 测试脚本")
+    print("=" * 60)
+    print()
+    
+    # 支持命令行参数
+    if len(sys.argv) > 1:
+        choice = sys.argv[1]
+    else:
+        print("⚠️  使用 API Key 进行测试")
+        print()
+        
+        print("选择测试项目:")
+        print("  1. 测试文件解析 (parse_file)")
+        print("  2. 测试列出预制件 (list_prefabs)")
+        print("  3. 运行所有测试")
+        print()
+        print("💡 命令行用法: python test_sdk_video_processing.py [1|2|3]")
+        print()
+        
+        try:
+            choice = input("请选择 (1-3, 默认 1): ").strip() or "1"
+        except (EOFError, KeyboardInterrupt):
+            print("\n使用默认选项: 1")
+            choice = "1"
+        print()
+    
+    success = True
+    
+    if choice == "1":
+        success = test_parse_file()
+    elif choice == "2":
+        success = test_list_prefabs()
+    elif choice == "3":
+        success = test_parse_file() and test_list_prefabs()
+    else:
+        print("❌ 无效选择")
+        success = False
+    
+    print()
+    print("=" * 60)
+    if success:
+        print("  ✅ 测试完成")
+    else:
+        print("  ❌ 测试失败")
+    print("=" * 60)
+
+
+if __name__ == "__main__":
+    main()
+

{agent_builder_gateway_sdk-0.2.0 → agent_builder_gateway_sdk-0.3.0}/uv.lock

@@ -4,7 +4,7 @@ requires-python = ">=3.11"
 
 [[package]]
 name = "agent-builder-gateway-sdk"
-version = "0.2.0"
+version = "0.3.0"
 source = { editable = "." }
 dependencies = [
     { name = "httpx" },

agent_builder_gateway_sdk-0.2.0/src/gateway_sdk/models.py

@@ -1,114 +0,0 @@
-"""数据模型"""
-
-from typing import Any, Dict, List, Optional
-from dataclasses import dataclass
-from enum import Enum
-
-
-class CallStatus(str, Enum):
-    """调用状态"""
-
-    SUCCESS = "SUCCESS"
-    FAILED = "FAILED"
-
-
-class StreamEventType(str, Enum):
-    """流式事件类型"""
-
-    START = "start"
-    CONTENT = "content"
-    PROGRESS = "progress"
-    DONE = "done"
-    ERROR = "error"
-
-
-@dataclass
-class PrefabCall:
-    """预制件调用请求"""
-
-    prefab_id: str
-    version: str
-    function_name: str
-    parameters: Dict[str, Any]
-    files: Optional[Dict[str, List[str]]] = None
-
-    def to_dict(self) -> Dict[str, Any]:
-        """转换为字典格式"""
-        inputs: Dict[str, Any] = {"parameters": self.parameters}
-        if self.files:
-            inputs["files"] = self.files
-
-        return {
-            "prefab_id": self.prefab_id,
-            "version": self.version,
-            "function_name": self.function_name,
-            "inputs": inputs,
-        }
-
-
-@dataclass
-class PrefabResult:
-    """预制件执行结果"""
-
-    status: CallStatus
-    output: Optional[Dict[str, Any]] = None
-    error: Optional[Dict[str, Any]] = None
-    job_id: Optional[str] = None
-
-    def is_success(self) -> bool:
-        """判断是否成功"""
-        return self.status == CallStatus.SUCCESS
-
-    def get(self, key: str, default: Any = None) -> Any:
-        """便捷获取输出字段"""
-        if self.output:
-            return self.output.get(key, default)
-        return default
-
-    def get_result(self) -> Any:
-        """获取业务结果"""
-        if self.output:
-            return self.output.get("result", self.output)
-        return None
-
-    def get_files(self) -> Dict[str, List[str]]:
-        """获取输出文件"""
-        if self.output:
-            return self.output.get("files", {})
-        return {}
-
-
-@dataclass
-class BatchResult:
-    """批量执行结果"""
-
-    job_id: str
-    status: str
-    results: List[PrefabResult]
-
-    def all_success(self) -> bool:
-        """判断是否全部成功"""
-        return all(r.is_success() for r in self.results)
-
-    def get_failed(self) -> List[PrefabResult]:
-        """获取失败的结果"""
-        return [r for r in self.results if not r.is_success()]
-
-
-@dataclass
-class StreamEvent:
-    """流式事件"""
-
-    type: StreamEventType
-    data: Any
-
-    @classmethod
-    def from_dict(cls, data: Dict[str, Any]) -> "StreamEvent":
-        """从字典创建"""
-        event_type = data.get("type", "content")
-        try:
-            event_type_enum = StreamEventType(event_type)
-        except ValueError:
-            event_type_enum = StreamEventType.CONTENT
-
-        return cls(type=event_type_enum, data=data.get("data"))