diff --git a/tests/integration/_support/proxy.py b/tests/integration/_support/proxy.py index 19d479085a9..c6ae4f30375 100644 --- a/tests/integration/_support/proxy.py +++ b/tests/integration/_support/proxy.py @@ -6,6 +6,8 @@ from types import FrameType from typing import Final from unittest.mock import patch +from litellm import run_server + _PREMIUM: Final = ( patch( # test-quality-ok: route entitlement only; license validation is outside these HTTP/DB contracts "litellm.proxy.auth.litellm_license.LicenseCheck.is_premium", return_value=True @@ -13,8 +15,6 @@ _PREMIUM: Final = ( ) _PREMIUM.start() -from litellm import run_server # noqa: E402 - def _exit_on_reraised_term(signum: int, frame: FrameType | None) -> None: sys.exit(0)