Merge branch '1.0-dev' into ishymko/678-push-notifications

Author: ishymko

src/a2a/server/request_handlers/default_request_handler.py

@@ -53,7 +53,11 @@
     TaskNotFoundError,
     UnsupportedOperationError,
 )
-from a2a.utils.task import apply_history_length
+from a2a.utils.task import (
+    apply_history_length,
+    validate_history_length,
+    validate_page_size,
+)
 from a2a.utils.telemetry import SpanKind, trace_class
 
 
@@ -123,6 +127,8 @@ async def on_get_task(
         context: ServerCallContext | None = None,
     ) -> Task | None:
         """Default handler for 'tasks/get'."""
+        validate_history_length(params)
+
         task_id = params.id
         task: Task | None = await self.task_store.get(task_id, context)
         if not task:
@@ -136,6 +142,10 @@ async def on_list_tasks(
         context: ServerCallContext | None = None,
     ) -> ListTasksResponse:
         """Default handler for 'tasks/list'."""
+        validate_history_length(params)
+        if params.HasField('page_size'):
+            validate_page_size(params.page_size)
+
         page = await self.task_store.list(params, context)
         for task in page.tasks:
             if not params.include_artifacts:
@@ -330,6 +340,8 @@ async def on_message_send(
         Starts the agent execution for the message and waits for the final
         result (Task or Message).
         """
+        validate_history_length(params.configuration)
+
         (
             _task_manager,
             task_id,

src/a2a/utils/constants.py

@@ -7,6 +7,9 @@
 DEFAULT_LIST_TASKS_PAGE_SIZE = 50
 """Default page size for the `tasks/list` method."""
 
+MAX_LIST_TASKS_PAGE_SIZE = 100
+"""Maximum page size for the `tasks/list` method."""
+
 
 # Transport protocol constants
 # These match the protocol binding values used in AgentCard

src/a2a/utils/task.py

@@ -13,6 +13,8 @@
     TaskState,
     TaskStatus,
 )
+from a2a.utils.constants import MAX_LIST_TASKS_PAGE_SIZE
+from a2a.utils.errors import InvalidParamsError, ServerError
 
 
 def new_task(request: Message) -> Task:
@@ -96,6 +98,16 @@ def HasField(self, field_name: Literal['history_length']) -> bool:  # noqa: N802
         ...
 
 
+def validate_history_length(config: HistoryLengthConfig | None) -> None:
+    """Validates that history_length is non-negative."""
+    if config and config.history_length < 0:
+        raise ServerError(
+            error=InvalidParamsError(
+                message='history length must be non-negative'
+            )
+        )
+
+
 def apply_history_length(
     task: Task, config: HistoryLengthConfig | None
 ) -> Task:
@@ -136,6 +148,24 @@ def apply_history_length(
     return task
 
 
+def validate_page_size(page_size: int) -> None:
+    """Validates that page_size is in range [1, 100].
+
+    See Also:
+        https://a2a-protocol.org/latest/specification/#314-list-tasks
+    """
+    if page_size < 1:
+        raise ServerError(
+            error=InvalidParamsError(message='minimum page size is 1')
+        )
+    if page_size > MAX_LIST_TASKS_PAGE_SIZE:
+        raise ServerError(
+            error=InvalidParamsError(
+                message=f'maximum page size is {MAX_LIST_TASKS_PAGE_SIZE}'
+            )
+        )
+
+
 _ENCODING = 'utf-8'