Update app.py

app.py

@@ -124,6 +124,7 @@ _search_query = RunnableBranch(
         )
         | CONDENSE_QUESTION_PROMPT
         | ChatOpenAI(temperature=0, api_key=os.environ['OPENAI_API_KEY'])
+        | conversational_memory 
         | StrOutputParser(),
     ),
     RunnableLambda(lambda x: x["question"]),
@@ -149,7 +150,6 @@ chain_neo4j = (
     )
     | qa_prompt
     | chat_model
-    | conversational_memory 
     | StrOutputParser()
 )