test

confluence_qa.py

@@ -50,7 +50,7 @@ class ConfluenceQA:
             template=template, input_variables=["context", "question"]
         )
         chain_type_kwargs = {"prompt": QA_CHAIN_PROMPT}
-        self.qa = RetrievalQA.from_chain_type(llm=self.llm, chain_type="stuff", retriever=self.db.as_retriever(), chain_type_kwargs=chain_type_kwargs)
+        self.qa = RetrievalQA.from_chain_type(llm=self.llm, chain_type="stuff", retriever=self.db.as_retriever(search_kwargs={"k":4}), chain_type_kwargs=chain_type_kwargs)
 
     def __init__(self,config:dict = {}) -> None:
         self.db=None