chore: 🤖 Auto-update A2A types from google-a2a/A2A (#126)

* chore: 🤖 Auto-update A2A types from a2aproject/A2A@628380e

Author: a2a-bot

src/a2a/types.py

@@ -464,9 +464,9 @@ class JSONRPCRequest(BaseModel):
     """
 
 
-class JSONRPCResult(BaseModel):
+class JSONRPCSuccessResponse(BaseModel):
     """
-    Represents a JSON-RPC 2.0 Result object.
+    Represents a JSON-RPC 2.0 Success Response object.
     """
 
     id: str | int | None = None
@@ -712,7 +712,7 @@ class TaskResubscriptionRequest(BaseModel):
     JSON-RPC request model for the 'tasks/resubscribe' method.
     """
 
-    id: str | int | None = None
+    id: str | int
     """
     An identifier established by the Client that MUST contain a String, Number.
     Numbers SHOULD NOT contain fractional parts.
@@ -821,7 +821,7 @@ class CancelTaskRequest(BaseModel):
     JSON-RPC request model for the 'tasks/cancel' method.
     """
 
-    id: str | int | None = None
+    id: str | int
     """
     An identifier established by the Client that MUST contain a String, Number.
     Numbers SHOULD NOT contain fractional parts.
@@ -864,7 +864,7 @@ class GetTaskPushNotificationConfigRequest(BaseModel):
     JSON-RPC request model for the 'tasks/pushNotificationConfig/get' method.
     """
 
-    id: str | int | None = None
+    id: str | int
     """
     An identifier established by the Client that MUST contain a String, Number.
     Numbers SHOULD NOT contain fractional parts.
@@ -910,7 +910,7 @@ class GetTaskRequest(BaseModel):
     JSON-RPC request model for the 'tasks/get' method.
     """
 
-    id: str | int | None = None
+    id: str | int
     """
     An identifier established by the Client that MUST contain a String, Number.
     Numbers SHOULD NOT contain fractional parts.
@@ -1017,7 +1017,7 @@ class SetTaskPushNotificationConfigRequest(BaseModel):
     JSON-RPC request model for the 'tasks/pushNotificationConfig/set' method.
     """
 
-    id: str | int | None = None
+    id: str | int
     """
     An identifier established by the Client that MUST contain a String, Number.
     Numbers SHOULD NOT contain fractional parts.
@@ -1086,9 +1086,7 @@ class Artifact(BaseModel):
 
 
 class GetTaskPushNotificationConfigResponse(
-    RootModel[
-        JSONRPCErrorResponse | GetTaskPushNotificationConfigSuccessResponse
-    ]
+    RootModel[JSONRPCErrorResponse | GetTaskPushNotificationConfigSuccessResponse]
 ):
     root: JSONRPCErrorResponse | GetTaskPushNotificationConfigSuccessResponse
     """
@@ -1195,7 +1193,7 @@ class SendMessageRequest(BaseModel):
     JSON-RPC request model for the 'message/send' method.
     """
 
-    id: str | int | None = None
+    id: str | int
     """
     An identifier established by the Client that MUST contain a String, Number.
     Numbers SHOULD NOT contain fractional parts.
@@ -1219,7 +1217,7 @@ class SendStreamingMessageRequest(BaseModel):
     JSON-RPC request model for the 'message/stream' method.
     """
 
-    id: str | int | None = None
+    id: str | int
     """
     An identifier established by the Client that MUST contain a String, Number.
     Numbers SHOULD NOT contain fractional parts.
@@ -1239,9 +1237,7 @@ class SendStreamingMessageRequest(BaseModel):
 
 
 class SetTaskPushNotificationConfigResponse(
-    RootModel[
-        JSONRPCErrorResponse | SetTaskPushNotificationConfigSuccessResponse
-    ]
+    RootModel[JSONRPCErrorResponse | SetTaskPushNotificationConfigSuccessResponse]
 ):
     root: JSONRPCErrorResponse | SetTaskPushNotificationConfigSuccessResponse
     """
@@ -1420,10 +1416,6 @@ class AgentCard(BaseModel):
     """
     The version of the agent - format is up to the provider.
     """
-    supportsAuthenticatedExtendedCard: bool | None = Field(default=None)
-    """
-    Optional field indicating there is an extended card available post authentication at the /agent/authenticatedExtendedCard endpoint.
-    """
 
 
 class Task(BaseModel):
@@ -1534,9 +1526,7 @@ class SendStreamingMessageSuccessResponse(BaseModel):
     """
 
 
-class CancelTaskResponse(
-    RootModel[JSONRPCErrorResponse | CancelTaskSuccessResponse]
-):
+class CancelTaskResponse(RootModel[JSONRPCErrorResponse | CancelTaskSuccessResponse]):
     root: JSONRPCErrorResponse | CancelTaskSuccessResponse
     """
     JSON-RPC response for the 'tasks/cancel' method.
@@ -1546,7 +1536,7 @@ class CancelTaskResponse(
 class GetTaskResponse(RootModel[JSONRPCErrorResponse | GetTaskSuccessResponse]):
     root: JSONRPCErrorResponse | GetTaskSuccessResponse
     """
-    JSON-RPC success response for the 'tasks/get' method.
+    JSON-RPC response for the 'tasks/get' method.
     """
 
 
@@ -1575,9 +1565,7 @@ class JSONRPCResponse(
     """
 
 
-class SendMessageResponse(
-    RootModel[JSONRPCErrorResponse | SendMessageSuccessResponse]
-):
+class SendMessageResponse(RootModel[JSONRPCErrorResponse | SendMessageSuccessResponse]):
     root: JSONRPCErrorResponse | SendMessageSuccessResponse
     """
     JSON-RPC response model for the 'message/send' method.

tests/test_types.py

@@ -913,6 +913,7 @@ def test_a2a_request_root_model() -> None:
         ),
     )
     set_push_notif_req_data: dict[str, Any] = {
+        'id': 1,
         'jsonrpc': '2.0',
         'method': 'tasks/pushNotificationConfig/set',
         'params': task_push_config.model_dump(),
@@ -932,6 +933,7 @@ def test_a2a_request_root_model() -> None:
     # GetTaskPushNotificationConfigRequest
     id_params = TaskIdParams(id='t2')
     get_push_notif_req_data: dict[str, Any] = {
+        'id': 1,
         'jsonrpc': '2.0',
         'method': 'tasks/pushNotificationConfig/get',
         'params': id_params.model_dump(),
@@ -973,6 +975,83 @@ def test_a2a_request_root_model() -> None:
         A2ARequest.model_validate(invalid_req_data)
 
 
+def test_a2a_request_root_model_id_validation() -> None:
+    # SendMessageRequest case
+    send_params = MessageSendParams(message=Message(**MINIMAL_MESSAGE_USER))
+    send_req_data: dict[str, Any] = {
+        'jsonrpc': '2.0',
+        'method': 'message/send',
+        'params': send_params.model_dump(),
+    }
+    with pytest.raises(ValidationError):
+        A2ARequest.model_validate(send_req_data)  # missing id
+
+    # SendStreamingMessageRequest case
+    send_subs_req_data: dict[str, Any] = {
+        'jsonrpc': '2.0',
+        'method': 'message/stream',
+        'params': send_params.model_dump(),
+    }
+    with pytest.raises(ValidationError):
+        A2ARequest.model_validate(send_subs_req_data)  # missing id
+
+    # GetTaskRequest case
+    get_params = TaskQueryParams(id='t2')
+    get_req_data: dict[str, Any] = {
+        'jsonrpc': '2.0',
+        'method': 'tasks/get',
+        'params': get_params.model_dump(),
+    }
+    with pytest.raises(ValidationError):
+        A2ARequest.model_validate(get_req_data)  # missing id
+
+    # CancelTaskRequest case
+    id_params = TaskIdParams(id='t2')
+    cancel_req_data: dict[str, Any] = {
+        'jsonrpc': '2.0',
+        'method': 'tasks/cancel',
+        'params': id_params.model_dump(),
+    }
+    with pytest.raises(ValidationError):
+        A2ARequest.model_validate(cancel_req_data)  # missing id
+
+    # SetTaskPushNotificationConfigRequest
+    task_push_config = TaskPushNotificationConfig(
+        taskId='t2',
+        pushNotificationConfig=PushNotificationConfig(
+            url='https://example.com', token='token'
+        ),
+    )
+    set_push_notif_req_data: dict[str, Any] = {
+        'jsonrpc': '2.0',
+        'method': 'tasks/pushNotificationConfig/set',
+        'params': task_push_config.model_dump(),
+        'taskId': 2,
+    }
+    with pytest.raises(ValidationError):
+        A2ARequest.model_validate(set_push_notif_req_data)  # missing id
+
+    # GetTaskPushNotificationConfigRequest
+    id_params = TaskIdParams(id='t2')
+    get_push_notif_req_data: dict[str, Any] = {
+        'jsonrpc': '2.0',
+        'method': 'tasks/pushNotificationConfig/get',
+        'params': id_params.model_dump(),
+        'taskId': 2,
+    }
+    with pytest.raises(ValidationError):
+        A2ARequest.model_validate(get_push_notif_req_data)
+
+    # TaskResubscriptionRequest
+    task_resubscribe_req_data: dict[str, Any] = {
+        'jsonrpc': '2.0',
+        'method': 'tasks/resubscribe',
+        'params': id_params.model_dump(),
+    }
+    with pytest.raises(ValidationError):
+        A2ARequest.model_validate(task_resubscribe_req_data)
+
+
 def test_content_type_not_supported_error():
     # Test ContentTypeNotSupportedError
     err = ContentTypeNotSupportedError(