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

smarta2a/server.py

@@ -48,6 +48,11 @@ from .types import (
     A2AStatus,
     A2AStreamResponse,
     TaskSendParams,
+    SetTaskPushNotificationRequest,
+    GetTaskPushNotificationRequest,
+    SetTaskPushNotificationResponse,
+    GetTaskPushNotificationResponse,
+    TaskPushNotificationConfig,
 )
 
 class SmartA2A:
@@ -129,6 +134,18 @@ class SmartA2A:
             self._register_handler("tasks/cancel", func, "task_cancel", "handler")
             return func
         return decorator
+    
+    def set_notification(self):
+        def decorator(func: Callable[[SetTaskPushNotificationRequest], None]) -> Callable:
+            self._register_handler("tasks/pushNotification/set", func, "set_notification", "handler")
+            return func
+        return decorator
+    
+    def get_notification(self):
+        def decorator(func: Callable[[GetTaskPushNotificationRequest], Union[TaskPushNotificationConfig, GetTaskPushNotificationResponse]]):
+            self._register_handler("tasks/pushNotification/get", func, "get_notification", "handler")
+            return func
+        return decorator
 
     async def process_request(self, request_data: dict) -> JSONRPCResponse:
         try:
@@ -141,6 +158,10 @@ class SmartA2A:
                 return self._handle_get_task(request_data)
             elif method == "tasks/cancel":
                 return self._handle_cancel_task(request_data)
+            elif method == "tasks/pushNotification/set":
+                return self._handle_set_notification(request_data)
+            elif method == "tasks/pushNotification/get":
+                return self._handle_get_notification(request_data)
             else:
                 return self._error_response(
                     request_data.get("id"),
@@ -422,6 +443,102 @@ class SmartA2A:
                 error=InternalError(data=str(e))
             )
 
+    def _handle_set_notification(self, request_data: dict) -> SetTaskPushNotificationResponse:
+        try:
+            request = SetTaskPushNotificationRequest.model_validate(request_data)
+            handler = self.handlers.get("tasks/pushNotification/set")
+            
+            if not handler:
+                return SetTaskPushNotificationResponse(
+                    id=request.id,
+                    error=MethodNotFoundError()
+                )
+
+            try:
+                # Execute handler (may or may not return something)
+                raw_result = handler(request)
+                
+                # If handler returns nothing - build success response from request params
+                if raw_result is None:
+                    return SetTaskPushNotificationResponse(
+                        id=request.id,
+                        result=request.params
+                    )
+                
+                # If handler returns a full response object
+                if isinstance(raw_result, SetTaskPushNotificationResponse):
+                    return raw_result
+                    
+
+            except Exception as e:
+                if isinstance(e, JSONRPCError):
+                    return SetTaskPushNotificationResponse(
+                        id=request.id,
+                        error=e
+                    )
+                return SetTaskPushNotificationResponse(
+                    id=request.id,
+                    error=InternalError(data=str(e))
+                )
+
+        except ValidationError as e:
+            return SetTaskPushNotificationResponse(
+                id=request_data.get("id"),
+                error=InvalidRequestError(data=e.errors())
+            )
+                      
+
+    def _handle_get_notification(self, request_data: dict) -> GetTaskPushNotificationResponse:
+        try:
+            request = GetTaskPushNotificationRequest.model_validate(request_data)
+            handler = self.handlers.get("tasks/pushNotification/get")
+            
+            if not handler:
+                return GetTaskPushNotificationResponse(
+                    id=request.id,
+                    error=MethodNotFoundError()
+                )
+            
+            try:
+                raw_result = handler(request)
+                
+                if isinstance(raw_result, GetTaskPushNotificationResponse):
+                    return raw_result
+                else:
+                    # Validate raw_result as TaskPushNotificationConfig
+                    config = TaskPushNotificationConfig.model_validate(raw_result)
+                    return GetTaskPushNotificationResponse(
+                        id=request.id,
+                        result=config
+                    )
+            except ValidationError as e:
+                return GetTaskPushNotificationResponse(
+                    id=request.id,
+                    error=InvalidParamsError(data=e.errors())
+                )
+            except Exception as e:
+                if isinstance(e, JSONRPCError):
+                    return GetTaskPushNotificationResponse(
+                        id=request.id,
+                        error=e
+                    )
+                return GetTaskPushNotificationResponse(
+                    id=request.id,
+                    error=InternalError(data=str(e))
+                )
+
+        except ValidationError as e:
+            return GetTaskPushNotificationResponse(
+                id=request_data.get("id"),
+                error=InvalidRequestError(data=e.errors())
+            )
+        except json.JSONDecodeError as e:
+            return GetTaskPushNotificationResponse(
+                id=request_data.get("id"),
+                error=JSONParseError(data=str(e))
+            )
+
+
     def _normalize_artifacts(self, content: Any) -> List[Artifact]:
         """Handle both A2AResponse content and regular returns"""
         if isinstance(content, Artifact):

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: smarta2a
-Version: 0.2.0
+Version: 0.2.1
 Summary: A Python package for creating servers and clients following Google's Agent2Agent protocol
 Project-URL: Homepage, https://github.com/siddharthsma/smarta2a
 Project-URL: Bug Tracker, https://github.com/siddharthsma/smarta2a/issues

smarta2a-0.2.1.dist-info/RECORD

@@ -0,0 +1,7 @@
+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,,

smarta2a-0.2.0.dist-info/RECORD

@@ -1,7 +0,0 @@
-smarta2a/__init__.py,sha256=f_RqeaAHiBXO0O8n2kR8EBLGM3ezNwmR-CV-xHeOHLM,182
-smarta2a/server.py,sha256=a5RTtROydJwxHK1nTYzv2nmRJvxhY-SsxuKDqE9_RwY,26117
-smarta2a/types.py,sha256=_UuFtOsnHIIqfQ2m_FiIBBp141iYmhpPGgxE0jmHSHg,10807
-smarta2a-0.2.0.dist-info/METADATA,sha256=4PZH8mK_4YNGnfAH4gbjPwZW0M9lG2yZYdJW8Dq6T3w,2478
-smarta2a-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-smarta2a-0.2.0.dist-info/licenses/LICENSE,sha256=ECMEVHuFkvpEmH-_A9HSxs_UnnsUqpCkiAYNHPCf2z0,1078
-smarta2a-0.2.0.dist-info/RECORD,,