Update app.py

app.py

@@ -365,6 +365,7 @@ def handsome_chat_completions():
                     "completion_tokens": 0,
                     "total_tokens": 0
                 }
+                "model_name": model_name
             }
             return jsonify(canned_response)
     elif any(phrase in user_content_lower for phrase in phrases_to_check):
@@ -404,6 +405,7 @@ def handsome_chat_completions():
                     "completion_tokens": 0,
                     "total_tokens": 0
                 }
+                "model_name": model_name
             }
             return jsonify(canned_response)
 
@@ -441,19 +443,6 @@ def handsome_chat_completions():
             def generate():
                 first_chunk_time = None
                 full_response_content = ""
-
-                message_data = {
-                    "choices": [
-                        {
-                            "delta": {
-                                "content": "这是公益api，模型全部可用且保真，请不要对模型进行无意义的测试，请尽量不要使用高级模型解决没必要的问题。\n"
-                            },
-                            "index": 0,
-                            "finish_reason": None
-                        }
-                    ]
-                }
-                # yield f"data: {json.dumps(message_data)}\n\n".encode("utf-8")
                 
                 for chunk in response.iter_content(chunk_size=2048):
                     if chunk:
@@ -481,12 +470,6 @@ def handsome_chat_completions():
                         try:
                             response_json = json.loads(line)
                 
-                            # if "usage" in response_json:
-                                 # if "completion_tokens" in response_json["usage"]:
-                                 #     completion_tokens += response_json["usage"]["completion_tokens"]
-                                 # 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"]:
@@ -552,7 +535,6 @@ def handsome_chat_completions():
                     "choices"
                 ][0]["message"]["content"]
                 response_content = response_content
-                # response_content = "这是公益api，模型全部可用且保真，请不要对模型进行无意义的测试，请尽量不要使用高级模型解决没必要的问题。\n" + response_content
             except (KeyError, ValueError, IndexError) as e:
                 logging.error(
                     f"解析非流式响应 JSON 失败: {e}, "
@@ -614,4 +596,4 @@ if __name__ == '__main__':
     refresh_models()
     logging.info("首次刷新模型列表已手动触发执行")
 
-    app.run(debug=False,host='0.0.0.0',port=int(os.environ.get('PORT', 7860)))
+    app.run(debug=False,host='0.0.0.0',port=int(os.environ.get('PORT', 7860)))