custom_logger for litellm - callback_func

This commit is contained in:
ishaan-jaff 2023-09-09 18:41:41 -07:00
parent 7dc694c704
commit eaaede608e
3 changed files with 44 additions and 2 deletions

View file

@ -1078,7 +1078,7 @@ def set_callbacks(callback_list, function_id=None):
else:
liteDebuggerClient = LiteDebugger(email=str(uuid.uuid4()))
elif callable(callback):
customLogger = CustomLogger()
customLogger = CustomLogger(callback_func=callback)
except Exception as e:
raise e