Merge pull request #6 from google-gemini/MarkDaoust-patch-4

meslubi2021 · web-flow · commit 4654d7e8bf64 · 2025-01-10T23:08:52.000+01:00
Fix error message, `google_search_grounding`, not the modes.
diff --git a/google/generativeai/types/content_types.py b/google/generativeai/types/content_types.py
@@ -728,7 +728,7 @@ def _make_tool(tool: ToolType) -> Tool:
             return Tool(google_search_retrieval=protos.GoogleSearchRetrieval())
         else:
             raise ValueError(
-                "The only string that can be passed as a tool is 'code_execution', or one of the specified values for the `mode` parameter for google_search_retrieval."
+                "The only strings that can be passed as a `tool` is 'code_execution' or 'google_search_retrieval'."
             )
     elif isinstance(tool, protos.CodeExecution):
         return Tool(code_execution=tool)

Original file line number	Diff line number	Diff line change
`@@ -728,7 +728,7 @@ def _make_tool(tool: ToolType) -> Tool:`
`728`	`728`	`return Tool(google_search_retrieval=protos.GoogleSearchRetrieval())`
`729`	`729`	`else:`
`730`	`730`	`raise ValueError(`
`731`		- "The only string that can be passed as a tool is 'code_execution', or one of the specified values for the `mode` parameter for google_search_retrieval."
	`731`	+ "The only strings that can be passed as a `tool` is 'code_execution' or 'google_search_retrieval'."
`732`	`732`	`)`
`733`	`733`	`elif isinstance(tool, protos.CodeExecution):`
`734`	`734`	`return Tool(code_execution=tool)`