Update usage in library

Author: holtskinner

src/a2a/client/helpers.py

@@ -15,8 +15,8 @@ def create_text_message_object(
         content: The text content of the message. Defaults to an empty string.
 
     Returns:
-        A `Message` object with a new UUID messageId.
+        A `Message` object with a new UUID message_id.
     """
     return Message(
-        role=role, parts=[Part(TextPart(text=content))], messageId=str(uuid4())
+        role=role, parts=[Part(TextPart(text=content))], message_id=str(uuid4())
     )

src/a2a/server/agent_execution/context.py

@@ -53,14 +53,14 @@ def __init__(  # noqa: PLR0913
         # match the request. Otherwise, create them
         if self._params:
             if task_id:
-                self._params.message.taskId = task_id
+                self._params.message.task_id = task_id
                 if task and task.id != task_id:
                     raise ServerError(InvalidParamsError(message='bad task id'))
             else:
                 self._check_or_generate_task_id()
             if context_id:
-                self._params.message.contextId = context_id
-                if task and task.contextId != context_id:
+                self._params.message.context_id = context_id
+                if task and task.context_id != context_id:
                     raise ServerError(
                         InvalidParamsError(message='bad context id')
                     )
@@ -148,17 +148,17 @@ def _check_or_generate_task_id(self) -> None:
         if not self._params:
             return
 
-        if not self._task_id and not self._params.message.taskId:
-            self._params.message.taskId = str(uuid.uuid4())
-        if self._params.message.taskId:
-            self._task_id = self._params.message.taskId
+        if not self._task_id and not self._params.message.task_id:
+            self._params.message.task_id = str(uuid.uuid4())
+        if self._params.message.task_id:
+            self._task_id = self._params.message.task_id
 
     def _check_or_generate_context_id(self) -> None:
         """Ensures a context ID is present, generating one if necessary."""
         if not self._params:
             return
 
-        if not self._context_id and not self._params.message.contextId:
-            self._params.message.contextId = str(uuid.uuid4())
-        if self._params.message.contextId:
-            self._context_id = self._params.message.contextId
+        if not self._context_id and not self._params.message.context_id:
+            self._params.message.context_id = str(uuid.uuid4())
+        if self._params.message.context_id:
+            self._context_id = self._params.message.context_id

src/a2a/server/agent_execution/simple_request_context_builder.py

@@ -18,7 +18,7 @@ def __init__(
 
         Args:
             should_populate_referred_tasks: If True, the builder will fetch tasks
-                referenced in `params.message.referenceTaskIds` and populate the
+                referenced in `params.message.reference_task_ids` and populate the
                 `related_tasks` field in the RequestContext. Defaults to False.
             task_store: The TaskStore instance to use for fetching referred tasks.
                 Required if `should_populate_referred_tasks` is True.
@@ -38,7 +38,7 @@ async def build(
 
         This method assembles the RequestContext object. If the builder was
         initialized with `should_populate_referred_tasks=True`, it fetches all tasks
-        referenced in `params.message.referenceTaskIds` from the `task_store`.
+        referenced in `params.message.reference_task_ids` from the `task_store`.
 
         Args:
             params: The parameters of the incoming message send request.
@@ -57,12 +57,12 @@ async def build(
             self._task_store
             and self._should_populate_referred_tasks
             and params
-            and params.message.referenceTaskIds
+            and params.message.reference_task_ids
         ):
             tasks = await asyncio.gather(
                 *[
                     self._task_store.get(task_id)
-                    for task_id in params.message.referenceTaskIds
+                    for task_id in params.message.reference_task_ids
                 ]
             )
             related_tasks = [x for x in tasks if x is not None]

src/a2a/server/apps/jsonrpc/fastapi_app.py

@@ -89,7 +89,7 @@ def add_routes_to_app(
         )(self._handle_requests)
         app.get(agent_card_url)(self._handle_get_agent_card)
 
-        if self.agent_card.supportsAuthenticatedExtendedCard:
+        if self.agent_card.supports_authenticated_extended_card:
             app.get(extended_agent_card_url)(
                 self._handle_get_authenticated_extended_agent_card
             )

src/a2a/server/apps/jsonrpc/jsonrpc_app.py

@@ -135,11 +135,11 @@ def __init__(
             agent_card=agent_card, request_handler=http_handler
         )
         if (
-            self.agent_card.supportsAuthenticatedExtendedCard
+            self.agent_card.supports_authenticated_extended_card
             and self.extended_agent_card is None
         ):
             logger.error(
-                'AgentCard.supportsAuthenticatedExtendedCard is True, but no extended_agent_card was provided. The /agent/authenticatedExtendedCard endpoint will return 404.'
+                'AgentCard.supports_authenticated_extended_card is True, but no extended_agent_card was provided. The /agent/authenticatedExtendedCard endpoint will return 404.'
             )
         self._context_builder = context_builder or DefaultCallContextBuilder()
 
@@ -421,7 +421,7 @@ async def _handle_get_authenticated_extended_agent_card(
         self, request: Request
     ) -> JSONResponse:
         """Handles GET requests for the authenticated extended agent card."""
-        if not self.agent_card.supportsAuthenticatedExtendedCard:
+        if not self.agent_card.supports_authenticated_extended_card:
             return JSONResponse(
                 {'error': 'Extended agent card not supported or not enabled.'},
                 status_code=404,
@@ -435,7 +435,7 @@ async def _handle_get_authenticated_extended_agent_card(
                     by_alias=True,
                 )
             )
-        # If supportsAuthenticatedExtendedCard is true, but no specific
+        # If supports_authenticated_extended_card is true, but no specific
         # extended_agent_card was provided during server initialization,
         # return a 404
         return JSONResponse(

src/a2a/server/apps/jsonrpc/starlette_app.py

@@ -86,7 +86,7 @@ def routes(
             ),
         ]
 
-        if self.agent_card.supportsAuthenticatedExtendedCard:
+        if self.agent_card.supports_authenticated_extended_card:
             app_routes.append(
                 Route(
                     extended_agent_card_url,

src/a2a/server/models.py

@@ -123,7 +123,7 @@ class TaskMixin:
     """Mixin providing standard task columns with proper type handling."""
 
     id: Mapped[str] = mapped_column(String(36), primary_key=True, index=True)
-    contextId: Mapped[str] = mapped_column(String(36), nullable=False)  # noqa: N815
+    context_id: Mapped[str] = mapped_column(String(36), nullable=False)
     kind: Mapped[str] = mapped_column(
         String(16), nullable=False, default='task'
     )
@@ -148,12 +148,12 @@ def task_metadata(cls) -> Mapped[dict[str, Any] | None]:
     def __repr__(self) -> str:
         """Return a string representation of the task."""
         repr_template = (
-            '<{CLS}(id="{ID}", contextId="{CTX_ID}", status="{STATUS}")>'
+            '<{CLS}(id="{ID}", context_id="{CTX_ID}", status="{STATUS}")>'
         )
         return repr_template.format(
             CLS=self.__class__.__name__,
             ID=self.id,
-            CTX_ID=self.contextId,
+            CTX_ID=self.context_id,
             STATUS=self.status,
         )
 
@@ -188,11 +188,11 @@ class TaskModel(TaskMixin, base):
         @override
         def __repr__(self) -> str:
             """Return a string representation of the task."""
-            repr_template = '<TaskModel[{TABLE}](id="{ID}", contextId="{CTX_ID}", status="{STATUS}")>'
+            repr_template = '<TaskModel[{TABLE}](id="{ID}", context_id="{CTX_ID}", status="{STATUS}")>'
             return repr_template.format(
                 TABLE=table_name,
                 ID=self.id,
-                CTX_ID=self.contextId,
+                CTX_ID=self.context_id,
                 STATUS=self.status,
             )
 

src/a2a/server/request_handlers/default_request_handler.py

@@ -126,7 +126,7 @@ async def on_cancel_task(
 
         task_manager = TaskManager(
             task_id=task.id,
-            context_id=task.contextId,
+            context_id=task.context_id,
             task_store=self.task_store,
             initial_message=None,
         )
@@ -140,7 +140,7 @@ async def on_cancel_task(
             RequestContext(
                 None,
                 task_id=task.id,
-                context_id=task.contextId,
+                context_id=task.context_id,
                 task=task,
             ),
             queue,
@@ -184,8 +184,8 @@ async def _setup_message_execution(
         """
         # Create task manager and validate existing task
         task_manager = TaskManager(
-            task_id=params.message.taskId,
-            context_id=params.message.contextId,
+            task_id=params.message.task_id,
+            context_id=params.message.context_id,
             task_store=self.task_store,
             initial_message=params.message,
         )
@@ -205,7 +205,7 @@ async def _setup_message_execution(
         request_context = await self._request_context_builder.build(
             params=params,
             task_id=task.id if task else None,
-            context_id=params.message.contextId,
+            context_id=params.message.context_id,
             task=task,
             context=context,
         )
@@ -218,10 +218,10 @@ async def _setup_message_execution(
         if (
             self._push_config_store
             and params.configuration
-            and params.configuration.pushNotificationConfig
+            and params.configuration.push_notification_config
         ):
             await self._push_config_store.set_info(
-                task_id, params.configuration.pushNotificationConfig
+                task_id, params.configuration.push_notification_config
             )
 
         queue = await self._queue_manager.create_or_tap(task_id)
@@ -366,13 +366,13 @@ async def on_set_task_push_notification_config(
         if not self._push_config_store:
             raise ServerError(error=UnsupportedOperationError())
 
-        task: Task | None = await self.task_store.get(params.taskId)
+        task: Task | None = await self.task_store.get(params.task_id)
         if not task:
             raise ServerError(error=TaskNotFoundError())
 
         await self._push_config_store.set_info(
-            params.taskId,
-            params.pushNotificationConfig,
+            params.task_id,
+            params.push_notification_config,
         )
 
         return params
@@ -404,7 +404,8 @@ async def on_get_task_push_notification_config(
             )
 
         return TaskPushNotificationConfig(
-            taskId=params.id, pushNotificationConfig=push_notification_config[0]
+            task_id=params.id,
+            pushNotificationConfig=push_notification_config[0],
         )
 
     async def on_resubscribe_to_task(
@@ -430,7 +431,7 @@ async def on_resubscribe_to_task(
 
         task_manager = TaskManager(
             task_id=task.id,
-            context_id=task.contextId,
+            context_id=task.context_id,
             task_store=self.task_store,
             initial_message=None,
         )
@@ -470,7 +471,7 @@ async def on_list_task_push_notification_config(
             for config in push_notification_config_list:
                 task_push_notification_config.append(
                     TaskPushNotificationConfig(
-                        taskId=params.id, pushNotificationConfig=config
+                        task_id=params.id, pushNotificationConfig=config
                     )
                 )
 
@@ -493,5 +494,5 @@ async def on_delete_task_push_notification_config(
             raise ServerError(error=TaskNotFoundError())
 
         await self._push_config_store.delete_info(
-            params.id, params.pushNotificationConfigId
+            params.id, params.push_notification_config_id
         )

src/a2a/server/request_handlers/grpc_handler.py

@@ -224,7 +224,7 @@ async def GetTaskPushNotificationConfig(
         return a2a_pb2.TaskPushNotificationConfig()
 
     @validate(
-        lambda self: self.agent_card.capabilities.pushNotifications,
+        lambda self: self.agent_card.capabilities.push_notifications,
         'Push notifications are not supported by the agent',
     )
     async def CreateTaskPushNotificationConfig(

src/a2a/server/request_handlers/jsonrpc_handler.py

@@ -255,7 +255,7 @@ async def get_push_notification_config(
             )
 
     @validate(
-        lambda self: self.agent_card.capabilities.pushNotifications,
+        lambda self: self.agent_card.capabilities.push_notifications,
         'Push notifications are not supported by the agent',
     )
     async def set_push_notification_config(