SuperAGI

1
"""add api_key and web_hook
2

3
Revision ID: 446884dcae58
4
Revises: 71e3980d55f5
5
Create Date: 2023-07-29 10:55:21.714245
6

7
"""
8
from alembic import op
9
import sqlalchemy as sa
10

11

12
# revision identifiers, used by Alembic.
13
revision = '446884dcae58'
14
down_revision = '2fbd6472112c'
15
branch_labels = None
16
depends_on = None
17

18

19
def upgrade() -> None:
20
    # ### commands auto generated by Alembic - please adjust! ###
21
    op.create_table('api_keys',
22
    sa.Column('id', sa.Integer(), nullable=False),
23
    sa.Column('org_id', sa.Integer(), nullable=True),
24
    sa.Column('name', sa.String(), nullable=True),
25
    sa.Column('key', sa.String(), nullable=True),
26
    sa.Column('created_at', sa.DateTime(), nullable=True),
27
    sa.Column('updated_at', sa.DateTime(), nullable=True),
28
    sa.Column('is_expired',sa.Boolean(),nullable=True,default=False),
29
    sa.PrimaryKeyConstraint('id')
30
    )
31
    op.create_table('webhooks',
32
    sa.Column('id', sa.Integer(), nullable=False),
33
    sa.Column('name', sa.String(), nullable=True),
34
    sa.Column('org_id', sa.Integer(), nullable=True),
35
    sa.Column('url', sa.String(), nullable=True),
36
    sa.Column('headers', sa.JSON(), nullable=True),
37
    sa.Column('created_at', sa.DateTime(), nullable=True),
38
    sa.Column('updated_at', sa.DateTime(), nullable=True),
39
    sa.Column('is_deleted',sa.Boolean(),nullable=True),
40
    sa.PrimaryKeyConstraint('id')
41
    )
42
    op.create_table('webhook_events',
43
    sa.Column('id', sa.Integer(), nullable=False),
44
    sa.Column('agent_id', sa.Integer(), nullable=True),
45
    sa.Column('run_id', sa.Integer(), nullable=True),
46
    sa.Column('event', sa.String(), nullable=True),
47
    sa.Column('status', sa.String(), nullable=True),
48
    sa.Column('errors', sa.Text(), nullable=True),
49
    sa.Column('created_at', sa.DateTime(), nullable=True),
50
    sa.Column('updated_at', sa.DateTime(), nullable=True),
51
    sa.PrimaryKeyConstraint('id')
52
    )
53

54
    #add index *********************
55
    # ### end Alembic commands ###
56

57

58
def downgrade() -> None:
59
    # ### commands auto generated by Alembic - please adjust! ###
60
    
61
    op.drop_table('webhooks')
62
    op.drop_table('api_keys')
63
    op.drop_table('webhook_events')
64

65
    # ### end Alembic commands ###
66