feat: add datasource OAuth client setup command and refactor related models

api/migrations/versions/2025_07_18_1409-bb3812d469dd_merge_rag_main.py

@@ -0,0 +1,25 @@
+"""empty message
+
+Revision ID: bb3812d469dd
+Revises: 15e40b74a6d2, 71f5020c6470
+Create Date: 2025-07-18 14:09:12.778358
+
+"""
+from alembic import op
+import models as models
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = 'bb3812d469dd'
+down_revision = ('15e40b74a6d2', '71f5020c6470')
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    pass
+
+
+def downgrade():
+    pass

api/migrations/versions/2025_07_18_1409-d4a76fde2724_datasource_oauth_1.py

@@ -0,0 +1,65 @@
+"""datasource_oauth_1
+
+Revision ID: d4a76fde2724
+Revises: bb3812d469dd
+Create Date: 2025-07-18 14:09:42.551752
+
+"""
+from alembic import op
+import models as models
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision = 'd4a76fde2724'
+down_revision = 'bb3812d469dd'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_table('tenant_plugin_auto_upgrade_strategies')
+    with op.batch_alter_table('datasource_oauth_params', schema=None) as batch_op:
+        batch_op.alter_column('plugin_id',
+               existing_type=sa.UUID(),
+               type_=sa.String(length=255),
+               existing_nullable=False)
+
+    with op.batch_alter_table('datasource_providers', schema=None) as batch_op:
+        batch_op.alter_column('plugin_id',
+               existing_type=sa.TEXT(),
+               type_=sa.String(length=255),
+               existing_nullable=False)
+
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('datasource_providers', schema=None) as batch_op:
+        batch_op.alter_column('plugin_id',
+               existing_type=sa.String(length=255),
+               type_=sa.TEXT(),
+               existing_nullable=False)
+
+    with op.batch_alter_table('datasource_oauth_params', schema=None) as batch_op:
+        batch_op.alter_column('plugin_id',
+               existing_type=sa.String(length=255),
+               type_=sa.UUID(),
+               existing_nullable=False)
+
+    op.create_table('tenant_plugin_auto_upgrade_strategies',
+    sa.Column('id', sa.UUID(), server_default=sa.text('uuid_generate_v4()'), autoincrement=False, nullable=False),
+    sa.Column('tenant_id', sa.UUID(), autoincrement=False, nullable=False),
+    sa.Column('strategy_setting', sa.VARCHAR(length=16), server_default=sa.text("'fix_only'::character varying"), autoincrement=False, nullable=False),
+    sa.Column('upgrade_time_of_day', sa.INTEGER(), autoincrement=False, nullable=False),
+    sa.Column('upgrade_mode', sa.VARCHAR(length=16), server_default=sa.text("'exclude'::character varying"), autoincrement=False, nullable=False),
+    sa.Column('exclude_plugins', postgresql.ARRAY(sa.VARCHAR(length=255)), autoincrement=False, nullable=False),
+    sa.Column('include_plugins', postgresql.ARRAY(sa.VARCHAR(length=255)), autoincrement=False, nullable=False),
+    sa.Column('created_at', postgresql.TIMESTAMP(), server_default=sa.text('CURRENT_TIMESTAMP'), autoincrement=False, nullable=False),
+    sa.Column('updated_at', postgresql.TIMESTAMP(), server_default=sa.text('CURRENT_TIMESTAMP'), autoincrement=False, nullable=False),
+    sa.PrimaryKeyConstraint('id', name=op.f('tenant_plugin_auto_upgrade_strategy_pkey')),
+    sa.UniqueConstraint('tenant_id', name=op.f('unique_tenant_plugin_auto_upgrade_strategy'), postgresql_include=[], postgresql_nulls_not_distinct=False)
+    )
+    # ### end Alembic commands ###