From 02c955babb1bba9d3b2dc5aced177acef2f0ade2 Mon Sep 17 00:00:00 2001
From: Kevin Hu <kevinhu.sh@gmail.com>
Date: Wed, 5 Mar 2025 14:37:51 +0800
Subject: [PATCH] Fix: parameter error. (#5641)

### What problem does this PR solve?

#5600

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 graphrag/general/extractor.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/graphrag/general/extractor.py b/graphrag/general/extractor.py
index 5576101c7..c2a9bf127 100644
--- a/graphrag/general/extractor.py
+++ b/graphrag/general/extractor.py
@@ -186,7 +186,7 @@ class Extractor:
             src_id: str,
             tgt_id: str,
             edges_data: list[dict],
-            all_relationships_data
+            all_relationships_data=None
     ):
         if not edges_data:
             return
@@ -229,7 +229,8 @@ class Extractor:
             source_id=source_id
         )
         self._set_relation_(src_id, tgt_id, edge_data)
-        all_relationships_data.append(edge_data)
+        if all_relationships_data is not None:
+            all_relationships_data.append(edge_data)
 
     async def _handle_entity_relation_summary(
             self,