gracefully end stream

supermemoryai · Apr 1, 2024 · a00bb30 · a00bb30
1 parent dbf3011
commit a00bb30
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/apps/cf-ai-backend/src/index.ts b/apps/cf-ai-backend/src/index.ts
@@ -136,7 +136,7 @@ export default {
 							const encodedChunk = converter.encode("data: " + JSON.stringify({ "response": chunkText }) + "\n\n");
 							controller.enqueue(encodedChunk);
 						}
-						const doneChunk = converter.encode("[DONE]");
+						const doneChunk = converter.encode("data: [DONE]");
 						controller.enqueue(doneChunk);
 						controller.close();
 					}
@@ -167,7 +167,7 @@ export default {
 							const encodedChunk = converter.encode("data: " + JSON.stringify({ "response": chunkText }) + "\n\n");
 							controller.enqueue(encodedChunk);
 						}
-						const doneChunk = converter.encode("[DONE]");
+						const doneChunk = converter.encode("data: [DONE]");
 						controller.enqueue(doneChunk);
 						controller.close();
 					}