From 749c7ea73dd7236a693ceb4704517abb5fe83d14 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 24 Oct 2023 17:40:58 -0700
Subject: [PATCH] (test) test custom input_callback

---
 litellm/tests/test_custom_logger.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/litellm/tests/test_custom_logger.py b/litellm/tests/test_custom_logger.py
index 21d99a1ee..2ba378aad 100644
--- a/litellm/tests/test_custom_logger.py
+++ b/litellm/tests/test_custom_logger.py
@@ -56,13 +56,21 @@ def send_slack_alert(
     else:
         print(f"Failed to send message to Slack. Status code: {response.status_code}")
         print(response.json())
-    
+
+def get_transformed_inputs(
+    kwargs,
+):
+    params_to_model = kwargs["additional_args"]["complete_input_dict"]
+    print("params to model", params_to_model)
+
 litellm.success_callback = [custom_callback, send_slack_alert]
 litellm.failure_callback = [send_slack_alert]
 
 
 litellm.set_verbose = True
 
+litellm.input_callback = [get_transformed_inputs]
+
 
 def test_chat_openai():
     try: