Feat/workflow phase2 (#4687)

2024-05-27 22:01:11 +08:00
parent 45deaee762
commit e852a21634
139 changed files with 5997 additions and 779 deletions
--- a/api/core/tools/provider/builtin/_positions.py
+++ b/api/core/tools/provider/builtin/_positions.py
@@ -1,6 +1,6 @@
 import os.path

-from core.tools.entities.user_entities import UserToolProvider
+from core.tools.entities.api_entities import UserToolProvider
 from core.utils.position_helper import get_position_map, sort_by_position_map


--- a/api/core/tools/provider/builtin/aippt/aippt.py
+++ b/api/core/tools/provider/builtin/aippt/aippt.py
@@ -1,3 +1,4 @@
+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.aippt.tools.aippt import AIPPTGenerateTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -9,3 +10,9 @@ class AIPPTProvider(BuiltinToolProviderController):
            AIPPTGenerateTool._get_api_token(credentials, user_id='__dify_system__')
        except Exception as e:
            raise ToolProviderCredentialValidationError(str(e))
+
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.PRODUCTIVITY, 
+            ToolLabelEnum.DESIGN,
+        ]
--- a/api/core/tools/provider/builtin/arxiv/arxiv.py
+++ b/api/core/tools/provider/builtin/arxiv/arxiv.py
@@ -1,3 +1,4 @@
+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.arxiv.tools.arxiv_search import ArxivSearchTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -7,7 +8,7 @@ class ArxivProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict) -> None:
        try:
            ArxivSearchTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -17,4 +18,9 @@ class ArxivProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.SEARCH,
+        ]
--- a/api/core/tools/provider/builtin/azuredalle/azuredalle.py
+++ b/api/core/tools/provider/builtin/azuredalle/azuredalle.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.azuredalle.tools.dalle3 import DallE3Tool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -9,7 +10,7 @@ class AzureDALLEProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict[str, Any]) -> None:
        try:
            DallE3Tool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -22,3 +23,8 @@ class AzureDALLEProvider(BuiltinToolProviderController):
            )
        except Exception as e:
            raise ToolProviderCredentialValidationError(str(e))
+
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.IMAGE
+        ]
--- a/api/core/tools/provider/builtin/bing/bing.py
+++ b/api/core/tools/provider/builtin/bing/bing.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.bing.tools.bing_web_search import BingSearchTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -9,7 +10,7 @@ class BingProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict[str, Any]) -> None:
        try:
            BingSearchTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).validate_credentials(
@@ -21,3 +22,8 @@ class BingProvider(BuiltinToolProviderController):
            )
        except Exception as e:
            raise ToolProviderCredentialValidationError(str(e))
+
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.SEARCH
+        ]
--- a/api/core/tools/provider/builtin/brave/brave.py
+++ b/api/core/tools/provider/builtin/brave/brave.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.brave.tools.brave_search import BraveSearchTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -9,7 +10,7 @@ class BraveProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict[str, Any]) -> None:
        try:
            BraveSearchTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -19,4 +20,9 @@ class BraveProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.SEARCH,
+        ]
--- a/api/core/tools/provider/builtin/chart/chart.py
+++ b/api/core/tools/provider/builtin/chart/chart.py
@@ -2,6 +2,7 @@ import matplotlib.pyplot as plt
 from fontTools.ttLib import TTFont
 from matplotlib.font_manager import findSystemFonts

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.chart.tools.line import LinearChartTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -44,7 +45,7 @@ class ChartProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict) -> None:
        try:
            LinearChartTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -54,4 +55,9 @@ class ChartProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.DESIGN, ToolLabelEnum.PRODUCTIVITY, ToolLabelEnum.UTILITIES
+        ]
--- a/api/core/tools/provider/builtin/code/code.py
+++ b/api/core/tools/provider/builtin/code/code.py
@@ -1,8 +1,14 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController


 class CodeToolProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict[str, Any]) -> None:
