opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-09-05 00:31:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(migration): avoid backfilling Tokener billing profiles

Browse Source
This commit is contained in:
Yansong Zhang 2026-09-03 17:29:14 +08:00
parent bac4ddd0f8
commit 9f5fdfccec
1 changed files with 0 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
api/migrations/versions/2026_09_03_1000-d4e5f6a7b8c9_add_tenant_model_billing_profiles.py
View File

@ -37,45 +37,6 @@ def upgrade():
),
sa.PrimaryKeyConstraint("tenant_id", name=op.f("tenant_model_billing_profile_pkey")),
)
bind = op.get_bind()
if bind.dialect.name == "postgresql":
op.execute(
sa.text(
"""
INSERT INTO tenant_model_billing_profiles (tenant_id, model_billing_source)
SELECT tenant_id, 'tokener'
FROM tenant_tokener_integrations
ON CONFLICT (tenant_id) DO UPDATE
SET model_billing_source = EXCLUDED.model_billing_source
"""
)
)
elif bind.dialect.name in {"mysql", "mariadb"}:
op.execute(
sa.text(
"""
INSERT INTO tenant_model_billing_profiles (tenant_id, model_billing_source)
SELECT tenant_id, 'tokener'
FROM tenant_tokener_integrations
ON DUPLICATE KEY UPDATE model_billing_source = VALUES(model_billing_source)
"""
)
)
else:
op.execute(
sa.text(
"""
INSERT INTO tenant_model_billing_profiles (tenant_id, model_billing_source)
SELECT integration.tenant_id, 'tokener'
FROM tenant_tokener_integrations AS integration
WHERE NOT EXISTS (
SELECT 1
FROM tenant_model_billing_profiles AS profile
WHERE profile.tenant_id = integration.tenant_id
)
"""
)
)
def downgrade():