Function addChunkToMessage

packages/ai-devtools/src/store/ai-context.tsx:470–514

(conversationId: string, chunk: Chunk)

  }

  function addChunkToMessage(conversationId: string, chunk: Chunk): void {
    const conv = state.conversations[conversationId]
    if (!conv) return

    if (chunk.messageId) {
      const messageIndex = conv.messages.findIndex(
        (msg) => msg.id === chunk.messageId,
      )

      if (messageIndex !== -1) {
        queueMessageChunk(conversationId, messageIndex, chunk)
        return
      } else {
        // Create new message with the chunk
        const newMessage: Message = {
          id: chunk.messageId,
          role: 'assistant',
          content: '',
          timestamp: chunk.timestamp,
          model: conv.model,
          chunks: [chunk],
        }
        setState(
          'conversations',
          conversationId,
          'messages',
          conv.messages.length,
          newMessage,
        )
        return
      }
    }

    // Find last assistant message
    for (let i = conv.messages.length - 1; i >= 0; i--) {
      const message = conv.messages[i]
      if (message && message.role === 'assistant') {
        queueMessageChunk(conversationId, i, chunk)
        return
      }
    }

    // Fallback: add to conversation's main chunks array if no assistant message found
    addChunk(conversationId, chunk)
  }

  function updateMessageUsage(
    conversationId: string,