Coverage for mindsdb / migrations / versions / 2023-05-31_aaecd7012a78_chatbot.py: 82%

1"""chatbot 

2 

3Revision ID: aaecd7012a78 

4Revises: b8be148dbc85 

5Create Date: 2023-05-18 10:53:35.256508 

6 

7""" 

8from alembic import op 

9import sqlalchemy as sa 

10import mindsdb.interfaces.storage.db # noqa 

11 

12 

13# revision identifiers, used by Alembic. 

14revision = 'aaecd7012a78' 

15down_revision = '6d748f2c7b0b' 

16branch_labels = None 

17depends_on = None 

18 

19 

20def upgrade(): 

21 # ### commands auto generated by Alembic - please adjust! ### 

22 op.create_table( 

23 'chat_bots', 

24 sa.Column('id', sa.Integer(), nullable=False), 

25 sa.Column('company_id', sa.Integer(), nullable=True), 

26 sa.Column('user_class', sa.Integer(), nullable=True), 

27 sa.Column('name', sa.String(), nullable=False), 

28 sa.Column('project_id', sa.Integer(), nullable=False), 

29 sa.Column('model_name', sa.String(), nullable=False), 

30 sa.Column('database_id', sa.Integer()), 

31 sa.Column('params', sa.JSON(), nullable=True), 

32 sa.Column('updated_at', sa.DateTime(), nullable=True), 

33 sa.Column('created_at', sa.DateTime(), nullable=True), 

34 sa.PrimaryKeyConstraint('id') 

35 ) 

36 

37 

38def downgrade(): 

39 

40 op.drop_table('chat_bots') 

41 # ### end Alembic commands ###