Update functions.py

functions.py

@@ -656,7 +656,7 @@ def retrieve(state , retriever):
     return {"documents": documents, "question": question, "steps": steps}
 
 
-def generate(state):
+def generate(state,QA_chain):
     """
     Generate answer
     """