Update app.py
Browse files
app.py
CHANGED
@@ -3,14 +3,26 @@ import requests
|
|
3 |
import json
|
4 |
import time
|
5 |
import random
|
|
|
6 |
|
7 |
app = Flask(__name__)
|
8 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
9 |
@app.route('/')
|
10 |
def index():
|
11 |
-
return "
|
12 |
|
13 |
-
@app.route('/
|
14 |
def handle_request():
|
15 |
try:
|
16 |
body = request.json
|
@@ -43,6 +55,8 @@ def handle_request():
|
|
43 |
selected_token = tokens[0].strip()
|
44 |
# 重新格式化为 "Bearer 随机选择的token"
|
45 |
selected_token = f"Bearer {selected_token}"
|
|
|
|
|
46 |
else:
|
47 |
return jsonify({"error": "Unauthorized: Missing Authorization header"}), 401
|
48 |
|
|
|
3 |
import json
|
4 |
import time
|
5 |
import random
|
6 |
+
import logging
|
7 |
|
8 |
app = Flask(__name__)
|
9 |
|
10 |
+
# 配置日志
|
11 |
+
logging.basicConfig(level=logging.INFO)
|
12 |
+
logger = logging.getLogger(__name__)
|
13 |
+
|
14 |
+
# 添加 StreamHandler 以实时输出日志到控制台
|
15 |
+
console_handler = logging.StreamHandler()
|
16 |
+
console_handler.setLevel(logging.INFO)
|
17 |
+
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
18 |
+
console_handler.setFormatter(formatter)
|
19 |
+
logger.addHandler(console_handler)
|
20 |
+
|
21 |
@app.route('/')
|
22 |
def index():
|
23 |
+
return "flux2api with siliconflow", 200
|
24 |
|
25 |
+
@app.route('/v1/chat/completions', methods=['POST'])
|
26 |
def handle_request():
|
27 |
try:
|
28 |
body = request.json
|
|
|
55 |
selected_token = tokens[0].strip()
|
56 |
# 重新格式化为 "Bearer 随机选择的token"
|
57 |
selected_token = f"Bearer {selected_token}"
|
58 |
+
# 记录命中的 token
|
59 |
+
logger.info(f"Request hit token: {selected_token}")
|
60 |
else:
|
61 |
return jsonify({"error": "Unauthorized: Missing Authorization header"}), 401
|
62 |
|