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

test: handle fireworks ai instability

Browse source
This commit is contained in:
Krrish Dholakia 2025-06-11 10:09:28 -07:00
parent e4ac1cdef2
commit ec52600f98
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/local_testing/test_embedding.py
View file

@ -806,6 +806,10 @@ def test_fireworks_embeddings():
assert cost > 0.0
print(response._hidden_params)
assert response._hidden_params["response_cost"] > 0.0
except litellm.RateLimitError as e:
pass
except litellm.InternalServerError as e:
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")