phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add message_logging on Custom Logger

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-09 15:59:42 -07:00
parent 176397cfca
commit af5b87a8de
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/integrations/custom_logger.py
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 CustomLogger: # https://docs.litellm.ai/docs/observability/custom_callback#callback-class
# Class variables or attributes # Class variables or attributes
def __init__(self) -> None: def __init__(self, message_logging: bool = True) -> None:
self.message_logging = message_logging
pass pass
def log_pre_api_call(self, model, messages, kwargs): def log_pre_api_call(self, model, messages, kwargs):