traia-iatp 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

traia_iatp/client/crewai_a2a_tools.py

@@ -32,7 +32,7 @@ _thread_local = threading.local()
 
 class A2AToolSchema(BaseModel):
     """Input schema for A2A tools."""
-    request: str = Field(description="The request or query to send to the A2A agent")
+    request: str = Field(description="The request or query to send to the A2A agent. Send only 'request' key when using this tool")
 
 
 class A2AToolConfig(BaseModel):

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/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.5.dist-info → traia_iatp-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: traia-iatp
-Version: 0.1.5
+Version: 0.1.7
 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
@@ -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.5.dist-info → traia_iatp-0.1.7.dist-info}/RECORD

@@ -5,7 +5,7 @@ traia_iatp/cli/__init__.py,sha256=nGgVg7pOjyIiWN5akKxqhKSHcpvU0VaDrH_vuhBEVcE,65
 traia_iatp/cli/main.py,sha256=MmBZxz5tob80_zJMeROKue39V2yvOeCp-8qkzzmFw28,20391
 traia_iatp/client/__init__.py,sha256=eo1oIMijV63WpklHEIftfoNWjEmhW4udcL7aIuZjf6M,259
 traia_iatp/client/a2a_client.py,sha256=fJwcPC9vmuH6Q4QSDUDGFNZyx9QQvuR9pPtbgKavkiM,10116
-traia_iatp/client/crewai_a2a_tools.py,sha256=RTmUzQmPnHx_wHhrAf-zPQjU9gSCyq-6S0RV1RDA7DQ,14317
+traia_iatp/client/crewai_a2a_tools.py,sha256=m989j13d8Ba1SDDNXuprBA0W7pjXvVApcCJWsBNliTI,14363
 traia_iatp/client/grpc_a2a_tools.py,sha256=pVt6UASXdt096ZNfcTqY0pjaTqLr-0Ptl8xd68rxRWQ,13003
 traia_iatp/client/root_path_a2a_client.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 traia_iatp/core/__init__.py,sha256=SdGs25MkNWok77wRtbCJFGkD85QAmYUAB40Fn9mHeXk,755
@@ -28,7 +28,7 @@ traia_iatp/mcp/templates/server.py.j2,sha256=zty-7UGU8CgWSbbFj-3tuJfrGJI79pBBYhc
 traia_iatp/registry/__init__.py,sha256=YGFGRqgRp0f09ND-FWTG9GQNwxz-HpUBYbt_KhLP-ZY,662
 traia_iatp/registry/atlas_search_indexes.json,sha256=_tXLH9us_AXonubuLKCX7uTDHuqw80u89E7Zt_tcdxI,6843
 traia_iatp/registry/embeddings.py,sha256=LvmQvQFbOBEZsQkCotBRLZWAyWDNYddRjaF5Ir8XEHw,11224
-traia_iatp/registry/iatp_search_api.py,sha256=CvXm0HWbSaspdxsdq4IJAIXvHSCybSA12PE2HHgrY-E,28984
+traia_iatp/registry/iatp_search_api.py,sha256=Y0X9MHjk95vEMxNHjUyCpd1egYXnRxMXcMccdl9O5Os,29255
 traia_iatp/registry/mongodb_registry.py,sha256=oPj8UXsZrlQZL3YAxBHpw2xeY5Ogauo95m6fmk4dzbY,32579
 traia_iatp/registry/readmes/ATLAS_SEARCH_INDEXES.md,sha256=vwYtLmqOfS-8f18CUc7D5z8meLl0O-GZ6n0cL1LXuy0,6049
 traia_iatp/registry/readmes/ATLAS_SEARCH_SETUP.md,sha256=6fpFmRI-ZYOugK6wW2Y20odlivqhjZrNWcOIa3Smzpo,4782
