mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-17 23:51:30 +00:00
036a380fa0
Some checks failed
ai-gateway image / ai-gateway release image (push) Waiting to run
LiteLLM Rust / rust-lint (push) Waiting to run
LiteLLM Rust / rust-test (push) Waiting to run
Terraform Modules / fmt, validate, test (aws) (push) Has been cancelled
Terraform Modules / fmt, validate, test (gcp) (push) Has been cancelled
Terraform Provider / gofmt, vet, build, test (push) Has been cancelled
Terraform Provider / Provider endpoints vs proxy OpenAPI schema (push) Has been cancelled
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
"""Batches suite's `client` fixture.
|
|
|
|
The shared lifecycle (resources/scoped_key), proxy liveness gate, and e2e marker
|
|
live in the parent tests/e2e/conftest.py. BatchClient holds the shared ProxyClient, so
|
|
the `resources` fixture cleans up keys through it; tests register file deletes and
|
|
batch cancels via `resources.defer(...)`.
|
|
|
|
Batch deployments (openai-batch, azure-batch, vertex-batch, ...) are registered
|
|
once per session via /model/new and deleted on teardown so they need not live in
|
|
the proxy config.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Final, Iterator
|
|
|
|
import pytest
|
|
|
|
from batch_client import BatchClient, build_client
|
|
from capabilities import PROVIDERS
|
|
from e2e_http import NoBody
|
|
from lifecycle import ResourceManager
|
|
from proxy_client import ProxyClient
|
|
|
|
|
|
def pytest_configure(config: pytest.Config) -> None:
|
|
config.addinivalue_line(
|
|
"markers",
|
|
"covers: registry cell a test covers, e.g. llm.batches.openai.basic.nonstream.works",
|
|
)
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def client(proxy: ProxyClient) -> BatchClient:
|
|
return build_client(proxy)
|
|
|
|
|
|
@pytest.fixture
|
|
def resources(client: BatchClient) -> Iterator[ResourceManager]:
|
|
manager: Final = ResourceManager(client=client.proxy, strict_cleanup=True)
|
|
yield manager
|
|
manager.teardown()
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def batch_deployments(client: BatchClient) -> Iterator[None]:
|
|
probe = client.proxy.probe("/health/liveliness", params=NoBody())
|
|
if not probe.healthy:
|
|
yield
|
|
return
|
|
|
|
registered: list[str] = []
|
|
try:
|
|
for provider in PROVIDERS:
|
|
registered.append(
|
|
client.create_model(provider.model, provider.litellm_params())
|
|
)
|
|
yield
|
|
finally:
|
|
for model_id in registered:
|
|
client.delete_model(model_id)
|