style: apply auto-formatting

Author: theailifestyle

examples/google_adk/french_translation_agent/adk_agent_executor.py

@@ -3,15 +3,19 @@
 import logging
 
 from collections.abc import AsyncGenerator, AsyncIterable
+from typing import Any
+from uuid import uuid4
 
-from adk_agent import create_french_translation_agent
 from google.adk import Runner
+from google.adk.agents import LlmAgent, RunConfig
 from google.adk.artifacts import InMemoryArtifactService
 from google.adk.events import Event
 from google.adk.memory.in_memory_memory_service import InMemoryMemoryService
 from google.adk.sessions import InMemorySessionService
 from google.genai import types as genai_types
+from pydantic import ConfigDict
 
+from a2a.client import A2AClient
 from a2a.server.agent_execution import AgentExecutor, RequestContext
 from a2a.server.events.event_queue import EventQueue
 from a2a.server.tasks import TaskUpdater
@@ -20,14 +24,26 @@
     FilePart,
     FileWithBytes,
     FileWithUri,
+    GetTaskRequest,
+    GetTaskSuccessResponse,
+    Message,
+    MessageSendParams,
     Part,
+    Role,
+    SendMessageRequest,
+    SendMessageSuccessResponse,
+    Task,
+    TaskQueryParams,
     TaskState,
     TaskStatus,
     TextPart,
     UnsupportedOperationError,
 )
+from a2a.utils import get_text_parts
 from a2a.utils.errors import ServerError
 
+from adk_agent import create_french_translation_agent
+
 
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.DEBUG)
@@ -51,12 +67,12 @@ def _run_agent(
         self,
         session_id: str,
         new_message: genai_types.Content,
-        task_updater: TaskUpdater,  # This parameter is not used in this method.
+        task_updater: TaskUpdater, # This parameter is not used in this method.
     ) -> AsyncGenerator[Event, None]:
         """Runs the ADK agent with the given message."""
         return self.runner.run_async(
             session_id=session_id,
-            user_id='self',  # The user ID for the ADK session.
+            user_id='self', # The user ID for the ADK session.
             new_message=new_message,
         )
 
@@ -72,9 +88,7 @@ async def _process_request(
         )
         session_id = session.id
         async for event in self._run_agent(
-            session_id,
-            new_message,
-            task_updater,  # Pass task_updater to _run_agent
+            session_id, new_message, task_updater # Pass task_updater to _run_agent
         ):
             logger.debug('Received ADK event: %s', event)
             if event.is_final_response():
@@ -84,7 +98,7 @@ async def _process_request(
                 task_updater.add_artifact(response)
                 task_updater.complete()
                 break
-            if not event.get_function_calls():
+            elif not event.get_function_calls():
                 # If it's not a final response and no function calls, it's an interim update.
                 logger.debug('Yielding update response')
                 task_updater.update_status(
@@ -95,10 +109,8 @@ async def _process_request(
                 )
             else:
                 # This agent does not use tools, so function calls are unexpected.
-                logger.debug(
-                    'Skipping event with function call: %s',
-                    event.get_function_calls(),
-                )
+                logger.debug('Skipping event with function call: %s', event.get_function_calls())
+
 
     async def execute(
         self,
@@ -131,12 +143,12 @@ async def _upsert_session(self, session_id: str):
 
 
 def convert_a2a_parts_to_genai(parts: list[Part]) -> list[genai_types.Part]:
-    """Converts a list of A2A Part objects to a list of Google GenAI Part objects."""
+    """Converts a list of A2A Part objects to a list of Google Gen AI Part objects."""
     return [convert_a2a_part_to_genai(part) for part in parts]
 
 
 def convert_a2a_part_to_genai(part: Part) -> genai_types.Part:
-    """Converts a single A2A Part object to a Google GenAI Part object."""
+    """Converts a single A2A Part object to a Google Gen AI Part object."""
     part = part.root
     if isinstance(part, TextPart):
         return genai_types.Part(text=part.text)
@@ -158,7 +170,7 @@ def convert_a2a_part_to_genai(part: Part) -> genai_types.Part:
 
 
 def convert_genai_parts_to_a2a(parts: list[genai_types.Part]) -> list[Part]:
-    """Converts a list of Google GenAI Part objects to a list of A2A Part objects."""
+    """Converts a list of Google Gen AI Part objects to a list of A2A Part objects."""
     return [
         convert_genai_part_to_a2a(part)
         for part in parts
@@ -167,7 +179,7 @@ def convert_genai_parts_to_a2a(parts: list[genai_types.Part]) -> list[Part]:
 
 
 def convert_genai_part_to_a2a(part: genai_types.Part) -> Part:
-    """Converts a single Google GenAI Part object to an A2A Part object."""
+    """Converts a single Google Gen AI Part object to an A2A Part object."""
     if part.text:
         return TextPart(text=part.text)
     if part.file_data: