Merge pull request #41627 from BerriAI/litellm_fireworks_minimax_m3_vision_tests

test(fireworks_ai): stop pinning vision support on minimax-m3
This commit is contained in:
kerry-berri 2026-09-17 12:28:23 -07:00 committed by GitHub
commit 0f5bc0ffa9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 7 additions and 7 deletions

View file

@ -973,12 +973,11 @@ def test_thinking_and_reasoning_effort_conflict_rejected():
)
def test_minimax_m3_supports_vision_from_model_map():
def test_llama_vision_supports_vision_from_model_map():
config = FireworksAIConfig()
for model in [
"fireworks_ai/accounts/fireworks/models/minimax-m3",
"fireworks_ai/minimax-m3",
"fireworks_ai/accounts/fireworks/models/llama-v3p2-11b-vision-instruct",
]:
assert supports_vision(model=model, custom_llm_provider="fireworks_ai") is True
assert config.get_provider_info(model)["supports_vision"] is True
@ -1052,7 +1051,7 @@ def test_transform_messages_helper_allows_vision_image_inputs():
]
out = config._transform_messages_helper(
messages, model="accounts/fireworks/models/minimax-m3", litellm_params={}
messages, model="accounts/fireworks/models/llama-v3p2-11b-vision-instruct", litellm_params={}
)
assert out == messages
@ -1117,7 +1116,7 @@ def test_transform_messages_helper_no_transform_inline():
}
]
out = config._transform_messages_helper(
messages, model="accounts/fireworks/models/minimax-m3", litellm_params={}
messages, model="accounts/fireworks/models/llama-v3p2-11b-vision-instruct", litellm_params={}
)
block = out[0]["content"][0]
assert block["image_url"] == url

View file

@ -3546,7 +3546,7 @@ _FIREWORKS_MODELS = [
"accounts/fireworks/models/minimax-m3",
512000,
512000,
True,
None,
True,
),
(
@ -3654,7 +3654,8 @@ def _assert_fireworks_entry(
assert info["supports_tool_choice"] is True
assert info["supports_reasoning"] is expected_reasoning
assert info["supports_response_schema"] is True
assert info["supports_vision"] is expected_vision
if expected_vision is not None:
assert info["supports_vision"] is expected_vision
@pytest.fixture