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/dingtalk/dingtalk.py
+++ b/api/core/tools/provider/builtin/dingtalk/dingtalk.py
@@ -1,4 +1,3 @@
-from core.tools.entities.values import ToolLabelEnum
 from core.tools.provider.builtin.dingtalk.tools.dingtalk_group_bot import DingTalkGroupBotTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController

@@ -7,8 +6,3 @@ class DingTalkProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict) -> None:
        DingTalkGroupBotTool()
        pass
-
-    def _get_tool_labels(self) -> list[ToolLabelEnum]:
-        return [
-            ToolLabelEnum.SOCIAL
-        ]
--- a/api/core/tools/provider/builtin/dingtalk/dingtalk.yaml
+++ b/api/core/tools/provider/builtin/dingtalk/dingtalk.yaml
@@ -10,4 +10,7 @@ identity:
    zh_Hans: 钉钉群机器人
    pt_BR: DingTalk group robot
  icon: icon.svg
+  tags:
+    - social
+    - productivity
 credentials_for_provider: