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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agent-builder-gateway-sdk
-Version: 0.3.0
+Version: 0.3.2
 Summary: Python SDK for Agent Builder Gateway - 用于 AI 构建的程序调用预制件
 Author: Agent Builder Team
 License: MIT
@@ -52,8 +52,16 @@ client = GatewayClient(jwt_token="your-jwt-token")
 
 # 或使用 API Key
 client = GatewayClient(api_key="sk-xxx")
+
+# 白名单模式（适用于 OpenHands 等白名单环境）
+client = GatewayClient()  # 无需提供认证信息
 ```
 
+**白名单模式说明**：
+- 如果你的环境已配置白名单（如 OpenHands、内部开发环境），可以直接创建客户端而无需提供认证信息
+- SDK 会以无鉴权模式发送请求，由 Gateway 基于 IP 白名单进行验证
+- 这种模式简化了开发流程，无需管理 token
+
 ### 调用预制件
 
 ```python

{agent_builder_gateway_sdk-0.3.0 → agent_builder_gateway_sdk-0.3.2}/README.md

@@ -32,8 +32,16 @@ client = GatewayClient(jwt_token="your-jwt-token")
 
 # 或使用 API Key
 client = GatewayClient(api_key="sk-xxx")
+
+# 白名单模式（适用于 OpenHands 等白名单环境）
+client = GatewayClient()  # 无需提供认证信息
 ```
 
+**白名单模式说明**：
+- 如果你的环境已配置白名单（如 OpenHands、内部开发环境），可以直接创建客户端而无需提供认证信息
+- SDK 会以无鉴权模式发送请求，由 Gateway 基于 IP 白名单进行验证
+- 这种模式简化了开发流程，无需管理 token
+
 ### 调用预制件
 
 ```python

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

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

{agent_builder_gateway_sdk-0.3.0 → agent_builder_gateway_sdk-0.3.2}/src/gateway_sdk/client.py

