server.py 71 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447
  1. # -*- coding: utf-8 -*-
  2. """搜索评估案例查看 server。
  3. 沿用 图文排版搜索评估.html 的版式(卡片 + dialog 详情 + rubric 评分条),
  4. 数据实时扫描 runs_full/*/form_*.json —— runs_full 下每新增一个 q 文件夹,刷新即出现。
  5. 分页:query → 三种形式(A/B/C) → 三个渠道 三行从上到下。
  6. 用法:python server.py [port] 默认 8770,浏览器开 http://0.0.0.0:8770
  7. """
  8. import json, re, glob, sys, pathlib, subprocess, threading
  9. from datetime import datetime
  10. from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
  11. from urllib.parse import urlparse, parse_qs
  12. try: # Windows 控制台默认 cp1252,中文 print 会崩,统一切 utf-8
  13. sys.stdout.reconfigure(encoding="utf-8")
  14. except Exception:
  15. pass
  16. HERE = pathlib.Path(__file__).parent
  17. SE = HERE.parent # search_eval:引擎脚本/procedure-dsl/evaluation 都在父目录
  18. sys.path.insert(0, str(HERE))
  19. sys.path.insert(0, str(SE)) # 复用兄弟模块 batch_extract_procedures / build_workflows
  20. import argparse
  21. _parser = argparse.ArgumentParser(description="工序解构 server (mode_procedure:写死 new 模式 + procedure 直写版)")
  22. _parser.add_argument("port", type=int, nargs="?", default=8771, help="运行端口(默认 8771)")
  23. _parser.add_argument("--mode", choices=["legacy", "new"], default="new", help="(mode_procedure 固定 new)")
  24. _args, _ = _parser.parse_known_args()
  25. PORT = _args.port
  26. MODE = "new" # mode_procedure 写死 new 模式
  27. RUNS_DIR_NAME = "runs_new"
  28. RUNS_DIR = HERE / RUNS_DIR_NAME # 产出目录在本文件夹 mode_procedure/runs_new
  29. PLAT = {"xhs": "小红书", "gzh": "公众号", "zhihu": "知乎", "x": "X", "bili": "B站", "douyin": "抖音",
  30. "sph": "视频号", "youtube": "YouTube", "github": "GitHub", "toutiao": "头条", "weibo": "微博"}
  31. KT = {"procedure": "工序", "step": "步骤", "tool": "工具"}
  32. # 从 taxonomy 取动作叶子/类型名,用于把 original_q 解析回原始维度(动作×类型 正交)
  33. # 路径优先级:search_eval/evaluation/(主源,IDE 编辑那份就是 runtime 实际读的)
  34. # → test_script/evaluation/(历史副本兜底)→ script/evaluation/(更老兜底)
  35. # 谁也找不到时整目录扫空,server 仍能起。
  36. EVALDIR = HERE / "evaluation"
  37. if not EVALDIR.exists():
  38. EVALDIR = HERE.parent.parent / "test_script" / "evaluation"
  39. if not EVALDIR.exists():
  40. EVALDIR = HERE.parent / "evaluation"
  41. try:
  42. _jm = json.load(open(EVALDIR / "judged_matrix.json", encoding="utf-8"))
  43. ACT_L1 = {a["name"]: a["l1"] for a in _jm["actions"]}
  44. ACTION_SET = set(ACT_L1)
  45. TYPE_SET = {t["name"] for t in _jm["types"]}
  46. ACTIONS_TAX = [{"name": a["name"], "l1": a["l1"], "l2": a.get("l2", "")} for a in _jm["actions"]]
  47. TYPES_TAX = [{"name": t["name"], "l1": t["l1"]} for t in _jm["types"]]
  48. # taxonomy 顺序沿用 judged_matrix(严格版);矩阵分值改用 type_action_scores(宽松版) —
  49. # 两份是同一组 27×50 cell 的独立 gemini judging,前者只 53 格到 tier3,后者 156 格到 score3
  50. _tas = json.load(open(EVALDIR / "type_action_scores.json", encoding="utf-8"))["scores"]
  51. _MATRIX = []
  52. for a in _jm["actions"]:
  53. row = []
  54. for t in _jm["types"]:
  55. rec = _tas.get(t["name"], {}).get(a["name"])
  56. row.append({"tier": rec["score"], "r": rec.get("reason", "")} if rec else {})
  57. _MATRIX.append(row)
  58. except Exception:
  59. ACT_L1, ACTION_SET, TYPE_SET, ACTIONS_TAX, TYPES_TAX, _MATRIX = {}, set(), set(), [], [], []
  60. ACTIVE_TASKS = {}
  61. ACTIVE_REEVALS = {}
  62. ACTIVE_SEARCH_EVALS = {}
  63. ACTIVE_BATCH_TASKS = {}
  64. def _gen_version() -> str:
  65. """生成版本号 v_月日时分。"""
  66. return "v_" + datetime.now().strftime("%m%d%H%M")
  67. def _read_meta_version(out_dir: pathlib.Path):
  68. """读 out_dir/_meta.json 里的版本号(用于把旧产物归档到 history/{旧版本}/)。"""
  69. try:
  70. meta = json.loads((out_dir / "_meta.json").read_text(encoding="utf-8"))
  71. return meta.get("version")
  72. except Exception:
  73. return None
  74. def _stamp_version(out_dir: pathlib.Path, version: str):
  75. """把版本号写进 out_dir/_meta.json(_write_meta 之后调用)。"""
  76. meta_path = out_dir / "_meta.json"
  77. try:
  78. meta = json.loads(meta_path.read_text(encoding="utf-8")) if meta_path.is_file() else {}
  79. except Exception:
  80. meta = {}
  81. meta["version"] = version
  82. try:
  83. meta_path.write_text(json.dumps(meta, ensure_ascii=False, indent=2), encoding="utf-8")
  84. except Exception:
  85. pass
  86. def _store_dsl(table: str, q: str, case_id: str, version: str, model: str, out_dir: pathlib.Path):
  87. """把刚生成的 workflow.json 写入数据库(agent_dsl / mode_dsl)。失败不阻断。"""
  88. try:
  89. import db
  90. wf = json.loads((out_dir / "workflow.json").read_text(encoding="utf-8"))
  91. src = None
  92. sp = out_dir / "_source.json"
  93. if sp.is_file():
  94. try:
  95. src = json.loads(sp.read_text(encoding="utf-8"))
  96. except Exception:
  97. src = None
  98. platform = case_id.split("_")[0] if "_" in case_id else None
  99. db.upsert_dsl(table, q, case_id, version, model, wf, source=src,
  100. platform=platform,
  101. post_title=(src or {}).get("title"),
  102. source_link=(src or {}).get("link"))
  103. except Exception as ex:
  104. print(f"[db] _store_dsl({table}) 失败(不阻断):{ex}")
  105. def backup_procedure_history(out_dir: pathlib.Path):
  106. if not out_dir.is_dir():
  107. return
  108. files_to_backup = [f for f in out_dir.iterdir() if f.is_file()]
  109. if not files_to_backup:
  110. return
  111. history_dir = out_dir / "history"
  112. # 按旧产物的版本号归档(v_MMDDHHMM);无版本号(历史遗留)回退时间戳
  113. label = _read_meta_version(out_dir) or datetime.now().strftime("%Y%m%d_%H%M%S")
  114. version_dir = history_dir / label
  115. version_dir.mkdir(parents=True, exist_ok=True)
  116. import shutil
  117. for f in files_to_backup:
  118. try:
  119. shutil.move(str(f), str(version_dir / f.name))
  120. except Exception as e:
  121. print(f"[backup] failed to move procedure file {f.name}: {e}")
  122. def backup_reeval_history(q_dir: pathlib.Path):
  123. if not q_dir.is_dir():
  124. return
  125. files_to_backup = list(q_dir.glob("form_*.json"))
  126. if not files_to_backup:
  127. return
  128. history_dir = q_dir / "history"
  129. timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
  130. version_dir = history_dir / timestamp
  131. version_dir.mkdir(parents=True, exist_ok=True)
  132. import shutil
  133. for f in files_to_backup:
  134. try:
  135. shutil.copy(str(f), str(version_dir / f.name))
  136. except Exception as e:
  137. print(f"[backup] failed to copy reeval file {f.name}: {e}")
  138. from batch_extract_procedures import _short_case, _source_to_dsl_input, _write_meta, _composite_score
  139. def _backend_note(model, engine="cyber_runner"):
  140. """根据引擎/模型判断计费后端,明确扣 OpenRouter 余额还是 Claude 订阅。"""
  141. if engine and engine != "cyber_runner":
  142. return "Claude SDK / OAuth(扣 Claude 订阅额度)"
  143. if "qwen" in (model or "").lower():
  144. return "Qwen 原生(阿里云,不走 OpenRouter)"
  145. return "OpenRouter(扣 OpenRouter 余额,非 Claude 订阅)"
  146. def _log_header(log_fh, title, model, engine="cyber_runner"):
  147. """在子进程日志开头写一行:本次用什么模型、走什么后端。"""
  148. line = f"[server] {title} · 模型 {model} · 后端 {_backend_note(model, engine)}\n"
  149. log_fh.write(line); log_fh.write("-" * 60 + "\n"); log_fh.flush()
  150. def run_model_extraction_task(q, folder_name, case_id, src_path, out_dir, model, version):
  151. """模型提取:跑 procedure_model_extract.py(单次大模型直出 workflow.json + 渲染)。
  152. 产物写 procedures/{folder}/_model/ 子目录,task_key 带 /_model 后缀,与 Agent 工序分开。"""
  153. task_key = f"{q}/{folder_name}/_model"
  154. log_path = out_dir / "_extract.log"
  155. try:
  156. out_dir.mkdir(parents=True, exist_ok=True)
  157. cmd = [sys.executable, "-u", str(HERE / "mode-dsl" / "procedure_model_extract.py"),
  158. str(src_path), "--out-dir", str(out_dir), "--model", model, "--case-id", folder_name]
  159. flags = subprocess.CREATE_NEW_PROCESS_GROUP if sys.platform == "win32" else 0
  160. with open(log_path, "w", encoding="utf-8", buffering=1) as log_fh:
  161. _log_header(log_fh, "🤖 工序提取(模型直出,无 agent/validate 循环)", model, "cyber_runner")
  162. proc = subprocess.Popen(cmd, stdout=log_fh, stderr=subprocess.STDOUT,
  163. cwd=str(HERE), creationflags=flags)
  164. ACTIVE_TASKS[task_key]["pid"] = proc.pid
  165. proc.wait()
  166. if proc.returncode == 0:
  167. # 模型工序独立展示在「大模型工序」子 tab(按 _model/*.html 检测),不进 gallery 合并
  168. _stamp_version(out_dir, version)
  169. _store_dsl("mode_dsl", q, case_id, version, model, out_dir) # 写库
  170. ACTIVE_TASKS[task_key]["status"] = "success"
  171. else:
  172. ACTIVE_TASKS[task_key]["status"] = "failed"
  173. ACTIVE_TASKS[task_key]["error"] = f"model extract exit code {proc.returncode}"
  174. except Exception as e:
  175. ACTIVE_TASKS[task_key]["status"] = "failed"
  176. ACTIVE_TASKS[task_key]["error"] = str(e)
  177. try:
  178. with open(log_path, "a", encoding="utf-8") as f_err:
  179. f_err.write(f"\n[server error] model extract failed: {e}\n")
  180. except Exception:
  181. pass
  182. def run_extraction_task(q, folder_name, case_id, src_path, out_dir, engine, model, version):
  183. task_key = f"{q}/{folder_name}"
  184. log_path = out_dir / "_extract.log"
  185. try:
  186. out_dir.mkdir(parents=True, exist_ok=True)
  187. if engine == "cyber_runner":
  188. script_path = HERE / "procedure-dsl" / "run_cyber.py"
  189. else:
  190. script_path = HERE / "procedure-dsl" / "run_procedure_dsl.py"
  191. cmd = [
  192. sys.executable, "-u", str(script_path),
  193. str(src_path),
  194. "--out-dir", str(out_dir),
  195. "--model", model,
  196. "--max-turns", "300"
  197. ]
  198. if engine != "cyber_runner":
  199. cmd.extend(["--max-retries", "3"])
  200. else:
  201. cmd.extend(["--skill", "procedure"]) # mode_procedure:run_cyber 用直写版 skill
  202. flags = subprocess.CREATE_NEW_PROCESS_GROUP if sys.platform == "win32" else 0
  203. with open(log_path, "w", encoding="utf-8", buffering=1) as log_fh:
  204. _log_header(log_fh, "🔧 工序提取(单帖)", model, engine)
  205. proc = subprocess.Popen(cmd, stdout=log_fh, stderr=subprocess.STDOUT,
  206. cwd=str(HERE / "procedure-dsl"), creationflags=flags) # cwd=本地 procedure-dsl(run_cyber 假设的工作目录,agent 相对路径才对)
  207. ACTIVE_TASKS[task_key]["pid"] = proc.pid
  208. proc.wait()
  209. if proc.returncode == 0:
  210. try:
  211. import build_workflows
  212. build_workflows.write_one(q, folder_name, runs_dir=RUNS_DIR)
  213. _stamp_version(out_dir, version)
  214. _store_dsl("agent_dsl", q, case_id, version, model, out_dir) # 写库
  215. ACTIVE_TASKS[task_key]["status"] = "success"
  216. except Exception as ex:
  217. ACTIVE_TASKS[task_key]["status"] = "failed"
  218. ACTIVE_TASKS[task_key]["error"] = f"Workflow compilation failed: {ex}"
  219. with open(log_path, "a", encoding="utf-8") as f_err:
  220. f_err.write(f"\n[server error] Workflow compilation failed: {ex}\n")
  221. else:
  222. ACTIVE_TASKS[task_key]["status"] = "failed"
  223. ACTIVE_TASKS[task_key]["error"] = f"Runner failed with exit code {proc.returncode}"
  224. except Exception as e:
  225. ACTIVE_TASKS[task_key]["status"] = "failed"
  226. ACTIVE_TASKS[task_key]["error"] = str(e)
  227. try:
  228. with open(log_path, "a", encoding="utf-8") as f_err:
  229. f_err.write(f"\n[server error] Extraction failed: {e}\n")
  230. except Exception:
  231. pass
  232. MODSET = {"文", "图", "视频", "音频"}
  233. TOOLQUAL = {"AI": "AI 模型", "软件": "桌面 APP", "电脑端": "桌面 APP", "在线": "云端 Web",
  234. "网页版": "云端 Web", "代码": "API·CLI", "命令行": "API·CLI", "插件": "插件扩展"}
  235. def parse_dims(oq):
  236. """把组合 query(如 '文 元素生成 提示词 教程')解析回 {动作, 类型, 动作L1, 约束}。"""
  237. toks = (oq or "").split()
  238. action = next((t for t in toks if t in ACTION_SET), None)
  239. type_ = next((t for t in toks if t in TYPE_SET), None)
  240. cons = None
  241. if toks:
  242. t0 = toks[0]
  243. if t0 in MODSET:
  244. cons = {"kind": "模态", "value": t0}
  245. elif t0 in TOOLQUAL:
  246. cons = {"kind": "工具类型", "value": TOOLQUAL[t0]}
  247. return {"action": action, "type": type_, "action_l1": ACT_L1.get(action, ""), "constraint": cons}
  248. def flat_scores(sc):
  249. f = {}
  250. for k, v in (sc or {}).items():
  251. if isinstance(v, dict):
  252. for kk, vv in v.items():
  253. try: f[kk] = int(vv)
  254. except Exception: pass
  255. else:
  256. try: f[k] = int(v)
  257. except Exception: pass
  258. return f
  259. def _recency_hard(date_str):
  260. """按 publish_timestamp 头 10 字符(YYYY-MM-DD)算硬时效:半年内=3 / 两年内=2 / 更早=1。
  261. 取代原 LLM 评的 recency 维度——脚本算更稳,发布时间在帖子抓取时就有,无需 LLM token。
  262. """
  263. try:
  264. d = datetime.strptime((date_str or "")[:10], "%Y-%m-%d")
  265. except (ValueError, TypeError):
  266. return None
  267. days = (datetime.now() - d).days
  268. if days <= 180: return 3
  269. if days <= 730: return 2
  270. return 1
  271. def adapt(r, run, form_name=None):
  272. p = r.get("post", {}); e = r.get("llm_evaluation", {})
  273. # 1. 解析 知识类型 (knowledge_type)
  274. kt = []
  275. kt_raw = e.get("知识类型") or e.get("knowledge_type") or []
  276. for k in kt_raw:
  277. if k in ("工序", "procedure"): kt.append("procedure")
  278. elif k in ("能力", "步骤", "step"): kt.append("step")
  279. elif k in ("工具", "tool"): kt.append("tool")
  280. fs = {}
  281. score_reasons = {}
  282. # 检测是否为 eval_prompt_sample-mod 里的新版 0-10 分数 schema
  283. is_mod_schema = "相关性" in e and isinstance(e["相关性"], dict) and ("和内容制作知识相关" in e["相关性"] or "和 query 相关" in e["相关性"])
  284. if is_mod_schema:
  285. # 新版 0-10 分数格式解析
  286. # 1. 相关性
  287. rel = e.get("相关性") or {}
  288. for subkey, item in rel.items():
  289. if isinstance(item, dict):
  290. score_val = item.get("得分")
  291. reason_val = item.get("理由")
  292. code_key = None
  293. if "内容制作" in subkey or "知识" in subkey:
  294. code_key = "relevance_production"
  295. elif "query" in subkey or "检索" in subkey:
  296. code_key = "relevance_query"
  297. if code_key and score_val is not None:
  298. try:
  299. fs[code_key] = float(score_val)
  300. if reason_val:
  301. score_reasons[code_key] = reason_val
  302. except Exception:
  303. pass
  304. # 2. 质量
  305. q_block = e.get("质量") or {}
  306. fixed = q_block.get("固定维度") or {}
  307. # 固定维度
  308. fixed_keys = {
  309. "时效性": "recency",
  310. "热度性": "popularity",
  311. "评论反馈": "feedback"
  312. }
  313. for cn, code in fixed_keys.items():
  314. item = fixed.get(cn)
  315. if isinstance(item, dict):
  316. score_val = item.get("得分")
  317. reason_val = item.get("理由")
  318. if score_val is not None:
  319. try:
  320. fs[code] = float(score_val)
  321. if reason_val:
  322. score_reasons[code] = reason_val
  323. except Exception:
  324. pass
  325. # 用例 (真实感, 表现力)
  326. usecase = fixed.get("用例") or {}
  327. usecase_keys = {
  328. "真实感": "realism",
  329. "表现力": "expressiveness"
  330. }
  331. for cn, code in usecase_keys.items():
  332. item = usecase.get(cn)
  333. if isinstance(item, dict):
  334. score_val = item.get("得分")
  335. reason_val = item.get("理由")
  336. if score_val is not None:
  337. try:
  338. fs[code] = float(score_val)
  339. if reason_val:
  340. score_reasons[code] = reason_val
  341. except Exception:
  342. pass
  343. # 动态维度
  344. dynamic = q_block.get("动态维度") or {}
  345. # 工序
  346. proc = dynamic.get("工序") or {}
  347. if proc:
  348. item = proc.get("流程完整性")
  349. if isinstance(item, dict):
  350. score_val = item.get("得分")
  351. reason_val = item.get("理由")
  352. if score_val is not None:
  353. try:
  354. fs["procedure_completeness"] = float(score_val)
  355. if reason_val:
  356. score_reasons["procedure_completeness"] = reason_val
  357. except Exception:
  358. pass
  359. field = proc.get("字段完整性") or {}
  360. field_keys = {
  361. "输入完整性": "procedure_input",
  362. "实现完整性": "procedure_implementation",
  363. "输出完整性": "procedure_output"
  364. }
  365. for cn, code in field_keys.items():
  366. item = field.get(cn)
  367. if isinstance(item, dict):
  368. score_val = item.get("得分")
  369. reason_val = item.get("理由")
  370. if score_val is not None:
  371. try:
  372. fs[code] = float(score_val)
  373. if reason_val:
  374. score_reasons[code] = reason_val
  375. except Exception:
  376. pass
  377. item = proc.get("泛化性")
  378. if isinstance(item, dict):
  379. score_val = item.get("得分")
  380. reason_val = item.get("理由")
  381. if score_val is not None:
  382. try:
  383. fs["procedure_generality"] = float(score_val)
  384. if reason_val:
  385. score_reasons["procedure_generality"] = reason_val
  386. except Exception:
  387. pass
  388. # 能力
  389. cap = dynamic.get("能力") or dynamic.get("步骤") or {}
  390. if cap:
  391. field = cap.get("字段完整性") or {}
  392. field_keys = {
  393. "输入完整性": "step_input",
  394. "实现完整性": "step_implementation",
  395. "输出完整性": "step_output"
  396. }
  397. for cn, code in field_keys.items():
  398. item = field.get(cn)
  399. if isinstance(item, dict):
  400. score_val = item.get("得分")
  401. reason_val = item.get("理由")
  402. if score_val is not None:
  403. try:
  404. fs[code] = float(score_val)
  405. if reason_val:
  406. score_reasons[code] = reason_val
  407. except Exception:
  408. pass
  409. item = cap.get("泛化性")
  410. if isinstance(item, dict):
  411. score_val = item.get("得分")
  412. reason_val = item.get("理由")
  413. if score_val is not None:
  414. try:
  415. fs["step_generality"] = float(score_val)
  416. if reason_val:
  417. score_reasons["step_generality"] = reason_val
  418. except Exception:
  419. pass
  420. # 工具
  421. tool = dynamic.get("工具") or {}
  422. if tool:
  423. tool_keys = {
  424. "能力边界覆盖": "tool_boundary",
  425. "有效比较": "tool_comparison",
  426. "参数/接口具体性": "tool_specificity",
  427. "实操示例": "tool_example",
  428. "版本&限制": "tool_limits"
  429. }
  430. for cn, code in tool_keys.items():
  431. item = tool.get(cn)
  432. if isinstance(item, dict):
  433. score_val = item.get("得分")
  434. reason_val = item.get("理由")
  435. if score_val is not None:
  436. try:
  437. fs[code] = float(score_val)
  438. if reason_val:
  439. score_reasons[code] = reason_val
  440. except Exception:
  441. pass
  442. else:
  443. # 兼容老版 1-5 分数 schema (带 "评分" 或 old-style flatness)
  444. is_new_schema = "评分" in e or "知识类型" in e or "制作相关性" in e
  445. CN_TO_EN = {
  446. "相关性": "relevance",
  447. "成品质量": "result_quality",
  448. "可信度": "credibility",
  449. "具体用例": "concrete_use_case",
  450. "完整性": "completeness",
  451. "步骤结构": "step_structure",
  452. "步骤可复现": "step_reproducibility",
  453. "步骤可复现性": "step_reproducibility",
  454. "能力定义": "capability_definition",
  455. "实现深度": "implementation_depth",
  456. "边界失败": "boundary_failure_eval",
  457. "通用性": "generality",
  458. "能力覆盖": "capability_coverage",
  459. "有效对比": "effective_comparison",
  460. "参数具体": "param_specificity",
  461. "实操示例": "worked_example",
  462. "实操用例": "worked_example",
  463. "示例完整": "worked_example",
  464. "版本限制": "version_limits",
  465. "版本说明": "version_limits",
  466. "限制说明": "version_limits",
  467. }
  468. if is_new_schema:
  469. pf = e.get("评分") or {}
  470. for cat, metrics in pf.items():
  471. if isinstance(metrics, dict):
  472. for metric, val in metrics.items():
  473. en_key = CN_TO_EN.get(metric, metric)
  474. if isinstance(val, dict) and "得分" in val:
  475. try: fs[en_key] = int(val["得分"])
  476. except Exception: pass
  477. elif isinstance(val, (int, float)):
  478. fs[en_key] = int(val)
  479. if isinstance(val, dict) and "理由" in val:
  480. score_reasons[en_key] = val["理由"]
  481. else:
  482. fs = flat_scores(e.get("scores", {}))
  483. # 计算均分 (overall)
  484. if is_mod_schema:
  485. rel_keys = {"relevance_production", "relevance_query"}
  486. rel_vals = [v for k, v in fs.items() if k in rel_keys]
  487. qual_vals = [v for k, v in fs.items() if k not in rel_keys]
  488. rel_avg = sum(rel_vals) / len(rel_vals) if rel_vals else None
  489. qual_avg = sum(qual_vals) / len(qual_vals) if qual_vals else None
  490. if rel_avg is not None and qual_avg is not None:
  491. overall = round((rel_avg + qual_avg) / 2, 1)
  492. elif rel_avg is not None:
  493. overall = round(rel_avg, 1)
  494. elif qual_avg is not None:
  495. overall = round(qual_avg, 1)
  496. else:
  497. overall = 0.0
  498. else:
  499. overall = round(sum(fs.values()) / len(fs), 1) if fs else 0
  500. anomaly = bool(e.get("error")) or not fs
  501. grade = p.get("_quality_grade", "")
  502. fb = r.get("found_by_queries", [])
  503. # 4. 解析 制作相关性 (production_relevance)
  504. if is_mod_schema:
  505. # 新版使用 "相关性" 中的 "和内容制作知识相关" 代表制作相关性
  506. production_relevance = fs.get("relevance_production")
  507. else:
  508. if is_new_schema:
  509. pr_block = e.get("制作相关性") or {}
  510. pr_raw = pr_block.get("得分") if isinstance(pr_block, dict) else pr_block
  511. if isinstance(pr_block, dict) and "理由" in pr_block:
  512. score_reasons["production_relevance"] = pr_block["理由"]
  513. else:
  514. pr_raw = e.get("production_relevance")
  515. try: production_relevance = int(float(pr_raw)) if pr_raw is not None else None
  516. except (TypeError, ValueError): production_relevance = None
  517. recency_hard = _recency_hard(p.get("publish_timestamp", ""))
  518. # 5. 解析 判定决策 (decision) 和 理由 (reason)
  519. reason = e.get("判定理由") or e.get("reason") or ""
  520. # 根据过滤指标决定是否保留 (过滤指标判定逻辑优先,不依赖文字匹配)
  521. is_discard = False
  522. # 制作相关性低于阈值则丢弃 (新版 0-10 满分,因此低于 4 丢弃;老版低于 2 丢弃)
  523. if production_relevance is not None:
  524. threshold = 4 if is_mod_schema else 2
  525. if production_relevance < threshold:
  526. is_discard = True
  527. # 时效性低于 2 被丢弃(发布时间超两年的老帖)
  528. if recency_hard is not None and recency_hard < 2:
  529. is_discard = True
  530. # 综合均分低于阈值被丢弃 (新版低于 6 丢弃;老版低于 3 丢弃)
  531. if overall is not None:
  532. threshold_ov = 6 if is_mod_schema else 3
  533. if overall < threshold_ov:
  534. is_discard = True
  535. decision = "discard" if is_discard else "report"
  536. # Find matching procedure html
  537. procedure_html = None
  538. case_id = r.get("case_id", "")
  539. title = p.get("title", "")
  540. run_dir = RUNS_DIR / run
  541. if run_dir.is_dir():
  542. # 1. 优先扫描该帖子对应的文件夹下的任何 HTML 文件 (不限名称)
  543. # 文件夹名格式: {form}_{platform}_{channel_content_id[:8]}
  544. content_id = r.get("channel_content_id") or ""
  545. if not content_id and case_id and "_" in case_id:
  546. content_id = case_id.split("_", 1)[1]
  547. plat_key = r.get("platform") or ""
  548. if form_name and plat_key and content_id:
  549. folder_name = f"{form_name}_{plat_key}_{content_id[:8]}"
  550. case_dir = run_dir / "procedures" / folder_name
  551. if case_dir.is_dir():
  552. html_files = list(case_dir.glob("*.html"))
  553. if html_files:
  554. procedure_html = f"runs_full/{run}/procedures/{folder_name}/{html_files[0].name}"
  555. # 1b. 兼容 fallback: 批量提取模式下直接以 case_id 命名的文件夹
  556. if not procedure_html and case_id:
  557. case_dir = run_dir / "procedures" / case_id
  558. if case_dir.is_dir():
  559. html_files = list(case_dir.glob("*.html"))
  560. if html_files:
  561. procedure_html = f"runs_full/{run}/procedures/{case_id}/{html_files[0].name}"
  562. # 2. 其次匹配标准文件名: case-{case_id}.html 或 {case_id}.html
  563. candidate_dirs = [run_dir, run_dir / "procedures"]
  564. if not procedure_html and case_id:
  565. named_files = [f"case-{case_id}.html", f"{case_id}.html"]
  566. for d_dir in candidate_dirs:
  567. if d_dir.is_dir():
  568. for name in named_files:
  569. if (d_dir / name).is_file():
  570. procedure_html = f"runs_full/{run}/procedures/{name}" if d_dir.name == "procedures" else f"runs_full/{run}/{name}"
  571. break
  572. if procedure_html:
  573. break
  574. # 3. 再次匹配 HTML 内部的标准声明 (meta 标签或 HTML 注释)
  575. if not procedure_html and case_id:
  576. for d_dir in candidate_dirs:
  577. if d_dir.is_dir():
  578. for html_path in d_dir.glob("*.html"):
  579. try:
  580. content = html_path.read_text(encoding="utf-8")
  581. if f'name="case-id" content="{case_id}"' in content or \
  582. f'name="case_id" content="{case_id}"' in content or \
  583. f'<!-- case_id: {case_id} -->' in content or \
  584. f'<!-- case-id: {case_id} -->' in content:
  585. procedure_html = f"runs_full/{run}/procedures/{html_path.name}" if d_dir.name == "procedures" else f"runs_full/{run}/{html_path.name}"
  586. break
  587. except Exception:
  588. continue
  589. if procedure_html:
  590. break
  591. # 4. 最后使用标题作为兜底模糊匹配
  592. if not procedure_html and title:
  593. for d_dir in candidate_dirs:
  594. if d_dir.is_dir():
  595. for html_path in d_dir.glob("*.html"):
  596. try:
  597. content = html_path.read_text(encoding="utf-8")
  598. if title in content:
  599. procedure_html = f"runs_full/{run}/procedures/{html_path.name}" if d_dir.name == "procedures" else f"runs_full/{run}/{html_path.name}"
  600. break
  601. except Exception:
  602. continue
  603. if procedure_html:
  604. break
  605. if procedure_html:
  606. procedure_html = procedure_html.replace("runs_full/", f"{RUNS_DIR_NAME}/")
  607. return {
  608. "case_id": r.get("case_id", ""),
  609. "platform": PLAT.get(r.get("platform"), r.get("platform")), "platformKey": r.get("platform"),
  610. "title": p.get("title", "") or "(无标题)", "date": (p.get("publish_timestamp", "") or "")[:10],
  611. "url": r.get("source_url", ""), "engagement": f'{p.get("like_count", 0)} 赞',
  612. "knowledge_type": kt, "decision": decision,
  613. "tools": [KT.get(k, k) for k in kt] + ([f"质量 {grade}"] if grade else []), "found_by": fb,
  614. "images": (p.get("images") or [])[:6], "text": p.get("body_text", "") or "",
  615. "scores": fs, "overall": overall, "reason": reason, "score_reasons": score_reasons,
  616. "grade": grade, "qscore": p.get("_quality_score", 0), "anomaly": anomaly,
  617. "production_relevance": production_relevance, "recency_hard": recency_hard,
  618. "run": run, "procedure_html": procedure_html,
  619. }
  620. def scan_runs():
  621. runs = {}
  622. for f in sorted(glob.glob(str(RUNS_DIR / "*" / "form_*.json"))):
  623. try:
  624. d = json.load(open(f, encoding="utf-8"))
  625. except Exception:
  626. continue
  627. run = pathlib.Path(f).parent.name
  628. form_name = d.get("form") or ""
  629. results = [adapt(r, run, form_name) for r in d.get("results", [])]
  630. report_val = sum(1 for r in results if r.get("decision") == "report" and not r.get("anomaly"))
  631. discard_val = sum(1 for r in results if r.get("decision") == "discard" and not r.get("anomaly"))
  632. runs.setdefault(run, []).append({
  633. "form": d.get("form"), "query": d.get("query"), "original_q": d.get("original_q", ""),
  634. "requirement": d.get("requirement", ""),
  635. "platforms": d.get("platforms", []), "total": d.get("total"),
  636. "report": report_val, "discard": discard_val,
  637. "results": results,
  638. })
  639. for v in runs.values():
  640. v.sort(key=lambda x: x.get("form") or "")
  641. if MODE == "new":
  642. def _mtime(name):
  643. try:
  644. return (RUNS_DIR / name).stat().st_mtime
  645. except Exception:
  646. return 0
  647. sorted_keys = sorted(runs.keys(), key=_mtime, reverse=True)
  648. else:
  649. def _qnum(name):
  650. m = re.search(r"\d+", name)
  651. return (int(m.group()) if m else 0, name)
  652. sorted_keys = [kv[0] for kv in sorted(runs.items(), key=lambda kv: _qnum(kv[0]))]
  653. out = []
  654. for k in sorted_keys:
  655. v = runs[k]
  656. oq = v[0].get("original_q") or v[0].get("query") or ""
  657. seen, hits = set(), 0 # 知识命中数 = 各形式采纳(report)且非异常、按 url 去重后的帖子数
  658. for f in v:
  659. for r in f.get("results", []):
  660. if r.get("decision") == "report" and not r.get("anomaly") and r.get("url") not in seen:
  661. seen.add(r.get("url")); hits += 1
  662. out.append({"key": k, "forms": v, "dims": parse_dims(oq), "original_q": oq,
  663. "hits": hits, "tot": sum((f.get("total") or 0) for f in v)})
  664. active_reevals = {k: v["status"] for k, v in ACTIVE_REEVALS.items()}
  665. active_batch_tasks = {k: v["status"] for k, v in ACTIVE_BATCH_TASKS.items()}
  666. return {"queries": out, "actions": ACTIONS_TAX, "types": TYPES_TAX, "matrix": _MATRIX, "active_reevals": active_reevals, "active_batch_tasks": active_batch_tasks}
  667. class H(BaseHTTPRequestHandler):
  668. def _send(self, code, body, ctype):
  669. b = body.encode("utf-8") if isinstance(body, str) else body
  670. self.send_response(code)
  671. if ctype.startswith("text/") or ctype == "application/json" or ctype == "application/javascript":
  672. self.send_header("Content-Type", ctype + "; charset=utf-8")
  673. else:
  674. self.send_header("Content-Type", ctype)
  675. self.send_header("Content-Length", str(len(b))); self.end_headers(); self.wfile.write(b)
  676. def do_GET(self):
  677. parsed = urlparse(self.path)
  678. path = parsed.path
  679. params = parse_qs(parsed.query)
  680. if path in ("/", "/index.html"):
  681. try:
  682. html_file = "new_query.html" if MODE == "new" else "index.html"
  683. page = (HERE / html_file).read_text(encoding="utf-8")
  684. self._send(200, page, "text/html")
  685. except Exception as e:
  686. self._send(500, f"Error reading page: {e}", "text/plain")
  687. elif path == "/api/data":
  688. self._send(200, json.dumps(scan_runs(), ensure_ascii=False), "application/json")
  689. elif path == "/api/procedure_status":
  690. q = (params.get("q") or [""])[0].strip()
  691. form = (params.get("form") or [""])[0].strip()
  692. case_id = (params.get("case_id") or [""])[0].strip()
  693. if not q or not form or not case_id:
  694. self._send(400, "missing q, form, or case_id", "text/plain")
  695. return
  696. variant = (params.get("variant") or ["agent"])[0].strip()
  697. req_version = (params.get("version") or [""])[0].strip()
  698. folder_name = f"{form}_{_short_case(case_id)}"
  699. out_dir = RUNS_DIR / q / "procedures" / folder_name
  700. rel_base = f"{RUNS_DIR_NAME}/{q}/procedures/{folder_name}"
  701. task_key = f"{q}/{folder_name}"
  702. if variant == "model": # 大模型工序在 _model/ 子目录
  703. out_dir = out_dir / "_model"
  704. rel_base = rel_base + "/_model"
  705. task_key = task_key + "/_model"
  706. # 指定了非当前版本 → 直接从 history/{version}/ 取该版本的 html
  707. if req_version and req_version != _read_meta_version(out_dir):
  708. hist = out_dir / "history" / req_version
  709. hf = list(hist.glob("*.html")) if hist.is_dir() else []
  710. if hf:
  711. self._send(200, json.dumps({"status": "success",
  712. "procedure_html": f"{rel_base}/history/{req_version}/{hf[0].name}"},
  713. ensure_ascii=False), "application/json")
  714. else:
  715. self._send(200, json.dumps({"status": "not_started",
  716. "error": f"版本 {req_version} 的页面文件已不存在"}, ensure_ascii=False), "application/json")
  717. return
  718. if task_key in ACTIVE_TASKS:
  719. task = ACTIVE_TASKS[task_key]
  720. res = {"status": task["status"], "error": task["error"]}
  721. if task["status"] == "success":
  722. html_files = list(out_dir.glob("*.html")) if out_dir.is_dir() else []
  723. if html_files:
  724. res["procedure_html"] = f"{rel_base}/{html_files[0].name}"
  725. self._send(200, json.dumps(res, ensure_ascii=False), "application/json")
  726. return
  727. html_files = list(out_dir.glob("*.html")) if out_dir.is_dir() else []
  728. if html_files:
  729. self._send(200, json.dumps({
  730. "status": "success",
  731. "procedure_html": f"{rel_base}/{html_files[0].name}"
  732. }, ensure_ascii=False), "application/json")
  733. return
  734. # 仅 agent 变体兼容旧的 case_id 命名目录
  735. if variant != "model" and case_id:
  736. fallback_dir = RUNS_DIR / q / "procedures" / case_id
  737. fb_html = list(fallback_dir.glob("*.html")) if fallback_dir.is_dir() else []
  738. if fb_html:
  739. self._send(200, json.dumps({
  740. "status": "success",
  741. "procedure_html": f"{RUNS_DIR_NAME}/{q}/procedures/{case_id}/{fb_html[0].name}"
  742. }, ensure_ascii=False), "application/json")
  743. return
  744. log_path = out_dir / "_extract.log"
  745. if not log_path.is_file() and variant != "model" and case_id:
  746. log_path = RUNS_DIR / q / "procedures" / case_id / "_extract.log"
  747. if log_path.is_file():
  748. self._send(200, json.dumps({"status": "failed", "error": "Not running, but no HTML output found (possibly crashed)."}, ensure_ascii=False), "application/json")
  749. return
  750. self._send(200, json.dumps({"status": "not_started"}, ensure_ascii=False), "application/json")
  751. elif path == "/api/procedure_versions":
  752. # 列某帖某变体的所有版本(DB 为主,文件系统兜底)。latest = 当前 _meta 版本。
  753. q = (params.get("q") or [""])[0].strip()
  754. form = (params.get("form") or [""])[0].strip()
  755. case_id = (params.get("case_id") or [""])[0].strip()
  756. variant = (params.get("variant") or ["agent"])[0].strip()
  757. folder_name = f"{form}_{_short_case(case_id)}"
  758. out_dir = RUNS_DIR / q / "procedures" / folder_name
  759. if variant == "model":
  760. out_dir = out_dir / "_model"
  761. latest = _read_meta_version(out_dir)
  762. by_ver = {}
  763. # DB(带 model / 工序数 / 时间)
  764. try:
  765. import db
  766. for r in db.fetch_versions("mode_dsl" if variant == "model" else "agent_dsl", q, case_id):
  767. if r.get("version"):
  768. by_ver[r["version"]] = {"version": r["version"], "model": r.get("model"),
  769. "procedure_count": r.get("procedure_count"),
  770. "created_time": str(r.get("created_time") or "")}
  771. except Exception:
  772. pass
  773. # 文件系统兜底(当前 + history/v_*)
  774. for v in ([latest] if latest else []) + (
  775. [d.name for d in (out_dir / "history").iterdir()
  776. if d.is_dir() and d.name.startswith("v_")] if (out_dir / "history").is_dir() else []):
  777. by_ver.setdefault(v, {"version": v, "model": None, "procedure_count": None, "created_time": ""})
  778. versions = sorted(by_ver.values(), key=lambda x: x["version"], reverse=True)
  779. for v in versions:
  780. v["is_latest"] = (v["version"] == latest)
  781. self._send(200, json.dumps({"versions": versions, "latest": latest}, ensure_ascii=False), "application/json")
  782. elif path == "/api/procedure_log":
  783. q = (params.get("q") or [""])[0].strip()
  784. form = (params.get("form") or [""])[0].strip()
  785. case_id = (params.get("case_id") or [""])[0].strip()
  786. if not q or not form or not case_id:
  787. self._send(400, "missing q, form, or case_id", "text/plain")
  788. return
  789. variant = (params.get("variant") or ["agent"])[0].strip()
  790. folder_name = f"{form}_{_short_case(case_id)}"
  791. if variant == "model":
  792. log_path = RUNS_DIR / q / "procedures" / folder_name / "_model" / "_extract.log"
  793. else:
  794. log_path = RUNS_DIR / q / "procedures" / folder_name / "_extract.log"
  795. if not log_path.is_file() and case_id:
  796. log_path = RUNS_DIR / q / "procedures" / case_id / "_extract.log"
  797. if not log_path.is_file():
  798. self._send(200, json.dumps({"log": ""}, ensure_ascii=False), "application/json")
  799. return
  800. try:
  801. content = log_path.read_text(encoding="utf-8", errors="replace")
  802. self._send(200, json.dumps({"log": content}, ensure_ascii=False), "application/json")
  803. except Exception as e:
  804. self._send(500, json.dumps({"error": str(e)}, ensure_ascii=False), "application/json")
  805. elif path == "/api/spec_content":
  806. file_name = (params.get("file") or [""])[0].strip()
  807. allowed = [
  808. "README.md",
  809. "tools.md",
  810. "extraction/phase1-skeleton.md",
  811. "extraction/phase2-normalize.md",
  812. "extraction/phase3-finalize.md",
  813. "taxonomy/type_suggestions.md"
  814. ]
  815. if file_name not in allowed:
  816. self._send(400, "invalid file parameter", "text/plain")
  817. return
  818. target_path = HERE / "procedure-dsl" / "spec" / file_name
  819. if not target_path.is_file():
  820. self._send(404, "spec file not found", "text/plain")
  821. return
  822. try:
  823. content = target_path.read_text(encoding="utf-8", errors="replace")
  824. self._send(200, json.dumps({"content": content}, ensure_ascii=False), "application/json")
  825. except Exception as e:
  826. self._send(500, json.dumps({"error": str(e)}, ensure_ascii=False), "application/json")
  827. elif path == "/api/reeval_status":
  828. q = (params.get("q") or [""])[0].strip()
  829. if not q:
  830. self._send(400, "missing q", "text/plain")
  831. return
  832. if q in ACTIVE_REEVALS:
  833. self._send(200, json.dumps({
  834. "status": ACTIVE_REEVALS[q]["status"],
  835. "error": ACTIVE_REEVALS[q].get("error")
  836. }, ensure_ascii=False), "application/json")
  837. else:
  838. self._send(200, json.dumps({"status": "not_started"}, ensure_ascii=False), "application/json")
  839. elif path == "/api/search_eval_status":
  840. q = (params.get("q") or [""])[0].strip()
  841. if not q:
  842. self._send(400, "missing q", "text/plain")
  843. return
  844. if q in ACTIVE_SEARCH_EVALS:
  845. self._send(200, json.dumps({
  846. "status": ACTIVE_SEARCH_EVALS[q]["status"],
  847. "error": ACTIVE_SEARCH_EVALS[q].get("error")
  848. }, ensure_ascii=False), "application/json")
  849. else:
  850. self._send(200, json.dumps({"status": "not_started"}, ensure_ascii=False), "application/json")
  851. elif path == "/api/batch_generate_status":
  852. q = (params.get("q") or [""])[0].strip()
  853. if not q:
  854. self._send(400, "missing q", "text/plain")
  855. return
  856. if q in ACTIVE_BATCH_TASKS:
  857. self._send(200, json.dumps({
  858. "status": ACTIVE_BATCH_TASKS[q]["status"],
  859. "error": ACTIVE_BATCH_TASKS[q].get("error")
  860. }, ensure_ascii=False), "application/json")
  861. else:
  862. self._send(200, json.dumps({"status": "not_started"}, ensure_ascii=False), "application/json")
  863. elif path == "/api/batch_generate_log":
  864. q = (params.get("q") or [""])[0].strip()
  865. if not q:
  866. self._send(400, "missing q", "text/plain")
  867. return
  868. safe_q = re.sub(r'[\x00-\x1f\\/*?:"<>|]', '', q).strip()
  869. log_path = RUNS_DIR / safe_q / "procedures" / "_batch_extract.log"
  870. if not log_path.is_file():
  871. self._send(200, json.dumps({"log": ""}, ensure_ascii=False), "application/json")
  872. return
  873. try:
  874. content = log_path.read_text(encoding="utf-8", errors="replace")
  875. self._send(200, json.dumps({"log": content}, ensure_ascii=False), "application/json")
  876. except Exception as e:
  877. self._send(500, json.dumps({"error": str(e)}, ensure_ascii=False), "application/json")
  878. elif path == "/api/search_eval_log":
  879. q = (params.get("q") or [""])[0].strip()
  880. if not q:
  881. self._send(400, "missing q", "text/plain")
  882. return
  883. safe_q = re.sub(r'[\x00-\x1f\\/*?:"<>|]', '', q).strip()
  884. log_path = RUNS_DIR / safe_q / "_search_eval.log"
  885. if not log_path.is_file():
  886. self._send(200, json.dumps({"log": ""}, ensure_ascii=False), "application/json")
  887. return
  888. try:
  889. content = log_path.read_text(encoding="utf-8", errors="replace")
  890. self._send(200, json.dumps({"log": content}, ensure_ascii=False), "application/json")
  891. except Exception as e:
  892. self._send(500, json.dumps({"error": str(e)}, ensure_ascii=False), "application/json")
  893. elif self.path.startswith("/runs_full/") or self.path.startswith("/runs_new/"):
  894. try:
  895. import urllib.parse
  896. clean_path = urllib.parse.unquote(self.path.split("?")[0])
  897. parts = clean_path.strip("/").split("/")
  898. target_file = HERE
  899. for part in parts:
  900. target_file = target_file / part
  901. runs_dir = HERE / parts[0]
  902. if runs_dir.resolve() in target_file.resolve().parents and target_file.is_file():
  903. content = target_file.read_bytes()
  904. ext = target_file.suffix.lower()
  905. ctype = "text/html"
  906. if ext in (".png", ".webp"):
  907. ctype = f"image/{ext[1:]}"
  908. elif ext in (".jpg", ".jpeg"):
  909. ctype = "image/jpeg"
  910. elif ext == ".json":
  911. ctype = "application/json"
  912. elif ext == ".js":
  913. ctype = "application/javascript"
  914. elif ext == ".css":
  915. ctype = "text/css"
  916. self._send(200, content, ctype)
  917. else:
  918. self._send(404, "not found", "text/plain")
  919. except Exception as e:
  920. self._send(500, f"Error: {e}", "text/plain")
  921. else:
  922. self._send(404, "not found", "text/plain")
  923. def do_POST(self):
  924. if self.path == "/api/run_search_eval":
  925. length = int(self.headers.get("Content-Length") or 0)
  926. raw = self.rfile.read(length).decode("utf-8") if length > 0 else "{}"
  927. try:
  928. payload = json.loads(raw)
  929. except Exception as e:
  930. self._send(400, json.dumps({"error": f"bad json: {e}"}), "application/json"); return
  931. q = (payload.get("query") or "").strip()
  932. platforms = (payload.get("platforms") or "xhs,zhihu").strip()
  933. if not q:
  934. self._send(400, json.dumps({"error": "missing query"}, ensure_ascii=False), "application/json"); return
  935. safe_q = re.sub(r'[\x00-\x1f\\/*?:"<>|]', '', q).strip()
  936. if not safe_q:
  937. self._send(400, json.dumps({"error": "invalid query name"}, ensure_ascii=False), "application/json"); return
  938. q_dir = RUNS_DIR / safe_q
  939. q_dir.mkdir(parents=True, exist_ok=True)
  940. temp_queries_file = q_dir / "temp_queries.json"
  941. try:
  942. with open(temp_queries_file, "w", encoding="utf-8") as f:
  943. json.dump([q], f, ensure_ascii=False)
  944. except Exception as e:
  945. self._send(500, json.dumps({"error": f"failed to write temp query: {e}"}), "application/json"); return
  946. log_path = q_dir / "_search_eval.log"
  947. ACTIVE_SEARCH_EVALS[q] = {
  948. "status": "running",
  949. "pid": None,
  950. "error": None
  951. }
  952. def run_search_eval_task(query_text, safe_query_name, queries_file, out_dir, target_platforms, log_file):
  953. try:
  954. cmd = [
  955. sys.executable, "-u", str(SE / "search_and_evaluate.py"),
  956. "--queries", str(queries_file),
  957. "--platforms", target_platforms,
  958. "--output-dir", str(out_dir),
  959. "--max-count", "20"
  960. ]
  961. import os
  962. env = os.environ.copy()
  963. env["PYTHONIOENCODING"] = "utf-8"
  964. env["PYTHONUTF8"] = "1"
  965. # 搜索评估未传 --eval-model → 用 search_and_evaluate 默认 gemini-flash-lite(OpenRouter)
  966. _se_model = cmd[cmd.index("--eval-model") + 1] if "--eval-model" in cmd else "google/gemini-3.1-flash-lite (默认)"
  967. flags = subprocess.CREATE_NEW_PROCESS_GROUP if sys.platform == "win32" else 0
  968. with open(log_file, "w", encoding="utf-8", buffering=1) as log_fh:
  969. _log_header(log_fh, "🔎 搜索评估", _se_model)
  970. proc = subprocess.Popen(cmd, stdout=log_fh, stderr=subprocess.STDOUT,
  971. cwd=str(SE), env=env, creationflags=flags)
  972. ACTIVE_SEARCH_EVALS[query_text]["pid"] = proc.pid
  973. proc.wait()
  974. if proc.returncode == 0:
  975. evaluated_file = out_dir / "evaluated.json"
  976. if evaluated_file.is_file():
  977. with open(evaluated_file, "r", encoding="utf-8") as rf:
  978. eval_data = json.load(rf)
  979. eval_data["form"] = "A"
  980. eval_data["query"] = query_text
  981. eval_data["original_q"] = query_text
  982. with open(out_dir / "form_A.json", "w", encoding="utf-8") as wf:
  983. json.dump(eval_data, wf, ensure_ascii=False, indent=2)
  984. ACTIVE_SEARCH_EVALS[query_text]["status"] = "success"
  985. else:
  986. ACTIVE_SEARCH_EVALS[query_text]["status"] = "failed"
  987. ACTIVE_SEARCH_EVALS[query_text]["error"] = "evaluated.json not found after execution"
  988. with open(log_file, "a", encoding="utf-8") as f_err:
  989. f_err.write("\n[server error] evaluated.json not found after execution\n")
  990. else:
  991. ACTIVE_SEARCH_EVALS[query_text]["status"] = "failed"
  992. ACTIVE_SEARCH_EVALS[query_text]["error"] = f"search_and_evaluate.py exited with code {proc.returncode}"
  993. except Exception as ex:
  994. ACTIVE_SEARCH_EVALS[query_text]["status"] = "failed"
  995. ACTIVE_SEARCH_EVALS[query_text]["error"] = str(ex)
  996. try:
  997. with open(log_file, "a", encoding="utf-8") as f_err:
  998. f_err.write(f"\n[server error] Exception: {ex}\n")
  999. except Exception:
  1000. pass
  1001. finally:
  1002. try:
  1003. if queries_file.is_file():
  1004. queries_file.unlink()
  1005. except Exception:
  1006. pass
  1007. t = threading.Thread(target=run_search_eval_task, args=(q, safe_q, temp_queries_file, q_dir, platforms, log_path))
  1008. t.daemon = True
  1009. t.start()
  1010. self._send(200, json.dumps({
  1011. "status": "started",
  1012. "query": q,
  1013. "log": f"{RUNS_DIR_NAME}/{safe_q}/_search_eval.log"
  1014. }, ensure_ascii=False), "application/json")
  1015. elif self.path == "/api/generate_procedure":
  1016. length = int(self.headers.get("Content-Length") or 0)
  1017. raw = self.rfile.read(length).decode("utf-8") if length > 0 else "{}"
  1018. try:
  1019. payload = json.loads(raw)
  1020. except Exception as e:
  1021. self._send(400, json.dumps({"error": f"bad json: {e}"}), "application/json"); return
  1022. q = (payload.get("q") or "").strip()
  1023. form = (payload.get("form") or "").strip()
  1024. case_id = (payload.get("case_id") or "").strip()
  1025. engine = (payload.get("engine") or "cyber_runner").strip()
  1026. model = (payload.get("model") or "google/gemini-3.1-flash-lite").strip()
  1027. if MODE != "new" and not re.match(r"^q\d+$", q):
  1028. self._send(400, json.dumps({"error": f"bad q (expect 'qNN'): {q!r}"}, ensure_ascii=False), "application/json"); return
  1029. if form not in ("A", "B", "C"):
  1030. self._send(400, json.dumps({"error": f"bad form: {form!r}"}, ensure_ascii=False), "application/json"); return
  1031. if not case_id:
  1032. self._send(400, json.dumps({"error": "missing case_id"}, ensure_ascii=False), "application/json"); return
  1033. q_dir = RUNS_DIR / q
  1034. form_file = q_dir / f"form_{form}.json"
  1035. if not form_file.is_file():
  1036. self._send(404, json.dumps({"error": f"form file not found: {form_file.name}"}, ensure_ascii=False), "application/json"); return
  1037. try:
  1038. with open(form_file, encoding="utf-8") as f:
  1039. form_data = json.load(f)
  1040. except Exception as e:
  1041. self._send(500, json.dumps({"error": f"failed to read form: {e}"}, ensure_ascii=False), "application/json"); return
  1042. matching_result = None
  1043. for r in form_data.get("results", []):
  1044. if r.get("case_id") == case_id:
  1045. matching_result = r
  1046. break
  1047. if not matching_result:
  1048. self._send(404, json.dumps({"error": f"case_id {case_id} not found in form {form}"}, ensure_ascii=False), "application/json"); return
  1049. folder_name = f"{form}_{_short_case(case_id)}"
  1050. out_dir = q_dir / "procedures" / folder_name
  1051. backup_procedure_history(out_dir)
  1052. out_dir.mkdir(parents=True, exist_ok=True)
  1053. src_path = out_dir / "_source.json"
  1054. try:
  1055. with open(src_path, "w", encoding="utf-8") as f:
  1056. json.dump(_source_to_dsl_input(matching_result), f, ensure_ascii=False, indent=2)
  1057. score = _composite_score(matching_result.get("llm_evaluation") or {})
  1058. _write_meta(out_dir, case_id=case_id, from_q=q, form=form, score=score)
  1059. version = _gen_version()
  1060. _stamp_version(out_dir, version)
  1061. except Exception as e:
  1062. self._send(500, json.dumps({"error": f"failed to write inputs: {e}"}, ensure_ascii=False), "application/json"); return
  1063. task_key = f"{q}/{folder_name}"
  1064. ACTIVE_TASKS[task_key] = {
  1065. "status": "running",
  1066. "start_time": datetime.now().isoformat(),
  1067. "pid": None,
  1068. "error": None
  1069. }
  1070. t = threading.Thread(target=run_extraction_task, args=(q, folder_name, case_id, src_path, out_dir, engine, model, version))
  1071. t.daemon = True
  1072. t.start()
  1073. self._send(200, json.dumps({
  1074. "status": "started",
  1075. "task_key": task_key,
  1076. "log": f"{RUNS_DIR_NAME}/{q}/procedures/{folder_name}/_extract.log"
  1077. }, ensure_ascii=False), "application/json")
  1078. elif self.path == "/api/extract_procedure_model":
  1079. # 模型提取:单次大模型直出 workflow.json(无 agent / 无 validate 循环)。
  1080. # 写同一个 procedures/{form}_{short_case}/,复用 procedure_status/log/展示。
  1081. length = int(self.headers.get("Content-Length") or 0)
  1082. raw = self.rfile.read(length).decode("utf-8") if length > 0 else "{}"
  1083. try:
  1084. payload = json.loads(raw)
  1085. except Exception as e:
  1086. self._send(400, json.dumps({"error": f"bad json: {e}"}), "application/json"); return
  1087. q = (payload.get("q") or "").strip()
  1088. form = (payload.get("form") or "").strip()
  1089. case_id = (payload.get("case_id") or "").strip()
  1090. model = (payload.get("model") or "anthropic/claude-sonnet-4-6").strip() # 大模型工序默认 claude
  1091. if form not in ("A", "B", "C"):
  1092. self._send(400, json.dumps({"error": f"bad form: {form!r}"}, ensure_ascii=False), "application/json"); return
  1093. if not case_id:
  1094. self._send(400, json.dumps({"error": "missing case_id"}, ensure_ascii=False), "application/json"); return
  1095. q_dir = RUNS_DIR / q
  1096. form_file = q_dir / f"form_{form}.json"
  1097. if not form_file.is_file():
  1098. self._send(404, json.dumps({"error": f"form file not found: {form_file.name}"}, ensure_ascii=False), "application/json"); return
  1099. try:
  1100. with open(form_file, encoding="utf-8") as f:
  1101. form_data = json.load(f)
  1102. except Exception as e:
  1103. self._send(500, json.dumps({"error": f"failed to read form: {e}"}, ensure_ascii=False), "application/json"); return
  1104. matching_result = next((r for r in form_data.get("results", []) if r.get("case_id") == case_id), None)
  1105. if not matching_result:
  1106. self._send(404, json.dumps({"error": f"case_id {case_id} not found in form {form}"}, ensure_ascii=False), "application/json"); return
  1107. folder_name = f"{form}_{_short_case(case_id)}"
  1108. out_dir = q_dir / "procedures" / folder_name / "_model" # 大模型工序独立子目录,不覆盖 Agent 工序
  1109. backup_procedure_history(out_dir)
  1110. out_dir.mkdir(parents=True, exist_ok=True)
  1111. src_path = out_dir / "_source.json"
  1112. try:
  1113. with open(src_path, "w", encoding="utf-8") as f:
  1114. json.dump(_source_to_dsl_input(matching_result), f, ensure_ascii=False, indent=2)
  1115. _write_meta(out_dir, case_id=case_id, from_q=q, form=form,
  1116. score=_composite_score(matching_result.get("llm_evaluation") or {}))
  1117. version = _gen_version()
  1118. _stamp_version(out_dir, version)
  1119. except Exception as e:
  1120. self._send(500, json.dumps({"error": f"failed to write inputs: {e}"}, ensure_ascii=False), "application/json"); return
  1121. task_key = f"{q}/{folder_name}/_model"
  1122. ACTIVE_TASKS[task_key] = {"status": "running", "start_time": datetime.now().isoformat(), "pid": None, "error": None}
  1123. t = threading.Thread(target=run_model_extraction_task, args=(q, folder_name, case_id, src_path, out_dir, model, version))
  1124. t.daemon = True
  1125. t.start()
  1126. self._send(200, json.dumps({
  1127. "status": "started", "task_key": task_key,
  1128. "log": f"{RUNS_DIR_NAME}/{q}/procedures/{folder_name}/_model/_extract.log"
  1129. }, ensure_ascii=False), "application/json")
  1130. elif self.path == "/api/reeval":
  1131. length = int(self.headers.get("Content-Length") or 0)
  1132. raw = self.rfile.read(length).decode("utf-8") if length > 0 else "{}"
  1133. try:
  1134. payload = json.loads(raw)
  1135. except Exception as e:
  1136. self._send(400, json.dumps({"error": f"bad json: {e}"}), "application/json"); return
  1137. q = (payload.get("q") or "").strip()
  1138. if MODE != "new" and not re.match(r"^q\d+$", q):
  1139. self._send(400, json.dumps({"error": f"bad q (expect 'qNN'): {q!r}"},
  1140. ensure_ascii=False), "application/json"); return
  1141. q_dir = RUNS_DIR / q
  1142. if not q_dir.is_dir():
  1143. self._send(404, json.dumps({"error": f"{RUNS_DIR_NAME}/{q} not found"}, ensure_ascii=False),
  1144. "application/json"); return
  1145. backup_reeval_history(q_dir)
  1146. log_path = q_dir / "_reeval.log"
  1147. try:
  1148. log_fh = open(log_path, "w", encoding="utf-8", buffering=1)
  1149. cmd = [sys.executable, "-u", str(SE / "batch_3forms.py"),
  1150. "--reeval", "--reeval-q", q, "--output-dir", str(RUNS_DIR)]
  1151. flags = subprocess.CREATE_NEW_PROCESS_GROUP if sys.platform == "win32" else 0
  1152. proc = subprocess.Popen(cmd, stdout=log_fh, stderr=subprocess.STDOUT,
  1153. cwd=str(SE), creationflags=flags)
  1154. ACTIVE_REEVALS[q] = {
  1155. "status": "running",
  1156. "pid": proc.pid,
  1157. "error": None
  1158. }
  1159. def wait_reeval(q_key, p_obj, fh):
  1160. try:
  1161. p_obj.wait()
  1162. if p_obj.returncode == 0:
  1163. ACTIVE_REEVALS[q_key]["status"] = "success"
  1164. else:
  1165. ACTIVE_REEVALS[q_key]["status"] = "failed"
  1166. ACTIVE_REEVALS[q_key]["error"] = f"Subprocess exited with code {p_obj.returncode}"
  1167. except Exception as ex:
  1168. ACTIVE_REEVALS[q_key]["status"] = "failed"
  1169. ACTIVE_REEVALS[q_key]["error"] = str(ex)
  1170. finally:
  1171. try:
  1172. fh.close()
  1173. except Exception:
  1174. pass
  1175. t = threading.Thread(target=wait_reeval, args=(q, proc, log_fh))
  1176. t.daemon = True
  1177. t.start()
  1178. self._send(200, json.dumps(
  1179. {"status": "started", "pid": proc.pid, "q": q,
  1180. "log": str(log_path.relative_to(HERE))},
  1181. ensure_ascii=False), "application/json")
  1182. except Exception as e:
  1183. self._send(500, json.dumps({"error": f"failed to start: {e}"},
  1184. ensure_ascii=False), "application/json")
  1185. elif self.path == "/api/batch_generate_procedure":
  1186. length = int(self.headers.get("Content-Length") or 0)
  1187. raw = self.rfile.read(length).decode("utf-8") if length > 0 else "{}"
  1188. try:
  1189. payload = json.loads(raw)
  1190. except Exception as e:
  1191. self._send(400, json.dumps({"error": f"bad json: {e}"}), "application/json"); return
  1192. q = (payload.get("q") or "").strip()
  1193. form = (payload.get("form") or "A").strip()
  1194. engine = (payload.get("engine") or "cyber_runner").strip()
  1195. model = (payload.get("model") or "google/gemini-3.1-flash-lite").strip()
  1196. try:
  1197. concurrency = int(payload.get("concurrency", 4))
  1198. except Exception:
  1199. concurrency = 4
  1200. concurrency = max(1, min(16, concurrency))
  1201. if not q:
  1202. self._send(400, json.dumps({"error": "missing q"}, ensure_ascii=False), "application/json"); return
  1203. q_dir = RUNS_DIR / q
  1204. form_file = q_dir / f"form_{form}.json"
  1205. if not form_file.is_file():
  1206. self._send(404, json.dumps({"error": f"form file not found: {form_file.name}"}, ensure_ascii=False), "application/json"); return
  1207. try:
  1208. with open(form_file, encoding="utf-8") as f:
  1209. form_data = json.load(f)
  1210. except Exception as e:
  1211. self._send(500, json.dumps({"error": f"failed to read form: {e}"}, ensure_ascii=False), "application/json"); return
  1212. valid_results = []
  1213. import copy
  1214. for r in form_data.get("results", []):
  1215. adapted = adapt(r, q, form)
  1216. if adapted.get("decision") == "report" and not adapted.get("anomaly"):
  1217. r_copy = copy.deepcopy(r)
  1218. original_case_id = r.get("case_id", "")
  1219. r_copy["case_id"] = f"{form}_{_short_case(original_case_id)}"
  1220. valid_results.append(r_copy)
  1221. if not valid_results:
  1222. self._send(400, json.dumps({"error": "没有可提取工序的帖子(所有帖子都已被过滤,或都存在解析异常)"}, ensure_ascii=False), "application/json"); return
  1223. procedures_dir = q_dir / "procedures"
  1224. procedures_dir.mkdir(parents=True, exist_ok=True)
  1225. batch_posts_path = procedures_dir / "temp_batch.json"
  1226. try:
  1227. with open(batch_posts_path, "w", encoding="utf-8") as f:
  1228. json.dump({"results": valid_results}, f, ensure_ascii=False, indent=2)
  1229. except Exception as e:
  1230. self._send(500, json.dumps({"error": f"failed to write temp_batch.json: {e}"}, ensure_ascii=False), "application/json"); return
  1231. ACTIVE_BATCH_TASKS[q] = {
  1232. "status": "running",
  1233. "start_time": datetime.now().isoformat(),
  1234. "pid": None,
  1235. "error": None
  1236. }
  1237. def run_batch_extraction():
  1238. try:
  1239. log_path = procedures_dir / "_batch_extract.log"
  1240. cmd = [
  1241. sys.executable, "-u", str(HERE / "procedure-dsl" / "run_cyber.py"),
  1242. str(batch_posts_path),
  1243. "--out-dir", str(procedures_dir),
  1244. "--batch",
  1245. "--batch-workers", str(concurrency),
  1246. "--model", model,
  1247. "--skill", "procedure", # mode_procedure:用直写版 skill
  1248. "--max-turns", "300"
  1249. ]
  1250. flags = subprocess.CREATE_NEW_PROCESS_GROUP if sys.platform == "win32" else 0
  1251. with open(log_path, "w", encoding="utf-8", buffering=1) as log_fh:
  1252. _log_header(log_fh, "🔧 工序提取(批量 --skill procedure)", model, engine)
  1253. proc = subprocess.Popen(cmd, stdout=log_fh, stderr=subprocess.STDOUT,
  1254. cwd=str(HERE / "procedure-dsl"), creationflags=flags) # cwd=本地 procedure-dsl(run_cyber 假设的工作目录,agent 相对路径才对)
  1255. ACTIVE_BATCH_TASKS[q]["pid"] = proc.pid
  1256. proc.wait()
  1257. if proc.returncode == 0:
  1258. try:
  1259. import build_workflows
  1260. build_workflows.write_run(q, runs_dir=RUNS_DIR)
  1261. ACTIVE_BATCH_TASKS[q]["status"] = "success"
  1262. except Exception as ex:
  1263. ACTIVE_BATCH_TASKS[q]["status"] = "failed"
  1264. ACTIVE_BATCH_TASKS[q]["error"] = f"Workflows compilation failed: {ex}"
  1265. else:
  1266. ACTIVE_BATCH_TASKS[q]["status"] = "failed"
  1267. ACTIVE_BATCH_TASKS[q]["error"] = f"Batch runner exited with code {proc.returncode}"
  1268. except Exception as ex:
  1269. ACTIVE_BATCH_TASKS[q]["status"] = "failed"
  1270. ACTIVE_BATCH_TASKS[q]["error"] = str(ex)
  1271. finally:
  1272. try:
  1273. if batch_posts_path.is_file():
  1274. batch_posts_path.unlink()
  1275. except Exception:
  1276. pass
  1277. t = threading.Thread(target=run_batch_extraction)
  1278. t.daemon = True
  1279. t.start()
  1280. self._send(200, json.dumps({
  1281. "status": "started",
  1282. "q": q,
  1283. "log": f"{RUNS_DIR_NAME}/{q}/procedures/_batch_extract.log"
  1284. }, ensure_ascii=False), "application/json")
  1285. elif self.path == "/api/save_spec":
  1286. length = int(self.headers.get("Content-Length") or 0)
  1287. raw = self.rfile.read(length).decode("utf-8") if length > 0 else "{}"
  1288. try:
  1289. payload = json.loads(raw)
  1290. except Exception as e:
  1291. self._send(400, json.dumps({"error": f"bad json: {e}"}), "application/json"); return
  1292. file_name = (payload.get("file") or "").strip()
  1293. content = payload.get("content") or ""
  1294. allowed = [
  1295. "README.md",
  1296. "tools.md",
  1297. "extraction/phase1-skeleton.md",
  1298. "extraction/phase2-normalize.md",
  1299. "extraction/phase3-finalize.md",
  1300. "taxonomy/type_suggestions.md"
  1301. ]
  1302. if file_name not in allowed:
  1303. self._send(400, json.dumps({"error": "invalid file parameter"}), "application/json"); return
  1304. target_path = HERE / "procedure-dsl" / "spec" / file_name
  1305. try:
  1306. target_path.parent.mkdir(parents=True, exist_ok=True)
  1307. target_path.write_text(content, encoding="utf-8")
  1308. self._send(200, json.dumps({"status": "ok"}, ensure_ascii=False), "application/json")
  1309. except Exception as e:
  1310. self._send(500, json.dumps({"error": str(e)}, ensure_ascii=False), "application/json")
  1311. else:
  1312. self._send(404, json.dumps({"error": "not found"}), "application/json")
  1313. def log_message(self, *a): pass
  1314. if __name__ == "__main__":
  1315. n = len(scan_runs()["queries"])
  1316. print(f"搜索评估查看 server ({MODE} 模式):http://0.0.0.0:{PORT} ({RUNS_DIR_NAME}/ 下 {n} 个 query,实时扫描)")
  1317. ThreadingHTTPServer(("0.0.0.0", PORT), H).serve_forever()