check fix

microsoft · Dec 27, 2024 · ec36d4e · ec36d4e
1 parent 9119ef3
commit ec36d4e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/graphrag/query/structured_search/basic_search/basic_context.py b/graphrag/query/structured_search/basic_search/basic_context.py
@@ -53,7 +53,9 @@ def build_context(
         # make a delimited table for the context; this imitates graphrag context building
         table = ["id|text"] + [f"{s['id']}|{s['text']}" for s in sources]
 
+        columns = pd.Index(["id", "text"])
+
         return ContextBuilderResult(
             context_chunks="\n\n".join(table),
-            context_records={"sources": pd.DataFrame(sources, columns=["id", "text"])},
+            context_records={"sources": pd.DataFrame(sources, columns=columns)},
         )