fix when user_api_key_dict is None

paneru-rajan · Jun 12, 2024 · 995b1e7 · 995b1e7
1 parent b10ac0d
commit 995b1e7
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/litellm/proxy/common_utils/management_endpoint_utils.py b/litellm/proxy/common_utils/management_endpoint_utils.py
@@ -23,7 +23,9 @@ async def wrapper(*args, **kwargs):
 
             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 @@ async def wrapper(*args, **kwargs):
 
             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