yangtb24 commited on
Commit
5e02bab
·
verified ·
1 Parent(s): 7a44e1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -15
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
- if key_is_valid(key):
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 字段