fix: Support project-based gemini model path for BuiltInCodeExecutor and all built-in tools

This is to support model path like :
projects/265104255505/locations/us-central1/publishers/google/models/gemini-2.0-flash-001"

PiperOrigin-RevId: 783413351

Author: seanzhougoogle

src/google/adk/code_executors/built_in_code_executor.py

@@ -12,11 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from google.genai import types
 from typing_extensions import override
 
 from ..agents.invocation_context import InvocationContext
 from ..models import LlmRequest
+from ..utils.model_name_utils import is_gemini_2_model
 from .base_code_executor import BaseCodeExecutor
 from .code_execution_utils import CodeExecutionInput
 from .code_execution_utils import CodeExecutionResult
@@ -39,7 +42,7 @@ def execute_code(
 
   def process_llm_request(self, llm_request: LlmRequest) -> None:
     """Pre-process the LLM request for Gemini 2.0+ models to use the code execution tool."""
-    if llm_request.model and llm_request.model.startswith("gemini-2"):
+    if is_gemini_2_model(llm_request.model):
       llm_request.config = llm_request.config or types.GenerateContentConfig()
       llm_request.config.tools = llm_request.config.tools or []
       llm_request.config.tools.append(

src/google/adk/tools/enterprise_search_tool.py

@@ -19,6 +19,8 @@
 from google.genai import types
 from typing_extensions import override
 
+from ..utils.model_name_utils import is_gemini_1_model
+from ..utils.model_name_utils import is_gemini_model
 from .base_tool import BaseTool
 from .tool_context import ToolContext
 
@@ -47,8 +49,8 @@ async def process_llm_request(
       tool_context: ToolContext,
       llm_request: LlmRequest,
   ) -> None:
-    if llm_request.model and 'gemini-' in llm_request.model:
-      if 'gemini-1' in llm_request.model and llm_request.config.tools:
+    if is_gemini_model(llm_request.model):
+      if is_gemini_1_model(llm_request.model) and llm_request.config.tools:
         raise ValueError(
             'Enterprise web search tool can not be used with other tools in'
             ' Gemini 1.x.'

src/google/adk/tools/google_search_tool.py

@@ -19,6 +19,8 @@
 from google.genai import types
 from typing_extensions import override
 
+from ..utils.model_name_utils import is_gemini_1_model
+from ..utils.model_name_utils import is_gemini_model
 from .base_tool import BaseTool
 from .tool_context import ToolContext
 
@@ -46,15 +48,15 @@ async def process_llm_request(
   ) -> None:
     llm_request.config = llm_request.config or types.GenerateContentConfig()
     llm_request.config.tools = llm_request.config.tools or []
-    if llm_request.model and 'gemini-1' in llm_request.model:
+    if is_gemini_1_model(llm_request.model):
       if llm_request.config.tools:
         raise ValueError(
             'Google search tool can not be used with other tools in Gemini 1.x.'
         )
       llm_request.config.tools.append(
           types.Tool(google_search_retrieval=types.GoogleSearchRetrieval())
       )
-    elif llm_request.model and 'gemini-' in llm_request.model:
+    elif is_gemini_model(llm_request.model):
       llm_request.config.tools.append(
           types.Tool(google_search=types.GoogleSearch())
       )

src/google/adk/tools/retrieval/vertex_ai_rag_retrieval.py

@@ -24,11 +24,12 @@
 from typing_extensions import override
 from vertexai.preview import rag
 
+from ...utils.model_name_utils import is_gemini_2_model
 from ..tool_context import ToolContext
 from .base_retrieval_tool import BaseRetrievalTool
 
 if TYPE_CHECKING:
-  from ...models.llm_request import LlmRequest
+  from ...models import LlmRequest
 
 logger = logging.getLogger('google_adk.' + __name__)
 
@@ -62,7 +63,7 @@ async def process_llm_request(
       llm_request: LlmRequest,
   ) -> None:
     # Use Gemini built-in Vertex AI RAG tool for Gemini 2 models.
-    if llm_request.model and llm_request.model.startswith('gemini-2'):
+    if is_gemini_2_model(llm_request.model):
       llm_request.config = (
           types.GenerateContentConfig()
           if not llm_request.config

src/google/adk/tools/url_context_tool.py

@@ -19,6 +19,8 @@
 from google.genai import types
 from typing_extensions import override
 
+from ..utils.model_name_utils import is_gemini_1_model
+from ..utils.model_name_utils import is_gemini_2_model
 from .base_tool import BaseTool
 from .tool_context import ToolContext
 
@@ -46,9 +48,9 @@ async def process_llm_request(
   ) -> None:
     llm_request.config = llm_request.config or types.GenerateContentConfig()
     llm_request.config.tools = llm_request.config.tools or []
-    if llm_request.model and 'gemini-1' in llm_request.model:
+    if is_gemini_1_model(llm_request.model):
       raise ValueError('Url context tool can not be used in Gemini 1.x.')
-    elif llm_request.model and 'gemini-2' in llm_request.model:
+    elif is_gemini_2_model(llm_request.model):
       llm_request.config.tools.append(
           types.Tool(url_context=types.UrlContext())
       )

src/google/adk/tools/vertex_ai_search_tool.py

@@ -20,6 +20,8 @@
 from google.genai import types
 from typing_extensions import override
 
+from ..utils.model_name_utils import is_gemini_1_model
+from ..utils.model_name_utils import is_gemini_model
 from .base_tool import BaseTool
 from .tool_context import ToolContext
 
@@ -86,8 +88,8 @@ async def process_llm_request(
       tool_context: ToolContext,
       llm_request: LlmRequest,
   ) -> None:
-    if llm_request.model and 'gemini-' in llm_request.model:
-      if 'gemini-1' in llm_request.model and llm_request.config.tools:
+    if is_gemini_model(llm_request.model):
+      if is_gemini_1_model(llm_request.model) and llm_request.config.tools:
         raise ValueError(
             'Vertex AI search tool can not be used with other tools in Gemini'
             ' 1.x.'

src/google/adk/utils/model_name_utils.py

@@ -0,0 +1,90 @@
+# Copyright 2025 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.
+
+"""Utilities for model name validation and parsing."""
+
+from __future__ import annotations
+
+import re
+from typing import Optional
+
+
+def extract_model_name(model_string: str) -> str:
+  """Extract the actual model name from either simple or path-based format.
+
+  Args:
+    model_string: Either a simple model name like "gemini-2.5-pro" or
+                  a path-based model name like "projects/.../models/gemini-2.0-flash-001"
+
+  Returns:
+    The extracted model name (e.g., "gemini-2.5-pro")
+  """
+  # Pattern for path-based model names
+  path_pattern = (
+      r'^projects/[^/]+/locations/[^/]+/publishers/[^/]+/models/(.+)$'
+  )
+  match = re.match(path_pattern, model_string)
+  if match:
+    return match.group(1)
+
+  # If it's not a path-based model, return as-is (simple model name)
+  return model_string
+
+
+def is_gemini_model(model_string: Optional[str]) -> bool:
+  """Check if the model is a Gemini model using regex patterns.
+
+  Args:
+    model_string: Either a simple model name or path-based model name
+
+  Returns:
+    True if it's a Gemini model, False otherwise
+  """
+  if not model_string:
+    return False
+
+  model_name = extract_model_name(model_string)
+  return re.match(r'^gemini-', model_name) is not None
+
+
+def is_gemini_1_model(model_string: Optional[str]) -> bool:
+  """Check if the model is a Gemini 1.x model using regex patterns.
+
+  Args:
+    model_string: Either a simple model name or path-based model name
+
+  Returns:
+    True if it's a Gemini 1.x model, False otherwise
+  """
+  if not model_string:
+    return False
+
+  model_name = extract_model_name(model_string)
+  return re.match(r'^gemini-1\.\d+', model_name) is not None
+
+
+def is_gemini_2_model(model_string: Optional[str]) -> bool:
+  """Check if the model is a Gemini 2.x model using regex patterns.
+
+  Args:
+    model_string: Either a simple model name or path-based model name
+
+  Returns:
+    True if it's a Gemini 2.x model, False otherwise
+  """
+  if not model_string:
+    return False
+
+  model_name = extract_model_name(model_string)
+  return re.match(r'^gemini-2\.\d+', model_name) is not None