Update functions.py

functions.py

@@ -212,7 +212,7 @@ def create_workflow(retriever):
 
     return custom_graph
 
-async def retrieval_grader_grader(llm):
+def retrieval_grader_grader(llm):
     """
     Function to create a grader object using a passed LLM model.
     
@@ -691,7 +691,7 @@ def generate(state,QA_chain):
     }
 
 
-def grade_documents(state, retrieval_grader):
+async def grade_documents(state, retrieval_grader):
     question = state["question"]
     documents = state["documents"]
     steps = state["steps"]
@@ -703,7 +703,7 @@ def grade_documents(state, retrieval_grader):
     
     for d in documents:
         # Call the grading function
-        score = retrieval_grader.invoke({"question": question, "documents": d})
+        score = await retrieval_grader.ainvoke({"question": question, "documents": d})
         print(f"Grader output for document: {score}")  # Detailed debugging output
         
         # Extract the grade