mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-20 00:11:50 +00:00
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:
commit
0f5bc0ffa9
2 changed files with 7 additions and 7 deletions
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue