Browse Source
fix(graphrag): variable refernence error (#2136)
### 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>tags/v0.11.0
dearjane
1 year ago
1 changed files with 2 additions and 2 deletions
+ 2
- 2
graphrag/search.py
View File
| s = s.query(bqry)[0: 6] | s = s.query(bqry)[0: 6] | ||||
| s = s.to_dict() | s = s.to_dict() | ||||
| txt_res = self.es.search(deepcopy(s), idxnm=idxnm, timeout="600s", src=src) | 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-"): | if merge_into_first(txt_res, "-Original Content-"): | ||||
| txt_ids = comm_ids[0:1] | |||||
| txt_ids = txt_ids[0:1] | |||||
| return self.SearchResult( | return self.SearchResult( | ||||
| total=len(ent_ids) + len(comm_ids) + len(txt_ids), | total=len(ent_ids) + len(comm_ids) + len(txt_ids), |
Loading…