chore: Ran formatting

Author: holtskinner

examples/google_adk/calendar_agent/__main__.py

@@ -1,31 +1,31 @@
 import logging
 import os
-import sys
 
 import click
 import uvicorn
 
-from adk_agent_executor import ADKAgentExecutor
 from adk_agent import create_agent
+from adk_agent_executor import ADKAgentExecutor
 from dotenv import load_dotenv
+from google.adk.artifacts import InMemoryArtifactService
+from google.adk.memory.in_memory_memory_service import InMemoryMemoryService
+from google.adk.runners import Runner
+from google.adk.sessions import InMemorySessionService
+from starlette.applications import Starlette
+from starlette.requests import Request
+from starlette.responses import PlainTextResponse
+from starlette.routing import Route
 
-from a2a.server.tasks import InMemoryTaskStore
 from a2a.server.apps import A2AStarletteApplication
 from a2a.server.request_handlers import DefaultRequestHandler
+from a2a.server.tasks import InMemoryTaskStore
 from a2a.types import (
     AgentAuthentication,
     AgentCapabilities,
     AgentCard,
     AgentSkill,
 )
-from google.adk.artifacts import InMemoryArtifactService
-from google.adk.memory.in_memory_memory_service import InMemoryMemoryService
-from google.adk.runners import Runner
-from google.adk.sessions import InMemorySessionService
-from starlette.applications import Starlette
-from starlette.routing import Route
-from starlette.requests import Request
-from starlette.responses import PlainTextResponse
+
 
 load_dotenv()
 

examples/google_adk/calendar_agent/adk_agent_executor.py

@@ -1,8 +1,8 @@
 import asyncio
 import logging
+
 from collections import namedtuple
 from collections.abc import AsyncGenerator
-from typing import Dict
 from urllib.parse import parse_qs, urlparse
 
 from google.adk import Runner
@@ -27,6 +27,7 @@
 from a2a.utils.errors import ServerError
 from a2a.utils.message import new_agent_text_message
 
+
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.DEBUG)
 
@@ -42,7 +43,7 @@
 class ADKAgentExecutor(AgentExecutor):
     """An AgentExecutor that runs an ADK-based Agent."""
 
-    _awaiting_auth: Dict[str, asyncio.Future]
+    _awaiting_auth: dict[str, asyncio.Future]
 
     def __init__(self, runner: Runner, card: AgentCard):
         self.runner = runner
