meshsl commited on
Commit
4cf77d8
·
verified ·
1 Parent(s): 203c55a

Update rag_system.py

Browse files
Files changed (1) hide show
  1. rag_system.py +1 -9
rag_system.py CHANGED
@@ -67,17 +67,9 @@ class RAGSystem:
67
  )
68
  return self.prompt_template
69
 
70
- GREETINGS = [
71
- "hi", "hello", "hey", "good morning", "good afternoon", "good evening"
72
- ]
73
 
74
  def ask_question(self, user_input):
75
- # التعامل مع التحية
76
- if user_input.lower() in [g.lower() for g in self.GREETINGS]:
77
- return (
78
- "Hi! Please ask a question related to any of the following Saudi cybersecurity documents: "
79
- "ECC (Essential Cybersecurity Controls), SCYWF (Saudi Cybersecurity Workforce Framework), or OSMACC (Organizations’ Social Media Accounts Cybersecurity Controls)."
80
- )
81
 
82
  # استرجاع الوثائق ذات الصلة
83
  retriever = self.vectorstore.as_retriever(search_kwargs={"k": 3})
 
67
  )
68
  return self.prompt_template
69
 
70
+
 
 
71
 
72
  def ask_question(self, user_input):
 
 
 
 
 
 
73
 
74
  # استرجاع الوثائق ذات الصلة
75
  retriever = self.vectorstore.as_retriever(search_kwargs={"k": 3})