smgc commited on
Commit
22d1a47
·
verified ·
1 Parent(s): 1953b4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -20
app.py CHANGED
@@ -2,14 +2,9 @@ from flask import Flask, request, jsonify, Response
2
  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 "flux2api with siliconflow", 200
12
-
13
  @app.route('/v1/chat/completions', methods=['POST'])
14
  def handle_request():
15
  try:
@@ -32,24 +27,10 @@ def handle_request():
32
  "guidance_scale": 1
33
  }
34
 
35
- # 从传入的 Authorization 头中随机选择一个 token
36
- authorization_header = request.headers.get('Authorization')
37
- if authorization_header:
38
- # 去掉 "Bearer " 前缀并分割 token
39
- tokens = authorization_header.replace("Bearer ", "").split(',')
40
- if len(tokens) > 1:
41
- selected_token = random.choice(tokens).strip()
42
- else:
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
-
49
  headers = {
50
  'accept': 'application/json',
51
  'content-type': 'application/json',
52
- 'Authorization': selected_token
53
  }
54
 
55
  response = requests.post(new_url, headers=headers, json=new_request_body)
 
2
  import requests
3
  import json
4
  import time
 
5
 
6
  app = Flask(__name__)
7
 
 
 
 
 
8
  @app.route('/v1/chat/completions', methods=['POST'])
9
  def handle_request():
10
  try:
 
27
  "guidance_scale": 1
28
  }
29
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
30
  headers = {
31
  'accept': 'application/json',
32
  'content-type': 'application/json',
33
+ 'Authorization': request.headers.get('Authorization')
34
  }
35
 
36
  response = requests.post(new_url, headers=headers, json=new_request_body)