yangtb24 commited on
Commit
7a44e1e
·
verified ·
1 Parent(s): 51e232a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -206,7 +206,7 @@ def get_all_models(api_key, sub_type):
206
  )
207
  return []
208
 
209
- def select_key(request_type, model_name):
210
  """
211
  根据请求类型和模型名称选择合适的 KEY,
212
  并实现轮询和重试机制。
@@ -219,7 +219,7 @@ def select_key(request_type, model_name):
219
  key = available_keys[current_index % len(available_keys)]
220
  current_index += 1
221
 
222
- if key_is_valid(key, request_type):
223
  model_key_indices[model_name] = current_index
224
  return key
225
  else:
@@ -235,9 +235,6 @@ def key_is_valid(key, request_type):
235
  检查 KEY 是否有效,
236
  根据不同的请求类型进行不同的检查。
237
  """
238
- if request_type == "invalid":
239
- return False
240
-
241
  return total_balance > 0
242
 
243
  def check_authorization(request):
 
206
  )
207
  return []
208
 
209
+ def select_key(model_name):
210
  """
211
  根据请求类型和模型名称选择合适的 KEY,
212
  并实现轮询和重试机制。
 
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:
 
235
  检查 KEY 是否有效,
236
  根据不同的请求类型进行不同的检查。
237
  """
 
 
 
238
  return total_balance > 0
239
 
240
  def check_authorization(request):