This commit is contained in:
Thomas Werkmeister 2026-08-27 17:17:57 -05:00 • committed by GitHub
commit 4f14b5b80b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -174,7 +174,11 @@ class AWSEventStreamDecoder:
except Exception as e:
verbose_logger.error("Error parsing message: %s. Attempting to combine with next event.", e)
continue
# remove custom sagemaker [DONE] message in stream
if accumulated_json.endswith("[DONE]"):
accumulated_json = accumulated_json[:-6].strip()
# Handle any remaining data after the iterator is exhausted
if accumulated_json:
try: