gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

minor fix

Browse source
This commit is contained in:
yongtenglei 2025-11-20 11:19:57 +08:00
parent aca4eb8b6a
commit 22af553b88
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
rag/nlp/search.py
View file

@ -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),