improvement(mothership): streaming state transitions

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(),
  })
  })