i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-09-01 17:00:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(migration): backfill 前清空 user_model_usage_counts 确保幂等性

Browse Source
This commit is contained in:
fawney19
2026-02-28 15:28:34 +08:00
parent 4fdebfc78e
commit 27047e5880
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
alembic/versions/20260228_1100_b2c3d4e5f6a7_cleanup_capability_settings.py
View File

@@ -169,7 +169,8 @@ def upgrade() -> None:
if not index_exists("user_model_usage_counts", "idx_user_model_usage_model"): if not index_exists("user_model_usage_counts", "idx_user_model_usage_model"):
op.create_index("idx_user_model_usage_model", "user_model_usage_counts", ["model"]) op.create_index("idx_user_model_usage_model", "user_model_usage_counts", ["model"])
# Backfill from existing usage records # Backfill from existing usage records (truncate first for idempotency)
conn.execute(sa.text("DELETE FROM user_model_usage_counts"))
rows = conn.execute( rows = conn.execute(
sa.text( sa.text(
"SELECT user_id, model, COUNT(*) AS cnt FROM usage" "SELECT user_id, model, COUNT(*) AS cnt FROM usage"