test: check both orders of a custom handler and a video entry under one name
Some checks failed
LiteLLM Rust / rust-lint (push) Has been cancelled
LiteLLM Rust / rust-test (push) Has been cancelled
LiteLLM Rust / rust-wheel (push) Has been cancelled

This commit is contained in:
mateo-berri 2026-11-10 01:44:58 -08:00
parent a278d6939d
commit bb7a2fda8f

View file

@ -2365,14 +2365,17 @@ def test_completion_passes_vector_store_params_to_custom_handler(
assert received_params.items() >= vector_store_params.items()
def test_completion_passes_vector_store_params_to_custom_handler_listed_after_other_entries(
monkeypatch: pytest.MonkeyPatch,
@pytest.mark.parametrize(
"shared_name_handlers",
((OpenAIVideoConfig(), OptionalParamsEchoHandler()), (OptionalParamsEchoHandler(), OpenAIVideoConfig())),
)
def test_completion_passes_vector_store_params_to_custom_handler_among_other_map_entries(
monkeypatch: pytest.MonkeyPatch, shared_name_handlers: tuple[CustomLLM | BaseVideoConfig, ...]
) -> None:
_register_handlers(
monkeypatch,
("other-handler", OptionalParamsEchoHandler()),
("echo-handler", OpenAIVideoConfig()),
("echo-handler", OptionalParamsEchoHandler()),
*(("echo-handler", handler) for handler in shared_name_handlers),
)
response: Final = litellm.completion(