cad-agent/dwg_review/checklist.py

"""审查清单加载 + 系统提示构建（清单驱动）。"""

import json

from . import config

DEFAULT_CHECKLIST = {
    "name": "内置通用清单",
    "categories": [
        {"title": "图层规范性", "severity": "medium",
         "checks": ["图元是否在正确图层", "是否存在空图层/命名异常图层"]},
        {"title": "文字", "severity": "medium",
         "checks": ["空文字", "错别字", "字高异常", "文字重叠"]},
        {"title": "标注", "severity": "high",
         "checks": ["关键尺寸缺失", "标注值异常"]},
        {"title": "几何", "severity": "medium",
         "checks": ["零长度/退化实体", "重复重叠实体", "未闭合轮廓"]},
        {"title": "干涉/碰撞", "severity": "high",
         "checks": ["构件/管线间非预期侵入", "同位置重复叠画", "引线/填充异常遮挡"]},
        {"title": "图面整体", "severity": "low",
         "checks": ["图面完整性", "内容是否超出图框"]},
    ],
}


def load_checklist(path=None):
    """加载审查清单 JSON；失败时回退内置默认清单。"""
    path = path or config.CHECKLIST_PATH
    try:
        with open(path, "r", encoding="utf-8") as f:
            data = json.load(f)
        if isinstance(data, dict) and data.get("categories"):
            return data
        print(f"[Warn] 审查清单 {path} 结构异常，使用内置默认清单。")
    except Exception as exc:  # noqa: BLE001
        print(f"[Warn] 无法加载审查清单 {path}: {exc}，使用内置默认清单。")
    return DEFAULT_CHECKLIST


def build_system_prompt(checklist, *, allow_tools=True):
    """根据审查清单动态生成 LLM 系统提示词。"""
    lines = [
        "你是一名资深建筑与工业厂房图纸审查专家。",
        "你将逐张收到某个图框（一张图纸）的：结构化图元数据、文字内容，以及该图框的视口截图。",
        "",
        f"请依据审查清单《{checklist.get('name', '审查清单')}》逐项审查，"
        "指出**具体问题**（尽量引用实体 handle 或文字内容），并标注严重程度：",
        "",
    ]
    for cat in checklist.get("categories", []):
        lines.append(f"【{cat.get('title', '')}】（严重度：{cat.get('severity', 'medium')}）")
        for chk in cat.get("checks", []):
            lines.append(f"  - {chk}")
        lines.append("")

    if allow_tools:
        lines += [
            "你可以调用**只读**取证工具进一步确认细节；**不要凭猜测下结论**。",
            "",
            "**看不清某处时，调用 `inspect_region` 指认位置**，它会自动按可读分辨率放大并返回清晰截图。",
            "`target` 三种指认方式任选其一：",
            '  - `{"handle": "3A4"}` —— 已知实体句柄；',
            '  - `{"label": "tile-3", "cx": 0.4, "cy": 0.6, "span": 0.2}` —— 某张已给截图上的一个位置',
            "    （`cx` 从左到右、`cy` **从上到下**，均为 0~1 归一化；`span` 为要看的范围占该截图的比例，默认 0.25）；",
            '  - `{"window": {"min_x": …, "min_y": …, "max_x": …, "max_y": …}}` —— 直接给世界坐标框。',
            '`purpose` 选 `"text"`（要读文字/标注，默认）或 `"line"`（只看走向/连接，覆盖更广）。',
            "返回的截图会带 `inspect-N` 标签，可继续作为 `label` 指认更细的位置（逐级下钻）。",
            "**你不需要判断截图够不够清楚**——`inspect_region` 会保证返回的图达到可读分辨率；",
            "若它提示区域过大，请缩小 `target`（或用 `span` 缩小范围）后重试。",
            "",
            "结构化数据不足以判断时：用 `get_entity_detail` / `find_duplicate_entities` /",
            "`find_zero_length` / `find_collisions` / `get_layers` 等查证后再下结论。",
            "排查「干涉/碰撞」时，先调 `find_collisions` 拿到**包围盒相交候选**"
            "（按重叠比降序，`min_overlap` 可调阈值），",
            "再对可疑的候选 handle 用 `inspect_region` 放大目视确认——"
            "**AABB 相交不等于真干涉**，务必核实后再下结论。",
            "",
        ]

    lines += [
        "要求：",
        "- 只报告你**有依据**的问题，不要臆测；数据与截图冲突时以数据为准。",
        "- 按清单条目组织结论，未命中的条目可省略。",
        "- 不要复述输入数据。",
        "",
        "输出格式（**必须严格遵守**）：",
        "在你完成全部取证、准备给出该图框的最终结论时，**只输出一个 JSON 对象**，"
        "不要输出任何解释文字，也不要用 markdown 代码块包裹。结构如下：",
        "{",
        '  "summary": "<该图框的一句话整体结论；若未发现问题，写“未发现明显问题”>",',
        '  "issues": [',
        "    {",
        '      "category": "<图层/文字/标注/几何/干涉/图面整体 之一>",',
        '      "severity": "<high|medium|low>",',
        '      "message": "<问题描述，简洁具体>",',
        '      "handle": "<该问题对应的实体句柄；无法定位到具体实体时省略此字段>"',
        "    }",
        "  ]",
        "}",
        "其中 `handle` 必须是你**在取证数据或工具返回里真实见过**的句柄，不要编造；"
        "无法确定具体实体时省略 `handle`，该问题仍会被记录。",
    ]
    if allow_tools:
        lines.append("（需要进一步确认时先调用工具；拿到结果后，最终再输出上述 JSON。）")
    return "\n".join(lines)


def checklist_summary(checklist):
    """用于规划阶段的一句话摘要。"""
    cats = checklist.get("categories", [])
    return f"{checklist.get('name', '清单')}（{len(cats)} 类：" + \
        "、".join(c.get("title", "") for c in cats) + "）"