4bcffcd64aa4_update_dataset_model_field_null_.py 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. """update_dataset_model_field_null_available
  2. Revision ID: 4bcffcd64aa4
  3. Revises: 853f9b9cd3b6
  4. Create Date: 2023-08-28 20:58:50.077056
  5. """
  6. import sqlalchemy as sa
  7. from alembic import op
  8. # revision identifiers, used by Alembic.
  9. revision = '4bcffcd64aa4'
  10. down_revision = '853f9b9cd3b6'
  11. branch_labels = None
  12. depends_on = None
  13. def upgrade():
  14. # ### commands auto generated by Alembic - please adjust! ###
  15. with op.batch_alter_table('datasets', schema=None) as batch_op:
  16. batch_op.alter_column('embedding_model',
  17. existing_type=sa.VARCHAR(length=255),
  18. nullable=True,
  19. existing_server_default=sa.text("'text-embedding-ada-002'::character varying"))
  20. batch_op.alter_column('embedding_model_provider',
  21. existing_type=sa.VARCHAR(length=255),
  22. nullable=True,
  23. existing_server_default=sa.text("'openai'::character varying"))
  24. # ### end Alembic commands ###
  25. def downgrade():
  26. # ### commands auto generated by Alembic - please adjust! ###
  27. with op.batch_alter_table('datasets', schema=None) as batch_op:
  28. batch_op.alter_column('embedding_model_provider',
  29. existing_type=sa.VARCHAR(length=255),
  30. nullable=False,
  31. existing_server_default=sa.text("'openai'::character varying"))
  32. batch_op.alter_column('embedding_model',
  33. existing_type=sa.VARCHAR(length=255),
  34. nullable=False,
  35. existing_server_default=sa.text("'text-embedding-ada-002'::character varying"))
  36. # ### end Alembic commands ###