[Enhancement] Added support for dynamic 'nr_retrieved_documents' value in config.yaml (#110)

* [Enhancement]: Update the workflow file

Signed-off-by: lochan paudel <lochanpaudel10@gmail.com>

* [Enhancement] Added support for dynamic 'k' value in config.yaml for retrieval

Signed-off-by: lochan paudel <lochanpaudel10@gmail.com>

---------

Signed-off-by: lochan paudel <lochanpaudel10@gmail.com>

Author: nahcol10

src/mvt/config.yaml

@@ -16,3 +16,4 @@ system_prompt: "You are an assistant for question-answering tasks that are relat
 use_query_rewriting: false # Set to true to enable query rewriting
 query_rewriting_prompt: "You are an assistant helping to rewrite user queries to make them more specific and effective for searching documents. The context is the Linux Foundation Decentralized Trust former Hyperledger. Please rewrite the human query to be more specific, detailed, and optimized for document retrieval, considering the context mentioned."
 logo_pth: "https://upload.wikimedia.org/wikipedia/en/thumb/e/e2/The_Founder_Institute_Logo.png/250px-The_Founder_Institute_Logo.png"
+nr_retrieved_documents: 5

src/mvt/main.py

@@ -47,7 +47,7 @@ def get_ragchain(filter):
     docsearch = FAISS.load_local(config_data["persist_directory"], embeddings, allow_dangerous_deserialization=True)
 
     # Define a retriever interface
-    retriever = docsearch.as_retriever(search_kwargs={"k": 5, "filter": filter})
+    retriever = docsearch.as_retriever(search_kwargs={"k": config_data["nr_retrieved_documents"], "filter": filter})
 
     # read prompt string from config file
     prompt_str = config_data["system_prompt"]