asteroid-odyssey 1.0.3__py3-none-any.whl → 1.1.0__py3-none-any.whl

asteroid_odyssey/client.py

@@ -15,7 +15,7 @@ from typing import Dict, Any, Optional, List, Union, Tuple
 from .openapi_client import (
     Configuration,
     ApiClient,
-    SDKApi,
+    APIApi,
     ExecutionApi,
     ExecutionStatusResponse,
     ExecutionResultResponse,
@@ -56,7 +56,7 @@ class AsteroidClient:
         )
         
         self.api_client = ApiClient(config)
-        self.sdk_api = SDKApi(self.api_client)
+        self.api_api = APIApi(self.api_client)
         self.execution_api = ExecutionApi(self.api_client)
         
     def execute_agent(self, agent_id: str, execution_data: Dict[str, Any], agent_profile_id: Optional[str] = None) -> str: 
@@ -77,12 +77,9 @@ class AsteroidClient:
         Example:
             execution_id = client.execute_agent('my-agent-id', {'input': 'some dynamic value'}, 'agent-profile-id')
         """
-        req = StructuredAgentExecutionRequest(
-            dynamic_data=execution_data,
-            agent_profile_id=agent_profile_id
-        )
+        req = StructuredAgentExecutionRequest(dynamic_data=execution_data, agent_profile_id=agent_profile_id)
         try:
-            response = self.sdk_api.execute_agent_structured(agent_id, req)
+            response = self.execution_api.execute_agent_structured(agent_id, req)
             return response.execution_id
         except ApiException as e:
             raise Exception(f"Failed to execute agent: {e}")
@@ -105,7 +102,7 @@ class AsteroidClient:
             print(status.status)
         """
         try:
-            return self.sdk_api.get_execution_status(execution_id)
+            return self.execution_api.get_execution_status(execution_id)
         except ApiException as e:
             raise Exception(f"Failed to get execution status: {e}")
     
@@ -127,12 +124,12 @@ class AsteroidClient:
             print(result)
         """
         try:
-            response = self.sdk_api.get_execution_result(execution_id)
+            response = self.execution_api.get_execution_result(execution_id)
             
             if response.error:
                 raise Exception(response.error)
             
-            return response.result or {}
+            return response.execution_result or {}
         except ApiException as e:
             raise Exception(f"Failed to get execution result: {e}")
     
@@ -273,7 +270,7 @@ class AsteroidClient:
             print(f"Recording available at: {recording_url}")
         """
         try:
-            response = self.sdk_api.get_browser_session_recording(execution_id)
+            response = self.execution_api.get_browser_session_recording(execution_id)
             return response.recording_url
         except ApiException as e:
             raise Exception(f"Failed to get browser session recording: {e}")

asteroid_odyssey/openapi_client/__init__.py

@@ -20,7 +20,6 @@ __version__ = "1.0.0"
 __all__ = [
     "APIApi",
     "ExecutionApi",
-    "SDKApi",
     "ApiResponse",
     "ApiClient",
     "Configuration",
@@ -46,7 +45,6 @@ __all__ = [
 # import apis into sdk package
 from asteroid_odyssey.openapi_client.api.api_api import APIApi as APIApi
 from asteroid_odyssey.openapi_client.api.execution_api import ExecutionApi as ExecutionApi
-from asteroid_odyssey.openapi_client.api.sdk_api import SDKApi as SDKApi
 
 # import ApiClient
 from asteroid_odyssey.openapi_client.api_response import ApiResponse as ApiResponse

asteroid_odyssey/openapi_client/api/__init__.py

@@ -3,5 +3,4 @@
 # import apis into api package
 from asteroid_odyssey.openapi_client.api.api_api import APIApi
 from asteroid_odyssey.openapi_client.api.execution_api import ExecutionApi
-from asteroid_odyssey.openapi_client.api.sdk_api import SDKApi