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.
This commit is contained in:
yryzhan-vitech 2026-07-27 22:54:08 +02:00
parent 394818e45b
commit fe2c2830f4

View file

@ -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 (