diff --git a/rag/nlp/search.py b/rag/nlp/search.py index 86416cd49..1e83969e4 100644 --- a/rag/nlp/search.py +++ b/rag/nlp/search.py @@ -258,6 +258,8 @@ class Dealer: q_denor = np.sqrt(np.sum([s*s for t,s in query_rfea.items() if t != PAGERANK_FLD])) for i in search_res.ids: nor, denor = 0, 0 + if not search_res.field[i].get(TAG_FLD): + continue for t, sc in eval(search_res.field[i].get(TAG_FLD, "{}")).items(): if t in query_rfea: nor += query_rfea[t] * sc