From 995b1e73157aec39c12fde0ec84ed0997ae63f51 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 11 Jun 2024 18:33:15 -0700 Subject: [PATCH] fix when user_api_key_dict is None --- litellm/proxy/common_utils/management_endpoint_utils.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/common_utils/management_endpoint_utils.py b/litellm/proxy/common_utils/management_endpoint_utils.py index 3b5340df340..006f6aaae39 100644 --- a/litellm/proxy/common_utils/management_endpoint_utils.py +++ b/litellm/proxy/common_utils/management_endpoint_utils.py @@ -23,7 +23,9 @@ def management_endpoint_wrapper(func): if kwargs is None: kwargs = {} - user_api_key_dict: UserAPIKeyAuth = kwargs.get("user_api_key_dict") + user_api_key_dict: UserAPIKeyAuth = ( + kwargs.get("user_api_key_dict") or UserAPIKeyAuth() + ) parent_otel_span = user_api_key_dict.parent_otel_span if parent_otel_span is not None: from litellm.proxy.proxy_server import open_telemetry_logger @@ -56,7 +58,9 @@ def management_endpoint_wrapper(func): if kwargs is None: kwargs = {} - user_api_key_dict: UserAPIKeyAuth = kwargs.get("user_api_key_dict") + user_api_key_dict: UserAPIKeyAuth = ( + kwargs.get("user_api_key_dict") or UserAPIKeyAuth() + ) parent_otel_span = user_api_key_dict.parent_otel_span if parent_otel_span is not None: from litellm.proxy.proxy_server import open_telemetry_logger