From 9cca25f87415a586d863ff12ae11fc201a7f9eaf Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 13 Jul 2024 21:46:14 -0700 Subject: [PATCH] test(test_end_users.py): fix test --- tests/test_end_users.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/test_end_users.py b/tests/test_end_users.py index 4ee894987..c358658c8 100644 --- a/tests/test_end_users.py +++ b/tests/test_end_users.py @@ -186,7 +186,7 @@ async def test_end_user_specific_region(): key = key_gen["key"] for _ in range(3): - client = AsyncOpenAI(api_key=key, base_url="http://0.0.0.0:4000") + client = AsyncOpenAI(api_key=key, base_url="http://0.0.0.0:4000", max_retries=0) print("SENDING USER PARAM - {}".format(end_user_obj["user_id"])) result = await client.chat.completions.with_raw_response.create( @@ -225,7 +225,7 @@ async def test_enduser_tpm_limits_non_master_key(): key = key_gen["key"] # chat completion 1 - client = AsyncOpenAI(api_key=key, base_url="http://0.0.0.0:4000") + client = AsyncOpenAI(api_key=key, base_url="http://0.0.0.0:4000", max_retries=0) # chat completion 2 passed = 0 @@ -267,7 +267,9 @@ async def test_enduser_tpm_limits_with_master_key(): ) # chat completion 1 - client = AsyncOpenAI(api_key="sk-1234", base_url="http://0.0.0.0:4000") + client = AsyncOpenAI( + api_key="sk-1234", base_url="http://0.0.0.0:4000", max_retries=0 + ) # chat completion 2 passed = 0