mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-27 01:22:18 +00:00
7975987107
Fifty six of the deleted tests turn out to assert the output of litellm code rather than the catalog lookup itself, things like map_openai_params, get_supported_openai_params, should_fake_stream, transform_request bodies, cost_per_token arithmetic, get_llm_provider routing, and provider config dispatch. They only happen to read shipped entries as inputs, so they belong in the later rewrite that injects a local model_cost, not in this deletion Each one is restored verbatim from origin/main along with the fixtures, helpers, constants and imports it needs, and tests/test_litellm/test_sambanova_model_metadata.py is restored wholesale Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
25 lines
937 B
Python
25 lines
937 B
Python
import json
|
|
from pathlib import Path
|
|
|
|
from litellm.litellm_core_utils.get_llm_provider_logic import get_llm_provider
|
|
|
|
|
|
def test_sambanova_minimax_m27_model_info():
|
|
model = "sambanova/MiniMax-M2.7"
|
|
json_path = Path(__file__).parents[2] / "model_prices_and_context_window.json"
|
|
with open(json_path) as f:
|
|
model_cost = json.load(f)
|
|
|
|
info = model_cost.get(model)
|
|
assert info is not None, f"{model} not found in model_prices_and_context_window.json"
|
|
assert info["litellm_provider"] == "sambanova"
|
|
assert info["mode"] == "chat"
|
|
assert info["input_cost_per_token"] > 0
|
|
assert info["output_cost_per_token"] > 0
|
|
assert info["supports_function_calling"] is True
|
|
assert info["supports_reasoning"] is True
|
|
assert info["supports_tool_choice"] is True
|
|
|
|
routed_model, provider, _, _ = get_llm_provider(model=model)
|
|
assert routed_model == "MiniMax-M2.7"
|
|
assert provider == "sambanova"
|