traia-iatp 0.1.6__tar.gz → 0.1.8__tar.gz

{traia_iatp-0.1.6/src/traia_iatp.egg-info → traia_iatp-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: traia-iatp
-Version: 0.1.6
+Version: 0.1.8
 Summary: Inter-Agent Transfer Protocol (IATP) - Enable AI Agents to utilize other AI Agents as tools
 Project-URL: Documentation, https://pypi.org/project/traia-iatp
 Project-URL: Source, https://github.com/Traia-IO/IATP
@@ -21,7 +21,7 @@ Requires-Dist: httpx>=0.28.1
 Requires-Dist: jinja2>=3.1.6
 Requires-Dist: openai>=1.90.0
 Requires-Dist: pydantic>=2.11.5
-Requires-Dist: pymongo>=4.13.2
+Requires-Dist: pymongo[aws]>=4.13.2
 Requires-Dist: python-dotenv>=1.1.0
 Requires-Dist: pytz>=2025.2
 Requires-Dist: requests>=2.32.3
@@ -29,6 +29,7 @@ Requires-Dist: rich>=13.9.4
 Requires-Dist: typer>=0.16.0
 Requires-Dist: uvicorn>=0.34.3
 Requires-Dist: agentops>=0.3.0
+Requires-Dist: dnspython>=2.7.0
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0.0; extra == "dev"
 Requires-Dist: pytest-cov>=4.0.0; extra == "dev"

{traia_iatp-0.1.6 → traia_iatp-0.1.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "traia-iatp"
-version = "0.1.6"
+version = "0.1.8"
 description = "Inter-Agent Transfer Protocol (IATP) - Enable AI Agents to utilize other AI Agents as tools"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -26,14 +26,15 @@ dependencies = [
     "jinja2>=3.1.6",
     "openai>=1.90.0",
     "pydantic>=2.11.5",
-    "pymongo>=4.13.2",
+    "pymongo[aws]>=4.13.2",
     "python-dotenv>=1.1.0",
     "pytz>=2025.2",
     "requests>=2.32.3",
     "rich>=13.9.4",
     "typer>=0.16.0",
     "uvicorn>=0.34.3",
-    "agentops>=0.3.0"
+    "agentops>=0.3.0",
+    "dnspython>=2.7.0",
 ]
 
 [project.optional-dependencies]

{traia_iatp-0.1.6 → traia_iatp-0.1.8}/src/traia_iatp/registry/iatp_search_api.py

@@ -89,7 +89,13 @@ class UtilityAgentInfo:
 
 def get_readonly_connection_string() -> str:
     """Get read-only MongoDB connection string."""
-    # Try X.509 certificate authentication first
+
+    # Try IAM access
+    cluster_host_name = os.environ.get('CLUSTER_HOST_NAME', None)
+    if cluster_host_name:
+        return f"mongodb+srv://{cluster_host_name}/?authMechanism=MONGODB-AWS&authSource=%24external&retryWrites=true&w=majority&appName=Traia-IATP-Cluster"
+
+    # Try X.509 certificate authentication next
     cert_file = os.getenv("MONGODB_X509_CERT_FILE")
     if cert_file and os.path.exists(cert_file):
         # Extract just the cluster hostname without query parameters

{traia_iatp-0.1.6 → traia_iatp-0.1.8}/src/traia_iatp/server/a2a_server.py

@@ -39,7 +39,7 @@ class UtilityAgencyExecutor(AgentExecutor):
             # Get the user's request from context
             user_message = context.get_last_user_message()
             if not user_message:
-                event_queue.enqueue_event(
+                await event_queue.enqueue_event(
                     new_agent_text_message("No user message provided")
                 )
                 return
@@ -73,6 +73,10 @@ class UtilityAgencyExecutor(AgentExecutor):
                 new_agent_text_message(f"Error processing request: {str(e)}")
             )
 
+    async def cancel(self, context: RequestContext, event_queue: EventQueue) -> None:
+        """Cancel all tasks."""
+        pass
+
 
 class MCPToolExecutor(AgentExecutor):
     """Agent executor for individual MCP tool execution."""
@@ -89,7 +93,7 @@ class MCPToolExecutor(AgentExecutor):
             # Get tool arguments from context
             user_message = context.get_last_user_message()
             if not user_message:
-                event_queue.enqueue_event(
+                await event_queue.enqueue_event(
                     new_agent_text_message("No arguments provided")
                 )
                 return
@@ -110,7 +114,7 @@ class MCPToolExecutor(AgentExecutor):
             
             try:
                 result = await mcp_client.call_tool(self.tool_name, arguments)
-                event_queue.enqueue_event(
+                await event_queue.enqueue_event(
                     new_agent_text_message(str(result))
                 )
             finally:
@@ -118,7 +122,7 @@ class MCPToolExecutor(AgentExecutor):
                 
         except Exception as e:
             logger.error(f"Error executing MCP tool {self.tool_name}: {e}")
-            event_queue.enqueue_event(
+            await event_queue.enqueue_event(
                 new_agent_text_message(f"Error executing tool: {str(e)}")
             )
 
@@ -194,7 +198,7 @@ def create_a2a_server(
     # Create the A2A application
     app = A2AStarletteApplication(
         agent_card=agent_card,
-        request_handler=request_handler
+        http_handler=request_handler
     )
     
     return app

{traia_iatp-0.1.6 → traia_iatp-0.1.8/src/traia_iatp.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: traia-iatp
-Version: 0.1.6
+Version: 0.1.8
 Summary: Inter-Agent Transfer Protocol (IATP) - Enable AI Agents to utilize other AI Agents as tools
 Project-URL: Documentation, https://pypi.org/project/traia-iatp
 Project-URL: Source, https://github.com/Traia-IO/IATP
@@ -21,7 +21,7 @@ Requires-Dist: httpx>=0.28.1
 Requires-Dist: jinja2>=3.1.6
 Requires-Dist: openai>=1.90.0
 Requires-Dist: pydantic>=2.11.5
-Requires-Dist: pymongo>=4.13.2
+Requires-Dist: pymongo[aws]>=4.13.2
 Requires-Dist: python-dotenv>=1.1.0
 Requires-Dist: pytz>=2025.2
 Requires-Dist: requests>=2.32.3
@@ -29,6 +29,7 @@ Requires-Dist: rich>=13.9.4
 Requires-Dist: typer>=0.16.0
 Requires-Dist: uvicorn>=0.34.3
 Requires-Dist: agentops>=0.3.0
+Requires-Dist: dnspython>=2.7.0
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0.0; extra == "dev"
 Requires-Dist: pytest-cov>=4.0.0; extra == "dev"

{traia_iatp-0.1.6 → traia_iatp-0.1.8}/src/traia_iatp.egg-info/requires.txt

@@ -9,7 +9,7 @@ httpx>=0.28.1
 jinja2>=3.1.6
 openai>=1.90.0
 pydantic>=2.11.5
-pymongo>=4.13.2
+pymongo[aws]>=4.13.2
 python-dotenv>=1.1.0
 pytz>=2025.2
 requests>=2.32.3
@@ -17,6 +17,7 @@ rich>=13.9.4
 typer>=0.16.0
 uvicorn>=0.34.3
 agentops>=0.3.0
+dnspython>=2.7.0
 
 [dev]
 pytest>=7.0.0