From e6cd1c97e40b36d9cdead0b0fcd286189bd23039 Mon Sep 17 00:00:00 2001 From: kerry Date: Sun, 11 Oct 2026 01:31:29 +0000 Subject: [PATCH] test(decisions): price classifier calls from decision rows Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../complexity_router/test_jev_classifier.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/tests/unit/router_strategy/complexity_router/test_jev_classifier.py b/tests/unit/router_strategy/complexity_router/test_jev_classifier.py index fea283cbe9c..7ef58695670 100644 --- a/tests/unit/router_strategy/complexity_router/test_jev_classifier.py +++ b/tests/unit/router_strategy/complexity_router/test_jev_classifier.py @@ -14,6 +14,7 @@ import litellm from litellm._logging import verbose_router_logger from litellm.constants import INTERNAL_CALL_ORIGIN_METADATA_KEY from litellm.integrations.custom_logger import CustomLogger +from litellm.litellm_core_utils.decisions_pricing import decisions_pricing_key from litellm.litellm_core_utils.logging_worker import GLOBAL_LOGGING_WORKER from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler from litellm.router_strategy.complexity_router.complexity_router import ComplexityRouter @@ -500,6 +501,11 @@ async def test_oss_routes_with_its_own_credentials_and_accounts_the_checkpoint( monkeypatch.setenv(f"{provider.upper()}_API_KEY", "oss-env-key") monkeypatch.setattr(litellm, "disable_aiohttp_transport", True) monkeypatch.setitem(litellm.model_cost, f"{provider}/{model}", {"input_cost_per_token": 0.01}) + monkeypatch.setitem( + litellm.model_cost, + decisions_pricing_key(provider, model), + {"input_cost_per_token": 0.01}, + ) recorder: Final = _UsageRecorder(f"{provider}/{model}") monkeypatch.setattr(litellm, "_async_success_callback", [recorder]) router: Final = ComplexityRouter( @@ -563,6 +569,11 @@ async def test_databricks_routes_through_the_serving_endpoint_and_accounts_the_e monkeypatch.setenv("DATABRICKS_TOKEN", "dapi-env-token") monkeypatch.setattr(litellm, "disable_aiohttp_transport", True) monkeypatch.setitem(litellm.model_cost, f"databricks/{endpoint}", {"input_cost_per_token": 0.01}) + monkeypatch.setitem( + litellm.model_cost, + decisions_pricing_key("databricks", endpoint), + {"input_cost_per_token": 0.01}, + ) recorder: Final = _UsageRecorder(f"databricks/{endpoint}") monkeypatch.setattr(litellm, "_async_success_callback", [recorder]) router: Final = ComplexityRouter(