scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-03 07:23:33 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 185

test: accept the mirrored cost map's extraction mode, new row fields and openinfer rows

Browse source 
The mirrored map carries extraction mode rows, explicit_cache_read_input_token_cost and supports_bedrock_native_structured_output, and openinfer rows for a provider this line has no model set for. stable/1.106.x widened the same checks in ebf92c9a45 and f44d4a70c0
This commit is contained in:
Yuneng Jiang 2026-10-27 18:08:19 -07:00
parent 99671ddcf8
commit 82925fc3c0
No known key found for this signature in database
2 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/unit/test_model_prices_and_context_window.py
View file

@ -632,7 +632,7 @@ class TestBedrockClaude35SonnetContextWindow:
assert not set(above_200k_fields).intersection(rows[model])
PROVIDER_LABELS_WITHOUT_A_MODEL_SET: Final = frozenset({"sagemaker", "bedrock_converse"})
PROVIDER_LABELS_WITHOUT_A_MODEL_SET: Final = frozenset({"sagemaker", "bedrock_converse", "openinfer"})
MODES_SERVED_OUTSIDE_THE_LLM_PROVIDER_REGISTRY: Final = frozenset({"search", "decisions", "evaluation"})
VERTEX_FAMILIES_A_VERTEX_WILDCARD_GRANT_DOES_NOT_LIST: Final = frozenset(
{

7
tests/unit/test_utils.py
View file

@ -951,6 +951,7 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
"cache_creation_input_token_cost_priority": {"type": "number"},
"cache_creation_input_token_cost_ultrafast": {"type": "number"},
"cache_read_input_token_cost": {"type": "number"},
"explicit_cache_read_input_token_cost": {"type": "number"},
"cache_read_input_token_cost_above_32k_tokens": {"type": "number"},
"cache_read_input_token_cost_above_100k_tokens": {"type": "number"},
"cache_read_input_token_cost_above_100k_tokens_batches": {"type": "number"},
@ -1058,6 +1059,7 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
"image_edit",
"embedding",
"evaluation",
"extraction",
"guardrail",
"image_generation",
"video_generation",
@ -1138,6 +1140,7 @@ def test_aaamodel_prices_and_context_window_json_is_valid():
"supports_box_questions": {"type": "boolean"},
"supports_vision": {"type": "boolean"},
"supports_web_search": {"type": "boolean"},
"supports_bedrock_native_structured_output": {"type": "boolean"},
"token_count_framing": {"type": "integer"},
"token_count_model": {
"type": "object",
@ -8683,9 +8686,9 @@ def test_models_by_provider():
continue
elif k == "sample_spec":
continue
elif v["litellm_provider"] == "sagemaker" or v["litellm_provider"] == "bedrock_converse":
elif v["litellm_provider"] in ("sagemaker", "bedrock_converse", "openinfer"):
continue
elif v.get("mode") in ("search", "decisions", "evaluation"):
elif v.get("mode") in ("search", "decisions", "evaluation", "extraction"):
continue
else:
providers.add(v["litellm_provider"])