distributed-a2a 0.1.9rc0__tar.gz → 0.1.9rc1__tar.gz

{distributed_a2a-0.1.9rc0/distributed_a2a.egg-info → distributed_a2a-0.1.9rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.9rc0
+Version: 0.1.9rc1
 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.9rc0 → distributed_a2a-0.1.9rc1}/distributed_a2a/model.py

@@ -1,5 +1,5 @@
 import os
-from typing import List, Any
+from typing import List, Any, Optional
 
 from langchain_core.language_models import BaseChatModel
 from langchain_openai import ChatOpenAI
@@ -13,6 +13,15 @@ class SkillConfig(BaseModel):
     tags: List[str] = Field(description="The tags associated with the skill")
 
 
+class RegistryItemConfig(BaseModel):
+    url: str = Field(description="The url of the registry")
+
+
+class RegistryConfig(BaseModel):
+    agent: RegistryItemConfig = Field(description="The agent registry configuration")
+    mcp: RegistryItemConfig = Field(description="The mcp registry configuration")
+
+
 class LLMConfig(BaseModel):
     base_url: str = Field(description="The base url of the LLM provider")
     model: str = Field(description="The model to use for the LLM e.g. gpt-3.5-turbo")
@@ -32,6 +41,7 @@ class CardConfig(BaseModel):
 
 
 class AgentItem(BaseModel):
+    registry: Optional[RegistryConfig] = Field(description="The registry configuration node", default=None)
     card: CardConfig = Field(description="The agent card configuration node")
     llm: LLMConfig = Field(description="The LLM configuration node")
     system_prompt: str = Field(description="The system prompt to use for the LLM or a path to a file containing the system prompt")
@@ -50,6 +60,7 @@ class AgentConfig(BaseModel):
 
 
 class RouterItem(BaseModel):
+    registry: Optional[RegistryConfig] = Field(description="The registry configuration node", default=None)
     card: CardConfig = Field(description="The router card configuration node")
     llm: LLMConfig = Field(description="The LLM configuration node")
 

{distributed_a2a-0.1.9rc0 → distributed_a2a-0.1.9rc1}/distributed_a2a/router.py

@@ -8,7 +8,7 @@ from fastapi import FastAPI
 
 from .executors import RoutingExecutor
 from .model import RouterConfig
-from .registry import DynamoDbRegistryLookup
+from .registry import DynamoDbRegistryLookup, AgentRegistryLookup
 from .server import CAPABILITIES, AGENT_CARD_TABLE
 
 
@@ -33,8 +33,7 @@ def load_router(router_dic: dict[str, Any]) -> FastAPI:
 
     executor = RoutingExecutor(
         router_config=router_config,
-        routing_tool=DynamoDbRegistryLookup(agent_card_tabel=AGENT_CARD_TABLE).as_tool()
-
+        routing_tool=AgentRegistryLookup(router_config.router.registry.agent.url).as_tool()
     )
 
     return A2ARESTFastAPIApplication(

{distributed_a2a-0.1.9rc0 → distributed_a2a-0.1.9rc1}/distributed_a2a/server.py

@@ -1,3 +1,4 @@
+import os
 import asyncio
 import time
 from contextlib import asynccontextmanager
@@ -13,7 +14,7 @@ from fastapi import FastAPI
 
 from .executors import RoutingAgentExecutor
 from .model import AgentConfig
-from .registry import DynamoDbRegistryLookup, registry_heart_beat
+from .registry import DynamoDbRegistryLookup, registry_heart_beat, AgentRegistryLookup
 
 CAPABILITIES = AgentCapabilities(streaming=False, push_notifications=False)
 
@@ -21,6 +22,7 @@ HEART_BEAT_INTERVAL_SEC = 5
 MAX_HEART_BEAT_MISSES = 3
 
 AGENT_CARD_TABLE = "agent-cards"
+AGENT_REGISTRY_URL = ""
 
 def get_expire_at() -> int:
     return int(time.time() + MAX_HEART_BEAT_MISSES * HEART_BEAT_INTERVAL_SEC)
@@ -69,20 +71,16 @@ def load_app(agent_config: Any) -> FastAPI:
         preferred_transport=agent_config.agent.card.preferred_transport_protocol,
         capabilities=CAPABILITIES
     )
-
-
+    agent_registry = AgentRegistryLookup(agent_config.agent.registry.agent.url)
     executor = RoutingAgentExecutor(agent_config=agent_config,
-                                    routing_tool=DynamoDbRegistryLookup(agent_card_tabel=AGENT_CARD_TABLE).as_tool())
+                                    routing_tool=agent_registry.as_tool())
 
     @asynccontextmanager
     async def lifespan(_: FastAPI) -> AsyncGenerator[None, Any]:
-        asyncio.create_task(heart_beat(name=agent_card.name, agent_card_table=AGENT_CARD_TABLE, agent_card=agent_card))
-        import os
-        registry_url = os.getenv("REGISTRY_URL")
-        if registry_url:
-            asyncio.create_task(registry_heart_beat(name=agent_card.name, registry_url=registry_url,
-                                                   agent_card=agent_card, interval_sec=HEART_BEAT_INTERVAL_SEC,
-                                                   get_expire_at=get_expire_at))
+        asyncio.create_task(registry_heart_beat(name=agent_card.name, registry_url=agent_config.agent.registry.agent.url,
+                                                agent_card=agent_card, interval_sec=HEART_BEAT_INTERVAL_SEC,
+                                                get_expire_at=get_expire_at))
+
         yield
 
 

{distributed_a2a-0.1.9rc0 → distributed_a2a-0.1.9rc1/distributed_a2a.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.9rc0
+Version: 0.1.9rc1
 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.9rc0 → distributed_a2a-0.1.9rc1}/pyproject.toml

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