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

fix(utils.py): add exception mapping for databricks errors

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-27 13:13:31 -07:00
parent 1a8f45e8da
commit 05ba34b9b7
2 changed files with 22 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_exceptions.py
View file

@ -770,7 +770,9 @@ def test_litellm_predibase_exception():
# print(f"accuracy_score: {accuracy_score}")
@pytest.mark.parametrize("provider", ["predibase", "vertex_ai_beta", "anthropic"])
@pytest.mark.parametrize(
"provider", ["predibase", "vertex_ai_beta", "anthropic", "databricks"]
)
def test_exception_mapping(provider):
"""
For predibase, run through a set of mock exceptions