distributed-a2a 0.1.5rc5__py3-none-any.whl → 0.1.5rc7__py3-none-any.whl

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, AgentConfig
+from .model import get_model, AgentConfig, LLMConfig
 
 
 class AgentResponse(BaseModel):
@@ -29,7 +29,7 @@ class StringResponse(AgentResponse):
 
 class StatusAgent[ResponseT: AgentResponse]:
 
-    def __init__(self, agent_config: AgentConfig, api_key: str, is_routing: bool, tools: list[BaseTool]):
+    def __init__(self, llm_config: LLMConfig, name: str, system_prompt: str, api_key: str, is_routing: bool, tools: list[BaseTool]):
         response_format: type[AgentResponse]
         if is_routing:
             response_format = RoutingResponse
@@ -38,14 +38,14 @@ class StatusAgent[ResponseT: AgentResponse]:
 
         self.agent = create_agent(
             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),
+                      model=llm_config.model,
+                      base_url=llm_config.base_url,
+                      reasoning_effort=llm_config.reasoning_effort),
             tools=tools,
             checkpointer=MemorySaver(),  # TODO replace by dynamodb
-            system_prompt=agent_config.agent.llm.system_prompt,
+            system_prompt=system_prompt,
             response_format=response_format,
-            name=agent_config.agent.card.name
+            name=name
         )
 
     async def __call__(self, message: str, context_id: str = None) -> ResponseT:

distributed_a2a/executors.py

@@ -7,7 +7,6 @@ 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
@@ -27,6 +26,7 @@ class RoutingAgentExecutor(AgentExecutor):
         super().__init__()
         api_key = os.environ.get(agent_config.agent.llm.api_key_env)
         self.agent = StatusAgent[StringResponse](
+            llm_config=agent_config.agent.llm,
             system_prompt=agent_config.agent.llm.system_prompt,
             name="Router",
             api_key=api_key,
@@ -34,6 +34,7 @@ class RoutingAgentExecutor(AgentExecutor):
             tools=[] if tools is None else tools,
         )
         self.routing_agent = StatusAgent[RoutingResponse](
+            llm_config=agent_config.agent.llm,
             system_prompt=ROUTING_SYSTEM_PROMPT,
             name="Router",
             api_key=api_key,

distributed_a2a/server.py

@@ -1,5 +1,4 @@
 import asyncio
-import logging
 import time
 from contextlib import asynccontextmanager
 from typing import Any
@@ -41,9 +40,8 @@ async def heart_beat(name: str, agent_card_table: str, agent_card: AgentCard):
 
 
 def load_app(agent_config: dict[str, Any]) -> FastAPI:
-    logging.info("Loading agent app")
-    logging.info(f"config dictionary: {agent_config}")
-    agent_config=AgentConfig.model_validate(agent_config)
+
+    agent_config= AgentConfig.model_validate(agent_config)
 
     skills = [AgentSkill(
         id=skill.id,
@@ -67,6 +65,7 @@ def load_app(agent_config: dict[str, Any]) -> FastAPI:
         default_output_modes=agent_config.agent.card.default_output_modes,
         skills=skills,
         preferred_transport=agent_config.agent.card.preferred_transport_protocol,
+        capabilities=CAPABILITIES
     )
 
 
@@ -84,5 +83,5 @@ def load_app(agent_config: dict[str, Any]) -> FastAPI:
         http_handler=DefaultRequestHandler(
             agent_executor=executor,
             task_store=InMemoryTaskStore() #TODO replace with dynamodb store
-        )
-    ).build(title=agent_card.name, lifespan=lifespan)
+
+        )).build(title=agent_card.name, lifespan=lifespan, root_path=agent_config.agent.card.name) #TODO use extra parameter

{distributed_a2a-0.1.5rc5.dist-info → distributed_a2a-0.1.5rc7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.5rc5
+Version: 0.1.5rc7
 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.5rc7.dist-info/RECORD

@@ -0,0 +1,12 @@
+distributed_a2a/__init__.py,sha256=1q_7gRfBCGe-7sF_9YKzevyS97XQhtuFnZiYHIqaU-0,120
+distributed_a2a/agent.py,sha256=CpIeEASjkIflGCiVo4f5c5_xUg3VwMr27l-o4cj73Mw,2597
+distributed_a2a/client.py,sha256=2974Uw8YUuyBytwxxJJKYsWXCpEaIbGmMUHDraITxJ0,4149
+distributed_a2a/executors.py,sha256=m8Z48r4gWYE_39wq-lS4Btk5lOsKTuS7JSYNYF2Ejaw,4171
+distributed_a2a/model.py,sha256=yXiqEBZfQh-B_ld5klUVSUuXxninYfZ1aC9IJmmJdMg,2323
+distributed_a2a/registry.py,sha256=197eZVR6TW0isOUE0VjWSWs7aCqbWqnBzcV8EVXAxrI,677
+distributed_a2a/server.py,sha256=stBQ926uT8Ye8KeBYR3cFIM5vqFOFa4Ozceai5aIWP0,3079
+distributed_a2a-0.1.5rc7.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
+distributed_a2a-0.1.5rc7.dist-info/METADATA,sha256=HiG7eJxznlt7ygJzCjOdj8joWP4nsQBGtvIIOWJYARQ,3106
+distributed_a2a-0.1.5rc7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+distributed_a2a-0.1.5rc7.dist-info/top_level.txt,sha256=23qJ8n5k7796BHDK7a58uuO-X4GV0EgUWcGi8NIn-0k,16
+distributed_a2a-0.1.5rc7.dist-info/RECORD,,

distributed_a2a-0.1.5rc5.dist-info/RECORD

@@ -1,12 +0,0 @@
-distributed_a2a/__init__.py,sha256=1q_7gRfBCGe-7sF_9YKzevyS97XQhtuFnZiYHIqaU-0,120
-distributed_a2a/agent.py,sha256=fW3htAZkZucuuowvROO0jcscpUMOsH-5loxBZOf6G4U,2642
-distributed_a2a/client.py,sha256=2974Uw8YUuyBytwxxJJKYsWXCpEaIbGmMUHDraITxJ0,4149
-distributed_a2a/executors.py,sha256=PMBAxeE3-ZY2YNiRbgzdksxrsJb4Q10kHzGjt020TEo,4104
-distributed_a2a/model.py,sha256=yXiqEBZfQh-B_ld5klUVSUuXxninYfZ1aC9IJmmJdMg,2323
-distributed_a2a/registry.py,sha256=197eZVR6TW0isOUE0VjWSWs7aCqbWqnBzcV8EVXAxrI,677
-distributed_a2a/server.py,sha256=65k3_p1cS77E-_wVOD1eBb5xjRMtDQvAPthNo2BZTro,3089
-distributed_a2a-0.1.5rc5.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
-distributed_a2a-0.1.5rc5.dist-info/METADATA,sha256=r9PU1E1b7kHMnXJBOiDYs4hCd0P3C7OfuPM4Z-qvNx0,3106
-distributed_a2a-0.1.5rc5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-distributed_a2a-0.1.5rc5.dist-info/top_level.txt,sha256=23qJ8n5k7796BHDK7a58uuO-X4GV0EgUWcGi8NIn-0k,16
-distributed_a2a-0.1.5rc5.dist-info/RECORD,,