From 02f242e9e81648ef2f4d6bf529e0821a87b4afb8 Mon Sep 17 00:00:00 2001 From: Michael Poluektov Date: Tue, 2 Jul 2024 11:52:46 +0100 Subject: [PATCH] keep title, task, function tags for pipelines --- backend/main.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/backend/main.py b/backend/main.py index 91b2495f6..b2fe38245 100644 --- 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]