[Add] : add HyDE RAG Fusion mid context

Author: seungwon2

genai/aws-gen-ai-kr/20_applications/02_qa_chatbot/04_web_ui/bedrock.py

@@ -117,24 +117,27 @@ def invoke(query, streaming_callback, parent, reranker, hyde, ragfusion, alpha):
     )
     response, pretty_contexts, similar_docs, augmentation = qa.invoke(query = query, complex_doc = True)
     print("-------> response")
-    print(response)
+    # print(response)
     print("-------> pretty_contexts -> 모든 컨텍스트 포함된 자료")
 
     def extract_elements_and_print(pretty_contexts):
         for context in pretty_contexts:
             print("context: \n")
-            print(context)
+    #         print(context)
 
-    print("######### SEMANTIC #########")
-    extract_elements_and_print(pretty_contexts[0])
-    print("######### KEYWORD #########")
-    extract_elements_and_print(pretty_contexts[1])
-    print("######### WITHOUT_RERANKER #########")
-    extract_elements_and_print(pretty_contexts[2])
-    print("######## SIMILAR_DOCS ##########")
-    extract_elements_and_print(pretty_contexts[3])
+    # print("######### SEMANTIC #########")
+    # extract_elements_and_print(pretty_contexts[0])
+    # print("######### KEYWORD #########")
+    # extract_elements_and_print(pretty_contexts[1])
+    # print("######### WITHOUT_RERANKER #########")
+    # extract_elements_and_print(pretty_contexts[2])
+    # print("######## SIMILAR_DOCS ##########")
+    # extract_elements_and_print(pretty_contexts[3])
+    # if hyde or ragfusion:
+    #     print("######## 중간답변 ##########")
+    #     print(augmentation)
     if hyde or ragfusion:
-        print("######## 중간답변 ##########")
-        print(augmentation)
+        return response, pretty_contexts, augmentation
+
 
     return response, pretty_contexts

genai/aws-gen-ai-kr/20_applications/02_qa_chatbot/04_web_ui/streamlit.py

@@ -156,6 +156,12 @@ def show_answer_with_multi_columns(answers):
             with st.chat_message("assistant"):
                 with st.expander("Context 확인하기 ⬇️"):
                     show_context_with_tab(contexts=msg["content"])
+                    
+        elif msg["role"] == "hyde_or_fusion":
+            with st.chat_message("assistant"):
+                with st.expander("중간 답변 확인하기 ⬇️"):
+                    msg["content"]
+                    
         elif msg["role"] == "assistant_column":
             # 'Separately' 옵션일 경우 multi column 으로 보여주지 않고 첫 번째 답변만 출력
             st.chat_message(msg["role"]).write(msg["content"][0]) 
@@ -188,19 +194,34 @@ def show_answer_with_multi_columns(answers):
         )
         # response 로 메세지, 링크, 레퍼런스(source_documents) 받아오게 설정된 것을 변수로 저장
         answer = response[0]
-        contexts = response[1] 
+        contexts = response[1]
+        if hyde or ragfusion:
+            mid_answer = response[2]
 
         # UI 출력
         st.chat_message("assistant").write(answer)
 
+        
+        if hyde:
+            with st.chat_message("assistant"):
+                with st.expander("HyDE 중간 생성 답변 ⬇️"):
+                    mid_answer
+            
+        if ragfusion:
+            with st.chat_message("assistant"):
+                with st.expander("RAG-Fusion 중간 생성 쿼리 ⬇️"):
+                    mid_answer  
         with st.chat_message("assistant"): 
-            with st.expander("정확도 별 답변 보기 ⬇️"):
+            with st.expander("정확도 별 컨텍스트 보기 ⬇️"):
                 show_context_with_tab(contexts)
-
+        
         # Session 메세지 저장
         st.session_state.messages.append({"role": "assistant", "content": answer})
-        st.session_state.messages.append({"role": "assistant_context", "content": contexts})
 
+        if hyde or ragfusion:
+            st.session_state.messages.append({"role": "hyde_or_fusion", "content": mid_answer})
+
+        st.session_state.messages.append({"role": "assistant_context", "content": contexts})
         # Thinking을 complete로 수동으로 바꾸어 줌
         st_cb._complete_current_thought()