-        pass
+        pass
+
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.PRODUCTIVITY
+        ]
--- a/api/core/tools/provider/builtin/dalle/dalle.py
+++ b/api/core/tools/provider/builtin/dalle/dalle.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.dalle.tools.dalle2 import DallE2Tool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -9,7 +10,7 @@ class DALLEProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict[str, Any]) -> None:
        try:
            DallE2Tool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -21,4 +22,9 @@ class DALLEProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.IMAGE, ToolLabelEnum.PRODUCTIVITY
+        ]
--- a/api/core/tools/provider/builtin/devdocs/devdocs.py
+++ b/api/core/tools/provider/builtin/devdocs/devdocs.py
@@ -1,3 +1,4 @@
+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.devdocs.tools.searchDevDocs import SearchDevDocsTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -7,7 +8,7 @@ class DevDocsProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict) -> None:
        try:
            SearchDevDocsTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -18,4 +19,9 @@ class DevDocsProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+    
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.SEARCH, ToolLabelEnum.PRODUCTIVITY
+        ]
--- a/api/core/tools/provider/builtin/dingtalk/dingtalk.py
+++ b/api/core/tools/provider/builtin/dingtalk/dingtalk.py
@@ -1,3 +1,4 @@
+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

@@ -6,3 +7,8 @@ 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/duckduckgo/duckduckgo.py
+++ b/api/core/tools/provider/builtin/duckduckgo/duckduckgo.py
@@ -1,3 +1,4 @@
+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.duckduckgo.tools.duckduckgo_search import DuckDuckGoSearchTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -7,7 +8,7 @@ class DuckDuckGoProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict) -> None:
        try:
            DuckDuckGoSearchTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -17,4 +18,9 @@ class DuckDuckGoProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.SEARCH
+        ]
--- a/api/core/tools/provider/builtin/feishu/feishu.py
+++ b/api/core/tools/provider/builtin/feishu/feishu.py
@@ -1,3 +1,4 @@
+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,3 +7,8 @@ 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/firecrawl/firecrawl.py
+++ b/api/core/tools/provider/builtin/firecrawl/firecrawl.py
@@ -1,3 +1,4 @@
+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.firecrawl.tools.crawl import CrawlTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -8,7 +9,7 @@ class FirecrawlProvider(BuiltinToolProviderController):
        try:
            # Example validation using the Crawl tool
            CrawlTool().fork_tool_runtime(
-                meta={"credentials": credentials}
+                runtime={"credentials": credentials}
            ).invoke(
                user_id='',
                tool_parameters={
@@ -20,4 +21,9 @@ class FirecrawlProvider(BuiltinToolProviderController):
                }
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.SEARCH, ToolLabelEnum.UTILITIES
+        ]
--- a/api/core/tools/provider/builtin/gaode/gaode.py
+++ b/api/core/tools/provider/builtin/gaode/gaode.py
@@ -2,6 +2,7 @@ import urllib.parse

 import requests

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController

@@ -24,3 +25,9 @@ class GaodeProvider(BuiltinToolProviderController):
                raise ToolProviderCredentialValidationError("Gaode API Key is invalid. {}".format(e))
        except Exception as e:
            raise ToolProviderCredentialValidationError(str(e))
+
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.UTILITIES, ToolLabelEnum.PRODUCTIVITY, 
+            ToolLabelEnum.WEATHER, ToolLabelEnum.TRAVEL
+        ]
--- a/api/core/tools/provider/builtin/github/github.py
+++ b/api/core/tools/provider/builtin/github/github.py
@@ -1,5 +1,6 @@
 import requests

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController

@@ -30,3 +31,8 @@ class GihubProvider(BuiltinToolProviderController):
                raise ToolProviderCredentialValidationError("Github API Key and Api Version is invalid. {}".format(e))
        except Exception as e:
            raise ToolProviderCredentialValidationError(str(e))
