feat(model/tools): filter unregistered tools and models

feat/2p
开坦克的贝塔 2 years ago
parent 7d5ebbb611
commit 87a4776272

@ -201,10 +201,15 @@ class ModelProviderFactory:
model_providers_path = os.path.dirname(current_path)
# get all folders path under model_providers_path that do not start with __
whitelist = [
"baichuan", "chatglm", "deepseek", "hunyuan", "minimax", "moonshot",
"siliconflow", "tongyi", "volcengine_maas",
"wenxin", "xinference", "yi", "zhipuai"
]
model_provider_dir_paths = [
os.path.join(model_providers_path, model_provider_dir)
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))
]

@ -17,4 +17,7 @@ class BuiltinToolProviderSort:
sorted_providers = sort_by_position_map(cls._position, providers, name_func)
return sorted_providers
blacklist = ['duckduckgo', 'brave', 'dalle', 'github', 'google', 'jina', 'slack', 'stablediffusion', 'youtube']
filtered_providers = [provider for provider in sorted_providers if provider.name not in blacklist]
return filtered_providers

Loading…
Cancel
Save