diff --git a/src/backend/langflow/alembic/versions/260dbcc8b680_adds_tables.py b/src/backend/langflow/alembic/versions/260dbcc8b680_adds_tables.py index 0752b6b23..729e1898f 100644 --- a/src/backend/langflow/alembic/versions/260dbcc8b680_adds_tables.py +++ b/src/backend/langflow/alembic/versions/260dbcc8b680_adds_tables.py @@ -23,7 +23,7 @@ def upgrade() -> None: # ### commands auto generated by Alembic - please adjust! ### conn = op.get_bind() - inspector = Inspector.from_engine(conn) + inspector = Inspector.from_engine(conn) # type: ignore # List existing tables existing_tables = inspector.get_table_names() # Drop 'flowstyle' table if it exists @@ -145,8 +145,8 @@ def upgrade() -> None: def downgrade() -> None: # ### commands auto generated by Alembic - please adjust! ### - conn = op.get_bind() - inspector = Inspector.from_engine(conn) + conn = op.get_bind() + inspector = Inspector.from_engine(conn) # type: ignore # List existing tables existing_tables = inspector.get_table_names() if "flow" in existing_tables: diff --git a/src/backend/langflow/alembic/versions/67cc006d50bf_add_profile_image_column.py b/src/backend/langflow/alembic/versions/67cc006d50bf_add_profile_image_column.py index ca0aa0542..6ce9316ac 100644 --- a/src/backend/langflow/alembic/versions/67cc006d50bf_add_profile_image_column.py +++ b/src/backend/langflow/alembic/versions/67cc006d50bf_add_profile_image_column.py @@ -7,9 +7,9 @@ Create Date: 2023-09-08 07:36:13.387318 """ from typing import Sequence, Union -from alembic import op import sqlalchemy as sa import sqlmodel +from alembic import op from sqlalchemy.engine.reflection import Inspector # revision identifiers, used by Alembic. @@ -22,7 +22,7 @@ depends_on: Union[str, Sequence[str], None] = None def upgrade() -> None: # ### commands auto generated by Alembic - please adjust! ### conn = op.get_bind() - inspector = Inspector.from_engine(conn) + inspector = Inspector.from_engine(conn) # type: ignore if "user" in inspector.get_table_names() and "profile_image" not in [ column["name"] for column in inspector.get_columns("user") ]: @@ -39,7 +39,7 @@ def upgrade() -> None: def downgrade() -> None: # ### commands auto generated by Alembic - please adjust! ### conn = op.get_bind() - inspector = Inspector.from_engine(conn) + inspector = Inspector.from_engine(conn) # type: ignore if "user" in inspector.get_table_names() and "profile_image" in [ column["name"] for column in inspector.get_columns("user") ]: