add source printing

langchain-streamlit-demo/app.py

@@ -463,7 +463,7 @@ if st.session_state.llm:
             #     ],
             # )
 
-            full_response: Union[str, None] = None
+            full_response: Union[str, dict[str, Any], None] = None
 
             message_placeholder = st.empty()
             sources_placeholder = st.empty()
@@ -530,10 +530,11 @@ if st.session_state.llm:
                         input_str,
                         config=get_config(callbacks),
                     )
-                    st.session_state.sources = response["source_documents"][:3]
+                    st.session_state.sources = response["source_documents"]
                     with st.sidebar.expander("Sources"):
-                        for source in st.session_state.sources:
+                        for e, source in enumerate(st.session_state.sources):
                             st.markdown("-" * 50)
+                            st.markdown(f"## Source {e+1}")
                             st.markdown(source.page_content)
                     # with sources_placeholder:
                     #     with st.expander("Sources", expanded=False):
@@ -604,14 +605,15 @@ if st.session_state.llm:
                     full_response["output_text"],
                     full_response["source_documents"],
                 )
-                st.session_state.sources = sources[:3]
+                st.session_state.sources = sources
                 message_placeholder.markdown(
                     text_response + "\n\n" + f"⏱️ {finish_time:.2f}s"
                 )
                 with sources_placeholder:
                     with st.expander("Sources", expanded=False):
-                        for source in st.session_state.sources:
+                        for e, source in enumerate(st.session_state.sources):
                             st.markdown("-" * 50)
+                            st.markdown(f"## Source {e+1}")
                             st.markdown(source.page_content)
 
                 # --- Tracing ---

langchain-streamlit-demo/defaults.py

@@ -48,7 +48,7 @@ DEFAULT_LANGSMITH_PROJECT = os.environ.get("LANGCHAIN_PROJECT")
 
 TEST_QUESTIONS = [
     "Come posso ottenerle le credenziali di accesso all'area riservata?",
-    "Quali prestazioni presenti nel checkup?",
+    "Quali prestazioni sono presenti nel checkup?",
     "La risonanza magnetica è coperta dalla polizza?",
     "Le visite odontoiatriche sono coperte dalla polizza?",
     "Come posso richiedere il checkup?",