mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-12 03:48:56 +00:00
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:
parent
a5e79564cb
commit
1c795e36e7
1 changed files with 2 additions and 2 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue