🤖 fix: restore MUX_OUTPUT/MUX_PROMPT in script execution history

Change-Id: I333264f12f60533456812a8cd48fd11dc5e87fec
Signed-off-by: Thomas Kosiewski <[email protected]>

Author: ThomasK33

src/browser/utils/messages/modelMessageTransform.ts

@@ -226,9 +226,15 @@ export function transformScriptMessagesForLLM(messages: MuxMessage[]): MuxMessag
       llmContent += `\nStdout/Stderr: (no output)`;
     }
 
-    // EXCLUDE MUX_OUTPUT and MUX_PROMPT from the LLM context for the script message itself.
-    // MUX_PROMPT is sent as a separate user message by ChatInput, so including it here would be duplication.
-    // MUX_OUTPUT is intended for user toasts, not LLM context.
+    // Include MUX_OUTPUT if present
+    if (result.outputFile?.trim()) {
+      llmContent += `\nMUX_OUTPUT:\n${result.outputFile.trim()}`;
+    }
+
+    // Include MUX_PROMPT if present
+    if (result.promptFile?.trim()) {
+      llmContent += `\nMUX_PROMPT:\n${result.promptFile.trim()}`;
+    }
 
     return {
       ...msg,

src/browser/utils/messages/transformScriptMessagesForLLM.test.ts

@@ -43,7 +43,7 @@ describe("transformScriptMessagesForLLM", () => {
     }
   });
 
-  it("should exclude MUX_OUTPUT and MUX_PROMPT from script execution messages (avoid duplication)", () => {
+  it("should include all fields (stdout, output, prompt) if present", () => {
     const scriptResult: BashToolResult = {
       success: true,
       output: "stdout stuff",
@@ -79,10 +79,8 @@ describe("transformScriptMessagesForLLM", () => {
     expect(textPart.type).toBe("text");
     if (textPart.type === "text") {
       expect(textPart.text).toContain("Stdout/Stderr:\nstdout stuff");
-      expect(textPart.text).not.toContain("MUX_OUTPUT");
-      expect(textPart.text).not.toContain("User toast");
-      expect(textPart.text).not.toContain("MUX_PROMPT");
-      expect(textPart.text).not.toContain("Model prompt");
+      expect(textPart.text).toContain("MUX_OUTPUT:\nUser toast");
+      expect(textPart.text).toContain("MUX_PROMPT:\nModel prompt");
     }
   });
 });