mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-11 22:51:28 +00:00
test_dd_llms_obs_redaction
This commit is contained in:
parent
eef864360e
commit
e4c31add58
1 changed files with 3 additions and 3 deletions
|
|
@ -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"] == [
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue