fix: do not use logUpdate if noConsoleClear

offline-ai · Jun 12, 2024 · bb5c825 · bb5c825
1 parent d0d51f8
commit bb5c825
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/lib/run-script.ts b/src/lib/run-script.ts
@@ -124,21 +124,27 @@ export async function runScript(filename: string, options: IRunScriptOptions) {
   if (stream) {
     runtime.on('llm-stream', async function(llmResult, content: string, count: number) {
       const runtime = this.target as AIScriptEx
-      const s = llmResult.content
-      llmLastContent += s
+      let s = llmResult.content
 
       if (quit) {
         runtime.abort('quit')
         process.emit('SIGINT')
       }
       if (count !== retryCount) {
         retryCount = count
-        llmLastContent += colors.blue(`<续:${count}>`)
+        s += colors.blue(`<续:${count}>`)
       }
+      llmLastContent += s
       // if (llmLastContent.length > 100) {
       //   llmLastContent = llmLastContent.slice(llmLastContent.length-100)
       // }
-      if (!isSilence && llmLastContent) {logUpdate(llmLastContent)}
+      if (!isSilence && llmLastContent) {
+        if (!options.noConsoleClear) {
+          logUpdate(llmLastContent)
+        } else {
+          process.stdout.write(s)
+        }
+      }
     })
   }