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 丢图框块，离线侧暂无等价手段），留待后续打磨。
59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
"""审查运行的共享上下文。"""
|
|
|
|
from dataclasses import dataclass
|
|
|
|
from . import config, evidence_model
|
|
from .cad_tools import CadTools
|
|
from .checklist import load_checklist
|
|
from .llm_client import LlmClient
|
|
from .mcp_client import McpClient
|
|
|
|
|
|
@dataclass
|
|
class ReviewContext:
|
|
"""一次审查运行的共享上下文。
|
|
|
|
`connect_cad=False`（离线模式）时 `client` / `tools` 均为 `None`——供 PDF / 图片
|
|
等**纯离线源**使用：这些源没有 CAD 可连，全链路（取证 / 审查 / 批注）都必须
|
|
先判 `ctx.tools is None` 再决定是否走 CAD 工具。`source` 记录本次运行的来源类型，
|
|
供规划阶段分发（CAD 走 `detect_drawing_frames`，文件源自带图框）。
|
|
|
|
`source_path` 为文件源（PDF / 图片）的磁盘路径；`source_reader` 是 `sources`
|
|
打开的读取器（缓存，避免每页重复打开 PDF）。
|
|
"""
|
|
|
|
client: McpClient
|
|
tools: CadTools
|
|
llm: LlmClient
|
|
checklist: dict
|
|
source: str = evidence_model.SOURCE_CAD
|
|
source_path: str = None
|
|
source_reader: object = None
|
|
max_frames: int = 0
|
|
frame_limit: int = 200
|
|
max_iters: int = 4
|
|
annotate: bool = False
|
|
dry_run: bool = False
|
|
|
|
@classmethod
|
|
def create(cls, *, need_llm=True, connect_cad=True, source=None, source_path=None,
|
|
**overrides):
|
|
client = McpClient() if connect_cad else None
|
|
# CadTools.__init__ 会立刻 list_tools() 去连 127.0.0.1:8080；离线模式必须跳过。
|
|
tools = CadTools(client) if connect_cad else None
|
|
llm = LlmClient() if need_llm else None
|
|
checklist = load_checklist()
|
|
params = dict(
|
|
client=client,
|
|
tools=tools,
|
|
llm=llm,
|
|
checklist=checklist,
|
|
source=source or evidence_model.SOURCE_CAD,
|
|
source_path=source_path,
|
|
max_frames=config.MAX_FRAMES,
|
|
frame_limit=config.FRAME_ENTITY_LIMIT,
|
|
max_iters=config.REVIEW_MAX_ITERS,
|
|
)
|
|
params.update(overrides)
|
|
return cls(**params)
|