Compare commits
3 Commits
Author | SHA1 | Date | |
---|---|---|---|
![]() |
96008f1f3d | ||
![]() |
c106a896a5 | ||
![]() |
87a4776272 |
@ -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))
|
||||||
]
|
]
|
||||||
|
|
||||||
|
@ -17,4 +17,7 @@ class BuiltinToolProviderSort:
|
|||||||
|
|
||||||
sorted_providers = sort_by_position_map(cls._position, providers, name_func)
|
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