Skip to content

Commit 270d6e7

Browse files
committed
Remove extensions from grpc methog get_card
1 parent 28b1d53 commit 270d6e7

File tree

2 files changed

+0
-24
lines changed

2 files changed

+0
-24
lines changed

src/a2a/client/transports/grpc.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -229,7 +229,6 @@ async def get_card(
229229

230230
card_pb = await self.stub.GetAgentCard(
231231
a2a_pb2.GetAgentCardRequest(),
232-
metadata=self._get_metadata(context), # probaby not needed
233232
)
234233
card = proto_utils.FromProto.agent_card(card_pb)
235234
self.agent_card = card

tests/client/test_grpc_client.py

Lines changed: 0 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -719,26 +719,3 @@ async def test_get_task_callback_with_extensions(
719719
metadata_dict = dict(kwargs['metadata'])
720720
assert HTTP_EXTENSION_HEADER in metadata_dict
721721
assert metadata_dict[HTTP_EXTENSION_HEADER] == 'test_extension'
722-
723-
@pytest.mark.asyncio
724-
async def test_get_card_with_extensions(
725-
self, mock_grpc_stub: AsyncMock, sample_agent_card: AgentCard
726-
):
727-
extensions = ['test_extension']
728-
transport = GrpcTransport(
729-
channel=AsyncMock(),
730-
agent_card=sample_agent_card,
731-
extensions=extensions,
732-
)
733-
transport.stub = mock_grpc_stub
734-
mock_grpc_stub.GetAgentCard.return_value = (
735-
proto_utils.ToProto.agent_card(sample_agent_card)
736-
)
737-
738-
await transport.get_card()
739-
740-
mock_grpc_stub.GetAgentCard.assert_awaited_once()
741-
_, kwargs = mock_grpc_stub.GetAgentCard.call_args
742-
metadata_dict = dict(kwargs['metadata'])
743-
assert HTTP_EXTENSION_HEADER in metadata_dict
744-
assert metadata_dict[HTTP_EXTENSION_HEADER] == 'test_extension'

0 commit comments

Comments
 (0)