+
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.UTILITIES
+        ]
--- a/api/core/tools/provider/builtin/google/google.py
+++ b/api/core/tools/provider/builtin/google/google.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.google.tools.google_search import GoogleSearchTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -9,7 +10,7 @@ class GoogleProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict[str, Any]) -> None:
        try:
            GoogleSearchTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -20,4 +21,9 @@ class GoogleProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+    
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.SEARCH
+        ]
--- a/api/core/tools/provider/builtin/jina/jina.py
+++ b/api/core/tools/provider/builtin/jina/jina.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController

@@ -9,4 +10,9 @@ class GoogleProvider(BuiltinToolProviderController):
        try:
            pass
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.SEARCH, ToolLabelEnum.PRODUCTIVITY
+        ]
--- a/api/core/tools/provider/builtin/judge0ce/judge0ce.py
+++ b/api/core/tools/provider/builtin/judge0ce/judge0ce.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.judge0ce.tools.executeCode import ExecuteCodeTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -9,7 +10,7 @@ class Judge0CEProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict[str, Any]) -> None:
        try:
            ExecuteCodeTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -20,4 +21,9 @@ class Judge0CEProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.OTHER, ToolLabelEnum.UTILITIES
+        ]
--- a/api/core/tools/provider/builtin/maths/maths.py
+++ b/api/core/tools/provider/builtin/maths/maths.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.maths.tools.eval_expression import EvaluateExpressionTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -16,3 +17,8 @@ class MathsProvider(BuiltinToolProviderController):
            )
        except Exception as e:
            raise ToolProviderCredentialValidationError(str(e))
+
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.UTILITIES, ToolLabelEnum.PRODUCTIVITY
+        ]
--- a/api/core/tools/provider/builtin/openweather/openweather.py
+++ b/api/core/tools/provider/builtin/openweather/openweather.py
@@ -1,5 +1,6 @@
 import requests

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController

@@ -34,3 +35,8 @@ class OpenweatherProvider(BuiltinToolProviderController):
                )
        except Exception as e:
            raise ToolProviderCredentialValidationError(str(e))
+
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.WEATHER
+        ]
--- a/api/core/tools/provider/builtin/pubmed/pubmed.py
+++ b/api/core/tools/provider/builtin/pubmed/pubmed.py
@@ -1,3 +1,4 @@
+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.pubmed.tools.pubmed_search import PubMedSearchTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -7,7 +8,7 @@ class PubMedProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict) -> None:
        try:
            PubMedSearchTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -17,4 +18,9 @@ class PubMedProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.MEDICAL, ToolLabelEnum.SEARCH
+        ]
--- a/api/core/tools/provider/builtin/qrcode/qrcode.py
+++ b/api/core/tools/provider/builtin/qrcode/qrcode.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.qrcode.tools.qrcode_generator import QRCodeGeneratorTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -14,3 +15,8 @@ class QRCodeProvider(BuiltinToolProviderController):
                                        })
        except Exception as e:
            raise ToolProviderCredentialValidationError(str(e))
+
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.UTILITIES
+        ]
--- a/api/core/tools/provider/builtin/searxng/searxng.py
+++ b/api/core/tools/provider/builtin/searxng/searxng.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.searxng.tools.searxng_search import SearXNGSearchTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -9,7 +10,7 @@ class SearXNGProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict[str, Any]) -> None:
        try:
            SearXNGSearchTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -22,4 +23,9 @@ class SearXNGProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.SEARCH, ToolLabelEnum.PRODUCTIVITY
+        ]
--- a/api/core/tools/provider/builtin/slack/slack.py
+++ b/api/core/tools/provider/builtin/slack/slack.py
@@ -1,3 +1,4 @@
+from core.tools.entities.values import ToolLabelEnum
 from core.tools.provider.builtin.slack.tools.slack_webhook import SlackWebhookTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController

@@ -6,3 +7,8 @@ class SlackProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict) -> None:
        SlackWebhookTool()
        pass
