Coverage for mindsdb / migrations / versions / 2024-06-06_cbedc4968d5d_store_llm_data.py: 82%

1"""store llm data 

2 

3Revision ID: cbedc4968d5d 

4Revises: 2958416fbe75 

5Create Date: 2024-06-06 13:59:45.158089 

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 = 'cbedc4968d5d' 

15down_revision = '2958416fbe75' 

16branch_labels = None 

17depends_on = None 

18 

19 

20def upgrade(): 

21 op.create_table( 

22 'llm_data', 

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

24 sa.Column('input', sa.String(), nullable=False), 

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

26 sa.Column('model_id', sa.String(), nullable=False), 

27 sa.Column('created_at', sa.DateTime(), server_default=sa.func.now()), 

28 sa.Column('updated_at', sa.DateTime(), server_default=sa.func.now(), onupdate=sa.func.now()), 

29 sa.PrimaryKeyConstraint('id') 

30 ) 

31 

32 

33def downgrade(): 

34 op.drop_table('llm_data')