Update app.py

app.py

@@ -37,6 +37,7 @@ vectorstore = FAISS(
     docstore=InMemoryDocstore(docstore),
     index_to_docstore_id=index_to_docstore_id
 )
+retriever = vectorstore.as_retriever()
 # ========= Step 3: 加载语言模型 =========
 model_name = "Qwen/Qwen1.5-1.8B-Chat"
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
@@ -114,7 +115,7 @@ def chat(user_input, history):
     ])
 
     # 🔍 1. 检索与用户问题最相关的语料
-    retrieved_docs = vectorstore.get_relevant_documents(user_input)
+    retrieved_docs = retriever.get_relevant_documents(user_input) 
     retrieved_context = "\n".join([doc.page_content for doc in retrieved_docs])
 
     example_pairs = cleaned_pairs[:2]