distributed-a2a 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

distributed_a2a/__init__.py

@@ -1,5 +1,7 @@
+from .client import RoutingA2AClient
 from .server import load_app
 
 __all__ = [
-    "load_app"
+    "load_app",
+    "RoutingA2AClient"
 ]

distributed_a2a/client.py

@@ -0,0 +1,110 @@
+import json
+import time
+from uuid import uuid4
+
+import httpx
+from a2a.client import ClientConfig, ClientFactory, A2ACardResolver, ClientEvent
+from a2a.client import create_text_message_object
+from a2a.types import (
+    AgentCard,
+    Message, TaskQueryParams, Task, Artifact, Part, TextPart
+)
+from a2a.types import TaskState
+
+
+class RemoteAgentConnection:
+    """A class to hold the connections to the remote agents."""
+
+    def __init__(self, agent_card: AgentCard, client: httpx.AsyncClient):
+        client_config = ClientConfig(
+            httpx_client=client,
+            supported_transports=[agent_card.preferred_transport],
+            streaming=agent_card.capabilities.streaming,
+            polling=True
+        )
+        client_factory = ClientFactory(config=client_config)
+        self.agent_client = client_factory.create(agent_card)
+
+    async def _send_message_to_agent(self, message_request: Message) -> Task:
+
+        responses: list[ClientEvent] = []
+        async for response in self.agent_client.send_message(message_request):
+            responses.append(response)
+
+        task_response: Task | None = None
+        match responses:
+            case [(task, _)]:
+                task_response = task
+            case _:
+                raise Exception("Wrong response format")
+        return task_response
+
+    async def _get_task(self, task_id: str) -> Task:
+        query_params: TaskQueryParams = TaskQueryParams(id=task_id)
+        response: Task = await self.agent_client.get_task(query_params)
+        return response
+
+    async def send_message(self, message_to_send: str, context_id, task_id: None | str = None,
+                           count=0) -> str | AgentCard:
+        message: Message = create_text_message_object(content=message_to_send)
+        message.message_id = str(uuid4())
+        message.context_id = context_id
+
+        response: Task
+        if task_id is None:
+            response = await self._send_message_to_agent(message)
+        else:
+            response = await self._get_task(task_id)
+
+        task_state = response.status.state
+        if task_state == TaskState.working or task_state == TaskState.submitted:
+            if count < 20:
+                time.sleep(1)
+                return await self.send_message(message_to_send, context_id, response.id, count + 1)
+            else:
+                raise Exception("Timeout waiting for agent to respond")
+
+        if task_state == TaskState.failed:
+            raise Exception("A2ATaskFailed")
+        elif task_state == TaskState.auth_required:
+            raise Exception("A2ATaskAuthRequired")
+
+        match response.artifacts:
+            case [Artifact(name='target_agent', parts=[Part(root=TextPart(text=agent_card))])]:
+                return AgentCard(**json.loads(agent_card))
+            case [Artifact(name='current_result', parts=[Part(root=TextPart(text=result))])]:
+                return result
+            case _:
+                raise Exception("Wrong response format")
+
+
+MAX_RECURSION_DEPTH = 10
+
+
+class RoutingA2AClient:
+    def __init__(self, initial_url: str):
+        self.url = initial_url
+        self.client = httpx.AsyncClient()
+        self.current_card: AgentCard | None = None
+
+    async def fetch_current_card(self):
+        card_resolver = A2ACardResolver(
+            self.client, self.url
+        )
+        self.current_card = (
+            await card_resolver.get_agent_card()
+        )
+
+    async def send_message(self, message: str, context_id: str, depth: int = 0) -> str:
+        if depth > MAX_RECURSION_DEPTH:
+            raise Exception("Maximum recursion depth exceeded. This is likely due to an infinite loop in your agent.")
+        if self.current_card is None:
+            await self.fetch_current_card()
+
+        agent_connection = RemoteAgentConnection(self.current_card, self.client)
+
+        agent_response: str | AgentCard = await agent_connection.send_message(message, context_id)
+        if isinstance(agent_response, AgentCard):
+            self.current_card = agent_response
+            return await self.send_message(message, context_id, depth + 1)
+        return agent_response

{distributed_a2a-0.1.1.dist-info → distributed_a2a-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.1
+Version: 0.1.3
 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.1.dist-info → distributed_a2a-0.1.3.dist-info}/RECORD

@@ -1,11 +1,12 @@
-distributed_a2a/__init__.py,sha256=fff-M86h-bxXMrtmNoHhL7mY-5vTx9ED1VJ9s2HVoXE,59
+distributed_a2a/__init__.py,sha256=1q_7gRfBCGe-7sF_9YKzevyS97XQhtuFnZiYHIqaU-0,120
 distributed_a2a/agent.py,sha256=dv4EBN70E4PoW9XxErwUEievDB7X-gyz8bmbG0Z-05U,2376
+distributed_a2a/client.py,sha256=2974Uw8YUuyBytwxxJJKYsWXCpEaIbGmMUHDraITxJ0,4149
 distributed_a2a/executors.py,sha256=xXiJydbb4modTdn7XmDflJ4QZ6v_SBeiIrLlcNErc9M,3951
 distributed_a2a/model.py,sha256=I-e0V1cqxbRNi2LdSZW8XPkgqncy9zX_dOXAusuqPrQ,331
 distributed_a2a/registry.py,sha256=197eZVR6TW0isOUE0VjWSWs7aCqbWqnBzcV8EVXAxrI,677
 distributed_a2a/server.py,sha256=C83rt-T3_EGQbDfPmncSvpa9PYbJxweHc5tKFmTOnvI,2601
-distributed_a2a-0.1.1.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
-distributed_a2a-0.1.1.dist-info/METADATA,sha256=SlsxcSx3Rjn5PfclsQ_6LE1MNojGVAo41bG0MBbgCtY,3604
-distributed_a2a-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-distributed_a2a-0.1.1.dist-info/top_level.txt,sha256=23qJ8n5k7796BHDK7a58uuO-X4GV0EgUWcGi8NIn-0k,16
-distributed_a2a-0.1.1.dist-info/RECORD,,
+distributed_a2a-0.1.3.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
+distributed_a2a-0.1.3.dist-info/METADATA,sha256=D0xe5WxwE8nfTlR8ZDoIP_b2FEFlkJDcqpKe6apr09c,3604
+distributed_a2a-0.1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+distributed_a2a-0.1.3.dist-info/top_level.txt,sha256=23qJ8n5k7796BHDK7a58uuO-X4GV0EgUWcGi8NIn-0k,16
+distributed_a2a-0.1.3.dist-info/RECORD,,