@@ -17,7 +17,7 @@ from .exceptions import (
 
 
 # Gateway 地址
-DEFAULT_GATEWAY_URL = "http://nodeport.sensedeal.vip:30566"
+DEFAULT_GATEWAY_URL = "http://agent-builder-gateway-test.sensedeal.vip"
 
 
 class GatewayClient:
@@ -38,14 +38,15 @@ class GatewayClient:
             api_key: API Key（优先级高）
             jwt_token: JWT Token（用于 AI 构建的程序）
             timeout: 请求超时时间（秒）
+        
+        Note:
+            如果不提供 api_key 或 jwt_token，客户端将在无鉴权模式下工作。
+            这适用于配置了白名单的环境（如 OpenHands）。
         """
         self.base_url = base_url.rstrip("/")
         self.auth = AuthManager(api_key=api_key, jwt_token=jwt_token)
         self.timeout = timeout
 
-        if not self.auth.is_authenticated():
-            raise AuthenticationError("必须提供 api_key 或 jwt_token")
-
     def run(
         self,
         prefab_id: str,

agent_builder_gateway_sdk-0.3.2/test_real_file.py

@@ -0,0 +1,64 @@
+#!/usr/bin/env python3
+"""测试真实的 S3 文件"""
+
+from gateway_sdk import GatewayClient
+
+def test_real_s3_file():
+    """使用真实的 S3 文件测试白名单模式"""
+    
+    print("🧪 测试白名单模式 + 真实 S3 文件")
+    
+    # 无需 token，白名单模式
+    client = GatewayClient()
+    
+    s3_url = "s3://cubeflow-dev/prefab-gateway/prefab-inputs/f94d31c2-154d-4363-8450-cc907117cbcb/2025/10/29/1761708431516-____.pdf"
+    
+    print(f"📄 S3 文件: {s3_url}")
+    print("📤 调用 file-processing-prefab...")
+    
+    try:
+        result = client.run(
+            prefab_id="file-processing-prefab",
+            version="0.1.5",
+            function_name="parse_file",
+            parameters={},
+            files={"input": [s3_url]}
+        )
+        
+        print(f"\n📊 SDK 层状态: {result.status}")
+        print(f"📋 Job ID: {result.job_id}")
+        
+        if result.is_success():
+            print("\n✅ 调用成功！")
+            
+            # 获取函数返回值
+            function_result = result.get_function_result()
+            if function_result:
+                print(f"\n📦 函数返回值:")
+                print(f"   success: {function_result.get('success')}")
+                print(f"   message: {function_result.get('message')}")
+                if function_result.get('content'):
+                    content = function_result.get('content')
+                    print(f"   content: {content[:200]}..." if len(content) > 200 else f"   content: {content}")
+                print(f"   input_file: {function_result.get('input_file')}")
+                print(f"   output_file: {function_result.get('output_file')}")
+                
+                if not function_result.get('success'):
+                    print(f"   error: {function_result.get('error')}")
+                    print(f"   error_code: {function_result.get('error_code')}")
+            
+            # 获取输出文件
+            output_files = result.get_files()
+            if output_files:
+                print(f"\n📁 输出文件: {output_files}")
+        else:
+            print(f"\n❌ 调用失败")
+            print(f"错误: {result.error}")
+            
+    except Exception as e:
+        print(f"\n❌ 异常: {e}")
+        import traceback
+        traceback.print_exc()
+
+if __name__ == "__main__":
+    test_real_s3_file()

agent_builder_gateway_sdk-0.3.2/test_whitelist_mode.py

@@ -0,0 +1,69 @@
+#!/usr/bin/env python3
+"""
+测试白名单模式（无鉴权）
+
+这个脚本测试 SDK 在白名单环境下是否能正常工作（不提供 api_key 或 jwt_token）
+"""
+
+from gateway_sdk import GatewayClient
+
+def test_whitelist_mode():
+    """测试白名单模式创建客户端"""
+    
+    print("🧪 测试 1: 无参数创建客户端")
+    try:
+        # ✅ 在白名单环境下，可以直接创建客户端
+        client = GatewayClient()
+        print("   ✅ 客户端创建成功（无鉴权模式）")
+        print(f"   - Gateway URL: {client.base_url}")
+        print(f"   - 超时设置: {client.timeout}秒")
+        print(f"   - 鉴权状态: {'已配置' if client.auth.is_authenticated() else '未配置（白名单模式）'}")
+    except Exception as e:
+        print(f"   ❌ 失败: {e}")
+        return False
+    
+    print("\n🧪 测试 2: 调用预制件（需要白名单环境）")
+    print("   ⚠️  注意: 这个测试只有在白名单环境下才能成功")
+    print("   如果不在白名单环境下，会返回 401 认证错误，这是正常的")
+    
+    try:
+        result = client.run(
+            prefab_id="video-processing-prefab",
+            version="0.3.1",
+            function_name="video_to_audio",
+            parameters={},
+            files={"input": ["s3://test-bucket/sample.mp4"]}
+        )
+        
+        if result.is_success():
+            print("   ✅ 预制件调用成功（白名单环境确认）")
+            function_result = result.get_function_result()
+            print(f"   - 返回状态: {function_result}")
+        else:
+            print(f"   ⚠️  预制件调用返回失败: {result.error}")
+            print("   这可能是业务逻辑错误，不是鉴权问题")
+            
+    except Exception as e:
+        error_msg = str(e)
+        if "401" in error_msg or "认证" in error_msg or "Unauthorized" in error_msg:
+            print(f"   ⚠️  认证失败（预期行为，不在白名单环境下）")
+            print(f"   错误信息: {error_msg}")
+        else:
+            print(f"   ❌ 其他错误: {error_msg}")
+    
+    print("\n" + "="*60)
+    print("✅ SDK 无参数创建功能测试完成")
+    print("="*60)
+    print("\n📝 使用说明:")
+    print("   在白名单环境（如 OpenHands）中，可以这样使用:")
+    print("   ```python")
+    print("   from gateway_sdk import GatewayClient")
+    print("   client = GatewayClient()  # 无需提供 token")
+    print("   result = client.run(...)")
+    print("   ```")
+    
+    return True
+
+if __name__ == "__main__":
+    test_whitelist_mode()
+

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

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