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:
Devin AI 2026-08-20 18:23:26 +00:00
parent b7017a7949
commit e1ff8b27fa
2 changed files with 9 additions and 6 deletions

View file

@ -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

View file

@ -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