phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) retry gemini-pro-vision 3 times

Browse source
This commit is contained in:
ishaan-jaff 2024-01-11 14:39:08 +05:30
parent c41b47dc8b
commit fccd81a444
1 changed files with 14 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

23
litellm/tests/test_google_ai_studio_gemini.py
View file

@ -6,29 +6,34 @@ sys.path.insert(
import litellm
from dotenv import load_dotenv
def generate_text():
try:
litellm.set_verbose = True
messages = [
{
"role": "user",
"content": [
{
"type": "text",
"text": "What is this image?"
},
{"type": "text", "text": "What is this image?"},
{
"type": "image_url",
"image_url": {
"url": "https://avatars.githubusercontent.com/u/17561003?v=4"
}
}
]
},
},
],
}
]
response = litellm.completion(model="gemini/gemini-pro-vision", messages=messages, stop="Hello world")
response = litellm.completion(
model="gemini/gemini-pro-vision",
messages=messages,
stop="Hello world",
num_retries=3,
)
print(response)
assert isinstance(response.choices[0].message.content, str) == True
except Exception as exception:
raise Exception("An error occurred during text generation:", exception)
generate_text()
# generate_text()