From 152456e10d7acfa1d8ecd878856fd1640784b16d Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 22 Jan 2024 12:16:16 -0800 Subject: [PATCH] (test) test_call_with_key_over_budget_stream --- litellm/tests/test_key_generate_prisma.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index effac5890..484c9d5a5 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -681,8 +681,8 @@ def test_call_with_key_over_budget_stream(prisma_client): async def test(): await litellm.proxy.proxy_server.prisma_client.connect() - request = NewUserRequest(max_budget=0.00001) - key = await new_user(request) + request = GenerateKeyRequest(max_budget=0.00001) + key = await generate_key_fn(request) print(key) generated_key = key.key @@ -739,5 +739,5 @@ def test_call_with_key_over_budget_stream(prisma_client): asyncio.run(test()) except Exception as e: error_detail = e.detail - assert "Authentication Error, ExceededBudget:" in error_detail + assert "Authentication Error, ExceededTokenBudget:" in error_detail print(vars(e))