phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 10:14:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(proxy_server.py): fix linting errors

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-23 21:42:31 -08:00
parent 19fb24cd15
commit c9e37b8fb8
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/proxy/proxy_server.py
View file

@ -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()