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

feat(redact_messages.py): allow remove sensitive key information before passing to logging integration

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-22 20:58:02 -07:00
parent 0b9e93d863
commit 548e4f53f8
7 changed files with 65 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/integrations/opentelemetry.py
View file

@ -7,6 +7,7 @@ from typing import TYPE_CHECKING, Any, Dict, Optional, Union
import litellm
from litellm._logging import verbose_logger
from litellm.integrations.custom_logger import CustomLogger
from litellm.litellm_core_utils.redact_messages import redact_user_api_key_info
from litellm.types.services import ServiceLoggerPayload
if TYPE_CHECKING:
@ -315,7 +316,9 @@ class OpenTelemetry(CustomLogger):
#############################################
metadata = litellm_params.get("metadata", {}) or {}
for key, value in metadata.items():
clean_metadata = redact_user_api_key_info(metadata=metadata)
for key, value in clean_metadata.items():
if self.is_primitive(value):
span.set_attribute("metadata.{}".format(key), value)