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>
This commit is contained in:
yucheng 2026-09-24 15:59:45 +00:00
parent a5e79564cb
commit 1c795e36e7

View file

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