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)

Files changed (1) hide show
  1. 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
- res = self.dataStore.search(fields, [], condition, [], OrderByExpr(), p, bs, index_name(tenant_id), kb_ids)
387
- dict_chunks = self.dataStore.getFields(res, fields)
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: