test(test_completion.py): handle together ai timeout

This commit is contained in:
Krrish Dholakia 2024-01-18 17:54:16 -08:00
parent 1e5efdfa37
commit 94ce524c63

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