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

fix(cost_calculator.py): respect litellm.suppress_debug_info for cost calc

Browse source 
Fixes https://github.com/BerriAI/litellm/issues/4818#issuecomment-2263795765
This commit is contained in:
Krrish Dholakia 2024-08-01 12:27:01 -07:00
parent dce8e3efb8
commit aad0bbb08c
1 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
litellm/cost_calculator.py
View file

@ -735,9 +735,16 @@ def response_cost_calculator(
)
return None
except Exception as e:
verbose_logger.warning(
"litellm.cost_calculator.py::response_cost_calculator - Returning None. Exception occurred - {}/n{}".format(
str(e), traceback.format_exc()
if litellm.suppress_debug_info: # allow cli tools to suppress this information.
verbose_logger.debug(
"litellm.cost_calculator.py::response_cost_calculator - Returning None. Exception occurred - {}/n{}".format(
str(e), traceback.format_exc()
)
)
else:
verbose_logger.warning(
"litellm.cost_calculator.py::response_cost_calculator - Returning None. Exception occurred - {}/n{}".format(
str(e), traceback.format_exc()
)
)
)
return None