落地报价模板 AI 解析与对象存储，并增加回复超时系统设置页。

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-09-18 17:42:46 +08:00
parent 982b3a2317
commit 527050bbf9
41 changed files with 4965 additions and 426 deletions
@@ -0,0 +1,122 @@
+# -*- coding: utf-8 -*-
+"""
+联调冒烟：用桌面海运样表抽单元格清单 → 真网千问 sheet_map。
+不经主账 HTTP（本机 8180 未起）；验证「提示词+模型」能否产出合法 mappingJson。
+"""
+from __future__ import annotations
+
+import json
+import os
+import sys
+from pathlib import Path
+
+ROOT = Path(__file__).resolve().parents[1]
+DESK = Path(r"C:/Users/Administrator/Desktop/海陆空报价单模板最新-0917/海陆空报价单模板最新-0917")
+
+
+def load_dotenv(path: Path) -> None:
+    if not path.is_file():
+        return
+    for line in path.read_text(encoding="utf-8", errors="replace").splitlines():
+        s = line.strip()
+        if not s or s.startswith("#") or "=" not in s:
+            continue
+        k, v = s.split("=", 1)
+        os.environ.setdefault(k.strip(), v.strip().strip('"').strip("'"))
+
+
+def build_inventory_openpyxl(xlsx: Path, max_lines: int = 1800) -> dict:
+    import openpyxl
+
+    wb = openpyxl.load_workbook(xlsx, data_only=False)
+    ws = wb.active
+    yellow_approx = 0
+    lines: list[str] = []
+    merges = {}
+    for rng in ws.merged_cells.ranges:
+        origin = f"{openpyxl.utils.get_column_letter(rng.min_col)}{rng.min_row}"
+        merges[origin] = str(rng)
+    for row in ws.iter_rows(max_row=120, max_col=20):
+        for cell in row:
+            val = cell.value
+            text = "" if val is None else str(val).strip()
+            fill = ""
+            try:
+                fg = cell.fill.fgColor
+                if fg and fg.rgb and isinstance(fg.rgb, str) and fg.rgb not in ("00000000", "0"):
+                    fill = fg.rgb[-6:].upper()
+                    if fill.startswith("FF") and len(fg.rgb) >= 8:
+                        fill = fg.rgb[-6:].upper()
+                    if fill in {"FFFF00", "FFEB9C", "FFC000", "FFFF99", "FFCC00", "FFD966", "FFF2CC", "FFE699"} or (
+                        len(fill) == 6 and int(fill[0:2], 16) > 200 and int(fill[2:4], 16) > 180 and int(fill[4:6], 16) < 160
+                    ):
+                        yellow_approx += 1
+            except Exception:
+                pass
+            addr = cell.coordinate
+            merge = merges.get(addr, "-")
+            if not text and merge == "-" and not fill:
+                continue
+            safe = text.replace("|", "/")[:120]
+            lines.append(f"{ws.title}!{addr} | {safe} | {merge} | {fill or '-'}")
+            if len(lines) >= max_lines:
+                break
+        if len(lines) >= max_lines:
+            break
+    return {
+        "sheet": ws.title,
+        "cellInventory": lines,
+        "cellCount": len(lines),
+        "yellowCount": yellow_approx,
+        "fileName": xlsx.name,
+        "transportMode": "sea",
+        "versionId": "smoke-local",
+    }
+
+
+def main() -> int:
+    load_dotenv(ROOT.parent / ".env")
+    # 本脚本强制允许出站，否则 Worker 配置会挡掉真网
+    os.environ["LLM_ALLOW_NETWORK"] = "true"
+    os.environ["LLM_DATA_USAGE_CONFIRMED"] = "1"
+    sys.path.insert(0, str(ROOT))
+
+    sea = DESK / "海运报价单模板.xlsx"
+    if not sea.is_file():
+        print("FAIL missing", sea)
+        return 2
+    payload = build_inventory_openpyxl(sea, max_lines=800)
+    print(
+        f"inventory sheet={payload['sheet']} cells={payload['cellCount']} yellow≈{payload['yellowCount']}",
+        flush=True,
+    )
+
+    from agent.llm.mode_sheet_map import invoke_sheet_map
+
+    print("calling qwen sheet_map …", flush=True)
+    result = invoke_sheet_map(payload, allow_network=True, timeout_seconds=90.0)
+    if not result.get("ok"):
+        print("FAIL AI", result.get("error"))
+        if result.get("raw_preview"):
+            print("raw_preview", result["raw_preview"][:300])
+        return 1
+    mapping = result["mapping"]
+    fields = mapping.get("fields") or {}
+    fees = mapping.get("feeRegions") or []
+    routes = mapping.get("routeRegions") or []
+    print("OK AI mapping")
+    print("  transportMode=", mapping.get("transportMode"))
+    print("  layoutType=", mapping.get("layoutType"))
+    print("  fields=", len(fields), sorted(fields.keys())[:12])
+    print("  feeRegions=", len(fees))
+    print("  routeRegions=", len(routes))
+    print("  unmappedRequired=", mapping.get("unmappedRequired"))
+    print("  confidence=", mapping.get("confidence"))
+    out = ROOT / "scripts" / "_last_sea_mapping.json"
+    out.write_text(json.dumps(mapping, ensure_ascii=False, indent=2), encoding="utf-8")
+    print("saved", out)
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())
@@ -0,0 +1,71 @@
+# -*- coding: utf-8 -*-
+"""本地冒烟：sheet_map 解析 + 桌面 8 张样表可读。不打千问。"""
+from __future__ import annotations
+
+import importlib.util
+import sys
+import types
+from pathlib import Path
+
+ROOT = Path(__file__).resolve().parents[1]
+DESK = Path(r"C:/Users/Administrator/Desktop/海陆空报价单模板最新-0917/海陆空报价单模板最新-0917")
+
+
+def load_sheet_map():
+    llm_mode = types.ModuleType("agent.llm.modes")
+
+    class LlmMode:
+        class SHEET_MAP:
+            value = "sheet_map"
+
+    llm_mode.LlmMode = LlmMode
+    sys.modules.setdefault("agent", types.ModuleType("agent"))
+    sys.modules.setdefault("agent.llm", types.ModuleType("agent.llm"))
+    sys.modules["agent.llm.modes"] = llm_mode
+    path = ROOT / "agent" / "llm" / "mode_sheet_map.py"
+    spec = importlib.util.spec_from_file_location("mode_sheet_map", path)
+    mod = importlib.util.module_from_spec(spec)
+    assert spec.loader is not None
+    spec.loader.exec_module(mod)
+    return mod
+
+
+def main() -> None:
+    mod = load_sheet_map()
+    out = mod.parse_mapping_json('{"fields":{"quote_no":"S!A1"},"feeRegions":[]}')
+    assert out["ok"], out
+    fenced = mod.strip_json_fence('```json\n{"a":1}\n```')
+    assert '"a"' in fenced or "a" in fenced
+    msg = mod.build_user_message(
+        {
+            "transportMode": "land",
+            "fileName": "陆运.xlsx",
+            "sheet": "Sheet1",
+            "cellInventory": ["Sheet1!A1 | 客户 | - | FFFF00"],
+        }
+    )
+    assert "陆运" in msg
+    prompt = ROOT / "agent" / "quote_templates" / "prompts" / "mapping_analyzer.txt"
+    assert prompt.is_file() and prompt.stat().st_size > 1000, prompt
+    print("sheet_map helpers OK")
+
+    assert DESK.is_dir(), f"missing desktop dir: {DESK}"
+    files = sorted(list(DESK.glob("*.xlsx")) + list(DESK.glob("*.xlsx.xlsx")))
+    assert len(files) >= 8, f"expected >=8 templates, got {len(files)}"
+    try:
+        import openpyxl
+    except ImportError:
+        print(f"desktop files present: {len(files)} (openpyxl not installed, skip cell count)")
+        for f in files:
+            print(" -", f.name)
+        return
+    for f in files:
+        wb = openpyxl.load_workbook(f, data_only=False)
+        ws = wb.active
+        n = sum(1 for row in ws.iter_rows(max_row=100, max_col=24) for c in row if c.value is not None)
+        print(f"{f.name}: sheet={ws.title} nonempty≈{n}")
+    print("desktop 8 templates readable OK")
+
+
+if __name__ == "__main__":
+    main()