change variable name in prompt

langchain-streamlit-demo/app.py

@@ -442,7 +442,7 @@ if st.session_state.llm:
                             config["max_concurrency"] = 5
                         # raw_results = st.session_state.doc_chain.batch(
                         #     [
-                        #         {"input": doc.page_content, "prompt": prompt}
+                        #         {"context": doc.page_content, "prompt": prompt}
                         #         for doc in st.session_state.texts
                         #     ],
                         #     config,

langchain-streamlit-demo/qagen.py

@@ -36,7 +36,7 @@ Do not provide additional commentary and do not wrap your response in Markdown f
 templ2 = """{prompt}
 Please create question/answer pairs, in the specified JSON format, for the following text:
 ----------------
-{input}"""
+{context}"""
 CHAT_PROMPT = ChatPromptTemplate.from_messages(
     [
         ("system", templ1),