Bläddra i källkod

keep title, task, function tags for pipelines

Michael Poluektov 10 månader sedan
förälder
incheckning
02f242e9e8
1 ändrade filer med 4 tillägg och 1 borttagningar
  1. 4 1
      backend/main.py

+ 4 - 1
backend/main.py

@@ -833,7 +833,10 @@ def filter_pipeline(payload, user):
             else:
             else:
                 pass
                 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"]:
         for key in ["title", "task", "function"]:
             if key in payload:
             if key in payload:
                 del payload[key]
                 del payload[key]