feat: ✨ add logWarn to console methods in runpromptcontext

microsoft · Oct 23, 2024 · 28e9232 · 28e9232
1 parent 488ea1a
commit 28e9232
Showing 1 changed file with 29 additions and 12 deletions.
diff --git a/packages/core/src/runpromptcontext.ts b/packages/core/src/runpromptcontext.ts
@@ -25,7 +25,7 @@ import {
 } from "./parameters"
 import { consoleLogFormat } from "./logging"
 import { isGlobMatch } from "./glob"
-import { arrayify, logError, logVerbose } from "./util"
+import { arrayify, logError, logVerbose, logWarn } from "./util"
 import { renderShellOutput } from "./chatrender"
 import { jinjaRender } from "./jinja"
 import { mustacheRender } from "./mustache"
@@ -73,18 +73,35 @@ export function createChatTurnGenerationContext(
 ): ChatTurnGenerationContext & { node: PromptNode } {
     const node: PromptNode = { children: [] }
 
-    const log = (...args: any[]) => {
-        const line = consoleLogFormat(...args)
-        if (line) {
-            trace.log(line)
-            logVerbose(line)
-        }
-    }
     const console = Object.freeze<PromptGenerationConsole>({
-        log,
-        debug: log,
-        warn: log,
-        error: log,
+        log: (...args: any[]) => {
+            const line = consoleLogFormat(...args)
+            if (line) {
+                trace.log(line)
+                process.stdout.write(line + "\n")
+            }
+        },
+        debug: (...args: any[]) => {
+            const line = consoleLogFormat(...args)
+            if (line) {
+                trace.log(line)
+                logVerbose(line)
+            }
+        },
+        warn: (...args: any[]) => {
+            const line = consoleLogFormat(...args)
+            if (line) {
+                trace.log(line)
+                logWarn(line)
+            }
+        },
+        error: (...args: any[]) => {
+            const line = consoleLogFormat(...args)
+            if (line) {
+                trace.log(line)
+                logError(line)
+            }
+        },
     })
 
     const ctx: ChatTurnGenerationContext & { node: PromptNode } = {