scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

test(test_completion.py): handle gemini instability

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-07 15:58:47 -07:00
parent f76261af35
commit 639f84d609
1 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
litellm/tests/test_completion.py
View file

@ -934,15 +934,18 @@ def test_completion_function_plus_image(model):
}
]
response = completion(
model=model,
messages=[image_message],
tool_choice=tool_choice,
tools=tools,
stream=False,
)
try:
response = completion(
model=model,
messages=[image_message],
tool_choice=tool_choice,
tools=tools,
stream=False,
)
print(response)
print(response)
except litellm.InternalServerError:
pass
@pytest.mark.parametrize(