基座(cad_mcp_frame): - 新增 RecvOneFrame: 以 '\n' 为消息边界分帧,去掉单次 recv(2048) 限制(上限 8MB); 用 select 短探测兼容不发 '\n' 的旧客户端;clientSocket 设 SO_RCVTIMEO 防卸载卡死 - JSON-RPC id 支持 string/number;无 id 的通知不回包 - isError 映射: 协议级错误保留顶层 error,工具执行错误转 result.isError=true - 响应补 '\n' 作为消息边界 Agent(新增 dwg_review 包): - mcp_client: 唯一 TCP/JSON-RPC 封装(单飞锁/分帧/重试/兼容 error 与 isError) - cad_tools: 从 tools/list 动态建表,含 15 个只读白名单 - phases: planning -> evidence -> review(受控只读取证子循环)-> report - annotator: 批注回写 AI_REVIEW 层,支持 dry-run 与撤销 - config: 配置全部走环境变量;Windows 下回退读注册表以读到新设的系统变量 其它: 新增 verify_mcp_frame.py 验证脚本;.gitignore 补 Python/AutoCAD 锁文件规则; ai.memory 记录架构决策与实测结论
76 lines
3.1 KiB
Python
76 lines
3.1 KiB
Python
"""审查清单加载 + 系统提示构建(清单驱动)。"""
|
||
|
||
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": "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 += [
|
||
"你可以调用**只读**取证工具进一步确认细节(例如放大局部、查实体详情、查重复实体)。",
|
||
"当结构化数据不足以判断时,先调用工具取证,再下结论。",
|
||
"",
|
||
]
|
||
|
||
lines += [
|
||
"要求:",
|
||
"- 只报告你**有依据**的问题,不要臆测;数据与截图冲突时以数据为准。",
|
||
"- 按清单条目组织结论,未命中的条目可省略。",
|
||
"- 若该图框无任何明显问题,明确说“未发现明显问题”。",
|
||
"- 不要复述输入数据。",
|
||
]
|
||
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) + ")"
|