AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-14 06:16:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(graphrag): variable refernence error (#2136)

Browse Source 
### What problem does this PR solve?

fix: Use wrong variable in graph rag step.

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)


Co-authored-by: 陈晓强 <chenxiaoqiang@cvte.com>
This commit is contained in:
dearjane 2024-08-28 16:35:42 +08:00 committed by GitHub
parent e627ee9ea4
commit 3abc9be1c2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
graphrag/search.py
View File

@ -93,9 +93,9 @@ class KGSearch(Dealer):
s = s.query(bqry)[0: 6]
s = s.to_dict()
txt_res = self.es.search(deepcopy(s), idxnm=idxnm, timeout="600s", src=src)
txt_ids = self.es.getDocIds(comm_res)
txt_ids = self.es.getDocIds(txt_res)
if merge_into_first(txt_res, "-Original Content-"):
txt_ids = comm_ids[0:1]
txt_ids = txt_ids[0:1]
return self.SearchResult(
total=len(ent_ids) + len(comm_ids) + len(txt_ids),