Codex: sync local edits in adk-python

Author: discopops

contributing/samples/agent_registry_agent/agent.py

@@ -33,17 +33,19 @@
 print(f"Listing agents in {project_id}/{location}...")
 agents = registry.list_agents()
 for agent in agents.get("agents", []):
-  print(f"- Agent: {agent.get('displayName')} ({agent.get('name')})")
+    print(f"- Agent: {agent.get('displayName')} ({agent.get('name')})")
 
 print(f"\nListing MCP servers in {project_id}/{location}...")
 mcp_servers = registry.list_mcp_servers()
 for server in mcp_servers.get("mcpServers", []):
-  print(f"- MCP Server: {server.get('displayName')} ({server.get('name')})")
+    print(f"- MCP Server: {server.get('displayName')} ({server.get('name')})")
 
 print(f"\nListing endpoints in {project_id}/{location}...")
 endpoints = registry.list_endpoints()
 for endpoint in endpoints.get("endpoints", []):
-  print(f"- Endpoint: {endpoint.get('displayName')} ({endpoint.get('name')})")
+    print(
+        f"- Endpoint: {endpoint.get('displayName')} ({endpoint.get('name')})"
+    )
 
 # Example of using a specific agent or MCP server from the registry:
 # (Note: These names should be full resource names as returned by list methods)

contributing/samples/live_bidi_streaming_parallel_tools_agent/agent.py

@@ -17,15 +17,15 @@
 
 
 def turn_on_lights():
-  """Turn on the lights."""
-  print("turn_on_lights")
-  return {"status": "OK"}
+    """Turn on the lights."""
+    print("turn_on_lights")
+    return {"status": "OK"}
 
 
 def turn_on_tv():
-  """Turn on the tv."""
-  print("turn_on_tv")
-  return {"status": "OK"}
+    """Turn on the tv."""
+    print("turn_on_tv")
+    return {"status": "OK"}
 
 
 root_agent = Agent(

contributing/samples/skills_agent/agent.py

@@ -17,7 +17,9 @@
 import pathlib
 
 from google.adk import Agent
-from google.adk.code_executors.unsafe_local_code_executor import UnsafeLocalCodeExecutor
+from google.adk.code_executors.unsafe_local_code_executor import (
+    UnsafeLocalCodeExecutor,
+)
 from google.adk.skills import load_skill_from_dir
 from google.adk.skills import models
 from google.adk.tools.base_tool import BaseTool
@@ -26,37 +28,37 @@
 
 
 class GetTimezoneTool(BaseTool):
-  """A tool to get the timezone for a given location."""
+    """A tool to get the timezone for a given location."""
 
-  def __init__(self):
-    super().__init__(
-        name="get_timezone",
-        description="Returns the timezone for a given location.",
-    )
+    def __init__(self):
+        super().__init__(
+            name="get_timezone",
+            description="Returns the timezone for a given location.",
+        )
 
-  def _get_declaration(self) -> types.FunctionDeclaration | None:
-    return types.FunctionDeclaration(
-        name=self.name,
-        description=self.description,
-        parameters_json_schema={
-            "type": "object",
-            "properties": {
-                "location": {
-                    "type": "string",
-                    "description": "The location to get the timezone for.",
+    def _get_declaration(self) -> types.FunctionDeclaration | None:
+        return types.FunctionDeclaration(
+            name=self.name,
+            description=self.description,
+            parameters_json_schema={
+                "type": "object",
+                "properties": {
+                    "location": {
+                        "type": "string",
+                        "description": "The location to get the timezone for.",
+                    },
                 },
+                "required": ["location"],
             },
-            "required": ["location"],
-        },
-    )
+        )
 
-  async def run_async(self, *, args: dict, tool_context) -> str:
-    return f"The timezone for {args['location']} is UTC+00:00."
+    async def run_async(self, *, args: dict, tool_context) -> str:
+        return f"The timezone for {args['location']} is UTC+00:00."
 
 
 def get_wind_speed(location: str) -> str:
-  """Returns the current wind speed for a given location."""
-  return f"The wind speed in {location} is 10 mph."
+    """Returns the current wind speed for a given location."""
+    return f"The wind speed in {location} is 10 mph."
 
 
 greeting_skill = models.Skill(

src/google/adk/a2a/agent/config.py

@@ -43,82 +43,82 @@
 
 
 class ParametersConfig(BaseModel):
-  """Configuration for the parameters passed to the A2A send_message request."""
+    """Configuration for the parameters passed to the A2A send_message request."""
 
-  request_metadata: Optional[dict[str, Any]] = None
-  client_call_context: Optional[ClientCallContext] = None
-  # TODO: Add support for requested_extension and
-  # message_send_configuration once they are supported by the A2A client.
-  #
-  # requested_extension: Optional[list[str]] = None
-  # message_send_configuration: Optional[MessageSendConfiguration] = None
+    request_metadata: Optional[dict[str, Any]] = None
+    client_call_context: Optional[ClientCallContext] = None
+    # TODO: Add support for requested_extension and
+    # message_send_configuration once they are supported by the A2A client.
+    #
+    # requested_extension: Optional[list[str]] = None
+    # message_send_configuration: Optional[MessageSendConfiguration] = None
 
 
 class RequestInterceptor(BaseModel):
-  """Interceptor for A2A requests."""
+    """Interceptor for A2A requests."""
 
-  before_request: Optional[
-      Callable[
-          [InvocationContext, A2AMessage, ParametersConfig],
-          Awaitable[tuple[Union[A2AMessage, Event], ParametersConfig]],
-      ]
-  ] = None
-  """Hook executed before the agent starts processing the request.
+    before_request: Optional[
+        Callable[
+            [InvocationContext, A2AMessage, ParametersConfig],
+            Awaitable[tuple[Union[A2AMessage, Event], ParametersConfig]],
+        ]
+    ] = None
+    """Hook executed before the agent starts processing the request.
 
     Returns an Event if the request should be aborted and the Event
     returned to the caller.
   """
 
-  after_request: Optional[
-      Callable[
-          [InvocationContext, A2AEvent, Event], Awaitable[Union[Event, None]]
-      ]
-  ] = None
-  """Hook executed after the agent has processed the request.
+    after_request: Optional[
+        Callable[
+            [InvocationContext, A2AEvent, Event], Awaitable[Union[Event, None]]
+        ]
+    ] = None
+    """Hook executed after the agent has processed the request.
 
     Returns None if the event should not be sent to the caller.
   """
 
 
 class A2aRemoteAgentConfig(BaseModel):
-  """Configuration for A2A remote agents."""
-
-  # Converts standard A2A Messages into ADK Event.
-  a2a_message_converter: A2AMessageToEventConverter = (
-      convert_a2a_message_to_event
-  )
-
-  # Converts an A2A Task into an ADK Event.
-  a2a_task_converter: A2ATaskToEventConverter = convert_a2a_task_to_event
-
-  # Converts A2A TaskStatusUpdateEvents into ADK Event.
-  a2a_status_update_converter: A2AStatusUpdateToEventConverter = (
-      convert_a2a_status_update_to_event
-  )
-
-  # Converts A2A TaskArtifactUpdateEvents into ADK Event.
-  a2a_artifact_update_converter: A2AArtifactUpdateToEventConverter = (
-      convert_a2a_artifact_update_to_event
-  )
-
-  # A low-level hook that converts individual A2A Message Parts
-  # into native ADK/GenAI Part objects.
-  # This is utilized internally by the other converters.
-  a2a_part_converter: A2APartToGenAIPartConverter = (
-      convert_a2a_part_to_genai_part
-  )
-
-  request_interceptors: Optional[list[RequestInterceptor]] = None
-
-  def __deepcopy__(self, memo):
-    cls = self.__class__
-    copied_values = {}
-    for k, v in self.__dict__.items():
-      if not k.startswith('_'):
-        if callable(v):
-          copied_values[k] = v
-        else:
-          copied_values[k] = copy.deepcopy(v, memo)
-    result = cls.model_construct(**copied_values)
-    memo[id(self)] = result
-    return result
+    """Configuration for A2A remote agents."""
+
+    # Converts standard A2A Messages into ADK Event.
+    a2a_message_converter: A2AMessageToEventConverter = (
+        convert_a2a_message_to_event
+    )
+
+    # Converts an A2A Task into an ADK Event.
+    a2a_task_converter: A2ATaskToEventConverter = convert_a2a_task_to_event
+
+    # Converts A2A TaskStatusUpdateEvents into ADK Event.
+    a2a_status_update_converter: A2AStatusUpdateToEventConverter = (
+        convert_a2a_status_update_to_event
+    )
+
+    # Converts A2A TaskArtifactUpdateEvents into ADK Event.
+    a2a_artifact_update_converter: A2AArtifactUpdateToEventConverter = (
+        convert_a2a_artifact_update_to_event
+    )
+
+    # A low-level hook that converts individual A2A Message Parts
+    # into native ADK/GenAI Part objects.
+    # This is utilized internally by the other converters.
+    a2a_part_converter: A2APartToGenAIPartConverter = (
+        convert_a2a_part_to_genai_part
+    )
+
+    request_interceptors: Optional[list[RequestInterceptor]] = None
+
+    def __deepcopy__(self, memo):
+        cls = self.__class__
+        copied_values = {}
+        for k, v in self.__dict__.items():
+            if not k.startswith("_"):
+                if callable(v):
+                    copied_values[k] = v
+                else:
+                    copied_values[k] = copy.deepcopy(v, memo)
+        result = cls.model_construct(**copied_values)
+        memo[id(self)] = result
+        return result