distributed-a2a 0.1.6rc2__tar.gz → 0.1.6rc4__tar.gz

{distributed_a2a-0.1.6rc2/distributed_a2a.egg-info → distributed_a2a-0.1.6rc4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.6rc2
+Version: 0.1.6rc4
 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.6rc2 → distributed_a2a-0.1.6rc4}/distributed_a2a/executors.py

@@ -83,9 +83,9 @@ class RoutingAgentExecutor(AgentExecutor):
 
 
 class RoutingExecutor(AgentExecutor):
-    def __init__(self, router_config: RouterConfig, routing_tool: BaseTool):
+    def __init__(self, router_config: RouterConfig, routing_tool: BaseTool) -> None:
         super().__init__()
-        api_key = os.environ.get(router_config.agent.llm.api_key_env)
+        api_key = os.environ.get(router_config.router.llm.api_key_env)
         self.routing_agent = StatusAgent[RoutingResponse](
             llm_config=router_config.agent.llm,
             system_prompt=ROUTING_SYSTEM_PROMPT,

{distributed_a2a-0.1.6rc2 → distributed_a2a-0.1.6rc4}/distributed_a2a/router.py

@@ -1,9 +1,10 @@
+import os
 from typing import Any
 
 from a2a.server.apps import A2ARESTFastAPIApplication
 from a2a.server.request_handlers import DefaultRequestHandler
 from a2a.server.tasks import InMemoryTaskStore
-from a2a.types import AgentCard
+from a2a.types import AgentCard, AgentSkill
 from fastapi import FastAPI
 
 from .executors import RoutingExecutor
@@ -17,14 +18,20 @@ def load_router(router_dic: dict[str, Any]) -> FastAPI:
     agent_card = AgentCard(
         name="Router",
         description="Agent to redirect to the best matching agent based on the agent card",
-        url="",
+        url=router_config.router.card.url,
         version="1.0.0",
-        default_input_modes=router_config.agent.card.default_input_modes,
-        default_output_modes=router_config.agent.card.default_output_modes,
-        skills=[],
-        preferred_transport=router_config.agent.card.preferred_transport_protocol,
+        default_input_modes=router_config.router.card.default_input_modes,
+        default_output_modes=router_config.router.card.default_output_modes,
+        skills=[AgentSkill(
+            id='routing',
+            name='Agent routing',
+            description='Identifies the most suitable agent for the given task and returns the agent card',
+            tags=['agent', 'routing']
+        )],
+        preferred_transport=router_config.router.card.preferred_transport_protocol,
         capabilities=CAPABILITIES
     )
+
     executor = RoutingExecutor(
         router_config=router_config,
         routing_tool=DynamoDbRegistryLookup(agent_card_tabel=AGENT_CARD_TABLE).as_tool()
@@ -38,4 +45,4 @@ def load_router(router_dic: dict[str, Any]) -> FastAPI:
             task_store=InMemoryTaskStore()  # TODO replace with dynamodb store
 
         )).build(title=agent_card.name,
-                 root_path=f"/{router_config.agent.card.name}")
+                 root_path=f"/{router_config.agent.card.name}")

{distributed_a2a-0.1.6rc2 → distributed_a2a-0.1.6rc4/distributed_a2a.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.6rc2
+Version: 0.1.6rc4
 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.6rc2 → distributed_a2a-0.1.6rc4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "distributed_a2a"
-version = "0.1.6rc2"
+version = "0.1.6rc4"
 description = "A library for building A2A agents with routing capabilities"
 readme = "README.md"
 requires-python = ">=3.10"