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

Co-authored-by: zhaobs <[email protected]>
langgenius · Dec 9, 2024 · 5c166b3 · 5c166b3
1 parent 230fa32
commit 5c166b3
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