diff --git a/docs/docs/integrations/llms/sagemaker.ipynb b/docs/docs/integrations/llms/sagemaker.ipynb
index f10e14d72ef..101f9d81ab0 100644
--- a/docs/docs/integrations/llms/sagemaker.ipynb
+++ b/docs/docs/integrations/llms/sagemaker.ipynb
@@ -141,7 +141,7 @@
     "    accepts = \"application/json\"\n",
     "\n",
     "    def transform_input(self, prompt: str, model_kwargs: Dict) -> bytes:\n",
-    "        input_str = json.dumps({prompt: prompt, **model_kwargs})\n",
+    "        input_str = json.dumps({\"inputs\": prompt, \"parameters\": model_kwargs})\n",
     "        return input_str.encode(\"utf-8\")\n",
     "\n",
     "    def transform_output(self, output: bytes) -> str:\n",
@@ -197,7 +197,7 @@
     "    accepts = \"application/json\"\n",
     "\n",
     "    def transform_input(self, prompt: str, model_kwargs: Dict) -> bytes:\n",
-    "        input_str = json.dumps({prompt: prompt, **model_kwargs})\n",
+    "        input_str = json.dumps({\"inputs\": prompt, \"parameters\": model_kwargs})\n",
     "        return input_str.encode(\"utf-8\")\n",
     "\n",
     "    def transform_output(self, output: bytes) -> str:\n",