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]