Use proper attribute for sagemaker request (#11362)

2025-06-05 07:47:43 +02:00 · 2025-06-05 07:47:43 +02:00 · d7982bb0af
commit d7982bb0af
parent 4019f79808
1 changed files with 1 additions and 1 deletions
--- a/litellm/llms/sagemaker/completion/handler.py
+++ b/litellm/llms/sagemaker/completion/handler.py
@ -626,7 +626,7 @@ class SagemakerLLM(BaseAWSLLM):
                inference_params[k] = v

        #### HF EMBEDDING LOGIC
-        data = json.dumps({"text_inputs": input}).encode("utf-8")
+        data = json.dumps({"inputs": input}).encode("utf-8")

        ## LOGGING
        request_str = f"""