smarta2a 0.3.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

smarta2a/server/subscription_service.py

@@ -1,109 +0,0 @@
-# Library imports
-from typing import Optional, List, Dict, Any, AsyncGenerator, Union
-from datetime import datetime
-from collections import defaultdict
-from uuid import uuid4
-from fastapi.responses import StreamingResponse
-from sse_starlette.sse import EventSourceResponse
-
-# Local imports
-from smarta2a.server.handler_registry import HandlerRegistry
-from smarta2a.server.state_manager import StateManager
-from smarta2a.utils.types import (
-    Message, StateData, SendTaskStreamingRequest, SendTaskStreamingResponse,
-    TaskSendParams, A2AStatus, A2AStreamResponse, TaskStatusUpdateEvent,
-    TaskStatus, TaskState, TaskArtifactUpdateEvent, Artifact, TextPart,
-    FilePart, DataPart, FileContent, MethodNotFoundError, TaskNotFoundError,
-    InternalError
-)
-
-class SubscriptionService:
-    def __init__(self, registry: HandlerRegistry, state_mgr: StateManager):
-        self.registry = registry
-        self.state_mgr = state_mgr
-
-    async def subscribe(self, request: SendTaskStreamingRequest, state: Optional[StateData]) -> StreamingResponse:
-        handler = self.registry.get_subscription("tasks/sendSubscribe")
-        if not handler:
-            err = SendTaskStreamingResponse(jsonrpc="2.0", id=request.id, error=MethodNotFoundError()).model_dump_json()
-            return EventSourceResponse(err)
-
-        session_id = state.sessionId if state else request.params.sessionId or str(uuid4())
-        history = state.history.copy() if state else [request.params.message]
-        metadata = state.metadata.copy() if state else (request.params.metadata or {})
-
-        async def event_stream():
-            try:
-                events = handler(request, state) if state else handler(request)
-                async for ev in self._normalize(request.params, events, history.copy(), metadata.copy(), session_id):
-                    yield f"data: {ev}\n\n"
-            except Exception as e:
-                err = TaskNotFoundError() if 'not found' in str(e).lower() else InternalError(data=str(e))
-                msg = SendTaskStreamingResponse(jsonrpc="2.0", id=request.id, error=err).model_dump_json()
-                yield f"data: {msg}\n\n"
-
-        return StreamingResponse(event_stream(), media_type="text/event-stream; charset=utf-8")
-
-    async def _normalize(
-        self,
-        params: TaskSendParams,
-        events: AsyncGenerator,
-        history: List[Message],
-        metadata: Dict[str, Any],
-        session_id: str
-    ) -> AsyncGenerator[str, None]:
-        artifact_state = defaultdict(lambda: {"index": 0, "last_chunk": False})
-        async for item in events:
-            if isinstance(item, SendTaskStreamingResponse):
-                yield item.model_dump_json()
-                continue
-
-            if isinstance(item, A2AStatus):
-                te = TaskStatusUpdateEvent(
-                    id=params.id,
-                    status=TaskStatus(state=TaskState(item.status), timestamp=datetime.now()),
-                    final=item.final or (item.status.lower() == TaskState.COMPLETED),
-                    metadata=item.metadata
-                )
-                yield SendTaskStreamingResponse(jsonrpc="2.0", id=params.id, result=te).model_dump_json()
-                continue
-
-            content_item = item
-            if not isinstance(item, A2AStreamResponse):
-                content_item = A2AStreamResponse(content=item)
-
-            parts: List[Union[TextPart, FilePart, DataPart]] = []
-            cont = content_item.content
-            if isinstance(cont, str): parts.append(TextPart(text=cont))
-            elif isinstance(cont, bytes): parts.append(FilePart(file=FileContent(bytes=cont)))
-            elif isinstance(cont, (TextPart, FilePart, DataPart)): parts.append(cont)
-            elif isinstance(cont, Artifact): parts.extend(cont.parts)
-            elif isinstance(cont, list):
-                for elem in cont:
-                    if isinstance(elem, str): parts.append(TextPart(text=elem))
-                    elif isinstance(elem, (TextPart, FilePart, DataPart)): parts.append(elem)
-                    elif isinstance(elem, Artifact): parts.extend(elem.parts)
-
-            idx = content_item.index
-            state = artifact_state[idx]
-            evt = TaskArtifactUpdateEvent(
-                id=params.id,
-                artifact=Artifact(
-                    parts=parts,
-                    index=idx,
-                    append=content_item.append or (state["index"] == idx),
-                    lastChunk=content_item.final or state["last_chunk"],
-                    metadata=content_item.metadata
-                )
-            )
-            if content_item.final:
-                state["last_chunk"] = True
-            state["index"] += 1
-
-            agent_msg = Message(role="agent", parts=evt.artifact.parts, metadata=evt.artifact.metadata)
-            new_hist = self.state_mgr.strategy.update_history(history, [agent_msg])
-            metadata = {**metadata, **(evt.artifact.metadata or {})}
-            self.state_mgr.update(StateData(session_id, new_hist, metadata))
-            history = new_hist
-
-            yield SendTaskStreamingResponse(jsonrpc="2.0", id=params.id, result=evt).model_dump_json()

smarta2a-0.3.0.dist-info/METADATA

