|
22 | 22 | AgentCard, |
23 | 23 | CancelTaskRequest, |
24 | 24 | CancelTaskResponse, |
| 25 | + GetAuthenticatedExtendedCardRequest, |
| 26 | + GetAuthenticatedExtendedCardResponse, |
25 | 27 | GetTaskPushNotificationConfigParams, |
26 | 28 | GetTaskPushNotificationConfigRequest, |
27 | 29 | GetTaskPushNotificationConfigResponse, |
28 | 30 | GetTaskRequest, |
29 | 31 | GetTaskResponse, |
30 | 32 | JSONRPCErrorResponse, |
31 | | - JSONRPCRequest, |
32 | 33 | Message, |
33 | 34 | MessageSendParams, |
34 | 35 | SendMessageRequest, |
@@ -349,23 +350,24 @@ async def get_card( |
349 | 350 | if not self._needs_extended_card: |
350 | 351 | return card |
351 | 352 |
|
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), |
355 | 357 | self._get_http_args(context), |
356 | 358 | context, |
357 | 359 | ) |
358 | 360 |
|
359 | 361 | response_data = await self._send_request( |
360 | | - JSONRPCRequest( |
361 | | - method='agent/getAuthenticatedExtendedCard', |
362 | | - params={}, |
363 | | - id=str(uuid4()), |
364 | | - ).model_dump(), |
| 362 | + payload, |
365 | 363 | modified_kwargs, |
366 | 364 | ) |
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 |
369 | 371 | self._needs_extended_card = False |
370 | 372 | return card |
371 | 373 |
|
|
0 commit comments