Method insertMessageAfter

packages/node-runtime/src/ai/chats.ts:778–835

(
    aiChatId: string,
    afterMessageId: string,
    role: AIMessageRole,
    content: string,
    contentBlocks?: ContentBlock[],
    tokenUsage?: TokenUsageData
  )

  }

  insertMessageAfter(
    aiChatId: string,
    afterMessageId: string,
    role: AIMessageRole,
    content: string,
    contentBlocks?: ContentBlock[],
    tokenUsage?: TokenUsageData
  ): AIMessage {
    const db = this.getDb()
    const now = Math.floor(Date.now() / 1000)
    const id = this.generateId('msg')

    const pendingDebug = role === 'assistant' ? this.pendingDebugContextMap.get(aiChatId) : undefined
    if (pendingDebug) {
      this.pendingDebugContextMap.delete(aiChatId)
    }

    const childRow = db
      .prepare('SELECT id FROM ai_message WHERE parent_id = ? AND ai_chat_id = ? LIMIT 1')
      .get(afterMessageId, aiChatId) as { id: string } | undefined

    db.prepare(
      `INSERT INTO ai_message (
         id, ai_chat_id, role, content, timestamp, data_keywords, data_message_count,
         content_blocks, token_usage, debug_context, parent_id, sibling_group_id, branch_index
       )
       VALUES (?, ?, ?, ?, ?, NULL, NULL, ?, ?, ?, ?, ?, 0)`
    ).run(
      id,
      aiChatId,
      role,
      content,
      now,
      contentBlocks ? JSON.stringify(contentBlocks) : null,
      tokenUsage ? JSON.stringify(tokenUsage) : null,
      pendingDebug ?? null,
      afterMessageId,
      id
    )

    if (childRow) {
      db.prepare('UPDATE ai_message SET parent_id = ? WHERE id = ?').run(id, childRow.id)
      db.prepare('UPDATE ai_chat SET updated_at = ? WHERE id = ?').run(now, aiChatId)
    } else {
      db.prepare('UPDATE ai_chat SET active_message_id = ?, updated_at = ? WHERE id = ?').run(id, now, aiChatId)
    }

    return {
      id,
      aiChatId,
      role,
      content,
      timestamp: now,
      parentId: afterMessageId,
      contentBlocks,
      tokenUsage,
    }
  }