phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Revert "changes"

Browse source 
This reverts commit fa88bc9632.
This commit is contained in:
Utkash Dubey 2025-03-10 14:46:21 -07:00
parent 1f17daf52c
commit c7a04140e9
19 changed files with 141 additions and 191 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/local_testing/test_router_utils.py
View file

@ -178,7 +178,8 @@ async def test_update_kwargs_before_fallbacks(call_type):
def test_router_get_model_info_wildcard_routes():
litellm.model_cost = litellm.get_locally_cached_model_cost_map()
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")
router = Router(
model_list=[
{
@ -199,7 +200,8 @@ def test_router_get_model_info_wildcard_routes():
@pytest.mark.asyncio
async def test_router_get_model_group_usage_wildcard_routes():
litellm.model_cost = litellm.get_locally_cached_model_cost_map()
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")
router = Router(
model_list=[
{
@ -295,7 +297,8 @@ async def test_call_router_callbacks_on_failure():
@pytest.mark.asyncio
async def test_router_model_group_headers():
litellm.model_cost = litellm.get_locally_cached_model_cost_map()
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")
from litellm.types.utils import OPENAI_RESPONSE_HEADERS
router = Router(
@ -327,7 +330,8 @@ async def test_router_model_group_headers():
@pytest.mark.asyncio
async def test_get_remaining_model_group_usage():
litellm.model_cost = litellm.get_locally_cached_model_cost_map()
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
litellm.model_cost = litellm.get_model_cost_map(url="")
from litellm.types.utils import OPENAI_RESPONSE_HEADERS
router = Router(