From 857adc8bed6e3605fa6cfcba698092fdf23b39f7 Mon Sep 17 00:00:00 2001 From: harish-berri Date: Wed, 29 Apr 2026 19:29:48 +0000 Subject: [PATCH] fix linting --- .../proxy/management_endpoints/key_management_endpoints.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/litellm/proxy/management_endpoints/key_management_endpoints.py b/litellm/proxy/management_endpoints/key_management_endpoints.py index c8814a1ce7c..ff48cfd7278 100644 --- a/litellm/proxy/management_endpoints/key_management_endpoints.py +++ b/litellm/proxy/management_endpoints/key_management_endpoints.py @@ -3248,10 +3248,11 @@ async def generate_key_helper_fn( # noqa: PLR0915 data=key_data, table_name="key" ) - key_data["token_id"] = getattr(create_key_response, "token", None) - if key_data["token_id"] is not None: + created_token_id = getattr(create_key_response, "token", None) + key_data["token_id"] = created_token_id + if isinstance(created_token_id, str): await InvalidVirtualKeyCache.delete_invalid_token_cache( - hashed_token=key_data["token_id"], + hashed_token=created_token_id, user_api_key_cache=user_api_key_cache, ) key_data["litellm_budget_table"] = getattr(