Update utils.py

utils.py

@@ -330,11 +330,11 @@ def grade_documents_direct(prompt, documents):
     filtered_docs = []
 
     for d in documents:
-        score = chain.invoke({"question": prompt, "context": d.page_content})
+        llm_chain = LLMChain(llm = model, prompt = grade_prompt)
+        score = llm_chain.run({"question": prompt, "context": d.page_content}})
         print("hier1..................................")
-        grade = score[0].binary_score
-        print("hier2..................................")
-        if grade == "ja":
+        
+        if not score == "nein":
             print("---Bewertung: Dokument ist relevant---")
             filtered_docs.append(d)
         else: