fix: update return type of context_records

https://github.com/microsoft/graphrag/blob/v1.0.1/graphrag/query/context_builder/builders.py#L21
Cinnamon · Dec 30, 2024 · 54f54ce · 54f54ce
1 parent e5b58a8
commit 54f54ce
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/libs/ktem/ktem/index/file/graph/pipelines.py b/libs/ktem/ktem/index/file/graph/pipelines.py
@@ -291,10 +291,10 @@ def _to_document(self, header: str, context_text: str) -> RetrievedDocument:
         )
 
     def format_context_records(self, context_records) -> list[RetrievedDocument]:
-        entities = context_records.get("entities", [])
-        relationships = context_records.get("relationships", [])
-        reports = context_records.get("reports", [])
-        sources = context_records.get("sources", [])
+        entities = context_records.get("entities", pd.DataFrame())
+        relationships = context_records.get("relationships", pd.DataFrame())
+        reports = context_records.get("reports", pd.DataFrame())
+        sources = context_records.get("sources", pd.DataFrame())
 
         docs = []