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.'