add message_logging on Custom Logger

This commit is contained in:
Ishaan Jaff
2024-09-09 15:59:42 -07:00
parent 15c761a56b
commit 715387c3c0
+2 -1
View File
@@ -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):