distributed-a2a 0.1.8__py3-none-any.whl → 0.1.9rc0__py3-none-any.whl

distributed_a2a/registry.py

@@ -1,16 +1,53 @@
+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[str]:
+    def get_agent_cards(self) -> list[dict]:
 
         items = self.table.scan().get("Items", [])
-        cards: list[str] = [it["card"] for it in items]
+        cards: list[dict] = [json.loads(it["card"]) for it in items]
         return cards
 
     def as_tool(self) -> StructuredTool:

distributed_a2a/server.py

@@ -13,7 +13,7 @@ from fastapi import FastAPI
 
 from .executors import RoutingAgentExecutor
 from .model import AgentConfig
-from .registry import DynamoDbRegistryLookup
+from .registry import DynamoDbRegistryLookup, registry_heart_beat
 
 CAPABILITIES = AgentCapabilities(streaming=False, push_notifications=False)
 
@@ -25,6 +25,8 @@ AGENT_CARD_TABLE = "agent-cards"
 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()})
@@ -75,6 +77,12 @@ def load_app(agent_config: Any) -> FastAPI:
     @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))
         yield
 
 

{distributed_a2a-0.1.8.dist-info → distributed_a2a-0.1.9rc0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.8
+Version: 0.1.9rc0
 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.dist-info/RECORD

@@ -0,0 +1,13 @@
+distributed_a2a/__init__.py,sha256=gO8sTT20O6XvDRzj7B322O9v6HxV-_eIlJD1wMlPYoI,171
+distributed_a2a/agent.py,sha256=qY775N3Br2YMuaHLALU4t6MO47syuEoTtFLLPRr8TyA,2910
+distributed_a2a/client.py,sha256=3tEBqu3HKEQyOFk5vsO1YiuKP2pZx-n6SCzsJUYNcyc,4601
+distributed_a2a/executors.py,sha256=1toayqLUan0f1INyMo5v02L9uCHMq6hp1gf0jVdjX6I,7152
+distributed_a2a/model.py,sha256=K4ltuBHTm_r5I-viITDt4GrxntwcUj1sOjLenHYB79k,2981
+distributed_a2a/registry.py,sha256=hwG3f2eqny6eSNsPdJnxlf6jFA-K8flO3loJXGmc5bg,2144
+distributed_a2a/router.py,sha256=NV8aKraFNrI53PxJAGXI5NtG3-Oy7x5ofHfgqeqsnsI,1780
+distributed_a2a/server.py,sha256=IZjDBBLYXYDYPFqBTyMxvGW20qMVweWXGSTP5jtgJKg,3540
+distributed_a2a-0.1.9rc0.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
+distributed_a2a-0.1.9rc0.dist-info/METADATA,sha256=zbcnqeXDcB8H88TT4apM50DRH7ZBAu9Clbka4UhEz2Y,3198
+distributed_a2a-0.1.9rc0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+distributed_a2a-0.1.9rc0.dist-info/top_level.txt,sha256=23qJ8n5k7796BHDK7a58uuO-X4GV0EgUWcGi8NIn-0k,16
+distributed_a2a-0.1.9rc0.dist-info/RECORD,,

distributed_a2a-0.1.8.dist-info/RECORD

@@ -1,13 +0,0 @@
-distributed_a2a/__init__.py,sha256=gO8sTT20O6XvDRzj7B322O9v6HxV-_eIlJD1wMlPYoI,171
-distributed_a2a/agent.py,sha256=qY775N3Br2YMuaHLALU4t6MO47syuEoTtFLLPRr8TyA,2910
-distributed_a2a/client.py,sha256=3tEBqu3HKEQyOFk5vsO1YiuKP2pZx-n6SCzsJUYNcyc,4601
-distributed_a2a/executors.py,sha256=1toayqLUan0f1INyMo5v02L9uCHMq6hp1gf0jVdjX6I,7152
-distributed_a2a/model.py,sha256=K4ltuBHTm_r5I-viITDt4GrxntwcUj1sOjLenHYB79k,2981
-distributed_a2a/registry.py,sha256=197eZVR6TW0isOUE0VjWSWs7aCqbWqnBzcV8EVXAxrI,677
-distributed_a2a/router.py,sha256=NV8aKraFNrI53PxJAGXI5NtG3-Oy7x5ofHfgqeqsnsI,1780
-distributed_a2a/server.py,sha256=6DWLHc4ju5WK5IdmPteetn7V8dnCttAYEf5W8bryTpk,3131
-distributed_a2a-0.1.8.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
-distributed_a2a-0.1.8.dist-info/METADATA,sha256=MReZhC86KBh_gV5fvDka98UIgIsuVE0EkzPAZ2DIKZc,3195
-distributed_a2a-0.1.8.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-distributed_a2a-0.1.8.dist-info/top_level.txt,sha256=23qJ8n5k7796BHDK7a58uuO-X4GV0EgUWcGi8NIn-0k,16
-distributed_a2a-0.1.8.dist-info/RECORD,,