From 6cd1fcdcf05734cff34d0dfed7e098da73a0a913 Mon Sep 17 00:00:00 2001 From: milan Date: Tue, 25 Aug 2026 18:37:26 +0000 Subject: [PATCH] test: drop unneeded proxy_server patches and ratchet lint budgets Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- ruff-strict-budget.json | 2 +- .../proxy/spend_tracking/test_spend_tracking_utils.py | 11 ++--------- type-discipline-budget.json | 2 +- 3 files changed, 4 insertions(+), 11 deletions(-) diff --git a/ruff-strict-budget.json b/ruff-strict-budget.json index 03318718fb5..9815ccfa23e 100644 --- a/ruff-strict-budget.json +++ b/ruff-strict-budget.json @@ -168,7 +168,7 @@ "limit": 3 }, "RET504": { - "limit": 176 + "limit": 175 }, "RUF012": { "limit": 240 diff --git a/tests/test_litellm/proxy/spend_tracking/test_spend_tracking_utils.py b/tests/test_litellm/proxy/spend_tracking/test_spend_tracking_utils.py index dab9415de77..843e1d1296f 100644 --- a/tests/test_litellm/proxy/spend_tracking/test_spend_tracking_utils.py +++ b/tests/test_litellm/proxy/spend_tracking/test_spend_tracking_utils.py @@ -3,13 +3,10 @@ import datetime import json from datetime import timezone from typing import Any, Final, cast - -from typing_extensions import ReadOnly, TypedDict +from unittest.mock import AsyncMock, MagicMock, patch import pytest - - -from unittest.mock import AsyncMock, MagicMock, patch +from typing_extensions import ReadOnly, TypedDict import litellm from litellm.constants import ( @@ -3526,8 +3523,6 @@ def _model_router_spend_log_kwargs(slp_model: str | None) -> _ModelRouterSpendLo } -@patch("litellm.proxy.proxy_server.master_key", None) -@patch("litellm.proxy.proxy_server.general_settings", {}) def test_get_logging_payload_uses_standard_logging_payload_model(): payload = get_logging_payload( kwargs=_model_router_spend_log_kwargs(slp_model="azure_ai/gpt-5-mini"), @@ -3538,8 +3533,6 @@ def test_get_logging_payload_uses_standard_logging_payload_model(): assert payload["model"] == "azure_ai/gpt-5-mini" -@patch("litellm.proxy.proxy_server.master_key", None) -@patch("litellm.proxy.proxy_server.general_settings", {}) def test_get_logging_payload_falls_back_to_kwargs_model_when_slp_model_missing(): payload = get_logging_payload( kwargs=_model_router_spend_log_kwargs(slp_model=None), diff --git a/type-discipline-budget.json b/type-discipline-budget.json index 05098546325..542762f1cea 100644 --- a/type-discipline-budget.json +++ b/type-discipline-budget.json @@ -30,7 +30,7 @@ "limit": 16673 }, "LIT011": { - "limit": 5588 + "limit": 5587 }, "LIT012": { "limit": 4510