laantungir/routstr-core
1
0
Fork 0
mirror of https://github.com/Routstr/routstr-core.git synced 2026-10-05 20:28:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #667 from Routstr/fix-tests

Browse Source 
test: register DeepSeek V4 pricing shim in cache pricing tests
This commit is contained in:
9qeklajc
2026-08-16 21:10:29 +02:00
committed by GitHub
parent 95817fd9a9 8218a103a3
commit 59e622af8a
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unit/test_cache_pricing.py
+9
View File
@@ -31,6 +31,15 @@ from routstr.payment.models import (
backfill_cache_pricing,
)
from routstr.upstream import GenericUpstreamProvider
from routstr.upstream.deepseek_v4_pricing_shim import register_deepseek_v4_pricing
@pytest.fixture(autouse=True)
def _deepseek_v4_pricing() -> None:
# litellm's bundled cost map lacks the DeepSeek V4 entries (they only
# appear when its remote map is reachable); production injects them at
# startup via this same shim.
register_deepseek_v4_pricing()
def _make_model(model_id: str, pricing: Pricing) -> Model: