test(test_streaming.py): commenting out sagemaker test due to aws key issue

litellm/tests/test_streaming.py

@@ -695,32 +695,32 @@ def test_completion_replicate_stream_bad_key():
 # test_completion_bedrock_claude_stream() 
 
 
-def test_completion_sagemaker_stream():
+# def test_completion_sagemaker_stream():
-    try:
+#     try:
-        response = completion(
+#         response = completion(
-            model="sagemaker/jumpstart-dft-meta-textgeneration-llama-2-7b", 
+#             model="sagemaker/jumpstart-dft-meta-textgeneration-llama-2-7b", 
-            messages=messages,
+#             messages=messages,
-            temperature=0.2,
+#             temperature=0.2,
-            max_tokens=80,
+#             max_tokens=80,
-            stream=True,
+#             stream=True,
-        )
+#         )
-        complete_response = ""
+#         complete_response = ""
-        has_finish_reason = False
+#         has_finish_reason = False
-        # Add any assertions here to check the response
+#         # Add any assertions here to check the response
-        for idx, chunk in enumerate(response):
+#         for idx, chunk in enumerate(response):
-            chunk, finished = streaming_format_tests(idx, chunk)
+#             chunk, finished = streaming_format_tests(idx, chunk)
-            has_finish_reason = finished
+#             has_finish_reason = finished
-            if finished:
+#             if finished:
-                break
+#                 break
-            complete_response += chunk
+#             complete_response += chunk
-        if has_finish_reason is False:
+#         if has_finish_reason is False:
-            raise Exception("finish reason not set for last chunk")
+#             raise Exception("finish reason not set for last chunk")
-        if complete_response.strip() == "": 
+#         if complete_response.strip() == "": 
-            raise Exception("Empty response received")
+#             raise Exception("Empty response received")
-    except InvalidRequestError as e: 
+#     except InvalidRequestError as e: 
-        pass
+#         pass
-    except Exception as e:
+#     except Exception as e:
-        pytest.fail(f"Error occurred: {e}")
+#         pytest.fail(f"Error occurred: {e}")
 
 # test_completion_sagemaker_stream()