@@ -1,103 +0,0 @@
-Metadata-Version: 2.4
-Name: smarta2a
-Version: 0.3.0
-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>
-License-File: LICENSE
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python :: 3
-Requires-Python: >=3.8
-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
-
-A Python package for creating a server following Google's Agent2Agent protocol
-
-## Features
-
-✅ **Full A2A Protocol Compliance** - Implements all required endpoints and response formats
-
-⚡ **Decorator-Driven Development** - Rapid endpoint configuration with type safety
-
-🧩 **Automatic Protocol Conversion** - Simple returns become valid A2A responses 
-
-🔀 **Flexible Response Handling** - Support for Tasks, Artifacts, Streaming, and raw protocol types if needed!
-
-🛡️ **Built-in Validation** - Automatic Pydantic validation of A2A schemas  
-
-⚡ **Single File Setup** - Get compliant in <10 lines of code
-
-🌍 **Production Ready** - CORS, async support, and error handling included
-
-## Installation
-
-```bash
-pip install smarta2a
-```
-
-## Simple Echo Server Implementation
-
-```python
-from smarta2a.server import SmartA2A
-
-app = SmartA2A("EchoServer")
-
-@app.on_send_task()
-def handle_task(request):
-    """Echo the input text back as a completed task"""
-    input_text = request.content[0].text
-    return f"Echo: {input_text}"
-
-if __name__ == "__main__":
-    app.run()
-```
-
-Automatically contructs the response:
-
-```json
-{
-  "jsonrpc": "2.0",
-  "id": "test",
-  "result": {
-    "id": "echo-task",
-    "status": {"state": "completed"},
-    "artifacts": [{
-      "parts": [{"type": "text", "text": "Echo: Hello!"}]
-    }]
-  }
-}
-```
-
-## Development
-
-To set up the development environment:
-
-```bash
-# Clone the repository
-git clone https://github.com/siddharthsma/smarta2a.git
-cd smarta2a
-
-# Create and activate virtual environment
-python -m venv venv
-source venv/bin/activate  # On Windows: venv\Scripts\activate
-
-# Install development dependencies
-pip install -e ".[dev]"
-```
-
-## License
-
-This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details. 

smarta2a-0.3.0.dist-info/RECORD

@@ -1,40 +0,0 @@
-smarta2a/__init__.py,sha256=T_EECYqWrxshix0FbgUv22zlKRX22HFU-HKXcYTOb3w,175
-smarta2a/agent/a2a_agent.py,sha256=zk6ZZtq4HrnKMDuZKnQPeKploV1IGXSZ1WmVPdIZAeY,1586
-smarta2a/agent/a2a_mcp_server.py,sha256=X_mxkgYgCA_dSNtCvs0rSlOoWYc-8d3Qyxv0e-a7NKY,1015
-smarta2a/archive/smart_mcp_client.py,sha256=NjyMR8xvrhy0-iFamj2XQnKp5qzkkFya7EQY4Kzl5dw,2574
-smarta2a/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-smarta2a/client/a2a_client.py,sha256=Or2uF_ZK9WBvL-9QQ6HOTAuJy9yUL3Yyxtwqi8uGMMo,11040
-smarta2a/client/mcp_client.py,sha256=ANjEzQFXEhKGQtNC9AjLI_IFtjv5twCCQhcB0TrwUpc,3534
-smarta2a/client/tools_manager.py,sha256=TX4r0KIT9ZZa-P8OQ48ePEGzGJpCchLCr5t5o17Szww,2352
-smarta2a/examples/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-smarta2a/examples/echo_server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-smarta2a/examples/echo_server/curl.txt,sha256=MYBbAgrU3PeCsksrrBc0xgaTD5Dro7xiTsCUz8Ocvt8,247
-smarta2a/examples/echo_server/main.py,sha256=i_XLQbRAv6XxkJX0vLV3CxHNANG41yAxgP-V-b4Iysc,1261
-smarta2a/examples/openai_delegator_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-smarta2a/examples/openai_delegator_agent/main.py,sha256=8jXBJtzfgoCH45YDaXiTgvitVIDkqnVcX6lNvxQF0xQ,1220
-smarta2a/examples/openai_weather_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-smarta2a/examples/openai_weather_agent/main.py,sha256=o8QqLu9JC3zRXRPGWhYdd7mVl3ooys5UwkPCPv2JHmQ,796
-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=oaJ5Yi5Bh5NGOo_sncevzQkhwqzeakOxVBWibB802Nc,10846
-smarta2a/server/__init__.py,sha256=f2X454Ll4vJc02V4JLJHTN-h8u0TBm4d_FkiO4t686U,53
-smarta2a/server/handler_registry.py,sha256=OVRG5dTvxB7qUNXgsqWxVNxIyRljUShSYxb1gtbi5XM,820
-smarta2a/server/server.py,sha256=WJGQVOprF2Hh_036eW4GJrRr3fLC0S_B3dAfONCgZrs,32180
-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=d04fNubP4BMmN5XTMYahwRHwNX2qcyEeiLzzQbpZppA,13177
-smarta2a-0.3.0.dist-info/METADATA,sha256=T2O-Z72460TCVmCL_idl2H5Av_Tf7_uwY79zV0PtvoE,2726
-smarta2a-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-smarta2a-0.3.0.dist-info/licenses/LICENSE,sha256=ECMEVHuFkvpEmH-_A9HSxs_UnnsUqpCkiAYNHPCf2z0,1078
-smarta2a-0.3.0.dist-info/RECORD,,

smarta2a-0.3.0.dist-info/licenses/LICENSE

@@ -1,21 +0,0 @@
-MIT License
-
-Copyright (c) 2025 Siddharth Ambegaonkar
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.