52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
|
"""修复 code-build-context 中列表推导式 r 与前面 genexp 的 r 作用域冲突（就地更新数据库 Agent）。"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
BACKEND = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|||
|
|
sys.path.insert(0, BACKEND)
|
|||
|
|
|
|||
|
|
from sqlalchemy.orm.attributes import flag_modified
|
|||
|
|
|
|||
|
|
from app.core.database import SessionLocal
|
|||
|
|
from app.models.agent import Agent
|
|||
|
|
|
|||
|
|
OLD_VEC = "vec_str = '\\n'.join((r.get('text') or r.get('content') or '') for r in right)"
|
|||
|
|
NEW_VEC = "vec_str = '\\n'.join((rec.get('text') or rec.get('content') or '') for rec in right)"
|
|||
|
|
OLD_KW = 'kw_lines = [f"{r}: {t}" for _, r, t in scored[:6]]'
|
|||
|
|
NEW_KW = 'kw_lines = [f"{role}: {text}" for _, role, text in scored[:6]]'
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main() -> int:
|
|||
|
|
name = os.environ.get("PATCH_AGENT_NAME", "知你客服11号")
|
|||
|
|
db = SessionLocal()
|
|||
|
|
try:
|
|||
|
|
a = db.query(Agent).filter(Agent.name == name).first()
|
|||
|
|
if not a:
|
|||
|
|
print("未找到", name, file=sys.stderr)
|
|||
|
|
return 1
|
|||
|
|
wf = a.workflow_config
|
|||
|
|
for n in wf.get("nodes", []):
|
|||
|
|
if n.get("id") != "code-build-context":
|
|||
|
|
continue
|
|||
|
|
c = n.get("data", {}).get("code", "")
|
|||
|
|
c2 = c.replace(OLD_VEC, NEW_VEC).replace(OLD_KW, NEW_KW)
|
|||
|
|
if c2 == c:
|
|||
|
|
print("无需替换（可能已修复或内容不同）")
|
|||
|
|
return 0
|
|||
|
|
n.setdefault("data", {})["code"] = c2
|
|||
|
|
a.workflow_config = wf
|
|||
|
|
flag_modified(a, "workflow_config")
|
|||
|
|
db.commit()
|
|||
|
|
print("已更新", name, "code-build-context")
|
|||
|
|
return 0
|
|||
|
|
print("未找到 code-build-context 节点", file=sys.stderr)
|
|||
|
|
return 1
|
|||
|
|
finally:
|
|||
|
|
db.close()
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
raise SystemExit(main())
|