diff --git a/litellm/llms/databricks.py b/litellm/llms/databricks.py index 1a276f52c9..363b222fea 100644 --- a/litellm/llms/databricks.py +++ b/litellm/llms/databricks.py @@ -795,6 +795,8 @@ class ModelResponseIterator: try: chunk = chunk.replace("data:", "") chunk = chunk.strip() + if chunk == "[DONE]": + raise StopAsyncIteration if len(chunk) > 0: json_chunk = json.loads(chunk) return self.chunk_parser(chunk=json_chunk)