feat(review): PDF/图片离线源适配(C1)—— 无需 CAD 的整条审查链路
支持直接审查 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 丢图框块,离线侧暂无等价手段),留待后续打磨。
This commit is contained in:
1 parent
d4a9442e30
commit
6143bb9ab2
32 files changed
+2617
-77
No files matched your search
+22
-4
@@ -2,7 +2,7 @@
|
||||
|
||||
from dataclasses import dataclass
|
||||
|
||||
from . import config
|
||||
from . import config, evidence_model
|
||||
from .cad_tools import CadTools
|
||||
from .checklist import load_checklist
|
||||
from .llm_client import LlmClient
|
||||
@@ -11,10 +11,24 @@ 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
|
||||
@@ -22,9 +36,11 @@ class ReviewContext:
|
||||
dry_run: bool = False
|
||||
|
||||
@classmethod
|
||||
def create(cls, *, need_llm=True, **overrides):
|
||||
client = McpClient()
|
||||
tools = CadTools(client)
|
||||
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(
|
||||
@@ -32,6 +48,8 @@ class ReviewContext:
|
||||
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,
|
||||
|
||||
Reference in new issue
Block a user