diff --git a/litellm/llms/sagemaker.py b/litellm/llms/sagemaker.py
index 2604875b47..94a940ddc2 100644
--- a/litellm/llms/sagemaker.py
+++ b/litellm/llms/sagemaker.py
@@ -365,7 +365,10 @@ def completion(
     ## RESPONSE OBJECT
     completion_response = json.loads(response)
     try:
-        completion_response_choices = completion_response[0]
+        if isinstance(completion_response, list):
+            completion_response_choices = completion_response[0]
+        else:
+            completion_response_choices = completion_response
         completion_output = ""
         if "generation" in completion_response_choices:
             completion_output += completion_response_choices["generation"]
@@ -580,7 +583,10 @@ async def async_completion(
         ## RESPONSE OBJECT
         completion_response = json.loads(response)
         try:
-            completion_response_choices = completion_response[0]
+            if isinstance(completion_response, list):
+                completion_response_choices = completion_response[0]
+            else:
+                completion_response_choices = completion_response
             completion_output = ""
             if "generation" in completion_response_choices:
                 completion_output += completion_response_choices["generation"]