|
|
|
@ -201,10 +201,15 @@ class ModelProviderFactory:
|
|
|
|
model_providers_path = os.path.dirname(current_path)
|
|
|
|
model_providers_path = os.path.dirname(current_path)
|
|
|
|
|
|
|
|
|
|
|
|
# get all folders path under model_providers_path that do not start with __
|
|
|
|
# get all folders path under model_providers_path that do not start with __
|
|
|
|
|
|
|
|
whitelist = [
|
|
|
|
|
|
|
|
"baichuan", "chatglm", "deepseek", "hunyuan", "minimax", "moonshot",
|
|
|
|
|
|
|
|
"tongyi",
|
|
|
|
|
|
|
|
"wenxin", "yi", "zhipuai"
|
|
|
|
|
|
|
|
]
|
|
|
|
model_provider_dir_paths = [
|
|
|
|
model_provider_dir_paths = [
|
|
|
|
os.path.join(model_providers_path, model_provider_dir)
|
|
|
|
os.path.join(model_providers_path, model_provider_dir)
|
|
|
|
for model_provider_dir in os.listdir(model_providers_path)
|
|
|
|
for model_provider_dir in os.listdir(model_providers_path)
|
|
|
|
if not model_provider_dir.startswith('__')
|
|
|
|
if model_provider_dir in whitelist
|
|
|
|
and os.path.isdir(os.path.join(model_providers_path, model_provider_dir))
|
|
|
|
and os.path.isdir(os.path.join(model_providers_path, model_provider_dir))
|
|
|
|
]
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
|
|
|