phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: use more descriptive flag

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-07 18:59:46 -07:00
parent b7e31638fd
commit ec0b511119
3 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_proxy_server.py
View file

@ -816,7 +816,7 @@ async def test_create_user_default_budget(prisma_client):
setattr(litellm.proxy.proxy_server, "prisma_client", prisma_client)
setattr(litellm.proxy.proxy_server, "master_key", "sk-1234")
setattr(litellm, "max_user_budget", 10)
setattr(litellm, "max_internal_user_budget", 10)
await litellm.proxy.proxy_server.prisma_client.connect()
user = f"ishaan {uuid.uuid4().hex}"
request = NewUserRequest(user_id=user) # create a key with no budget
@ -834,5 +834,5 @@ async def test_create_user_default_budget(prisma_client):
assert (
mock_client.call_args.kwargs["data"]["max_budget"]
== litellm.max_user_budget
== litellm.max_internal_user_budget
)