Parcourir la source

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 il y a 10 mois
Parent
révision
321e9f3719
Aucun compte lié à l'adresse e-mail de l'auteur
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1
    1
      rag/nlp/search.py

+ 1
- 1
rag/nlp/search.py Voir le fichier

@@ -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:

Chargement…
Annuler
Enregistrer