From e407b185ee9bd9e21033ea4429cdfd989748a103 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 27 Nov 2023 09:08:20 -0800
Subject: [PATCH] (feat) completion:sagemaker - better debugging

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

diff --git a/litellm/llms/sagemaker.py b/litellm/llms/sagemaker.py
index 04a08a4512..ce36367de0 100644
--- a/litellm/llms/sagemaker.py
+++ b/litellm/llms/sagemaker.py
@@ -127,10 +127,18 @@ def completion(
     }
 
     ## LOGGING
+    request_str = f"""
+    response = client.invoke_endpoint(
+        EndpointName={model},
+        ContentType="application/json",
+        Body={data},
+        CustomAttributes="accept_eula=true",
+    )
+    """
     logging_obj.pre_call(
             input=prompt,
             api_key="",
-            additional_args={"complete_input_dict": data},
+            additional_args={"complete_input_dict": data, "request_str": request_str},
         )
     ## COMPLETION CALL
     response = client.invoke_endpoint(