feat: Knowledge base workflow release, version management function (#4405)

* feat: Knowledge base workflow release, version management function

* feat: Knowledge base workflow release, version management function

Author: shaohuzhang1

apps/knowledge/api/knowledge_version.py

@@ -0,0 +1,96 @@
+# coding=utf-8
+"""
+    @project: MaxKB
+    @Author：虎虎
+    @file： application_version.py
+    @date：2025/6/4 17:33
+    @desc:
+"""
+from drf_spectacular.types import OpenApiTypes
+from drf_spectacular.utils import OpenApiParameter
+
+from common.mixins.api_mixin import APIMixin
+from common.result import ResultSerializer, ResultPageSerializer
+from knowledge.serializers.knowledge_version import KnowledgeVersionModelSerializer
+
+
+class KnowledgeListVersionResult(ResultSerializer):
+    def get_data(self):
+        return KnowledgeVersionModelSerializer(many=True)
+
+
+class KnowledgePageVersionResult(ResultPageSerializer):
+    def get_data(self):
+        return KnowledgeVersionModelSerializer(many=True)
+
+
+class KnowledgeWorkflowVersionResult(ResultSerializer):
+    def get_data(self):
+        return KnowledgeVersionModelSerializer()
+
+
+class KnowledgeVersionAPI(APIMixin):
+    @staticmethod
+    def get_parameters():
+        return [
+            OpenApiParameter(
+                name="workspace_id",
+                description="工作空间id",
+                type=OpenApiTypes.STR,
+                location='path',
+                required=True,
+            ),
+            OpenApiParameter(
+                name="knowledge_id",
+                description="knowledge ID",
+                type=OpenApiTypes.STR,
+                location='path',
+                required=True,
+            )
+        ]
+
+
+class KnowledgeVersionOperateAPI(APIMixin):
+    @staticmethod
+    def get_parameters():
+        return [
+            OpenApiParameter(
+                name="knowledge_version_id",
+                description="工作流版本id",
+                type=OpenApiTypes.STR,
+                location='path',
+                required=True,
+            )
+            , *KnowledgeVersionAPI.get_parameters()
+        ]
+
+    @staticmethod
+    def get_response():
+        return KnowledgeWorkflowVersionResult
+
+
+class KnowledgeVersionListAPI(APIMixin):
+    @staticmethod
+    def get_parameters():
+        return [
+            OpenApiParameter(
+                name="name",
+                description="Version Name",
+                type=OpenApiTypes.STR,
+                required=False,
+            )
+            , *KnowledgeVersionOperateAPI.get_parameters()]
+
+    @staticmethod
+    def get_response():
+        return KnowledgeListVersionResult
+
+
+class KnowledgeVersionPageAPI(APIMixin):
+    @staticmethod
+    def get_parameters():
+        return KnowledgeVersionListAPI.get_parameters()
+
+    @staticmethod
+    def get_response():
+        return KnowledgePageVersionResult

apps/knowledge/api/knowledge_workflow.py

@@ -1,6 +1,9 @@
 # coding=utf-8
+from drf_spectacular.types import OpenApiTypes
+from drf_spectacular.utils import OpenApiParameter
 
 from common.mixins.api_mixin import APIMixin
+from knowledge.serializers.knowledge_workflow import KnowledgeWorkflowActionRequestSerializer
 
 
 class KnowledgeWorkflowApi(APIMixin):
@@ -9,3 +12,55 @@ class KnowledgeWorkflowApi(APIMixin):
 
 class KnowledgeWorkflowVersionApi(APIMixin):
     pass
+
+
+class KnowledgeWorkflowActionApi(APIMixin):
+    @staticmethod
+    def get_request():
+        return KnowledgeWorkflowActionRequestSerializer
+
+    @staticmethod
+    def get_parameters():
+        return [
+            OpenApiParameter(
+                name="workspace_id",
+                description="工作空间id",
+                type=OpenApiTypes.STR,
+                location='path',
+                required=True,
+            ),
+            OpenApiParameter(
+                name="knowledge_id",
+                description="知识库id",
+                type=OpenApiTypes.STR,
+                location='path',
+                required=True,
+            )
+        ]
+
+    class Operate(APIMixin):
+        @staticmethod
+        def get_parameters():
+            return [
+                OpenApiParameter(
+                    name="workspace_id",
+                    description="工作空间id",
+                    type=OpenApiTypes.STR,
+                    location='path',
+                    required=True,
+                ),
+                OpenApiParameter(
+                    name="knowledge_id",
+                    description="知识库id",
+                    type=OpenApiTypes.STR,
+                    location='path',
+                    required=True,
+                ),
+                OpenApiParameter(
+                    name="knowledge_action_id",
+                    description="知识库执行id",
+                    type=OpenApiTypes.STR,
+                    location='path',
+                    required=True,
+                )
+            ]

apps/knowledge/migrations/0007_remove_knowledgeworkflowversion_workflow_and_more.py

@@ -0,0 +1,22 @@
+# Generated by Django 5.2.8 on 2025-12-01 06:28
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('knowledge', '0006_paragraph_chunks'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='knowledgeworkflowversion',
+            name='workflow',
+        ),
+        migrations.AddField(
+            model_name='knowledgeworkflowversion',
+            name='name',
+            field=models.CharField(default='', max_length=128, verbose_name='版本名称'),
+        ),
+    ]

apps/knowledge/models/knowledge.py

