solace-agent-mesh 1.11.2__py3-none-any.whl

solace_agent_mesh/gateway/http_sse/alembic/versions/20251023_add_project_users_table.py

@@ -0,0 +1,72 @@
+"""Add project_users table for multi-user project access
+
+Revision ID: 20251023_add_project_users
+Revises: 20251023_remove_created_by_user_id
+Create Date: 2025-10-23
+
+"""
+from typing import Sequence, Union
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'add_project_users_001'
+down_revision: Union[str, None] = 'add_projects_table_001'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """
+    Create project_users table to enable multi-user access to projects.
+    
+    This table tracks which users have access to which projects and their roles.
+    It supports future features for collaborative project management.
+    """
+    # Create project_users table
+    op.create_table(
+        'project_users',
+        sa.Column('id', sa.String(), nullable=False),
+        sa.Column('project_id', sa.String(), nullable=False),
+        sa.Column('user_id', sa.String(), nullable=False),
+        sa.Column('role', sa.String(), nullable=False),
+        sa.Column('added_at', sa.BigInteger(), nullable=False),
+        sa.Column('added_by_user_id', sa.String(), nullable=False),
+        sa.PrimaryKeyConstraint('id'),
+        sa.ForeignKeyConstraint(['project_id'], ['projects.id'], ondelete='CASCADE'),
+        sa.UniqueConstraint('project_id', 'user_id', name='uq_project_user')
+    )
+    
+    # Create indexes for efficient queries
+    op.create_index(
+        'ix_project_users_project_id',
+        'project_users',
+        ['project_id']
+    )
+    
+    op.create_index(
+        'ix_project_users_user_id',
+        'project_users',
+        ['user_id']
+    )
+    
+    # Create composite index for common query pattern (user accessing specific project)
+    op.create_index(
+        'ix_project_users_user_project',
+        'project_users',
+        ['user_id', 'project_id']
+    )
+
+
+def downgrade() -> None:
+    """
+    Remove project_users table and related indexes.
+    """
+    # Drop indexes first
+    op.drop_index('ix_project_users_user_project', table_name='project_users')
+    op.drop_index('ix_project_users_user_id', table_name='project_users')
+    op.drop_index('ix_project_users_project_id', table_name='project_users')
+    
+    # Drop the table
+    op.drop_table('project_users')

solace_agent_mesh/gateway/http_sse/alembic/versions/20251023_add_soft_delete_and_search.py

@@ -0,0 +1,109 @@
+"""Add soft delete and search support for sessions and projects
+
+Revision ID: add_soft_delete_search_001
+Revises: remove_is_global_001
+Create Date: 2025-10-23 16:20:00.000000
+
+"""
+from typing import Sequence, Union
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy import inspect
+
+# revision identifiers, used by Alembic.
+revision: str = 'soft_del_search_001'
+down_revision: Union[str, Sequence[str], None] = 'add_project_users_001'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Add soft delete columns and search indexes."""
+    bind = op.get_bind()
+    inspector = inspect(bind)
+    
+    # Add soft delete columns to sessions table
+    if 'sessions' in inspector.get_table_names():
+        sessions_columns = [col['name'] for col in inspector.get_columns('sessions')]
+        
+        if 'deleted_at' not in sessions_columns:
+            op.add_column('sessions', sa.Column('deleted_at', sa.BigInteger(), nullable=True))
+        
+        if 'deleted_by' not in sessions_columns:
+            op.add_column('sessions', sa.Column('deleted_by', sa.String(), nullable=True))
+        
+        # Create index on deleted_at for efficient filtering
+        try:
+            op.create_index('ix_sessions_deleted_at', 'sessions', ['deleted_at'])
+        except Exception:
+            pass  # Index might already exist
+        
+        # Create composite index for user queries with soft delete
+        try:
+            op.create_index('ix_sessions_user_deleted', 'sessions', ['user_id', 'deleted_at'])
+        except Exception:
+            pass
+    
+    # Add soft delete columns to projects table
+    if 'projects' in inspector.get_table_names():
+        projects_columns = [col['name'] for col in inspector.get_columns('projects')]
+        
+        if 'deleted_at' not in projects_columns:
+            op.add_column('projects', sa.Column('deleted_at', sa.BigInteger(), nullable=True))
+        
+        if 'deleted_by' not in projects_columns:
+            op.add_column('projects', sa.Column('deleted_by', sa.String(), nullable=True))
+        
+        # Create index on deleted_at for efficient filtering
+        try:
+            op.create_index('ix_projects_deleted_at', 'projects', ['deleted_at'])
+        except Exception:
+            pass
+        
+        # Create composite index for user queries with soft delete
+        try:
+            op.create_index('ix_projects_user_deleted', 'projects', ['user_id', 'deleted_at'])
+        except Exception:
+            pass
+
+
+def downgrade() -> None:
+    """Remove soft delete columns and search indexes."""
+    bind = op.get_bind()
+    inspector = inspect(bind)
+
+    # Drop soft delete indexes from sessions
+    if 'sessions' in inspector.get_table_names():
+        try:
+            op.drop_index('ix_sessions_user_deleted', table_name='sessions')
+        except Exception:
+            pass
+        
+        try:
+            op.drop_index('ix_sessions_deleted_at', table_name='sessions')
+        except Exception:
+            pass
+        
+        sessions_columns = [col['name'] for col in inspector.get_columns('sessions')]
+        if 'deleted_by' in sessions_columns:
+            op.drop_column('sessions', 'deleted_by')
+        if 'deleted_at' in sessions_columns:
+            op.drop_column('sessions', 'deleted_at')
+    
+    # Drop indexes and columns from projects
+    if 'projects' in inspector.get_table_names():
+        try:
+            op.drop_index('ix_projects_user_deleted', table_name='projects')
+        except Exception:
+            pass
+        
+        try:
+            op.drop_index('ix_projects_deleted_at', table_name='projects')
+        except Exception:
+            pass
+        
+        projects_columns = [col['name'] for col in inspector.get_columns('projects')]
+        if 'deleted_by' in projects_columns:
+            op.drop_column('projects', 'deleted_by')
+        if 'deleted_at' in projects_columns:
+            op.drop_column('projects', 'deleted_at')

solace_agent_mesh/gateway/http_sse/alembic/versions/20251024_add_default_agent_to_projects.py

@@ -0,0 +1,26 @@
+"""add default agent to projects
+
+Revision ID: default_agent_001
+Revises: soft_del_search_001
+Create Date: 2025-01-24 01:13:00.000000
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = 'default_agent_001'
+down_revision = 'soft_del_search_001'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    """Add default_agent_id column to projects table."""
+    op.add_column('projects', sa.Column('default_agent_id', sa.String(), nullable=True))
+
+
+def downgrade():
+    """Remove default_agent_id column from projects table."""
+    op.drop_column('projects', 'default_agent_id')

solace_agent_mesh/gateway/http_sse/alembic/versions/20251024_add_projects_table.py

