fix: Update google_search_tool.py to support updated Gemini LIVE model naming

rayira · copybara-github · commit 766e7f40fa45 · 2025-06-25T19:56:09.000-07:00
Merge google#1518 ## Description Fixes [google#1512](google#1512) by updating google_search_tool.py to support new Gemini LIVE model naming ## Changes - Update the model name checking in google_search_tool.py COPYBARA_INTEGRATE_REVIEW=google#1518 from rayira:rayira-patch-1 4c98f88 PiperOrigin-RevId: 775941268
diff --git 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())
       )

Original file line number	Diff line number	Diff line change
`@@ -54,7 +54,7 @@ async def process_llm_request(`
`54`	`54`	`llm_request.config.tools.append(`
`55`	`55`	`types.Tool(google_search_retrieval=types.GoogleSearchRetrieval())`
`56`	`56`	`)`
`57`		`- elif llm_request.model and 'gemini-2' in llm_request.model:`
	`57`	`+ elif llm_request.model and 'gemini-' in llm_request.model:`
`58`	`58`	`llm_request.config.tools.append(`
`59`	`59`	`types.Tool(google_search=types.GoogleSearch())`
`60`	`60`	`)`