expand rerank range

infiniflow · Oct 8, 2024 · d2a9c53 · d2a9c53
1 parent 5cc9981
commit d2a9c53
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rag/nlp/search.py b/rag/nlp/search.py
@@ -364,7 +364,7 @@ def retrieval(self, question, embd_mdl, tenant_id, kb_ids, page, page_size, simi
         if not question:
             return ranks
         RERANK_PAGE_LIMIT = 3
-        req = {"kb_ids": kb_ids, "doc_ids": doc_ids, "size": page_size*RERANK_PAGE_LIMIT,
+        req = {"kb_ids": kb_ids, "doc_ids": doc_ids, "size": max(page_size*RERANK_PAGE_LIMIT, 128),
                "question": question, "vector": True, "topk": top,
                "similarity": similarity_threshold,
                "available_int": 1}