@@ -0,0 +1,135 @@
+"""Add projects table and project_id to sessions
+
+Revision ID: add_projects_table_001
+Revises: 20251015_session_idx
+Create Date: 2025-10-24 10:00:00.000000
+
+"""
+from typing import Sequence, Union
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy import inspect
+
+# revision identifiers, used by Alembic.
+revision: str = 'add_projects_table_001'
+down_revision: Union[str, Sequence[str], None] = '20251015_session_idx'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Create projects table and add project_id to sessions."""
+    bind = op.get_bind()
+    inspector = inspect(bind)
+    existing_tables = inspector.get_table_names()
+    dialect_name = bind.dialect.name
+
+    # Create projects table if it doesn't exist
+    # Note: This table is created without is_global and created_by_user_id columns
+    # as those were removed in the squashed migration
+    if 'projects' not in existing_tables:
+        op.create_table('projects',
+            sa.Column('id', sa.String(), nullable=False),
+            sa.Column('name', sa.String(), nullable=False),
+            sa.Column('user_id', sa.String(), nullable=False),
+            sa.Column('description', sa.Text(), nullable=True),
+            sa.Column('system_prompt', sa.Text(), nullable=True),
+            sa.Column('created_at', sa.BigInteger(), nullable=False),
+            sa.Column('updated_at', sa.BigInteger(), nullable=True),
+            sa.PrimaryKeyConstraint('id')
+        )
+
+    # Add project_id column to sessions if it doesn't exist
+    sessions_columns = [col['name'] for col in inspector.get_columns('sessions')]
+    if 'project_id' not in sessions_columns:
+        if dialect_name == 'sqlite':
+            # SQLite doesn't support ALTER TABLE ADD CONSTRAINT, recreate table
+            op.create_table(
+                'sessions_new',
+                sa.Column('id', sa.String(), nullable=False),
+                sa.Column('name', sa.String(), nullable=True),
+                sa.Column('user_id', sa.String(), nullable=False),
+                sa.Column('agent_id', sa.String(), nullable=True),
+                sa.Column('created_time', sa.BigInteger(), nullable=False),
+                sa.Column('updated_time', sa.BigInteger(), nullable=False),
+                sa.Column('project_id', sa.String(), nullable=True),
+                sa.ForeignKeyConstraint(['project_id'], ['projects.id']),
+                sa.PrimaryKeyConstraint('id')
+            )
+
+            # Copy data from old table
+            op.execute("""
+                INSERT INTO sessions_new (id, name, user_id, agent_id, created_time, updated_time)
+                SELECT id, name, user_id, agent_id, created_time, updated_time
+                FROM sessions
+            """)
+
+            # Drop old table
+            op.drop_table('sessions')
+
+            # Rename new table
+            op.rename_table('sessions_new', 'sessions')
+
+            # Recreate indexes
+            op.create_index('ix_sessions_user_id', 'sessions', ['user_id'])
+            op.create_index('ix_sessions_project_id', 'sessions', ['project_id'])
+        else:
+            # PostgreSQL, MySQL - standard ALTER TABLE
+            op.add_column('sessions', sa.Column('project_id', sa.String(), nullable=True))
+            op.create_index('ix_sessions_project_id', 'sessions', ['project_id'])
+            op.create_foreign_key(
+                'fk_sessions_project_id',
+                'sessions',
+                'projects',
+                ['project_id'],
+                ['id']
+            )
+
+
+def downgrade() -> None:
+    """Downgrade schema - removes project-related changes."""
+    bind = op.get_bind()
+    inspector = inspect(bind)
+    dialect_name = bind.dialect.name
+
+    # Drop project_id column from sessions if it exists
+    sessions_columns = [col['name'] for col in inspector.get_columns('sessions')]
+    if 'project_id' in sessions_columns:
+        if dialect_name == 'sqlite':
+            # SQLite doesn't support ALTER TABLE DROP CONSTRAINT, recreate table
+            op.create_table(
+                'sessions_old',
+                sa.Column('id', sa.String(), nullable=False),
+                sa.Column('name', sa.String(), nullable=True),
+                sa.Column('user_id', sa.String(), nullable=False),
+                sa.Column('agent_id', sa.String(), nullable=True),
+                sa.Column('created_time', sa.BigInteger(), nullable=False),
+                sa.Column('updated_time', sa.BigInteger(), nullable=False),
+                sa.PrimaryKeyConstraint('id')
+            )
+
+            # Copy data from current table (excluding project_id)
+            op.execute("""
+                INSERT INTO sessions_old (id, name, user_id, agent_id, created_time, updated_time)
+                SELECT id, name, user_id, agent_id, created_time, updated_time
+                FROM sessions
+            """)
+
+            # Drop current table
+            op.drop_table('sessions')
+
+            # Rename old table back
+            op.rename_table('sessions_old', 'sessions')
+
+            # Recreate index on user_id
+            op.create_index('ix_sessions_user_id', 'sessions', ['user_id'])
+        else:
+            # PostgreSQL, MySQL - standard ALTER TABLE
+            op.drop_constraint('fk_sessions_project_id', 'sessions', type_='foreignkey')
+            op.drop_index('ix_sessions_project_id', 'sessions')
+            op.drop_column('sessions', 'project_id')
+
+    # Drop projects table if it exists
+    existing_tables = inspector.get_table_names()
+    if 'projects' in existing_tables:
+        op.drop_table('projects')

solace_agent_mesh/gateway/http_sse/alembic/versions/20251108_create_prompt_tables_with_sharing.py

@@ -0,0 +1,154 @@
+"""Create prompt tables with role-based sharing support
+
+Revision ID: 20251108_prompt_tables_complete
+Revises: 20251103_merge_heads
+Create Date: 2025-11-08
+
+This is a squashed migration that combines:
+- Initial prompt tables creation
+- Prompt pinning feature
+- Role-based sharing via prompt_group_users table
+"""
+
+from collections.abc import Sequence
+import sqlalchemy as sa
+from alembic import op
+
+revision: str = "20251108_prompt_tables_complete"
+down_revision: str | Sequence[str] | None = "default_agent_001"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    """Create complete prompt schema with sharing support."""
+    
+    # 1. Create prompts table (individual prompt versions)
+    op.create_table(
+        'prompts',
+        sa.Column('id', sa.String(), nullable=False),
+        sa.Column('prompt_text', sa.Text(), nullable=False),
+        sa.Column('group_id', sa.String(), nullable=False),
+        sa.Column('user_id', sa.String(), nullable=False),
+        sa.Column('version', sa.Integer(), nullable=False, server_default='1'),
+        sa.Column('created_at', sa.BigInteger(), nullable=False),
+        sa.Column('updated_at', sa.BigInteger(), nullable=False),
+        sa.PrimaryKeyConstraint('id')
+    )
+    
+    # Indexes for prompts
+    op.create_index('ix_prompts_group_id', 'prompts', ['group_id'], unique=False)
+    op.create_index('ix_prompts_user_id', 'prompts', ['user_id'], unique=False)
+    
+    # 2. Create prompt_groups table (prompt collections)
+    op.create_table(
+        'prompt_groups',
+        sa.Column('id', sa.String(), nullable=False),
+        sa.Column('name', sa.String(length=255), nullable=False),
+        sa.Column('description', sa.Text(), nullable=True),
+        sa.Column('category', sa.String(length=100), nullable=True),
+        sa.Column('command', sa.String(length=50), nullable=True),
+        sa.Column('user_id', sa.String(), nullable=False),
+        sa.Column('author_name', sa.String(length=255), nullable=True),
+        sa.Column('production_prompt_id', sa.String(), nullable=True),
+        sa.Column('is_shared', sa.Boolean(), nullable=False, server_default='false'),
+        sa.Column('is_pinned', sa.Boolean(), nullable=False, server_default='false'),
+        sa.Column('created_at', sa.BigInteger(), nullable=False),
+        sa.Column('updated_at', sa.BigInteger(), nullable=False),
+        sa.ForeignKeyConstraint(
+            ['production_prompt_id'],
+            ['prompts.id'],
+            ondelete='SET NULL'
+        ),
+        sa.PrimaryKeyConstraint('id')
+    )
+    
+    # Indexes for prompt_groups
+    op.create_index('ix_prompt_groups_name', 'prompt_groups', ['name'], unique=False)
+    op.create_index('ix_prompt_groups_category', 'prompt_groups', ['category'], unique=False)
+    op.create_index('ix_prompt_groups_command', 'prompt_groups', ['command'], unique=True)
+    op.create_index('ix_prompt_groups_user_id', 'prompt_groups', ['user_id'], unique=False)
+    op.create_index('ix_prompt_groups_is_pinned', 'prompt_groups', ['is_pinned'], unique=False)
+    
+    # 3. Add foreign key from prompts to prompt_groups
+    with op.batch_alter_table('prompts', schema=None) as batch_op:
+        batch_op.create_foreign_key(
+            'fk_prompts_group_id',
+            'prompt_groups',
+            ['group_id'],
+            ['id'],
+            ondelete='CASCADE'
+        )
+    
+    # 4. Create prompt_group_users table (role-based sharing)
+    op.create_table(
+        'prompt_group_users',
+        sa.Column('id', sa.String(), nullable=False),
+        sa.Column('prompt_group_id', sa.String(), nullable=False),
+        sa.Column('user_id', sa.String(), nullable=False),
+        sa.Column('role', sa.String(), nullable=False, server_default='viewer'),
+        sa.Column('added_at', sa.BigInteger(), nullable=False),
+        sa.Column('added_by_user_id', sa.String(), nullable=False),
+        sa.PrimaryKeyConstraint('id'),
+        sa.ForeignKeyConstraint(
+            ['prompt_group_id'], 
+            ['prompt_groups.id'], 
+            ondelete='CASCADE'
+        ),
+        sa.UniqueConstraint('prompt_group_id', 'user_id', name='uq_prompt_group_user')
+    )
+    
+    # Indexes for prompt_group_users (optimized for sharing queries)
+    op.create_index(
+        'ix_prompt_group_users_prompt_group_id',
+        'prompt_group_users',
+        ['prompt_group_id'],
+        unique=False
+    )
+    
+    op.create_index(
+        'ix_prompt_group_users_user_id',
+        'prompt_group_users',
+        ['user_id'],
+        unique=False
+    )
+    
+    # Composite index for fast access checks
+    op.create_index(
+        'ix_prompt_group_users_user_prompt',
+        'prompt_group_users',
+        ['user_id', 'prompt_group_id'],
+        unique=False
+    )
+    
+    # Optional: Index for role-based queries
+    op.create_index(
+        'ix_prompt_group_users_user_role',
+        'prompt_group_users',
+        ['user_id', 'role'],
+        unique=False
+    )
+
+
+def downgrade() -> None:
+    """Remove all prompt-related tables."""
+    
+    # Drop prompt_group_users table and indexes
+    op.drop_index('ix_prompt_group_users_user_role', table_name='prompt_group_users')
+    op.drop_index('ix_prompt_group_users_user_prompt', table_name='prompt_group_users')
+    op.drop_index('ix_prompt_group_users_user_id', table_name='prompt_group_users')
+    op.drop_index('ix_prompt_group_users_prompt_group_id', table_name='prompt_group_users')
+    op.drop_table('prompt_group_users')
+    
+    # Drop prompt_groups indexes and table
+    op.drop_index('ix_prompt_groups_is_pinned', table_name='prompt_groups')
+    op.drop_index('ix_prompt_groups_user_id', table_name='prompt_groups')
+    op.drop_index('ix_prompt_groups_command', table_name='prompt_groups')
+    op.drop_index('ix_prompt_groups_category', table_name='prompt_groups')
+    op.drop_index('ix_prompt_groups_name', table_name='prompt_groups')
+    op.drop_table('prompt_groups')
+    
+    # Drop prompts indexes and table
+    op.drop_index('ix_prompts_user_id', table_name='prompts')
+    op.drop_index('ix_prompts_group_id', table_name='prompts')
+    op.drop_table('prompts')

solace_agent_mesh/gateway/http_sse/alembic/versions/20251115_add_parent_task_id.py

@@ -0,0 +1,32 @@
+"""Add parent_task_id to tasks table
+
+Revision ID: 20251115_add_parent_task_id
+Revises: 20251108_prompt_tables_complete
+Create Date: 2025-11-15
+
+"""
+
+from collections.abc import Sequence
+import sqlalchemy as sa
+from alembic import op
+
+revision: str = "20251115_add_parent_task_id"
+down_revision: str | Sequence[str] | None = "20251108_prompt_tables_complete"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    """Add parent_task_id column to tasks table for efficient task hierarchy queries."""
+    op.add_column(
+        "tasks", sa.Column("parent_task_id", sa.String(), nullable=True)
+    )
+    op.create_index(
+        "ix_tasks_parent_task_id", "tasks", ["parent_task_id"], unique=False
+    )
+
+
+def downgrade() -> None:
+    """Remove parent_task_id column from tasks table."""
+    op.drop_index("ix_tasks_parent_task_id", table_name="tasks")
+    op.drop_column("tasks", "parent_task_id")

solace_agent_mesh/gateway/http_sse/alembic/versions/20251126_add_background_task_fields.py

@@ -0,0 +1,47 @@
+"""Add background task execution fields
+
+Revision ID: 20251126_background_tasks
+Revises: 20251202_versioned_prompt_fields
+Create Date: 2025-11-26 17:30:00.000000
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '20251126_background_tasks'
+down_revision = '20251202_versioned_prompt_fields'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    """Add fields to support background task execution."""
+    # Add execution_mode column (foreground | background)
+    op.add_column('tasks', sa.Column('execution_mode', sa.String(20), nullable=True, server_default='foreground'))
+    
+    # Add last_activity_time for timeout detection
+    op.add_column('tasks', sa.Column('last_activity_time', sa.BigInteger, nullable=True))
+    
+    # Add background_execution_enabled flag
+    op.add_column('tasks', sa.Column('background_execution_enabled', sa.Boolean, nullable=True, server_default='false'))
+    
+    # Add max_execution_time_ms for timeout configuration
+    op.add_column('tasks', sa.Column('max_execution_time_ms', sa.BigInteger, nullable=True))
+    
+    # Create index on execution_mode for efficient queries
+    op.create_index('idx_tasks_execution_mode', 'tasks', ['execution_mode'])
+    
+    # Create index on last_activity_time for timeout monitoring
+    op.create_index('idx_tasks_last_activity', 'tasks', ['last_activity_time'])
+
+
+def downgrade():
+    """Remove background task execution fields."""
+    op.drop_index('idx_tasks_last_activity', table_name='tasks')
+    op.drop_index('idx_tasks_execution_mode', table_name='tasks')
+    op.drop_column('tasks', 'max_execution_time_ms')
+    op.drop_column('tasks', 'background_execution_enabled')
+    op.drop_column('tasks', 'last_activity_time')
+    op.drop_column('tasks', 'execution_mode')

