yym68686 commited on
Commit
e86d80a
·
1 Parent(s): fc14950

Fix the issue where the original model name cannot be requested.

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -92,12 +92,12 @@ class ModelRequestHandler:
92
  model = model.split("/")[1]
93
  for provider in config['providers']:
94
  if provider['provider'] == provider_name:
95
- models_list = list(provider['model'].keys()) + list(provider['model'].values())
96
  if (model and model_name in models_list) or (model == "*" and model_name in models_list):
97
  provider_rules.append(provider_name)
98
  provider_list = []
99
  for provider in config['providers']:
100
- if model_name in list(provider['model'].keys()) + list(provider['model'].values()) and ((provider_rules and provider['provider'] in provider_rules) or provider_rules == []):
101
  provider_list.append(provider)
102
  return provider_list
103
 
 
92
  model = model.split("/")[1]
93
  for provider in config['providers']:
94
  if provider['provider'] == provider_name:
95
+ models_list = provider['model'].keys()
96
  if (model and model_name in models_list) or (model == "*" and model_name in models_list):
97
  provider_rules.append(provider_name)
98
  provider_list = []
99
  for provider in config['providers']:
100
+ if model_name in provider['model'].keys() and ((provider_rules and provider['provider'] in provider_rules) or provider_rules == []):
101
  provider_list.append(provider)
102
  return provider_list
103