Update app.py

app.py

@@ -110,8 +110,10 @@ def main():
     )
     
     retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type=search_type, k=k, chunk_size=550, chunk_overlap=40)
- 
-    
+    retrieval_grader = retrieval_grader()
+    toxicity_grader = toxicity_grader()
+    hallucination_grader =  hallucination_grader()
+    answer_grader = answer_grader()
 
 
     # Graph
@@ -120,7 +122,7 @@ 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", grade_documents)  # grade documents
+    workflow.add_node("grade_documents", lambda state: grade_documents(state, retrieval_grader ))  # grade documents
     workflow.add_node("generate", generate)  # generatae
     workflow.add_node("web_search", web_search)  # web search
     workflow.add_node("transform_query", transform_query)
@@ -130,7 +132,7 @@ def main():
     workflow.set_entry_point("ask_question")
     workflow.add_conditional_edges(
         "ask_question",
-        lambda state: grade_question_toxicity(state),
+        lambda state: grade_question_toxicity(state, toxicity_grader),
     
         {
         "good": "retrieve",
@@ -142,7 +144,7 @@ def main():
     workflow.add_edge("retrieve", "grade_documents")
     workflow.add_conditional_edges(
         "grade_documents",
-        lambda state: decide_to_generate(state),
+        decide_to_generate,
         {
             "search": "web_search",
             "generate": "generate",
@@ -152,7 +154,7 @@ def main():
     workflow.add_edge("web_search", "generate")
     workflow.add_conditional_edges(
         "generate",
-        lambda state: grade_generation_v_documents_and_question(state),
+        lambda state: grade_generation_v_documents_and_question(state, hallucination_grader, answer_grader),
         {
             "not supported": "generate",
             "useful": END,