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

fix - vertex ai exceptions

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-04 21:32:10 -07:00
parent 157b2b3a06
commit 4d1806bc95
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_exceptions.py
View file

@ -585,7 +585,7 @@ def test_router_completion_vertex_exception():
pytest.fail("Request should have failed - bad api key") pytest.fail("Request should have failed - bad api key")
except Exception as e: except Exception as e:
print("exception: ", e) print("exception: ", e)
assert "model: vertex_ai/gemini-pro" in str(e) assert "Model: gemini-pro" in str(e)
assert "model_group: vertex-gemini-pro" in str(e) assert "model_group: vertex-gemini-pro" in str(e)
assert "deployment: vertex_ai/gemini-pro" in str(e) assert "deployment: vertex_ai/gemini-pro" in str(e)
@ -604,7 +604,7 @@ def test_litellm_completion_vertex_exception():
pytest.fail("Request should have failed - bad api key") pytest.fail("Request should have failed - bad api key")
except Exception as e: except Exception as e:
print("exception: ", e) print("exception: ", e)
assert "model: vertex_ai/gemini-pro" in str(e) assert "Model: gemini-pro" in str(e)
assert "vertex_project: bad-project" in str(e) assert "vertex_project: bad-project" in str(e)