|
|
@@ -0,0 +1,159 @@
|
|
|
+# -*- coding: utf-8 -*-
|
|
|
+r"""问答管线（P12）：从详情层 `serve.py` 搬来的 `_ask_worker`/`_ask_step_text` 与相关常量。
|
|
|
+
|
|
|
+★搬运改动（仅三处，其余逐字保留）：
|
|
|
+ 1. `_ask_worker` 加参数 `st`（详情层是"线程 + 全局 ASK"，这里改"同步 + 传入状态字典"）；
|
|
|
+ 2. 体内 `ASK['x']` → `st['x']`（`_on_step` 在 worker 内定义，闭包自然拿到同一个 st）；
|
|
|
+ 3. 对外只暴露 `ask_run(q, model, cat, xrev, lang)`：建 st、按 `ask_start` 的 hints 逻辑拼词、跑管线、收尾 state。
|
|
|
+"""
|
|
|
+from __future__ import annotations
|
|
|
+
|
|
|
+import pathlib
|
|
|
+import sys
|
|
|
+from typing import Any
|
|
|
+
|
|
|
+from app_common.app_common_guanlan.api import paths as _P
|
|
|
+
|
|
|
+ASK_CAT_HINTS = {
|
|
|
+ '设备类': '(设备状态类: 从本体的机组档案、对象查询与机制检索三条路取证)',
|
|
|
+ '检修排程决策': '(检修排程类: 用 ont_scenario/ont_get 取排期沙盘与验收树; 建议级动作注明待人裁)',
|
|
|
+ '损失分析': '(损失分析类: 用 ont_loss/ont_scenario 取证; 电量kWh为主, 金额必带电价区间0.391~0.85口径; 限电类损失≠设备责任)',
|
|
|
+ '安全风险预测': '(风险预测类: 用 ont_risk/ont_mechanism_search 取证; 区分升级中/稳定高位/盲区, 未列≠正常, 不可判≠阴性)',
|
|
|
+}
|
|
|
+
|
|
|
+ASK_CAT_HINTS_EN = {
|
|
|
+ '设备类': '(Equipment state: use unit profiles, object queries and mechanism search for evidence)',
|
|
|
+ '检修排程决策': '(Maintenance scheduling: use scenarios and acceptance trees; mark advisory actions as pending human decision)',
|
|
|
+ '损失分析': '(Loss analysis: use loss accounts and scenarios; energy in kWh/MWh, monetary value must carry the tariff-range caveat)',
|
|
|
+ '安全风险预测': '(Risk prediction: distinguish escalating, stable-high and blind-spot cases; no listing is not proof of normality)',
|
|
|
+}
|
|
|
+
|
|
|
+MODEL_SPEC = {
|
|
|
+ 'qwen3:8b': dict(nm='轻量 8B', loc='本地', t='约 30 到 60 秒', use='默认档: 快速简答问答 (用户裁 2026-09-07)'),
|
|
|
+ 'qwen3.8:27b': dict(nm='标准 27B', loc='本地', t='约 2 到 4 分钟', use='复杂问题档: 多台横向比对与机制推断 (用户裁 2026-09-07)'),
|
|
|
+ 'qwen3:30b-a3b': dict(nm='快 30B (a3b)', loc='本地', t='约 20 到 60 秒', use='8B 校闸未过时的自动升档目标; 也可直接选'),
|
|
|
+ 'deepseek-r1:14b': dict(nm='本机 DeepSeek 14B', loc='本地', t='约 1 到 3 分钟', use='交叉审核审核档 (作答请优先 qwen 27B; 用户裁 2026-09-07)'),
|
|
|
+ 'deepseek-r1:32b': dict(nm='本机 DeepSeek 32B', loc='本地', t='约 3 到 8 分钟', use='备用: 更强更慢, 引用格式不稳 (约三成概率不出文)'),
|
|
|
+ 'deepseek-chat': dict(nm='DeepSeek', loc='云端', t='约 3 到 10 秒', use='默认档, 状态查询与机制推断',
|
|
|
+ warn='内容会外发到模型厂商, 场名与台号已去标识化'),
|
|
|
+ 'kimi-fast': dict(nm='Kimi', loc='云端', t='约 5 到 20 秒', use='长文归纳',
|
|
|
+ warn='内容会外发到模型厂商, 场名与台号已去标识化'),
|
|
|
+ 'qwen-plus': dict(nm='通义千问', loc='云端', t='约 5 到 20 秒', use='备用云端档',
|
|
|
+ warn='内容会外发到模型厂商, 场名与台号已去标识化'),
|
|
|
+ 'deepseek-v4-pro': dict(nm='云端 DS', loc='云端', t='约 2 到 6 分钟', use='复杂推理与交叉审核',
|
|
|
+ warn='内容会外发到模型厂商, 场名与台号已去标识化'),
|
|
|
+}
|
|
|
+
|
|
|
+
|
|
|
+def _ask_step_text(k, d='', lang='zh'):
|
|
|
+ k = '' if k is None else str(k)
|
|
|
+ d = '' if d is None else str(d)
|
|
|
+ if lang != 'en':
|
|
|
+ return f'{k}· {d}' if d else k
|
|
|
+ key = _ASK_STEP_KEY_EN.get(k, k)
|
|
|
+ detail = d.strip()
|
|
|
+ m = re.search(r'第\s*(\d+\s*/\s*\d+)\s*轮', detail)
|
|
|
+ if m:
|
|
|
+ detail = 'round ' + m.group(1).replace(' ', '')
|
|
|
+ else:
|
|
|
+ m = re.search(r'(\d{1,2})\s*一站式证据包', detail)
|
|
|
+ if m:
|
|
|
+ alias = 'focus unit'
|
|
|
+ try:
|
|
|
+ from src.windscada.deid import unitize_answer
|
|
|
+ alias = unitize_answer(f"WTG{int(m.group(1)):02d}")
|
|
|
+ except Exception:
|
|
|
+ pass
|
|
|
+ detail = f'{alias} evidence pack'
|
|
|
+ elif '闭环' in detail and '证据' in detail:
|
|
|
+ n = ''.join(re.findall(r'\d+', detail))
|
|
|
+ detail = f'closure and trend evidence: {n} records' if n else 'closure and trend evidence'
|
|
|
+ elif '全场' in detail and '判级' in detail:
|
|
|
+ n = ''.join(re.findall(r'\d+', detail))
|
|
|
+ detail = f'fleet ratings: {n} units' if n else 'fleet ratings'
|
|
|
+ elif '换模型' in detail or '复核' in detail:
|
|
|
+ detail = 'reviewing with another model'
|
|
|
+ elif re.search(r'[\u3400-\u9fff\u3000-\u303f\uff00-\uffef]', detail):
|
|
|
+ detail = ''
|
|
|
+ if re.search(r'[\u3400-\u9fff\u3000-\u303f\uff00-\uffef]', key):
|
|
|
+ key = 'Working'
|
|
|
+ if re.search(r'[\u3400-\u9fff\u3000-\u303f\uff00-\uffef]', detail):
|
|
|
+ detail = ''
|
|
|
+ return f'{key} · {detail}' if detail else key
|
|
|
+
|
|
|
+
|
|
|
+def _ask_worker(q, model, lang='zh', st=None):
|
|
|
+ import subprocess, threading, time as _t
|
|
|
+ out = _P.ont() / 'ask_last.out'
|
|
|
+ try:
|
|
|
+ if MODEL_SPEC.get(model, {}).get('loc') == '本地': # 本地档 (qwen3 / deepseek-r1) 走直连快路径; 原按 'qwen' 前缀硬判, 加 r1 档后改按档表 loc
|
|
|
+ # 本地模型走直连快路径 (fast_agent: ollama chat+tools 同进程, 砍 dsh/node/MCP 三层;
|
|
|
+ # 实测 8b 116s→19s; 同一套工具与闸, 服务端强制双闸). deepseek 云端仍走 dsh 完整编排.
|
|
|
+ threading.Thread(target=_warm, args=(model,), daemon=True).start()
|
|
|
+ sys.path.insert(0, str(pathlib.Path('.').resolve()))
|
|
|
+ sys.path.insert(0, str(pathlib.Path('scripts').resolve()))
|
|
|
+ from src.ontology.fast_agent import answer_escalate as _fast # 2026-09-07: 8B 拦了自动升 a3b 重答 (用户令)
|
|
|
+
|
|
|
+ def _on_step(k, d=''):
|
|
|
+ st['step'] = _ask_step_text(k, d, lang)
|
|
|
+ st.setdefault('steps', []).append(st['step'])
|
|
|
+
|
|
|
+ def _on_partial(txt):
|
|
|
+ # 先快后准 (用户裁 2026-09-07): 30 s 内先出初答 (契约直出), 状态 partial, 前端继续轮询; 27B 出来后整段替换
|
|
|
+ ASK.update(answer=txt, state='partial', partial=True)
|
|
|
+
|
|
|
+ r = _fast(q, model, on_step=_on_step, lang=lang, on_partial=_on_partial)
|
|
|
+ _esc = f" · 升档 {r['escalated_from']}→{r['model']}" if r.get('escalated_from') else ''
|
|
|
+ ans = r['answer'] + f"\n\n({r['gate']} · {r['rounds']}轮 · 直连快路径{_esc})"
|
|
|
+ out.write_text(ans, encoding='utf-8') # 答案是中文: 缺 encoding 在中文 Windows 上按 GBK 写, 可能崩
|
|
|
+ from ontology_p2_verify import verify as _v0
|
|
|
+ ASK.update(answer=ans, verify=_v0(ans, lang=lang), partial=False, refined_from=r.get('escalated_from'))
|
|
|
+ # 交叉审核: 换一个本地模型复核同一份证据 (同模型自审 = 自己给自己判卷)
|
|
|
+ if st.get('xrev'):
|
|
|
+ _on_step('交叉审核', '换模型复核中')
|
|
|
+ from src.ontology.fast_agent import cross_review as _xr
|
|
|
+ # 审核档 = DeepSeek r1:14b (用户裁 2026-09-07); 作答本身是 DeepSeek 时换 qwen 27B 审 —— 始终跨家族 (同家族自审 = 自己判卷)
|
|
|
+ _rm = 'deepseek-r1:14b' if not model.startswith('deepseek') else 'qwen3.8:27b'
|
|
|
+ st['review'] = _xr(q, r['answer'], r.get('facts') or [], model=_rm)
|
|
|
+ ASK.update(state='done')
|
|
|
+ return
|
|
|
+ if False:
|
|
|
+ q = '/no_think ' + q # (快路径内已带 /no_think; dsh 路径仅 deepseek)
|
|
|
+ cmd = ['deploy/dsh/dsh_run.sh', '--profile', 'headless',
|
|
|
+ '--patch', 'deploy/dsh/headless.patch.ontology.yml']
|
|
|
+ mp = ASK_MODELS.get(model)
|
|
|
+ if mp:
|
|
|
+ cmd += ['--patch', mp]
|
|
|
+ if lang == 'en':
|
|
|
+ cmd.append(q + ' Answer in English only. End every conclusion sentence with [object/id]. Finally run claim_check.')
|
|
|
+ else:
|
|
|
+ cmd.append(q + ' 每个结论句末尾用[对象id]标注来源。最后调用claim_check。')
|
|
|
+ # 2026-09-16: 走 src.proc.run —— 本函数跑在 detail 服务进程里 (无可见控制台), 裸 spawn
|
|
|
+ # 会让 Windows 给这条"云端档问答"新建可见控制台窗口 (本地档不走这里)。
|
|
|
+ from src import proc as _proc
|
|
|
+ r = _proc.run(cmd, capture_output=True, text=True, timeout=600)
|
|
|
+ ans = r.stdout.strip() or ('(空输出) stderr: ' + r.stderr[-500:])
|
|
|
+ out.write_text(ans, encoding='utf-8')
|
|
|
+ sys.path.insert(0, str(pathlib.Path('scripts').resolve()))
|
|
|
+ from ontology_p2_verify import verify as _v
|
|
|
+ ASK.update(answer=ans, verify=_v(ans, lang=lang), state='done')
|
|
|
+ except Exception as e:
|
|
|
+ _msg = str(e)
|
|
|
+ if 'Connection refused' in _msg or 'Errno 61' in _msg or '11434' in _msg:
|
|
|
+ _msg = '本机模型未启动 (Ollama 127.0.0.1:11434 拒连); 离线版不转云, 请先启动 Ollama' # 2026-09-07 真停实测: 原样抛 urlopen 错误给用户
|
|
|
+ ASK.update(state='error', err=_msg[:300])
|
|
|
+
|
|
|
+
|
|
|
+def ask_run(q: str, model: str = 'qwen3:8b', cat: str = '', xrev: bool = False, lang: str = 'zh') -> dict[str, Any]:
|
|
|
+ """同步跑一次问答，返回末态 dict（前端契约不变：状态由 Java 持有，客户端轮询 /api/ask_status）。"""
|
|
|
+ st: dict[str, Any] = dict(state='running', q=q, model=model, answer='', verify=None, err='',
|
|
|
+ step='启动' if lang != 'en' else 'Starting', steps=[], review=None,
|
|
|
+ xrev=bool(xrev), lang=lang)
|
|
|
+ hints = ASK_CAT_HINTS_EN if lang == 'en' else ASK_CAT_HINTS
|
|
|
+ q2 = (hints.get(cat, '') + ' ' + q).strip()
|
|
|
+ try:
|
|
|
+ _ask_worker(q2, model, lang, st)
|
|
|
+ except Exception as e: # noqa: BLE001
|
|
|
+ st['err'] = str(e)[:300]
|
|
|
+ st['state'] = 'error' if st.get('err') else 'done'
|
|
|
+ return st
|
zhouyang.xie