feat: add template store dialog and related functionality for knowledge templates

Author: liuruibin

apps/knowledge/serializers/knowledge_workflow.py

@@ -5,6 +5,7 @@
 from functools import reduce
 from typing import Dict, List
 
+import requests
 import uuid_utils.compat as uuid
 from django.core.cache import cache
 from django.db import transaction
@@ -25,7 +26,9 @@
 from common.exception.app_exception import AppApiException
 from common.field.common import UploadedFileField
 from common.result import result
+from common.utils.common import bytes_to_uploaded_file
 from common.utils.common import restricted_loads, generate_uuid
+from common.utils.logger import maxkb_logger
 from common.utils.rsa_util import rsa_long_decrypt
 from common.utils.tool_code import ToolExecutor
 from knowledge.models import KnowledgeScope, Knowledge, KnowledgeType, KnowledgeWorkflow, KnowledgeWorkflowVersion
@@ -70,6 +73,7 @@ class KnowledgeWorkflowActionListQuerySerializer(serializers.Serializer):
     user_name = serializers.CharField(required=False, label=_('Name'), allow_blank=True, allow_null=True)
     state = serializers.CharField(required=False, label=_("State"), allow_blank=True, allow_null=True)
 
+
 class KBWFInstance:
 
     def __init__(self, knowledge_workflow: dict, function_lib_list: List[dict], version: str, tool_list: List[dict]):
@@ -81,6 +85,7 @@ def __init__(self, knowledge_workflow: dict, function_lib_list: List[dict], vers
     def get_tool_list(self):
         return [*(self.tool_list or []), *(self.function_lib_list or [])]
 
+
 class KnowledgeWorkflowActionSerializer(serializers.Serializer):
     workspace_id = serializers.CharField(required=True, label=_('workspace id'))
     knowledge_id = serializers.UUIDField(required=True, label=_('knowledge id'))
@@ -248,13 +253,32 @@ def save_workflow(self, instance: Dict):
 
             knowledge_workflow.save()
             save_workflow_mapping(instance.get('work_flow', {}), ResourceType.KNOWLEDGE, str(knowledge_id))
+
+            # 处理 work_flow_template
+            if instance.get('work_flow_template') is not None:
+                template_instance = instance.get('work_flow_template')
+                download_url = template_instance.get('downloadUrl')
+                # 查找匹配的版本名称
+                res = requests.get(download_url, timeout=5)
+                KnowledgeWorkflowSerializer.Import(data={
+                    'user_id': self.data.get('user_id'),
+                    'workspace_id': self.data.get('workspace_id'),
+                    'knowledge_id': str(knowledge_id),
+                }).import_({'file': bytes_to_uploaded_file(res.content, 'file.kbwf')}, is_import_tool=True)
+
+                try:
+                    requests.get(template_instance.get('downloadCallbackUrl'), timeout=5)
+                except Exception as e:
+                    maxkb_logger.error(f"callback appstore tool download error: {e}")
+
             return {**KnowledgeModelSerializer(knowledge).data, 'document_list': []}
 
     class Import(serializers.Serializer):
         user_id = serializers.UUIDField(required=True, label=_('user id'))
         workspace_id = serializers.CharField(required=True, label=_('workspace id'))
         knowledge_id = serializers.UUIDField(required=True, label=_('knowledge id'))
 
+        @transaction.atomic
         def import_(self, instance: dict, is_import_tool, with_valid=True):
             if with_valid:
                 self.is_valid()
@@ -296,8 +320,10 @@ def import_(self, instance: dict, is_import_tool, with_valid=True):
                 update_tool_map,
             )
             tool_model_list = [self.to_tool(tool, workspace_id, user_id) for tool in tool_list]
-            KnowledgeWorkflow.objects.filter(workspace_id=workspace_id,knowledge_id=knowledge_id).update(
-                work_flow=work_flow
+            KnowledgeWorkflow.objects.filter(workspace_id=workspace_id, knowledge_id=knowledge_id).update_or_create(
+                knowledge_id=knowledge_id,
+                workspace_id=workspace_id,
+                defaults={'work_flow': work_flow}
             )
 
             if is_import_tool:
@@ -373,7 +399,6 @@ def export(self, with_valid=True):
             except Exception as e:
                 return result.error(str(e), response_status=status.HTTP_500_INTERNAL_SERVER_ERROR)
 
-
     class Operate(serializers.Serializer):
         user_id = serializers.UUIDField(required=True, label=_('user id'))
         workspace_id = serializers.CharField(required=True, label=_('workspace id'))
@@ -416,6 +441,23 @@ def edit(self, instance: Dict):
                                                                  'work_flow': instance.get('work_flow')
                                                              })
                 return self.one()
