yangtb24 commited on
Commit
36a66d2
·
verified ·
1 Parent(s): dc11322

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -324,6 +324,7 @@ def handsome_chat_completions():
324
  user_content = extract_user_content(data.get("messages", []))
325
 
326
  phrases_to_check = ["hello", "你好", "什么模型", "签到", "社工", "你是谁", "冷笑话", "只回答", "Netflix", "response", "A="]
 
327
  canned_response_content = "这是公益api,模型全部可用且保真,请不要对模型进行无意义的测试,请尽量不要使用高级模型解决没必要的问题。\n换个话题吧,请不要对模型进行无意义的测试,请尽量不要使用高级模型解决没必要的问题。"
328
 
329
  user_content_lower = user_content.lower()
@@ -349,6 +350,7 @@ def handsome_chat_completions():
349
  "model_name": model_name
350
  }
351
 
 
352
  yield f"data: {json.dumps(model_data)}\n\n".encode("utf-8")
353
  yield f"data: [DONE]\n\n".encode("utf-8")
354
  return Response(
@@ -374,7 +376,7 @@ def handsome_chat_completions():
374
  "model_name": model_name
375
  }
376
  return jsonify(canned_response)
377
- elif any(phrase in user_content_lower for phrase in phrases_to_check):
378
  logging.info("成功拦截一次!")
379
  if data.get("stream", False):
380
  def generate_canned_stream():
 
324
  user_content = extract_user_content(data.get("messages", []))
325
 
326
  phrases_to_check = ["hello", "你好", "什么模型", "签到", "社工", "你是谁", "冷笑话", "只回答", "Netflix", "response", "A="]
327
+ phrases_to_check_lower = [phrase.lower() for phrase in phrases_to_check] # Lowercase the phrases
328
  canned_response_content = "这是公益api,模型全部可用且保真,请不要对模型进行无意义的测试,请尽量不要使用高级模型解决没必要的问题。\n换个话题吧,请不要对模型进行无意义的测试,请尽量不要使用高级模型解决没必要的问题。"
329
 
330
  user_content_lower = user_content.lower()
 
350
  "model_name": model_name
351
  }
352
 
353
+ yield f"data: {json.dumps(message_data)}\n\n".encode("utf-8")
354
  yield f"data: {json.dumps(model_data)}\n\n".encode("utf-8")
355
  yield f"data: [DONE]\n\n".encode("utf-8")
356
  return Response(
 
376
  "model_name": model_name
377
  }
378
  return jsonify(canned_response)
379
+ elif any(phrase in user_content_lower for phrase in phrases_to_check_lower): # Use the lowercased phrases
380
  logging.info("成功拦截一次!")
381
  if data.get("stream", False):
382
  def generate_canned_stream():