refine using rerank model (infiniflow#2553)

### What problem does this PR solve? infiniflow#2552 ### Type of change - [x] Performance Improvement
Halfknow · Sep 24, 2024 · f2a47c0 · f2a47c0
1 parent c06a098
commit f2a47c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rag/nlp/search.py b/rag/nlp/search.py
@@ -348,7 +348,7 @@ def rerank_by_model(self, rerank_mdl, sres, query, tkweight=0.3,
             ins_tw.append(tks)
 
         tksim = self.qryr.token_similarity(keywords, ins_tw)
-        vtsim,_ = rerank_mdl.similarity(" ".join(keywords), [rmSpace(" ".join(tks)) for tks in ins_tw])
+        vtsim,_ = rerank_mdl.similarity(query, [rmSpace(" ".join(tks)) for tks in ins_tw])
 
         return tkweight*np.array(tksim) + vtweight*vtsim, tksim, vtsim