ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter3/structured-knowledge-extraction/test_extraction_coverage.py

@@ -0,0 +1,18 @@
+from campaign import missing_extraction_rows
+
+
+def test_missing_extraction_rows_finds_omissions_across_batches():
+    rows = [{"id": "case-a"}, {"id": "case-b"}, {"id": "case-c"}]
+    outputs = [
+        {"cases": [{"id": "case-a"}]},
+        {"cases": [{"id": "case-c"}]},
+    ]
+
+    assert missing_extraction_rows(rows, outputs) == [{"id": "case-b"}]
+
+
+def test_missing_extraction_rows_ignores_null_ids_and_empty_outputs():
+    rows = [{"id": "case-a"}]
+    outputs = [{"cases": [{"id": None}]}, {}, {"cases": None}]
+
+    assert missing_extraction_rows(rows, outputs) == rows