From e33161b4f8650e8bcb36c650c4e2d1fe79ae2526 Mon Sep 17 00:00:00 2001
From: Yanfei Chen <yanfeichen@google.com>
Date: Tue, 8 Jul 2025 13:28:40 -0700
Subject: [PATCH] fix: Support project-based gemini model path to use
 enterprise_web_search_tool

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

diff --git a/src/google/adk/tools/enterprise_search_tool.py b/src/google/adk/tools/enterprise_search_tool.py
index f3380bc5..e3ce0af4 100644
--- a/src/google/adk/tools/enterprise_search_tool.py
+++ b/src/google/adk/tools/enterprise_search_tool.py
@@ -47,8 +47,8 @@ class EnterpriseWebSearchTool(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(
             'Enterprise web search tool can not be used with other tools in'
             ' Gemini 1.x.'