ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: skip datazone models

Browse Source
This commit is contained in:
Krrish Dholakia 2025-03-12 16:43:38 -07:00
parent 1250ae6645
commit 81d8a35348
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/litellm_utils_tests/test_supports_tool_choice.py
View File

@ -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