scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

Fix log spamming for custom sagemaker [DONE] chunk in streams

Browse source
This commit is contained in:
Thomas Werkmeister 2026-06-29 12:07:55 +02:00 • committed by GitHub
parent 8e30cfbeb1
commit 9cffbdb1ba
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/llms/sagemaker/common_utils.py
View file

@ -173,7 +173,11 @@ class AWSEventStreamDecoder:
except Exception as e:
verbose_logger.error(f"Error parsing message: {e}. Attempting to combine with next event.")
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: