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/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
+        ]