Fixes for tool comments

Author: dmandar

src/a2a/client/auth/interceptor.py

@@ -33,8 +33,10 @@ async def intercept(
         context: ClientCallContext | None,
     ) -> tuple[dict[str, Any], dict[str, Any]]:
         """Applies authentication headers to the request if credentials are available."""
-        if not all(
-            (agent_card, agent_card.security, agent_card.securitySchemes)
+        if (
+            agent_card is None
+            or agent_card.security is None
+            or agent_card.securitySchemes is None
         ):
             return request_payload, http_kwargs
 

src/a2a/client/client.py

@@ -248,7 +248,7 @@ async def send_message(
             context,
         )
         response_data = await self._send_request(payload, modified_kwargs)
-        return SendMessageResponse(response_data)
+        return SendMessageResponse.model_validate(response_data)
 
     async def send_message_streaming(
         self,
@@ -297,7 +297,9 @@ async def send_message_streaming(
         ) as event_source:
             try:
                 async for sse in event_source.aiter_sse():
-                    yield SendStreamingMessageResponse(json.loads(sse.data))
+                    yield SendStreamingMessageResponse.model_validate(
+                        json.loads(sse.data)
+                    )
             except SSEError as e:
                 raise A2AClientHTTPError(
                     400,
@@ -377,14 +379,7 @@ async def get_task(
             context,
         )
         response_data = await self._send_request(payload, modified_kwargs)
-        return GetTaskResponse(response_data)
-
-        return GetTaskResponse(
-            **await self._send_request(
-                request.model_dump(mode='json', exclude_none=True),
-                http_kwargs,
-            )
-        )
+        return GetTaskResponse.model_validate(response_data)
 
     async def cancel_task(
         self,
@@ -419,7 +414,7 @@ async def cancel_task(
             context,
         )
         response_data = await self._send_request(payload, modified_kwargs)
-        return CancelTaskResponse(response_data)
+        return CancelTaskResponse.model_validate(response_data)
 
     async def set_task_callback(
         self,
@@ -454,7 +449,9 @@ async def set_task_callback(
             context,
         )
         response_data = await self._send_request(payload, modified_kwargs)
-        return SetTaskPushNotificationConfigResponse(response_data)
+        return SetTaskPushNotificationConfigResponse.model_validate(
+            response_data
+        )
 
     async def get_task_callback(
         self,
@@ -489,4 +486,6 @@ async def get_task_callback(
             context,
         )
         response_data = await self._send_request(payload, modified_kwargs)
-        return GetTaskPushNotificationConfigResponse(response_data)
+        return GetTaskPushNotificationConfigResponse.model_validate(
+            response_data
+        )