feat: ✨ enhance output handling and add model alias reset

microsoft · Jan 11, 2025 · 9771d16 · 9771d16
1 parent 735b68c
commit 9771d16
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/packages/core/src/promptrunner.ts b/packages/core/src/promptrunner.ts
@@ -271,8 +271,11 @@ export async function runTemplate(
             changelogs,
             edits,
         } = chatResult
-        let { annotations } = chatResult
-        if (chatResult.text) output.appendContent(chatResult.text)
+        let { text, annotations } = chatResult
+        const userContent = output.content
+        if (userContent) {
+            text = userContent + "\n\n" + text
+        }
 
         // Reporting and tracing output
         if (fences?.length)
@@ -324,7 +327,7 @@ export async function runTemplate(
             annotations,
             changelogs,
             fileEdits,
-            text: output.content,
+            text,
             version,
             fences,
             frames,

diff --git a/packages/core/src/testhost.ts b/packages/core/src/testhost.ts
@@ -79,6 +79,9 @@ export class TestHost implements RuntimeHost {
         return { ok: true }
     }
 
+    clearModelAlias(source: "cli" | "env" | "config" | "script"): void {
+        delete this.modelAliases[source]
+    }
     setModelAlias(
         source: "cli" | "env" | "config",
         id: string,