Update app.py

app.py

@@ -119,17 +119,17 @@ def main():
     # Define the nodes
     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", lambda state: grade_documents(state, retrieval_grader_grader() ))  # grade documents
+    workflow.add_node("grade_documents", lambda state: grade_documents(state, retrieval_grader_grader(llm) ))  # grade documents
     workflow.add_node("generate", generate)  # generatae
     workflow.add_node("web_search", web_search)  # web search
-    workflow.add_node("transform_query", transform_query)
+    workflow.add_node("transform_query", lambda state: transform_query(state,create_question_rewriter(llm) ))
 
 
     # Build graph
     workflow.set_entry_point("ask_question")
     workflow.add_conditional_edges(
         "ask_question",
-        lambda state: grade_question_toxicity(state, create_toxicity_checker()),
+        lambda state: grade_question_toxicity(state, create_toxicity_checker(llm)),
     
         {
         "good": "retrieve",
@@ -151,7 +151,7 @@ def main():
     workflow.add_edge("web_search", "generate")
     workflow.add_conditional_edges(
         "generate",
-        lambda state: grade_generation_v_documents_and_question(state, create_hallucination_checker(), create_helpfulness_checker()),
+        lambda state: grade_generation_v_documents_and_question(state, create_hallucination_checker(llm), create_helpfulness_checker(llm)),
         {
             "not supported": "generate",
             "useful": END,