mirror of
https://github.com/usestrix/strix.git
synced 2026-10-11 03:37:54 +00:00
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
"""Vertex AI and Bedrock support ship with every install, not as extras."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import importlib
|
|
import tomllib
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parent.parent
|
|
|
|
|
|
def _pyproject() -> dict[str, object]:
|
|
return tomllib.loads((ROOT / "pyproject.toml").read_text(encoding="utf-8"))
|
|
|
|
|
|
@pytest.mark.parametrize("package", ["google-auth", "boto3"])
|
|
def test_provider_packages_are_regular_dependencies(package: str) -> None:
|
|
project = _pyproject()["project"]
|
|
assert isinstance(project, dict)
|
|
assert any(req.startswith(package) for req in project["dependencies"])
|
|
assert "optional-dependencies" not in project
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"module",
|
|
[
|
|
"google.auth",
|
|
"google.auth.transport.requests",
|
|
"google.oauth2.service_account",
|
|
"boto3",
|
|
"litellm.llms.vertex_ai.vertex_llm_base",
|
|
"litellm.llms.bedrock.base_aws_llm",
|
|
],
|
|
)
|
|
def test_provider_modules_import(module: str) -> None:
|
|
importlib.import_module(module)
|
|
|
|
|
|
def test_pyinstaller_spec_bundles_google_auth() -> None:
|
|
spec = (ROOT / "strix.spec").read_text(encoding="utf-8")
|
|
assert "collect_submodules('google.auth')" in spec
|
|
assert "'google.auth'," not in spec
|
|
assert "'google.oauth2'," not in spec
|