From d5b1cc7150b00baa897ddd83c473ce00719cb774 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 23 Jan 2024 08:36:49 -0800 Subject: [PATCH] (ci/cd) run prisma.connect() before test --- litellm/tests/test_key_generate_prisma.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index e3ad40889..f50df4570 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -658,6 +658,7 @@ def test_key_generate_with_custom_auth(prisma_client): async def test(): try: + await litellm.proxy.proxy_server.prisma_client.connect() request = GenerateKeyRequest() key = await generate_key_fn(request) pytest.fail(f"Expected an exception. Got {key}")