mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-13 03:59:12 +00:00
3090f4b23e
* test(integration): wait for customer spend rows and use owned OpenRouter cost * ci: isolate integration workers and clean up resources after crashes * docs: describe isolated extensions workers and shard census * fix(ci): restore integration service and fixture guarantees * fix(integration): preserve replay IDs and native route fixtures * fix(tests): scope integration harness imports to each unit case * fix(ci): preserve fixtures and wait for owned workers * fix(ci): isolate nested runs and telemetry cases * fix(ci): make worker readiness and fault tests deterministic * chore(ci): drop MCP fixture repair already covered by main * chore(ci): keep the local catalog fix from main --------- Co-authored-by: yuneng <yuneng@berri.ai>
34 lines
1.7 KiB
Python
34 lines
1.7 KiB
Python
import base64
|
|
from pathlib import Path
|
|
from typing import Final
|
|
|
|
import pytest
|
|
from pydantic import JsonValue
|
|
|
|
|
|
@pytest.mark.parametrize("response_id", ("resp_original", "", "resp_original;part", "resp_response_id:original"))
|
|
def test_response_id_uses_the_deployment_id_and_preserves_provider_identity_and_the_complete_body(
|
|
response_id: str, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
monkeypatch.syspath_prepend(str(Path(__file__).resolve().parents[2]))
|
|
from integration.translation.runner import expected_response_for_deployment
|
|
|
|
original: Final = f"litellm:custom_llm_provider:openai;model_id:recorded-deployment;response_id:{response_id}"
|
|
expected: Final[dict[str, JsonValue]] = {
|
|
"id": "resp_" + base64.b64encode(original.encode()).decode(),
|
|
"content": [{"text": "answer"}],
|
|
}
|
|
result: Final = expected_response_for_deployment(expected, "isolated-deployment")
|
|
updated: Final = f"litellm:custom_llm_provider:openai;model_id:isolated-deployment;response_id:{response_id}"
|
|
assert result == {**expected, "id": "resp_" + base64.b64encode(updated.encode()).decode()}
|
|
assert expected["id"] == "resp_" + base64.b64encode(original.encode()).decode()
|
|
with pytest.raises(AssertionError, match="must identify the selected deployment"):
|
|
expected_response_for_deployment(expected, None)
|
|
|
|
|
|
def test_unencoded_provider_identity_is_preserved_exactly(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.syspath_prepend(str(Path(__file__).resolve().parents[2]))
|
|
from integration.translation.runner import expected_response_for_deployment
|
|
|
|
expected: Final[dict[str, JsonValue]] = {"id": "resp_original", "content": "answer"}
|
|
assert expected_response_for_deployment(expected, "isolated-deployment") is expected
|