Update app.py

app.py

@@ -307,20 +307,41 @@ def handsome_chat_completions():
         return jsonify({"error": "Unauthorized"}), 401
 
     data = request.get_json()
-    logging.info(f"Request data: {data}") 
+    logging.info(f"Request data: {data}")
     if not data or 'model' not in data:
         return jsonify({"error": "Invalid request data"}), 400
     if data['model'] not in models["text"] and data['model'] not in models["image"]:
         return jsonify({"error": "Invalid model"}), 400
 
     model_name = data['model']
-    
+
     request_type = determine_request_type(
         model_name,
         models["text"] + models["image"],
         models["free_text"] + models["free_image"]
     )
     
+    user_content = extract_user_content(data.get("messages", []))
+    
+    if "hi" in user_content.lower() or "hello" in user_content.lower():
+        canned_response = {
+            "choices": [
+                {
+                    "message": {
+                        "content": "这是公益api，模型全部可用且保真，请不要对模型进行无意义的测试，请尽量不要使用高级模型解决没必要的问题。"
+                    },
+                    "index": 0,
+                    "finish_reason": "stop"
+                }
+            ],
+            "usage": {
+                "prompt_tokens": 0,
+                "completion_tokens": 0,
+                "total_tokens": 0
+            }
+        }
+        return jsonify(canned_response)
+
     api_key = select_key(request_type, model_name)
 
     if not api_key:
@@ -401,7 +422,6 @@ def handsome_chat_completions():
                                  # if "prompt_tokens" in response_json["usage"]:
                                  #     prompt_tokens += response_json["usage"]["prompt_tokens"]
                 
-                
                             if "choices" in response_json:
                                 for choice in response_json["choices"]:
                                     if "delta" in choice and "content" in choice["delta"]: