"""执行visibility字段迁移"""
import sys
sys.path.insert(0, '.')

from sqlalchemy import text
from app.database import engine

def run_migration():
    with engine.connect() as conn:
        conn.execute(text("""
            ALTER TABLE aigcspace.models 
            ADD COLUMN IF NOT EXISTS visibility VARCHAR(20) DEFAULT 'user'
        """))
        conn.commit()
        print("Migration completed: visibility column added")

if __name__ == "__main__":
    run_migration()