tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-20 02:28:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
Download Patch File Download Diff File Expand all files Collapse all files
tests/litellm_utils_tests/test_utils.py
+3 -17
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():