From 62963146fed633e19100b194c4a3c3d16af2ffee Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 17 Jan 2024 12:23:45 -0800 Subject: [PATCH] (test) prisma /key/gen --- litellm/tests/test_key_generate_prisma.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index 534a0ca817..5b98a886be 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -60,7 +60,7 @@ def test_generate_and_call_with_valid_key(): try: async def test(): - await litellm.litellm.proxy.proxy_server.prisma_client.connect() + await litellm.proxy.proxy_server.prisma_client.connect() request = NewUserRequest() key = await new_user(request) print(key) @@ -87,7 +87,7 @@ def test_call_with_invalid_key(): try: async def test(): - await litellm.litellm.proxy.proxy_server.prisma_client.connect() + await litellm.proxy.proxy_server.prisma_client.connect() generated_key = "bad-key" bearer_token = "Bearer " + generated_key @@ -114,7 +114,7 @@ def test_call_with_invalid_model(): try: async def test(): - await litellm.litellm.proxy.proxy_server.prisma_client.connect() + await litellm.proxy.proxy_server.prisma_client.connect() request = NewUserRequest(models=["mistral"]) key = await new_user(request) print(key) @@ -150,7 +150,7 @@ def test_call_with_valid_model(): try: async def test(): - await litellm.litellm.proxy.proxy_server.prisma_client.connect() + await litellm.proxy.proxy_server.prisma_client.connect() request = NewUserRequest(models=["mistral"]) key = await new_user(request) print(key) @@ -182,7 +182,7 @@ def test_call_with_key_over_budget(): try: async def test(): - await litellm.litellm.proxy.proxy_server.prisma_client.connect() + await litellm.proxy.proxy_server.prisma_client.connect() request = NewUserRequest(max_budget=0.00001) key = await new_user(request) print(key) @@ -254,7 +254,7 @@ def test_call_with_key_over_budget_stream(): try: async def test(): - await litellm.litellm.proxy.proxy_server.prisma_client.connect() + await litellm.proxy.proxy_server.prisma_client.connect() request = NewUserRequest(max_budget=0.00001) key = await new_user(request) print(key)