1  from alembic import op
 2  import sqlalchemy as sa
 3  
 4  # revision identifiers, used by Alembic.
 5  revision = '007'
 6  down_revision = '006'
 7  branch_labels = None
 8  depends_on = None
 9  
10  def upgrade():
11      try:
12          op.add_column('output', sa.Column('latency_ms', sa.Integer(), nullable=True))
13      except Exception as e:
14          print(f"Error adding latency_ms column: {e}")
15  
16  def downgrade():
17      try:
18          op.drop_column('output', 'latency_ms')
19      except Exception as e:
20          print(f"Error dropping latency_ms column: {e}")