phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix test provider budgets

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-19 17:26:31 -08:00
parent 95f21722a0
commit 7ceee027c8
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/local_testing/test_provider_budgets.py
View file

@ -39,8 +39,6 @@ async def test_provider_budgets_e2e_test():
} }
router = Router( router = Router(
routing_strategy="provider-budget-routing",
routing_strategy_args=provider_budget_config,
model_list=[ model_list=[
{ {
"model_name": "gpt-3.5-turbo", # openai model name "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"}, "model_info": {"id": "openai-model-id"},
}, },
], ],
provider_budget_config=provider_budget_config,
redis_host=os.getenv("REDIS_HOST"), redis_host=os.getenv("REDIS_HOST"),
redis_port=int(os.getenv("REDIS_PORT")), redis_port=int(os.getenv("REDIS_PORT")),
redis_password=os.getenv("REDIS_PASSWORD"), redis_password=os.getenv("REDIS_PASSWORD"),
@ -99,8 +98,6 @@ async def test_provider_budgets_e2e_test_expect_to_fail():
} }
router = Router( router = Router(
routing_strategy="provider-budget-routing",
routing_strategy_args=provider_budget_config,
model_list=[ model_list=[
{ {
"model_name": "anthropic/*", # openai model name "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_host=os.getenv("REDIS_HOST"),
redis_port=int(os.getenv("REDIS_PORT")), redis_port=int(os.getenv("REDIS_PORT")),
redis_password=os.getenv("REDIS_PASSWORD"), redis_password=os.getenv("REDIS_PASSWORD"),
provider_budget_config=provider_budget_config,
) )
response = await router.acompletion( response = await router.acompletion(