Function createThinkingChunks

packages/ai-client/tests/test-utils.ts:352–403

(
  thinkingContent: string,
  textContent: string = '',
  messageId: string = 'msg-1',
  model: string = 'test',
)

 * Helper to create thinking chunks (AG-UI format using STEP_FINISHED for thinking)
 */
export function createThinkingChunks(
  thinkingContent: string,
  textContent: string = '',
  messageId: string = 'msg-1',
  model: string = 'test',
): Array<StreamChunk> {
  const chunks: Array<StreamChunk> = []
  let accumulatedThinking = ''
  const runId = `run-${messageId}`
  const stepId = `step-${messageId}`

  // Add thinking chunks via STEP_FINISHED events
  for (const chunk of thinkingContent) {
    accumulatedThinking += chunk
    chunks.push({
      type: 'STEP_FINISHED',
      stepName: stepId,
      stepId,
      model,
      timestamp: Date.now(),
      delta: chunk,
      content: accumulatedThinking,
    } as StreamChunk)
  }

  // Optionally add text content after thinking
  if (textContent) {
    let accumulatedText = ''
    for (const chunk of textContent) {
      accumulatedText += chunk
      chunks.push({
        type: 'TEXT_MESSAGE_CONTENT',
        messageId,
        model,
        timestamp: Date.now(),
        delta: chunk,
        content: accumulatedText,
      } as StreamChunk)
    }
  }

  chunks.push({
    type: 'RUN_FINISHED',
    runId,
    threadId: `thread-${messageId}`,
    model,
    timestamp: Date.now(),
    finishReason: 'stop',
  } as StreamChunk)

  return chunks
}