bug fix: triples extraction chunk sort without chunk_order (#1584)

SciPhi-AI · Nov 13, 2024 · 8a2723a · 8a2723a
1 parent a58ed6f
commit 8a2723a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py/core/pipes/kg/triples_extraction.py b/py/core/pipes/kg/triples_extraction.py
@@ -289,7 +289,7 @@ async def _run_logic(  # type: ignore
 
         # sort the extractions accroding to chunk_order field in metadata in ascending order
         extractions = sorted(
-            extractions, key=lambda x: x.metadata["chunk_order"]
+            extractions, key=lambda x: x.metadata.get("chunk_order", float('inf'))
         )
 
         # group these extractions into groups of extraction_merge_count