[Fix] 🐛 BingAPI ChatStream DONE flag

Harry-zklcdc · Jan 23, 2024 · 8949567 · 8949567
1 parent e7f243d
commit 8949567
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/api/v1/chat.go b/api/v1/chat.go
@@ -118,6 +118,7 @@ func ChatHandler(w http.ResponseWriter, r *http.Request) {
 				}
 				w.Write([]byte("data: "))
 				w.Write(resData)
+				w.Write([]byte("\n"))
 				break
 			}
 			resData, err := json.Marshal(resp)
@@ -140,6 +141,8 @@ func ChatHandler(w http.ResponseWriter, r *http.Request) {
 				}(globalChat.GetCookies())
 			}
 		}
+		w.Write([]byte("data: [DONE]\n"))
+		flusher.Flush()
 	} else {
 		text, err := chat.Chat(prompt, msg)
 		if err != nil {