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

fix(migration): body_rules 回填限定 codex 提供商类型

Browse Source 
原迁移仅按 api_format 过滤，会误回填所有 openai:cli 端点;
现 JOIN providers 表增加 provider_type = 'codex' 条件。
This commit is contained in:
fawney19
2026-03-06 23:44:37 +08:00
parent 90760da499
commit 1f3693d3a2
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
alembic/versions/20260302_1500_dd0278c0a28c_backfill_codex_default_body_rules.py
View File

@@ -1,6 +1,6 @@
"""backfill_codex_default_body_rules """backfill_codex_default_body_rules
Backfill default body_rules for openai:cli endpoints Backfill default body_rules for codex providers with openai:cli endpoints
that currently have body_rules IS NULL. that currently have body_rules IS NULL.
Rules: Rules:
@@ -48,17 +48,20 @@ _DEFAULT_BODY_RULES = [
def upgrade() -> None: def upgrade() -> None:
conn = op.get_bind() conn = op.get_bind()
# 幂等性: 仅回填 body_rules 为空（SQL NULL 或 JSON null）的记录 # 幂等性: 仅回填 codex 提供商中 body_rules 为空（SQL NULL 或 JSON null）的记录
rules_json = json.dumps(_DEFAULT_BODY_RULES, ensure_ascii=False) rules_json = json.dumps(_DEFAULT_BODY_RULES, ensure_ascii=False)
result = conn.execute( result = conn.execute(
sa.text(""" sa.text("""
UPDATE provider_endpoints UPDATE provider_endpoints pe
SET body_rules = CAST(:rules AS json), SET body_rules = CAST(:rules AS json),
updated_at = CURRENT_TIMESTAMP updated_at = CURRENT_TIMESTAMP
WHERE api_format = :fmt FROM providers p
AND (body_rules IS NULL OR body_rules::text = 'null') WHERE pe.provider_id = p.id
AND p.provider_type = :ptype
AND pe.api_format = :fmt
AND (pe.body_rules IS NULL OR pe.body_rules::text = 'null')
"""), """),
{"rules": rules_json, "fmt": _TARGET_FORMATS[0]}, {"rules": rules_json, "ptype": "codex", "fmt": _TARGET_FORMATS[0]},
) )
if result.rowcount: if result.rowcount:
print(f" backfilled body_rules for {result.rowcount} endpoint(s)") print(f" backfilled body_rules for {result.rowcount} endpoint(s)")