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

test(test_key_generate_prisma.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-29 20:30:43 -07:00
parent 3810b050c1
commit cbf35087c7
1 changed files with 13 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

18
litellm/tests/test_key_generate_prisma.py
View file

@ -576,9 +576,17 @@ def test_call_with_proxy_over_budget_stream(prisma_client):
litellm_proxy_budget_name = f"litellm-proxy-budget-{time.time()}" litellm_proxy_budget_name = f"litellm-proxy-budget-{time.time()}"
setattr( setattr(
litellm.proxy.proxy_server, litellm.proxy.proxy_server,
"litellm_proxy_budget_name", "litellm_proxy_admin_name",
litellm_proxy_budget_name, litellm_proxy_budget_name,
) )
setattr(litellm, "max_budget", 0.00001)
from litellm.proxy.proxy_server import user_api_key_cache
user_api_key_cache.set_cache(
key="{}:spend".format(litellm_proxy_budget_name), value=0
)
setattr(litellm.proxy.proxy_server, "user_api_key_cache", user_api_key_cache)
from litellm._logging import verbose_proxy_logger from litellm._logging import verbose_proxy_logger
import logging import logging
@ -589,10 +597,10 @@ def test_call_with_proxy_over_budget_stream(prisma_client):
async def test(): async def test():
await litellm.proxy.proxy_server.prisma_client.connect() await litellm.proxy.proxy_server.prisma_client.connect()
## CREATE PROXY + USER BUDGET ## ## CREATE PROXY + USER BUDGET ##
request = NewUserRequest( # request = NewUserRequest(
max_budget=0.00001, user_id=litellm_proxy_budget_name # max_budget=0.00001, user_id=litellm_proxy_budget_name
) # )
await new_user(request) # await new_user(request)
request = NewUserRequest() request = NewUserRequest()
key = await new_user(request) key = await new_user(request)
print(key) print(key)