add k param

app.py

@@ -106,7 +106,7 @@ def QAQuery_p(question: str):
     global vectordb_p
     # vectordb = Chroma(persist_directory='db', embedding_function=embeddings)
     retriever = vectordb_p.as_retriever()
-    retriever.search_kwargs['k'] = 3
+    retriever.search_kwargs['k'] = os.getenv("search_kwargs_k")
     # retriever.search_kwargs['fetch_k'] = 100
 
     qa = RetrievalQA.from_chain_type(llm=chat, chain_type="stuff", retriever=retriever, return_source_documents = True)
@@ -122,7 +122,7 @@ def QAQuery_p(question: str):
     print("Source:", res['source_documents'])
     response = res['result']
     # response = res['source_documents']
-    source = "test source function"
+    source = res['source_documents']
     return response, source
 
 def CreatDb():