(ci/cd) add print_verbose for /key/generate

litellm/proxy/proxy_server.py

@@ -896,6 +896,7 @@ async def generate_key_helper_fn(
             "max_budget": max_budget,
             "user_email": user_email,
         }
+        print_verbose("PrismaClient: Before Insert Data")
         new_verification_token = await prisma_client.insert_data(
             data=verification_token_data
         )
@@ -1769,6 +1770,7 @@ async def generate_key_fn(
     - expires: (datetime) Datetime object for when key expires.
     - user_id: (str) Unique user id - used for tracking spend across multiple keys for same user id.
     """
+    print_verbose("entered /key/generate")
     data_json = data.json()  # type: ignore
     response = await generate_key_helper_fn(**data_json)
     return GenerateKeyResponse(

litellm/proxy/utils.py

@@ -417,6 +417,9 @@ class PrismaClient:
                 db_data["token"] = hashed_token
                 max_budget = db_data.pop("max_budget", None)
                 user_email = db_data.pop("user_email", None)
+                print_verbose(
+                    "PrismaClient: Before upsert into litellm_verificationtoken"
+                )
                 new_verification_token = await self.db.litellm_verificationtoken.upsert(  # type: ignore
                     where={
                         "token": hashed_token,

litellm/tests/test_proxy_server_keys.py

@@ -64,7 +64,7 @@ async def wrapper_startup_event():
 import asyncio
 
 
-@pytest.yield_fixture
+@pytest.fixture
 def event_loop():
     """Create an instance of the default event loop for each test case."""
     policy = asyncio.WindowsSelectorEventLoopPolicy()
@@ -99,7 +99,7 @@ def test_add_new_key(client):
             "aliases": {"mistral-7b": "gpt-3.5-turbo"},
             "duration": "20m",
         }
-        print("testing proxy server")
+        print("testing proxy server - test_add_new_key")
         # Your bearer token
         token = os.getenv("PROXY_MASTER_KEY")