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

asteroid_odyssey/client.py

@@ -10,11 +10,12 @@ without modifying any generated files.
 
 import time
 import os
+import logging
 from typing import Dict, Any, Optional, List, Union, Tuple
 from .openapi_client import (
     Configuration,
     ApiClient,
-    SDKApi,
+    APIApi,
     ExecutionApi,
     ExecutionStatusResponse,
     ExecutionResultResponse,
@@ -55,17 +56,17 @@ 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]) -> str: 
+    def execute_agent(self, agent_id: str, execution_data: Dict[str, Any], agent_profile_id: Optional[str] = None) -> str: 
         """
         Execute an agent with the provided parameters.
         
         Args:
             agent_id: The ID of the agent to execute
-            agent_profile_id: The ID of the agent profile
             execution_data: The execution parameters
+            agent_profile_id: Optional ID of the agent profile
             
         Returns:
             The execution ID
@@ -74,11 +75,11 @@ class AsteroidClient:
             Exception: If the execution request fails
             
         Example:
-            execution_id = client.execute_structured_agent('my-agent-id', 'agent-profile-id', {'input': 'some dynamic value'})
+            execution_id = client.execute_agent('my-agent-id', {'input': 'some dynamic value'}, 'agent-profile-id')
         """
-        req = StructuredAgentExecutionRequest(dynamic_data=execution_data)
+        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}")
@@ -101,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}")
     
@@ -123,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}")
     
@@ -269,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}")
@@ -278,9 +279,23 @@ class AsteroidClient:
         """Context manager entry."""
         return self
     
-    def __exit__(self, exc_type, exc_value, traceback):
-        """Context manager exit."""
-        pass
+    def __exit__(self, exc_type, exc_value, tb):
+        """Context manager exit: clean up API client connection pool."""
+        try:
+            # Try to grab the pool_manager; if any attr is missing, skip
+            try:
+                pool_manager = self.api_client.rest_client.pool_manager
+            except AttributeError:
+                pool_manager = None
+
+            if pool_manager:
+                pool_manager.clear()
+        except Exception as e:
+            # Log but don't mask the original exception (if any)
+            logging.warning("Failed to clear connection pool: %s", e)
+
+        # Returning False allows any exception in the 'with' block to propagate
+        return False
 
 
 # Convenience functions that mirror the TypeScript SDK pattern
@@ -302,23 +317,23 @@ def create_client(api_key: str, base_url: Optional[str] = None) -> AsteroidClien
     """
     return AsteroidClient(api_key, base_url)
 
-def execute_agent(client: AsteroidClient, agent_id: str, agent_profile_id: str, execution_data: Dict[str, Any]) -> str:
+def execute_agent(client: AsteroidClient, agent_id: str, execution_data: Dict[str, Any], agent_profile_id: Optional[str] = None) -> str:
     """
     Execute an agent with the provided parameters.
     
     Args:
         client: The AsteroidClient instance
         agent_id: The ID of the agent to execute
-        agent_profile_id: The ID of the agent profile
         execution_data: The execution parameters
+        agent_profile_id: Optional ID of the agent profile
         
     Returns:
         The execution ID
         
     Example:
-        execution_id = execute_agent(client, 'my-agent-id', {'input': 'some dynamic value'})
+        execution_id = execute_agent(client, 'my-agent-id', {'input': 'some dynamic value'}, 'agent-profile-id')
     """
-    return client.execute_agent(agent_id, agent_profile_id, execution_data)
+    return client.execute_agent(agent_id, execution_data, agent_profile_id)
 
 
 

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