Upload 4 files

app.py

@@ -254,7 +254,18 @@ def chat_completions():
                     }
                     yield f"data: {json.dumps(data)}\n\n"
 
-            yield "data: [DONE]\n\n"
+            data = {
+                        'choices': [
+                            {
+                                'delta': {},
+                                'finish_reason': 'stop',
+                                'index': 0
+                            }
+                        ],
+                        'object': 'chat.completion.chunk'
+                    }
+
+            yield f"data: {json.dumps(data)}\n\n"
             logger.info(f"200!")
 
         except Exception as e:
@@ -266,7 +277,18 @@ def chat_completions():
                 }
             }
             yield f"data: {json.dumps(error_data)}\n\n"
-            yield "data: [DONE]\n\n"
+            data = {
+                        'choices': [
+                            {
+                                'delta': {},
+                                'finish_reason': 'stop',
+                                'index': 0
+                            }
+                        ],
+                        'object': 'chat.completion.chunk'
+                    }
+
+            yield f"data: {json.dumps(data)}\n\n"
 
     attempt = 0
     success = False