diff --git a/api/migrations/versions/2025_11_18_1400-a7b4e8f2c9d1_add_enduser_authentication_provider.py b/api/migrations/versions/2025_11_18_1400-a7b4e8f2c9d1_add_enduser_authentication_provider.py index 2668b9c752..f4e1a504dd 100644 --- a/api/migrations/versions/2025_11_18_1400-a7b4e8f2c9d1_add_enduser_authentication_provider.py +++ b/api/migrations/versions/2025_11_18_1400-a7b4e8f2c9d1_add_enduser_authentication_provider.py @@ -23,19 +23,18 @@ def upgrade(): sa.Column( "id", models.types.StringUUID(), - server_default=sa.text("uuid_generate_v4()"), nullable=False, ), sa.Column( "name", sa.String(length=256), - server_default=sa.text("'API KEY 1'::character varying"), + server_default="API KEY 1", nullable=False, ), sa.Column("tenant_id", models.types.StringUUID(), nullable=False), sa.Column("end_user_id", models.types.StringUUID(), nullable=False), - sa.Column("provider", sa.String(length=256), nullable=False), - sa.Column("encrypted_credentials", sa.Text(), nullable=True), + sa.Column("provider", sa.Text(), nullable=False), + sa.Column("encrypted_credentials", sa.Text(), server_default="", nullable=False), sa.Column( "created_at", sa.DateTime(), @@ -51,23 +50,22 @@ def upgrade(): sa.Column( "credential_type", sa.String(length=32), - server_default=sa.text("'api-key'::character varying"), + server_default="api-key", nullable=False, ), sa.Column("expires_at", sa.BigInteger(), server_default=sa.text("-1"), nullable=False), sa.PrimaryKeyConstraint("id", name="tool_enduser_authentication_provider_pkey"), sa.UniqueConstraint( - "tenant_id", - "provider", "end_user_id", + "provider", "name", name="unique_enduser_authentication_provider", ), ) op.create_index( - "tool_enduser_authentication_provider_tenant_id_idx", + "tool_enduser_authentication_provider_name_idx", "tool_enduser_authentication_providers", - ["tenant_id"], + ["name"], unique=False, ) op.create_index( @@ -76,17 +74,27 @@ def upgrade(): ["end_user_id"], unique=False, ) + op.create_index( + "tool_enduser_authentication_provider_provider_idx", + "tool_enduser_authentication_providers", + ["provider"], + unique=False, + ) # ### end Alembic commands ### def downgrade(): # ### commands auto generated by Alembic - please adjust! ### + op.drop_index( + "tool_enduser_authentication_provider_provider_idx", + table_name="tool_enduser_authentication_providers", + ) op.drop_index( "tool_enduser_authentication_provider_end_user_id_idx", table_name="tool_enduser_authentication_providers", ) op.drop_index( - "tool_enduser_authentication_provider_tenant_id_idx", + "tool_enduser_authentication_provider_name_idx", table_name="tool_enduser_authentication_providers", ) op.drop_table("tool_enduser_authentication_providers")