mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-17 23:51:30 +00:00
fix: rename _load_backup_json call sites to _load_model_cost_json
This commit is contained in:
parent
7290bcdf1c
commit
89a5a44ee0
1 changed files with 6 additions and 6 deletions
|
|
@ -53,33 +53,33 @@ class TestDeepSeekModelCostEntries:
|
|||
capability flags as their bare-name counterparts in the JSON files."""
|
||||
|
||||
def test_deepseek_chat_supports_response_schema_in_backup(self):
|
||||
data = _load_backup_json()
|
||||
data = _load_model_cost_json()
|
||||
entry = data.get("deepseek/deepseek-chat", {})
|
||||
assert entry.get("supports_response_schema") is True
|
||||
|
||||
def test_deepseek_reasoner_supports_response_schema_in_backup(self):
|
||||
data = _load_backup_json()
|
||||
data = _load_model_cost_json()
|
||||
entry = data.get("deepseek/deepseek-reasoner", {})
|
||||
assert entry.get("supports_response_schema") is True
|
||||
|
||||
def test_deepseek_chat_supports_system_messages_in_backup(self):
|
||||
data = _load_backup_json()
|
||||
data = _load_model_cost_json()
|
||||
entry = data.get("deepseek/deepseek-chat", {})
|
||||
assert entry.get("supports_system_messages") is True
|
||||
|
||||
def test_deepseek_reasoner_supports_system_messages_in_backup(self):
|
||||
data = _load_backup_json()
|
||||
data = _load_model_cost_json()
|
||||
entry = data.get("deepseek/deepseek-reasoner", {})
|
||||
assert entry.get("supports_system_messages") is True
|
||||
|
||||
def test_deepseek_chat_max_input_tokens_matches_bare_in_backup(self):
|
||||
data = _load_backup_json()
|
||||
data = _load_model_cost_json()
|
||||
bare = data.get("deepseek-chat", {})
|
||||
prefixed = data.get("deepseek/deepseek-chat", {})
|
||||
assert prefixed.get("max_input_tokens") == bare.get("max_input_tokens")
|
||||
|
||||
def test_deepseek_reasoner_max_output_tokens_matches_bare_in_backup(self):
|
||||
data = _load_backup_json()
|
||||
data = _load_model_cost_json()
|
||||
bare = data.get("deepseek-reasoner", {})
|
||||
prefixed = data.get("deepseek/deepseek-reasoner", {})
|
||||
assert prefixed.get("max_output_tokens") == bare.get("max_output_tokens")
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue