Skip to content

Commit ec36d4e

Browse files
committed
check fix
1 parent 9119ef3 commit ec36d4e

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

graphrag/query/structured_search/basic_search/basic_context.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,9 @@ def build_context(
5353
# make a delimited table for the context; this imitates graphrag context building
5454
table = ["id|text"] + [f"{s['id']}|{s['text']}" for s in sources]
5555

56+
columns = pd.Index(["id", "text"])
57+
5658
return ContextBuilderResult(
5759
context_chunks="\n\n".join(table),
58-
context_records={"sources": pd.DataFrame(sources, columns=["id", "text"])},
60+
context_records={"sources": pd.DataFrame(sources, columns=columns)},
5961
)

0 commit comments

Comments
 (0)