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