distributed-a2a 0.1.6rc8__tar.gz → 0.1.9rc0__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.6rc8
+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
@@ -19,16 +19,17 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: langchain>=0.1.0
-Requires-Dist: langchain-core>=0.1.0
-Requires-Dist: langchain-openai>=0.0.5
-Requires-Dist: langgraph>=0.0.20
-Requires-Dist: langgraph-dynamodb-checkpoint>=0.2.6.4
-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
+Requires-Dist: langchain==1.2.3
+Requires-Dist: langchain-core==1.2.7
+Requires-Dist: langchain-openai==1.1.7
+Requires-Dist: langgraph==1.0.5
+Requires-Dist: langgraph-dynamodb-checkpoint==0.2.6.4
+Requires-Dist: pydantic==2.12.5
+Requires-Dist: boto3==1.42.25
+Requires-Dist: a2a-sdk==0.3.22
+Requires-Dist: a2a-types==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.6rc8 → distributed_a2a-0.1.9rc0}/distributed_a2a/agent.py

@@ -11,7 +11,7 @@ from .model import get_model, LLMConfig
 
 
 class AgentResponse(BaseModel):
-    status: Literal[TaskState.rejected, TaskState.completed, TaskState.rejected, TaskState.failed] = Field(
+    status: Literal[TaskState.rejected, TaskState.completed, TaskState.failed] = Field(
         description=(
             f'You should select status as {TaskState.rejected} for requests that fall outside your area of expertise.'
             f'You should select status as {TaskState.completed} if the request is fully addressed and no further input is needed. '

distributed_a2a-0.1.9rc0/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.6rc8 → distributed_a2a-0.1.9rc0}/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.6rc8 → distributed_a2a-0.1.9rc0/distributed_a2a.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.6rc8
+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
@@ -19,16 +19,17 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: langchain>=0.1.0
-Requires-Dist: langchain-core>=0.1.0
-Requires-Dist: langchain-openai>=0.0.5
-Requires-Dist: langgraph>=0.0.20
-Requires-Dist: langgraph-dynamodb-checkpoint>=0.2.6.4
-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
+Requires-Dist: langchain==1.2.3
+Requires-Dist: langchain-core==1.2.7
+Requires-Dist: langchain-openai==1.1.7
+Requires-Dist: langgraph==1.0.5
+Requires-Dist: langgraph-dynamodb-checkpoint==0.2.6.4
+Requires-Dist: pydantic==2.12.5
+Requires-Dist: boto3==1.42.25
+Requires-Dist: a2a-sdk==0.3.22
+Requires-Dist: a2a-types==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.9rc0/distributed_a2a.egg-info/requires.txt

@@ -0,0 +1,11 @@
+langchain==1.2.3
+langchain-core==1.2.7
+langchain-openai==1.1.7
+langgraph==1.0.5
+langgraph-dynamodb-checkpoint==0.2.6.4
+pydantic==2.12.5
+boto3==1.42.25
+a2a-sdk==0.3.22
+a2a-types==0.1.0
+build==1.4.0
+twine==6.2.0

{distributed_a2a-0.1.6rc8 → distributed_a2a-0.1.9rc0}/pyproject.toml

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

distributed_a2a-0.1.9rc0/requirements.txt

@@ -0,0 +1,11 @@
+langchain==1.2.3
+langchain-core==1.2.7
+langchain-openai==1.1.7
+langgraph==1.0.5
+langgraph-dynamodb-checkpoint==0.2.6.4
+pydantic==2.12.5
+boto3==1.42.25
+a2a-sdk==0.3.22
+a2a-types==0.1.0
+build==1.4.0
+twine==6.2.0

distributed_a2a-0.1.6rc8/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")

distributed_a2a-0.1.6rc8/distributed_a2a.egg-info/requires.txt

@@ -1,10 +0,0 @@
-langchain>=0.1.0
-langchain-core>=0.1.0
-langchain-openai>=0.0.5
-langgraph>=0.0.20
-langgraph-dynamodb-checkpoint>=0.2.6.4
-pydantic>=2.0.0
-boto3>=1.28.0
-a2a>=0.1.0
-build>=1.4.0
-twine>=6.2.0

distributed_a2a-0.1.6rc8/requirements.txt

@@ -1,10 +0,0 @@
-langchain>=0.1.0
-langchain-core>=0.1.0
-langchain-openai>=0.0.5
-langgraph>=0.0.20
-langgraph-dynamodb-checkpoint>=0.2.6.4
-pydantic>=2.0.0
-boto3>=1.28.0
-a2a>=0.1.0
-build>=1.4.0
-twine>=6.2.0