improvement(mothership): streaming state transitions (#4439)

* improvement(mothership): improve streaming state transitions

* address comments

Author: icecrasher321

apps/sim/app/api/copilot/chat/stop/route.test.ts

@@ -29,6 +29,16 @@ const {
   mockSql: vi.fn((strings: TemplateStringsArray, ...values: unknown[]) => ({ strings, values })),
 }))
 
+vi.mock('@sim/db/schema', () => ({
+  copilotChats: {
+    id: 'copilotChats.id',
+    userId: 'copilotChats.userId',
+    workspaceId: 'copilotChats.workspaceId',
+    messages: 'copilotChats.messages',
+    conversationId: 'copilotChats.conversationId',
+  },
+}))
+
 vi.mock('@sim/db', () => ({
   db: {
     select: mockSelect,
@@ -140,6 +150,7 @@ describe('copilot chat stop route', () => {
       workspaceId: 'ws-1',
       chatId: 'chat-1',
       type: 'completed',
+      streamId: 'stream-1',
     })
   })
 })

apps/sim/app/api/copilot/chat/stop/route.ts

@@ -111,6 +111,7 @@ export const POST = withRouteHandler((req: NextRequest) =>
           workspaceId: updated.workspaceId,
           chatId,
           type: 'completed',
+          streamId,
         })
       }
 

apps/sim/app/api/copilot/chat/stream/route.ts

@@ -248,6 +248,7 @@ async function handleResumeRequestBody({
       events: batchEvents,
       previewSessions,
       status: run.status,
+      ...(run.chatId ? { chatId: run.chatId } : {}),
     })
   }
 

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

@@ -27,6 +27,7 @@ const mothershipEventsHandler = createWorkspaceSSE({
           send('task_status', {
             chatId: event.chatId,
             type: event.type,
+            ...(event.streamId ? { streamId: event.streamId } : {}),
             timestamp: Date.now(),
           })
         })