Skip to content

Commit 2cd2fcd

Browse files
committed
Fix remaining broken tests
1 parent d632deb commit 2cd2fcd

2 files changed

Lines changed: 15 additions & 13 deletions

File tree

src/a2a/client/transports/jsonrpc.py

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -22,13 +22,14 @@
2222
AgentCard,
2323
CancelTaskRequest,
2424
CancelTaskResponse,
25+
GetAuthenticatedExtendedCardRequest,
26+
GetAuthenticatedExtendedCardResponse,
2527
GetTaskPushNotificationConfigParams,
2628
GetTaskPushNotificationConfigRequest,
2729
GetTaskPushNotificationConfigResponse,
2830
GetTaskRequest,
2931
GetTaskResponse,
3032
JSONRPCErrorResponse,
31-
JSONRPCRequest,
3233
Message,
3334
MessageSendParams,
3435
SendMessageRequest,
@@ -349,23 +350,24 @@ async def get_card(
349350
if not self._needs_extended_card:
350351
return card
351352

352-
_, modified_kwargs = await self._apply_interceptors(
353-
'agent/getAuthenticatedExtendedCard',
354-
{},
353+
request = GetAuthenticatedExtendedCardRequest(id=str(uuid4()))
354+
payload, modified_kwargs = await self._apply_interceptors(
355+
request.method,
356+
request.model_dump(mode='json', exclude_none=True),
355357
self._get_http_args(context),
356358
context,
357359
)
358360

359361
response_data = await self._send_request(
360-
JSONRPCRequest(
361-
method='agent/getAuthenticatedExtendedCard',
362-
params={},
363-
id=str(uuid4()),
364-
).model_dump(),
362+
payload,
365363
modified_kwargs,
366364
)
367-
card = AgentCard.model_validate(response_data)
368-
self.agent_card = card
365+
response = GetAuthenticatedExtendedCardResponse.model_validate(
366+
response_data
367+
)
368+
if isinstance(response.root, JSONRPCErrorResponse):
369+
raise A2AClientJSONRPCError(response.root)
370+
self.agent_card = response.root.result
369371
self._needs_extended_card = False
370372
return card
371373

src/a2a/server/apps/rest/rest_app.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -196,14 +196,14 @@ def routes(self) -> dict[tuple[str, str], Callable[[Request], Any]]:
196196
('/v1/message:send', 'POST'): functools.partial(
197197
self._handle_request, self.handler.on_message_send
198198
),
199-
('/v1/message:stream', 'GET'): functools.partial(
199+
('/v1/message:stream', 'POST'): functools.partial(
200200
self._handle_streaming_request,
201201
self.handler.on_message_send_stream,
202202
),
203203
('/v1/tasks/{id}:cancel', 'POST'): functools.partial(
204204
self._handle_request, self.handler.on_cancel_task
205205
),
206-
('/v1/tasks/{id}:subscribe', 'POST'): functools.partial(
206+
('/v1/tasks/{id}:subscribe', 'GET'): functools.partial(
207207
self._handle_streaming_request,
208208
self.handler.on_resubscribe_to_task,
209209
),

0 commit comments

Comments
 (0)