phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix - track vertex_location and vertex_project in vertex exceptions

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-18 12:53:33 -07:00
parent 192e0842c6
commit 930f8712e4
1 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/utils.py
View file

@ -7302,6 +7302,7 @@ def exception_type(
original_exception, original_exception,
custom_llm_provider, custom_llm_provider,
completion_kwargs={}, completion_kwargs={},
extra_kwargs={},
): ):
global user_logger_fn, liteDebuggerClient global user_logger_fn, liteDebuggerClient
exception_mapping_worked = False exception_mapping_worked = False
@ -7799,15 +7800,14 @@ def exception_type(
if completion_kwargs is not None: if completion_kwargs is not None:
# 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 {} error_str += f"\nmodel: {_model}\n"
_vertex_project = completion_kwargs.get("vertex_project") if extra_kwargs is not None:
_vertex_location = completion_kwargs.get("vertex_location") _vertex_project = extra_kwargs.get("vertex_project")
_vertex_location = extra_kwargs.get("vertex_location")
_metadata = _kwargs.get("metadata", {}) or {} _metadata = extra_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"
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: