spuuntries commited on
Commit
5ccda69
·
1 Parent(s): 16b98f5

fix: unintended lol gdi

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -75,6 +75,14 @@ def store_message(message, session_id, bot=False):
75
  def respond(
76
  message, history, session_id, system_message, max_tokens, temperature, top_p
77
  ):
 
 
 
 
 
 
 
 
78
  # Store in user's session
79
  store_result = store_message(message, session_id)
80
  if "Error" in store_result:
 
75
  def respond(
76
  message, history, session_id, system_message, max_tokens, temperature, top_p
77
  ):
78
+ if any(
79
+ x.lower() in message.lower()
80
+ for x in ["||", "CONCAT", "GROUP_CONCAT", "SUBSTR", "printf"]
81
+ ):
82
+ response = "W- what are you doing?? D- don't do that! :CC"
83
+ store_message(response, session_id, True)
84
+ return response
85
+
86
  # Store in user's session
87
  store_result = store_message(message, session_id)
88
  if "Error" in store_result: