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

test(test_completion.py): handle internal server error in test

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-07 10:21:37 -07:00
parent 3cf9148a4a
commit 3646e3e3a4
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(