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 字段
|