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