tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-17 10:26:07 +00:00
Code Issues Packages Projects Releases Wiki 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:09 -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
litellm/cost_calculator.py
+11 -4
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