minor update

added search parameter k into config

code/config.yml

@@ -5,6 +5,7 @@ embedding_options:
   db_option : 'FAISS' # str
   db_path : 'vectorstores' # str
   model : 'sentence-transformers/all-MiniLM-L6-v2' # str [sentence-transformers/all-MiniLM-L6-v2, text-embedding-ada-002']
+  search_top_k : 5 # int
 llm_params: 
   use_history: True # bool
   llm_loader: 'openai' # str [ctransformers, openai]

code/modules/llm_tutor.py

@@ -47,7 +47,7 @@ class LLMTutor:
             qa_chain = ConversationalRetrievalChain.from_llm(
                 llm=llm,
                 chain_type="stuff",
-                retriever=db.as_retriever(search_kwargs={"k": 3}),
+                retriever=db.as_retriever(search_kwargs={"k": self.config["embedding_options"]["search_top_k"]}),
                 return_source_documents=True,
                 memory=memory,
                 combine_docs_chain_kwargs={"prompt": prompt},
@@ -56,7 +56,7 @@ class LLMTutor:
             qa_chain = RetrievalQA.from_chain_type(
                 llm=llm,
                 chain_type="stuff",
-                retriever=db.as_retriever(search_kwargs={"k": 3}),
+                retriever=db.as_retriever(search_kwargs={"k": self.config["embedding_options"]["search_top_k"]}),
                 return_source_documents=True,
                 chain_type_kwargs={"prompt": prompt},
             )