Fix stream

Author: Sg312

apps/sim/app/api/mothership/execute/route.ts

@@ -200,6 +200,7 @@ export const POST = withRouteHandler(async (req: NextRequest) => {
 
       const stream = new ReadableStream<Uint8Array>({
         start(controller) {
+          let forwardedAssistantContent = ''
           const send = (event: unknown) => {
             if (!cancelled) {
               controller.enqueue(encodeNdjson(event))
@@ -221,7 +222,14 @@ export const POST = withRouteHandler(async (req: NextRequest) => {
                   event.payload.channel === MothershipStreamV1TextChannel.assistant &&
                   event.payload.text
                 ) {
-                  send({ type: 'chunk', content: event.payload.text })
+                  const text = event.payload.text
+                  const content = text.startsWith(forwardedAssistantContent)
+                    ? text.slice(forwardedAssistantContent.length)
+                    : text
+                  if (content) {
+                    forwardedAssistantContent += content
+                    send({ type: 'chunk', content })
+                  }
                 }
               })
               allowExplicitAbort = false

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

@@ -164,6 +164,7 @@ async function readMothershipExecuteResponse(response: Response): Promise<Mother
 function createMothershipStreamingExecution(
   response: Response,
   fallbackChatId: string,
+  blockId: string,
   options: {
     onCancel?: (reason?: unknown) => void
     onDone?: () => void
@@ -265,6 +266,7 @@ function createMothershipStreamingExecution(
     execution: {
       success: true,
       output,
+      blockId,
       logs: [],
       metadata: {
         duration: 0,
@@ -438,7 +440,7 @@ export class MothershipBlockHandler implements BlockHandler {
 
       if (isContentSelectedForStreaming(ctx, block)) {
         cleanupImmediately = false
-        return createMothershipStreamingExecution(response, chatId, {
+        return createMothershipStreamingExecution(response, chatId, block.id, {
           onCancel: (reason) => {
             if (!abortController.signal.aborted) {
               abortController.abort(reason ?? 'mothership_stream_cancelled')