test: update tests

This commit is contained in:
Krrish Dholakia 2025-05-19 20:48:26 -07:00
parent 8eccc787ba
commit b091832733

View file

@ -478,8 +478,8 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
},
}
# prod_json = "./model_prices_and_context_window.json"
prod_json = "../../model_prices_and_context_window.json"
prod_json = "./model_prices_and_context_window.json"
# prod_json = "../../model_prices_and_context_window.json"
with open(prod_json, "r") as model_prices_file:
actual_json = json.load(model_prices_file)
assert isinstance(actual_json, dict)
@ -639,8 +639,8 @@ async def test_supports_tool_choice():
"""
# Load model prices
litellm._turn_on_debug()
path = "../../model_prices_and_context_window.json"
# path = "./model_prices_and_context_window.json"
# path = "../../model_prices_and_context_window.json"
path = "./model_prices_and_context_window.json"
with open(path, "r") as f:
model_prices = json.load(f)
litellm.model_cost = model_prices