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

fix(llm_cost_calc/google.py): fix google embedding cost calculation

Browse source 
Fixes https://github.com/BerriAI/litellm/issues/4630
This commit is contained in:
Krrish Dholakia 2024-07-11 11:52:18 -07:00
parent e3470d8e91
commit 2163434ff3
6 changed files with 133 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/litellm_core_utils/litellm_logging.py
View file

@ -1275,7 +1275,7 @@ class Logging:
f"Model={self.model}; cost={self.model_call_details['response_cost']}"
)
except litellm.NotFoundError as e:
verbose_logger.error(
verbose_logger.warning(
f"Model={self.model} not found in completion cost map. Setting 'response_cost' to None"
)
self.model_call_details["response_cost"] = None