tracer-cloud--opensre/tests/packaging/test_litellm_bundle_contract.py

"""Regression tests for LiteLLM package data in frozen release binaries.

Azure OpenAI and ``OPENSRE_LLM_TRANSPORT=litellm`` import ``litellm`` at runtime.
LiteLLM reads JSON price/context files from its package directory on import; the
release PyInstaller build must bundle them under ``_internal/litellm/`` or the
binary crashes with ``FileNotFoundError`` (see issue #3631).

Workflow contract assertions live in ``tests/github_ci/test_release_workflow.py``.
"""

from __future__ import annotations

from pathlib import Path

# LiteLLM loads this file during import; it must be present in frozen bundles.
_REQUIRED_LITELLM_DATA_FILE = "model_prices_and_context_window_backup.json"


def test_litellm_package_ships_required_price_context_backup() -> None:
    """Dev installs must still expose LiteLLM's price/context JSON (sanity check)."""
    import litellm

    data_path = Path(litellm.__file__).parent / _REQUIRED_LITELLM_DATA_FILE
    assert data_path.is_file(), f"expected LiteLLM data file at {data_path}"