+
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.SOCIAL
+        ]
--- a/api/core/tools/provider/builtin/spark/spark.py
+++ b/api/core/tools/provider/builtin/spark/spark.py
@@ -1,5 +1,6 @@
 import json

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.spark.tools.spark_img_generation import spark_response
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -38,3 +39,8 @@ class SparkProvider(BuiltinToolProviderController):
                )
        except Exception as e:
            raise ToolProviderCredentialValidationError(str(e))
+
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.IMAGE
+        ]
--- a/api/core/tools/provider/builtin/stability/stability.py
+++ b/api/core/tools/provider/builtin/stability/stability.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.provider.builtin.stability.tools.base import BaseStabilityAuthorization
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController

@@ -12,4 +13,9 @@ class StabilityToolProvider(BuiltinToolProviderController, BaseStabilityAuthoriz
        """
        This method is responsible for validating the credentials.
        """
-        self.sd_validate_credentials(credentials)
+        self.sd_validate_credentials(credentials)
+
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.IMAGE
+        ]
--- a/api/core/tools/provider/builtin/stablediffusion/stablediffusion.py
+++ b/api/core/tools/provider/builtin/stablediffusion/stablediffusion.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.stablediffusion.tools.stable_diffusion import StableDiffusionTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -9,9 +10,14 @@ class StableDiffusionProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict[str, Any]) -> None:
        try:
            StableDiffusionTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).validate_models()
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+    
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.IMAGE
+        ]
--- a/api/core/tools/provider/builtin/stackexchange/stackexchange.py
+++ b/api/core/tools/provider/builtin/stackexchange/stackexchange.py
@@ -1,3 +1,4 @@
+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.stackexchange.tools.searchStackExQuestions import SearchStackExQuestionsTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -7,7 +8,7 @@ class StackExchangeProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict) -> None:
        try:
            SearchStackExQuestionsTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -22,4 +23,9 @@ class StackExchangeProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.SEARCH, ToolLabelEnum.UTILITIES
+        ]
--- a/api/core/tools/provider/builtin/tavily/tavily.py
+++ b/api/core/tools/provider/builtin/tavily/tavily.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.tavily.tools.tavily_search import TavilySearchTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -9,7 +10,7 @@ class TavilyProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict[str, Any]) -> None:
        try:
            TavilySearchTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -26,4 +27,9 @@ class TavilyProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.SEARCH
+        ]
--- a/api/core/tools/provider/builtin/time/time.py
+++ b/api/core/tools/provider/builtin/time/time.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.time.tools.current_time import CurrentTimeTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -13,4 +14,9 @@ class WikiPediaProvider(BuiltinToolProviderController):
                tool_parameters={},
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.UTILITIES
+        ]
--- a/api/core/tools/provider/builtin/trello/trello.py
+++ b/api/core/tools/provider/builtin/trello/trello.py
@@ -2,6 +2,7 @@ from typing import Any

 import requests

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController

@@ -31,4 +32,9 @@ class TrelloProvider(BuiltinToolProviderController):
            raise ToolProviderCredentialValidationError("Error validating Trello credentials")
        except requests.exceptions.RequestException as e:
            # Handle other exceptions, such as connection errors
-            raise ToolProviderCredentialValidationError("Error validating Trello credentials")
+            raise ToolProviderCredentialValidationError("Error validating Trello credentials")
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.PRODUCTIVITY
+        ]
--- a/api/core/tools/provider/builtin/twilio/twilio.py
+++ b/api/core/tools/provider/builtin/twilio/twilio.py
@@ -3,6 +3,7 @@ from typing import Any
 from twilio.base.exceptions import TwilioRestException
 from twilio.rest import Client

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController

@@ -26,4 +27,9 @@ class TwilioProvider(BuiltinToolProviderController):
        except KeyError as e:
            raise ToolProviderCredentialValidationError(f"Missing required credential: {e}") from e
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.SOCIAL
+        ]
--- a/api/core/tools/provider/builtin/vectorizer/vectorizer.py
+++ b/api/core/tools/provider/builtin/vectorizer/vectorizer.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.vectorizer.tools.vectorizer import VectorizerTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -9,7 +10,7 @@ class VectorizerProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict[str, Any]) -> None:
        try:
            VectorizerTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -20,4 +21,9 @@ class VectorizerProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.PRODUCTIVITY, ToolLabelEnum.IMAGE
