From 09514751b553fd7713970af44033acf8f6e9f75a Mon Sep 17 00:00:00 2001
From: Michael Poluektov <78477503+michaelpoluektov@users.noreply.github.com>
Date: Tue, 2 Jul 2024 10:57:56 +0100
Subject: [PATCH] Update main.py

---
 backend/main.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/backend/main.py b/backend/main.py
index 70aa52bba3..91b2495f6e 100644
--- a/backend/main.py
+++ b/backend/main.py
@@ -834,11 +834,9 @@ def filter_pipeline(payload, user):
                 pass
 
     if "pipeline" not in app.state.MODELS[model_id]:
-        if "title" in payload:
-            del payload["title"]
-
-        if "task" in payload:
-            del payload["task"]
+        for key in ["title", "task", "function"]:
+            if key in payload:
+                del payload[key]
 
     return payload