scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

test fix test_pick_cheapest_chat_model_from_llm_provider

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-26 11:13:21 -07:00
parent baa9b34950
commit 1822c45c0e
1 changed files with 3 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
tests/litellm_utils_tests/test_utils.py
View file

@ -1532,26 +1532,12 @@ def test_supports_vision_gemini():
def test_pick_cheapest_chat_model_from_llm_provider():
from litellm.litellm_core_utils.llm_request_utils import (
LitellmCoreRequestUtils,
pick_cheapest_chat_models_from_llm_provider,
)
assert (
len(
LitellmCoreRequestUtils.pick_cheapest_chat_models_from_llm_provider(
"openai", n=3
)
)
== 3
)
assert len(pick_cheapest_chat_models_from_llm_provider("openai", n=3)) == 3
assert (
len(
LitellmCoreRequestUtils.pick_cheapest_chat_models_from_llm_provider(
"unknown", n=1
)
)
== 0
)
assert len(pick_cheapest_chat_models_from_llm_provider("unknown", n=1)) == 0
def test_get_potential_model_names():