diff --git a/backend/open_webui/migrations/versions/a0b1c2d3e4f5_add_memory_user_id_index.py b/backend/open_webui/migrations/versions/a0b1c2d3e4f5_add_memory_user_id_index.py index a52ade7711..b73e654149 100644 --- a/backend/open_webui/migrations/versions/a0b1c2d3e4f5_add_memory_user_id_index.py +++ b/backend/open_webui/migrations/versions/a0b1c2d3e4f5_add_memory_user_id_index.py @@ -6,6 +6,7 @@ Create Date: 2025-09-15 03:00:00.000000 """ +import sqlalchemy as sa from alembic import op revision = 'a0b1c2d3e4f5' @@ -15,7 +16,12 @@ depends_on = None def upgrade(): - op.create_index('ix_memory_user_id', 'memory', ['user_id']) + conn = op.get_bind() + inspector = sa.inspect(conn) + existing_indexes = {idx['name'] for idx in inspector.get_indexes('memory')} + + if 'ix_memory_user_id' not in existing_indexes: + op.create_index('ix_memory_user_id', 'memory', ['user_id']) def downgrade(): diff --git a/backend/open_webui/migrations/versions/a5c220713937_add_reply_to_id_column_to_message.py b/backend/open_webui/migrations/versions/a5c220713937_add_reply_to_id_column_to_message.py index d54ae24edb..0108e44614 100644 --- a/backend/open_webui/migrations/versions/a5c220713937_add_reply_to_id_column_to_message.py +++ b/backend/open_webui/migrations/versions/a5c220713937_add_reply_to_id_column_to_message.py @@ -19,16 +19,18 @@ depends_on: Union[str, Sequence[str], None] = None def upgrade() -> None: + conn = op.get_bind() + inspector = sa.inspect(conn) + msg_cols = {c['name'] for c in inspector.get_columns('message')} + # Add 'reply_to_id' column to the 'message' table for replying to messages - op.add_column( - 'message', - sa.Column('reply_to_id', sa.Text(), nullable=True), - ) - pass + if 'reply_to_id' not in msg_cols: + op.add_column( + 'message', + sa.Column('reply_to_id', sa.Text(), nullable=True), + ) def downgrade() -> None: # Remove 'reply_to_id' column from the 'message' table op.drop_column('message', 'reply_to_id') - - pass