Update app.py
Browse files
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 |
-
|
| 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(
|