Coverage for mindsdb / migrations / versions / 2023-06-16_9d6271bb2c38_update_chat_bots_table.py: 57%

1"""update_chat_bots_table 

2 

3Revision ID: 9d6271bb2c38 

4Revises: aaecd7012a78 

5Create Date: 2023-06-16 12:08:50.170809 

6 

7""" 

8from alembic import op 

9import sqlalchemy as sa 

10 

11 

12# revision identifiers, used by Alembic. 

13revision = '9d6271bb2c38' 

14down_revision = 'aaecd7012a78' 

15branch_labels = None 

16depends_on = None 

17 

18 

19def upgrade(): 

20 with op.batch_alter_table('chat_bots', schema=None) as batch_op: 

21 batch_op.add_column(sa.Column('chat_engine', sa.String(), nullable=True)) 

22 batch_op.add_column(sa.Column('is_running', sa.Boolean(), default=True)) 

23 

24 

25def downgrade(): 

26 with op.batch_alter_table('chat_bots', schema=None) as batch_op: 

27 batch_op.drop_column('chat_engine') 

28 batch_op.drop_column('is_running')