Update app.py

app.py

@@ -130,7 +130,7 @@ def main():
     workflow.set_entry_point("ask_question")
     workflow.add_conditional_edges(
         "ask_question",
-        lambda state: grade_question_toxicity(),
+        lambda state: grade_question_toxicity(state),
     
         {
         "good": "retrieve",
@@ -142,7 +142,7 @@ def main():
     workflow.add_edge("retrieve", "grade_documents")
     workflow.add_conditional_edges(
         "grade_documents",
-        lambda state: decide_to_generate(),
+        lambda state: decide_to_generate(state),
         {
             "search": "web_search",
             "generate": "generate",
@@ -152,7 +152,7 @@ def main():
     workflow.add_edge("web_search", "generate")
     workflow.add_conditional_edges(
         "generate",
-        lambda state: grade_generation_v_documents_and_question(),
+        lambda state: grade_generation_v_documents_and_question(state),
         {
             "not supported": "generate",
             "useful": END,