From 8250c2a2eb481288bca2b464ef3376da6ce2c81b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 28 May 2024 12:40:11 -0700 Subject: [PATCH] Fix column name check in folder_id column (#1999) * bug: fix column name check in folder_id column --- .../alembic/versions/012fb73ac359_add_folder_table.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/alembic/versions/012fb73ac359_add_folder_table.py b/src/backend/base/langflow/alembic/versions/012fb73ac359_add_folder_table.py index b9ebc2ca2..84ebd9ff7 100644 --- a/src/backend/base/langflow/alembic/versions/012fb73ac359_add_folder_table.py +++ b/src/backend/base/langflow/alembic/versions/012fb73ac359_add_folder_table.py @@ -48,7 +48,8 @@ def upgrade() -> None: with op.batch_alter_table("folder", schema=None) as batch_op: batch_op.create_index(batch_op.f("ix_folder_name"), ["name"], unique=False) - if "folder_id" not in inspector.get_columns("flow"): + column_names = [column["name"] for column in inspector.get_columns("flow")] + if "folder_id" not in column_names: with op.batch_alter_table("flow", schema=None) as batch_op: batch_op.add_column(sa.Column("folder_id", sqlmodel.sql.sqltypes.GUID(), nullable=True)) batch_op.create_foreign_key("flow_folder_id_fkey", "folder", ["folder_id"], ["id"]) @@ -62,7 +63,8 @@ def downgrade() -> None: inspector = Inspector.from_engine(conn) # type: ignore table_names = inspector.get_table_names() # ### commands auto generated by Alembic - please adjust! ### - if "folder_id" in inspector.get_columns("flow"): + column_names = [column["name"] for column in inspector.get_columns("flow")] + if "folder_id" in column_names: with op.batch_alter_table("flow", schema=None) as batch_op: batch_op.add_column(sa.Column("folder", sa.VARCHAR(), nullable=True)) batch_op.drop_constraint("flow_folder_id_fkey", type_="foreignkey")