keep title, task, function tags for pipelines

02f242e9 · Michael Poluektov · 09514751 · 02f242e9
Commit 02f242e9 authored Jul 02, 2024 by Michael Poluektov
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

backend/main.py backend/main.py +4 -1

No files found.
--- a/backend/main.py
+++ b/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]