rag q&a

langchain-streamlit-demo/app.py

@@ -26,7 +26,7 @@ from langchain.vectorstores import FAISS
 from langsmith.client import Client
 from streamlit_feedback import streamlit_feedback
 
-from qagen import combine_qa_pair_lists, get_rag_qa_gen_chain
+from qagen import get_rag_qa_gen_chain
 from summarize import get_summarization_chain
 
 __version__ = "0.0.10"
@@ -448,7 +448,9 @@ if st.session_state.llm:
                         #     config,
                         # )
                         raw_results = st.session_state.doc_chain.invoke(prompt, config)
-                        results = combine_qa_pair_lists(raw_results).QuestionAnswerPairs
+                        # print(raw_results)
+                        # results = combine_qa_pair_lists(raw_results).QuestionAnswerPairs
+                        results = raw_results.QuestionAnswerPairs
 
                         def _to_str(idx, qap):
                             question_piece = f"{idx}. **Q:** {qap.question}"