distributed-a2a 0.1.5__py3-none-any.whl → 0.1.5rc1__py3-none-any.whl

distributed_a2a/agent.py

@@ -4,10 +4,10 @@ from a2a.types import TaskState
 from langchain.agents import create_agent
 from langchain_core.runnables import RunnableConfig
 from langchain_core.tools import BaseTool
-from langgraph_dynamodb_checkpoint import DynamoDBSaver
+from langgraph.checkpoint.memory import MemorySaver
 from pydantic import BaseModel, Field
 
-from .model import get_model, AgentConfig, LLMConfig
+from .model import get_model, AgentConfig
 
 
 class AgentResponse(BaseModel):
@@ -20,42 +20,32 @@ 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, agent_config: AgentConfig, 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=f"checkpoint_saver_{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),
+                      model=agent_config.agent.llm.model,
+                      base_url=agent_config.agent.llm.base_url,
+                      reasoning_effort=agent_config.agent.llm.reasoning_effort),
             tools=tools,
-            checkpointer=saver,
-            system_prompt=system_prompt,
+            checkpointer=MemorySaver(),  # TODO replace by dynamodb
+            system_prompt=agent_config.agent.llm.system_prompt,
             response_format=response_format,
-            name=name
+            name=agent_config.agent.card.name
         )
 
     async def __call__(self, message: str, context_id: str = None) -> ResponseT:

distributed_a2a/executors.py

@@ -7,6 +7,7 @@ from a2a.server.events import EventQueue
 from a2a.types import TaskStatusUpdateEvent, TaskStatus, TaskState, TaskArtifactUpdateEvent, Artifact
 from a2a.utils import new_text_artifact
 from langchain_core.tools import BaseTool
+from openai import api_key
 
 from .agent import StatusAgent, RoutingResponse, StringResponse
 from .model import AgentConfig
@@ -26,15 +27,13 @@ class RoutingAgentExecutor(AgentExecutor):
         super().__init__()
         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.system_prompt,
-            name=agent_config.agent.card.name,
+            system_prompt=agent_config.agent.llm.system_prompt,
+            name="Router",
             api_key=api_key,
             is_routing=False,
             tools=[] if tools is None else tools,
         )
         self.routing_agent = StatusAgent[RoutingResponse](
-            llm_config=agent_config.agent.llm,
             system_prompt=ROUTING_SYSTEM_PROMPT,
             name="Router",
             api_key=api_key,

distributed_a2a/model.py

@@ -1,52 +1,43 @@
-import os
-from typing import List, Any
+from typing import List
 
 from langchain_core.language_models import BaseChatModel
 from langchain_openai import ChatOpenAI
-from pydantic import BaseModel, Field
+from pydantic import BaseModel
 
 
 class SkillConfig(BaseModel):
-    id: str = Field(description="The id of the skill e.g. weather")
-    name: str = Field(description="The name of the skill e.g. weather")
-    description: str = Field(description="A short description of the skill")
-    tags: List[str] = Field(description="The tags associated with the skill")
+    id: str
+    name: str
+    description: str
+    tags: List[str]
 
 
 class LLMConfig(BaseModel):
-    base_url: str = Field(description="The base url of the LLM provider")
-    model: str = Field(description="The model to use for the LLM e.g. gpt-3.5-turbo")
-    api_key_env: str = Field(description="The environment variable containing the api key for the LLM provider")
-    reasoning_effort: str  = Field(description="The reasoning effort to use for the LLM e.g. high", default="high")
+    base_url: str
+    model: str
+    api_key_env: str
+    reasoning_effort: str
+    system_prompt: str
 
 
 class CardConfig(BaseModel):
-    name: str = Field(description="The name of the agent" )
-    description: str = Field(description="A short description of the agent")
-    version: str = Field(description="The version of the agent")
-    default_input_modes: List[str]  = Field(description="The default input modes supported by the agent", default=["text","text/plaintext"])
-    default_output_modes: List[str] = Field(description="The default output modes supported by the agent", default=["text","text/plaintext"])
-    preferred_transport_protocol: str = Field(description="The preferred transport protocol for the agent", default="HTTP+JSON")
-    url: str = Field(description="The url of the agent")
-    skills: List[SkillConfig] = Field(description="The skills supported by the agent", default=[])
+    name: str
+    version: str
+    default_input_modes: List[str]
+    default_output_modes: List[str]
+    preferred_transport_protocol: str
+    url: str
+    description: str
+    skills: List[SkillConfig]
 
 
 class AgentItem(BaseModel):
-    card: CardConfig = Field(description="The agent card configuration node")
-    llm: LLMConfig = Field(description="The LLM configuration node")
-    system_prompt: str = Field(description="The system prompt to use for the LLM or a path to a file containing the system prompt")
-
-    def __init__(self, /, **data: Any) -> None:
-        prompt_or_path= data['system_prompt']
-        if os.path.exists(prompt_or_path):
-            with open(prompt_or_path, "r", encoding="utf-8") as f:
-                data['system_prompt'] =  f.read()
-
-        super().__init__(**data)
+    card: CardConfig
+    llm: LLMConfig
 
 
 class AgentConfig(BaseModel):
-    agent: AgentItem = Field(description="The agent configuration node")
+    agent: AgentItem
 
 
 

distributed_a2a/server.py

@@ -57,15 +57,14 @@ def load_app(agent_config: dict[str, Any]) -> FastAPI:
     ))
 
     agent_card = AgentCard(
-        name=agent_config.agent.card.name,
-        description=agent_config.agent.card.description,
-        url=agent_config.agent.card.url,
-        version=agent_config.agent.card.version,
-        default_input_modes=agent_config.agent.card.default_input_modes,
-        default_output_modes=agent_config.agent.card.default_output_modes,
+        name=agent_config.agent.name,
+        description=agent_config.agent.description,
+        url=agent_config.agent.url,
+        version=agent_config.agent.version,
+        default_input_modes=agent_config.agent.default_input_modes,
+        default_output_modes=agent_config.agent.default_output_modes,
         skills=skills,
-        preferred_transport=agent_config.agent.card.preferred_transport_protocol,
-        capabilities=CAPABILITIES
+        preferred_transport=agent_config.agent.preferred_transport_protocol,
     )
 
 
