Rename fields to camel_case, fix grpc import

Author: mikeas1

src/a2a/server/request_handlers/grpc_handler.py

@@ -1,25 +1,21 @@
 # ruff: noqa: N802
 import contextlib
 import logging
-
 from abc import ABC, abstractmethod
 from collections.abc import AsyncIterable, Sequence
 
-
 try:
     import grpc
     import grpc.aio
+    from grpc.aio import Metadata
 except ImportError as e:
     raise ImportError(
         'GrpcHandler requires grpcio and grpcio-tools to be installed. '
         'Install with: '
         "'pip install a2a-sdk[grpc]'"
     ) from e
 
-from grpc.aio import Metadata
-
 import a2a.grpc.a2a_pb2_grpc as a2a_grpc
-
 from a2a import types
 from a2a.auth.user import UnauthenticatedUser
 from a2a.extensions.common import (
@@ -34,7 +30,6 @@
 from a2a.utils.errors import ServerError
 from a2a.utils.helpers import validate, validate_async_generator
 
-
 logger = logging.getLogger(__name__)
 
 # For now we use a trivial wrapper on the grpc context object

tests/extensions/test_common.py

@@ -33,8 +33,8 @@ def test_find_extension_by_uri():
         version='1.0',
         url='http://test.com',
         skills=[],
-        defaultInputModes=['text/plain'],
-        defaultOutputModes=['text/plain'],
+        default_input_modes=['text/plain'],
+        default_output_modes=['text/plain'],
         capabilities=AgentCapabilities(extensions=[ext1, ext2]),
     )
 
@@ -50,8 +50,8 @@ def test_find_extension_by_uri_no_extensions():
         version='1.0',
         url='http://test.com',
         skills=[],
-        defaultInputModes=['text/plain'],
-        defaultOutputModes=['text/plain'],
+        default_input_modes=['text/plain'],
+        default_output_modes=['text/plain'],
         capabilities=AgentCapabilities(extensions=None),
     )
 

tests/server/apps/jsonrpc/test_jsonrpc_app.py

@@ -1,7 +1,6 @@
 from unittest.mock import AsyncMock, MagicMock
 
 import pytest
-from starlette.applications import Starlette
 from starlette.testclient import TestClient
 
 # Attempt to import StarletteBaseUser, fallback to MagicMock if not available
@@ -21,10 +20,10 @@
     RequestHandler,
 )  # For mock spec
 from a2a.types import (
-    AgentCapabilities,
     AgentCard,
     Message,
     MessageSendParams,
+    Part,
     Role,
     SendMessageRequest,
     SendMessageResponse,
@@ -107,9 +106,9 @@ def mock_handler(self):
             root=SendMessageSuccessResponse(
                 id='1',
                 result=Message(
-                    messageId='test',
+                    message_id='test',
                     role=Role.agent,
-                    parts=[TextPart(text='response message')],
+                    parts=[Part(TextPart(text='response message'))],
                 ),
             )
         )
@@ -138,9 +137,9 @@ def test_request_with_single_extension(self, client, mock_handler):
                 id='1',
                 params=MessageSendParams(
                     message=Message(
-                        messageId='1',
+                        message_id='1',
                         role=Role.user,
-                        parts=[TextPart(text='hi')],
+                        parts=[Part(TextPart(text='hi'))],
                     )
                 ),
             ).model_dump(),
@@ -163,9 +162,9 @@ def test_request_with_comma_separated_extensions(
                 id='1',
                 params=MessageSendParams(
                     message=Message(
-                        messageId='1',
+                        message_id='1',
                         role=Role.user,
-                        parts=[TextPart(text='hi')],
+                        parts=[Part(TextPart(text='hi'))],
                     )
                 ),
             ).model_dump(),
@@ -190,9 +189,9 @@ def test_request_with_comma_separated_extensions_no_space(
                 id='1',
                 params=MessageSendParams(
                     message=Message(
-                        messageId='1',
+                        message_id='1',
                         role=Role.user,
-                        parts=[TextPart(text='hi')],
+                        parts=[Part(TextPart(text='hi'))],
                     )
                 ),
             ).model_dump(),
@@ -217,9 +216,9 @@ def test_request_with_multiple_extension_headers(
                 id='1',
                 params=MessageSendParams(
                     message=Message(
-                        messageId='1',
+                        message_id='1',
                         role=Role.user,
-                        parts=[TextPart(text='hi')],
+                        parts=[Part(TextPart(text='hi'))],
                     )
                 ),
             ).model_dump(),
@@ -238,9 +237,9 @@ def side_effect(request, context: ServerCallContext):
                 root=SendMessageSuccessResponse(
                     id='1',
                     result=Message(
-                        messageId='test',
+                        message_id='test',
                         role=Role.agent,
-                        parts=[TextPart(text='response message')],
+                        parts=[Part(TextPart(text='response message'))],
                     ),
                 )
             )
@@ -253,9 +252,9 @@ def side_effect(request, context: ServerCallContext):
                 id='1',
                 params=MessageSendParams(
                     message=Message(
-                        messageId='1',
+                        message_id='1',
                         role=Role.user,
-                        parts=[TextPart(text='hi')],
+                        parts=[Part(TextPart(text='hi'))],
                     )
                 ),
             ).model_dump(),