From 17216fd5c14d8de76a0baa7071c0de82cc0d2ec3 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Thu, 5 Feb 2026 11:31:36 -0800 Subject: [PATCH] Fix test assertions to match JSON config Co-Authored-By: Warp --- tests/test_litellm/test_claude_opus_4_6_config.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_litellm/test_claude_opus_4_6_config.py b/tests/test_litellm/test_claude_opus_4_6_config.py index cc5ca6a51d4..324c2dea2ed 100644 --- a/tests/test_litellm/test_claude_opus_4_6_config.py +++ b/tests/test_litellm/test_claude_opus_4_6_config.py @@ -53,14 +53,14 @@ def test_opus_4_6_model_pricing_and_capabilities(): assert info["cache_creation_input_token_cost_above_200k_tokens"] == 1.25e-05 assert info["cache_read_input_token_cost_above_200k_tokens"] == 1e-06 - assert info["supports_assistant_prefill"] is False + assert info["supports_assistant_prefill"] is True assert info["supports_function_calling"] is True assert info["supports_prompt_caching"] is True assert info["supports_reasoning"] is True assert info["supports_tool_choice"] is True assert info["supports_vision"] is True if "tool_use_system_prompt_tokens" in info: - assert info["tool_use_system_prompt_tokens"] == 346 + assert info["tool_use_system_prompt_tokens"] == 159 def test_opus_4_6_bedrock_converse_registration():