mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-11 03:38:38 +00:00
8520626e7a
* fix(lens): pin worker dependencies and support approved image digests * fix(lens): include locked dependencies and release identity in build context * fix(lens): select the dev worker package for SHA-tagged charts
81 lines
3.5 KiB
Python
81 lines
3.5 KiB
Python
from importlib.metadata import Distribution, PackageNotFoundError, PathDistribution
|
|
from pathlib import Path
|
|
from typing import Final
|
|
|
|
import pytest
|
|
|
|
from litellm.proxy.lens.release import worker_image
|
|
|
|
|
|
@pytest.mark.parametrize("tag", ("v1.2.3", "v1.2.3-rc.4", "v1.2.3-dev.5", "branch-main-1234567"))
|
|
def test_install_command_follows_the_gateway_release(monkeypatch: pytest.MonkeyPatch, tag: str) -> None:
|
|
monkeypatch.setenv("LITELLM_RELEASE_TAG", tag)
|
|
monkeypatch.delenv("LENS_WORKER_IMAGE", raising=False)
|
|
assert worker_image() == f"ghcr.io/berriai/litellm-lens-worker:{tag}"
|
|
|
|
|
|
def test_private_registry_override_keeps_its_exact_digest(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
image: Final = "registry.example/lens-worker@sha256:" + "a" * 64
|
|
monkeypatch.setenv("LITELLM_RELEASE_TAG", "branch-main-1234567")
|
|
monkeypatch.setenv("LENS_WORKER_IMAGE", image)
|
|
assert worker_image() == image
|
|
|
|
|
|
def test_source_build_uses_the_separate_development_package(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
tag: Final = "sha-" + "a" * 40
|
|
monkeypatch.setenv("LITELLM_RELEASE_TAG", tag)
|
|
monkeypatch.delenv("LENS_WORKER_IMAGE", raising=False)
|
|
assert worker_image() == f"ghcr.io/berriai/litellm-lens-worker-dev:{tag}"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"installed,expected",
|
|
(("1.2.3", "v1.2.3"), ("1.2.3rc4", "v1.2.3-rc.4"), ("1.2.3.dev5", "v1.2.3-dev.5")),
|
|
)
|
|
def test_python_installs_recommend_the_matching_worker(
|
|
monkeypatch: pytest.MonkeyPatch, tmp_path: Path, installed: str, expected: str
|
|
) -> None:
|
|
from litellm.proxy.lens import release
|
|
|
|
metadata: Final = tmp_path / "litellm.dist-info"
|
|
metadata.mkdir()
|
|
metadata.joinpath("METADATA").write_text(f"Name: litellm\nVersion: {installed}\n")
|
|
|
|
def installed_distribution(name: str) -> Distribution:
|
|
assert name == "litellm"
|
|
return PathDistribution(metadata)
|
|
|
|
monkeypatch.delenv("LITELLM_RELEASE_TAG", raising=False)
|
|
monkeypatch.delenv("LENS_WORKER_IMAGE", raising=False)
|
|
monkeypatch.setattr(release, "distribution", installed_distribution)
|
|
monkeypatch.setattr(release, "__file__", str(tmp_path / "litellm/proxy/lens/release.py"))
|
|
assert release.release_tag() == expected
|
|
assert worker_image() == f"ghcr.io/berriai/litellm-lens-worker:{expected}"
|
|
|
|
|
|
@pytest.mark.parametrize("source", ("checkout", "direct-install", "unversioned-container", "missing-package"))
|
|
def test_unknown_source_never_falls_back_to_a_package_version_or_image_override(
|
|
monkeypatch: pytest.MonkeyPatch, tmp_path: Path, source: str
|
|
) -> None:
|
|
from litellm.proxy.lens import release
|
|
|
|
metadata: Final = tmp_path / "litellm.dist-info"
|
|
metadata.mkdir()
|
|
metadata.joinpath("METADATA").write_text("Name: litellm\nVersion: 1.2.3\n")
|
|
if source == "direct-install":
|
|
metadata.joinpath("direct_url.json").write_text('{"url":"file:///checkout","dir_info":{"editable":true}}')
|
|
|
|
def installed_distribution(name: str) -> Distribution:
|
|
if source == "missing-package":
|
|
raise PackageNotFoundError(name)
|
|
return PathDistribution(metadata)
|
|
|
|
monkeypatch.delenv("LITELLM_RELEASE_TAG", raising=False)
|
|
monkeypatch.setenv("LENS_WORKER_IMAGE", "registry.example/lens-worker:old")
|
|
monkeypatch.setattr(release, "distribution", installed_distribution)
|
|
if source != "checkout":
|
|
monkeypatch.setattr(release, "__file__", str(tmp_path / "litellm/proxy/lens/release.py"))
|
|
if source == "unversioned-container":
|
|
monkeypatch.setenv("LITELLM_RELEASE_TAG", "")
|
|
assert release.release_tag() == ""
|
|
assert worker_image() == ""
|