Fix the bug that cannot adapt to the available model name.
Browse files
main.py
CHANGED
|
@@ -222,13 +222,25 @@ class ModelRequestHandler:
|
|
| 222 |
# print("provider_rules", provider_rules)
|
| 223 |
for item in provider_rules:
|
| 224 |
for provider in config['providers']:
|
| 225 |
-
|
| 226 |
-
|
| 227 |
-
|
| 228 |
-
|
| 229 |
-
else:
|
| 230 |
-
if model_name in provider['model'].keys():
|
| 231 |
provider_list.append(provider)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 232 |
# import json
|
| 233 |
# for provider in provider_list:
|
| 234 |
# print(json.dumps(provider, indent=4, ensure_ascii=False))
|
|
|
|
| 222 |
# print("provider_rules", provider_rules)
|
| 223 |
for item in provider_rules:
|
| 224 |
for provider in config['providers']:
|
| 225 |
+
# print("provider", provider, provider['provider'] == item, item)
|
| 226 |
+
if "/" in item:
|
| 227 |
+
if provider['provider'] == item.split("/")[0]:
|
| 228 |
+
if model_name in provider['model'].keys() and item.split("/")[1] == model_name:
|
|
|
|
|
|
|
| 229 |
provider_list.append(provider)
|
| 230 |
+
elif provider['provider'] == item:
|
| 231 |
+
if model_name in provider['model'].keys():
|
| 232 |
+
provider_list.append(provider)
|
| 233 |
+
else:
|
| 234 |
+
pass
|
| 235 |
+
|
| 236 |
+
# if provider['provider'] == item:
|
| 237 |
+
# if "/" in item:
|
| 238 |
+
# if item.split("/")[1] == model_name:
|
| 239 |
+
# provider_list.append(provider)
|
| 240 |
+
# else:
|
| 241 |
+
# if model_name in provider['model'].keys():
|
| 242 |
+
# provider_list.append(provider)
|
| 243 |
+
|
| 244 |
# import json
|
| 245 |
# for provider in provider_list:
|
| 246 |
# print(json.dumps(provider, indent=4, ensure_ascii=False))
|