test_provider_budgets_e2e_test

This commit is contained in:
Ishaan Jaff 2024-11-19 13:57:56 -08:00
parent ba36ecc65a
commit 3d296926ae

View file

@ -60,6 +60,9 @@ async def test_provider_budgets_e2e_test():
"model_info": {"id": "openai-model-id"},
},
],
redis_host=os.getenv("REDIS_HOST"),
redis_port=int(os.getenv("REDIS_PORT")),
redis_password=os.getenv("REDIS_PASSWORD"),
)
response = await router.acompletion(
@ -106,6 +109,9 @@ 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"),
)
response = await router.acompletion(