distributed-a2a 0.1.9rc11__py3-none-any.whl → 0.1.9rc13__py3-none-any.whl

distributed_a2a/executors.py

@@ -63,7 +63,7 @@ class RoutingAgentExecutor(AgentExecutor):
                                                               final=False,
                                                               context_id=context.context_id,
                                                               task_id=context.task_id))
-        self.reinitialize_agent_with_tools()
+        await self.reinitialize_agent_with_tools()
         agent_response: StringResponse = await self.agent(message=context.get_user_input(),
                                                           context_id=context.context_id)
 
@@ -104,12 +104,16 @@ class RoutingAgentExecutor(AgentExecutor):
             context_id=context.context_id,
             task_id=context.task_id))
 
-    def reinitialize_agent_with_tools(self):
+    async def reinitialize_agent_with_tools(self):
         mcp_tools = self.mcp_registry.get_mcp_tool_for_agent(self.agent_config.agent.card.name)
+        if not mcp_tools:
+            #no mcp tool found no need to reinitialize Agent
+            return
+
         logger.info(f"Agent {self.agent_config.agent.card.name} has access to the following tools: {mcp_tools}")
         tools = {tool["name"]: {"url": tool["url"], "transport": tool["protocol"]} for tool in mcp_tools}
         mcp_client = MultiServerMCPClient(tools)
-        mcp_tools = mcp_client.get_tools()
+        mcp_tools = await mcp_client.get_tools()
 
         self.agent = StatusAgent[StringResponse](
             llm_config=self.agent_config.agent.llm,
@@ -122,7 +126,7 @@ class RoutingAgentExecutor(AgentExecutor):
 
 
 class RoutingExecutor(AgentExecutor):
-    def __init__(self, router_config: RouterConfig, routing_tool: BaseTool) -> None:
+    def __init__(self, router_config: RouterConfig) -> None:
         super().__init__()
         api_key = os.environ.get(router_config.router.llm.api_key_env)
         if api_key is None:
@@ -134,7 +138,7 @@ class RoutingExecutor(AgentExecutor):
             name=router_config.router.card.name,
             api_key=api_key,
             is_routing=True,
-            tools=[routing_tool]
+            tools=[]
         )
 
     async def cancel(self, context: RequestContext, event_queue: EventQueue) -> None:

distributed_a2a/router.py

@@ -34,7 +34,6 @@ def load_router(router_dic: dict[str, Any]) -> FastAPI:
 
     executor = RoutingExecutor(
         router_config=router_config,
-        routing_tool=AgentRegistryLookup(router_config.router.registry.agent.url).as_tool()
     )
 
     return A2ARESTFastAPIApplication(

{distributed_a2a-0.1.9rc11.dist-info → distributed_a2a-0.1.9rc13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.9rc11
+Version: 0.1.9rc13
 Summary: A library for building A2A agents with routing capabilities
 Home-page: https://github.com/Barra-Technologies/distributed-a2a
 Author: Fabian Bell

{distributed_a2a-0.1.9rc11.dist-info → distributed_a2a-0.1.9rc13.dist-info}/RECORD

@@ -1,13 +1,13 @@
 distributed_a2a/__init__.py,sha256=gO8sTT20O6XvDRzj7B322O9v6HxV-_eIlJD1wMlPYoI,171
 distributed_a2a/agent.py,sha256=qY775N3Br2YMuaHLALU4t6MO47syuEoTtFLLPRr8TyA,2910
 distributed_a2a/client.py,sha256=3tEBqu3HKEQyOFk5vsO1YiuKP2pZx-n6SCzsJUYNcyc,4601
-distributed_a2a/executors.py,sha256=3W2bzoiOjPKXY5uAeUdMVjykhG6IKv5Azse7Ssa3KU8,9387
+distributed_a2a/executors.py,sha256=CBjCZiM2YNSZslyZPqdHLDNFLbqeCNDYDSnVm6GJ3wY,9476
 distributed_a2a/model.py,sha256=rW9jY0OIR8L_I77zz47JQMd-ucQQnnfNw_HIS-YvrcM,3533
 distributed_a2a/registry.py,sha256=lTXdS3PHXXMPXNjZDxyNRJjhJ1uZJOxgOgpokQp7boc,2050
-distributed_a2a/router.py,sha256=l1kDn5_iYNq6QZj8JsouLC6zz5vdK-qGSpj4KODwXY8,1834
+distributed_a2a/router.py,sha256=Uk8TQdcwbH5a30lfDm0iA0B1usQg3x6A4BnK-UbDy4A,1742
 distributed_a2a/server.py,sha256=-__c-BKzJWSXpW21NOp6faUIdjWV_o32jqv42_lPi2g,3352
-distributed_a2a-0.1.9rc11.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
-distributed_a2a-0.1.9rc11.dist-info/METADATA,sha256=VLpU5fyKmRYMj3perY0ZLfftOZsky5nJc3YnLGj5P3M,3244
-distributed_a2a-0.1.9rc11.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-distributed_a2a-0.1.9rc11.dist-info/top_level.txt,sha256=23qJ8n5k7796BHDK7a58uuO-X4GV0EgUWcGi8NIn-0k,16
-distributed_a2a-0.1.9rc11.dist-info/RECORD,,
+distributed_a2a-0.1.9rc13.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
+distributed_a2a-0.1.9rc13.dist-info/METADATA,sha256=2AIngK6hTX74qrOP_UrigWCfYAeO70_tc-SdZnX9Bso,3244
+distributed_a2a-0.1.9rc13.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+distributed_a2a-0.1.9rc13.dist-info/top_level.txt,sha256=23qJ8n5k7796BHDK7a58uuO-X4GV0EgUWcGi8NIn-0k,16
+distributed_a2a-0.1.9rc13.dist-info/RECORD,,