code: def upgrade():
      # ### commands auto generated by Alembic - please adjust! ###
      with op.batch_alter_table("chat_bots", schema=None) as batch_op:
          batch_op.drop_column("chat_engine")
          batch_op.drop_column("user_class")
          batch_op.drop_column("company_id")
          batch_op.drop_column("is_running")
  
imports: from alembic import op
  import sqlalchemy
  import mindsdb.interfaces.storage.db
label_desc: Use image node_id 1 for calling a global function with example usage: upgrade() without return types