SuperAGI

1
"""open_ai_error_handling
2

3
Revision ID: 661ec8a4c32e
4
Revises: 40affbf3022b
5
Create Date: 2023-09-07 10:41:07.462436
6

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

11

12
# revision identifiers, used by Alembic.
13
revision = '661ec8a4c32e'
14
down_revision = 'c4f2f6ba602a'
15
branch_labels = None
16
depends_on = None
17

18

19
def upgrade() -> None:
20
    # ### commands auto generated by Alembic - please adjust! ###
21
    op.add_column('agent_execution_feeds', sa.Column('error_message', sa.String(), nullable=True))
22
    op.add_column('agent_executions', sa.Column('last_shown_error_id', sa.Integer(), nullable=True))
23
    # ### end Alembic commands ###
24

25

26
def downgrade() -> None:
27
    # ### commands auto generated by Alembic - please adjust! ###
28
    op.drop_column('agent_executions', 'last_shown_error_id')
29
    op.drop_column('agent_execution_feeds', 'error_message')
30
    # ### end Alembic commands ###
31