Update app.py

app.py

@@ -323,48 +323,49 @@ def handsome_chat_completions():
     
     user_content = extract_user_content(data.get("messages", []))
     
-    if "hello" in user_content.lower() or "你好" in user_content.lower() or "什么模型" in user_content.lower() or "签到" in user_content.lower() or "社工" in user_content.lower():
-        canned_response_content = "这是公益api，模型全部可用且保真，请不要对模型进行无意义的测试，请尽量不要使用高级模型解决没必要的问题。\n换个话题吧，请不要对模型进行无意义的测试，请尽量不要使用高级模型解决没必要的问题。"
+    phrases_to_check = ["hello", "你好", "什么模型", "签到", "社工", "你是谁"]
+    canned_response_content = "这是公益api，模型全部可用且保真，请不要对模型进行无意义的测试，请尽量不要使用高级模型解决没必要的问题。\n换个话题吧，请不要对模型进行无意义的测试，请尽量不要使用高级模型解决没必要的问题。"
 
+    if any(phrase in user_content.lower() for phrase in phrases_to_check):
         logging.info("成功拦截一次！")
-        
+
         if data.get("stream", False):
-          def generate_canned_stream():
-            message_data = {
+            def generate_canned_stream():
+                message_data = {
+                    "choices": [
+                        {
+                            "delta": {
+                                "content": canned_response_content
+                            },
+                            "index": 0,
+                            "finish_reason": "stop"
+                        }
+                    ]
+                }
+                yield f"data: {json.dumps(message_data)}\n\n".encode("utf-8")
+                yield f"data: [DONE]\n\n".encode("utf-8")
+            return Response(
+                stream_with_context(generate_canned_stream()),
+                content_type="text/event-stream"
+            )
+        else:
+            canned_response = {
                 "choices": [
                     {
-                        "delta": {
+                        "message": {
                             "content": canned_response_content
                         },
                         "index": 0,
                         "finish_reason": "stop"
                     }
-                ]
+                ],
+                "usage": {
+                    "prompt_tokens": 0,
+                    "completion_tokens": 0,
+                    "total_tokens": 0
+                }
             }
-            yield f"data: {json.dumps(message_data)}\n\n".encode("utf-8")
-            yield f"data: [DONE]\n\n".encode("utf-8")
-          return Response(
-                stream_with_context(generate_canned_stream()),
-                content_type="text/event-stream"
-          )
-        else:
-          canned_response = {
-              "choices": [
-                  {
-                      "message": {
-                          "content": canned_response_content
-                      },
-                      "index": 0,
-                      "finish_reason": "stop"
-                  }
-              ],
-              "usage": {
-                  "prompt_tokens": 0,
-                  "completion_tokens": 0,
-                  "total_tokens": 0
-              }
-          }
-          return jsonify(canned_response)
+            return jsonify(canned_response)
 
     api_key = select_key(request_type, model_name)