From d66b6c0559ed7339c1fb5f2a457190b8c9a1ac29 Mon Sep 17 00:00:00 2001 From: Yuhong Sun Date: Thu, 21 Mar 2024 12:27:56 -0700 Subject: [PATCH] Fix Tag Document Source Enum (#1240) --- ...3b470d1a_remove_documentsource_from_tag.py | 36 +++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 backend/alembic/versions/91fd3b470d1a_remove_documentsource_from_tag.py diff --git a/backend/alembic/versions/91fd3b470d1a_remove_documentsource_from_tag.py b/backend/alembic/versions/91fd3b470d1a_remove_documentsource_from_tag.py new file mode 100644 index 000000000..b8f1a7292 --- /dev/null +++ b/backend/alembic/versions/91fd3b470d1a_remove_documentsource_from_tag.py @@ -0,0 +1,36 @@ +"""Remove DocumentSource from Tag + +Revision ID: 91fd3b470d1a +Revises: 173cae5bba26 +Create Date: 2024-03-21 12:05:23.956734 + +""" +from alembic import op +import sqlalchemy as sa +from danswer.configs.constants import DocumentSource + +# revision identifiers, used by Alembic. +revision = "91fd3b470d1a" +down_revision = "173cae5bba26" +branch_labels = None +depends_on = None + + +def upgrade() -> None: + op.alter_column( + "tag", + "source", + type_=sa.String(length=50), + existing_type=sa.Enum(DocumentSource, native_enum=False), + existing_nullable=False, + ) + + +def downgrade() -> None: + op.alter_column( + "tag", + "source", + type_=sa.Enum(DocumentSource, native_enum=False), + existing_type=sa.String(length=50), + existing_nullable=False, + )