fix(test_custom_callbacks_input.py): unit tests for 'turn_off_message_logging'

ensure no raw request is logged either
This commit is contained in:
Krrish Dholakia 2024-06-07 15:39:15 -07:00
parent 51fb199329
commit f73b6033fd
4 changed files with 72 additions and 9 deletions

View file

@ -1,5 +1,10 @@
from typing import List, Optional, Union, Dict, Tuple, Literal
from typing_extensions import TypedDict
from enum import Enum
class LiteLLMCommonStrings(Enum):
redacted_by_litellm = "redacted by litellm. 'litellm.turn_off_message_logging=True'"
class CostPerToken(TypedDict):