fix: step3 _normalize_rule 为缺失 section 的 table/text source 补齐字段 - Closes #53

LLM 生成的 source 有时缺少 section 字段，导致 Layer A schema 验证失败。在 _normalize_rule 中添加防御性处理：从兄弟 source 或 rule path 推断 section。 Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-06-02 15:46:59 +08:00
parent 6d0a5284e7
commit 1477dbdd18
2 changed files with 67 additions and 0 deletions
@@ -465,3 +465,49 @@ class TestNormalizeRule:
        normalized = _normalize_rule(rule)
        assert normalized["trigger"]["operator"] == "AND"
        assert normalized["trigger"]["conditions"][0]["operator"] == ">="
+
+    def test_normalize_source_missing_section_from_sibling(self):
+        """Table/text sources without section get it from sibling sources."""
+        rule = {
+            "trigger": {"conditions": [{"signal": "x", "operator": "==", "value": "1"}]},
+            "sources": [
+                {"type": "table", "section": "3.1.1 系统限制", "row": 1},
+                {"type": "text", "text_snippet": "missing section"},
+            ],
+        }
+        normalized = _normalize_rule(rule)
+        assert normalized["sources"][1]["section"] == "3.1.1 系统限制"
+
+    def test_normalize_source_missing_section_from_path(self):
+        """Table/text sources without section and no sibling fall back to rule path."""
+        rule = {
+            "trigger": {"conditions": [{"signal": "x", "operator": "==", "value": "1"}]},
+            "path": "4.2 关闭流程 > decision_speed > action_disable",
+            "sources": [
+                {"type": "table", "row": 3, "text_snippet": "no section anywhere"},
+            ],
+        }
+        normalized = _normalize_rule(rule)
+        assert normalized["sources"][0]["section"] == "4.2 关闭流程"
+
+    def test_normalize_source_keeps_existing_section(self):
+        """Sources that already have section are not modified."""
+        rule = {
+            "trigger": {"conditions": [{"signal": "x", "operator": "==", "value": "1"}]},
+            "sources": [
+                {"type": "table", "section": "1.0 概述", "row": 1},
+            ],
+        }
+        normalized = _normalize_rule(rule)
+        assert normalized["sources"][0]["section"] == "1.0 概述"
+
+    def test_normalize_source_skips_logic_tree(self):
+        """Logic tree sources are not touched (don't need section)."""
+        rule = {
+            "trigger": {"conditions": [{"signal": "x", "operator": "==", "value": "1"}]},
+            "sources": [
+                {"type": "logic_tree", "image_id": "img1", "node_ids": ["n1"]},
+            ],
+        }
+        normalized = _normalize_rule(rule)
+        assert "section" not in normalized["sources"][0]