Fix the issue where the original model name cannot be requested.
Browse files
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 =
|
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
|
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 |
|