beyhan/graphify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix build() dropping hyperedges when merging multiple extractions

Browse Source
This commit is contained in:
Safi
2026-04-06 16:06:31 +01:00
parent 38d967fc63
commit 6570488658
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
graphify/build.py
+2 -1
View File
@@ -33,10 +33,11 @@ def build_from_json(extraction: dict) -> nx.Graph:
def build(extractions: list[dict]) -> nx.Graph: def build(extractions: list[dict]) -> nx.Graph:
"""Merge multiple extraction results into one 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: for ext in extractions:
combined["nodes"].extend(ext.get("nodes", [])) combined["nodes"].extend(ext.get("nodes", []))
combined["edges"].extend(ext.get("edges", [])) combined["edges"].extend(ext.get("edges", []))
combined["hyperedges"].extend(ext.get("hyperedges", []))
combined["input_tokens"] += ext.get("input_tokens", 0) combined["input_tokens"] += ext.get("input_tokens", 0)
combined["output_tokens"] += ext.get("output_tokens", 0) combined["output_tokens"] += ext.get("output_tokens", 0)
return build_from_json(combined) return build_from_json(combined)