fix GetAgentCard in gRPC

Author: sokoliva

src/a2a/compat/v0_3/grpc_handler.py

@@ -352,10 +352,29 @@ async def GetAgentCard(
         card_to_serve = self.agent_card
         if self.card_modifier:
             card_to_serve = await maybe_await(self.card_modifier(card_to_serve))
-        return proto_utils.ToProto.agent_card(
+        compat_card = proto_utils.ToProto.agent_card(
             conversions.to_compat_agent_card(card_to_serve)
         )
 
+        if self.agent_card.capabilities.extended_agent_card:
+
+            async def _handler(
+                server_context: ServerCallContext,
+            ) -> a2a_v0_3_pb2.AgentCard:
+                req_v03 = types_v03.GetAuthenticatedExtendedCardRequest(id=0)
+                res_v03 = await self.handler03.on_get_extended_agent_card(
+                    req_v03, server_context
+                )
+                if res_v03:
+                    return proto_utils.ToProto.agent_card(res_v03)
+                return compat_card
+
+            return await self._handle_unary(
+                context, _handler, a2a_v0_3_pb2.AgentCard()
+            )
+
+        return compat_card
+
     async def DeleteTaskPushNotificationConfig(
         self,
         request: a2a_v0_3_pb2.DeleteTaskPushNotificationConfigRequest,
@@ -379,23 +398,3 @@ async def _handler(
             return empty_pb2.Empty()
 
         return await self._handle_unary(context, _handler, empty_pb2.Empty())
-
-    async def GetExtendedCard(
-        self,
-        request: a2a_v0_3_pb2.GetAgentCardRequest,
-        context: grpc.aio.ServicerContext,
-    ) -> a2a_v0_3_pb2.AgentCard:
-        """Get the authenticated extended agent card (v0.3)."""
-
-        async def _handler(
-            server_context: ServerCallContext,
-        ) -> a2a_v0_3_pb2.AgentCard:
-            req_v03 = types_v03.GetAuthenticatedExtendedCardRequest(id=0)
-            res_v03 = await self.handler03.on_get_extended_agent_card(
-                req_v03, server_context
-            )
-            return proto_utils.ToProto.agent_card(res_v03)
-
-        return await self._handle_unary(
-            context, _handler, a2a_v0_3_pb2.AgentCard()
-        )

tests/compat/v0_3/test_grpc_handler.py

@@ -37,7 +37,7 @@ def sample_agent_card() -> a2a_pb2.AgentCard:
         capabilities=a2a_pb2.AgentCapabilities(
             streaming=True,
             push_notifications=True,
-            extended_agent_card=True,
+            extended_agent_card=False,
         ),
         supported_interfaces=[
             a2a_pb2.AgentInterface(
@@ -451,7 +451,7 @@ async def test_get_agent_card_success(
         url='http://jsonrpc.v03.com',
         version='1.0.0',
         protocol_version='0.3',
-        supports_authenticated_extended_card=True,
+        supports_authenticated_extended_card=False,
         preferred_transport='JSONRPC',
         capabilities=a2a_v0_3_pb2.AgentCapabilities(
             streaming=True,
@@ -468,13 +468,13 @@ async def test_get_extended_card_success(
     mock_grpc_context: AsyncMock,
     sample_agent_card: a2a_pb2.AgentCard,
 ) -> None:
+    handler.agent_card.capabilities.extended_agent_card = True
     request = a2a_v0_3_pb2.GetAgentCardRequest()
     mock_request_handler.on_get_extended_agent_card.return_value = (
         sample_agent_card
     )
-    mock_request_handler.extended_agent_card = sample_agent_card
 
-    response = await handler.GetExtendedCard(request, mock_grpc_context)
+    response = await handler.GetAgentCard(request, mock_grpc_context)
 
     expected_res = a2a_v0_3_pb2.AgentCard(
         name='Test Agent',