diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py index c9ba4c215..a2605dd55 100644 --- a/litellm/proxy/proxy_server.py +++ b/litellm/proxy/proxy_server.py @@ -117,7 +117,9 @@ app.add_middleware( allow_methods=["*"], allow_headers=["*"], ) -def log_input_output(request, response): +def log_input_output(request, response, custom_logger=None): + if custom_logger is not None: + custom_logger(request, response) global otel_logging if otel_logging != True: return