Compare commits

...

3 Commits

Author SHA1 Message Date
开坦克的贝塔
96008f1f3d feat: 白名单中去除 maas 平台 2024-06-27 17:49:43 +08:00
开坦克的贝塔
c106a896a5 feat: optimize whitelist for model providers 2024-06-21 10:56:20 +08:00
开坦克的贝塔
87a4776272 feat(model/tools): filter unregistered tools and models 2024-06-19 17:34:11 +08:00
2 changed files with 10 additions and 2 deletions

View File

@ -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",
"tongyi",
"wenxin", "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))
]

View File

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