From b2fc7740b363a4e33ec99c7377f396f5cee40b5a Mon Sep 17 00:00:00 2001
From: Google Team Member <genai-sdk-bot@google.com>
Date: Thu, 12 Jun 2025 17:29:25 -0700
Subject: [PATCH] fix: Support project-based gemini model path to use
 google_search_tool

PiperOrigin-RevId: 770858301
---
 src/google/adk/tools/vertex_ai_search_tool.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/google/adk/tools/vertex_ai_search_tool.py b/src/google/adk/tools/vertex_ai_search_tool.py
index c370e2a7..5449f509 100644
--- a/src/google/adk/tools/vertex_ai_search_tool.py
+++ b/src/google/adk/tools/vertex_ai_search_tool.py
@@ -76,8 +76,8 @@ class VertexAiSearchTool(BaseTool):
       tool_context: ToolContext,
       llm_request: LlmRequest,
   ) -> None:
-    if llm_request.model and llm_request.model.startswith('gemini-'):
-      if llm_request.model.startswith('gemini-1') and llm_request.config.tools:
+    if llm_request.model and 'gemini-' in llm_request.model:
+      if 'gemini-1' in 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.'