修复流传输完成的日志记录

LLM-Red-Team · Dec 10, 2024 · 1648f3b · 1648f3b
1 parent 78e0ef8
commit 1648f3b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/api/controllers/chat.ts b/src/api/controllers/chat.ts
@@ -767,7 +767,10 @@ function createTransStream(model: string, stream: any, refConvId: string, endCal
   );
   stream.once(
     "close",
-    () => !transStream.closed && transStream.end("data: [DONE]\n\n")
+    () => {
+      !transStream.closed && transStream.end("data: [DONE]\n\n");
+      endCallback && endCallback();
+    }
   );
   return transStream;
 }