open-webui/backend/open_webui/migrations/versions/018012973d35_add_indexes.py

"""Add indexes

Revision ID: 018012973d35
Revises: d31026856c01
Create Date: 2025-08-13 03:00:00.000000

"""

from alembic import op
import sqlalchemy as sa

revision = '018012973d35'
down_revision = 'd31026856c01'
branch_labels = None
depends_on = None


def upgrade():
    # Chat table indexes
    op.create_index('folder_id_idx', 'chat', ['folder_id'])
    op.create_index('user_id_pinned_idx', 'chat', ['user_id', 'pinned'])
    op.create_index('user_id_archived_idx', 'chat', ['user_id', 'archived'])
    op.create_index('updated_at_user_id_idx', 'chat', ['updated_at', 'user_id'])
    op.create_index('folder_id_user_id_idx', 'chat', ['folder_id', 'user_id'])

    # Tag table index
    op.create_index('user_id_idx', 'tag', ['user_id'])

    # Function table index
    op.create_index('is_global_idx', 'function', ['is_global'])


def downgrade():
    # Chat table indexes
    op.drop_index('folder_id_idx', table_name='chat')
    op.drop_index('user_id_pinned_idx', table_name='chat')
    op.drop_index('user_id_archived_idx', table_name='chat')
    op.drop_index('updated_at_user_id_idx', table_name='chat')
    op.drop_index('folder_id_user_id_idx', table_name='chat')

    # Tag table index
    op.drop_index('user_id_idx', table_name='tag')

    # Function table index

    op.drop_index('is_global_idx', table_name='function')