Update delete_workflow (#459)

modelscope · Nov 28, 2024 · 7d285e8 · 7d285e8
1 parent 6178c29
commit 7d285e8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/agentscope/studio/_app.py b/src/agentscope/studio/_app.py
@@ -764,6 +764,11 @@ def _delete_workflow() -> Response:
     if not filename:
         return jsonify({"error": "Filename is required"})
 
+    if not filename.endswith(".json"):
+        return jsonify({"error": "Only JSON files can be deleted"})
+
+    filename = os.path.basename(filename)
+
     filepath = os.path.join(user_dir, filename)
     if not os.path.exists(filepath):
         return jsonify({"error": "File not found"})