From 657048865852c6a503f32d3e432ec34bbdbe5caf Mon Sep 17 00:00:00 2001 From: Safi Date: Mon, 6 Apr 2026 14:48:00 +0100 Subject: [PATCH] fix build() dropping hyperedges when merging multiple extractions --- graphify/build.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/graphify/build.py b/graphify/build.py index 0975d7a5..5e65ceeb 100644 --- a/graphify/build.py +++ b/graphify/build.py @@ -33,10 +33,11 @@ def build_from_json(extraction: dict) -> nx.Graph: def build(extractions: list[dict]) -> nx.Graph: """Merge multiple extraction results into one graph.""" - combined: dict = {"nodes": [], "edges": [], "input_tokens": 0, "output_tokens": 0} + combined: dict = {"nodes": [], "edges": [], "hyperedges": [], "input_tokens": 0, "output_tokens": 0} for ext in extractions: combined["nodes"].extend(ext.get("nodes", [])) combined["edges"].extend(ext.get("edges", [])) + combined["hyperedges"].extend(ext.get("hyperedges", [])) combined["input_tokens"] += ext.get("input_tokens", 0) combined["output_tokens"] += ext.get("output_tokens", 0) return build_from_json(combined)