018012973d35_add_indexes.py 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. """Add indexes
  2. Revision ID: 018012973d35
  3. Revises: d31026856c01
  4. Create Date: 2025-08-13 03:00:00.000000
  5. """
  6. from alembic import op
  7. import sqlalchemy as sa
  8. revision = "018012973d35"
  9. down_revision = "d31026856c01"
  10. branch_labels = None
  11. depends_on = None
  12. def upgrade():
  13. # Chat table indexes
  14. op.create_index("folder_id_idx", "chat", ["folder_id"])
  15. op.create_index("user_id_pinned_idx", "chat", ["user_id", "pinned"])
  16. op.create_index("user_id_archived_idx", "chat", ["user_id", "archived"])
  17. op.create_index("updated_at_user_id_idx", "chat", ["updated_at", "user_id"])
  18. op.create_index("folder_id_user_id_idx", "chat", ["folder_id", "user_id"])
  19. # Tag table index
  20. op.create_index("user_id_idx", "tag", ["user_id"])
  21. # Function table index
  22. op.create_index("is_global_idx", "function", ["is_global"])
  23. def downgrade():
  24. # Chat table indexes
  25. op.drop_index("folder_id_idx", table_name="chat")
  26. op.drop_index("user_id_pinned_idx", table_name="chat")
  27. op.drop_index("user_id_archived_idx", table_name="chat")
  28. op.drop_index("updated_at_user_id_idx", table_name="chat")
  29. op.drop_index("folder_id_user_id_idx", table_name="chat")
  30. # Tag table index
  31. op.drop_index("user_id_idx", table_name="tag")
  32. # Function table index
  33. op.drop_index("is_global_idx", table_name="function")