fix: make response to stream

unit-mesh · Dec 1, 2023 · fe3026f · fe3026f
1 parent bac87c4
commit fe3026f
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/editor/src/components/editor/action/command-functions.ts b/editor/src/components/editor/action/command-functions.ts
@@ -71,13 +71,18 @@ export const CommandFunctions = Extension.create({
 
 						switch (action.outputForm) {
 							case OutputForm.STREAMING:
-								const content = await fetch("/api/completion/yiyan", {
+								const response = await fetch("/api/completion/yiyan", {
 									method: "POST",
 									body: JSON.stringify({ prompt: prompt }),
-								}).then(it => it.text());
+								})
+
+								await response.body?.pipeThrough(new TextDecoderStream()).pipeTo(new WritableStream({
+									write: (chunk) => {
+										const pos = actionExecutor.position(editor.state.selection);
+										editor.chain().focus().insertContentAt(pos, chunk).run();
+									}
+								}));
 
-								const pos = actionExecutor.position(editor.state.selection);
-								editor.chain().focus().insertContentAt(pos, content).run();
 								break;
 							case OutputForm.NORMAL:
 								const msg = await fetch("/api/completion/yiyan", {