phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix test_master_key_hashing

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-21 17:56:09 -07:00
parent 0ea1f367d7
commit f0f1d50734
2 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/management_helpers/utils.py
View file

@ -266,7 +266,7 @@ def management_endpoint_wrapper(func):
)
_http_request: Request = kwargs.get("http_request")
parent_otel_span = user_api_key_dict.parent_otel_span
parent_otel_span = getattr(user_api_key_dict, "parent_otel_span", None)
if parent_otel_span is not None:
from litellm.proxy.proxy_server import open_telemetry_logger
@ -310,7 +310,7 @@ def management_endpoint_wrapper(func):
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 = getattr(user_api_key_dict, "parent_otel_span", None)
if parent_otel_span is not None:
from litellm.proxy.proxy_server import open_telemetry_logger