From c6856632427c083070feb78240508fabb501c8b2 Mon Sep 17 00:00:00 2001
From: Yunlu Wen <wylswz@163.com>
Date: Thu, 7 Aug 2025 10:49:06 +0800
Subject: [PATCH] feat: add knowledge pipeline creation feature

---
 .../console/datasets/rag_pipeline/rag_pipeline.py      |  2 ++
 api/controllers/console/wraps.py                       | 10 ++++++++++
 api/services/feature_service.py                        |  6 ++++++
 3 files changed, 18 insertions(+)

diff --git a/api/controllers/console/datasets/rag_pipeline/rag_pipeline.py b/api/controllers/console/datasets/rag_pipeline/rag_pipeline.py
index 93976bd6f5..f496149deb 100644
--- a/api/controllers/console/datasets/rag_pipeline/rag_pipeline.py
+++ b/api/controllers/console/datasets/rag_pipeline/rag_pipeline.py
@@ -9,6 +9,7 @@ from controllers.console.wraps import (
     account_initialization_required,
     enterprise_license_required,
     setup_required,
+    knowledge_pipeline_publish_enabled,
 )
 from extensions.ext_database import db
 from libs.login import login_required
@@ -116,6 +117,7 @@ class PublishCustomizedPipelineTemplateApi(Resource):
     @login_required
     @account_initialization_required
     @enterprise_license_required
+    @knowledge_pipeline_publish_enabled
     def post(self, pipeline_id: str):
         parser = reqparse.RequestParser()
         parser.add_argument(
diff --git a/api/controllers/console/wraps.py b/api/controllers/console/wraps.py
index d862dac373..2df41ba840 100644
--- a/api/controllers/console/wraps.py
+++ b/api/controllers/console/wraps.py
@@ -261,3 +261,13 @@ def is_allow_transfer_owner(view):
         abort(403)
 
     return decorated
+
+
+def knowledge_pipeline_publish_enabled(view):
+    @wraps(view)
+    def decorated(*args, **kwargs):
+        features = FeatureService.get_features(current_user.current_tenant_id)
+        if features.knowledge_pipeline.publish_enabled:
+            return view(*args, **kwargs)
+        abort(403)
+    return decorated
diff --git a/api/services/feature_service.py b/api/services/feature_service.py
index 1441e6ce16..43c8a5fec4 100644
--- a/api/services/feature_service.py
+++ b/api/services/feature_service.py
@@ -87,6 +87,8 @@ class WebAppAuthModel(BaseModel):
     allow_email_code_login: bool = False
     allow_email_password_login: bool = False
 
+class KnowledgePipeline(BaseModel):
+    publish_enabled: bool = False
 
 class PluginInstallationScope(StrEnum):
     NONE = "none"
@@ -126,6 +128,7 @@ class FeatureModel(BaseModel):
     is_allow_transfer_workspace: bool = True
     # pydantic configs
     model_config = ConfigDict(protected_namespaces=())
+    knowledge_pipeline: KnowledgePipeline = KnowledgePipeline()
 
 
 class KnowledgeRateLimitModel(BaseModel):
@@ -265,6 +268,9 @@ class FeatureService:
         if "knowledge_rate_limit" in billing_info:
             features.knowledge_rate_limit = billing_info["knowledge_rate_limit"]["limit"]
 
+        if "knowledge_pipeline_publish_enabled" in billing_info:
+            features.knowledge_pipeline.publish_enabled = billing_info["knowledge_pipeline_publish_enabled"]
+
     @classmethod
     def _fulfill_params_from_enterprise(cls, features: SystemFeatureModel):
         enterprise_info = EnterpriseService.get_info()