mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-11 22:51:28 +00:00
test: update tests
This commit is contained in:
parent
8eccc787ba
commit
b091832733
1 changed files with 4 additions and 4 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue