fix sagemaker test

litellm/llms/sagemaker.py

@@ -387,7 +387,7 @@ class SagemakerLLM(BaseAWSLLM):
             original_response=completion_response,
             additional_args={"complete_input_dict": _data},
         )
-        print_verbose(f"raw model_response: {response}")
+        print_verbose(f"raw model_response: {completion_response}")
         ## RESPONSE OBJECT
         try:
             if isinstance(completion_response, list):

litellm/tests/test_sagemaker.py

@@ -84,7 +84,7 @@ async def test_completion_sagemaker(sync_mode):
 
 
 @pytest.mark.asyncio()
-@pytest.mark.parametrize("sync_mode", [True])
+@pytest.mark.parametrize("sync_mode", [False, True])
 async def test_completion_sagemaker_stream(sync_mode):
     try:
         litellm.set_verbose = False