scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-20 00:11:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 77

fix(fireworks-ai): drop banned typing.cast to a suppressed import for the copied pricing entry

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
kerry 2026-09-16 00:57:05 +00:00
parent 5eab1feb20
commit 405a127838
1 changed files with 9 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
litellm/llms/fireworks_ai/cost_calculator.py
View file

@ -2,9 +2,11 @@
For calculating cost of fireworks ai serverless inference models.
"""
from collections.abc import Mapping
from datetime import datetime
from typing import Final, cast
from typing import (
Final,
cast, # noqa: TID251 # the fallback entry is a dict copy of a ReadOnly TypedDict; no cast-free way to retype it
)
from litellm.constants import (
FIREWORKS_AI_4_B,
@ -78,14 +80,11 @@ def _with_cache_read_fallback(model_info: ModelInfo) -> ModelInfo:
return model_info
effective: Final[dict[str, object]] = dict(model_info)
effective["cache_read_input_token_cost"] = input_rate
off_peak: Final = effective.get("off_peak_pricing")
if isinstance(off_peak, Mapping):
off_peak_map: Final[Mapping[str, object]] = cast(Mapping[str, object], off_peak)
if "cache_read_input_token_cost" not in off_peak_map:
effective["off_peak_pricing"] = {
**off_peak_map,
"cache_read_input_token_cost": off_peak_map.get("input_cost_per_token", input_rate),
}
off_peak: Final = model_info.get("off_peak_pricing")
if off_peak is not None and "cache_read_input_token_cost" not in off_peak:
off_peak_copy: Final[dict[str, object]] = dict(off_peak)
off_peak_copy["cache_read_input_token_cost"] = off_peak_copy.get("input_cost_per_token", input_rate)
effective["off_peak_pricing"] = off_peak_copy
return cast(ModelInfo, effective)