From 3d296926aeee4941f17c73ab446281f220b54af3 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 19 Nov 2024 13:57:56 -0800 Subject: [PATCH] test_provider_budgets_e2e_test --- tests/local_testing/test_provider_budgets.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/local_testing/test_provider_budgets.py b/tests/local_testing/test_provider_budgets.py index a26302e6d..17371da54 100644 --- a/tests/local_testing/test_provider_budgets.py +++ b/tests/local_testing/test_provider_budgets.py @@ -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(