@@ -83,5 +82,5 @@ def load_app(agent_config: dict[str, Any]) -> FastAPI:
         http_handler=DefaultRequestHandler(
             agent_executor=executor,
             task_store=InMemoryTaskStore() #TODO replace with dynamodb store
-
-        )).build(title=agent_card.name, lifespan=lifespan, root_path=f"/{agent_config.agent.card.name}") #TODO use extra parameter
+        )
+    ).build(title=agent_card.name, lifespan=lifespan)

{distributed_a2a-0.1.5.dist-info → distributed_a2a-0.1.5rc1.dist-info}/METADATA

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

@@ -0,0 +1,12 @@
+distributed_a2a/__init__.py,sha256=1q_7gRfBCGe-7sF_9YKzevyS97XQhtuFnZiYHIqaU-0,120
+distributed_a2a/agent.py,sha256=fW3htAZkZucuuowvROO0jcscpUMOsH-5loxBZOf6G4U,2642
+distributed_a2a/client.py,sha256=2974Uw8YUuyBytwxxJJKYsWXCpEaIbGmMUHDraITxJ0,4149
+distributed_a2a/executors.py,sha256=PMBAxeE3-ZY2YNiRbgzdksxrsJb4Q10kHzGjt020TEo,4104
+distributed_a2a/model.py,sha256=uyAfvCuuuHoeWnDa2h7-0OqfRbry6VWdhQBbyr3vpro,994
+distributed_a2a/registry.py,sha256=197eZVR6TW0isOUE0VjWSWs7aCqbWqnBzcV8EVXAxrI,677
+distributed_a2a/server.py,sha256=T6KGK9y5j4oq5ewkAEJgVGe945BVNnmz0Fe4FFBR6n8,2948
+distributed_a2a-0.1.5rc1.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
+distributed_a2a-0.1.5rc1.dist-info/METADATA,sha256=uVUJcAP6EE4e1uo1q-jUmmI1OEiPtREhFO1XP8zBViU,3106
+distributed_a2a-0.1.5rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+distributed_a2a-0.1.5rc1.dist-info/top_level.txt,sha256=23qJ8n5k7796BHDK7a58uuO-X4GV0EgUWcGi8NIn-0k,16
+distributed_a2a-0.1.5rc1.dist-info/RECORD,,

distributed_a2a-0.1.5.dist-info/RECORD

@@ -1,12 +0,0 @@
-distributed_a2a/__init__.py,sha256=1q_7gRfBCGe-7sF_9YKzevyS97XQhtuFnZiYHIqaU-0,120
-distributed_a2a/agent.py,sha256=ZI6sFISU5nELEqzVXbZaf7osqVH095raxjmrFzipoDA,2858
-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=eVccqBos4nMQqe9feyglGlMmUYIGz2t5jmBbMVt-NGE,3085
-distributed_a2a-0.1.5.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
-distributed_a2a-0.1.5.dist-info/METADATA,sha256=YKbtMhInCY7o5Z0qRoMKKuQPpc-mwrkpGFH8qXglEag,3157
-distributed_a2a-0.1.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-distributed_a2a-0.1.5.dist-info/top_level.txt,sha256=23qJ8n5k7796BHDK7a58uuO-X4GV0EgUWcGi8NIn-0k,16
-distributed_a2a-0.1.5.dist-info/RECORD,,