SuperAGI

1
"""renaming templates
2

3
Revision ID: d9b3436197eb
4
Revises: 3356a2f89a33
5
Create Date: 2023-06-10 09:28:28.262705
6

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

11

12
# revision identifiers, used by Alembic.
13
revision = 'd9b3436197eb'
14
down_revision = '3356a2f89a33'
15
branch_labels = None
16
depends_on = None
17

18

19
def upgrade() -> None:
20
    op.rename_table('agent_templates', 'agent_workflows')
21
    op.rename_table('agent_template_steps', 'agent_workflow_steps')
22
    with op.batch_alter_table('agent_workflow_steps') as bop:
23
        bop.alter_column('agent_template_id', new_column_name='agent_workflow_id')
24
    with op.batch_alter_table('agents') as bop:
25
        bop.alter_column('agent_template_id', new_column_name='agent_workflow_id')
26

27

28
def downgrade() -> None:
29
    op.rename_table('agent_workflows', 'agent_templates')
30
    op.rename_table('agent_workflow_steps', 'agent_template_steps')
31
    with op.batch_alter_table('agent_templates') as bop:
32
        bop.alter_column('agent_workflow_id', new_column_name='agent_template_id')
33
    with op.batch_alter_table('agents') as bop:
34
        bop.alter_column('agent_workflow_id', new_column_name='agent_template_id')
35