phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(test) gemini pro vision

Browse source
This commit is contained in:
ishaan-jaff 2023-12-16 18:31:55 +05:30
parent 774a725ccb
commit db188507b9
1 changed files with 31 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

31
litellm/tests/test_amazing_vertex_completion.py
View file

@ -192,3 +192,34 @@ async def test_async_vertexai_streaming_response():
pytest.fail(f"An exception occurred: {e}")
# asyncio.run(test_async_vertexai_streaming_response())
def test_gemini_pro_vision():
try:
litellm.set_verbose = True
litellm.num_retries=0
resp = litellm.completion(
model = "vertex_ai/gemini-pro-vision",
messages=[
{
"role": "user",
"content": [
{
"type": "text",
"text": "Whats in this image?"
},
{
"type": "image_url",
"image_url": {
"url": "https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg"
}
}
]
}
],
)
print(resp)
except Exception as e:
import traceback
traceback.print_exc()
raise e
test_gemini_pro_vision()