From ed8c666922796792af72ff7ce1fe1fc3e1d30f06 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 4 Dec 2023 08:44:20 -0800
Subject: [PATCH] (feat) proxy: custom_logger for I/O logging

---
 litellm/proxy/proxy_server.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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