Kevin Hu
commited on
Commit
·
077390b
1
Parent(s):
38b46ea
Fix raptor bug. (#4192)
Browse files### What problem does this PR solve?
### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
- rag/nlp/search.py +2 -2
rag/nlp/search.py
CHANGED
@@ -383,8 +383,8 @@ class Dealer:
|
|
383 |
res = []
|
384 |
bs = 128
|
385 |
for p in range(0, max_count, bs):
|
386 |
-
|
387 |
-
dict_chunks = self.dataStore.getFields(
|
388 |
if dict_chunks:
|
389 |
res.extend(dict_chunks.values())
|
390 |
if len(dict_chunks.values()) < bs:
|
|
|
383 |
res = []
|
384 |
bs = 128
|
385 |
for p in range(0, max_count, bs):
|
386 |
+
es_res = self.dataStore.search(fields, [], condition, [], OrderByExpr(), p, bs, index_name(tenant_id), kb_ids)
|
387 |
+
dict_chunks = self.dataStore.getFields(es_res, fields)
|
388 |
if dict_chunks:
|
389 |
res.extend(dict_chunks.values())
|
390 |
if len(dict_chunks.values()) < bs:
|