gjm
6143bb9ab2
支持直接审查 PDF 与位图图纸，全链路离线（不连 CAD、不连 8080 端口）。 渲染在内存内完成，复用既有墨迹热力图 + 自适应瓦片规划。 - 新增 dwg_review/sources.py - detect_source / plan_file / collect 统一入口 - PdfReader：pypdfium2 惰性导入，页序即图框序，处理 /Rotate 与 crop 边距 - ImageReader：世界单位 = 源像素，snap_box 整数对齐保证窗口与像素严格一致 - ink_grid：纯 PIL 灰度阈值 → 占用网格（不引入 numpy） - ReviewContext.create(connect_cad=False)：离线时 client/tools 均为 None - 全链路 ctx.tools is None 守卫：review 只读子循环 / inspect_region / annotator / checks 的窗口类检查 / 绘图区最大化 - checklist.build_system_prompt(allow_tools=False, offline=True)：离线提示词 - __main__：位置参数 source，按扩展名自动判定源类型并决定是否连 CAD - config 新增 8 个常量（PDF_MAX_PAGES / 文本最小字数 / 各源缩放上限 / 墨迹阈值等） - tiling.content_predicate 与 heatmap.cell_box 提升为公开函数 验证： - pytest 457 passed - verify_pdf_source.py：PDF 23/23、图片 9/9（不需 CAD），含实测瓦片字高像素与 旋转映射自检（文本区墨迹 0.311 vs 忽略 /Rotate 反向对照 0.145） - 真实 LLM 端到端：20 页 HVAC 图纸前 3 页跑通，结论具体可核验且未编造 handle 已知局限：页边线在墨迹网格里会把整页连成单簇，瓦片常退化为整页象限（CAD 侧 靠 drop_oversized 丢图框块，离线侧暂无等价手段），留待后续打磨。
80 lines
2.7 KiB
Python
80 lines
2.7 KiB
Python
"""阶段 4：汇总报告 + 落盘。
|
|
|
|
审查结论在阶段 3 已是结构化对象（`{summary, issues[]}`），
|
|
因此这里**不再**需要第二次 LLM 调用来抽取结构。
|
|
"""
|
|
|
|
import json
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
|
|
from .. import config
|
|
from ..findings import flatten_issues
|
|
|
|
_SUMMARY_SYSTEM = (
|
|
"你是图纸审查负责人。请把下面各图框的审查意见汇总为一份结构化的审查报告："
|
|
"先给整体结论，再按图框列出问题，最后给出整改建议。语言简洁、条目化。"
|
|
)
|
|
|
|
|
|
def format_finding(finding):
|
|
"""把一个 findings 元素渲染成给汇总 LLM 的文本块。"""
|
|
lines = [f"【图框 {finding.get('frame')}（图幅 {finding.get('sheet') or '—'}）】"]
|
|
summary = finding.get("summary")
|
|
if summary:
|
|
lines.append(str(summary))
|
|
for issue in finding.get("issues") or []:
|
|
tag = "确定性" if issue.get("source") == "deterministic" else "LLM"
|
|
flag = "" if issue.get("verified", True) else "（未在证据中定位，仅供参考）"
|
|
handle = issue.get("handle")
|
|
where = f" handle={handle}" if handle else ""
|
|
lines.append(
|
|
f"- [{issue.get('severity', '?')}][{tag}] {issue.get('message', '')}{where}{flag}"
|
|
)
|
|
return "\n".join(lines)
|
|
|
|
|
|
def summarize(ctx, findings, log=print):
|
|
"""用 LLM 把各图框结论汇总为总报告。"""
|
|
if not findings:
|
|
return "未识别到可审查的图框，或所有图框均无有效包围盒。"
|
|
|
|
report_input = "\n\n".join(format_finding(f) for f in findings)
|
|
try:
|
|
return ctx.llm.chat_text(_SUMMARY_SYSTEM, report_input)
|
|
except Exception as exc: # noqa: BLE001
|
|
log(f"[汇总失败] {exc}")
|
|
return report_input
|
|
|
|
|
|
def save_report(markdown, findings, plan, issues=None, log=print):
|
|
"""落盘 Markdown + JSON 报告，返回两个路径。
|
|
|
|
`issues` 省略时从 findings 里拍平派生（供批注回写）。
|
|
"""
|
|
if issues is None:
|
|
issues = flatten_issues(findings)
|
|
|
|
out_dir = Path(config.REPORT_DIR)
|
|
out_dir.mkdir(parents=True, exist_ok=True)
|
|
stamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
|
md_path = out_dir / f"review_{stamp}.md"
|
|
json_path = out_dir / f"review_{stamp}.json"
|
|
|
|
md_path.write_text(markdown, encoding="utf-8")
|
|
json_path.write_text(
|
|
json.dumps(
|
|
{
|
|
"generated_at": stamp,
|
|
"findings": findings,
|
|
"issues": issues,
|
|
"frames": plan.get("frames"),
|
|
},
|
|
ensure_ascii=False,
|
|
indent=2,
|
|
),
|
|
encoding="utf-8",
|
|
)
|
|
log(f"[报告] 已写入 {md_path} 与 {json_path}")
|
|
return md_path, json_path
|