File tree Expand file tree Collapse file tree 2 files changed +0
-24
lines changed
src/a2a/client/transports Expand file tree Collapse file tree 2 files changed +0
-24
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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'
You can’t perform that action at this time.
0 commit comments