diff --git a/tests/litellm_utils_tests/test_supports_tool_choice.py b/tests/litellm_utils_tests/test_supports_tool_choice.py index 3b512f41f9..a68aea4f5a 100644 --- a/tests/litellm_utils_tests/test_supports_tool_choice.py +++ b/tests/litellm_utils_tests/test_supports_tool_choice.py @@ -117,7 +117,9 @@ async def test_supports_tool_choice(): """ # Load model prices litellm._turn_on_debug() - with open("./model_prices_and_context_window.json", "r") as f: + local_path = "../../model_prices_and_context_window.json" + prod_path = "./model_prices_and_context_window.json" + with open(local_path, "r") as f: model_prices = json.load(f) litellm.model_cost = model_prices config_manager = ProviderConfigManager() @@ -133,6 +135,8 @@ async def test_supports_tool_choice(): or any(provider in model_name for provider in OLD_PROVIDERS) or model_info["litellm_provider"] in OLD_PROVIDERS or model_name in block_list + or "azure/eu" in model_name + or "azure/us" in model_name ): continue