@@ -159,9 +159,8 @@ class KnowledgeWorkflowVersion(AppModelMixin):
     """
     id = models.UUIDField(primary_key=True, max_length=128, default=uuid.uuid7, editable=False, verbose_name="主键id")
     knowledge = models.ForeignKey(Knowledge, on_delete=models.CASCADE, verbose_name="知识库", db_constraint=False)
-    workflow = models.ForeignKey(KnowledgeWorkflow, on_delete=models.CASCADE, verbose_name="工作流",
-                                 db_constraint=False, related_name='versions')
     workspace_id = models.CharField(max_length=64, verbose_name="工作空间id", default="default", db_index=True)
+    name = models.CharField(verbose_name="版本名称", max_length=128, default="")
     work_flow = models.JSONField(verbose_name="工作流数据", default=dict)
     publish_user_id = models.UUIDField(verbose_name="发布者id", max_length=128, default=None, null=True)
     publish_user_name = models.CharField(verbose_name="发布者名称", max_length=128, default="")

apps/knowledge/serializers/knowledge_version.py

@@ -0,0 +1,108 @@
+# coding=utf-8
+"""
+    @project: MaxKB
+    @Author：虎虎
+    @file： KnowledgeVersionSerializer.py
+    @date：2025/11/28 18:00
+    @desc:
+"""
+from typing import Dict
+
+from django.db.models import QuerySet
+from rest_framework import serializers
+
+from django.utils.translation import gettext_lazy as _
+
+from common.db.search import page_search
+from common.exception.app_exception import AppApiException
+from knowledge.models import KnowledgeWorkflowVersion, Knowledge
+
+
+class KnowledgeWorkflowVersionEditSerializer(serializers.Serializer):
+    name = serializers.CharField(required=False, max_length=128, allow_null=True, allow_blank=True,
+                                 label=_("Version Name"))
+
+
+class KnowledgeVersionModelSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = KnowledgeWorkflowVersion
+        fields = ['id', 'name', 'workspace_id', 'knowledge_id', 'work_flow', 'publish_user_id', 'publish_user_name',
+                  'create_time',
+                  'update_time']
+
+
+class KnowledgeWorkflowVersionQuerySerializer(serializers.Serializer):
+    knowledge_id = serializers.UUIDField(required=True, label=_("Knowledge ID"))
+    name = serializers.CharField(required=False, allow_null=True, allow_blank=True,
+                                 label=_("summary"))
+
+
+class KnowledgeWorkflowVersionSerializer(serializers.Serializer):
+    workspace_id = serializers.CharField(required=False, label=_("Workspace ID"))
+
+    class Query(serializers.Serializer):
+        workspace_id = serializers.CharField(required=False, allow_null=True, allow_blank=True, label=_("Workspace ID"))
+
+        def get_query_set(self, query):
+            query_set = QuerySet(KnowledgeWorkflowVersion).filter(knowledge_id=query.get('knowledge_id'))
+            if 'name' in query and query.get('name') is not None:
+                query_set = query_set.filter(name__contains=query.get('name'))
+            if 'workspace_id' in self.data and self.data.get('workspace_id') is not None:
+                query_set = query_set.filter(workspace_id=self.data.get('workspace_id'))
+            return query_set.order_by("-create_time")
+
+        def list(self, query, with_valid=True):
+            if with_valid:
+                self.is_valid(raise_exception=True)
+                KnowledgeWorkflowVersionQuerySerializer(data=query).is_valid(raise_exception=True)
+            query_set = self.get_query_set(query)
+            return [KnowledgeVersionModelSerializer(v).data for v in query_set]
+
+        def page(self, query, current_page, page_size, with_valid=True):
+            if with_valid:
+                self.is_valid(raise_exception=True)
+            return page_search(current_page, page_size,
+                               self.get_query_set(query),
+                               post_records_handler=lambda v: KnowledgeVersionModelSerializer(v).data)
+
+    class Operate(serializers.Serializer):
+        workspace_id = serializers.CharField(required=False, allow_null=True, allow_blank=True, label=_("Workspace ID"))
+        knowledge_id = serializers.UUIDField(required=True, label=_("Knowledge ID"))
+        knowledge_version_id = serializers.UUIDField(required=True,
+                                                     label=_("Knowledge version ID"))
+
+        def is_valid(self, *, raise_exception=False):
+            super().is_valid(raise_exception=True)
+            workspace_id = self.data.get('workspace_id')
+            query_set = QuerySet(Knowledge).filter(id=self.data.get('knowledge_id'))
+            if workspace_id:
+                query_set = query_set.filter(workspace_id=workspace_id)
+            if not query_set.exists():
+                raise AppApiException(500, _('Knowledge id does not exist'))
+
+        def one(self, with_valid=True):
+            if with_valid:
+                self.is_valid(raise_exception=True)
+            knowledge_version = QuerySet(KnowledgeWorkflowVersion).filter(knowledge_id=self.data.get('knowledge_id'),
+                                                                          id=self.data.get(
+                                                                              'knowledge_version_id')).first()
+            if knowledge_version is not None:
+                return KnowledgeVersionModelSerializer(knowledge_version).data
+            else:
+                raise AppApiException(500, _('Workflow version does not exist'))
+
+        def edit(self, instance: Dict, with_valid=True):
+            if with_valid:
+                self.is_valid(raise_exception=True)
+                KnowledgeWorkflowVersionEditSerializer(data=instance).is_valid(raise_exception=True)
+            knowledge_version = QuerySet(KnowledgeWorkflowVersion).filter(knowledge_id=self.data.get('knowledge_id'),
+                                                                          id=self.data.get(
+                                                                              'knowledge_version_id')).first()
+            if knowledge_version is not None:
+                name = instance.get('name', None)
+                if name is not None and len(name) > 0:
+                    knowledge_version.name = name
+                knowledge_version.save()
+                return KnowledgeVersionModelSerializer(knowledge_version).data
+            else:
+                raise AppApiException(500, _('Workflow version does not exist'))