scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 87

test(test_completion.py): handle together ai timeout

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-18 17:54:16 -08:00
parent 1e5efdfa37
commit 94ce524c63
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_completion.py
View file

@ -1269,6 +1269,8 @@ def test_completion_together_ai():
"Cost for completion call together-computer/llama-2-70b: ",
f"${float(cost):.10f}",
)
except litellm.Timeout as e:
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")