phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

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
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}")