@@ -39,7 +39,7 @@ traia_iatp/registry/readmes/MONGODB_X509_AUTH.md,sha256=VCKzHfklSN4kg-ycdkf2OQdp
 traia_iatp/registry/readmes/README.md,sha256=rb8XSQLUUqV-rpD0G7QLlWbvuse2wRXby-bJ4Pv23Iw,7178
 traia_iatp/registry/readmes/REFACTORING_SUMMARY.md,sha256=DwnvOmtCb0kGcqBDR7R2EdpaGxh8z7GycxTyYP5msyA,6038
 traia_iatp/server/__init__.py,sha256=sh_tdZJhF1PO2GBTfPMy9UI4Dqz1TQBqnJTBNA68VgA,385
-traia_iatp/server/a2a_server.py,sha256=3-RZngcoOZn_T9bAEtJrSDeFWGFomfXYMaIRNeHP2TI,7701
+traia_iatp/server/a2a_server.py,sha256=lD9s3KSvwIpROL6QAHAg5Y_Rf7gOF6Nr1c75GJ9aitI,7854
 traia_iatp/server/example_template_usage.py,sha256=5hiI9eOxvT0Cow6Khm63IHZIuadNDPDEVOHC94nR5Fc,2294
 traia_iatp/server/iatp_server_agent_generator.py,sha256=uMVO9jODPXyLs6DxGSfhJl01N6l4_CShu88O3I9UZqQ,8880
 traia_iatp/server/iatp_server_template_generator.py,sha256=AALMsYXITlHWC_UUOoBZe0qEOKBjIAgDYEIVJjHdbkc,9772
@@ -65,9 +65,9 @@ traia_iatp/utils/__init__.py,sha256=D739Fc3KWPlAaPZ-3XM7uRvBAA7V8b4t6PBYJvMSsig,
 traia_iatp/utils/docker_utils.py,sha256=mPD09TPJKEe9SKjrR5E_o9z8xle-iCZq6JLzHZ6JXFk,8634
 traia_iatp/utils/general.py,sha256=g6sRKzalBPwRWobBg3a7pz4eNIvvxv8RefoUNFdHm4o,1503
 traia_iatp/utils/iatp_utils.py,sha256=RrGzk2ENvGIwuuMigOUrK5Yne0esl1r_K2P2iVygUgM,4680
-traia_iatp-0.1.5.dist-info/licenses/LICENSE,sha256=R8vcu8GZdTTXF40sbOll11-FbarDm7_PiuaGzQp9whw,1065
-traia_iatp-0.1.5.dist-info/METADATA,sha256=e74bs-LpRmNtuG7w8ylvtDCnHiot4ummAyfExn1UMv8,13218
-traia_iatp-0.1.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-traia_iatp-0.1.5.dist-info/entry_points.txt,sha256=K4p9Z2lBdPEybzXfed3JbUtrVHDdg60CRnJ178df7K0,55
-traia_iatp-0.1.5.dist-info/top_level.txt,sha256=FozpD3vmZ4WP01hjSDUt5ENn0ttnIxxm1tPrTiIsyzI,11
-traia_iatp-0.1.5.dist-info/RECORD,,
+traia_iatp-0.1.7.dist-info/licenses/LICENSE,sha256=R8vcu8GZdTTXF40sbOll11-FbarDm7_PiuaGzQp9whw,1065
+traia_iatp-0.1.7.dist-info/METADATA,sha256=kkicBHG78RSC5h_Y53kNUcnPPBel8qUXHsugo2bRhmM,13250
+traia_iatp-0.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+traia_iatp-0.1.7.dist-info/entry_points.txt,sha256=K4p9Z2lBdPEybzXfed3JbUtrVHDdg60CRnJ178df7K0,55
+traia_iatp-0.1.7.dist-info/top_level.txt,sha256=FozpD3vmZ4WP01hjSDUt5ENn0ttnIxxm1tPrTiIsyzI,11
+traia_iatp-0.1.7.dist-info/RECORD,,