scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-15 09:12:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 311

test(cost_map): keep the exact fill-missing assertion for Claude thinking rules

Browse source 
Build the expected baseline part from the loaded map so a rule that leaks an extra key into Claude entries still fails, and wrap lines over 120 characters.
This commit is contained in:
ryan-crabbe-berri 2026-11-02 22:50:53 -08:00
parent 203f692b0b
commit 67a1482ee1
1 changed files with 13 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
tests/unit/litellm_core_utils/test_fallback_generalizations.py
View file

@ -915,11 +915,12 @@ def test_shipped_claude_thinking_rules_backfill_only_anthropic(shipped_cost_map)
assert info.get("supports_adaptive_thinking") is None
assert info.get("supports_legacy_thinking") is None
assert info.get("max_input_tokens") is None
assert match_fill_missing_generalizations("claude-sonnet-4-6", "anthropic").items() >= {
assert match_fill_missing_generalizations("claude-sonnet-4-6", "anthropic") == {
**_claude_baseline()["model_info"],
"supports_adaptive_thinking": True,
"supports_legacy_thinking": True,
"supports_tool_search": True,
}.items()
}
assert match_fill_missing_generalizations("claude-sonnet-4-6", "perplexity") is None
@ -964,7 +965,13 @@ def test_shipped_tool_search_rule_fills_mapped_claude_entries_without_flag(shipp
assert match_fill_missing_generalizations("claude-opus-5", "perplexity") is None
CLAUDE_BASELINE_FILL_PROVIDERS: Final = ("anthropic", "azure_ai", "bedrock", "bedrock_converse", "vertex_ai-anthropic_models")
CLAUDE_BASELINE_FILL_PROVIDERS: Final = (
"anthropic",
"azure_ai",
"bedrock",
"bedrock_converse",
"vertex_ai-anthropic_models",
)
def _claude_baseline() -> Mapping[str, object]:
@ -989,7 +996,9 @@ def test_shipped_claude_baseline_fills_only_missing_keys_on_mapped_entries(shipp
"""A mapped Claude entry from a first-party host gets each baseline key it leaves out,
while every baseline key the entry sets keeps the entry's own value."""
baseline: Final = _claude_baseline()["model_info"]
entries: Final = tuple((key, entry) for key, entry in _mapped_claude_chat_entries() if entry["litellm_provider"] == provider)
entries: Final = tuple(
(key, entry) for key, entry in _mapped_claude_chat_entries() if entry["litellm_provider"] == provider
)
assert any(baseline.keys() - entry.keys() for _, entry in entries)
assert any(_fields_overriding_baseline(entry, baseline) for _, entry in entries)