chore: import upstream snapshot with attribution

apps/sim/lib/copilot/request/context/request-context.ts

@@ -0,0 +1,33 @@
+import { generateId } from '@sim/utils/id'
+import { TraceCollector } from '@/lib/copilot/request/trace'
+import type { StreamingContext } from '@/lib/copilot/request/types'
+
+/**
+ * Create a fresh StreamingContext.
+ */
+export function createStreamingContext(overrides?: Partial<StreamingContext>): StreamingContext {
+  return {
+    chatId: undefined,
+    executionId: undefined,
+    runId: undefined,
+    messageId: generateId(),
+    accumulatedContent: '',
+    finalAssistantContent: '',
+    sawMainToolCall: false,
+    contentBlocks: [],
+    toolCalls: new Map(),
+    pendingToolPromises: new Map(),
+    currentThinkingBlock: null,
+    subagentThinkingBlocks: new Map(),
+    isInThinkingBlock: false,
+    subAgentContent: {},
+    subAgentToolCalls: {},
+    pendingContent: '',
+    streamComplete: false,
+    wasAborted: false,
+    errors: [],
+    activeFileIntents: new Map(),
+    trace: new TraceCollector(),
+    ...overrides,
+  }
+}

apps/sim/lib/copilot/request/context/result.test.ts

@@ -0,0 +1,97 @@
+/**
+ * @vitest-environment node
+ */
+import { describe, expect, it } from 'vitest'
+import { MothershipStreamV1ToolOutcome } from '@/lib/copilot/generated/mothership-stream-v1'
+import { FunctionExecute } from '@/lib/copilot/generated/tool-catalog-v1'
+import { buildToolCallSummaries } from '@/lib/copilot/request/context/result'
+import { TraceCollector } from '@/lib/copilot/request/trace'
+import type { StreamingContext } from '@/lib/copilot/request/types'
+
+function makeContext(): StreamingContext {
+  return {
+    chatId: undefined,
+    requestId: undefined,
+    executionId: undefined,
+    runId: undefined,
+    messageId: 'msg-1',
+    accumulatedContent: '',
+    finalAssistantContent: '',
+    sawMainToolCall: false,
+    contentBlocks: [],
+    toolCalls: new Map(),
+    pendingToolPromises: new Map(),
+    awaitingAsyncContinuation: undefined,
+    currentThinkingBlock: null,
+    subagentThinkingBlocks: new Map(),
+    isInThinkingBlock: false,
+    subAgentContent: {},
+    subAgentToolCalls: {},
+    pendingContent: '',
+    streamComplete: false,
+    wasAborted: false,
+    errors: [],
+    trace: new TraceCollector(),
+  }
+}
+
+describe('buildToolCallSummaries', () => {
+  it.concurrent('keeps pending tools as pending instead of defaulting to success', () => {
+    const context = makeContext()
+    context.toolCalls.set('tool-1', {
+      id: 'tool-1',
+      name: 'download_to_workspace_file',
+      status: 'pending',
+      startTime: 1,
+    })
+
+    const summaries = buildToolCallSummaries(context)
+
+    expect(summaries).toHaveLength(1)
+    expect(summaries[0]?.status).toBe('pending')
+  })
+
+  it.concurrent('keeps executing tools as executing when no result exists yet', () => {
+    const context = makeContext()
+    context.toolCalls.set('tool-2', {
+      id: 'tool-2',
+      name: FunctionExecute.id,
+      status: 'executing',
+      startTime: 1,
+    })
+
+    const summaries = buildToolCallSummaries(context)
+
+    expect(summaries).toHaveLength(1)
+    expect(summaries[0]?.status).toBe('executing')
+  })
+
+  it.concurrent(
+    'preserves canonical terminal statuses instead of deriving them from result success',
+    () => {
+      const context = makeContext()
+      context.toolCalls.set('tool-3', {
+        id: 'tool-3',
+        name: 'download_to_workspace_file',
+        status: MothershipStreamV1ToolOutcome.cancelled,
+        result: { success: false },
+        error: 'Stopped by user',
+        startTime: 1,
+        endTime: 2,
+      })
+
+      const summaries = buildToolCallSummaries(context)
+
+      expect(summaries).toHaveLength(1)
+      expect(summaries[0]).toEqual({
+        id: 'tool-3',
+        name: 'download_to_workspace_file',
+        status: MothershipStreamV1ToolOutcome.cancelled,
+        params: undefined,
+        result: undefined,
+        error: 'Stopped by user',
+        durationMs: 1,
+      })
+    }
+  )
+})

apps/sim/lib/copilot/request/context/result.ts

@@ -0,0 +1,18 @@
+import { getToolCallStateOutput } from '@/lib/copilot/request/tool-call-state'
+import type { StreamingContext, ToolCallSummary } from '@/lib/copilot/request/types'
+
+/**
+ * Build a ToolCallSummary array from the streaming context.
+ */
+export function buildToolCallSummaries(context: StreamingContext): ToolCallSummary[] {
+  return Array.from(context.toolCalls.values()).map((toolCall) => ({
+    id: toolCall.id,
+    name: toolCall.name,
+    status: toolCall.status,
+    params: toolCall.params,
+    result: getToolCallStateOutput(toolCall),
+    error: toolCall.error,
+    durationMs:
+      toolCall.endTime && toolCall.startTime ? toolCall.endTime - toolCall.startTime : undefined,
+  }))
+}