diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index c34a701c6..2f7184761 100644 --- a/litellm/proxy/proxy_server.py +++ b/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( diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index f0ee85462..bc61a6666 100644 --- a/litellm/proxy/utils.py +++ b/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, diff --git a/litellm/tests/test_proxy_server_keys.py b/litellm/tests/test_proxy_server_keys.py index b372681a5..3399007e8 100644 --- a/litellm/tests/test_proxy_server_keys.py +++ b/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")