114eed84c228_remove_tool_id_from_model_invoke.py 900 B

1234567891011121314151617181920212223242526272829303132
  1. """remove tool id from model invoke
  2. Revision ID: 114eed84c228
  3. Revises: c71211c8f604
  4. Create Date: 2024-01-10 04:40:57.257824
  5. """
  6. import sqlalchemy as sa
  7. from alembic import op
  8. from sqlalchemy.dialects import postgresql
  9. # revision identifiers, used by Alembic.
  10. revision = '114eed84c228'
  11. down_revision = 'c71211c8f604'
  12. branch_labels = None
  13. depends_on = None
  14. def upgrade():
  15. # ### commands auto generated by Alembic - please adjust! ###
  16. with op.batch_alter_table('tool_model_invokes', schema=None) as batch_op:
  17. batch_op.drop_column('tool_id')
  18. # ### end Alembic commands ###
  19. def downgrade():
  20. # ### commands auto generated by Alembic - please adjust! ###
  21. with op.batch_alter_table('tool_model_invokes', schema=None) as batch_op:
  22. batch_op.add_column(sa.Column('tool_id', postgresql.UUID(), autoincrement=False, nullable=False))
  23. # ### end Alembic commands ###