From fe2c2830f48e7767610b592230cb7c04a76926e5 Mon Sep 17 00:00:00 2001 From: yryzhan-vitech Date: Mon, 27 Jul 2026 22:54:08 +0200 Subject: [PATCH] chore(proxy): import the drain helpers as standalone statements Both names sat as continuation lines inside existing multi-line import blocks, where CPython attributes the bytecode to the first line of the block, so they were untrackable by coverage while still counting as added lines. --- litellm/proxy/proxy_server.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 3b1b60027ac..0d3caf69322 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -241,8 +241,8 @@ from litellm.constants import ( PROXY_BUDGET_RESCHEDULER_MAX_TIME, PROXY_BUDGET_RESCHEDULER_MIN_TIME, PROXY_CONFIG_RELOAD_INTERVAL_SECONDS, - PROXY_SHUTDOWN_SPEND_DRAIN_TIMEOUT_SECONDS, ) +from litellm.constants import PROXY_SHUTDOWN_SPEND_DRAIN_TIMEOUT_SECONDS from litellm.exceptions import RejectedRequestError from litellm.integrations.custom_guardrail import ModifyResponseException from litellm.integrations.custom_logger import CustomLogger @@ -548,9 +548,9 @@ from litellm.proxy.utils import ( migrate_passwords_to_scrypt_async, model_dump_with_preserved_fields, prefetch_config_params, - update_daily_tag_spend, update_spend, ) +from litellm.proxy.utils import update_daily_tag_spend from litellm.proxy.video_endpoints.endpoints import router as video_router from litellm.repositories.credentials_repository import CredentialsRepository from litellm.router import (