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))
|