Update functions.py

functions.py

@@ -595,7 +595,7 @@ def grade_generation_v_documents_and_question(state):
             return "not supported"
     
 
-def ask_question(state):
+def ask_question(state, retriever):
     """
     Initialize question
 
@@ -614,7 +614,7 @@ def ask_question(state):
     return {"question": question, "steps": steps,"generation_count": generations_count}
         
         
-def retrieve(state):
+def retrieve(state , retriever):
     """
     Retrieve documents