Update retrieval.py

retrieval.py

@@ -54,8 +54,13 @@ def initialize_llmchain(
         memory_key="chat_history", output_key="answer", return_messages=True
     )
     # retriever=vector_db.as_retriever(search_type="similarity", search_kwargs={'k': 3})
-    retriever = vector_db.as_retriever()
-
+    # retriever = vector_db.as_retriever()
+    # Улучшение: параметризованный поиск
+    retriever = vector_db.as_retriever(
+    search_kwargs={"k": 3},
+    search_type="mmr" # Maximal Marginal Relevance
+    )
+    
     progress(0.8, desc="Defining retrieval chain...")
     with open('prompt_template.json', 'r') as file:
     	system_prompt = json.load(file)