fix: align tests and implementation with proto definition updates

This commit resolves extensive test failures caused by recent Protocol Buffer changes in a2a_pb2.

Key changes:
- Removed invalid 'AgentCard.url' references; updated 'RestTransport' and tests to use 'supported_interfaces'.
- Updated 'AgentCard' capability assertions to use 'capabilities.extended_agent_card'.
- Fixed 'BaseClient.send_message' usage in integration tests to use 'request' kwarg instead of 'message'.
- Updated assertion logic in integration tests to check 'received_params.message'.
- verified all tests pass including database integration tests.

Signed-off-by: Luca Muscariello <muscariello@ieee.org>

Author: muscariello

scripts/grpc_gen_post_processor.py

@@ -11,7 +11,7 @@
 from pathlib import Path
 
 
-def process_generated_code(src_folder: str = 'src/a2a/grpc') -> None:
+def process_generated_code(src_folder: str = 'src/a2a/types') -> None:
     """Post processor for the generated code."""
     dir_path = Path(src_folder)
     print(dir_path)

src/a2a/client/base_client.py

@@ -76,7 +76,7 @@ async def send_message(
             ),
         )
         send_message_request = SendMessageRequest(
-            request=request, configuration=config, metadata=request_metadata
+            message=request, configuration=config, metadata=request_metadata
         )
 
         if not self._config.streaming or not self._card.capabilities.streaming:
