From db024ad7b70c88400c71ecfd50fd6864df6e679d Mon Sep 17 00:00:00 2001 From: Weves Date: Wed, 20 Sep 2023 13:14:19 -0700 Subject: [PATCH] Provide default value for 'retrieval_enabled' column in migration --- .../versions/d5645c915d0e_remove_deletion_attempt_table.py | 1 + .../versions/dba7f71618f5_danswer_custom_tool_flow.py | 5 ++++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/backend/alembic/versions/d5645c915d0e_remove_deletion_attempt_table.py b/backend/alembic/versions/d5645c915d0e_remove_deletion_attempt_table.py index 120d63e16..964809942 100644 --- a/backend/alembic/versions/d5645c915d0e_remove_deletion_attempt_table.py +++ b/backend/alembic/versions/d5645c915d0e_remove_deletion_attempt_table.py @@ -18,6 +18,7 @@ depends_on = None def upgrade() -> None: op.drop_table("deletion_attempt") + sa.Enum(name="deletionstatus").drop(op.get_bind(), checkfirst=False) def downgrade() -> None: diff --git a/backend/alembic/versions/dba7f71618f5_danswer_custom_tool_flow.py b/backend/alembic/versions/dba7f71618f5_danswer_custom_tool_flow.py index 100f37148..bef3657de 100644 --- a/backend/alembic/versions/dba7f71618f5_danswer_custom_tool_flow.py +++ b/backend/alembic/versions/dba7f71618f5_danswer_custom_tool_flow.py @@ -18,8 +18,11 @@ depends_on = None def upgrade() -> None: op.add_column( - "persona", sa.Column("retrieval_enabled", sa.Boolean(), nullable=False) + "persona", + sa.Column("retrieval_enabled", sa.Boolean(), nullable=True), ) + op.execute("UPDATE persona SET retrieval_enabled = true") + op.alter_column("persona", "retrieval_enabled", nullable=False) def downgrade() -> None: