mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-07 08:26:10 +00:00
test: satisfy test-quality gate in consolidated registry tests
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
parent
b7017a7949
commit
e1ff8b27fa
2 changed files with 9 additions and 6 deletions
|
|
@ -1065,10 +1065,16 @@ def test_generic_cost_per_token_gpt56_terra_cache_costs_by_tier_and_context(
|
|||
],
|
||||
)
|
||||
def test_generic_cost_per_token_gpt56_cyber(
|
||||
model, prompt_tokens, input_rate, cache_write_rate, cache_read_rate, output_rate
|
||||
model,
|
||||
prompt_tokens,
|
||||
input_rate,
|
||||
cache_write_rate,
|
||||
cache_read_rate,
|
||||
output_rate,
|
||||
monkeypatch,
|
||||
):
|
||||
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
|
||||
litellm.model_cost = litellm.get_model_cost_map(url="")
|
||||
monkeypatch.setenv("LITELLM_LOCAL_MODEL_COST_MAP", "True")
|
||||
monkeypatch.setattr(litellm, "model_cost", litellm.get_model_cost_map(url=""))
|
||||
|
||||
cached_tokens = 50000
|
||||
cache_write_tokens = 40000
|
||||
|
|
|
|||
|
|
@ -12,9 +12,6 @@ values in both the primary price map and the ``litellm/`` backup, and verify
|
|||
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
|
||||
sys.path.insert(0, os.path.abspath("../.."))
|
||||
|
||||
import litellm
|
||||
from litellm import completion_cost
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue