From 0250492d9583d3991cd78e8286cc2bb19a893f6d Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 17 Jan 2024 14:54:29 -0800 Subject: [PATCH] (fix) /key/delete --- litellm/proxy/_types.py | 6 +----- litellm/proxy/proxy_server.py | 7 +++++-- 2 files changed, 6 insertions(+), 7 deletions(-) diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index 039db119cb0..42136593eb7 100644 --- a/litellm/proxy/_types.py +++ b/litellm/proxy/_types.py @@ -167,12 +167,8 @@ class GenerateKeyResponse(LiteLLMBase): user_id: str -class _DeleteKeyObject(LiteLLMBase): - key: str - - class DeleteKeyRequest(LiteLLMBase): - keys: List[_DeleteKeyObject] + keys: List[str] class NewUserRequest(GenerateKeyRequest): diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index 471e26d3dfc..153c1633d72 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -2045,8 +2045,11 @@ async def delete_key_fn(request: Request, data: DeleteKeyRequest): try: keys = data.keys - deleted_keys = await delete_verification_token(tokens=keys) - assert len(keys) == deleted_keys + result = await delete_verification_token(tokens=keys) + verbose_proxy_logger.debug("/key/delete - deleted_keys=", result) + + number_deleted_keys = len(result["deleted_keys"]) + assert len(keys) == number_deleted_keys return {"deleted_keys": keys} except Exception as e: raise HTTPException(