From 856ab1ad256a8f2ec9e63f5a0afe0bf71f728ab9 Mon Sep 17 00:00:00 2001 From: harish-berri Date: Wed, 29 Apr 2026 18:11:09 +0000 Subject: [PATCH] Refactor import statements in key_management_endpoints.py - Updated the import statements in `generate_key_helper_fn` for better readability by formatting them into a multi-line structure. This change enhances code clarity without altering functionality. --- .../proxy/management_endpoints/key_management_endpoints.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/management_endpoints/key_management_endpoints.py b/litellm/proxy/management_endpoints/key_management_endpoints.py index 588ed81fe3b..c8814a1ce7c 100644 --- a/litellm/proxy/management_endpoints/key_management_endpoints.py +++ b/litellm/proxy/management_endpoints/key_management_endpoints.py @@ -3032,7 +3032,11 @@ async def generate_key_helper_fn( # noqa: PLR0915 access_group_ids: Optional[list] = None, budget_limits: Optional[list] = None, # multiple concurrent budget windows ): - from litellm.proxy.proxy_server import premium_user, prisma_client, user_api_key_cache + from litellm.proxy.proxy_server import ( + premium_user, + prisma_client, + user_api_key_cache, + ) if prisma_client is None: raise Exception(