Update main.py

09514751 · Michael Poluektov · GitHub · afd74213 · 09514751
Unverified Commit 09514751 authored Jul 02, 2024 by Michael Poluektov Committed by GitHub Jul 02, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 5 deletions

backend/main.py backend/main.py +3 -5

No files found.
--- 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:
+        for key in ["title", "task", "function"]:
-            del payload["title"]
+            if key in payload:
+                del payload[key]
-        if "task" in payload:
-            del payload["task"]
    return payload