@@ -95,13 +96,13 @@ async def _process_request(
                 # Break out of event handling loop -- no more work will be done until the authorization
                 # is received.
                 break
-            elif event.is_final_response():
+            if event.is_final_response():
                 parts = convert_genai_parts_to_a2a(event.content.parts)
                 logger.debug('Yielding final response: %s', parts)
                 task_updater.add_artifact(parts)
                 task_updater.complete()
                 break
-            elif not event.get_function_calls():
+            if not event.get_function_calls():
                 logger.debug('Yielding update response')
                 task_updater.update_status(
                     TaskState.working,
@@ -245,23 +246,21 @@ def convert_a2a_part_to_genai(part: Part) -> types.Part:
     part = part.root
     if isinstance(part, TextPart):
         return types.Part(text=part.text)
-    elif isinstance(part, FilePart):
+    if isinstance(part, FilePart):
         if isinstance(part.file, FileWithUri):
             return types.Part(
                 file_data=types.FileData(
                     file_uri=part.file.uri, mime_type=part.file.mime_type
                 )
             )
-        elif isinstance(part.file, FileWithBytes):
+        if isinstance(part.file, FileWithBytes):
             return types.Part(
                 inline_data=types.Blob(
                     data=part.file.bytes, mime_type=part.file.mime_type
                 )
             )
-        else:
-            raise ValueError(f'Unsupported file type: {type(part.file)}')
-    else:
-        raise ValueError(f'Unsupported part type: {type(part)}')
+        raise ValueError(f'Unsupported file type: {type(part.file)}')
+    raise ValueError(f'Unsupported part type: {type(part)}')
 
 
 def convert_genai_parts_to_a2a(parts: list[types.Part]) -> list[Part]:
@@ -277,14 +276,14 @@ def convert_genai_part_to_a2a(part: types.Part) -> Part:
     """Convert a single Google GenAI Part type into an A2A Part type."""
     if part.text:
         return TextPart(text=part.text)
-    elif part.file_data:
+    if part.file_data:
         return FilePart(
             file=FileWithUri(
                 uri=part.file_data.file_uri,
                 mime_type=part.file_data.mime_type,
             )
         )
-    elif part.inline_data:
+    if part.inline_data:
         return Part(
             root=FilePart(
                 file=FileWithBytes(
@@ -293,14 +292,13 @@ def convert_genai_part_to_a2a(part: types.Part) -> Part:
                 )
             )
         )
-    else:
-        raise ValueError(f'Unsupported part type: {part}')
+    raise ValueError(f'Unsupported part type: {part}')
 
 
 def get_auth_request_function_call(event: Event) -> types.FunctionCall:
     """Get the special auth request function call from the event"""
     if not (event.content and event.content.parts):
-        return
+        return None
     for part in event.content.parts:
         if (
             part

examples/helloworld/agent_executor.py

@@ -2,9 +2,6 @@
 
 from a2a.server.agent_execution import AgentExecutor, RequestContext
 from a2a.server.events import EventQueue
-from a2a.types import (
-    Task,
-)
 from a2a.utils import new_agent_text_message
 
 

src/a2a/server/agent_execution/__init__.py

@@ -1,4 +1,5 @@
 from a2a.server.agent_execution.agent_executor import AgentExecutor
 from a2a.server.agent_execution.context import RequestContext
 
+
 __all__ = ['AgentExecutor', 'RequestContext']

src/a2a/server/agent_execution/context.py

@@ -5,7 +5,6 @@
     Message,
     MessageSendParams,
     Task,
-    TextPart,
 )
 from a2a.utils import get_message_text
 from a2a.utils.errors import ServerError

src/a2a/server/apps/__init__.py

@@ -2,4 +2,4 @@
 from a2a.server.apps.starlette_app import A2AStarletteApplication
 
 
-__all__ = ['HttpApp', 'A2AStarletteApplication']
+__all__ = ['A2AStarletteApplication', 'HttpApp']

src/a2a/server/apps/starlette_app.py

@@ -1,9 +1,17 @@
-from collections.abc import AsyncGenerator
 import json
 import logging
 import traceback
+
+from collections.abc import AsyncGenerator
 from typing import Any
 
+from pydantic import ValidationError
+from sse_starlette.sse import EventSourceResponse
+from starlette.applications import Starlette
+from starlette.requests import Request
+from starlette.responses import JSONResponse, Response
+from starlette.routing import Route
+
 from a2a.server.request_handlers.jsonrpc_handler import (
     JSONRPCHandler,
     RequestHandler,
@@ -29,12 +37,6 @@
     UnsupportedOperationError,
 )
 from a2a.utils.errors import MethodNotImplementedError
-from pydantic import ValidationError
-from sse_starlette.sse import EventSourceResponse
-from starlette.applications import Starlette
-from starlette.requests import Request
-from starlette.responses import JSONResponse, Response
-from starlette.routing import Route
 
 
 logger = logging.getLogger(__name__)
@@ -116,7 +118,7 @@ async def _handle_requests(self, request: Request) -> Response:
             return await self._process_non_streaming_request(
                 request_id, a2a_request
             )
-        except MethodNotImplementedError as e:
+        except MethodNotImplementedError:
             traceback.print_exc()
             return self._generate_error_response(
                 request_id, A2AError(root=UnsupportedOperationError())

src/a2a/server/events/__init__.py

@@ -1,14 +1,19 @@
 from a2a.server.events.event_consumer import EventConsumer
 from a2a.server.events.event_queue import Event, EventQueue
-from a2a.server.events.queue_manager import QueueManager, TaskQueueExists, NoTaskQueue
 from a2a.server.events.in_memory_queue_manager import InMemoryQueueManager
+from a2a.server.events.queue_manager import (
+    NoTaskQueue,
+    QueueManager,
+    TaskQueueExists,
+)
+
 
 __all__ = [
     'Event',
     'EventConsumer',
     'EventQueue',
+    'InMemoryQueueManager',
+    'NoTaskQueue',
     'QueueManager',
     'TaskQueueExists',
-    'NoTaskQueue',
-    'InMemoryQueueManager',
 ]

src/a2a/server/events/in_memory_queue_manager.py

@@ -1,6 +1,12 @@
 import asyncio
+
 from a2a.server.events.event_queue import EventQueue
-from a2a.server.events.queue_manager import QueueManager, TaskQueueExists, NoTaskQueue
+from a2a.server.events.queue_manager import (
+    NoTaskQueue,
+    QueueManager,
+    TaskQueueExists,
+)
+
 
 class InMemoryQueueManager(QueueManager):
     """InMemoryQueueManager is used for a single binary management.

src/a2a/server/events/queue_manager.py

@@ -1,4 +1,5 @@
 from abc import ABC, abstractmethod
+
 from a2a.server.events.event_queue import EventQueue
 
 
@@ -29,5 +30,6 @@ async def create_or_tap(self, task_id: str) -> EventQueue:
 class TaskQueueExists(Exception):
     pass
 
+
 class NoTaskQueue(Exception):
     pass