From 72e1c69e6d0349ad69355ad82d161e4e2481ed3a Mon Sep 17 00:00:00 2001 From: Cursor Agent Date: Sat, 26 Sep 2026 04:25:44 +0000 Subject: [PATCH] test(router): annotate the budget filter return values The two helpers return the deployments the filter kept and the debug string. Name that as a sequence of model-list entries. Co-authored-by: abeed09 --- tests/unit/router_strategy/test_budget_limiter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/router_strategy/test_budget_limiter.py b/tests/unit/router_strategy/test_budget_limiter.py index 336e1cd96d9..169d4b58ca3 100644 --- a/tests/unit/router_strategy/test_budget_limiter.py +++ b/tests/unit/router_strategy/test_budget_limiter.py @@ -174,7 +174,7 @@ def _apply_budget_filter( request_tags: Sequence[str], spend_key: str, spend: float, -): +) -> tuple[Sequence[DeploymentTypedDict], str]: limiter: Final = RouterBudgetLimiting(dual_cache=DualCache(), provider_budget_config=None) limiter.provider_budget_config = provider_budget limiter.deployment_budget_config = deployment_budget @@ -190,7 +190,7 @@ def _apply_budget_filter( ) -def _filter_one(scope: _Scope, config: BudgetConfig, spend: float): +def _filter_one(scope: _Scope, config: BudgetConfig, spend: float) -> tuple[Sequence[DeploymentTypedDict], str]: match scope: case "provider": return _apply_budget_filter(