Update app.py

app.py

@@ -323,31 +323,71 @@ def handsome_chat_completions():
     
     user_content = extract_user_content(data.get("messages", []))
     
-    phrases_to_check = ["hello", "你好", "什么模型", "签到", "社工", "你是谁", "'content': 'hi'"]
+    phrases_to_check = ["hello", "你好", "什么模型", "签到", "社工", "你是谁"]
     canned_response_content = "这是公益api，模型全部可用且保真，请不要对模型进行无意义的测试，请尽量不要使用高级模型解决没必要的问题。\n换个话题吧，请不要对模型进行无意义的测试，请尽量不要使用高级模型解决没必要的问题。"
-
-    if any(phrase in user_content.lower() for phrase in phrases_to_check):
-        logging.info("成功拦截一次！")
-
+    
+    user_content_lower = user_content.lower()
+    
+    if user_content_lower == "hi":
+        logging.info("成功拦截一次！(仅hi)")
         if data.get("stream", False):
-            def generate_canned_stream():
-                message_data = {
-                    "choices": [
-                        {
-                            "delta": {
-                                "content": canned_response_content
-                            },
-                            "index": 0,
-                            "finish_reason": "stop"
-                        }
-                    ]
+                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": [
+                    {
+                        "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"
-            )
+            }
+            return jsonify(canned_response)
+    elif 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 = {
+                        "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": [