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

test(test_end_users.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-13 21:46:14 -07:00
parent d0fb685c56
commit 9cca25f874
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/test_end_users.py
View file

@ -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