mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-23 00:41:40 +00:00
18a1491bd2
Children spawned as [sys.executable, -c, ...] put the working directory first on sys.path, so under 'make test-rust-extension' a source checkout shadows the installed wheel and the child imports a litellm with no compiled extension. A shared helper spawns them with -I and asserts the child resolved the same litellm.__file__ as the parent, and a new TQ009 rule flags un-isolated sys.executable spawns. Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from collections.abc import Mapping
|
|
from typing import Final
|
|
|
|
import litellm
|
|
|
|
PARENT_LITELLM_FILE: Final = "LITELLM_TEST_PARENT_LITELLM_FILE"
|
|
|
|
_PROLOGUE: Final = (
|
|
"import os as _os, litellm as _litellm; _parent = _os.environ.pop({key!r}); "
|
|
'assert _litellm.__file__ == _parent, f"child imported litellm from {{_litellm.__file__}}, parent from {{_parent}}"; '
|
|
"del _os, _litellm, _parent\n"
|
|
)
|
|
|
|
|
|
def run_child_interpreter(
|
|
source: str, *, env: Mapping[str, str] | None = None, timeout: float
|
|
) -> subprocess.CompletedProcess[str]:
|
|
"""Run `source` in a fresh interpreter that imports the same `litellm` as this process.
|
|
|
|
`-I` keeps the working directory off sys.path so a source checkout cannot shadow an
|
|
installed wheel, and the prologue fails fast with both paths if the child still
|
|
resolves a different package.
|
|
"""
|
|
environment: Final = {**(os.environ if env is None else env), PARENT_LITELLM_FILE: litellm.__file__}
|
|
return subprocess.run(
|
|
[sys.executable, "-I", "-c", _PROLOGUE.format(key=PARENT_LITELLM_FILE) + source],
|
|
capture_output=True,
|
|
text=True,
|
|
timeout=timeout,
|
|
env=environment,
|
|
)
|