mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-19 00:01:29 +00:00
test: drop the fireworks vision flag pin that reads the shipped cost map
get_provider_info is a passthrough over the cost map entry, so asserting supports_vision on named fireworks models pins a vendor capability rather than litellm behavior Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
parent
7975987107
commit
eff323682e
1 changed files with 0 additions and 13 deletions
|
|
@ -4,7 +4,6 @@ from unittest.mock import MagicMock, patch
|
|||
import pytest
|
||||
|
||||
import litellm
|
||||
from litellm import supports_vision
|
||||
from litellm.constants import SESSION_ID_GENERATED_METADATA_KEY
|
||||
from litellm.llms.fireworks_ai.chat.transformation import FireworksAIConfig
|
||||
from litellm.llms.fireworks_ai.common_utils import get_fireworks_session_id
|
||||
|
|
@ -939,18 +938,6 @@ def test_thinking_and_reasoning_effort_conflict_rejected():
|
|||
)
|
||||
|
||||
|
||||
def test_llama_vision_supports_vision_from_model_map():
|
||||
config = FireworksAIConfig()
|
||||
|
||||
for model in [
|
||||
"fireworks_ai/accounts/fireworks/models/llama-v3p2-11b-vision-instruct",
|
||||
"fireworks_ai/accounts/fireworks/models/minimax-m3",
|
||||
"fireworks_ai/minimax-m3",
|
||||
]:
|
||||
assert supports_vision(model=model, custom_llm_provider="fireworks_ai") is True
|
||||
assert config.get_provider_info(model)["supports_vision"] is True
|
||||
|
||||
|
||||
def test_transform_messages_helper_rejects_file_blocks():
|
||||
config = FireworksAIConfig()
|
||||
messages = [
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue