asteroid-odyssey 1.0.2__tar.gz → 1.0.3__tar.gz

{asteroid_odyssey-1.0.2 → asteroid_odyssey-1.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: asteroid-odyssey
-Version: 1.0.2
+Version: 1.0.3
 Summary: A Python SDK for browser automation using Asteroid platform.
 Author-email: David Mlcoch <founders@asteroid.com>
 License-Expression: MIT

{asteroid_odyssey-1.0.2 → asteroid_odyssey-1.0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "asteroid-odyssey"
-version = "1.0.2"
+version = "1.0.3"
 authors = [
   { name = "David Mlcoch", email = "founders@asteroid.com" },
 ]

{asteroid_odyssey-1.0.2 → asteroid_odyssey-1.0.3}/src/asteroid_odyssey/client.py

@@ -10,6 +10,7 @@ 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,
@@ -58,14 +59,14 @@ class AsteroidClient:
         self.sdk_api = SDKApi(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,9 +75,12 @@ 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)
             return response.execution_id
@@ -278,9 +282,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 +320,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-1.0.2 → asteroid_odyssey-1.0.3}/src/asteroid_odyssey.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: asteroid-odyssey
-Version: 1.0.2
+Version: 1.0.3
 Summary: A Python SDK for browser automation using Asteroid platform.
 Author-email: David Mlcoch <founders@asteroid.com>
 License-Expression: MIT