Run ruff format

Author: holtskinner

src/a2a/server/apps/jsonrpc/jsonrpc_app.py

@@ -317,7 +317,7 @@ async def _handle_requests(self, request: Request) -> Response:  # noqa: PLR0911
                 )
             raise e
         except Exception as e:
-            logger.error(f'Unhandled exception: {e}')
+            logger.exception(f'Unhandled exception: {e}')
             traceback.print_exc()
             return self._generate_error_response(
                 request_id, A2AError(root=InternalError(message=str(e)))

src/a2a/server/events/event_consumer.py

@@ -141,10 +141,10 @@ async def consume_all(self) -> AsyncGenerator[Event]:
                 if self.queue.is_closed():
                     break
             except ValidationError as e:
-                logger.error(f'Invalid event format received: {e}')
+                logger.exception(f'Invalid event format received: {e}')
                 continue
             except Exception as e:
-                logger.error(
+                logger.exception(
                     f'Stopping event consumption due to exception: {e}'
                 )
                 self._exception = e

src/a2a/server/events/in_memory_queue_manager.py

@@ -34,7 +34,7 @@ async def add(self, task_id: str, queue: EventQueue) -> None:
         """
         async with self._lock:
             if task_id in self._task_queue:
-                raise TaskQueueExists()
+                raise TaskQueueExists
             self._task_queue[task_id] = queue
 
     async def get(self, task_id: str) -> EventQueue | None:
@@ -67,7 +67,7 @@ async def close(self, task_id: str) -> None:
         """
         async with self._lock:
             if task_id not in self._task_queue:
-                raise NoTaskQueue()
+                raise NoTaskQueue
             queue = self._task_queue.pop(task_id)
             await queue.close()
 

src/a2a/server/request_handlers/default_request_handler.py

@@ -303,7 +303,7 @@ async def on_message_send(
             )
 
         except Exception as e:
-            logger.error(f'Agent execution failed. Error: {e}')
+            logger.exception(f'Agent execution failed. Error: {e}')
             raise
         finally:
             if interrupted_or_non_blocking:

src/a2a/server/tasks/base_push_notification_sender.py

@@ -66,7 +66,7 @@ async def _dispatch_notification(
             )
             return True
         except Exception as e:
-            logger.error(
+            logger.exception(
                 f'Error sending push-notification for task_id={task.id} to URL: {url}. Error: {e}'
             )
             return False

src/a2a/server/tasks/database_push_notification_config_store.py

@@ -175,7 +175,7 @@ def _from_orm(
                     decrypted_payload
                 )
             except (json.JSONDecodeError, ValidationError) as e:
-                logger.error(
+                logger.exception(
                     'Failed to parse decrypted push notification config for task %s, config %s. '
                     'Data is corrupted or not valid JSON after decryption.',
                     model_instance.task_id,
@@ -201,7 +201,7 @@ def _from_orm(
             return PushNotificationConfig.model_validate_json(payload)
         except (json.JSONDecodeError, ValidationError) as e:
             if self._fernet:
-                logger.error(
+                logger.exception(
                     'Failed to parse push notification config for task %s, config %s. '
                     'Decryption failed and the data is not valid JSON. '
                     'This likely indicates the data is corrupted or encrypted with a different key.',
@@ -210,7 +210,7 @@ def _from_orm(
                 )
             else:
                 # if no key is configured and the payload is not valid JSON.
-                logger.error(
+                logger.exception(
                     'Failed to parse push notification config for task %s, config %s. '
                     'Data is not valid JSON and no encryption key is configured.',
                     model_instance.task_id,
@@ -253,7 +253,7 @@ async def get_info(self, task_id: str) -> list[PushNotificationConfig]:
                 try:
                     configs.append(self._from_orm(model))
                 except ValueError as e:
-                    logger.error(
+                    logger.exception(
                         'Could not deserialize push notification config for task %s, config %s: %s',
                         model.task_id,
                         model.config_id,

src/a2a/utils/telemetry.py

@@ -213,7 +213,7 @@ async def async_wrapper(*args, **kwargs) -> Any:
                             span, args, kwargs, result, exception
                         )
                     except Exception as attr_e:
-                        logger.error(
+                        logger.exception(
                             f'attribute_extractor error in span {actual_span_name}: {attr_e}'
                         )
 
@@ -247,7 +247,7 @@ def sync_wrapper(*args, **kwargs) -> Any:
                             span, args, kwargs, result, exception
                         )
                     except Exception as attr_e:
-                        logger.error(
+                        logger.exception(
                             f'attribute_extractor error in span {actual_span_name}: {attr_e}'
                         )
 

tests/client/test_base_client.py

@@ -1,4 +1,4 @@
-from unittest.mock import AsyncMock, MagicMock, patch
+from unittest.mock import AsyncMock, MagicMock
 
 import pytest
 
@@ -20,8 +20,7 @@
 
 @pytest.fixture
 def mock_transport():
-    transport = AsyncMock(spec=ClientTransport)
-    return transport
+    return AsyncMock(spec=ClientTransport)
 
 
 @pytest.fixture

tests/client/test_client_task_manager.py

@@ -1,21 +1,23 @@
-import pytest
 from unittest.mock import AsyncMock, Mock, patch
+
+import pytest
+
 from a2a.client.client_task_manager import ClientTaskManager
 from a2a.client.errors import (
     A2AClientInvalidArgsError,
     A2AClientInvalidStateError,
 )
 from a2a.types import (
+    Artifact,
+    Message,
+    Part,
+    Role,
     Task,
-    TaskStatus,
+    TaskArtifactUpdateEvent,
     TaskState,
+    TaskStatus,
     TaskStatusUpdateEvent,
-    TaskArtifactUpdateEvent,
-    Message,
-    Role,
-    Part,
     TextPart,
-    Artifact,
 )
 
 

tests/server/apps/jsonrpc/test_serialization.py

@@ -1,8 +1,8 @@
 from unittest import mock
 
 import pytest
-from fastapi import FastAPI
 
+from fastapi import FastAPI
 from pydantic import ValidationError
 from starlette.testclient import TestClient