1  from alembic import op
 2  import sqlalchemy as sa
 3  
 4  revision = '016'
 5  down_revision = '015'
 6  branch_labels = None
 7  depends_on = None
 8  
 9  def upgrade():
10      try:
11          op.add_column('retrieval_events', sa.Column('chunk_id', sa.String(255), nullable=True, index=True))
12          op.add_column('retrieval_events', sa.Column('chunk_token_length', sa.Integer(), nullable=True))
13          op.add_column('retrieval_events', sa.Column('chunk_text_length', sa.Integer(), nullable=True))
14      except Exception as e:
15          print(f"Error adding chunk analytics columns: {e}")
16  
17  def downgrade():
18      try:
19          op.drop_column('retrieval_events', 'chunk_id')
20          op.drop_column('retrieval_events', 'chunk_token_length')
21          op.drop_column('retrieval_events', 'chunk_text_length')
22      except Exception as e:
23          print(f"Error dropping chunk analytics columns: {e}")