From c9e37b8fb881130ca546e95865d13705364a4aa1 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 23 Nov 2023 21:42:31 -0800 Subject: [PATCH] fix(proxy_server.py): fix linting errors --- litellm/proxy/proxy_server.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index c9193d7504..1d1931fa74 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -355,10 +355,13 @@ async def generate_key_helper_fn(duration_str: str, models: list, aliases: dict, async def delete_verification_token(tokens: List[str]): global prisma_client try: - # Assuming 'db' is your Prisma Client instance - deleted_tokens = await prisma_client.litellm_verificationtoken.delete_many( - where={"token": {"in": tokens}} - ) + if prisma_client: + # Assuming 'db' is your Prisma Client instance + deleted_tokens = await prisma_client.litellm_verificationtoken.delete_many( + where={"token": {"in": tokens}} + ) + else: + raise Exception except Exception as e: traceback.print_exc() raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR) @@ -666,7 +669,7 @@ async def generate_key_fn(request: Request): ) @router.post("/key/delete", dependencies=[Depends(user_api_key_auth)]) -async def generate_key_fn(request: Request): +async def delete_key_fn(request: Request): try: data = await request.json()