feat: support define tags in tool yaml (#4763)

2024-05-29 15:19:14 +08:00
parent b189faca52
commit 74f38eacda
82 changed files with 187 additions and 255 deletions
--- a/api/core/tools/provider/builtin/feishu/feishu.py
+++ b/api/core/tools/provider/builtin/feishu/feishu.py
@@ -1,4 +1,3 @@
-from core.tools.entities.values import ToolLabelEnum
 from core.tools.provider.builtin.feishu.tools.feishu_group_bot import FeishuGroupBotTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController

@@ -6,9 +5,3 @@ from core.tools.provider.builtin_tool_provider import BuiltinToolProviderControl
 class FeishuProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict) -> None:
        FeishuGroupBotTool()
-        pass
-
-    def _get_tool_labels(self) -> list[ToolLabelEnum]:
-        return [
-            ToolLabelEnum.SOCIAL
-        ]
--- a/api/core/tools/provider/builtin/feishu/feishu.yaml
+++ b/api/core/tools/provider/builtin/feishu/feishu.yaml
@@ -10,4 +10,7 @@ identity:
    zh_Hans: 飞书群机器人
    pt_BR: Feishu group bot
  icon: icon.svg
+  tags:
+    - social
+    - productivity
 credentials_for_provider: