Bladeren bron

fix: stop rerank by model when search result is empty (#4203)

### What problem does this PR solve?


stop rerank by model when search result is empty, otherwise rerank may
raise an error (qwen).

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

Co-authored-by: 刘博 <liubo@ynby.cn>
tags/v0.15.1
Bo Liu 10 maanden geleden
bovenliggende
commit
321e9f3719
No account linked to committer's email address
1 gewijzigde bestanden met toevoegingen van 1 en 1 verwijderingen
  1. 1
    1
      rag/nlp/search.py

+ 1
- 1
rag/nlp/search.py Bestand weergeven

@@ -313,7 +313,7 @@ class Dealer:
ranks["total"] = sres.total

if page <= RERANK_PAGE_LIMIT:
if rerank_mdl:
if rerank_mdl and sres.total > 0:
sim, tsim, vsim = self.rerank_by_model(rerank_mdl,
sres, question, 1 - vector_similarity_weight, vector_similarity_weight)
else:

Laden…
Annuleren
Opslaan