Update functions.py

functions.py

@@ -401,7 +401,7 @@ def create_helpfulness_checker(llm):
     return helpfulness_grader
 
 
-def grade_document_relevance(question: str, document: str):
+def grade_document_relevance(question: str, document: str, retrieval_grader):
     input_data = {"documents": documents,"question": question, }
     try:
         result = retrieval_grader.invoke(input_data)
@@ -504,7 +504,7 @@ def create_question_rewriter(llm):
     return question_rewriter
 
 
-def transform_query(state):
+def transform_query(state, question_rewriter):
     """
     Transform the query to produce a better question.
 
@@ -553,7 +553,7 @@ def format_google_results(google_results):
     return formatted_documents
 
 
-def grade_generation_v_documents_and_question(state):
+def grade_generation_v_documents_and_question(state,hallucination_grader,answer_grader ):
     """
     Determines whether the generation is grounded in the document and answers the question.
     """