phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix when user_api_key_dict is None

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-11 18:33:15 -07:00
parent b10ac0d1f3
commit 995b1e7315
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/common_utils/management_endpoint_utils.py
View file

@ -23,7 +23,9 @@ def management_endpoint_wrapper(func):
if kwargs is None: if kwargs is None:
kwargs = {} 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 parent_otel_span = user_api_key_dict.parent_otel_span
if parent_otel_span is not None: if parent_otel_span is not None:
from litellm.proxy.proxy_server import open_telemetry_logger from litellm.proxy.proxy_server import open_telemetry_logger
@ -56,7 +58,9 @@ def management_endpoint_wrapper(func):
if kwargs is None: if kwargs is None:
kwargs = {} 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 parent_otel_span = user_api_key_dict.parent_otel_span
if parent_otel_span is not None: if parent_otel_span is not None:
from litellm.proxy.proxy_server import open_telemetry_logger from litellm.proxy.proxy_server import open_telemetry_logger