mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-18 08:25:10 +00:00
test(test_amazing_vertex_completion.py): fix test
This commit is contained in:
@@ -548,6 +548,7 @@ async def test_gemini_pro_vision(provider, sync_mode):
|
||||
resp = await litellm.acompletion(
|
||||
model="{}/gemini-1.5-flash-preview-0514".format(provider),
|
||||
messages=[
|
||||
{"role": "system", "content": "Be a good bot"},
|
||||
{
|
||||
"role": "user",
|
||||
"content": [
|
||||
@@ -559,7 +560,7 @@ async def test_gemini_pro_vision(provider, sync_mode):
|
||||
},
|
||||
},
|
||||
],
|
||||
}
|
||||
},
|
||||
],
|
||||
)
|
||||
print(resp)
|
||||
@@ -568,7 +569,7 @@ async def test_gemini_pro_vision(provider, sync_mode):
|
||||
|
||||
# DO Not DELETE this ASSERT
|
||||
# Google counts the prompt tokens for us, we should ensure we use the tokens from the orignal response
|
||||
assert prompt_tokens == 263 # the gemini api returns 263 to us
|
||||
assert prompt_tokens == 267 # the gemini api returns 267 to us
|
||||
|
||||
except litellm.RateLimitError as e:
|
||||
pass
|
||||
|
||||
Reference in New Issue
Block a user