distributed-a2a 0.1.5rc19__tar.gz → 0.1.6rc1__tar.gz

{distributed_a2a-0.1.5rc19/distributed_a2a.egg-info → distributed_a2a-0.1.6rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.5rc19
+Version: 0.1.6rc1
 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.5rc19 → distributed_a2a-0.1.6rc1}/distributed_a2a/__init__.py

@@ -1,7 +1,9 @@
 from .client import RoutingA2AClient
+from .router import load_router
 from .server import load_app
 
 __all__ = [
     "load_app",
+    "load_router",
     "RoutingA2AClient"
 ]

{distributed_a2a-0.1.5rc19 → distributed_a2a-0.1.6rc1}/distributed_a2a/agent.py

@@ -4,7 +4,6 @@ from a2a.types import TaskState
 from langchain.agents import create_agent
 from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import BaseTool
-from langgraph.checkpoint.memory import MemorySaver
 from langgraph_dynamodb_checkpoint import DynamoDBSaver
 from pydantic import BaseModel, Field
 

{distributed_a2a-0.1.5rc19 → distributed_a2a-0.1.6rc1}/distributed_a2a/executors.py

@@ -9,7 +9,7 @@ from a2a.utils import new_text_artifact
 from langchain_core.tools import BaseTool
 
 from .agent import StatusAgent, RoutingResponse, StringResponse
-from .model import AgentConfig
+from .model import AgentConfig, RouterConfig
 
 logger = logging.getLogger(__name__)
 
@@ -80,3 +80,47 @@ class RoutingAgentExecutor(AgentExecutor):
             final=True,
             context_id=context.context_id,
             task_id=context.task_id))
+
+
+class RoutingExecutor(AgentExecutor):
+    def __init__(self, router_config: RouterConfig, routing_tool: BaseTool):
+        super().__init__()
+        api_key = os.environ.get(router_config.agent.llm.api_key_env)
+        self.routing_agent = StatusAgent[RoutingResponse](
+            llm_config=router_config.agent.llm,
+            system_prompt=ROUTING_SYSTEM_PROMPT,
+            name=router_config.router.card.name,
+            api_key=api_key,
+            is_routing=True,
+            tools=[routing_tool]
+        )
+
+    async def cancel(self, context: RequestContext, event_queue: EventQueue) -> None:
+        raise NotImplementedError
+
+    async def execute(self, context: RequestContext, event_queue: EventQueue) -> None:
+        await event_queue.enqueue_event(TaskStatusUpdateEvent(status=TaskStatus(state=TaskState.working),
+                                                              final=False,
+                                                              context_id=context.context_id,
+                                                              task_id=context.task_id))
+
+        agent_response: RoutingResponse = await self.routing_agent(message=context.get_user_input(),
+                                                                   context_id=context.context_id)
+        agent_name: str = json.loads(agent_response.agent_card)["name"]
+        logger.info(f"Request with id {context.context_id} got rejected and will be rerouted to a '{agent_name}'.",
+                    extra={"card": agent_response.agent_card})
+        artifact = new_text_artifact(name='target_agent', description='New target agent for request.',
+                                     text=agent_response.agent_card)
+
+        # publish actual result
+        await event_queue.enqueue_event(TaskArtifactUpdateEvent(append=False,
+                                                                context_id=context.context_id,
+                                                                task_id=context.task_id,
+                                                                last_chunk=True,
+                                                                artifact=artifact))
+        # set and publish the final status
+        await event_queue.enqueue_event(TaskStatusUpdateEvent(status=TaskStatus(
+            state=TaskState(agent_response.status)),
+            final=True,
+            context_id=context.context_id,
+            task_id=context.task_id))

{distributed_a2a-0.1.5rc19 → distributed_a2a-0.1.6rc1}/distributed_a2a/model.py

@@ -49,6 +49,13 @@ class AgentConfig(BaseModel):
     agent: AgentItem = Field(description="The agent configuration node")
 
 
+class RouterItem(BaseModel):
+    card: CardConfig = Field(description="The router card configuration node")
+    llm: LLMConfig = Field(description="The LLM configuration node")
+
+
+class RouterConfig(BaseModel):
+    router: RouterItem = Field(description="The router configuration node")
 
 def get_model(api_key: str, model: str, base_url: str, reasoning_effort: str) -> BaseChatModel:
     return ChatOpenAI(

distributed_a2a-0.1.6rc1/distributed_a2a/router.py

@@ -0,0 +1,41 @@
+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 fastapi import FastAPI
+
+from distributed_a2a.distributed_a2a.executors import RoutingExecutor
+from distributed_a2a.distributed_a2a.model import RouterConfig
+from distributed_a2a.distributed_a2a.registry import DynamoDbRegistryLookup
+from distributed_a2a.distributed_a2a.server import CAPABILITIES, AGENT_CARD_TABLE
+
+
+def load_router(router_dic: dict[str, Any]) -> FastAPI:
+    router_config = RouterConfig.model_validate(router_dic)
+    agent_card = AgentCard(
+        name="Router",
+        description="Agent to redirect to the best matching agent based on the agent 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,
+        capabilities=CAPABILITIES
+    )
+    executor = RoutingExecutor(
+        router_config=router_config,
+        routing_tool=DynamoDbRegistryLookup(agent_card_tabel=AGENT_CARD_TABLE).as_tool()
+
+    )
+
+    return A2ARESTFastAPIApplication(
+        agent_card=agent_card,
+        http_handler=DefaultRequestHandler(
+            agent_executor=executor,
+            task_store=InMemoryTaskStore()  # TODO replace with dynamodb store
+
+        )).build(title=agent_card.name,
+                 root_path=f"/{router_config.agent.card.name}")

{distributed_a2a-0.1.5rc19 → distributed_a2a-0.1.6rc1/distributed_a2a.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.5rc19
+Version: 0.1.6rc1
 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.5rc19 → distributed_a2a-0.1.6rc1}/distributed_a2a.egg-info/SOURCES.txt

@@ -10,8 +10,8 @@ distributed_a2a/client.py
 distributed_a2a/executors.py
 distributed_a2a/model.py
 distributed_a2a/registry.py
+distributed_a2a/router.py
 distributed_a2a/server.py
-distributed_a2a/tool_registry_test.py
 distributed_a2a.egg-info/PKG-INFO
 distributed_a2a.egg-info/SOURCES.txt
 distributed_a2a.egg-info/dependency_links.txt

{distributed_a2a-0.1.5rc19 → distributed_a2a-0.1.6rc1}/pyproject.toml

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

distributed_a2a-0.1.5rc19/distributed_a2a/tool_registry_test.py

@@ -1,6 +0,0 @@
-from distributed_a2a.distributed_a2a.registry import ToolRegistry
-
-if __name__ == '__main__':
-    reg = ToolRegistry(tools_table="mcp_tools")
-    res = reg.get_tool_by_name(tool_name="my_second_tool")
-    print(res)