yangtb24 commited on
Commit
f836ee7
·
verified ·
1 Parent(s): 7a01782

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -18
app.py CHANGED
@@ -205,17 +205,6 @@ def get_all_models(api_key, sub_type):
205
  )
206
  return []
207
 
208
- def determine_request_type(model_name, model_list, free_model_list):
209
- """
210
- 根据用户请求的模型判断请求类型。
211
- """
212
- if model_name in free_model_list:
213
- return "free"
214
- elif model_name in model_list:
215
- return "paid"
216
- else:
217
- return "unknown"
218
-
219
  def select_key(request_type, model_name):
220
  """
221
  根据请求类型和模型名称选择合适的 KEY,
@@ -505,13 +494,7 @@ def handsome_chat_completions():
505
 
506
  model_name = data['model']
507
 
508
- request_type = determine_request_type(
509
- model_name,
510
- text_models + image_models,
511
- free_text_models + free_image_models
512
- )
513
-
514
- api_key = select_key(request_type, model_name)
515
 
516
  if not api_key:
517
  return jsonify(
 
205
  )
206
  return []
207
 
 
 
 
 
 
 
 
 
 
 
 
208
  def select_key(request_type, model_name):
209
  """
210
  根据请求类型和模型名称选择合适的 KEY,
 
494
 
495
  model_name = data['model']
496
 
497
+ api_key = select_key(model_name)
 
 
 
 
 
 
498
 
499
  if not api_key:
500
  return jsonify(