🤖 fix: include stdout/stderr/MUX_OUTPUT/MUX_PROMPT in LLM context

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

Author: ThomasK33

src/browser/components/Messages/ScriptExecutionMessage.tsx

@@ -71,7 +71,7 @@ export const ScriptExecutionMessage: React.FC<ScriptExecutionMessageProps> = ({
               {formatDuration(result.wall_duration_ms)}
             </div>
             <div className="text-foreground-secondary text-[11px]">
-              Visible only to you; never sent to the model.
+              Visible to you and the model.
             </div>
           </DetailSection>
 

src/browser/utils/messages/modelMessageTransform.ts

@@ -209,42 +209,38 @@ export function injectModeTransition(
  * - Preserves the rest of the message structure (id, role, other metadata)
  */
 export function transformScriptMessagesForLLM(messages: MuxMessage[]): MuxMessage[] {
-  return messages.flatMap((msg) => {
+  return messages.map((msg) => {
     if (msg.metadata?.muxMetadata?.type !== "script-execution") {
-      return [msg];
+      return msg;
     }
 
     const scriptMeta = msg.metadata.muxMetadata;
     const result = scriptMeta.result;
-    // Only include the message if the script explicitly wrote to MUX_PROMPT (promptFile)
-    // or MUX_OUTPUT (outputFile).
-    // This respects the "Visible only to you" promise for stdout/stderr (result.output).
-    const promptContent = result.promptFile?.trim();
-    const outputContent = result.outputFile?.trim();
-    const hasContent = (promptContent && promptContent.length > 0) || (outputContent && outputContent.length > 0);
+    
+    let llmContent = `Script '${scriptMeta.scriptName}' executed (exit code ${result.exitCode}).`;
 
-    if (hasContent) {
-      let llmContent = `Script '${scriptMeta.scriptName}' executed (exit code ${result.exitCode}).`;
-      
-      if (outputContent && outputContent.length > 0) {
-        llmContent += `\nMUX_OUTPUT:\n${outputContent}`;
-      }
-      
-      if (promptContent && promptContent.length > 0) {
-        llmContent += `\nMUX_PROMPT:\n${promptContent}`;
-      }
+    // Include Stdout/Stderr if present
+    if (result.output) {
+      llmContent += `\nStdout/Stderr:\n${result.output}`;
+    } else {
+      llmContent += `\nStdout/Stderr: (no output)`;
+    }
+
+    // Include MUX_OUTPUT if present
+    if (result.outputFile && result.outputFile.trim()) {
+      llmContent += `\nMUX_OUTPUT:\n${result.outputFile.trim()}`;
+    }
 
-      return [
-        {
-          ...msg,
-          parts: [{ type: "text", text: llmContent }],
-          // Keep metadata for debugging but ensure downstream consumers use the new parts
-        },
-      ];
+    // Include MUX_PROMPT if present
+    if (result.promptFile && result.promptFile.trim()) {
+      llmContent += `\nMUX_PROMPT:\n${result.promptFile.trim()}`;
     }
 
-    // Otherwise, exclude from LLM context entirely
-    return [];
+    return {
+      ...msg,
+      parts: [{ type: "text", text: llmContent }],
+      // Keep metadata for debugging but ensure downstream consumers use the new parts
+    };
   });
 }