From 715387c3c0f7dcd2181b74a018cc4ab697aa4a50 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 9 Sep 2024 15:59:42 -0700 Subject: [PATCH] add message_logging on Custom Logger --- litellm/integrations/custom_logger.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/integrations/custom_logger.py b/litellm/integrations/custom_logger.py index ce0caf32bc..e5d3dfd56c 100644 --- a/litellm/integrations/custom_logger.py +++ b/litellm/integrations/custom_logger.py @@ -15,7 +15,8 @@ from litellm.types.utils import AdapterCompletionStreamWrapper, ModelResponse class CustomLogger: # https://docs.litellm.ai/docs/observability/custom_callback#callback-class # Class variables or attributes - def __init__(self) -> None: + def __init__(self, message_logging: bool = True) -> None: + self.message_logging = message_logging pass def log_pre_api_call(self, model, messages, kwargs):