🐛 Bug: Fix the bug where the model cannot be found.
Browse files
main.py
CHANGED
|
@@ -706,11 +706,15 @@ def get_provider_list(provider_rules, config, request_model):
|
|
| 706 |
# print("provider_rules", provider_rules)
|
| 707 |
for item in provider_rules:
|
| 708 |
for provider in config['providers']:
|
| 709 |
-
|
|
|
|
|
|
|
| 710 |
new_provider = copy.deepcopy(provider)
|
| 711 |
-
model_dict = get_model_dict(provider)
|
| 712 |
-
model_name_split = "/".join(item.split("/")[1:])
|
| 713 |
# old: new
|
|
|
|
|
|
|
|
|
|
|
|
|
| 714 |
new_provider["model"] = [{model_dict[model_name_split]: request_model}]
|
| 715 |
if request_model in model_dict.keys() and model_name_split == request_model:
|
| 716 |
provider_list.append(new_provider)
|
|
|
|
| 706 |
# print("provider_rules", provider_rules)
|
| 707 |
for item in provider_rules:
|
| 708 |
for provider in config['providers']:
|
| 709 |
+
model_dict = get_model_dict(provider)
|
| 710 |
+
model_name_split = "/".join(item.split("/")[1:])
|
| 711 |
+
if "/" in item and provider['provider'] == item.split("/")[0] and model_name_split in model_dict.keys():
|
| 712 |
new_provider = copy.deepcopy(provider)
|
|
|
|
|
|
|
| 713 |
# old: new
|
| 714 |
+
# print("item", item)
|
| 715 |
+
# print("model_dict", model_dict)
|
| 716 |
+
# print("model_name_split", model_name_split)
|
| 717 |
+
# print("request_model", request_model)
|
| 718 |
new_provider["model"] = [{model_dict[model_name_split]: request_model}]
|
| 719 |
if request_model in model_dict.keys() and model_name_split == request_model:
|
| 720 |
provider_list.append(new_provider)
|