Browse Source
Fix raptor bug. (#4192)
### What problem does this PR solve? ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue)tags/v0.15.1
Kevin Hu
10 months ago
1 changed files with 2 additions and 2 deletions
+ 2
- 2
rag/nlp/search.py
View File
| 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) | |||||
| dict_chunks = self.dataStore.getFields(res, fields) | |||||
| es_res = self.dataStore.search(fields, [], condition, [], OrderByExpr(), p, bs, index_name(tenant_id), kb_ids) | |||||
| 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: |
Loading…