Update app.py

app.py

@@ -96,7 +96,7 @@ def get_conversation_chain(vectorstore):
     
     
     prompt = hub.pull("rlm/rag-prompt")
-    rag_chain = retriever | prompt  | llm | StrOutputParser()
+    rag_chain = ("context": retriever | prompt  | llm | StrOutputParser())
 
 
     return rag_chain
@@ -123,7 +123,7 @@ def main():
 
 
     if user_question := st.text_input("Ask a question about your documents:"):
-        handle_userinput(user_question, vectorstore, conversation)
+        handle_userinput(user_question)
     
    
     with st.sidebar: