合并远程报价模板 AI 解析与超时设置：保留私聊图片和附件询价。

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-09-18 17:47:48 +08:00
parent 65ef75f1ef 527050bbf9
commit 19f94252f0
41 changed files with 4972 additions and 438 deletions
@@ -0,0 +1,150 @@
+"""
+报价模板 sheet_map：JSON 剥离、缺项检测、补跑单测（不打外网）。
+"""
+
+from __future__ import annotations
+
+import unittest
+from types import SimpleNamespace
+
+from agent.llm.mode_sheet_map import (
+    assess_mapping_gaps,
+    build_user_message,
+    invoke_sheet_map,
+    parse_mapping_json,
+    strip_json_fence,
+)
+
+
+class TestSheetMapParse(unittest.TestCase):
+    def test_strip_fence(self) -> None:
+        raw = '```json\n{"schemaVersion":"quote-template-mapping-v1","fields":{}}\n```'
+        self.assertIn("schemaVersion", strip_json_fence(raw))
+
+    def test_parse_ok(self) -> None:
+        raw = '{"schemaVersion":"quote-template-mapping-v1","fields":{"quote_no":"S!A1"},"feeRegions":[]}'
+        out = parse_mapping_json(raw)
+        self.assertTrue(out["ok"])
+        self.assertEqual(out["mapping"]["fields"]["quote_no"], "S!A1")
+
+    def test_parse_invalid(self) -> None:
+        out = parse_mapping_json("not-json")
+        self.assertFalse(out["ok"])
+
+    def test_user_message_contains_inventory(self) -> None:
+        msg = build_user_message(
+            {
+                "transportMode": "sea",
+                "fileName": "海运报价单模板.xlsx",
+                "sheet": "Sheet1",
+                "cellInventory": ["Sheet1!A1 | 报价日期 | - | FFFF00"],
+            }
+        )
+        self.assertIn("海运", msg)
+        self.assertIn("Sheet1!A1", msg)
+        self.assertIn("自检", msg)
+
+    def test_user_message_gaps_block(self) -> None:
+        msg = build_user_message(
+            {
+                "transportMode": "sea",
+                "fileName": "x.xlsx",
+                "sheet": "S",
+                "cellInventory": ["S!A1 | x | - | -"],
+            },
+            gaps=["费用区缺失：始发港费用"],
+        )
+        self.assertIn("补跑缺口", msg)
+        self.assertIn("始发港费用", msg)
+
+    def test_assess_gaps_sea_fee_and_fields(self) -> None:
+        payload = {
+            "transportMode": "sea",
+            "cellInventory": [
+                "海运询价单!A18 | 始发港费用/Origin Charges | - | -",
+                "海运询价单!A30 | 海运费用/Ocean Freight | - | -",
+                "海运询价单!A35 | 目的港费用/Destination Charges | - | -",
+                "海运询价单!E6 | 提货地址/Place of Receipt | - | -",
+                "海运询价单!A11 | 货值/Cargo Value | - | -",
+            ],
+        }
+        mapping = {
+            "fields": {"quote_no": "海运询价单!B3"},
+            "feeRegions": [],
+        }
+        gaps = assess_mapping_gaps(payload, mapping)
+        self.assertTrue(any("始发港费用" in g for g in gaps))
+        self.assertTrue(any("海运费用" in g for g in gaps))
+        self.assertTrue(any("目的港费用" in g for g in gaps))
+        self.assertTrue(any("pickup_address" in g for g in gaps))
+        self.assertTrue(any("cargo_value" in g for g in gaps))
+
+    def test_assess_gaps_complete_empty(self) -> None:
+        payload = {
+            "transportMode": "sea",
+            "cellInventory": [
+                "S!A18 | 始发港费用/Origin Charges | - | -",
+                "S!E6 | 提货地址 | - | -",
+            ],
+        }
+        mapping = {
+            "fields": {"pickup_address": "S!H6"},
+            "feeRegions": [
+                {
+                    "id": "origin_charges",
+                    "title": "始发港费用",
+                    "tmsSource": "polCostItems",
+                    "stopAtText": ["x"],
+                    "anchor": "S!A19",
+                }
+            ],
+        }
+        self.assertEqual(assess_mapping_gaps(payload, mapping), [])
+
+    def test_invoke_retries_when_gaps(self) -> None:
+        calls: list[str] = []
+
+        def fake_chat(*, mode, messages, temperature=0.0):  # noqa: ARG001
+            calls.append(messages[-1]["content"])
+            if len(calls) == 1:
+                content = (
+                    '{"schemaVersion":"quote-template-mapping-v1",'
+                    '"fields":{"quote_no":"S!B3"},"feeRegions":[]}'
+                )
+            else:
+                content = (
+                    '{"schemaVersion":"quote-template-mapping-v1",'
+                    '"fields":{"quote_no":"S!B3","pickup_address":"S!H6","cargo_value":"S!B11"},'
+                    '"feeRegions":[{"id":"origin_charges","title":"始发港费用",'
+                    '"tmsSource":"polCostItems","anchor":"S!A19","stopAtText":["海运费用"],'
+                    '"grow":"insert_before_stop","styleCopyRows":1}]}'
+                )
+            return SimpleNamespace(ok=True, content=content, provider="stub", error="")
+
+        payload = {
+            "versionId": "v1",
+            "transportMode": "sea",
+            "fileName": "sea.xlsx",
+            "sheet": "S",
+            "cellInventory": [
+                "S!A18 | 始发港费用/Origin Charges | - | -",
+                "S!E6 | 提货地址/Place of Receipt | - | -",
+                "S!A11 | 货值/Cargo Value | - | -",
+            ],
+        }
+        out = invoke_sheet_map(payload, chat_fn=fake_chat, allow_network=False)
+        self.assertTrue(out["ok"])
+        self.assertEqual(out.get("rounds"), 2)
+        self.assertEqual(len(calls), 2)
+        self.assertIn("补跑缺口", calls[1])
+        self.assertIn("pickup_address", out["mapping"]["fields"])
+        self.assertEqual(len(out["mapping"]["feeRegions"]), 1)
+        self.assertIn("mappingGaps", out["mapping"])
+        self.assertIsInstance(out["mapping"]["mappingGaps"], list)
+        # 第二轮已补齐清单内缺口 → mappingGaps 为空且与 gaps_after 一致
+        self.assertEqual(out.get("gaps_after"), out["mapping"]["mappingGaps"])
+        self.assertEqual(out["mapping"]["mappingGaps"], [])
+
+
+if __name__ == "__main__":
+    unittest.main()