yym68686 commited on
Commit
163e912
·
1 Parent(s): 7b515b4

Fix the bug that cannot adapt to the available model name.

Browse files
Files changed (1) hide show
  1. main.py +18 -6
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
- if provider['provider'] == item:
226
- if "/" in item:
227
- if item.split("/")[1] == model_name:
228
- provider_list.append(provider)
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))