scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 178

fix(mypy): use explicit None check for cache rate values to satisfy type checker

Browse source
This commit is contained in:
Ishaan Jaffer 2026-04-14 19:03:56 -07:00
parent e20d9df1b6
commit 84c507bc14
No known key found for this signature in database
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/cost_calculator.py
View file

@ -1611,10 +1611,12 @@ def completion_cost( # noqa: PLR0915
if (_cr or _cc) and model:
try:
_mi = litellm.get_model_info(model=model, custom_llm_provider=custom_llm_provider)
if _cr and _mi.get("cache_read_input_token_cost"):
_cache_read_cost = float(_cr) * float(_mi["cache_read_input_token_cost"])
if _cc and _mi.get("cache_creation_input_token_cost"):
_cache_creation_cost = float(_cc) * float(_mi["cache_creation_input_token_cost"])
_cr_rate = _mi.get("cache_read_input_token_cost")
if _cr and _cr_rate is not None:
_cache_read_cost = float(_cr) * float(_cr_rate)
_cc_rate = _mi.get("cache_creation_input_token_cost")
if _cc and _cc_rate is not None:
_cache_creation_cost = float(_cc) * float(_cc_rate)
except Exception:
pass
_store_cost_breakdown_in_logging_obj(