From 7ceee027c8193f986c344acff4318f4ace671be3 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 19 Nov 2024 17:26:31 -0800 Subject: [PATCH] fix test provider budgets --- tests/local_testing/test_provider_budgets.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/local_testing/test_provider_budgets.py b/tests/local_testing/test_provider_budgets.py index 2622a79e0..5e685cae6 100644 --- a/tests/local_testing/test_provider_budgets.py +++ b/tests/local_testing/test_provider_budgets.py @@ -39,8 +39,6 @@ async def test_provider_budgets_e2e_test(): } router = Router( - routing_strategy="provider-budget-routing", - routing_strategy_args=provider_budget_config, model_list=[ { "model_name": "gpt-3.5-turbo", # openai model name @@ -60,6 +58,7 @@ async def test_provider_budgets_e2e_test(): "model_info": {"id": "openai-model-id"}, }, ], + provider_budget_config=provider_budget_config, redis_host=os.getenv("REDIS_HOST"), redis_port=int(os.getenv("REDIS_PORT")), redis_password=os.getenv("REDIS_PASSWORD"), @@ -99,8 +98,6 @@ async def test_provider_budgets_e2e_test_expect_to_fail(): } router = Router( - routing_strategy="provider-budget-routing", - routing_strategy_args=provider_budget_config, model_list=[ { "model_name": "anthropic/*", # openai model name @@ -112,6 +109,7 @@ async def test_provider_budgets_e2e_test_expect_to_fail(): redis_host=os.getenv("REDIS_HOST"), redis_port=int(os.getenv("REDIS_PORT")), redis_password=os.getenv("REDIS_PASSWORD"), + provider_budget_config=provider_budget_config, ) response = await router.acompletion(