yym68686 commited on
Commit
2f54462
·
1 Parent(s): ffacd74

🐛 Bug: Fix the bug where the model cannot be found.

Browse files
Files changed (1) hide show
  1. main.py +7 -3
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
- if "/" in item and provider['provider'] == item.split("/")[0]:
 
 
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)