Update candidate number

app.py

@@ -56,6 +56,7 @@ clear_label = ["清除","清除","Clear"]
 submit_label = ["提交","提交","Submit"]
 
 threshold = 0.6
+candidate = 1
 
 # %%
 # helper functions for prompt processing for this LLM
@@ -119,7 +120,7 @@ def answer(input_text,context=None):
 
 # helper function for RAG
 def helper_rag(text):
-    docs_out = vectordb.similarity_search_with_relevance_scores(text,k=3)
+    docs_out = vectordb.similarity_search_with_relevance_scores(text,k=candidate)
     #docs_out = vectordb.max_marginal_relevance_search(text,k=5,fetch_k = 20, lambda_mult = 0.5)
     context = []
     for doc in docs_out: