From 1822c45c0ec3d0ac200e273618cd2b95fa3c5962 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 26 Mar 2025 11:13:21 -0700 Subject: [PATCH] test fix test_pick_cheapest_chat_model_from_llm_provider --- tests/litellm_utils_tests/test_utils.py | 20 +++----------------- 1 file changed, 3 insertions(+), 17 deletions(-) diff --git a/tests/litellm_utils_tests/test_utils.py b/tests/litellm_utils_tests/test_utils.py index ef9bf30c13..fea225e4a3 100644 --- a/tests/litellm_utils_tests/test_utils.py +++ b/tests/litellm_utils_tests/test_utils.py @@ -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():