distributed-a2a 0.1.4__tar.gz → 0.1.5rc0__tar.gz

{distributed_a2a-0.1.4/distributed_a2a.egg-info → distributed_a2a-0.1.5rc0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.4
+Version: 0.1.5rc0
 Summary: A library for building A2A agents with routing capabilities
 Home-page: https://github.com/Barra-Technologies/distributed-a2a
 Author: Fabian Bell
@@ -26,6 +26,8 @@ Requires-Dist: langgraph>=0.0.20
 Requires-Dist: pydantic>=2.0.0
 Requires-Dist: boto3>=1.28.0
 Requires-Dist: a2a>=0.1.0
+Requires-Dist: build>=1.4.0
+Requires-Dist: twine>=6.2.0
 Dynamic: author
 Dynamic: home-page
 Dynamic: license-file

{distributed_a2a-0.1.4 → distributed_a2a-0.1.5rc0}/distributed_a2a/agent.py

@@ -7,7 +7,7 @@ from langchain_core.tools import BaseTool
 from langgraph.checkpoint.memory import MemorySaver
 from pydantic import BaseModel, Field
 
-from .model import get_model
+from .model import get_model, AgentConfig
 
 
 class AgentResponse(BaseModel):
@@ -29,7 +29,7 @@ class StringResponse(AgentResponse):
 
 class StatusAgent[ResponseT: AgentResponse]:
 
-    def __init__(self, system_prompt: str, name: str, api_key: str, is_routing: bool, tools: list[BaseTool]):
+    def __init__(self, agent_config: AgentConfig, api_key: str, is_routing: bool, tools: list[BaseTool]):
         response_format: type[AgentResponse]
         if is_routing:
             response_format = RoutingResponse
@@ -37,12 +37,15 @@ class StatusAgent[ResponseT: AgentResponse]:
             response_format = StringResponse
 
         self.agent = create_agent(
-            get_model(api_key),
+            get_model(api_key=api_key,
+                      model=agent_config.agent.llm.model,
+                      base_url=agent_config.agent.llm.base_url,
+                      reasoning_effort=agent_config.agent.llm.reasoning_effort),
             tools=tools,
             checkpointer=MemorySaver(),  # TODO replace by dynamodb
-            system_prompt=system_prompt,
+            system_prompt=agent_config.agent.llm.system_prompt,
             response_format=response_format,
-            name=name
+            name=agent_config.agent.card.name
         )
 
     async def __call__(self, message: str, context_id: str = None) -> ResponseT:

{distributed_a2a-0.1.4 → distributed_a2a-0.1.5rc0}/distributed_a2a/executors.py

@@ -1,13 +1,16 @@
 import json
 import logging
+import os
 
 from a2a.server.agent_execution import AgentExecutor, RequestContext
 from a2a.server.events import EventQueue
 from a2a.types import TaskStatusUpdateEvent, TaskStatus, TaskState, TaskArtifactUpdateEvent, Artifact
 from a2a.utils import new_text_artifact
 from langchain_core.tools import BaseTool
+from openai import api_key
 
 from .agent import StatusAgent, RoutingResponse, StringResponse
+from .model import AgentConfig
 
 logger = logging.getLogger(__name__)
 
@@ -20,10 +23,11 @@ You are a helpful routing assistant which routes user requests to specialized re
 
 class RoutingAgentExecutor(AgentExecutor):
 
-    def __init__(self, api_key: str, system_prompt: str, routing_tool: BaseTool, tools: list[BaseTool] | None = None):
+    def __init__(self, agent_config: AgentConfig, routing_tool: BaseTool, tools: list[BaseTool] | None = None):
         super().__init__()
+        api_key = os.environ.get(agent_config.agent.llm.api_key_env)
         self.agent = StatusAgent[StringResponse](
-            system_prompt=system_prompt,
+            system_prompt=agent_config.agent.llm.system_prompt,
             name="Router",
             api_key=api_key,
             is_routing=False,

distributed_a2a-0.1.5rc0/distributed_a2a/model.py

@@ -0,0 +1,50 @@
+from typing import List
+
+from langchain_core.language_models import BaseChatModel
+from langchain_openai import ChatOpenAI
+from pydantic import BaseModel
+
+
+class SkillConfig(BaseModel):
+    id: str
+    name: str
+    description: str
+    tags: List[str]
+
+
+class LLMConfig(BaseModel):
+    base_url: str
+    model: str
+    api_key_env: str
+    reasoning_effort: str
+    system_prompt: str
+
+
+class CardConfig(BaseModel):
+    name: str
+    version: str
+    default_input_modes: List[str]
+    default_output_modes: List[str]
+    preferred_transport_protocol: str
+    url: str
+    description: str
+    skills: List[SkillConfig]
+
+
+class AgentItem(BaseModel):
+    card: CardConfig
+    llm: LLMConfig
+
+
+class AgentConfig(BaseModel):
+    agent: AgentItem
+
+
+
+def get_model(api_key: str, model: str, base_url: str, reasoning_effort: str) -> BaseChatModel:
+    return ChatOpenAI(
+        model=model,
+        base_url=base_url,
+        api_key=lambda: api_key,
+        reasoning_effort=reasoning_effort
+    )

{distributed_a2a-0.1.4 → distributed_a2a-0.1.5rc0}/distributed_a2a/server.py

@@ -1,16 +1,18 @@
 import asyncio
 import time
 from contextlib import asynccontextmanager
+from typing import Any
 
 import boto3
 from a2a.server.apps import A2ARESTFastAPIApplication
 from a2a.server.request_handlers import DefaultRequestHandler
 from a2a.server.tasks import InMemoryTaskStore
 from a2a.types import AgentSkill, \
-    AgentCapabilities, AgentCard, TransportProtocol
+    AgentCapabilities, AgentCard
 from fastapi import FastAPI
 
 from .executors import RoutingAgentExecutor
+from .model import AgentConfig
 from .registry import DynamoDbRegistryLookup
 
 CAPABILITIES = AgentCapabilities(streaming=False, push_notifications=False)
@@ -35,34 +37,43 @@ async def heart_beat(name: str, agent_card_table: str, agent_card: AgentCard):
         )
 
 
-def load_app(name: str, description: str, skills: list[AgentSkill], api_key: str, system_prompt: str,
-             host: str) -> FastAPI:
 
-    routing_skill = AgentSkill(
+
+def load_app(agent_config: dict[str, Any]) -> FastAPI:
+
+    agent_config= AgentConfig.model_validate(agent_config)
+
+    skills = [AgentSkill(
+        id=skill.id,
+        name=skill.name,
+        description=skill.description,
+        tags=skill.tags)
+    for skill in agent_config.skills]
+    skills.append(AgentSkill(
         id='routing',
         name='Agent routing',
         description='Identifies the most suitable agent for the given task and returns the agent card',
         tags=['agent', 'routing']
-    )
+    ))
 
     agent_card = AgentCard(
-        name=name,
-        description=description,
-        url=host,
-        version='1.0.0',
-        default_input_modes=['text', 'text/plain'],
-        default_output_modes=['text', 'text/plain'],
-        capabilities=CAPABILITIES,
-        skills=skills + [routing_skill],
-        preferred_transport=TransportProtocol.http_json
+        name=agent_config.agent.name,
+        description=agent_config.agent.description,
+        url=agent_config.agent.url,
+        version=agent_config.agent.version,
+        default_input_modes=agent_config.agent.default_input_modes,
+        default_output_modes=agent_config.agent.default_output_modes,
+        skills=skills,
+        preferred_transport=agent_config.agent.preferred_transport_protocol,
     )
 
 
-    executor = RoutingAgentExecutor(api_key=api_key, system_prompt=system_prompt, routing_tool=DynamoDbRegistryLookup(agent_card_tabel=AGENT_CARD_TABLE).as_tool())
+    executor = RoutingAgentExecutor(agent_config=agent_config,
+                                    routing_tool=DynamoDbRegistryLookup(agent_card_tabel=AGENT_CARD_TABLE).as_tool())
 
     @asynccontextmanager
     async def lifespan(_: FastAPI):
-        asyncio.create_task(heart_beat(name=name, agent_card_table=AGENT_CARD_TABLE, agent_card=agent_card))
+        asyncio.create_task(heart_beat(name=agent_card.name, agent_card_table=AGENT_CARD_TABLE, agent_card=agent_card))
         yield
 
 
@@ -70,6 +81,6 @@ def load_app(name: str, description: str, skills: list[AgentSkill], api_key: str
         agent_card=agent_card,
         http_handler=DefaultRequestHandler(
             agent_executor=executor,
-            task_store=InMemoryTaskStore()
+            task_store=InMemoryTaskStore() #TODO replace with dynamodb store
         )
     ).build(title=name, lifespan=lifespan)

{distributed_a2a-0.1.4 → distributed_a2a-0.1.5rc0/distributed_a2a.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.4
+Version: 0.1.5rc0
 Summary: A library for building A2A agents with routing capabilities
 Home-page: https://github.com/Barra-Technologies/distributed-a2a
 Author: Fabian Bell
@@ -26,6 +26,8 @@ Requires-Dist: langgraph>=0.0.20
 Requires-Dist: pydantic>=2.0.0
 Requires-Dist: boto3>=1.28.0
 Requires-Dist: a2a>=0.1.0
+Requires-Dist: build>=1.4.0
+Requires-Dist: twine>=6.2.0
 Dynamic: author
 Dynamic: home-page
 Dynamic: license-file

{distributed_a2a-0.1.4 → distributed_a2a-0.1.5rc0}/distributed_a2a.egg-info/requires.txt

@@ -5,3 +5,5 @@ langgraph>=0.0.20
 pydantic>=2.0.0
 boto3>=1.28.0
 a2a>=0.1.0
+build>=1.4.0
+twine>=6.2.0

{distributed_a2a-0.1.4 → distributed_a2a-0.1.5rc0}/pyproject.toml

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

{distributed_a2a-0.1.4 → distributed_a2a-0.1.5rc0}/requirements.txt

@@ -5,3 +5,5 @@ langgraph>=0.0.20
 pydantic>=2.0.0
 boto3>=1.28.0
 a2a>=0.1.0
+build>=1.4.0
+twine>=6.2.0

distributed_a2a-0.1.4/distributed_a2a/model.py

@@ -1,11 +0,0 @@
-from langchain_core.language_models import BaseChatModel
-from langchain_openai import ChatOpenAI
-
-
-def get_model(api_key: str) -> BaseChatModel:
-    return ChatOpenAI(
-        model="google/gemini-2.5-flash",
-        base_url="https://openrouter.ai/api/v1",
-        api_key=lambda: api_key,
-        reasoning_effort="medium"
-    )