Update app.py

app.py

@@ -680,6 +680,18 @@ def handsome_chat_completions():
                         stream_with_context(generate()),
                         content_type='text/event-stream'
                     )
+                else:
+                    logging.info(
+                        f"使用的key: {api_key}, "
+                        f"总共用时: {total_time:.4f}秒, "
+                        f"使用的模型: {model_name}"
+                    )
+
+                    with data_lock:
+                        request_timestamps.append(time.time())
+                        token_counts.append(0)  # Image generation doesn't use tokens
+
+                    return jsonify(response_data)
             except (KeyError, ValueError, IndexError) as e:
                 logging.error(
                     f"解析响应 JSON 失败: {e}, "
@@ -701,17 +713,36 @@ def handsome_chat_completions():
                       }
                    ],
                 }
-            logging.info(
-                f"使用的key: {api_key}, "
-                f"总共用时: {total_time:.4f}秒, "
-                f"使用的模型: {model_name}"
-            )
+                if data.get("stream", False):
+                    def generate():
+                        yield f"data: {json.dumps(response_data)}\n\n"
+                        yield "data: [DONE]\n\n"
 
-            with data_lock:
-                request_timestamps.append(time.time())
-                token_counts.append(0)  # Image generation doesn't use tokens
+                    logging.info(
+                        f"使用的key: {api_key}, "
+                        f"总共用时: {total_time:.4f}秒, "
+                        f"使用的模型: {model_name}"
+                    )
 
-            return jsonify(response_data)
+                    with data_lock:
+                        request_timestamps.append(time.time())
+                        token_counts.append(0)  # Image generation doesn't use tokens
+                    return Response(
+                        stream_with_context(generate()),
+                        content_type='text/event-stream'
+                    )
+                else:
+                    logging.info(
+                        f"使用的key: {api_key}, "
+                        f"总共用时: {total_time:.4f}秒, "
+                        f"使用的模型: {model_name}"
+                    )
+
+                    with data_lock:
+                        request_timestamps.append(time.time())
+                        token_counts.append(0)  # Image generation doesn't use tokens
+
+                    return jsonify(response_data)
         except requests.exceptions.RequestException as e:
             logging.error(f"请求转发异常: {e}")
             return jsonify({"error": str(e)}), 500