phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-28 04:04:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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 9c0179d132
commit e8bab67ea4
1 changed files with 2 additions and 0 deletions
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: ", "Cost for completion call together-computer/llama-2-70b: ",
f"${float(cost):.10f}", f"${float(cost):.10f}",
) )
except litellm.Timeout as e:
pass
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")