Add default fallback chunk

premAI-io · Mar 5, 2024 · 6c16bc4 · 6c16bc4
1 parent 42d2fba
commit 6c16bc4
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/prem_utils/connectors/anthropic.py b/prem_utils/connectors/anthropic.py
@@ -56,6 +56,19 @@ def parse_chunk(self, chunk):
                     }
                 ],
             }
+        else:
+            return {
+                "id": 0,
+                "model": None,
+                "object": None,
+                "created": None,
+                "choices": [
+                    {
+                        "delta": {"content": "", "role": "assistant"},
+                        "finish_reason": None,
+                    }
+                ],
+            }
 
     def preprocess_messages(self, messages):
         system_prompt = ""