fix(databricks.py): handle DONE chunk from databricks

2024-07-27 16:51:45 -07:00 · 2024-07-27 16:51:45 -07:00 · 6f9c29d39b
commit 6f9c29d39b
parent f76cad210c
1 changed files with 2 additions and 0 deletions
--- 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)