From e4c31add58cba7191d3caaf980eaefc7322fe82a Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 8 Nov 2025 14:23:42 -0800 Subject: [PATCH] test_dd_llms_obs_redaction --- .../integrations/datadog/test_datadog_llm_observability.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_litellm/integrations/datadog/test_datadog_llm_observability.py b/tests/test_litellm/integrations/datadog/test_datadog_llm_observability.py index f6b8af503f..464cb0026e 100644 --- a/tests/test_litellm/integrations/datadog/test_datadog_llm_observability.py +++ b/tests/test_litellm/integrations/datadog/test_datadog_llm_observability.py @@ -3,7 +3,7 @@ import os import sys from datetime import datetime, timedelta, timezone from typing import Optional -from unittest.mock import Mock, patch, MagicMock +from unittest.mock import MagicMock, Mock, patch import pytest @@ -406,13 +406,13 @@ async def test_dd_llms_obs_redaction(mock_env_vars): assert ( dd_llms_obs_logger.logged_standard_logging_payload["messages"][0]["content"] - == LiteLLMCommonStrings.redacted_by_litellm.value + == "redacted-by-litellm" ) assert ( dd_llms_obs_logger.logged_standard_logging_payload["response"]["choices"][0][ "message" ]["content"] - == LiteLLMCommonStrings.redacted_by_litellm.value + == "redacted-by-litellm" ) assert test_s3_logger.logged_standard_logging_payload["messages"] == [