kanshan/graphrag
1
0
Fork 0
mirror of https://github.com/microsoft/graphrag.git synced 2026-01-14 00:57:23 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Fix multi-index search (#2063)

Browse Source
This commit is contained in:
Nathan Evans 2025-09-18 12:49:56 -07:00 committed by GitHub
parent 6c66b7c30f
commit 2bf7e7c018
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 15 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.semversioner/next-release/patch-20250918192431890892.json Normal file
View File

@ -0,0 +1,4 @@
{
"type": "patch",
"description": "Fix multi-index search."
}

21
graphrag/utils/api.py
View File

@ -168,57 +168,58 @@ def update_context_data(
"""
updated_context_data = {}
for key in context_data:
entries = context_data[key].to_dict(orient="records")
updated_entry = []
if key == "reports":
updated_entry = [
dict(
{k: entry[k] for k in entry},
entry,
index_name=links["community_reports"][int(entry["id"])][
"index_name"
],
index_id=links["community_reports"][int(entry["id"])]["id"],
)
for entry in context_data[key]
for entry in entries
]
if key == "entities":
updated_entry = [
dict(
{k: entry[k] for k in entry},
entry,
entity=entry["entity"].split("-")[0],
index_name=links["entities"][int(entry["id"])]["index_name"],
index_id=links["entities"][int(entry["id"])]["id"],
)
for entry in context_data[key]
for entry in entries
]
if key == "relationships":
updated_entry = [
dict(
{k: entry[k] for k in entry},
entry,
source=entry["source"].split("-")[0],
target=entry["target"].split("-")[0],
index_name=links["relationships"][int(entry["id"])]["index_name"],
index_id=links["relationships"][int(entry["id"])]["id"],
)
for entry in context_data[key]
for entry in entries
]
if key == "claims":
updated_entry = [
dict(
{k: entry[k] for k in entry},
entry,
entity=entry["entity"].split("-")[0],
index_name=links["covariates"][int(entry["id"])]["index_name"],
index_id=links["covariates"][int(entry["id"])]["id"],
)
for entry in context_data[key]
for entry in entries
]
if key == "sources":
updated_entry = [
dict(
{k: entry[k] for k in entry},
entry,
index_name=links["text_units"][int(entry["id"])]["index_name"],
index_id=links["text_units"][int(entry["id"])]["id"],
)
for entry in context_data[key]
for entry in entries
]
updated_context_data[key] = updated_entry
return updated_context_data