mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-20 00:11:50 +00:00
test - gemini-1.5-pro-latest
This commit is contained in:
parent
63805873e7
commit
81e4d59357
1 changed files with 1 additions and 1 deletions
|
|
@ -2489,7 +2489,7 @@ def test_completion_deep_infra_mistral():
|
|||
# Gemini tests
|
||||
def test_completion_gemini():
|
||||
litellm.set_verbose = True
|
||||
model_name = "gemini/gemini-pro"
|
||||
model_name = "gemini/gemini-1.5-pro-latest"
|
||||
messages = [{"role": "user", "content": "Hey, how's it going?"}]
|
||||
try:
|
||||
response = completion(model=model_name, messages=messages)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue