update k number to 6

src/Rag.py

@@ -17,7 +17,6 @@ embedder = None
 index = None
 llm_client = None
 
-# 添加线程锁以确保多用户并发安全
 import threading
 _model_lock = threading.Lock()
 
@@ -41,7 +40,6 @@ def save_embeddings(embedder_name, embeddings):
     Save embeddings to a .npy file.
     """
     file_path = os.path.join("data", "embeddings", f"{embedder_name.replace('/', '_')}.npy")
-    # 确保目录存在
     os.makedirs(os.path.dirname(file_path), exist_ok=True)
     np.save(file_path, embeddings)
     print(f"Saved embeddings for {embedder_name}...")
@@ -360,7 +358,7 @@ def depression_assistant(query, model_name="meta-llama/Llama-3.3-70B-Instruct-Tu
     global db, referenced_tables_db, embedder, index, llm_client
     
     t1 = time.perf_counter()
-    results = faiss_search(query, embedder, db, index, referenced_tables_db, k=3)
+    results = faiss_search(query, embedder, db, index, referenced_tables_db)
     t2 = time.perf_counter()
     print(f"[Time] FAISS search done in {t2 - t1:.2f} seconds.")
 

src/app.py

@@ -32,7 +32,6 @@ def get_llm_client(client_type, api_key):
         )
     return None
 
-# ===== 辅助函数 =====
 def get_current_model_info():
     if "cached_model_info" in st.session_state and st.session_state.cached_model_info:
         return st.session_state.cached_model_info