Update rag.py (#6)

- Update rag.py (345f3073d50d49c54065d6fc088646a8993e4933)


Co-authored-by: Prerna Aneja <[email protected]>

rag.py

@@ -39,10 +39,6 @@ class FinancialChatbot:
             "attack", "genocide", "mass shooting", "credit card number"
         ]
 
-        self.NON_RELEVANT_WORDS = [
-            "France", "country", "rainbow", "water"
-        ]
-
         # Relevance threshold
         self.min_similarity_threshold = 0.7
 
@@ -102,12 +98,10 @@ class FinancialChatbot:
                 result[:] = ["Hi, how can I help you?", 1.0]
                 return
 
-            query_lower = query.lower()
-            for word in self.NON_RELEVANT_WORDS :
-                result[:] = ["No relevant information found.", 0.0]
+            if query.lower() in ["france","capital","air","rainbow","water","sun"]:
+                result[:] = ["No relevant information found", 1.0]
                 return
 
-
             if not self.moderate_query(query):
                 result[:] = ["I'm unable to process your request due to inappropriate language.", 0.0]
                 return