(ci/cd) run prisma.connect() before test

2024-01-23 08:36:49 -08:00 · 2024-01-23 08:36:49 -08:00 · d5b1cc7150
commit d5b1cc7150
parent 5ad0971c5a
1 changed files with 1 additions and 0 deletions
--- 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}")