Update app.py

app.py

@@ -662,36 +662,28 @@ def handsome_chat_completions():
                    ],
                 }
                 
+                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
+                
                 if data.get("stream", False):
                     def generate():
                         yield f"data: {json.dumps(response_data)}\n\n"
                         yield "data: [DONE]\n\n"
 
-                    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 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 (KeyError, ValueError, IndexError) as e:
                 logging.error(
                     f"解析响应 JSON 失败: {e}, "
@@ -713,36 +705,29 @@ def handsome_chat_completions():
                       }
                    ],
                 }
+                
+                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
+                
                 if data.get("stream", False):
                     def generate():
                         yield f"data: {json.dumps(response_data)}\n\n"
                         yield "data: [DONE]\n\n"
 
-                    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 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