distributed-a2a 0.1.5rc16__py3-none-any.whl → 0.1.5rc18__py3-none-any.whl

distributed_a2a/agent.py

@@ -5,6 +5,7 @@ from langchain.agents import create_agent
 from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import BaseTool
 from langgraph.checkpoint.memory import MemorySaver
+from langgraph_dynamodb_checkpoint import DynamoDBSaver
 from pydantic import BaseModel, Field
 
 from .model import get_model, AgentConfig, LLMConfig
@@ -20,29 +21,39 @@ class AgentResponse(BaseModel):
         )
     )
 
+
 class RoutingResponse(AgentResponse):
     agent_card: str = Field(description="The stringified json of the agent card to be returned to the user")
 
+
 class StringResponse(AgentResponse):
     response: str = Field(description="The main response to be returned to the user")
 
 
 class StatusAgent[ResponseT: AgentResponse]:
 
-    def __init__(self, llm_config: LLMConfig, name: str, system_prompt: str, api_key: str, is_routing: bool, tools: list[BaseTool]):
+    def __init__(self, llm_config: LLMConfig, name: str, system_prompt: str, api_key: str, is_routing: bool,
+                 tools: list[BaseTool]):
         response_format: type[AgentResponse]
         if is_routing:
             response_format = RoutingResponse
         else:
             response_format = StringResponse
 
+
+        saver = DynamoDBSaver(
+            table_name="your-dynamodb-table-name",
+            max_read_request_units=20,  ## TODO find correct value for app
+            max_write_request_units=20,  ## TODO find correct value for app
+            ttl_seconds=86400
+        )
         self.agent = create_agent(
             get_model(api_key=api_key,
                       model=llm_config.model,
                       base_url=llm_config.base_url,
                       reasoning_effort=llm_config.reasoning_effort),
             tools=tools,
-            checkpointer=MemorySaver(),  # TODO replace by dynamodb
+            checkpointer=saver,
             system_prompt=system_prompt,
             response_format=response_format,
             name=name

distributed_a2a/executors.py

@@ -27,7 +27,7 @@ class RoutingAgentExecutor(AgentExecutor):
         api_key = os.environ.get(agent_config.agent.llm.api_key_env)
         self.agent = StatusAgent[StringResponse](
             llm_config=agent_config.agent.llm,
-            system_prompt=agent_config.agent.get_system_prompt(),
+            system_prompt=agent_config.agent.system_prompt,
             name=agent_config.agent.card.name,
             api_key=api_key,
             is_routing=False,

distributed_a2a/server.py

@@ -77,7 +77,6 @@ def load_app(agent_config: dict[str, Any]) -> FastAPI:
         asyncio.create_task(heart_beat(name=agent_card.name, agent_card_table=AGENT_CARD_TABLE, agent_card=agent_card))
         yield
 
-
     return A2ARESTFastAPIApplication(
         agent_card=agent_card,
         http_handler=DefaultRequestHandler(

distributed_a2a/tool_registry_test.py

@@ -0,0 +1,6 @@
+from distributed_a2a.distributed_a2a.registry import ToolRegistry
+
+if __name__ == '__main__':
+    reg = ToolRegistry(tools_table="mcp_tools")
+    res = reg.get_tool_by_name(tool_name="my_second_tool")
+    print(res)

{distributed_a2a-0.1.5rc16.dist-info → distributed_a2a-0.1.5rc18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.5rc16
+Version: 0.1.5rc18
 Summary: A library for building A2A agents with routing capabilities
 Home-page: https://github.com/Barra-Technologies/distributed-a2a
 Author: Fabian Bell
@@ -23,6 +23,7 @@ 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

distributed_a2a-0.1.5rc18.dist-info/RECORD

@@ -0,0 +1,13 @@
+distributed_a2a/__init__.py,sha256=1q_7gRfBCGe-7sF_9YKzevyS97XQhtuFnZiYHIqaU-0,120
+distributed_a2a/agent.py,sha256=PsbnTQLqP45AmLQb0uzbfPRMjdCZ2kARK0m3rDb77FM,2910
+distributed_a2a/client.py,sha256=2974Uw8YUuyBytwxxJJKYsWXCpEaIbGmMUHDraITxJ0,4149
+distributed_a2a/executors.py,sha256=TNHO3eEIlKWObqBblcx_XiP5KahcfaGK1YuajWmQ5rE,4187
+distributed_a2a/model.py,sha256=pvuoVg8QChYq21us49wZa7Pv-BlIU-mlfN2auVQvqdY,2676
+distributed_a2a/registry.py,sha256=197eZVR6TW0isOUE0VjWSWs7aCqbWqnBzcV8EVXAxrI,677
+distributed_a2a/server.py,sha256=iTowUWA3Kwa1eCpF7SdEu7jF9oxpi5Gj9j1On3eoArg,3084
+distributed_a2a/tool_registry_test.py,sha256=xqnFKzogI8c6K3HoXjCXCgT4IlZEPQj3hlKTIwdYNyE,215
+distributed_a2a-0.1.5rc18.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
+distributed_a2a-0.1.5rc18.dist-info/METADATA,sha256=6gkP288uHWYZ79b6h3x-k2fFY9vOzo-DaRVTEGSuLtE,3161
+distributed_a2a-0.1.5rc18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+distributed_a2a-0.1.5rc18.dist-info/top_level.txt,sha256=23qJ8n5k7796BHDK7a58uuO-X4GV0EgUWcGi8NIn-0k,16
+distributed_a2a-0.1.5rc18.dist-info/RECORD,,

distributed_a2a-0.1.5rc16.dist-info/RECORD

@@ -1,12 +0,0 @@
-distributed_a2a/__init__.py,sha256=1q_7gRfBCGe-7sF_9YKzevyS97XQhtuFnZiYHIqaU-0,120
-distributed_a2a/agent.py,sha256=CpIeEASjkIflGCiVo4f5c5_xUg3VwMr27l-o4cj73Mw,2597
-distributed_a2a/client.py,sha256=2974Uw8YUuyBytwxxJJKYsWXCpEaIbGmMUHDraITxJ0,4149
-distributed_a2a/executors.py,sha256=5Zp1fHERVAGFWZ3yYUCiAlLl2EO8-k5AhfTW_hqQ49I,4193
-distributed_a2a/model.py,sha256=pvuoVg8QChYq21us49wZa7Pv-BlIU-mlfN2auVQvqdY,2676
-distributed_a2a/registry.py,sha256=197eZVR6TW0isOUE0VjWSWs7aCqbWqnBzcV8EVXAxrI,677
-distributed_a2a/server.py,sha256=eVccqBos4nMQqe9feyglGlMmUYIGz2t5jmBbMVt-NGE,3085
-distributed_a2a-0.1.5rc16.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
-distributed_a2a-0.1.5rc16.dist-info/METADATA,sha256=KQmM0yOX12PWi_5gJjvQjnUN6dbohO_5O-zAdvYBqc4,3107
-distributed_a2a-0.1.5rc16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-distributed_a2a-0.1.5rc16.dist-info/top_level.txt,sha256=23qJ8n5k7796BHDK7a58uuO-X4GV0EgUWcGi8NIn-0k,16
-distributed_a2a-0.1.5rc16.dist-info/RECORD,,