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