Skip to content

Commit 6023427

Browse files
docs(gemini.md): cleanup
1 parent 2486a10 commit 6023427

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

docs/my-website/docs/providers/gemini.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -655,7 +655,7 @@ import os
655655

656656
os.environ["GEMINI_API_KEY"] = ".."
657657

658-
tools = [{"googleSearchRetrieval": {}}] # 👈 ADD GOOGLE SEARCH
658+
tools = [{"googleSearch": {}}] # 👈 ADD GOOGLE SEARCH
659659

660660
response = completion(
661661
model="gemini/gemini-2.0-flash",
@@ -691,7 +691,7 @@ curl -X POST 'http://0.0.0.0:4000/chat/completions' \
691691
-d '{
692692
"model": "gemini-2.0-flash",
693693
"messages": [{"role": "user", "content": "What is the weather in San Francisco?"}],
694-
"tools": [{"googleSearchRetrieval": {}}]
694+
"tools": [{"googleSearch": {}}]
695695
}
696696
'
697697
```

docs/my-website/docs/providers/vertex.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

369369
resp = litellm.completion(
370370
model="vertex_ai/gemini-1.0-pro-001",
@@ -391,7 +391,7 @@ client = OpenAI(
391391
response = 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

397397
print(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

534534
resp = litellm.completion(
535535
model="vertex_ai/gemini-1.0-pro-001",

0 commit comments

Comments
 (0)