Update app.py

app.py

@@ -118,8 +118,8 @@ def main():
     workflow = StateGraph(GraphState)
 
     # Define the nodes
-    workflow.add_node("ask_question", ask_question(retriever))
-    workflow.add_node("retrieve", retrieve(retriever))  
+    workflow.add_node("ask_question", lambda state: ask_question(state, retriever))
+    workflow.add_node("retrieve", lambda state: retrieve(state, retriever))  
     workflow.add_node("grade_documents", grade_documents)  # grade documents
     workflow.add_node("generate", generate)  # generatae
     workflow.add_node("web_search", web_search)  # web search