+            if instance.get("work_flow_template"):
+                template_instance = instance.get('work_flow_template')
+                download_url = template_instance.get('downloadUrl')
+                # 查找匹配的版本名称
+                res = requests.get(download_url, timeout=5)
+                KnowledgeWorkflowSerializer.Import(data={
+                    'user_id': self.data.get('user_id'),
+                    'workspace_id': self.data.get('workspace_id'),
+                    'knowledge_id': str(self.data.get('knowledge_id')),
+                }).import_({'file': bytes_to_uploaded_file(res.content, 'file.kbwf')}, is_import_tool=False)
+
+                try:
+                    requests.get(template_instance.get('downloadCallbackUrl'), timeout=5)
+                except Exception as e:
+                    maxkb_logger.error(f"callback appstore tool download error: {e}")
+
+                return self.one()
 
         def one(self):
             self.is_valid(raise_exception=True)

ui/src/api/tool/store.ts

@@ -32,6 +32,20 @@ const getStoreToolList: (param?: any, loading?: Ref<boolean>) => Promise<Result<
   return get('/workspace/store/tool', param, loading)
 }
 
+const getStoreKBList: (param?: any, loading?: Ref<boolean>) => Promise<Result<any>> = (
+  param,
+  loading,
+) => {
+  return get('/workspace/store/knowledge_template', param, loading)
+}
+
+const getStoreAppList: (param?: any, loading?: Ref<boolean>) => Promise<Result<any>> = (
+  param,
+  loading,
+) => {
+  return get('/workspace/store/application_template', param, loading)
+}
+
 /**
  * 工具商店-添加系统内置
  */
@@ -57,6 +71,8 @@ const addStoreTool: (
 export default {
   getInternalToolList,
   getStoreToolList,
+  getStoreKBList,
+  getStoreAppList,
   addInternalTool,
   addStoreTool
 }

ui/src/views/knowledge-workflow/index.vue

@@ -26,6 +26,13 @@
         </el-button>
       </div>
       <div v-else-if="!route.path.includes('share/')">
+        <el-button
+          class="ml-8"
+          v-if="permissionPrecise.create()"
+          @click="openTemplateStoreDialog()"
+        >
+          {{ $t('模版中心') }}
+        </el-button>
         <el-button @click="showPopover = !showPopover">
           <AppIcon iconName="app-add-outlined" class="mr-4" />
           {{ $t('workflow.setting.addComponent') }}
@@ -161,6 +168,7 @@
       v-click-outside="clickoutsideHistory"
       @refreshVersion="refreshVersion"
     />
+  <TemplateStoreDialog ref="templateStoreDialogRef" :api-type="apiType" source="work_flow" @refresh="getDetail"/>
   </div>
 </template>
 <script setup lang="ts">
@@ -186,6 +194,7 @@ import permissionMap from '@/permission'
 import { WorkflowMode } from '@/enums/application'
 import { loadSharedApi } from '@/utils/dynamics-api/shared-api'
 import { knowledgeBaseNode } from '@/workflow/common/data'
+import TemplateStoreDialog from "@/views/knowledge/template-store/TemplateStoreDialog.vue";
 provide('getResourceDetail', () => detail)
 provide('workflowMode', WorkflowMode.Knowledge)
 provide('loopWorkflowMode', WorkflowMode.KnowledgeLoop)
@@ -649,6 +658,12 @@ const toImportDoc = () => {
   }
 }
 
+const templateStoreDialogRef = ref()
+function openTemplateStoreDialog() {
+  templateStoreDialogRef.value?.open(folderId)
+}
+
+
 /**
  * 定时保存
  */

ui/src/views/knowledge/component/KnowledgeListContainer.vue

@@ -35,6 +35,13 @@
             <el-option v-for="u in user_options" :key="u.id" :value="u.id" :label="u.nick_name" />
           </el-select>
         </div>
+        <el-button
+          class="ml-8"
+          v-if="!isShared && permissionPrecise.create()"
+          @click="openTemplateStoreDialog()"
+        >
+          {{ $t('模版中心') }}
+        </el-button>
         <el-dropdown trigger="click" v-if="!isShared && permissionPrecise.create()">
           <el-button type="primary" class="ml-8">
             {{ $t('common.create') }}
@@ -305,6 +312,7 @@
     ref="ResourceAuthorizationDrawerRef"
     v-if="apiType === 'workspace'"
   />
+  <TemplateStoreDialog ref="templateStoreDialogRef" :api-type="apiType" @refresh="getList" />
 </template>
 
 <script lang="ts" setup>
