Update setup_code.py

setup_code.py

@@ -179,10 +179,17 @@ class Relevant_Documents_Agent:
     def is_relevant(self, matches_text, user_query_plus_conversation) -> bool:
       prompt = f"Please confirm that the text in angle brackets: <{matches_text}>, is relevant to the text in double square brackets: [[{user_query_plus_conversation}]]. Return Yes or No"
       #st.write(f"is_relevant prompt {prompt}")
-      response = get_completion(self.client, prompt)
+      # response = get_completion(self.client, prompt)
       #st.write(f"is_relevant response {response}")
 
-      return is_Yes(response)
+      count = 0
+
+      for i in range(3):
+          response = get_completion(self.client, prompt)
+          count += int(is_Yes(response))
+
+      # return is_Yes(response)
+      return count >= 2
 
 class OpenAI_Agent:
   def __init__(self, model="gpt-3.5-turbo", key_filename="open_ai_key.txt"):