AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-04-23 22:50:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix raptor bug. (#4192)

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


### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu 2024-12-23 18:59:48 +08:00 committed by GitHub
parent cf62230548
commit c373dba0bc
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
rag/nlp/search.py
View File

@ -383,8 +383,8 @@ class Dealer:
res = [] res = []
bs = 128 bs = 128
for p in range(0, max_count, bs): for p in range(0, max_count, bs):
res = self.dataStore.search(fields, [], condition, [], OrderByExpr(), p, bs, index_name(tenant_id), kb_ids) es_res = self.dataStore.search(fields, [], condition, [], OrderByExpr(), p, bs, index_name(tenant_id), kb_ids)
dict_chunks = self.dataStore.getFields(res, fields) dict_chunks = self.dataStore.getFields(es_res, fields)
if dict_chunks: if dict_chunks:
res.extend(dict_chunks.values()) res.extend(dict_chunks.values())
if len(dict_chunks.values()) < bs: if len(dict_chunks.values()) < bs: