Remove more native enums

This commit is contained in:
Weves
2023-11-01 12:38:17 -07:00
committed by Chris Weaver
parent 61c9343a7e
commit 5c9c70dffb

View File

@ -0,0 +1,35 @@
"""forcibly remove more enum types from postgres
Revision ID: 77d07dffae64
Revises: d61e513bef0a
Create Date: 2023-11-01 12:33:01.999617
"""
from alembic import op
from sqlalchemy import String
# revision identifiers, used by Alembic.
revision = "77d07dffae64"
down_revision = "d61e513bef0a"
branch_labels = None
depends_on = None
def upgrade() -> None:
# In a PR:
# https://github.com/danswer-ai/danswer/pull/397/files#diff-f05fb341f6373790b91852579631b64ca7645797a190837156a282b67e5b19c2
# we directly changed some previous migrations. This caused some users to have native enums
# while others wouldn't. This has caused some issues when adding new fields to these enums.
# This migration manually changes the enum types to ensure that nobody uses native enums.
op.alter_column("query_event", "selected_search_flow", type_=String)
op.alter_column("query_event", "feedback", type_=String)
op.alter_column("document_retrieval_feedback", "feedback", type_=String)
op.execute("DROP TYPE IF EXISTS searchtype")
op.execute("DROP TYPE IF EXISTS qafeedbacktype")
op.execute("DROP TYPE IF EXISTS searchfeedbacktype")
def downgrade() -> None:
# We don't want Native Enums, do nothing
pass