Update functions.py

functions.py

@@ -189,9 +189,9 @@ def embed_text(query,corpus,title,embedding_model,_emb_tok,chain_type='stuff'):
                                            prompt=PROMPT, 
                                            )
 
-        answer = chain({"input_documents": docs, "question": query}, return_only_outputs=True)
+        answer = chain({"input_documents": docs, "question": query}, return_only_outputs=False)
 
-        return answer['output_text']
+        return answer
 
     elif chain_type == 'Refined':
 
@@ -200,9 +200,9 @@ def embed_text(query,corpus,title,embedding_model,_emb_tok,chain_type='stuff'):
 )
         chain = load_qa_chain(OpenAI(temperature=0), chain_type="refine", return_refine_steps=False,
                      question_prompt=initial_qa_prompt, refine_prompt=refine_prompt)
-        answer = chain({"input_documents": docs, "question": query}, return_only_outputs=True)
+        answer = chain({"input_documents": docs, "question": query}, return_only_outputs=False)
 
-        return answer['output_text']
+        return answer
     
 @st.experimental_singleton(suppress_st_warning=True)
 def get_spacy():