scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-09 22:31:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 86

test original file instead of the backup

Browse source
This commit is contained in:
Utkash Dubey 2025-03-10 20:09:05 -07:00
parent 412289a0d0
commit 14736283bc
1 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/litellm/test_model_prices_and_context_window_schema.py
View file

@ -1,5 +1,4 @@
import os
import litellm
import json
from jsonschema import validate
def test_model_prices_and_context_window_json_is_valid():
@ -100,8 +99,8 @@ def test_model_prices_and_context_window_json_is_valid():
},
}
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
actual_json = litellm.get_model_cost_map(url='')
with open("./model_prices_and_context_window.json", "r") as model_prices_file:
actual_json = json.load(model_prices_file)
assert isinstance(actual_json, dict)
temporarily_removed = actual_json.pop('sample_spec', None) # remove the sample, whose schema is inconsistent with the real data