@@ -329,6 +337,7 @@ import { i18n_name } from '@/utils/common'
 import { SourceTypeEnum } from '@/enums/common'
 import { loadSharedApi } from '@/utils/dynamics-api/shared-api'
 import permissionMap from '@/permission'
+import TemplateStoreDialog from "@/views/knowledge/template-store/TemplateStoreDialog.vue";
 const router = useRouter()
 const route = useRoute()
 const { folder, user, knowledge } = useStore()
@@ -541,6 +550,11 @@ function refreshFolder() {
   emit('refreshFolder')
 }
 
+const templateStoreDialogRef = ref()
+function openTemplateStoreDialog() {
+  templateStoreDialogRef.value?.open(folder.currentFolder.id)
+}
+
 onMounted(() => {
   if (apiType.value !== 'workspace') {
     folder.setCurrentFolder({

ui/src/views/knowledge/create-component/CreateWorkflowKnowledgeDialog.vue

@@ -52,15 +52,19 @@ const dialogVisible = ref<boolean>(false)
 const currentFolder = ref<any>(null)
 
 const workflowDefault = ref(knowledgeTemplate.default)
+const workflowTemplate = ref()
 
 watch(dialogVisible, (bool) => {
   if (!bool) {
     currentFolder.value = null
   }
 })
 
-const open = (folder: string) => {
+const open = (folder: string, workflow?: any) => {
   currentFolder.value = folder
+  if (workflow) {
+    workflowTemplate.value = workflow
+  }
   dialogVisible.value = true
 }
 
@@ -69,6 +73,7 @@ const submitHandle = async () => {
     const obj = {
       folder_id: currentFolder.value?.id,
       work_flow: workflowDefault.value,
+      work_flow_template: workflowTemplate.value,
       ...BaseFormRef.value.form,
     }
     loadSharedApi({ type: 'knowledge', systemType: apiType.value })

ui/src/views/knowledge/template-store/InternalDescDrawer.vue

@@ -0,0 +1,93 @@
+<template>
+  <el-drawer v-model="visibleInternalDesc" size="60%" :append-to-body="true">
+    <template #header>
+      <div class="flex align-center" style="margin-left: -8px">
+        <el-button class="cursor mr-4" link @click.prevent="visibleInternalDesc = false">
+          <el-icon :size="20">
+            <Back />
+          </el-icon>
+        </el-button>
+        <h4>详情</h4>
+      </div>
+    </template>
+
+    <div>
+      <div class="card-header">
+        <div class="flex-between">
+          <div class="title flex align-center">
+            <el-avatar
+              v-if="isAppIcon(toolDetail?.icon)"
+              shape="square"
+              :size="64"
+              style="background: none"
+              class="mr-8"
+            >
+              <img :src="toolDetail?.icon" alt="" />
+            </el-avatar>
+            <el-avatar
+              v-else-if="toolDetail?.name"
+              :name="toolDetail?.name"
+              pinyinColor
+              shape="square"
+              :size="64"
+              class="mr-8"
+            />
+            <div class="ml-16">
+              <h3 class="mb-8">{{ toolDetail.name }}</h3>
+              <el-text type="info" v-if="toolDetail?.desc">
+                {{ toolDetail.desc }}
+              </el-text>
+            </div>
+          </div>
+          <div @click.stop>
+            <el-button type="primary" @click="addInternalTool(toolDetail)">
+              {{ $t('common.add') }}
+            </el-button>
+          </div>
+        </div>
+      </div>
+      <MdPreview
+        ref="editorRef"
+        editorId="preview-only"
+        :modelValue="markdownContent"
+        style="background: none"
+        noImgZoomIn
+      />
+    </div>
+  </el-drawer>
+</template>
+
+<script setup lang="ts">
+import { ref, watch } from 'vue'
+import { cloneDeep } from 'lodash'
+import { isAppIcon, numberFormat } from '@/utils/common'
+const emit = defineEmits(['refresh', 'addTool'])
+
+const visibleInternalDesc = ref(false)
+const markdownContent = ref('')
+const toolDetail = ref<any>({})
+
+watch(visibleInternalDesc, (bool) => {
+  if (!bool) {
+    markdownContent.value = ''
+  }
+})
+
+const open = (data: any, detail: any) => {
+  toolDetail.value = detail
+  if (data) {
+    markdownContent.value = cloneDeep(data)
+  }
+  visibleInternalDesc.value = true
+}
+
+const addInternalTool = (data: any) => {
+  emit('addTool', data)
+  visibleInternalDesc.value = false
+}
+
+defineExpose({
+  open,
+})
+</script>
+<style lang="scss"></style>