chore: Formatting Fixes (#148)

Author: holtskinner

src/a2a/client/client.py

@@ -1,26 +1,37 @@
 import json
 import logging
+
 from collections.abc import AsyncGenerator
 from typing import Any
 from uuid import uuid4
 
 import httpx
+
 from httpx_sse import SSEError, aconnect_sse
 from pydantic import ValidationError
 
 from a2a.client.errors import A2AClientHTTPError, A2AClientJSONError
-from a2a.types import (AgentCard, CancelTaskRequest, CancelTaskResponse,
-                       GetTaskPushNotificationConfigRequest,
-                       GetTaskPushNotificationConfigResponse, GetTaskRequest,
-                       GetTaskResponse, SendMessageRequest,
-                       SendMessageResponse, SendStreamingMessageRequest,
-                       SendStreamingMessageResponse,
-                       SetTaskPushNotificationConfigRequest,
-                       SetTaskPushNotificationConfigResponse)
+from a2a.types import (
+    AgentCard,
+    CancelTaskRequest,
+    CancelTaskResponse,
+    GetTaskPushNotificationConfigRequest,
+    GetTaskPushNotificationConfigResponse,
+    GetTaskRequest,
+    GetTaskResponse,
+    SendMessageRequest,
+    SendMessageResponse,
+    SendStreamingMessageRequest,
+    SendStreamingMessageResponse,
+    SetTaskPushNotificationConfigRequest,
+    SetTaskPushNotificationConfigResponse,
+)
 from a2a.utils.telemetry import SpanKind, trace_class
 
+
 logger = logging.getLogger(__name__)
 
+
 class A2ACardResolver:
     """Agent Card resolver."""
 
@@ -160,6 +171,7 @@ async def get_client_from_agent_card_url(
             agent_card_path: The path to the agent card endpoint, relative to the base URL.
             http_kwargs: Optional dictionary of keyword arguments to pass to the
                 underlying httpx.get request when fetching the agent card.
+
         Returns:
             An initialized `A2AClient` instance.
 
@@ -169,7 +181,9 @@ async def get_client_from_agent_card_url(
         """
         agent_card: AgentCard = await A2ACardResolver(
             httpx_client, base_url=base_url, agent_card_path=agent_card_path
-        ).get_agent_card(http_kwargs=http_kwargs) # Fetches public card by default
+        ).get_agent_card(
+            http_kwargs=http_kwargs
+        )  # Fetches public card by default
         return A2AClient(httpx_client=httpx_client, agent_card=agent_card)
 
     async def send_message(

src/a2a/server/events/event_queue.py

@@ -14,12 +14,7 @@
 logger = logging.getLogger(__name__)
 
 
-Event = (
-    Message
-    | Task
-    | TaskStatusUpdateEvent
-    | TaskArtifactUpdateEvent
-)
+Event = Message | Task | TaskStatusUpdateEvent | TaskArtifactUpdateEvent
 """Type alias for events that can be enqueued."""
 
 

src/a2a/server/tasks/task_updater.py

@@ -96,7 +96,7 @@ def complete(self, message: Message | None = None):
     def failed(self, message: Message | None = None):
         """Marks the task as failed and publishes a final status update."""
         self.update_status(TaskState.failed, message=message, final=True)
-    
+
     def reject(self, message: Message | None = None):
         """Marks the task as rejected and publishes a final status update."""
         self.update_status(TaskState.rejected, message=message, final=True)

tests/client/test_client.py

@@ -1,24 +1,45 @@
 import json
+
 from collections.abc import AsyncGenerator
 from typing import Any
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import httpx
 import pytest
+
 from httpx_sse import EventSource, ServerSentEvent
-from pydantic import ValidationError as PydanticValidationError
-
-from a2a.client import (A2ACardResolver, A2AClient, A2AClientHTTPError,
-                        A2AClientJSONError, create_text_message_object)
-from a2a.types import (A2ARequest, AgentCapabilities, AgentCard, AgentSkill,
-                       CancelTaskRequest, CancelTaskResponse,
-                       CancelTaskSuccessResponse, GetTaskRequest,
-                       GetTaskResponse, InvalidParamsError,
-                       JSONRPCErrorResponse, MessageSendParams, Role,
-                       SendMessageRequest, SendMessageResponse,
-                       SendMessageSuccessResponse, SendStreamingMessageRequest,
-                       SendStreamingMessageResponse, TaskIdParams,
-                       TaskNotCancelableError, TaskQueryParams)
+
+from a2a.client import (
+    A2ACardResolver,
+    A2AClient,
+    A2AClientHTTPError,
+    A2AClientJSONError,
+    create_text_message_object,
+)
+from a2a.types import (
+    A2ARequest,
+    AgentCapabilities,
+    AgentCard,
+    AgentSkill,
+    CancelTaskRequest,
+    CancelTaskResponse,
+    CancelTaskSuccessResponse,
+    GetTaskRequest,
+    GetTaskResponse,
+    InvalidParamsError,
+    JSONRPCErrorResponse,
+    MessageSendParams,
+    Role,
+    SendMessageRequest,
+    SendMessageResponse,
+    SendMessageSuccessResponse,
+    SendStreamingMessageRequest,
+    SendStreamingMessageResponse,
+    TaskIdParams,
+    TaskNotCancelableError,
+    TaskQueryParams,
+)
+
 
 AGENT_CARD = AgentCard(
     name='Hello World Agent',
@@ -100,7 +121,9 @@ class TestA2ACardResolver:
     BASE_URL = 'http://example.com'
     AGENT_CARD_PATH = '/.well-known/agent.json'
     FULL_AGENT_CARD_URL = f'{BASE_URL}{AGENT_CARD_PATH}'
-    EXTENDED_AGENT_CARD_PATH = '/agent/authenticatedExtendedCard' # Default path
+    EXTENDED_AGENT_CARD_PATH = (
+        '/agent/authenticatedExtendedCard'  # Default path
+    )
 
     @pytest.mark.asyncio
     async def test_init_strips_slashes(self, mock_httpx_client: AsyncMock):
@@ -141,11 +164,12 @@ async def test_get_agent_card_success_public_only(
 
     @pytest.mark.asyncio
     async def test_get_agent_card_success_with_specified_path_for_extended_card(
-        self, mock_httpx_client: AsyncMock):
+        self, mock_httpx_client: AsyncMock
+    ):
         extended_card_response = AsyncMock(spec=httpx.Response)
         extended_card_response.status_code = 200
-        extended_card_response.json.return_value = AGENT_CARD_EXTENDED.model_dump(
-            mode='json'
+        extended_card_response.json.return_value = (
+            AGENT_CARD_EXTENDED.model_dump(mode='json')
         )
 
         # Mock the single call for the extended card
@@ -156,20 +180,26 @@ async def test_get_agent_card_success_with_specified_path_for_extended_card(
             base_url=self.BASE_URL,
             agent_card_path=self.AGENT_CARD_PATH,
         )
-        
+
         # Fetch the extended card by providing its relative path and example auth
-        auth_kwargs = {"headers": {"Authorization": "Bearer test token"}}
+        auth_kwargs = {'headers': {'Authorization': 'Bearer test token'}}
         agent_card_result = await resolver.get_agent_card(
             relative_card_path=self.EXTENDED_AGENT_CARD_PATH,
-            http_kwargs=auth_kwargs
+            http_kwargs=auth_kwargs,
         )
 
-        expected_extended_url = f'{self.BASE_URL}/{self.EXTENDED_AGENT_CARD_PATH.lstrip("/")}'
-        mock_httpx_client.get.assert_called_once_with(expected_extended_url, **auth_kwargs)
+        expected_extended_url = (
+            f'{self.BASE_URL}/{self.EXTENDED_AGENT_CARD_PATH.lstrip("/")}'
+        )
+        mock_httpx_client.get.assert_called_once_with(
+            expected_extended_url, **auth_kwargs
+        )
         extended_card_response.raise_for_status.assert_called_once()
 
         assert isinstance(agent_card_result, AgentCard)
-        assert agent_card_result == AGENT_CARD_EXTENDED # Should return the extended card
+        assert (
+            agent_card_result == AGENT_CARD_EXTENDED
+        )  # Should return the extended card
 
     @pytest.mark.asyncio
     async def test_get_agent_card_validation_error(
@@ -178,19 +208,29 @@ async def test_get_agent_card_validation_error(
         mock_response = AsyncMock(spec=httpx.Response)
         mock_response.status_code = 200
         # Data that will cause a Pydantic ValidationError
-        mock_response.json.return_value = {"invalid_field": "value", "name": "Test Agent"} 
+        mock_response.json.return_value = {
+            'invalid_field': 'value',
+            'name': 'Test Agent',
+        }
         mock_httpx_client.get.return_value = mock_response
 
         resolver = A2ACardResolver(
             httpx_client=mock_httpx_client, base_url=self.BASE_URL
         )
         # The call that is expected to raise an error should be within pytest.raises
         with pytest.raises(A2AClientJSONError) as exc_info:
-            await resolver.get_agent_card() # Fetches from default path
-        
-        assert f'Failed to validate agent card structure from {self.FULL_AGENT_CARD_URL}' in str(exc_info.value)
-        assert 'invalid_field' in str(exc_info.value) # Check if Pydantic error details are present
-        assert mock_httpx_client.get.call_count == 1 # Should only be called once
+            await resolver.get_agent_card()  # Fetches from default path
+
+        assert (
+            f'Failed to validate agent card structure from {self.FULL_AGENT_CARD_URL}'
+            in str(exc_info.value)
+        )
+        assert 'invalid_field' in str(
+            exc_info.value
+        )  # Check if Pydantic error details are present
+        assert (
+            mock_httpx_client.get.call_count == 1
+        )  # Should only be called once
 
     @pytest.mark.asyncio
     async def test_get_agent_card_http_status_error(
@@ -217,7 +257,10 @@ async def test_get_agent_card_http_status_error(
             await resolver.get_agent_card()
 
         assert exc_info.value.status_code == 404
-        assert f'Failed to fetch agent card from {self.FULL_AGENT_CARD_URL}' in str(exc_info.value)
+        assert (
+            f'Failed to fetch agent card from {self.FULL_AGENT_CARD_URL}'
+            in str(exc_info.value)
+        )
         assert 'Not Found' in str(exc_info.value)
         mock_httpx_client.get.assert_called_once_with(self.FULL_AGENT_CARD_URL)
 
@@ -242,7 +285,10 @@ async def test_get_agent_card_json_decode_error(
             await resolver.get_agent_card()
 
         # Assertions using exc_info must be after the with block
-        assert f'Failed to parse JSON for agent card from {self.FULL_AGENT_CARD_URL}' in str(exc_info.value)
+        assert (
+            f'Failed to parse JSON for agent card from {self.FULL_AGENT_CARD_URL}'
+            in str(exc_info.value)
+        )
         assert 'Expecting value' in str(exc_info.value)
         mock_httpx_client.get.assert_called_once_with(self.FULL_AGENT_CARD_URL)
 
@@ -263,7 +309,10 @@ async def test_get_agent_card_request_error(
             await resolver.get_agent_card()
 
         assert exc_info.value.status_code == 503
-        assert f'Network communication error fetching agent card from {self.FULL_AGENT_CARD_URL}' in str(exc_info.value)
+        assert (
+            f'Network communication error fetching agent card from {self.FULL_AGENT_CARD_URL}'
+            in str(exc_info.value)
+        )
         assert 'Network issue' in str(exc_info.value)
         mock_httpx_client.get.assert_called_once_with(self.FULL_AGENT_CARD_URL)
 

tests/server/tasks/test_task_updater.py

@@ -212,9 +212,7 @@ def test_new_agent_message(self, task_updater, sample_parts):
         assert message.parts == sample_parts
         assert message.metadata is None
 
-    def test_new_agent_message_with_metadata(
-        self, task_updater, sample_parts
-    ):
+    def test_new_agent_message_with_metadata(self, task_updater, sample_parts):
         """Test creating a new agent message with metadata and final=True."""
         metadata = {'key': 'value'}
 

tests/server/test_integration.py

@@ -1,21 +1,36 @@
 import asyncio
+
 from typing import Any
 from unittest import mock
 
 import pytest
+
 from starlette.responses import JSONResponse
 from starlette.routing import Route
 from starlette.testclient import TestClient
 
 from a2a.server.apps.starlette_app import A2AStarletteApplication
-from a2a.types import (AgentCapabilities, AgentCard, Artifact, DataPart,
-                       InternalError, InvalidRequestError, JSONParseError,
-                       Part, PushNotificationConfig, Task,
-                       TaskArtifactUpdateEvent, TaskPushNotificationConfig,
-                       TaskState, TaskStatus, TextPart,
-                       UnsupportedOperationError)
+from a2a.types import (
+    AgentCapabilities,
+    AgentCard,
+    Artifact,
+    DataPart,
+    InternalError,
+    InvalidRequestError,
+    JSONParseError,
+    Part,
+    PushNotificationConfig,
+    Task,
+    TaskArtifactUpdateEvent,
+    TaskPushNotificationConfig,
+    TaskState,
+    TaskStatus,
+    TextPart,
+    UnsupportedOperationError,
+)
 from a2a.utils.errors import MethodNotImplementedError
 
+
 # === TEST SETUP ===
 
 MINIMAL_AGENT_SKILL: dict[str, Any] = {
@@ -135,7 +150,7 @@ def test_authenticated_extended_agent_card_endpoint_not_supported(
     # So, building the app and trying to hit it should result in 404 from Starlette itself
     client = TestClient(app_instance.build())
     response = client.get('/agent/authenticatedExtendedCard')
-    assert response.status_code == 404 # Starlette's default for no route
+    assert response.status_code == 404  # Starlette's default for no route
 
 
 def test_authenticated_extended_agent_card_endpoint_supported_with_specific_extended_card(
@@ -144,7 +159,9 @@ def test_authenticated_extended_agent_card_endpoint_supported_with_specific_exte
     handler: mock.AsyncMock,
 ):
     """Test extended card endpoint returns the specific extended card when provided."""
-    agent_card.supportsAuthenticatedExtendedCard = True # Main card must support it
+    agent_card.supportsAuthenticatedExtendedCard = (
+        True  # Main card must support it
+    )
     app_instance = A2AStarletteApplication(
         agent_card, handler, extended_agent_card=extended_agent_card_fixture
     )
@@ -157,10 +174,9 @@ def test_authenticated_extended_agent_card_endpoint_supported_with_specific_exte
     assert data['name'] == extended_agent_card_fixture.name
     assert data['version'] == extended_agent_card_fixture.version
     assert len(data['skills']) == len(extended_agent_card_fixture.skills)
-    assert any(
-        skill['id'] == 'skill-extended' for skill in data['skills']
-    ), "Extended skill not found in served card"
-
+    assert any(skill['id'] == 'skill-extended' for skill in data['skills']), (
+        'Extended skill not found in served card'
+    )
 
 
 def test_agent_card_custom_url(

tests/test_types.py

@@ -1491,8 +1491,8 @@ def test_a2a_error_validation_and_serialization() -> None:
 
 def test_subclass_enums() -> None:
     """validate subtype enum types"""
-    assert "cookie" == In.cookie
+    assert In.cookie == 'cookie'
 
-    assert "user" == Role.user
+    assert Role.user == 'user'
 
-    assert "working" == TaskState.working
+    assert TaskState.working == 'working'