"""刷新全部 Agent rules（种子模板更新后重写）"""
import sys
sys.path.insert(0, r'C:\Users\xinmiclaw\Desktop\雅文内容创造工作台\server')
from sqlmodel import Session, select
from app.db import engine
from app.models import Agent
from app.seed import RULES, AGENTS

with Session(engine) as s:
    for name, role, tags, kb_names, rules_key in AGENTS:
        a = s.exec(select(Agent).where(Agent.name == name)).first()
        if a:
            a.rules = RULES[rules_key].format(name=name) if "{name}" in RULES[rules_key] else RULES[rules_key]
            s.add(a)
    s.commit()
    print("全部 Agent rules 已刷新（含颜色标记规范）")