Update app.py

app.py

@@ -153,6 +153,7 @@ _search_query = RunnableBranch(
 #Answer:"""
 
 
+# Define the ChatPromptTemplate
 prompt = ChatPromptTemplate.from_messages(
     [
         (
@@ -165,9 +166,7 @@ prompt = ChatPromptTemplate.from_messages(
     ]
 )
 
-
-
-#qa_prompt = ChatPromptTemplate.from_template(prompt)
+# Ensure 'context' and 'question' map correctly to the prompt's expected variables
 qa_prompt = prompt
 
 
@@ -175,8 +174,9 @@ qa_prompt = prompt
 chain_neo4j = (
     RunnableParallel(
         {
-            "context": _search_query | retriever_neo4j,
-            "question": RunnablePassthrough(),
+            "chat_history": _search_query,
+            "input": RunnablePassthrough(),
+            "agent_scratchpad": retriever_neo4j,
         }
     )
     | qa_prompt
@@ -184,10 +184,11 @@ chain_neo4j = (
     | StrOutputParser()
 )
 
+
 # Define the function to get the response
 def get_response(question):
     try:
-        return chain_neo4j.invoke({"question": question})
+        return chain_neo4j.invoke({"input": question, "chat_history": [], "agent_scratchpad": ""})
     except Exception as e:
         return f"Error: {str(e)}"