@@ -91,12 +91,12 @@ async def send_message(
             if response.HasField('task'):
                 stream_response.task.CopyFrom(response.task)
                 client_event = (stream_response, response.task)
-            elif response.HasField('msg'):
-                stream_response.msg.CopyFrom(response.msg)
+            elif response.HasField('message'):
+                stream_response.message.CopyFrom(response.message)
                 client_event = (stream_response, None)
             else:
-                # Response must have either task or msg
-                raise ValueError('Response has neither task nor msg')
+                # Response must have either task or message
+                raise ValueError('Response has neither task nor message')
 
             await self.consume(client_event, self._card)
             yield client_event
@@ -116,7 +116,7 @@ async def _process_stream(
             client_event: ClientEvent
             # When we get a message in the stream then we don't expect any
             # further messages so yield and return
-            if stream_response.HasField('msg'):
+            if stream_response.HasField('message'):
                 client_event = (stream_response, None)
                 await self.consume(client_event, self._card)
                 yield client_event

src/a2a/client/client_factory.py

@@ -202,14 +202,9 @@ def create(
           If there is no valid matching of the client configuration with the
           server configuration, a `ValueError` is raised.
         """
-        server_preferred = (
-            card.preferred_transport or TRANSPORT_PROTOCOLS_JSONRPC
-        )
-        server_set = {server_preferred: card.url}
-        if card.additional_interfaces:
-            server_set.update(
-                {x.protocol_binding: x.url for x in card.additional_interfaces}
-            )
+        server_set = {
+            x.protocol_binding: x.url for x in card.supported_interfaces
+        }
         client_set = self._config.supported_protocol_bindings or [
             TRANSPORT_PROTOCOLS_JSONRPC
         ]
@@ -268,19 +263,15 @@ def minimal_agent_card(
     if transports is None:
         transports = []
     return AgentCard(
-        url=url,
-        preferred_transport=transports[0] if transports else None,
-        additional_interfaces=[
-            AgentInterface(protocol_binding=t, url=url) for t in transports[1:]
-        ]
-        if len(transports) > 1
-        else [],
-        supports_authenticated_extended_card=True,
-        capabilities=AgentCapabilities(),
+        supported_interfaces=[
+            AgentInterface(protocol_binding=t, url=url) for t in transports
+        ],
+        capabilities=AgentCapabilities(extended_agent_card=True),
         default_input_modes=[],
         default_output_modes=[],
         description='',
         skills=[],
         version='',
         name='',
+        protocol_versions=['v1'],
     )

src/a2a/client/client_task_manager.py

@@ -82,7 +82,7 @@ async def process(
             ClientError: If the task ID in the event conflicts with the TaskManager's ID
                          when the TaskManager's ID is already set.
         """
-        if event.HasField('msg'):
+        if event.HasField('message'):
             # Messages are not processed here.
             return None
 

src/a2a/client/transports/grpc.py

@@ -53,9 +53,7 @@ def __init__(
         self.channel = channel
         self.stub = a2a_pb2_grpc.A2AServiceStub(channel)
         self._needs_extended_card = (
-            agent_card.supports_authenticated_extended_card
-            if agent_card
-            else True
+            agent_card.capabilities.extended_agent_card if agent_card else True
         )
         self.extensions = extensions
 

src/a2a/client/transports/jsonrpc.py

@@ -56,7 +56,12 @@ def __init__(
         if url:
             self.url = url
         elif agent_card:
-            self.url = agent_card.url
+            if agent_card.supported_interfaces:
+                self.url = agent_card.supported_interfaces[0].url
+            else:
+                # Fallback or error if no interfaces?
+                # For compatibility we might check if 'url' attr exists (it does not on proto anymore)
+                raise ValueError('AgentCard has no supported interfaces')
         else:
             raise ValueError('Must provide either agent_card or url')
 

src/a2a/client/transports/rest.py

@@ -26,6 +26,7 @@
     Task,
     TaskPushNotificationConfig,
 )
+from a2a.utils.constants import TRANSPORT_HTTP_JSON, TRANSPORT_JSONRPC
 from a2a.utils.telemetry import SpanKind, trace_class
 
 
@@ -48,7 +49,18 @@ def __init__(
         if url:
             self.url = url
         elif agent_card:
-            self.url = agent_card.url
+            for interface in agent_card.supported_interfaces:
+                if interface.protocol_binding in (
+                    TRANSPORT_HTTP_JSON,
+                    TRANSPORT_JSONRPC,
+                ):
+                    self.url = interface.url
+                    break
+            else:
+                raise ValueError(
+                    f'AgentCard does not support {TRANSPORT_HTTP_JSON} '
+                    f'or {TRANSPORT_JSONRPC}'
+                )
         else:
             raise ValueError('Must provide either agent_card or url')
         if self.url.endswith('/'):
@@ -57,9 +69,7 @@ def __init__(
         self.agent_card = agent_card
         self.interceptors = interceptors or []
         self._needs_extended_card = (
-            agent_card.supports_authenticated_extended_card
-            if agent_card
-            else True
+            agent_card.capabilities.extended_agent_card if agent_card else True
         )
         self.extensions = extensions
 

src/a2a/server/agent_execution/context.py

@@ -65,13 +65,13 @@ def __init__(  # noqa: PLR0913
         # match the request. Otherwise, create them
         if self._params:
             if task_id:
-                self._params.request.task_id = 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.request.context_id = context_id
+                self._params.message.context_id = context_id
                 if task and task.context_id != context_id:
                     raise ServerError(
                         InvalidParamsError(message='bad context id')
@@ -93,7 +93,7 @@ def get_user_input(self, delimiter: str = '\n') -> str:
         if not self._params:
             return ''
 
-        return get_message_text(self._params.request, delimiter)
+        return get_message_text(self._params.message, delimiter)
 
     def attach_related_task(self, task: Task) -> None:
         """Attaches a related task to the context.
@@ -109,7 +109,7 @@ def attach_related_task(self, task: Task) -> None:
     @property
     def message(self) -> Message | None:
         """The incoming `Message` object from the request, if available."""
-        return self._params.request if self._params else None
+        return self._params.message if self._params else None
 
     @property
     def related_tasks(self) -> list[Task]:
@@ -176,23 +176,23 @@ def _check_or_generate_task_id(self) -> None:
         if not self._params:
             return
 
-        if not self._task_id and not self._params.request.task_id:
-            self._params.request.task_id = self._task_id_generator.generate(
+        if not self._task_id and not self._params.message.task_id:
+            self._params.message.task_id = self._task_id_generator.generate(
                 IDGeneratorContext(context_id=self._context_id)
             )
-        if self._params.request.task_id:
-            self._task_id = self._params.request.task_id
+        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.request.context_id:
-            self._params.request.context_id = (
+        if not self._context_id and not self._params.message.context_id:
+            self._params.message.context_id = (
                 self._context_id_generator.generate(
                     IDGeneratorContext(task_id=self._task_id)
                 )
             )
-        if self._params.request.context_id:
-            self._context_id = self._params.request.context_id
+        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.request.reference_task_ids` 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.
@@ -57,12 +57,12 @@ async def build(
             self._task_store
             and self._should_populate_referred_tasks
             and params
-            and params.request.reference_task_ids
+            and params.message.reference_task_ids
         ):
             tasks = await asyncio.gather(
                 *[
                     self._task_store.get(task_id)
-                    for task_id in params.request.reference_task_ids
+                    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

@@ -149,7 +149,7 @@ def add_routes_to_app(
                 self._handle_get_agent_card
             )
 
-        if self.agent_card.supports_authenticated_extended_card:
+        if self.agent_card.capabilities.extended_agent_card:
             app.get(extended_agent_card_url)(
                 self._handle_get_authenticated_extended_agent_card
             )