diff --git a/tests/litellm/llms/sagemaker/test_sagemaker_common_utils.py b/tests/litellm/llms/sagemaker/test_sagemaker_common_utils.py index e19c41f1ac3..3b53107148b 100644 --- a/tests/litellm/llms/sagemaker/test_sagemaker_common_utils.py +++ b/tests/litellm/llms/sagemaker/test_sagemaker_common_utils.py @@ -17,6 +17,8 @@ async def test_aiter_bytes_unicode_decode_error(): Ensures stream processing continues despite the error. + + Relevant issue: https://github.com/BerriAI/litellm/issues/9165 """ # Create an instance of AWSEventStreamDecoder decoder = AWSEventStreamDecoder(model="test-model") @@ -56,6 +58,8 @@ async def test_aiter_bytes_valid_chunk_followed_by_unicode_error(): """ Test that valid chunks are processed correctly even when followed by Unicode decode errors. This ensures errors don't corrupt or prevent processing of valid data that came before. + + Relevant issue: https://github.com/BerriAI/litellm/issues/9165 """ decoder = AWSEventStreamDecoder(model="test-model")