浏览代码

keep title, task, function tags for pipelines

Michael Poluektov 10 月之前
父节点
当前提交
02f242e9e8
共有 1 个文件被更改,包括 4 次插入1 次删除
  1. 4 1
      backend/main.py

+ 4 - 1
backend/main.py

@@ -833,7 +833,10 @@ def filter_pipeline(payload, user):
             else:
                 pass
 
-    if "pipeline" not in app.state.MODELS[model_id]:
+    keep_extras = (
+        "pipeline" in app.state.MODELS[model_id] or "pipe" in app.state.MODELS[model_id]
+    )
+    if not keep_extras:
         for key in ["title", "task", "function"]:
             if key in payload:
                 del payload[key]