diff --git a/src/google/adk/tools/google_search_tool.py b/src/google/adk/tools/google_search_tool.py index d116e4b6c8e..9fe387df3ba 100644 --- a/src/google/adk/tools/google_search_tool.py +++ b/src/google/adk/tools/google_search_tool.py @@ -54,7 +54,7 @@ async def process_llm_request( llm_request.config.tools.append( types.Tool(google_search_retrieval=types.GoogleSearchRetrieval()) ) - elif llm_request.model and 'gemini-2' in llm_request.model: + elif llm_request.model and 'gemini-' in llm_request.model: llm_request.config.tools.append( types.Tool(google_search=types.GoogleSearch()) )