smarta2a 0.2.1__py3-none-any.whl → 0.2.3__py3-none-any.whl

smarta2a/{types.py → utils/types.py}

@@ -1,4 +1,4 @@
-from typing import Union, Any
+from typing import Union, Any, Dict
 from pydantic import BaseModel, Field, TypeAdapter
 from typing import Literal, List, Annotated, Optional
 from datetime import datetime
@@ -57,7 +57,7 @@ Part = Annotated[Union[TextPart, FilePart, DataPart], Field(discriminator="type"
 
 
 class Message(BaseModel):
-    role: Literal["user", "agent"]
+    role: Literal["user", "agent", "system"] # Added system role for system messages
     parts: List[Part]
     metadata: dict[str, Any] | None = None
 
@@ -357,6 +357,59 @@ class AgentCard(BaseModel):
     defaultOutputModes: List[str] = ["text"]
     skills: List[AgentSkill]
 
+    def pretty_print(self, include_separators: bool = False) -> str:
+        """Returns formatted string, optionally wrapped in separators"""
+        output = []
+        output.append(f"Name: {self.name}")
+        
+        if self.description:
+            output.append(f"Description: {self.description}")
+            
+        output.append(f"URL: {self.url}")
+        
+        if self.provider:
+            output.append(f"Provider Organization: {self.provider.organization}")
+
+        # Capabilities handling
+        capabilities = []
+        if self.capabilities.streaming:
+            capabilities.append("Streaming")
+        if self.capabilities.pushNotifications:
+            capabilities.append("Push Notifications")
+        if self.capabilities.stateTransitionHistory:
+            capabilities.append("State Transition History")
+        output.append("Capabilities: " + ", ".join(capabilities))
+        
+        # Skills handling
+        skills_output = ["Skills:"]
+        for skill in self.skills:
+            skills_output.append(f"  {skill.name} [{skill.id}]")
+            
+            if skill.description:
+                skills_output.append(f"    Description: {skill.description}")
+                
+            if skill.tags:
+                skills_output.append(f"    Tags: {', '.join(skill.tags)}")
+                
+            if skill.examples:
+                skills_output.append("    Examples:")
+                skills_output.extend([f"      - {ex}" for ex in skill.examples])
+                
+            if skill.inputModes:
+                skills_output.append(f"    Input Modes: {', '.join(skill.inputModes)}")
+                
+            if skill.outputModes:
+                skills_output.append(f"    Output Modes: {', '.join(skill.outputModes)}")
+                
+            skills_output.append("")
+
+        output.extend(skills_output)
+        result = "\n".join(output).strip()
+        
+        if include_separators:
+            return f"---\n{result}\n---"
+        return result
+
 
 class A2AClientError(Exception):
     pass
@@ -388,6 +441,8 @@ They are used to help with the implementation of the server.
 class A2AResponse(BaseModel):
     status: Union[TaskStatus, str]
     content: Union[str, List[Any], Part, Artifact, List[Part], List[Artifact]]
+    sessionId: Optional[str] = None           
+    metadata: Optional[dict[str, Any]] = None
     
     @model_validator(mode="after")
     def validate_state(self) -> 'A2AResponse':
@@ -422,3 +477,8 @@ class A2AStreamResponse(BaseModel):
     append: bool = False
     final: bool = False
     metadata: dict[str, Any] | None = None
+
+class StateData(BaseModel):
+    sessionId: str
+    history: List[Message]
+    metadata: Dict[str, Any]

{smarta2a-0.2.1.dist-info → smarta2a-0.2.3.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: smarta2a
-Version: 0.2.1
-Summary: A Python package for creating servers and clients following Google's Agent2Agent protocol
+Version: 0.2.3
+Summary: A simple Python framework (built on top of FastAPI) for creating Agents following Google's Agent2Agent protocol
 Project-URL: Homepage, https://github.com/siddharthsma/smarta2a
 Project-URL: Bug Tracker, https://github.com/siddharthsma/smarta2a/issues
 Author-email: Siddharth Ambegaonkar <siddharthsma@gmail.com>
@@ -10,10 +10,16 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.8
-Requires-Dist: fastapi
-Requires-Dist: pydantic
-Requires-Dist: sse-starlette
-Requires-Dist: uvicorn
+Requires-Dist: anyio>=4.9.0
+Requires-Dist: fastapi>=0.115.12
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: mcp>=0.1.0
+Requires-Dist: openai>=1.0.0
+Requires-Dist: pydantic>=2.11.3
+Requires-Dist: sse-starlette>=2.2.1
+Requires-Dist: starlette>=0.46.2
+Requires-Dist: typing-extensions>=4.13.2
+Requires-Dist: uvicorn>=0.34.1
 Description-Content-Type: text/markdown
 
 # SmartA2A
@@ -45,7 +51,7 @@ pip install smarta2a
 ## Simple Echo Server Implementation
 
 ```python
-from smarta2a import SmartA2A
+from smarta2a.server import SmartA2A
 
 app = SmartA2A("EchoServer")
 

smarta2a-0.2.3.dist-info/RECORD

@@ -0,0 +1,32 @@
+smarta2a/__init__.py,sha256=T_EECYqWrxshix0FbgUv22zlKRX22HFU-HKXcYTOb3w,175
+smarta2a/agent/a2a_agent.py,sha256=LrqWNgZra_dSHGmLZubhFMjZCgZ1BYdroXGuFFzv8Rk,1183
+smarta2a/agent/a2a_mcp_server.py,sha256=X_mxkgYgCA_dSNtCvs0rSlOoWYc-8d3Qyxv0e-a7NKY,1015
+smarta2a/archive/mcp_client.py,sha256=Fj64Kw5HFXZ0SHImWTsxTHIeP-3DgvP1nRiY_Jdgm0Q,3305
+smarta2a/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+smarta2a/client/a2a_client.py,sha256=Tixof6e2EXrGbShSJls_S009i9VbO9QizyCDzbzSISE,9890
+smarta2a/client/smart_mcp_client.py,sha256=-P_qwY3lnvL2uF-4e4ZdL3U0hyqMgVrD_u2cfhugQpc,2413
+smarta2a/client/tools_manager.py,sha256=bTbVwxR3hYsbQ87ClaX1cnswJ34ZAEeZ0Xi8N5q47Q0,2247
+smarta2a/history_update_strategies/__init__.py,sha256=x5WtiE9rG5ze8d8hA6E6wJOciBhWHa_ZgGgoIAZcXEQ,213
+smarta2a/history_update_strategies/append_strategy.py,sha256=j7Qbhs69Wwr-HBLB8GJ3-mEPaBSHiBV2xz9ZZi86k2w,312
+smarta2a/history_update_strategies/history_update_strategy.py,sha256=n2sfIGu8ztKI7gJTwRX26m4tZr28B8Xdhrk6RlBFlI8,373
+smarta2a/model_providers/__init__.py,sha256=2FhblAUiwG9Xv27yEpuuz0VrnIZ-rlpgIuPwm-UIX5U,147
+smarta2a/model_providers/base_llm_provider.py,sha256=6QjTUjYEnvHZji4_VWZz6CvLYKLyutxRUfIeH3seQg4,424
+smarta2a/model_providers/openai_provider.py,sha256=bLNWlsrrSFTXxbKsTX_xXfJFuzfWmndKRKqCy0h6ZB8,10830
+smarta2a/server/__init__.py,sha256=f2X454Ll4vJc02V4JLJHTN-h8u0TBm4d_FkiO4t686U,53
+smarta2a/server/handler_registry.py,sha256=OVRG5dTvxB7qUNXgsqWxVNxIyRljUShSYxb1gtbi5XM,820
+smarta2a/server/server.py,sha256=nqMsMeGJEvkj2kaF9U7yRnYAtj07vPCkQOdQALmaCws,31782
+smarta2a/server/state_manager.py,sha256=Uc4BNr2kQvi7MAEh3CmHsKV2bP-Q8bYbGADQ35iHmZo,1350
+smarta2a/server/subscription_service.py,sha256=fWqNNY0xmRksc_SZl4xt5fOPtZTQacfzou1-RMyaEd4,5188
+smarta2a/server/task_service.py,sha256=TXVnFeS9ofAqH2z_7BOfk5uDmoZKv9irHHQSIuurI70,7650
+smarta2a/state_stores/__init__.py,sha256=vafxAqpwvag_cYFH2XKGk3DPmJIWJr4Ioey30yLFkVQ,220
+smarta2a/state_stores/base_state_store.py,sha256=LFI-LThPLf7M9z_CcXWCswajxMAtMx9tMFFVhZU0fM8,521
+smarta2a/state_stores/inmemory_state_store.py,sha256=MgFGc7HxccrBxEqhVqKJ3bV-RnV1koU6iJd5m3rhhjA,682
+smarta2a/utils/__init__.py,sha256=5db5VgDGgbMUGEF-xuyaC3qrgRQkUE9WAITkFSiNqSA,702
+smarta2a/utils/prompt_helpers.py,sha256=jLETieoeBJLQXcGzwFeoKT5b2pS4tJ5770lPLImtKLo,1439
+smarta2a/utils/task_builder.py,sha256=wqSyfVHNTaXuGESu09dhlaDi7D007gcN3-8tH-nPQ40,5159
+smarta2a/utils/task_request_builder.py,sha256=6cOGOqj2Rg43xWM03GRJQzlIZHBptsMCJRp7oD-TDAQ,3362
+smarta2a/utils/types.py,sha256=mtkn78SrXyzjnOw6OBcTHlnhGfHgxlhKxJsRAVlUaEQ,13078
+smarta2a-0.2.3.dist-info/METADATA,sha256=XJBiT_AOoaltJAppDSd0D0y-bNKTwqxOuQiymgDj_Vs,2726
+smarta2a-0.2.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+smarta2a-0.2.3.dist-info/licenses/LICENSE,sha256=ECMEVHuFkvpEmH-_A9HSxs_UnnsUqpCkiAYNHPCf2z0,1078
+smarta2a-0.2.3.dist-info/RECORD,,

smarta2a-0.2.1.dist-info/RECORD

@@ -1,7 +0,0 @@
-smarta2a/__init__.py,sha256=f_RqeaAHiBXO0O8n2kR8EBLGM3ezNwmR-CV-xHeOHLM,182
-smarta2a/server.py,sha256=lFUlAz4TEeEXCs2bgJmMTN7JpZTwYEKXyt4KWEQwS6U,30915
-smarta2a/types.py,sha256=_UuFtOsnHIIqfQ2m_FiIBBp141iYmhpPGgxE0jmHSHg,10807
-smarta2a-0.2.1.dist-info/METADATA,sha256=H4os9rBnBzByJI1FWr7vYW2w8Or8PInU7XtmwsHCIcY,2478
-smarta2a-0.2.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-smarta2a-0.2.1.dist-info/licenses/LICENSE,sha256=ECMEVHuFkvpEmH-_A9HSxs_UnnsUqpCkiAYNHPCf2z0,1078
-smarta2a-0.2.1.dist-info/RECORD,,