mainfile cgpt 4

app.py

@@ -70,7 +70,7 @@ def create_conversational_qa_chain(retriever, condense_llm, answer_llm):
     ) | CONDENSE_QUESTION_PROMPT | condense_llm
 
     retrieval_chain = RunnableLambda(
-        lambda x: {"standalone_question": x}
+        lambda x: x['standalone_question']
     ) | retriever | _combine_documents
 
     answer_chain = ANSWER_PROMPT | answer_llm
@@ -136,8 +136,8 @@ def app():
                     "chat_history": chistory,
                 }
             )
-                st.write(response)
-        message = {"role": "assistant", "content": response}
+                st.write(response['generate_answer'])
+        message = {"role": "assistant", "content": response['generate_answer']}
         st.session_state.messages.append(message)
 
 if __name__ == '__main__':