Merge branch 'main' into copybara/891919371

Author: rohityan

contributing/samples/agent_registry_agent/agent.py

@@ -18,6 +18,7 @@
 
 from google.adk.agents.llm_agent import LlmAgent
 from google.adk.integrations.agent_registry import AgentRegistry
+from google.adk.models.google_llm import Gemini
 
 # Project and location can be set via environment variables:
 # GOOGLE_CLOUD_PROJECT and GOOGLE_CLOUD_LOCATION
@@ -27,6 +28,8 @@
 # Initialize Agent Registry client
 registry = AgentRegistry(project_id=project_id, location=location)
 
+# List agents, MCP servers, and endpoints resource names from the registry.
+# They can be used to initialize the agent, toolset, and model below.
 print(f"Listing agents in {project_id}/{location}...")
 agents = registry.list_agents()
 for agent in agents.get("agents", []):
@@ -37,6 +40,11 @@
 for server in mcp_servers.get("mcpServers", []):
   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')})")
+
 # 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)
 
@@ -52,8 +60,19 @@
     f"projects/{project_id}/locations/{location}/mcpServers/MCP_SERVER_NAME"
 )
 
+# 3. Getting a specific model endpoint configuration
+# This returns a string like:
+# "projects/adk12345/locations/us-central1/publishers/google/models/gemini-2.5-flash"
+# TODO: Replace ENDPOINT_NAME with your endpoint name
+model_name = registry.get_model_name(
+    f"projects/{project_id}/locations/{location}/endpoints/ENDPOINT_NAME"
+)
+
+# Initialize the model using the resolved model name from registry.
+gemini_model = Gemini(model=model_name)
+
 root_agent = LlmAgent(
-    model="gemini-2.5-flash",
+    model=gemini_model,
     name="discovery_agent",
     instruction=(
         "You have access to tools and sub-agents discovered via Registry."

contributing/samples/live_bidi_streaming_parallel_tools_agent/README.md

@@ -0,0 +1,38 @@
+# Simple Live (Bidi-Streaming) Agent with Parallel Tools
+This project provides a basic example of a live, [bidirectional streaming](https://google.github.io/adk-docs/streaming/) agent that demonstrates parallel tool execution.
+
+## Getting Started
+
+Follow these steps to get the agent up and running:
+
+1.  **Start the ADK Web Server**
+    Open your terminal, navigate to the root directory that contains the
+    `live_bidi_streaming_parallel_tools_agent` folder, and execute the following
+    command:
+    ```bash
+    adk web
+    ```
+
+2.  **Access the ADK Web UI**
+    Once the server is running, open your web browser and navigate to the URL 
+    provided in the terminal (it will typically be `http://localhost:8000`).
+
+3.  **Select the Agent**
+    In the top-left corner of the ADK Web UI, use the dropdown menu to select 
+    this agent (`live_bidi_streaming_parallel_tools_agent`).
+
+4.  **Start Streaming**
+    Click on the **Audio** icon located near the chat input 
+    box to begin the streaming session.
+
+5.  **Interact with the Agent**
+    You can now begin talking to the agent, and it will respond in real-time.
+    Try asking it to perform multiple actions at once, for example: "Turn on the
+    lights and the TV at the same time." The agent will be able to invoke both
+    `turn_on_lights` and `turn_on_tv` tools in parallel.
+
+## Usage Notes
+
+* You only need to click the **Audio** button once to initiate the
+ stream. The current version does not support stopping and restarting the stream
+  by clicking the button again during a session.

contributing/samples/live_bidi_streaming_parallel_tools_agent/__init__.py

@@ -0,0 +1,15 @@
+# Copyright 2026 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from . import agent

contributing/samples/live_bidi_streaming_parallel_tools_agent/agent.py

@@ -0,0 +1,36 @@
+# Copyright 2026 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+from google.adk.agents.llm_agent import Agent
+
+
+def turn_on_lights():
+  """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"}
+
+
+root_agent = Agent(
+    model="gemini-live-2.5-flash-native-audio",
+    name="Home_helper",
+    instruction="Be polite and answer all user's questions.",
+    tools=[turn_on_lights, turn_on_tv],
+)

contributing/samples/skills_agent/agent.py

@@ -54,9 +54,9 @@ async def run_async(self, *, args: dict, tool_context) -> str:
     return f"The timezone for {args['location']} is UTC+00:00."
 
 
-def get_current_humidity(location: str) -> str:
-  """Returns the current humidity for a given location."""
-  return f"The humidity in {location} is 45%."
+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."
 
 
 greeting_skill = models.Skill(
@@ -87,7 +87,7 @@ def get_current_humidity(location: str) -> str:
 # be used in production environments.
 my_skill_toolset = SkillToolset(
     skills=[greeting_skill, weather_skill],
-    additional_tools=[GetTimezoneTool(), get_current_humidity],
+    additional_tools=[GetTimezoneTool(), get_wind_speed],
     code_executor=UnsafeLocalCodeExecutor(),
 )
 

contributing/samples/skills_agent/skills/weather-skill/SKILL.md

@@ -1,8 +1,12 @@
 ---
 name: weather-skill
 description: A skill that provides weather information based on reference data.
+metadata:
+  adk_additional_tools:
+    - get_wind_speed
 ---
 
 Step 1: Check 'references/weather_info.md' for the current weather.
 Step 2: If humidity is requested, use run 'scripts/get_humidity.py' with the `location` argument.
-Step 3: Provide the update to the user.
+Step 3: If wind speed is requested, use the `get_wind_speed` tool.
+Step 4: Provide the update to the user.

contributing/samples/skills_agent/skills/weather_skill/SKILL.md

@@ -16,6 +16,7 @@
 
 from __future__ import annotations
 
+import copy
 from typing import Any
 from typing import Awaitable
 from typing import Callable
@@ -108,3 +109,16 @@ class A2aRemoteAgentConfig(BaseModel):
   )
 
   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

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

@@ -38,6 +38,7 @@
 from typing_extensions import override
 
 from ...runners import Runner
+from ...sessions import base_session_service
 from ...utils.context_utils import Aclosing
 from ..agent.interceptors.new_integration_extension import _NEW_A2A_ADK_INTEGRATION_EXTENSION
 from ..converters.from_adk_event import create_error_status_event
@@ -287,6 +288,8 @@ async def _resolve_session(
         app_name=runner.app_name,
         user_id=user_id,
         session_id=session_id,
+        # Checking existence doesn't require event history.
+        config=base_session_service.GetSessionConfig(num_recent_events=0),
     )
     if session is None:
       session = await runner.session_service.create_session(

src/google/adk/a2a/executor/a2a_agent_executor_impl.py

@@ -73,6 +73,10 @@ class OAuth2Auth(BaseModelWithConfig):
   # tool or adk can generate the auth_uri with the state info thus client
   # can verify the state
   auth_uri: Optional[str] = None
+  # A unique value generated at the start of the OAuth flow to bind the user's
+  # session to the authorization request. This value is typically stored with
+  # user session and passed to backend for validation.
+  nonce: Optional[str] = None
   state: Optional[str] = None
   # tool or adk can decide the redirect_uri if they don't want client to decide
   redirect_uri: Optional[str] = None