phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 10:14:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge 95bfb230e4 into b82af5b826

Browse source
This commit is contained in:
Harings Rob 2025-04-24 01:03:24 -07:00 • committed by GitHub
commit 2ea6ffa36e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/utils.py
View file

@ -2291,6 +2291,11 @@ def register_model(model_cost: Union[str, dict]): # noqa: PLR0915
loaded_model_cost = litellm.get_model_cost_map(url=model_cost)
for key, value in loaded_model_cost.items():
## Change, if present, input_cost_per_token and output_cost_per_token to float
if "input_cost_per_token" in value:
value["input_cost_per_token"] = float(value["input_cost_per_token"])
if "output_cost_per_token" in value:
value["output_cost_per_token"] = float(value["output_cost_per_token"])
## get model info ##
try:
existing_model: dict = cast(dict, get_model_info(model=key))