From 22af553b887bfe2a4ad39e2385ad91a076c03d22 Mon Sep 17 00:00:00 2001 From: yongtenglei Date: Thu, 20 Nov 2025 11:19:57 +0800 Subject: [PATCH] minor fix --- rag/nlp/search.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/rag/nlp/search.py b/rag/nlp/search.py index 6b88dab30..a479e5d3f 100644 --- a/rag/nlp/search.py +++ b/rag/nlp/search.py @@ -422,10 +422,10 @@ class Dealer: tsim = sim vsim = sim - if not sim: + sim_np = np.array(sim, dtype=np.float64) + if sim_np.size == 0: return ranks - sim_np = np.array(sim, dtype=np.float64) sorted_idx = np.argsort(sim_np * -1) valid_idx = [int(i) for i in sorted_idx if sim_np[i] >= similarity_threshold] @@ -461,7 +461,7 @@ class Dealer: "kb_id": chunk["kb_id"], "important_kwd": chunk.get("important_kwd", []), "image_id": chunk.get("img_id", ""), - "similarity": float(sim[i]), + "similarity": float(sim_np[i]), "vector_similarity": float(vsim[i]), "term_similarity": float(tsim[i]), "vector": chunk.get(vector_column, zero_vector),