feat: 白名单中去除 maas 平台

feat: optimize whitelist for model providers
feat(model/tools): filter unregistered tools and models
2026-01-19 11:45:05 +08:00 · 2024-06-27 17:49:43 +08:00 · 2024-06-21 10:56:20 +08:00 · 2024-06-19 17:34:11 +08:00
2 changed files with 10 additions and 2 deletions
--- a/api/core/model_runtime/model_providers/model_provider_factory.py
+++ b/api/core/model_runtime/model_providers/model_provider_factory.py
@ -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))
        ]

--- a/api/core/tools/provider/builtin/_positions.py
+++ b/api/core/tools/provider/builtin/_positions.py
@ -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
+
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