scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-26 06:03:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 35

test(count_tokens): make strict error envelope coverage explicit

Browse source
This commit is contained in:
mateo-berri 2026-10-19 14:30:57 -07:00
parent d522e2e09b
commit e9e7e32142
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tests/integration/providers/test_count_tokens_error_shapes_wire.py
View file

@ -1,5 +1,6 @@
import json
from collections.abc import Iterator
from pathlib import Path
from typing import Final
import anthropic
@ -29,13 +30,22 @@ _PROVIDER_KEY_VARIABLES: Final = (
@pytest.fixture(scope="module")
def keyless_proxy(tmp_path_factory: pytest.TempPathFactory) -> Iterator[OwnedProxy]:
def provider_error_config(tmp_path_factory: pytest.TempPathFactory) -> Path:
config: Final = tmp_path_factory.mktemp("count-tokens-provider-errors") / "config.yaml"
base: Final = Path(__file__).resolve().parents[1] / "proxy_config.yaml"
config.write_text(base.read_text() + "\nlitellm_settings:\n disable_token_counter: true\n")
return config
@pytest.fixture(scope="module")
def keyless_proxy(tmp_path_factory: pytest.TempPathFactory, provider_error_config: Path) -> Iterator[OwnedProxy]:
with (
gateway_from_environment() as gateway,
owned_proxy_process(
gateway,
tmp_path_factory.mktemp("count-tokens-error-shapes"),
{},
config=provider_error_config,
remove_environment=_PROVIDER_KEY_VARIABLES,
workers=1,
) as owned,