Method migrateDatabase

packages/node-runtime/src/ai/chats.ts:204–291

(db: Database.Database)

  }

  private migrateDatabase(db: Database.Database): void {
    try {
      const hasLegacyConversationTable = this.tableExists(db, 'ai_conversation')

      if (hasLegacyConversationTable) {
        const convColumns = this.getTableColumns(db, 'ai_conversation')
        if (!convColumns.includes('assistant_id')) {
          db.exec(`ALTER TABLE ai_conversation ADD COLUMN assistant_id TEXT DEFAULT '${DEFAULT_GENERAL_ID}'`)
        }
        if (!convColumns.includes('active_message_id')) {
          db.exec('ALTER TABLE ai_conversation ADD COLUMN active_message_id TEXT')
        }

        db.exec(`
          INSERT OR IGNORE INTO ai_chat (
            id, session_id, title, assistant_id, active_message_id, created_at, updated_at
          )
          SELECT id, session_id, title, COALESCE(assistant_id, '${DEFAULT_GENERAL_ID}'),
                 active_message_id, created_at, updated_at
          FROM ai_conversation
        `)
      }

      this.ensureMessageMigrationColumns(db, 'ai_message')

      const messageColumns = this.getTableColumns(db, 'ai_message')
      const hadLegacyConversationId = messageColumns.includes('conversation_id')
      const needsMessageTreeBackfill =
        !messageColumns.includes('parent_id') ||
        !messageColumns.includes('sibling_group_id') ||
        !messageColumns.includes('branch_index') ||
        hadLegacyConversationId

      if (hadLegacyConversationId && !messageColumns.includes('ai_chat_id')) {
        db.exec(`
          DROP INDEX IF EXISTS idx_ai_message_conversation;
          ALTER TABLE ai_message RENAME TO ai_message_legacy;

          CREATE TABLE ai_message (
            id TEXT PRIMARY KEY,
            ai_chat_id TEXT NOT NULL,
            role TEXT NOT NULL,
            content TEXT NOT NULL,
            timestamp INTEGER NOT NULL,
            data_keywords TEXT,
            data_message_count INTEGER,
            content_blocks TEXT,
            parent_id TEXT,
            sibling_group_id TEXT,
            branch_index INTEGER DEFAULT 0,
            debug_context TEXT,
            token_usage TEXT,
            FOREIGN KEY(ai_chat_id) REFERENCES ai_chat(id) ON DELETE CASCADE
          );

          INSERT INTO ai_message (
            id, ai_chat_id, role, content, timestamp, data_keywords, data_message_count,
            content_blocks, parent_id, sibling_group_id, branch_index, debug_context, token_usage