solace_agent_mesh/gateway/http_sse/alembic/versions/20251202_add_versioned_fields_to_prompts.py

@@ -0,0 +1,52 @@
+"""Add versioned metadata fields to prompts table
+
+Revision ID: 20251202_versioned_prompt_fields
+Revises: 20251115_add_parent_task_id
+Create Date: 2025-12-02
+
+This migration adds name, description, category, and command fields to the prompts table
+to enable full versioning of all prompt metadata, not just the prompt_text content.
+"""
+
+from collections.abc import Sequence
+import sqlalchemy as sa
+from alembic import op
+
+revision: str = "20251202_versioned_prompt_fields"
+down_revision: str | Sequence[str] | None = "20251115_add_parent_task_id"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    """Add versioned metadata fields to prompts table."""
+    
+    # Add new columns to prompts table
+    with op.batch_alter_table('prompts', schema=None) as batch_op:
+        batch_op.add_column(sa.Column('name', sa.String(length=255), nullable=True))
+        batch_op.add_column(sa.Column('description', sa.Text(), nullable=True))
+        batch_op.add_column(sa.Column('category', sa.String(length=100), nullable=True))
+        batch_op.add_column(sa.Column('command', sa.String(length=50), nullable=True))
+    
+    # Migrate existing data: copy metadata from prompt_groups to prompts
+    # This ensures existing prompt versions have the metadata from their group
+    connection = op.get_bind()
+    connection.execute(sa.text("""
+        UPDATE prompts 
+        SET name = pg.name,
+            description = pg.description,
+            category = pg.category,
+            command = pg.command
+        FROM prompt_groups pg
+        WHERE prompts.group_id = pg.id
+    """))
+
+
+def downgrade() -> None:
+    """Remove versioned metadata fields from prompts table."""
+    
+    with op.batch_alter_table('prompts', schema=None) as batch_op:
+        batch_op.drop_column('command')
+        batch_op.drop_column('category')
+        batch_op.drop_column('description')
+        batch_op.drop_column('name')