Fix the bug in the matching model algorithm.
Browse files
main.py
CHANGED
@@ -94,9 +94,13 @@ class ModelRequestHandler:
|
|
94 |
if "/" in model:
|
95 |
provider_name = model.split("/")[0]
|
96 |
model = model.split("/")[1]
|
|
|
97 |
for provider in config['providers']:
|
98 |
if provider['provider'] == provider_name:
|
99 |
-
models_list
|
|
|
|
|
|
|
100 |
if (model and model_name in models_list) or (model == "*" and model_name in models_list):
|
101 |
provider_rules.append(provider_name)
|
102 |
else:
|
|
|
94 |
if "/" in model:
|
95 |
provider_name = model.split("/")[0]
|
96 |
model = model.split("/")[1]
|
97 |
+
models_list = []
|
98 |
for provider in config['providers']:
|
99 |
if provider['provider'] == provider_name:
|
100 |
+
models_list.extend(list(provider['model'].keys()))
|
101 |
+
# print("models_list", models_list)
|
102 |
+
# print("model_name", model_name)
|
103 |
+
# print("model", model)
|
104 |
if (model and model_name in models_list) or (model == "*" and model_name in models_list):
|
105 |
provider_rules.append(provider_name)
|
106 |
else:
|
utils.py
CHANGED
@@ -17,7 +17,7 @@ def update_config(config_data):
|
|
17 |
config_data['providers'][index] = provider
|
18 |
api_keys_db = config_data['api_keys']
|
19 |
api_list = [item["api"] for item in api_keys_db]
|
20 |
-
# logger.info(json.dumps(
|
21 |
return config_data, api_keys_db, api_list
|
22 |
|
23 |
# 读取YAML配置文件
|
|
|
17 |
config_data['providers'][index] = provider
|
18 |
api_keys_db = config_data['api_keys']
|
19 |
api_list = [item["api"] for item in api_keys_db]
|
20 |
+
# logger.info(json.dumps(config_data, indent=4, ensure_ascii=False))
|
21 |
return config_data, api_keys_db, api_list
|
22 |
|
23 |
# 读取YAML配置文件
|