@@ -364,7 +364,7 @@ from litellm import completion
364364# # SETUP ENVIRONMENT
365365# !gcloud auth application-default login - run this to add vertex credentials to your env
366366
367- tools = [{" googleSearchRetrieval " : {}}] # 👈 ADD GOOGLE SEARCH
367+ tools = [{" googleSearch " : {}}] # 👈 ADD GOOGLE SEARCH
368368
369369resp = litellm.completion(
370370 model = " vertex_ai/gemini-1.0-pro-001" ,
@@ -391,7 +391,7 @@ client = OpenAI(
391391response = client.chat.completions.create(
392392 model = " gemini-pro" ,
393393 messages = [{" role" : " user" , " content" : " Who won the world cup?" }],
394- tools = [{" googleSearchRetrieval " : {}}],
394+ tools = [{" googleSearch " : {}}],
395395)
396396
397397print (response)
@@ -410,7 +410,7 @@ curl http://localhost:4000/v1/chat/completions \
410410 ],
411411 "tools": [
412412 {
413- "googleSearchRetrieval ": {}
413+ "googleSearch ": {}
414414 }
415415 ]
416416 }'
@@ -529,7 +529,7 @@ from litellm import completion
529529
530530# !gcloud auth application-default login - run this to add vertex credentials to your env
531531
532- tools = [{" googleSearchRetrieval " : {" disable_attributon" : False }}] # 👈 ADD GOOGLE SEARCH
532+ tools = [{" googleSearch " : {" disable_attributon" : False }}] # 👈 ADD GOOGLE SEARCH
533533
534534resp = litellm.completion(
535535 model = " vertex_ai/gemini-1.0-pro-001" ,
0 commit comments