Lint fixes

Author: holtskinner

examples/google_adk/calendar_agent/__main__.py

@@ -36,12 +36,15 @@
 @click.option('--host', 'host', default='localhost')
 @click.option('--port', 'port', default=10007)
 def main(host: str, port: int):
-    # Verify an API key is set. Not required if using Vertex AI APIs, since those can use gcloud credentials.
-    if not os.getenv('GOOGLE_GENAI_USE_VERTEXAI') == 'TRUE':
-        if not os.getenv('GOOGLE_API_KEY'):
-            raise Exception(
-                'GOOGLE_API_KEY environment variable not set and GOOGLE_GENAI_USE_VERTEXAI is not TRUE.'
-            )
+    # Verify an API key is set.
+    # Not required if using Vertex AI APIs.
+    if os.getenv('GOOGLE_GENAI_USE_VERTEXAI') != 'TRUE' and not os.getenv(
+        'GOOGLE_API_KEY'
+    ):
+        raise ValueError(
+            'GOOGLE_API_KEY environment variable not set and '
+            'GOOGLE_GENAI_USE_VERTEXAI is not TRUE.'
+        )
 
     skill = AgentSkill(
         id='check_availability',

examples/google_adk/calendar_agent/adk_agent_executor.py

@@ -63,7 +63,7 @@ async def _process_request(
         new_message: types.Content,
         session_id: str,
         task_updater: TaskUpdater,
-    ):
+    ) -> None:
         session_id = self._upsert_session(
             session_id,
         ).id
@@ -160,7 +160,7 @@ async def _complete_auth_processing(
         session_id: str,
         auth_details: ADKAuthDetails,
         task_updater: TaskUpdater,
-    ):
+    ) -> None:
         logger.debug('Waiting for auth event')
         try:
             auth_uri = await asyncio.wait_for(
@@ -296,7 +296,7 @@ def convert_genai_part_to_a2a(part: types.Part) -> Part:
 
 
 def get_auth_request_function_call(event: Event) -> types.FunctionCall:
-    """Get the special auth request function call from the event"""
+    """Get the special auth request function call from the event."""
     if not (event.content and event.content.parts):
         return None
     for part in event.content.parts:
@@ -308,12 +308,13 @@ def get_auth_request_function_call(event: Event) -> types.FunctionCall:
             and part.function_call.id in event.long_running_tool_ids
         ):
             return part.function_call
+    return None
 
 
 def get_auth_config(
     auth_request_function_call: types.FunctionCall,
 ) -> AuthConfig:
-    """Extracts the AuthConfig object from the arguments of the auth request function call"""
+    """Extracts the AuthConfig object from the arguments of the auth request function call."""
     if not auth_request_function_call.args or not (
         auth_config := auth_request_function_call.args.get('auth_config')
     ):

src/a2a/server/agent_execution/context.py

@@ -19,8 +19,10 @@ def __init__(
         task_id: str | None = None,
         context_id: str | None = None,
         task: Task | None = None,
-        related_tasks: list[Task] = [],
+        related_tasks: list[Task] = None,
     ):
+        if related_tasks is None:
+            related_tasks = []
         self._params = request
         self._task_id = task_id
         self._context_id = context_id