diff --git a/tests/test_litellm/reasoning_effort_grid_v4/grid_spec.py b/tests/test_litellm/reasoning_effort_grid_v4/grid_spec.py index 7ba676118a..cbd2cb5a9e 100644 --- a/tests/test_litellm/reasoning_effort_grid_v4/grid_spec.py +++ b/tests/test_litellm/reasoning_effort_grid_v4/grid_spec.py @@ -114,7 +114,7 @@ _BEDROCK_REQ = frozenset({"AWS_ACCESS_KEY_ID", "AWS_SECRET_ACCESS_KEY"}) _CAPS_OPUS_4_7: FrozenSet[str] = frozenset( {"supports_xhigh_reasoning_effort", "supports_max_reasoning_effort"} ) -_CAPS_OPUS_4_6: FrozenSet[str] = frozenset({"supports_max_reasoning_effort"}) +_CAPS_4_6: FrozenSet[str] = frozenset({"supports_max_reasoning_effort"}) _CAPS_NONE: FrozenSet[str] = frozenset() @@ -131,7 +131,7 @@ ANTHROPIC_DIRECT_MODELS: Tuple[ModelEntry, ...] = ( model="anthropic/claude-sonnet-4-6", mode="adaptive", required_env=_ANTHROPIC_REQ, - caps=_CAPS_NONE, + caps=_CAPS_4_6, ), ModelEntry( alias="claude-haiku-4-5", @@ -156,14 +156,14 @@ AZURE_AI_MODELS: Tuple[ModelEntry, ...] = ( model="azure_ai/claude-opus-4-6", mode="adaptive", required_env=_AZURE_FOUNDRY_REQ, - caps=_CAPS_OPUS_4_6, + caps=_CAPS_4_6, ), ModelEntry( alias="azure-claude-sonnet-4-6", model="azure_ai/claude-sonnet-4-6", mode="adaptive", required_env=_AZURE_FOUNDRY_REQ, - caps=_CAPS_NONE, + caps=_CAPS_4_6, ), ModelEntry( alias="azure-claude-haiku-4-5", @@ -190,7 +190,7 @@ VERTEX_AI_MODELS: Tuple[ModelEntry, ...] = ( mode="adaptive", extra_params=(("vertex_location", "us-east5"),), required_env=_VERTEX_REQ, - caps=_CAPS_OPUS_4_6, + caps=_CAPS_4_6, ), ModelEntry( alias="vertex-claude-sonnet-4-6", @@ -198,7 +198,7 @@ VERTEX_AI_MODELS: Tuple[ModelEntry, ...] = ( mode="adaptive", extra_params=(("vertex_location", "us-east5"),), required_env=_VERTEX_REQ, - caps=_CAPS_NONE, + caps=_CAPS_4_6, ), ModelEntry( alias="vertex-claude-haiku-4-5", @@ -226,7 +226,7 @@ BEDROCK_CONVERSE_MODELS: Tuple[ModelEntry, ...] = ( mode="adaptive", extra_params=(("aws_region_name", "us-east-1"),), required_env=_BEDROCK_REQ, - caps=_CAPS_OPUS_4_6, + caps=_CAPS_4_6, ), ModelEntry( alias="bedrock-claude-sonnet-4-6", @@ -234,7 +234,7 @@ BEDROCK_CONVERSE_MODELS: Tuple[ModelEntry, ...] = ( mode="adaptive", extra_params=(("aws_region_name", "us-east-1"),), required_env=_BEDROCK_REQ, - caps=_CAPS_NONE, + caps=_CAPS_4_6, ), ModelEntry( alias="bedrock-claude-sonnet-4-5", @@ -254,7 +254,7 @@ BEDROCK_INVOKE_CHAT_MODELS: Tuple[ModelEntry, ...] = ( mode="adaptive", extra_params=(("aws_region_name", "us-east-1"),), required_env=_BEDROCK_REQ, - caps=_CAPS_OPUS_4_6, + caps=_CAPS_4_6, ), ModelEntry( alias="bedrock-invoke-claude-sonnet-4-6", @@ -262,7 +262,7 @@ BEDROCK_INVOKE_CHAT_MODELS: Tuple[ModelEntry, ...] = ( mode="adaptive", extra_params=(("aws_region_name", "us-east-1"),), required_env=_BEDROCK_REQ, - caps=_CAPS_NONE, + caps=_CAPS_4_6, ), ModelEntry( alias="bedrock-invoke-claude-opus-4-5",