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

(test) test_call_with_key_over_budget_stream

Browse source
This commit is contained in:
ishaan-jaff 2024-01-22 12:16:16 -08:00
parent e846b8fca8
commit 152456e10d
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_key_generate_prisma.py
View file

@ -681,8 +681,8 @@ def test_call_with_key_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()
request = NewUserRequest(max_budget=0.00001) request = GenerateKeyRequest(max_budget=0.00001)
key = await new_user(request) key = await generate_key_fn(request)
print(key) print(key)
generated_key = key.key generated_key = key.key
@ -739,5 +739,5 @@ def test_call_with_key_over_budget_stream(prisma_client):
asyncio.run(test()) asyncio.run(test())
except Exception as e: except Exception as e:
error_detail = e.detail error_detail = e.detail
assert "Authentication Error, ExceededBudget:" in error_detail assert "Authentication Error, ExceededTokenBudget:" in error_detail
print(vars(e)) print(vars(e))