Update setup_code.py

setup_code.py

@@ -188,6 +188,7 @@ class Relevant_Documents_Agent:
           response = get_completion(self.client, prompt)
           count += int(is_Yes(response))
 
+      st.write(count)
       # return is_Yes(response)
       return count >= 2