distributed-a2a 0.1.5rc2__tar.gz → 0.1.5rc4__tar.gz

{distributed_a2a-0.1.5rc2/distributed_a2a.egg-info → distributed_a2a-0.1.5rc4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.5rc2
+Version: 0.1.5rc4
 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.5rc4/distributed_a2a/model.py

@@ -0,0 +1,50 @@
+from typing import List
+
+from langchain_core.language_models import BaseChatModel
+from langchain_openai import ChatOpenAI
+from pydantic import BaseModel, Field
+
+
+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")
+
+
+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")
+    system_prompt: str = Field(description="The system prompt to use for the LLM")
+
+
+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=[])
+
+
+class AgentItem(BaseModel):
+    card: CardConfig = Field(description="The agent card configuration node")
+    llm: LLMConfig = Field(description="The LLM configuration node")
+
+
+class AgentConfig(BaseModel):
+    agent: AgentItem = Field(description="The agent configuration node")
+
+
+
+def get_model(api_key: str, model: str, base_url: str, reasoning_effort: str) -> BaseChatModel:
+    return ChatOpenAI(
+        model=model,
+        base_url=base_url,
+        api_key=lambda: api_key,
+        reasoning_effort=reasoning_effort
+    )

{distributed_a2a-0.1.5rc2 → distributed_a2a-0.1.5rc4/distributed_a2a.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed_a2a
-Version: 0.1.5rc2
+Version: 0.1.5rc4
 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.5rc2 → distributed_a2a-0.1.5rc4}/pyproject.toml

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

distributed_a2a-0.1.5rc2/distributed_a2a/model.py

@@ -1,50 +0,0 @@
-from typing import List
-
-from langchain_core.language_models import BaseChatModel
-from langchain_openai import ChatOpenAI
-from pydantic import BaseModel
-
-
-class SkillConfig(BaseModel):
-    id: str
-    name: str
-    description: str
-    tags: List[str]
-
-
-class LLMConfig(BaseModel):
-    base_url: str
-    model: str
-    api_key_env: str
-    reasoning_effort: str
-    system_prompt: str
-
-
-class CardConfig(BaseModel):
-    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
-    llm: LLMConfig
-
-
-class AgentConfig(BaseModel):
-    agent: AgentItem
-
-
-
-def get_model(api_key: str, model: str, base_url: str, reasoning_effort: str) -> BaseChatModel:
-    return ChatOpenAI(
-        model=model,
-        base_url=base_url,
-        api_key=lambda: api_key,
-        reasoning_effort=reasoning_effort
-    )