scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-14 09:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 214

test(proxy): align structured output metadata expectation

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
mateo 2026-11-06 21:18:43 +00:00
parent 3f8f615086
commit 2ecad649ec
1 changed files with 11 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
tests/unit/proxy/proxy_server/test_routes_model_info.py
View file

@ -491,16 +491,16 @@ def test_model_info_reports_null_cost_for_unpriced_deployment_and_zero_for_decla
@pytest.mark.parametrize(
("deployment_model_info", "native"),
("deployment_model_info", "reported_native"),
[
({}, True),
({"supports_bedrock_native_structured_output": False}, False),
({"supports_bedrock_native_structured_output": False}, True),
({"supports_native_structured_output": True, "supports_bedrock_native_structured_output": False}, True),
],
ids=["native-by-default", "bedrock-opt-out", "explicit-generic-flag"],
ids=["native-by-default", "deprecated-opt-out", "explicit-generic-flag"],
)
def test_model_info_reports_a_deployment_bedrock_native_output_opt_out(
local_model_cost_map: None, deployment_model_info: dict[str, object], native: bool
def test_model_info_reports_the_map_default_without_reading_the_deprecated_opt_out(
local_model_cost_map: None, deployment_model_info: dict[str, object], reported_native: bool
) -> None:
info: Final = proxy_server._enrich_model_info_with_litellm_data(
{
@ -510,7 +510,12 @@ def test_model_info_reports_a_deployment_bedrock_native_output_opt_out(
}
)["model_info"]
assert info["supports_native_structured_output"] is native
assert info["supports_native_structured_output"] is reported_native
if "supports_bedrock_native_structured_output" in deployment_model_info:
assert (
info["supports_bedrock_native_structured_output"]
is deployment_model_info["supports_bedrock_native_structured_output"]
)
@pytest.mark.parametrize(