From 6f9c29d39b537fcd4fe42f50f7c5184d92302b2a Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Sat, 27 Jul 2024 16:51:45 -0700
Subject: [PATCH] fix(databricks.py): handle DONE chunk from databricks

---
 litellm/llms/databricks.py | 2 ++
 1 file changed, 2 insertions(+)

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)