소스 검색

keep title, task, function tags for pipelines

Michael Poluektov 10 달 전
부모
커밋
02f242e9e8
1개의 변경된 파일4개의 추가작업 그리고 1개의 파일을 삭제
  1. 4 1
      backend/main.py

+ 4 - 1
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]