Compare commits
3 Commits
Author | SHA1 | Date | |
---|---|---|---|
![]() |
96008f1f3d | ||
![]() |
c106a896a5 | ||
![]() |
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",
|
||||
"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))
|
||||
]
|
||||
|
||||
|
@ -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…
Reference in New Issue
Block a user