From 1c795e36e7950ec9f1283a5b5e83136139fe9cdd Mon Sep 17 00:00:00 2001 From: yucheng Date: Thu, 24 Sep 2026 15:59:45 +0000 Subject: [PATCH] test(integration): keep the run_server import at the top of the entitlement shim (LIT-8429) Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- tests/integration/_support/proxy.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)