neerajkalyank commited on
Commit
fceab00
·
verified ·
1 Parent(s): aae68bd

Update hrms_guardrails.py

Browse files
Files changed (1) hide show
  1. hrms_guardrails.py +4 -5
hrms_guardrails.py CHANGED
@@ -1,11 +1,10 @@
1
  HRMS_KEYWORDS = [
2
- "leave", "attendance", "payroll", "salary", "onboarding",
3
- "hr", "policy", "performance", "appraisal", "compliance"
4
  ]
5
 
6
  def is_hrms_query(query: str) -> bool:
7
- q = query.lower()
8
- return any(word in q for word in HRMS_KEYWORDS)
9
 
10
  def hrms_refusal():
11
- return "I can assist only with HRMS-related topics such as leave, payroll policies, attendance, onboarding, and performance."
 
1
  HRMS_KEYWORDS = [
2
+ "leave", "attendance", "payroll", "policy",
3
+ "onboarding", "hr", "performance", "appraisal"
4
  ]
5
 
6
  def is_hrms_query(query: str) -> bool:
7
+ return any(word in query.lower() for word in HRMS_KEYWORDS)
 
8
 
9
  def hrms_refusal():
10
+ return "I can help only with HRMS-related topics like leave, attendance, payroll policies, onboarding, and performance."