distributed-a2a 0.1.8__tar.gz → 0.1.9rc1__tar.gz

{distributed_a2a-0.1.8/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.8
+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.8 → 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.9rc1/distributed_a2a/registry.py

@@ -0,0 +1,55 @@
+import json
+import asyncio
+import boto3
+import requests
+from langchain_core.tools import StructuredTool
+from a2a.types import AgentCard
+
+
+async def registry_heart_beat(name: str, registry_url: str, agent_card: AgentCard, interval_sec: int,
+                              get_expire_at: callable) -> None:
+    registry = AgentRegistryLookup(registry_url=registry_url)
+    while True:
+        try:
+            registry.put_agent_card(name=name, agent_card=agent_card.model_dump(), expire_at=get_expire_at())
+        except Exception as e:
+            print(f"Failed to send heart beat to registry: {e}")
+        await asyncio.sleep(interval_sec)
+
+
+class AgentRegistryLookup:
+    def __init__(self, registry_url: str):
+        self.registry_url = registry_url
+
+    def get_agent_cards(self) -> list[dict]:
+        response = requests.get(url=f"{self.registry_url}/agent-cards", timeout=30)
+        response.raise_for_status()
+        return response.json()
+
+    def put_agent_card(self, name: str, agent_card: dict, expire_at: int) -> None:
+        response = requests.put(
+            url=f"{self.registry_url}/agent-card/{name}",
+            params={"expire_at": str(expire_at)},
+            json=agent_card,
+            timeout=30
+        )
+        response.raise_for_status()
+
+    def as_tool(self) -> StructuredTool:
+        return StructuredTool.from_function(func=lambda: self.get_agent_cards(), name="agent_card_lookup",
+                                            description="Gets all available agent cards")
+
+class DynamoDbRegistryLookup:
+    def __init__(self, agent_card_tabel: str):
+        dynamo = boto3.resource("dynamodb", region_name="eu-central-1")
+        self.table = dynamo.Table(agent_card_tabel)
+
+    def get_agent_cards(self) -> list[dict]:
+
+        items = self.table.scan().get("Items", [])
+        cards: list[dict] = [json.loads(it["card"]) for it in items]
+        return cards
+
+    def as_tool(self) -> StructuredTool:
+        return StructuredTool.from_function(func=lambda: self.get_agent_cards(), name="agent_card_lookup",
+                                            description="Gets all available agent cards")

{distributed_a2a-0.1.8 → 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.8 → 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
+from .registry import DynamoDbRegistryLookup, registry_heart_beat, AgentRegistryLookup
 
 CAPABILITIES = AgentCapabilities(streaming=False, push_notifications=False)
 
@@ -21,10 +22,13 @@ 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)
 
+
+
 async def heart_beat(name: str, agent_card_table: str, agent_card: AgentCard) -> None:
     table = boto3.resource("dynamodb", region_name="eu-central-1").Table(agent_card_table)
     table.put_item(Item={"id": name, "card": agent_card.model_dump_json(), "expireAt": get_expire_at()})
@@ -67,14 +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))
+        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.8 → 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.8
+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.8 → distributed_a2a-0.1.9rc1}/pyproject.toml

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

distributed_a2a-0.1.8/distributed_a2a/registry.py

@@ -1,18 +0,0 @@
-import boto3
-from langchain_core.tools import StructuredTool
-
-
-class DynamoDbRegistryLookup:
-    def __init__(self, agent_card_tabel: str):
-        dynamo = boto3.resource("dynamodb", region_name="eu-central-1")
-        self.table = dynamo.Table(agent_card_tabel)
-
-    def get_agent_cards(self) -> list[str]:
-
-        items = self.table.scan().get("Items", [])
-        cards: list[str] = [it["card"] for it in items]
-        return cards
-
-    def as_tool(self) -> StructuredTool:
-        return StructuredTool.from_function(func=lambda: self.get_agent_cards(), name="agent_card_lookup",
-                                            description="Gets all available agent cards")