Update functions.py

functions.py

@@ -326,7 +326,7 @@ def create_toxicity_checker(llm):
     return toxicity_grader
 
 
-def grade_question_toxicity(state):
+def grade_question_toxicity(state, toxicity_grader):
     """
     Grades the question for toxicity.