Update app.py

app.py

@@ -324,23 +324,45 @@ def handsome_chat_completions():
     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
+        canned_response_content = "这是公益api，模型全部可用且保真，请不要对模型进行无意义的测试，请尽量不要使用高级模型解决没必要的问题。"
+        
+        if data.get("stream", False):
+          def generate_canned_stream():
+            message_data = {
+                "choices": [
+                    {
+                        "delta": {
+                            "content": canned_response_content
+                        },
+                        "index": 0,
+                        "finish_reason": "stop"
+                    }
+                ]
             }
-        }
-        return jsonify(canned_response)
+            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)
 
     api_key = select_key(request_type, model_name)
 
@@ -534,6 +556,7 @@ def handsome_chat_completions():
         logging.error(f"请求转发异常: {e}")
         return jsonify({"error": str(e)}), 500
 
+
 if __name__ == '__main__':
     logging.info(f"环境变量：{os.environ}")