Formatting/Linting

Author: holtskinner

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

@@ -12,9 +12,10 @@
 from sse_starlette.sse import EventSourceResponse
 from starlette.applications import Starlette
 from starlette.authentication import BaseUser
+from starlette.exceptions import HTTPException
 from starlette.requests import Request
 from starlette.responses import JSONResponse, Response
-from starlette.exceptions import HTTPException
+from starlette.status import HTTP_413_REQUEST_ENTITY_TOO_LARGE
 
 from a2a.auth.user import UnauthenticatedUser
 from a2a.auth.user import User as A2AUser
@@ -234,10 +235,12 @@ async def _handle_requests(self, request: Request) -> Response:
                 A2AError(root=InvalidRequestError(data=json.loads(e.json()))),
             )
         except HTTPException as e:
-            if e.status_code == 413: # Payload Too Large
-                 return self._generate_error_response(
+            if e.status_code == HTTP_413_REQUEST_ENTITY_TOO_LARGE:
+                return self._generate_error_response(
                     request_id,
-                    A2AError(root=InvalidRequestError(message="Payload too large")),
+                    A2AError(
+                        root=InvalidRequestError(message='Payload too large')
+                    ),
                 )
             raise e
         except Exception as e:

tests/server/apps/jsonrpc/test_serialization.py

@@ -1,7 +1,8 @@
-import json
 from unittest import mock
 
 import pytest
+
+from pydantic import ValidationError
 from starlette.testclient import TestClient
 
 from a2a.server.apps import A2AFastAPIApplication, A2AStarletteApplication
@@ -10,15 +11,14 @@
     AgentCapabilities,
     AgentCard,
     In,
+    InvalidRequestError,
+    JSONParseError,
     Message,
     Part,
     Role,
     SecurityScheme,
     TextPart,
-    JSONParseError,
-    InvalidRequestError,
 )
-from pydantic import ValidationError
 
 
 @pytest.fixture
@@ -107,7 +107,10 @@ def test_handle_invalid_json(agent_card_with_api_key: AgentCard):
     app_instance = A2AStarletteApplication(agent_card_with_api_key, handler)
     client = TestClient(app_instance.build())
 
-    response = client.post('/', content='{ "jsonrpc": "2.0", "method": "test", "id": 1, "params": { "key": "value" }')
+    response = client.post(
+        '/',
+        content='{ "jsonrpc": "2.0", "method": "test", "id": 1, "params": { "key": "value" }',
+    )
     assert response.status_code == 200
     data = response.json()
     assert data['error']['code'] == JSONParseError().code
@@ -119,12 +122,12 @@ def test_handle_oversized_payload(agent_card_with_api_key: AgentCard):
     app_instance = A2AStarletteApplication(agent_card_with_api_key, handler)
     client = TestClient(app_instance.build())
 
-    large_string = "a" * 2_000_000  # 2MB string
+    large_string = 'a' * 2_000_000  # 2MB string
     payload = {
-        "jsonrpc": "2.0",
-        "method": "test",
-        "id": 1,
-        "params": {"data": large_string},
+        'jsonrpc': '2.0',
+        'method': 'test',
+        'id': 1,
+        'params': {'data': large_string},
     }
 
     # Starlette/FastAPI's default max request size is around 1MB.
@@ -139,7 +142,7 @@ def test_handle_oversized_payload(agent_card_with_api_key: AgentCard):
             data = response.json()
             assert data['error']['code'] == InvalidRequestError().code
         else:
-             assert response.status_code == 413
+            assert response.status_code == 413
     except Exception as e:
         # Depending on server setup, it might just drop the connection for very large payloads
         assert isinstance(e, (ConnectionResetError, RuntimeError))
@@ -151,25 +154,25 @@ def test_handle_unicode_characters(agent_card_with_api_key: AgentCard):
     app_instance = A2AStarletteApplication(agent_card_with_api_key, handler)
     client = TestClient(app_instance.build())
 
-    unicode_text = "こんにちは世界" # "Hello world" in Japanese
+    unicode_text = 'こんにちは世界'  # "Hello world" in Japanese
     unicode_payload = {
-        "jsonrpc": "2.0",
-        "method": "message/send",
-        "id": "unicode_test",
-        "params": {
-            "message": {
-                "role": "user",
-                "parts": [{"kind": "text", "text": unicode_text}],
-                "messageId": "msg-unicode"
+        'jsonrpc': '2.0',
+        'method': 'message/send',
+        'id': 'unicode_test',
+        'params': {
+            'message': {
+                'role': 'user',
+                'parts': [{'kind': 'text', 'text': unicode_text}],
+                'messageId': 'msg-unicode',
             }
-        }
+        },
     }
 
     # Mock a handler for this method
     handler.on_message_send.return_value = Message(
         role=Role.agent,
-        parts=[Part(root=TextPart(text=f"Received: {unicode_text}"))],
-        messageId="response-unicode"
+        parts=[Part(root=TextPart(text=f'Received: {unicode_text}'))],
+        messageId='response-unicode',
     )
 
     response = client.post('/', json=unicode_payload)
@@ -180,4 +183,4 @@ def test_handle_unicode_characters(agent_card_with_api_key: AgentCard):
     assert response.status_code == 200
     data = response.json()
     assert 'error' not in data or data['error'] is None
-    assert data['result']['parts'][0]['text'] == f"Received: {unicode_text}"
+    assert data['result']['parts'][0]['text'] == f'Received: {unicode_text}'