Update app.py

app.py

@@ -17,7 +17,7 @@ folder = snapshot_download(repo_id="umaiku/faiss_index", repo_type="dataset", lo
 embeddings = HuggingFaceEmbeddings(model_name="intfloat/multilingual-e5-small")
 
 vector_db = FAISS.load_local("faiss_index", embeddings, allow_dangerous_deserialization=True)
-retriever = vector_db.as_retriever()
+retriever = vector_db.as_retriever(search_kwargs={"k": 1})
 
 def respond(
     message,
@@ -27,7 +27,7 @@ def respond(
     temperature,
     top_p,
 ):
-    messages = [{"role": "system", "content": system_message, "context": retriever}]
+    messages = [{"role": "system", "content": system_message}]
 
     for val in history:
         if val[0]: