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

fix - show _vertex_project, _vertex_location in exceptions

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-18 11:48:43 -07:00
parent c589841b7e
commit 192e0842c6
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/utils.py
View file

@ -7800,14 +7800,22 @@ def exception_type(
# add model, deployment and model_group to the exception message # add model, deployment and model_group to the exception message
_model = completion_kwargs.get("model") _model = completion_kwargs.get("model")
_kwargs = completion_kwargs.get("kwargs", {}) or {} _kwargs = completion_kwargs.get("kwargs", {}) or {}
_vertex_project = completion_kwargs.get("vertex_project")
_vertex_location = completion_kwargs.get("vertex_location")
_metadata = _kwargs.get("metadata", {}) or {} _metadata = _kwargs.get("metadata", {}) or {}
_model_group = _metadata.get("model_group") _model_group = _metadata.get("model_group")
_deployment = _metadata.get("deployment") _deployment = _metadata.get("deployment")
error_str += f"\nmodel: {_model}\n" error_str += f"\nmodel: {_model}\n"
if _model_group is not None: if _model_group is not None:
error_str += f"model_group: {_model_group}\n" error_str += f"model_group: {_model_group}\n"
if _deployment is not None: if _deployment is not None:
error_str += f"deployment: {_deployment}\n" error_str += f"deployment: {_deployment}\n"
if _vertex_project is not None:
error_str += f"vertex_project: {_vertex_project}\n"
if _vertex_location is not None:
error_str += f"vertex_location: {_vertex_location}\n"
if ( if (
"Vertex AI API has not been used in project" in error_str "Vertex AI API has not been used in project" in error_str