yujunjun/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2025-09-26 08:34:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix: doc_aggs issue. (#8418)

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

#8406

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu 2025-06-23 14:54:01 +08:00 committed by GitHub
parent 81a4c0698c
commit d4e6e2bd21
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
rag/nlp/search.py
View File

@ -391,14 +391,20 @@ class Dealer:
for i in idx: for i in idx:
if sim[i] < similarity_threshold: if sim[i] < similarity_threshold:
break break
if len(ranks["chunks"]) >= page_size:
if aggs:
continue
break
id = sres.ids[i] id = sres.ids[i]
chunk = sres.field[id] chunk = sres.field[id]
dnm = chunk.get("docnm_kwd", "") dnm = chunk.get("docnm_kwd", "")
did = chunk.get("doc_id", "") did = chunk.get("doc_id", "")
if len(ranks["chunks"]) >= page_size:
if aggs:
if dnm not in ranks["doc_aggs"]:
ranks["doc_aggs"][dnm] = {"doc_id": did, "count": 0}
ranks["doc_aggs"][dnm]["count"] += 1
continue
break
position_int = chunk.get("position_int", []) position_int = chunk.get("position_int", [])
d = { d = {
"chunk_id": id, "chunk_id": id,