Merge branch '1.0-dev' into ishymko/integration-package-test

Author: ishymko

docs/migrations/v1_0/database/zero_downtime.md

@@ -62,7 +62,7 @@ Enable the v0.3 conversion utilities in your application entry point (e.g., `mai
 
 ```python
 from a2a.server.tasks import DatabaseTaskStore, DatabasePushNotificationConfigStore
-from a2a.compat.v0_3.conversions import (
+from a2a.compat.v0_3.model_conversions import (
     core_to_compat_task_model,
     core_to_compat_push_notification_config_model,
 )
@@ -126,7 +126,7 @@ This allows v1.0 instances to read *all* existing data regardless of when it was
 
 ## 🧩 Resources
 - **[a2a-db CLI](../../../../src/a2a/migrations/README.md)**: The primary tool for executing schema migrations.
-- **[Compatibility Conversions](../../../../src/a2a/compat/v0_3/conversions.py)**: Source for classes like `core_to_compat_task_model` used in Step 2.
+- **[Compatibility Conversions](../../../../src/a2a/compat/v0_3/model_conversions.py)**: Source for model conversion functions `core_to_compat_task_model` and `core_to_compat_push_notification_config_model` used in Step 2.
 - **[Task Store Implementation](../../../../src/a2a/server/tasks/database_task_store.py)**: The `DatabaseTaskStore` which handles the version-aware read/write logic.
 - **[Push Notification Store Implementation](../../../../src/a2a/server/tasks/database_push_notification_config_store.py)**: The `DatabasePushNotificationConfigStore` which handles the version-aware read/write logic.
 

pyproject.toml

@@ -199,6 +199,8 @@ indent-width = 4 # Google Style Guide §3.4: 4 spaces
 target-version = "py310" # Minimum Python version
 
 [tool.ruff.lint]
+preview = true
+explicit-preview-rules = true
 ignore = [
     "COM812", # Trailing comma missing.
     "FBT001", # Boolean positional arg in function definition

src/a2a/client/transports/grpc.py

@@ -47,7 +47,8 @@
     TaskPushNotificationConfig,
 )
 from a2a.utils.constants import PROTOCOL_VERSION_CURRENT, VERSION_HEADER
-from a2a.utils.errors import A2A_REASON_TO_ERROR
+from a2a.utils.errors import A2A_REASON_TO_ERROR, A2AError
+from a2a.utils.proto_utils import bad_request_to_validation_errors
 from a2a.utils.telemetry import SpanKind, trace_class
 
 
@@ -61,17 +62,23 @@ def _map_grpc_error(e: grpc.aio.AioRpcError) -> NoReturn:
 
     # Use grpc_status to cleanly extract the rich Status from the call
     status = rpc_status.from_call(cast('grpc.Call', e))
+    data = None
 
     if status is not None:
+        exception_cls: type[A2AError] | None = None
         for detail in status.details:
             if detail.Is(error_details_pb2.ErrorInfo.DESCRIPTOR):
                 error_info = error_details_pb2.ErrorInfo()
                 detail.Unpack(error_info)
-
                 if error_info.domain == 'a2a-protocol.org':
                     exception_cls = A2A_REASON_TO_ERROR.get(error_info.reason)
-                    if exception_cls:
-                        raise exception_cls(status.message) from e
+            elif detail.Is(error_details_pb2.BadRequest.DESCRIPTOR):
+                bad_request = error_details_pb2.BadRequest()
+                detail.Unpack(bad_request)
+                data = {'errors': bad_request_to_validation_errors(bad_request)}
+
+        if exception_cls:
+            raise exception_cls(status.message, data=data) from e
 
     raise A2AClientError(f'gRPC Error {e.code().name}: {e.details()}') from e
 

src/a2a/client/transports/jsonrpc.py

@@ -318,9 +318,10 @@ def _create_jsonrpc_error(self, error_dict: dict[str, Any]) -> Exception:
         """Creates the appropriate A2AError from a JSON-RPC error dictionary."""
         code = error_dict.get('code')
         message = error_dict.get('message', str(error_dict))
+        data = error_dict.get('data')
 
         if isinstance(code, int) and code in _JSON_RPC_ERROR_CODE_TO_A2A_ERROR:
-            return _JSON_RPC_ERROR_CODE_TO_A2A_ERROR[code](message)
+            return _JSON_RPC_ERROR_CODE_TO_A2A_ERROR[code](message, data=data)
 
         # Fallback to general A2AClientError
         return A2AClientError(f'JSON-RPC Error {code}: {message}')

src/a2a/compat/v0_3/conversions.py

@@ -1,16 +1,11 @@
 import base64
 
-from typing import TYPE_CHECKING, Any
-
-
-if TYPE_CHECKING:
-    from cryptography.fernet import Fernet
+from typing import Any
 
 from google.protobuf.json_format import MessageToDict, ParseDict
 
 from a2a.compat.v0_3 import types as types_v03
 from a2a.compat.v0_3.versions import is_legacy_version
-from a2a.server.models import PushNotificationConfigModel, TaskModel
 from a2a.types import a2a_pb2 as pb2_v10
 from a2a.utils import constants, errors
 
@@ -1378,77 +1373,3 @@ def to_compat_get_extended_agent_card_request(
 ) -> types_v03.GetAuthenticatedExtendedCardRequest:
     """Convert get extended agent card request to v0.3 compat type."""
     return types_v03.GetAuthenticatedExtendedCardRequest(id=request_id)
-
-
-def core_to_compat_task_model(task: pb2_v10.Task, owner: str) -> TaskModel:
-    """Converts a 1.0 core Task to a TaskModel using v0.3 JSON structure."""
-    compat_task = to_compat_task(task)
-    data = compat_task.model_dump(mode='json')
-
-    return TaskModel(
-        id=task.id,
-        context_id=task.context_id,
-        owner=owner,
-        status=data.get('status'),
-        history=data.get('history'),
-        artifacts=data.get('artifacts'),
-        task_metadata=data.get('metadata'),
-        protocol_version='0.3',
-    )
-
-
-def compat_task_model_to_core(task_model: TaskModel) -> pb2_v10.Task:
-    """Converts a TaskModel with v0.3 structure to a 1.0 core Task."""
-    compat_task = types_v03.Task(
-        id=task_model.id,
-        context_id=task_model.context_id,
-        status=types_v03.TaskStatus.model_validate(task_model.status),
-        artifacts=(
-            [types_v03.Artifact.model_validate(a) for a in task_model.artifacts]
-            if task_model.artifacts
-            else []
-        ),
-        history=(
-            [types_v03.Message.model_validate(h) for h in task_model.history]
-            if task_model.history
-            else []
-        ),
-        metadata=task_model.task_metadata,
-    )
-    return to_core_task(compat_task)
-
-
-def core_to_compat_push_notification_config_model(
-    task_id: str,
-    config: pb2_v10.TaskPushNotificationConfig,
-    owner: str,
-    fernet: 'Fernet | None' = None,
-) -> PushNotificationConfigModel:
-    """Converts a 1.0 core TaskPushNotificationConfig to a PushNotificationConfigModel using v0.3 JSON structure."""
-    compat_config = to_compat_push_notification_config(config)
-
-    json_payload = compat_config.model_dump_json().encode('utf-8')
-    data_to_store = fernet.encrypt(json_payload) if fernet else json_payload
-
-    return PushNotificationConfigModel(
-        task_id=task_id,
-        config_id=config.id,
-        owner=owner,
-        config_data=data_to_store,
-        protocol_version='0.3',
-    )
-
-
-def compat_push_notification_config_model_to_core(
-    model_instance: str, task_id: str
-) -> pb2_v10.TaskPushNotificationConfig:
-    """Converts a PushNotificationConfigModel with v0.3 structure back to a 1.0 core TaskPushNotificationConfig."""
-    inner_config = types_v03.PushNotificationConfig.model_validate_json(
-        model_instance
-    )
-    return to_core_task_push_notification_config(
-        types_v03.TaskPushNotificationConfig(
-            task_id=task_id,
-            push_notification_config=inner_config,
-        )
-    )

src/a2a/compat/v0_3/model_conversions.py

@@ -0,0 +1,92 @@
+"""Database model conversions for v0.3 compatibility."""
+
+from typing import TYPE_CHECKING
+
+
+if TYPE_CHECKING:
+    from cryptography.fernet import Fernet
+
+
+from a2a.compat.v0_3 import types as types_v03
+from a2a.compat.v0_3.conversions import (
+    to_compat_push_notification_config,
+    to_compat_task,
+    to_core_task,
+    to_core_task_push_notification_config,
+)
+from a2a.server.models import PushNotificationConfigModel, TaskModel
+from a2a.types import a2a_pb2 as pb2_v10
+
+
+def core_to_compat_task_model(task: pb2_v10.Task, owner: str) -> TaskModel:
+    """Converts a 1.0 core Task to a TaskModel using v0.3 JSON structure."""
+    compat_task = to_compat_task(task)
+    data = compat_task.model_dump(mode='json')
+
+    return TaskModel(
+        id=task.id,
+        context_id=task.context_id,
+        owner=owner,
+        status=data.get('status'),
+        history=data.get('history'),
+        artifacts=data.get('artifacts'),
+        task_metadata=data.get('metadata'),
+        protocol_version='0.3',
+    )
+
+
+def compat_task_model_to_core(task_model: TaskModel) -> pb2_v10.Task:
+    """Converts a TaskModel with v0.3 structure to a 1.0 core Task."""
+    compat_task = types_v03.Task(
+        id=task_model.id,
+        context_id=task_model.context_id,
+        status=types_v03.TaskStatus.model_validate(task_model.status),
+        artifacts=(
+            [types_v03.Artifact.model_validate(a) for a in task_model.artifacts]
+            if task_model.artifacts
+            else []
+        ),
+        history=(
+            [types_v03.Message.model_validate(h) for h in task_model.history]
+            if task_model.history
+            else []
+        ),
+        metadata=task_model.task_metadata,
+    )
+    return to_core_task(compat_task)
+
+
+def core_to_compat_push_notification_config_model(
+    task_id: str,
+    config: pb2_v10.TaskPushNotificationConfig,
+    owner: str,
+    fernet: 'Fernet | None' = None,
+) -> PushNotificationConfigModel:
+    """Converts a 1.0 core TaskPushNotificationConfig to a PushNotificationConfigModel using v0.3 JSON structure."""
+    compat_config = to_compat_push_notification_config(config)
+
+    json_payload = compat_config.model_dump_json().encode('utf-8')
+    data_to_store = fernet.encrypt(json_payload) if fernet else json_payload
+
+    return PushNotificationConfigModel(
+        task_id=task_id,
+        config_id=config.id,
+        owner=owner,
+        config_data=data_to_store,
+        protocol_version='0.3',
+    )
+
+
+def compat_push_notification_config_model_to_core(
+    model_instance: str, task_id: str
+) -> pb2_v10.TaskPushNotificationConfig:
+    """Converts a PushNotificationConfigModel with v0.3 structure back to a 1.0 core TaskPushNotificationConfig."""
+    inner_config = types_v03.PushNotificationConfig.model_validate_json(
+        model_instance
+    )
+    return to_core_task_push_notification_config(
+        types_v03.TaskPushNotificationConfig(
+            task_id=task_id,
+            push_notification_config=inner_config,
+        )
+    )

src/a2a/server/request_handlers/__init__.py

@@ -5,7 +5,10 @@
 from a2a.server.request_handlers.default_request_handler import (
     DefaultRequestHandler,
 )
-from a2a.server.request_handlers.request_handler import RequestHandler
+from a2a.server.request_handlers.request_handler import (
+    RequestHandler,
+    validate_request_params,
+)
 from a2a.server.request_handlers.response_helpers import (
     build_error_response,
     prepare_response_object,
@@ -43,4 +46,5 @@ def __init__(self, *args, **kwargs):
     'RequestHandler',
     'build_error_response',
     'prepare_response_object',
+    'validate_request_params',
 ]

src/a2a/server/request_handlers/default_request_handler.py

@@ -18,7 +18,10 @@
     InMemoryQueueManager,
     QueueManager,
 )
-from a2a.server.request_handlers.request_handler import RequestHandler
+from a2a.server.request_handlers.request_handler import (
+    RequestHandler,
+    validate_request_params,
+)
 from a2a.server.tasks import (
     PushNotificationConfigStore,
     PushNotificationEvent,
@@ -118,6 +121,7 @@ def __init__(  # noqa: PLR0913
         # asyncio tasks and to surface unexpected exceptions.
         self._background_tasks = set()
 
+    @validate_request_params
     async def on_get_task(
         self,
         params: GetTaskRequest,
@@ -133,6 +137,7 @@ async def on_get_task(
 
         return apply_history_length(task, params)
 
+    @validate_request_params
     async def on_list_tasks(
         self,
         params: ListTasksRequest,
@@ -154,6 +159,7 @@ async def on_list_tasks(
 
         return page
 
+    @validate_request_params
     async def on_cancel_task(
         self,
         params: CancelTaskRequest,
@@ -317,6 +323,7 @@ async def _send_push_notification_if_needed(
         ):
             await self._push_sender.send_notification(task_id, event)
 
+    @validate_request_params
     async def on_message_send(
         self,
         params: SendMessageRequest,
@@ -386,6 +393,7 @@ async def push_notification_callback(event: Event) -> None:
 
         return result
 
+    @validate_request_params
     async def on_message_send_stream(
         self,
         params: SendMessageRequest,
@@ -474,6 +482,7 @@ async def _cleanup_producer(
         async with self._running_agents_lock:
             self._running_agents.pop(task_id, None)
 
+    @validate_request_params
     async def on_create_task_push_notification_config(
         self,
         params: TaskPushNotificationConfig,
@@ -499,6 +508,7 @@ async def on_create_task_push_notification_config(
 
         return params
 
+    @validate_request_params
     async def on_get_task_push_notification_config(
         self,
         params: GetTaskPushNotificationConfigRequest,
@@ -530,6 +540,7 @@ async def on_get_task_push_notification_config(
 
         raise InternalError(message='Push notification config not found')
 
+    @validate_request_params
     async def on_subscribe_to_task(
         self,
         params: SubscribeToTaskRequest,
@@ -572,6 +583,7 @@ async def on_subscribe_to_task(
         async for event in result_aggregator.consume_and_emit(consumer):
             yield event
 
+    @validate_request_params
     async def on_list_task_push_notification_configs(
         self,
         params: ListTaskPushNotificationConfigsRequest,
@@ -597,6 +609,7 @@ async def on_list_task_push_notification_configs(
             configs=push_notification_config_list
         )
 
+    @validate_request_params
     async def on_delete_task_push_notification_config(
         self,
         params: DeleteTaskPushNotificationConfigRequest,