+        ]
--- a/api/core/tools/provider/builtin/webscraper/webscraper.py
+++ b/api/core/tools/provider/builtin/webscraper/webscraper.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.webscraper.tools.webscraper import WebscraperTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -9,7 +10,7 @@ class WebscraperProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict[str, Any]) -> None:
        try:
            WebscraperTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -20,4 +21,9 @@ class WebscraperProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.PRODUCTIVITY
+        ]
--- a/api/core/tools/provider/builtin/wecom/wecom.py
+++ b/api/core/tools/provider/builtin/wecom/wecom.py
@@ -1,3 +1,4 @@
+from core.tools.entities.values import ToolLabelEnum
 from core.tools.provider.builtin.wecom.tools.wecom_group_bot import WecomGroupBotTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController

@@ -6,3 +7,8 @@ class WecomProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict) -> None:
        WecomGroupBotTool()
        pass
+
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.SOCIAL
+        ]
--- a/api/core/tools/provider/builtin/wikipedia/wikipedia.py
+++ b/api/core/tools/provider/builtin/wikipedia/wikipedia.py
@@ -1,3 +1,4 @@
+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.wikipedia.tools.wikipedia_search import WikiPediaSearchTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -7,7 +8,7 @@ class WikiPediaProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict) -> None:
        try:
            WikiPediaSearchTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -17,4 +18,9 @@ class WikiPediaProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.SEARCH
+        ]
--- a/api/core/tools/provider/builtin/wolframalpha/wolframalpha.py
+++ b/api/core/tools/provider/builtin/wolframalpha/wolframalpha.py
@@ -1,5 +1,6 @@
 from typing import Any

+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.wolframalpha.tools.wolframalpha import WolframAlphaTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -9,7 +10,7 @@ class GoogleProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict[str, Any]) -> None:
        try:
            WolframAlphaTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -19,4 +20,9 @@ class GoogleProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+    
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.PRODUCTIVITY, ToolLabelEnum.UTILITIES
+        ]
--- a/api/core/tools/provider/builtin/yahoo/yahoo.py
+++ b/api/core/tools/provider/builtin/yahoo/yahoo.py
@@ -1,3 +1,4 @@
+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.yahoo.tools.ticker import YahooFinanceSearchTickerTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -7,7 +8,7 @@ class YahooFinanceProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict) -> None:
        try:
            YahooFinanceSearchTickerTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -17,4 +18,9 @@ class YahooFinanceProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+    
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.BUSINESS, ToolLabelEnum.FINANCE
+        ]
--- a/api/core/tools/provider/builtin/youtube/youtube.py
+++ b/api/core/tools/provider/builtin/youtube/youtube.py
@@ -1,3 +1,4 @@
+from core.tools.entities.values import ToolLabelEnum
 from core.tools.errors import ToolProviderCredentialValidationError
 from core.tools.provider.builtin.youtube.tools.videos import YoutubeVideosAnalyticsTool
 from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -7,7 +8,7 @@ class YahooFinanceProvider(BuiltinToolProviderController):
    def _validate_credentials(self, credentials: dict) -> None:
        try:
            YoutubeVideosAnalyticsTool().fork_tool_runtime(
-                meta={
+                runtime={
                    "credentials": credentials,
                }
            ).invoke(
@@ -19,4 +20,9 @@ class YahooFinanceProvider(BuiltinToolProviderController):
                },
            )
        except Exception as e:
-            raise ToolProviderCredentialValidationError(str(e))
+            raise ToolProviderCredentialValidationError(str(e))
+        
+    def _get_tool_labels(self) -> list[ToolLabelEnum]:
+        return [
+            ToolLabelEnum.VIDEOS
+        ]