diff --git a/docker/build_and_push.Dockerfile b/docker/build_and_push.Dockerfile index 3cfc134fb..9a44f2919 100644 --- a/docker/build_and_push.Dockerfile +++ b/docker/build_and_push.Dockerfile @@ -69,7 +69,7 @@ FROM python:3.12.3-slim AS runtime RUN apt-get update \ && apt-get upgrade -y \ - && apt-get install git -y \ + && apt-get install -y git libpq5 \ && apt-get clean \ && rm -rf /var/lib/apt/lists/* \ && useradd user -u 1000 -g 0 --no-create-home --home-dir /app/data diff --git a/docker/build_and_push_base.Dockerfile b/docker/build_and_push_base.Dockerfile index 4d40eb1d0..9ec1cdb34 100644 --- a/docker/build_and_push_base.Dockerfile +++ b/docker/build_and_push_base.Dockerfile @@ -73,7 +73,7 @@ FROM python:3.12.3-slim AS runtime RUN apt-get update \ && apt-get upgrade -y \ - && apt-get install git -y \ + && apt-get install -y git libpq5 \ && apt-get clean \ && rm -rf /var/lib/apt/lists/* \ && useradd user -u 1000 -g 0 --no-create-home --home-dir /app/data diff --git a/src/backend/Dockerfile b/src/backend/Dockerfile index ba1f3c946..df673a7be 100644 --- a/src/backend/Dockerfile +++ b/src/backend/Dockerfile @@ -3,7 +3,7 @@ FROM langflowai/backend_build as backend_build FROM python:3.10-slim WORKDIR /app -RUN apt-get update && apt-get install git -y +RUN apt-get update && apt-get install -y git libpq-dev gcc COPY --from=backend_build /app/dist/*.whl /app/ RUN pip install langflow-*.whl diff --git a/src/backend/base/langflow/alembic/versions/f3b2d1f1002d_add_column_access_type_to_flow.py b/src/backend/base/langflow/alembic/versions/f3b2d1f1002d_add_column_access_type_to_flow.py index 575a151b7..56f110e32 100644 --- a/src/backend/base/langflow/alembic/versions/f3b2d1f1002d_add_column_access_type_to_flow.py +++ b/src/backend/base/langflow/alembic/versions/f3b2d1f1002d_add_column_access_type_to_flow.py @@ -21,13 +21,17 @@ depends_on: Union[str, Sequence[str], None] = None def upgrade() -> None: conn = op.get_bind() + access_type_enum = sa.Enum('PRIVATE', 'PUBLIC', name='access_type_enum') + access_type_enum.create(conn, checkfirst=True) with op.batch_alter_table('flow', schema=None) as batch_op: if not migration.column_exists(table_name='flow', column_name='access_type', conn=conn): - batch_op.add_column(sa.Column('access_type', sa.Enum('PRIVATE', 'PUBLIC', name='access_type_enum'), server_default='private', nullable=False)) - + batch_op.add_column(sa.Column('access_type', access_type_enum, server_default=sa.text("'PRIVATE'"), nullable=False)) def downgrade() -> None: conn = op.get_bind() with op.batch_alter_table('flow', schema=None) as batch_op: if migration.column_exists(table_name='flow', column_name='access_type', conn=conn): batch_op.drop_column('access_type') + + access_type_enum = sa.EnuM('PRIVATE', 'PUBLIC', name='access_type_enum') + access_type_enum.drop(conn, checkfirst=True) diff --git a/src/backend/base/langflow/services/database/models/flow/model.py b/src/backend/base/langflow/services/database/models/flow/model.py index 13ec269df..a706c7ad9 100644 --- a/src/backend/base/langflow/services/database/models/flow/model.py +++ b/src/backend/base/langflow/services/database/models/flow/model.py @@ -33,8 +33,8 @@ HEX_COLOR_LENGTH = 7 class AccessTypeEnum(str, Enum): - PRIVATE = "private" - PUBLIC = "public" + PRIVATE = "PRIVATE" + PUBLIC = "PUBLIC" class FlowBase(SQLModel): @@ -59,7 +59,7 @@ class FlowBase(SQLModel): values_callable=lambda enum: [member.value for member in enum], ), nullable=False, - server_default=text("'private'"), + server_default=text("'PRIVATE'"), ), )