fix(pi): stream only final plan content

Author: Bill Leoutsakos

apps/sim/executor/handlers/pi/pi-handler.test.ts

@@ -931,4 +931,41 @@ describe('PiBlockHandler', () => {
     expect(text).toContain('streamed')
     expect(result.execution.output.content).toBe('streamed')
   })
+
+  it('streams only the canonical final document for Plan mode', async () => {
+    mockRunCloudPlan.mockImplementation(async (_params, runCtx) => {
+      runCtx.onEvent({ type: 'text', text: 'Inspecting files...' })
+      runCtx.onEvent({ type: 'final', text: '# Final Plan\n\n1. Make the change.' })
+      return {
+        totals: {
+          finalText: '# Final Plan\n\n1. Make the change.',
+          inputTokens: 0,
+          outputTokens: 0,
+          toolCalls: [],
+        },
+      }
+    })
+
+    const result = (await handler.execute(ctx({ stream: true, selectedOutputs: ['blk'] }), block, {
+      mode: 'cloud_plan',
+      task: 'plan it',
+      model: 'claude',
+      owner: 'o',
+      repo: 'r',
+      githubToken: 'ghp',
+    })) as StreamingExecution
+
+    const reader = result.stream.getReader()
+    const decoder = new TextDecoder()
+    let text = ''
+    for (;;) {
+      const { done, value } = await reader.read()
+      if (done) break
+      text += decoder.decode(value)
+    }
+
+    expect(text).toBe('# Final Plan\n\n1. Make the change.')
+    expect(text).not.toContain('Inspecting files...')
+    expect(result.execution.output.content).toBe('# Final Plan\n\n1. Make the change.')
+  })
 })

apps/sim/executor/handlers/pi/pi-handler.ts

@@ -510,6 +510,7 @@ export class PiBlockHandler implements BlockHandler {
           try {
             const result = await backend(params, {
               onEvent: (event) => {
+                if (params.mode === 'cloud_plan') return
                 const text = streamTextForEvent(event)
                 if (text) controller.enqueue(encoder.encode(text))
               },
@@ -519,6 +520,9 @@ export class PiBlockHandler implements BlockHandler {
               controller.error(new Error(result.totals.errorMessage))
               return
             }
+            if (params.mode === 'cloud_plan' && result.totals.finalText) {
+              controller.enqueue(encoder.encode(result.totals.finalText))
+            }
             Object.assign(
               output,
               this.buildOutput(