Coverage for mindsdb / migrations / versions / 2023-09-06_d44ab65a6a35_add_agents_table.py: 80%

1"""add_agents_table 

2 

3Revision ID: d44ab65a6a35 

4Revises: ad04ee0bd385 

5Create Date: 2023-09-06 11:32:08.777661 

6 

7""" 

8from alembic import op 

9import sqlalchemy as sa 

10 

11 

12# revision identifiers, used by Alembic. 

13revision = 'd44ab65a6a35' 

14down_revision = 'ad04ee0bd385' 

15branch_labels = None 

16depends_on = None 

17 

18 

19def upgrade(): 

20 op.create_table( 

21 'agents', 

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

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

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

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

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

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

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

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

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

31 sa.PrimaryKeyConstraint('id')) 

32 

33 

34def downgrade(): 

35 op.drop_table('agents')