Method forkAIChat

packages/node-runtime/src/ai/chats.ts:678–744

(sourceAIChatId: string, upToMessageId: string, title?: string)

  }

  forkAIChat(sourceAIChatId: string, upToMessageId: string, title?: string): AIChat {
    const db = this.getDb()
    const source = this.getAIChat(sourceAIChatId)
    if (!source) {
      throw new Error('Source AI chat not found')
    }

    const activePath = this.getActivePathRows(sourceAIChatId)
    const cutIndex = activePath.findIndex((row) => row.id === upToMessageId)
    if (cutIndex < 0) {
      throw new Error('Message not on active path')
    }

    const messagesToCopy = activePath.slice(0, cutIndex + 1)
    const now = Math.floor(Date.now() / 1000)
    const newConvId = this.generateId('conv')
    const forkTitle = title || `${source.title || 'Untitled'} (fork)`

    db.prepare(
      `INSERT INTO ai_chat (id, session_id, title, assistant_id, active_message_id, created_at, updated_at)
       VALUES (?, ?, ?, ?, NULL, ?, ?)`
    ).run(newConvId, source.sessionId, forkTitle, source.assistantId, now, now)

    const idMap = new Map<string, string>()
    let lastNewId: string | null = null

    for (const row of messagesToCopy) {
      const newMsgId = this.generateId('msg')
      idMap.set(row.id, newMsgId)
      const newParentId = row.parentId ? (idMap.get(row.parentId) ?? null) : null

      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, ?, ?, 0)`
      ).run(
        newMsgId,
        newConvId,
        row.role,
        row.content,
        row.timestamp,
        row.dataKeywords,
        row.dataMessageCount,
        row.contentBlocks,
        row.tokenUsage,
        newParentId,
        newMsgId
      )
      lastNewId = newMsgId
    }

    if (lastNewId) {
      db.prepare('UPDATE ai_chat SET active_message_id = ? WHERE id = ?').run(lastNewId, newConvId)
    }

    return {