distributed-a2a 0.1.5rc17__tar.gz → 0.1.5rc18__tar.gz

{distributed_a2a-0.1.5rc17/distributed_a2a.egg-info → distributed_a2a-0.1.5rc18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.5rc17
+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.5rc17 → distributed_a2a-0.1.5rc18}/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-0.1.5rc17 → distributed_a2a-0.1.5rc18}/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-0.1.5rc18/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.5rc17 → distributed_a2a-0.1.5rc18/distributed_a2a.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.5rc17
+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.5rc17 → distributed_a2a-0.1.5rc18}/distributed_a2a.egg-info/SOURCES.txt

@@ -11,6 +11,7 @@ distributed_a2a/executors.py
 distributed_a2a/model.py
 distributed_a2a/registry.py
 distributed_a2a/server.py
+distributed_a2a/tool_registry_test.py
 distributed_a2a.egg-info/PKG-INFO
 distributed_a2a.egg-info/SOURCES.txt
 distributed_a2a.egg-info/dependency_links.txt

{distributed_a2a-0.1.5rc17 → distributed_a2a-0.1.5rc18}/distributed_a2a.egg-info/requires.txt

@@ -2,6 +2,7 @@ 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

{distributed_a2a-0.1.5rc17 → distributed_a2a-0.1.5rc18}/pyproject.toml

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

{distributed_a2a-0.1.5rc17 → distributed_a2a-0.1.5rc18}/requirements.txt

@@ -2,8 +2,9 @@ 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
+twine>=6.2.0