2024_09_24_0922-6af6a521a53e_update_retrieval_resource.py 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. """update-retrieval-resource
  2. Revision ID: 6af6a521a53e
  3. Revises: ec3df697ebbb
  4. Create Date: 2024-09-24 09:22:43.570120
  5. """
  6. from alembic import op
  7. import models as models
  8. import sqlalchemy as sa
  9. from sqlalchemy.dialects import postgresql
  10. # revision identifiers, used by Alembic.
  11. revision = '6af6a521a53e'
  12. down_revision = 'd57ba9ebb251'
  13. branch_labels = None
  14. depends_on = None
  15. def upgrade():
  16. # ### commands auto generated by Alembic - please adjust! ###
  17. with op.batch_alter_table('dataset_retriever_resources', schema=None) as batch_op:
  18. batch_op.alter_column('document_id',
  19. existing_type=sa.UUID(),
  20. nullable=True)
  21. batch_op.alter_column('data_source_type',
  22. existing_type=sa.TEXT(),
  23. nullable=True)
  24. batch_op.alter_column('segment_id',
  25. existing_type=sa.UUID(),
  26. nullable=True)
  27. # ### end Alembic commands ###
  28. def downgrade():
  29. # ### commands auto generated by Alembic - please adjust! ###
  30. with op.batch_alter_table('dataset_retriever_resources', schema=None) as batch_op:
  31. batch_op.alter_column('segment_id',
  32. existing_type=sa.UUID(),
  33. nullable=False)
  34. batch_op.alter_column('data_source_type',
  35. existing_type=sa.TEXT(),
  36. nullable=False)
  37. batch_op.alter_column('document_id',
  38. existing_type=sa.UUID(),
  39. nullable=False)
  40. # ### end Alembic commands ###