38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
"""修复：按种子定义给 Agent 设置 kb_ids（库名字 → id）"""
|
||
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, KnowledgeBase
|
||
|
||
# 与 seed.py AGENTS 定义一致
|
||
AGENT_KBS = {
|
||
"主控 Agent": [],
|
||
"开场段 Agent": ["风格档案库"],
|
||
"心法段 Agent": ["风格档案库", "训练营框架库"],
|
||
"技法段 Agent": ["风格档案库", "直播原文库"],
|
||
"产品销售 Agent": ["产品知识库", "风格档案库"],
|
||
"收尾段 Agent": ["风格档案库"],
|
||
"钩子段 Agent": ["训练营框架库"],
|
||
"结构质检": ["质检规则库"],
|
||
"内容质检": ["质检规则库", "产品知识库"],
|
||
"风格尺度质检": ["质检规则库"],
|
||
"跨稿质检": ["备稿库", "训练营框架库"],
|
||
"组合 Agent": [],
|
||
"风格顾问": ["风格档案库"],
|
||
"产品顾问": ["产品知识库"],
|
||
}
|
||
|
||
with Session(engine) as s:
|
||
kbs = {k.name: k.id for k in s.exec(select(KnowledgeBase)).all()}
|
||
print("现有库:", kbs)
|
||
for a in s.exec(select(Agent)).all():
|
||
names = AGENT_KBS.get(a.name, [])
|
||
ids = [kbs[n] for n in names if n in kbs]
|
||
missing = [n for n in names if n not in kbs]
|
||
a.kb_ids = ids
|
||
s.add(a)
|
||
print(f"{a.name}: {ids}{'（缺库:' + str(missing) + '）' if missing else ''}")
|
||
s.commit()
|
||
print("完成")
|