From f2cd8d88a211cf805750a9084f414bceba83f992 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E2=80=9Chu?= <812791840@qq.com>
Date: Thu, 11 Dec 2025 09:55:08 +0800
Subject: [PATCH] Update search.py

---
 rag/nlp/search.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/rag/nlp/search.py b/rag/nlp/search.py
index f5dd2d4de..a9ac243d7 100644
--- a/rag/nlp/search.py
+++ b/rag/nlp/search.py
@@ -430,6 +430,9 @@ class Dealer:
         sorted_idx = np.argsort(sim_np * -1)
 
         valid_idx = [int(i) for i in sorted_idx if sim_np[i] >= similarity_threshold]
+        # use top to limit the maximum number of returned results
+        top = min(len(valid_idx), top)
+        valid_idx = valid_idx[:top]
         filtered_count = len(valid_idx)
         ranks["total"] = int(filtered_count)