diff --git a/rag/nlp/search.py b/rag/nlp/search.py index ea8340629..914b49c20 100644 --- a/rag/nlp/search.py +++ b/rag/nlp/search.py @@ -348,7 +348,7 @@ class Dealer: 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