Update app.py
Browse files
app.py
CHANGED
@@ -218,25 +218,12 @@ def select_key(model_name):
|
|
218 |
for _ in range(len(available_keys)):
|
219 |
key = available_keys[current_index % len(available_keys)]
|
220 |
current_index += 1
|
221 |
-
|
222 |
-
|
223 |
-
model_key_indices[model_name] = current_index
|
224 |
-
return key
|
225 |
-
else:
|
226 |
-
logging.warning(
|
227 |
-
f"KEY {key} 无效或达到限制,尝试下一个 KEY"
|
228 |
-
)
|
229 |
|
230 |
model_key_indices[model_name] = 0
|
231 |
return None
|
232 |
|
233 |
-
def key_is_valid(key, request_type):
|
234 |
-
"""
|
235 |
-
检查 KEY 是否有效,
|
236 |
-
根据不同的请求类型进行不同的检查。
|
237 |
-
"""
|
238 |
-
return total_balance > 0
|
239 |
-
|
240 |
def check_authorization(request):
|
241 |
"""
|
242 |
检查请求头中的 Authorization 字段
|
|
|
218 |
for _ in range(len(available_keys)):
|
219 |
key = available_keys[current_index % len(available_keys)]
|
220 |
current_index += 1
|
221 |
+
model_key_indices[model_name] = current_index
|
222 |
+
return key
|
|
|
|
|
|
|
|
|
|
|
|
|
223 |
|
224 |
model_key_indices[model_name] = 0
|
225 |
return None
|
226 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
227 |
def check_authorization(request):
|
228 |
"""
|
229 |
检查请求头中的 Authorization 字段
|