Переглянути джерело

Fix: doc_aggs issue. (#8418)

### What problem does this PR solve?

#8406

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.20.0
Kevin Hu 4 місяці тому
джерело
коміт
d4e6e2bd21
Аккаунт користувача з таким Email не знайдено
1 змінених файлів з 10 додано та 4 видалено
  1. 10
    4
      rag/nlp/search.py

+ 10
- 4
rag/nlp/search.py Переглянути файл

@@ -391,14 +391,20 @@ class Dealer:
for i in idx:
if sim[i] < similarity_threshold:
break
if len(ranks["chunks"]) >= page_size:
if aggs:
continue
break

id = sres.ids[i]
chunk = sres.field[id]
dnm = chunk.get("docnm_kwd", "")
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", [])
d = {
"chunk_id": id,

Завантаження…
Відмінити
Зберегти