Update app.py

app.py

@@ -45,7 +45,7 @@ def invoke(openai_api_key, use_rag, prompt):
     else:
         #vector_db = Chroma(persist_directory = CHROMA_DIR, embedding_function = OpenAIEmbeddings())
         llm = ChatOpenAI(model_name = MODEL_NAME, openai_api_key = openai_api_key, temperature = 0)
-        qa_chain = RetrievalQA.from_chain_type(llm, chain_type_kwargs = {"prompt": QA_CHAIN_PROMPT})
+        qa_chain = RetrievalQA.from_chain_type(llm, retriever = None, return_source_documents = True, cchain_type_kwargs = {"prompt": QA_CHAIN_PROMPT})
     result = qa_chain({"query": prompt})
     #print(result)
     return result["result"]