fix: tags could not be saved when the Workflow Tool was created (#11481)

Co-authored-by: zhaobs <zhaobs@cailian.net>
langgenius · Dec 16, 2024 · 83a5dcf · 83a5dcf
1 parent e11a876
commit 83a5dcf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/api/controllers/console/workspace/tool_providers.py b/api/controllers/console/workspace/tool_providers.py
@@ -368,6 +368,7 @@ def post(self):
             description=args["description"],
             parameters=args["parameters"],
             privacy_policy=args["privacy_policy"],
+            labels=args["labels"],
         )
 
 

diff --git a/api/services/tools/workflow_tools_manage_service.py b/api/services/tools/workflow_tools_manage_service.py
@@ -81,6 +81,10 @@ def create_workflow_tool(
         db.session.add(workflow_tool_provider)
         db.session.commit()
 
+        if labels is not None:
+            ToolLabelManager.update_tool_labels(
+                ToolTransformService.workflow_provider_to_controller(workflow_tool_provider), labels
+            )
         return {"result": "success"}
 
     @classmethod