tweaks

pages/3_🕮_Docs_Demo.py

@@ -96,10 +96,25 @@ rag_chain = (
     | StrOutputParser()
 )
 
+rag_chain.invoke("What is the difference between Fee and Easement?")
+
+
 # +
-# rag_chain.invoke("What is the difference between Fee and Easement?")
 
+from langchain_core.runnables import RunnableParallel
+
+rag_chain_from_docs = (
+    RunnablePassthrough.assign(context=(lambda x: format_docs(x["context"])))
+    | prompt
+    | llm
+    | StrOutputParser()
+)
+
+rag_chain_with_source = RunnableParallel(
+    {"context": retriever, "question": RunnablePassthrough()}
+).assign(answer=rag_chain_from_docs)
 
+rag_chain_with_source.invoke("What is the difference between Fee and Easement?")
 
 # +
 from langchain.memory.chat_message_histories import StreamlitChatMessageHistory