Merge branch 'main' of https://github.com/maxkb-dev/maxkb

README.md

@@ -39,7 +39,8 @@ docker run -d --name=maxkb --restart=always -p 8080:8080 -v C:/maxkb:/var/lib/po
 
 - 你也可以通过 [1Panel 应用商店](https://apps.fit2cloud.com/1panel) 快速部署 MaxKB；
 - 如果是内网环境，推荐使用 [离线安装包](https://community.fit2cloud.com/#/products/maxkb/downloads) 进行安装部署；
-- MaxKB 产品版本分为社区版和专业版，详情请参见：[MaxKB 产品版本对比](https://maxkb.cn/pricing.html)。
+- MaxKB 产品版本分为社区版和专业版，详情请参见：[MaxKB 产品版本对比](https://maxkb.cn/pricing.html)；
+- 如果您需要向团队介绍 MaxKB，可以使用这个 [官方 PPT 材料](https://maxkb.cn/download/introduce-maxkb_202411.pdf)。
 
 如你有更多问题，可以查看使用手册，或者通过论坛与我们交流。
 

apps/application/flow/i_step_node.py

@@ -32,7 +32,7 @@ def write_context(step_variable: Dict, global_variable: Dict, node, workflow):
         if workflow.is_result(node, NodeResult(step_variable, global_variable)) and 'answer' in step_variable:
             answer = step_variable['answer']
             yield answer
-            workflow.append_answer(answer)
+            node.answer_text = answer
     if global_variable is not None:
         for key in global_variable:
             workflow.context[key] = global_variable[key]

apps/application/flow/step_node/document_extract_node/i_document_extract_node.py

@@ -9,12 +9,7 @@ from common.util.field_message import ErrMessage
 
 
 class DocumentExtractNodeSerializer(serializers.Serializer):
-    # 需要查询的数据集id列表
+    document_list = serializers.ListField(required=False, error_messages=ErrMessage.list("文档"))
-    file_list = serializers.ListField(required=True, child=serializers.UUIDField(required=True),
-                                      error_messages=ErrMessage.list("数据集id列表"))
-
-    def is_valid(self, *, raise_exception=False):
-        super().is_valid(raise_exception=True)
 
 
 class IDocumentExtractNode(INode):
@@ -24,7 +19,9 @@ class IDocumentExtractNode(INode):
         return DocumentExtractNodeSerializer
 
     def _run(self):
-        return self.execute(**self.flow_params_serializer.data)
+        res = self.workflow_manage.get_reference_field(self.node_params_serializer.data.get('document_list')[0],
+                                                       self.node_params_serializer.data.get('document_list')[1:])
+        return self.execute(document=res, **self.flow_params_serializer.data)
 
-    def execute(self, file_list, **kwargs) -> NodeResult:
+    def execute(self, document, **kwargs) -> NodeResult:
         pass

apps/application/flow/step_node/document_extract_node/impl/base_document_extract_node.py

@@ -1,11 +1,34 @@
 # coding=utf-8
+from django.db.models import QuerySet
 
+from application.flow.i_step_node import NodeResult
 from application.flow.step_node.document_extract_node.i_document_extract_node import IDocumentExtractNode
+from dataset.models import File
 
 
 class BaseDocumentExtractNode(IDocumentExtractNode):
-    def execute(self, file_list, **kwargs):
+    def execute(self, document, **kwargs):
-        pass
+        self.context['document_list'] = document
+        content = ''
+        spliter = '\n-----------------------------------\n'
+        if len(document) > 0:
+            for doc in document:
+                file = QuerySet(File).filter(id=doc['file_id']).first()
+                file_type = doc['name'].split('.')[-1]
+                if file_type.lower() in ['txt', 'md', 'csv', 'html']:
+                    content += spliter + doc['name'] + '\n' + file.get_byte().tobytes().decode('utf-8')
+
+
+        return NodeResult({'content': content}, {})
 
     def get_details(self, index: int, **kwargs):
-        pass
+        return {
+            'name': self.node.properties.get('stepName'),
+            "index": index,
+            'run_time': self.context.get('run_time'),
+            'type': self.node.type,
+            'content': self.context.get('content'),
+            'status': self.status,
+            'err_message': self.err_message,
+            'document_list': self.context.get('document_list')
+        }

apps/application/flow/step_node/image_understand_step_node/i_image_understand_node.py

@@ -18,7 +18,7 @@ class ImageUnderstandNodeSerializer(serializers.Serializer):
 
     is_result = serializers.BooleanField(required=False, error_messages=ErrMessage.boolean('是否返回内容'))
 
-    image_list = serializers.ListField(required=False, error_messages=ErrMessage.list("图片仅1张"))
+    image_list = serializers.ListField(required=False, error_messages=ErrMessage.list("图片"))
 
 
 class IImageUnderstandNode(INode):

apps/application/flow/step_node/image_understand_step_node/impl/base_image_understand_node.py

@@ -25,7 +25,7 @@ def _write_context(node_variable: Dict, workflow_variable: Dict, node: INode, wo
     node.context['question'] = node_variable['question']
     node.context['run_time'] = time.time() - node.context['start_time']
     if workflow.is_result(node, NodeResult(node_variable, workflow_variable)):
-        workflow.answer += answer
+        node.answer_text = answer
 
 
 def write_context_stream(node_variable: Dict, workflow_variable: Dict, node: INode, workflow):

apps/application/flow/step_node/start_node/impl/base_start_node.py

@@ -52,8 +52,12 @@ class BaseStartStepNode(IStarNode):
         """
         开始节点 初始化全局变量
         """
-        return NodeResult({'question': question, 'image': self.workflow_manage.image_list},
+        node_variable = {
-                          workflow_variable)
+            'question': question,
+            'image': self.workflow_manage.image_list,
+            'document': self.workflow_manage.document_list
+        }
+        return NodeResult(node_variable, workflow_variable)
 
     def get_details(self, index: int, **kwargs):
         global_fields = []

apps/application/flow/workflow_manage.py

@@ -240,16 +240,20 @@ class NodeChunk:
 class WorkflowManage:
     def __init__(self, flow: Flow, params, work_flow_post_handler: WorkFlowPostHandler,
                  base_to_response: BaseToResponse = SystemToResponse(), form_data=None, image_list=None,
+                 document_list=None,
                  start_node_id=None,
                  start_node_data=None, chat_record=None):
         if form_data is None:
             form_data = {}
         if image_list is None:
             image_list = []
+        if document_list is None:
+            document_list = []
         self.start_node = None
         self.start_node_result_future = None
         self.form_data = form_data
         self.image_list = image_list
+        self.document_list = document_list
         self.params = params
         self.flow = flow
         self.lock = threading.Lock()
@@ -511,7 +515,7 @@ class WorkflowManage:
             if index == 0:
                 result.append(answer.get('content'))
                 continue
-            if answer.get('type') != answer_text_list[index - 1]:
+            if answer.get('type') != answer_text_list[index - 1].get('type'):
                 result.append(answer.get('content'))
             else:
                 result[-1] += answer.get('content')

apps/application/migrations/0011_application_model_params_setting.py

@@ -13,6 +13,6 @@ class Migration(migrations.Migration):
         migrations.AddField(
             model_name='application',
             name='model_params_setting',
-            field=models.JSONField(default={}, verbose_name='模型参数相关设置'),
+            field=models.JSONField(default=dict, verbose_name='模型参数相关设置'),
         ),
     ]

apps/application/migrations/0017_application_tts_model_params_setting.py

@@ -13,6 +13,6 @@ class Migration(migrations.Migration):
         migrations.AddField(
             model_name='application',
             name='tts_model_params_setting',
-            field=models.JSONField(default={}, verbose_name='模型参数相关设置'),
+            field=models.JSONField(default=dict, verbose_name='模型参数相关设置'),
         ),
     ]

apps/application/migrations/0019_application_file_upload_enable_and_more.py

@@ -23,7 +23,7 @@ class Migration(migrations.Migration):
         migrations.AddField(
             model_name='application',
             name='file_upload_setting',
-            field=models.JSONField(default={}, verbose_name='文件上传相关设置'),
+            field=models.JSONField(default=dict, verbose_name='文件上传相关设置'),
         ),
         migrations.AddField(
             model_name='chatrecord',

apps/application/models/application.py

@@ -48,8 +48,8 @@ class Application(AppModelMixin):
     model = models.ForeignKey(Model, on_delete=models.SET_NULL, db_constraint=False, blank=True, null=True)
     dataset_setting = models.JSONField(verbose_name="数据集参数设置", default=get_dataset_setting_dict)
     model_setting = models.JSONField(verbose_name="模型参数相关设置", default=get_model_setting_dict)
-    model_params_setting = models.JSONField(verbose_name="模型参数相关设置", default={})
+    model_params_setting = models.JSONField(verbose_name="模型参数相关设置", default=dict)
-    tts_model_params_setting = models.JSONField(verbose_name="模型参数相关设置", default={})
+    tts_model_params_setting = models.JSONField(verbose_name="模型参数相关设置", default=dict)
     problem_optimization = models.BooleanField(verbose_name="问题优化", default=False)
     icon = models.CharField(max_length=256, verbose_name="应用icon", default="/ui/favicon.ico")
     work_flow = models.JSONField(verbose_name="工作流数据", default=dict)
@@ -67,7 +67,7 @@ class Application(AppModelMixin):
     tts_type = models.CharField(verbose_name="语音播放类型", max_length=20, default="BROWSER")
     clean_time = models.IntegerField(verbose_name="清理时间", default=180)
     file_upload_enable = models.BooleanField(verbose_name="文件上传是否启用", default=False)
-    file_upload_setting = models.JSONField(verbose_name="文件上传相关设置", default={})
+    file_upload_setting = models.JSONField(verbose_name="文件上传相关设置", default=dict)
 
 
     @staticmethod

apps/application/serializers/chat_message_serializers.py

@@ -230,7 +230,8 @@ class ChatMessageSerializer(serializers.Serializer):
     client_id = serializers.CharField(required=True, error_messages=ErrMessage.char("客户端id"))
     client_type = serializers.CharField(required=True, error_messages=ErrMessage.char("客户端类型"))
     form_data = serializers.DictField(required=False, error_messages=ErrMessage.char("全局变量"))
-    image_list = serializers.ListField(required=False, error_messages=ErrMessage.list("图片仅1张"))
+    image_list = serializers.ListField(required=False, error_messages=ErrMessage.list("图片"))
+    document_list = serializers.ListField(required=False, error_messages=ErrMessage.list("文档"))
 
     def is_valid_application_workflow(self, *, raise_exception=False):
         self.is_valid_intraday_access_num()
@@ -322,6 +323,7 @@ class ChatMessageSerializer(serializers.Serializer):
         client_type = self.data.get('client_type')
         form_data = self.data.get('form_data')
         image_list = self.data.get('image_list')
+        document_list = self.data.get('document_list')
         user_id = chat_info.application.user_id
         chat_record_id = self.data.get('chat_record_id')
         chat_record = None
@@ -336,7 +338,7 @@ class ChatMessageSerializer(serializers.Serializer):
                                            'client_id': client_id,
                                            'client_type': client_type,
                                            'user_id': user_id}, WorkFlowPostHandler(chat_info, client_id, client_type),
-                                          base_to_response, form_data, image_list, self.data.get('runtime_node_id'),
+                                          base_to_response, form_data, image_list, document_list, self.data.get('runtime_node_id'),
                                           self.data.get('node_data'), chat_record)
         r = work_flow_manage.run()
         return r

apps/application/views/chat_views.py

@@ -132,6 +132,8 @@ class ChatView(APIView):
 
                                                'image_list': request.data.get(
                                                    'image_list') if 'image_list' in request.data else [],
+                                               'document_list': request.data.get(
+                                                   'document_list') if 'document_list' in request.data else [],
                                                'client_type': request.auth.client_type,
                                                'runtime_node_id': request.data.get('runtime_node_id', None),
                                                'node_data': request.data.get('node_data', {}),

ui/package.json

@@ -17,6 +17,7 @@
     "@ctrl/tinycolor": "^4.1.0",
     "@logicflow/core": "^1.2.27",
     "@logicflow/extension": "^1.2.27",
+    "@antv/layout": "^0.3.1",
     "@vueuse/core": "^10.9.0",
     "@wecom/jssdk": "^2.1.0",
     "axios": "^0.28.0",

ui/src/api/type/application.ts

@@ -39,7 +39,8 @@ interface chatType {
   record_id: string
   chat_id: string
   vote_status: string
-  status?: number
+  status?: number,
+  execution_details: any[]
 }
 
 export class ChatRecordManage {

ui/src/components/ai-chat/component/answer-content/index.vue

@@ -1,14 +1,20 @@
 <template>
   <div class="item-content mb-16 lighter">
-    <div v-for="(answer_text, index) in chatRecord.answer_text_list" :key="index">
+    <template v-for="(answer_text, index) in chatRecord.answer_text_list" :key="index">
       <div class="avatar">
         <img v-if="application.avatar" :src="application.avatar" height="32px" width="32px" />
         <LogoIcon v-else height="32px" width="32px" />
       </div>
       <div class="content">
-        <el-card shadow="always" class="dialog-card">
+        <el-card shadow="always" class="dialog-card mb-8">
           <MdRenderer
-            v-if="answer_text"
+            v-if="
+              (chatRecord.write_ed === undefined || chatRecord.write_ed === true) && !answer_text
+            "
+            source=" 抱歉，没有查找到相关内容，请重新描述您的问题或提供更多信息。"
+          ></MdRenderer>
+          <MdRenderer
+            v-else-if="answer_text"
             :source="answer_text"
             :send-message="chatMessage"
           ></MdRenderer>
@@ -23,16 +29,18 @@
             <KnowledgeSource :data="chatRecord" :type="application.type" />
           </div>
         </el-card>
-        <OperationButton
-          :type="type"
-          :application="application"
-          :chat-record="chatRecord"
-          :loading="loading"
-          :start-chat="startChat"
-          :stop-chat="stopChat"
-          :regenerationChart="regenerationChart"
-        ></OperationButton>
       </div>
+    </template>
+    <div class="content">
+      <OperationButton
+        :type="type"
+        :application="application"
+        :chat-record="chatRecord"
+        :loading="loading"
+        :start-chat="startChat"
+        :stop-chat="stopChat"
+        :regenerationChart="regenerationChart"
+      ></OperationButton>
     </div>
   </div>
 </template>

ui/src/components/ai-chat/component/chat-input-operate/index.vue

@@ -20,10 +20,12 @@
 
       <div class="operate flex align-center">
         <span v-if="props.applicationDetails.file_upload_enable" class="flex align-center">
+<!--            accept="image/jpeg, image/png, image/gif"-->
           <el-upload
             action="#"
             :auto-upload="false"
             :show-file-list="false"
+            :accept="[...imageExtensions, ...documentExtensions].map((ext) => '.' + ext).join(',')"
             :on-change="(file: any, fileList: any) => uploadFile(file, fileList)"
           >
             <el-button text>
@@ -126,6 +128,13 @@ const localLoading = computed({
     emit('update:loading', v)
   }
 })
+
+
+const imageExtensions = ['jpg', 'jpeg', 'png', 'gif', 'bmp']
+const documentExtensions = ['pdf', 'docx', 'txt', 'xls', 'xlsx', 'md', 'html', 'csv']
+const videoExtensions = ['mp4', 'avi', 'mov', 'mkv', 'flv']
+const audioExtensions = ['mp3', 'wav', 'aac', 'flac']
+
 const uploadFile = async (file: any, fileList: any) => {
   const { maxFiles, fileLimit } = props.applicationDetails.file_upload_setting
   if (fileList.length > maxFiles) {
@@ -141,7 +150,18 @@ const uploadFile = async (file: any, fileList: any) => {
   const formData = new FormData()
   for (const file of fileList) {
     formData.append('file', file.raw, file.name)
-    uploadFileList.value.push(file)
+    //
+    const extension = file.name.split('.').pop().toLowerCase() // 获取文件后缀名并转为小写
+
+    if (imageExtensions.includes(extension)) {
+      uploadImageList.value.push(file)
+    } else if (documentExtensions.includes(extension)) {
+      uploadDocumentList.value.push(file)
+    } else if (videoExtensions.includes(extension)) {
+      // videos.push(file)
+    } else if (audioExtensions.includes(extension)) {
+      // audios.push(file)
+    }
   }
 
   if (!chatId_context.value) {
@@ -158,21 +178,29 @@ const uploadFile = async (file: any, fileList: any) => {
     )
     .then((response) => {
       fileList.splice(0, fileList.length)
-      uploadFileList.value.forEach((file: any) => {
+      uploadImageList.value.forEach((file: any) => {
         const f = response.data.filter((f: any) => f.name === file.name)
         if (f.length > 0) {
           file.url = f[0].url
           file.file_id = f[0].file_id
         }
       })
-      console.log(uploadFileList.value)
+      uploadDocumentList.value.forEach((file: any) => {
+        const f = response.data.filter((f: any) => f.name === file.name)
+        if (f.length > 0) {
+          file.url = f[0].url
+          file.file_id = f[0].file_id
+        }
+      })
+      console.log(uploadDocumentList.value, uploadImageList.value)
     })
 }
 const recorderTime = ref(0)
 const startRecorderTime = ref(false)
 const recorderLoading = ref(false)
 const inputValue = ref<string>('')
-const uploadFileList = ref<Array<any>>([])
+const uploadImageList = ref<Array<any>>([])
+const uploadDocumentList = ref<Array<any>>([])
 const mediaRecorderStatus = ref(true)
 // 定义响应式引用
 const mediaRecorder = ref<any>(null)
@@ -289,15 +317,20 @@ const handleTimeChange = () => {
     handleTimeChange()
   }, 1000)
 }
+
 function sendChatHandle(event: any) {
   if (!event.ctrlKey) {
     // 如果没有按下组合键ctrl，则会阻止默认事件
     event.preventDefault()
     if (!isDisabledChart.value && !props.loading && !event.isComposing) {
       if (inputValue.value.trim()) {
-        props.sendMessage(inputValue.value, { image_list: uploadFileList.value })
+        props.sendMessage(inputValue.value, {
+          image_list: uploadImageList.value,
+          document_list: uploadDocumentList.value
+        })
         inputValue.value = ''
-        uploadFileList.value = []
+        uploadImageList.value = []
+        uploadDocumentList.value = []
         quickInputRef.value.textareaStyle.height = '45px'
       }
     }

ui/src/components/ai-chat/component/question-content/index.vue

@@ -22,10 +22,19 @@
 </template>
 <script setup lang="ts">
 import { type chatType } from '@/api/type/application'
-defineProps<{
+import { onMounted } from 'vue'
+const props = defineProps<{
   application: any
   chatRecord: chatType
 }>()
+
+onMounted(() => {
+  if (props.chatRecord.execution_details?.length > 0) {
+    props.chatRecord.execution_details[0].image_list.forEach((image: any) => {
+      console.log('image', image.name, image.url)
+    })
+  }
+})
 </script>
 <style lang="scss" scoped>
 </style>

ui/src/components/ai-chat/index.vue

@@ -370,7 +370,7 @@ function chatMessage(chat?: any, problem?: string, re_chat?: boolean, other_para
  */
 function getSourceDetail(row: any) {
   logApi.getRecordDetail(id || props.appId, row.chat_id, row.record_id, loading).then((res) => {
-    const exclude_keys = ['answer_text', 'id']
+    const exclude_keys = ['answer_text', 'id', 'answer_text_list']
     Object.keys(res.data).forEach((key) => {
       if (!exclude_keys.includes(key)) {
         row[key] = res.data[key]

ui/src/components/dynamics-form/items/table/TableCheckbox.vue

@@ -72,7 +72,7 @@ const props = defineProps<{
   modelValue?: Array<any>
 }>()
 const rowTemp = ref<any>()
-const evalF = (text: string, row: any) => {
+const evalF: (text: string, row: any) => string = (text: string, row: any) => {
   rowTemp.value = row
   return eval(text)
 }
@@ -167,8 +167,8 @@ watch(
 
 const activeText = computed(() => {
   if (props.modelValue) {
-    const rows = option_list.value.filter(
+    const rows = option_list.value.filter((f: any) =>
-      (f: any) => props.modelValue?.includes(f[valueField.value])
+      props.modelValue?.includes(f[valueField.value])
     )
     if (rows) {
       if (rows.length > 3) {

ui/src/components/icons/index.ts

@@ -1228,6 +1228,28 @@ export const iconMap: any = {
       ])
     }
   },
+
+  'app-beautify': {
+    iconReader: () => {
+      return h('i', [
+        h(
+          'svg',
+          {
+            style: { height: '100%', width: '100%' },
+            viewBox: '0 0 1024 1024',
+            version: '1.1',
+            xmlns: 'http://www.w3.org/2000/svg'
+          },
+          [
+            h('path', {
+              d: 'M739.6864 689.92l4.2496 3.584 136.4992 135.936a34.1504 34.1504 0 0 1-43.9296 51.968l-4.1984-3.584-136.5504-135.936a34.1504 34.1504 0 0 1 43.9296-51.968zM663.4496 151.552a34.1504 34.1504 0 0 1 51.2512 30.464l-5.9392 216.6272 156.4672 146.1248a34.1504 34.1504 0 0 1-8.6528 55.808l-4.8128 1.792-202.8032 61.0816-87.4496 197.12a34.1504 34.1504 0 0 1-56.32 9.216l-3.2768-4.096-119.5008-178.432-209.9712-24.064a34.1504 34.1504 0 0 1-26.1632-50.176l2.7648-4.3008 129.28-171.7248-42.5472-212.3776a34.1504 34.1504 0 0 1 40.448-40.1408l4.6592 1.3312 198.912 72.3456z m-18.6368 89.7536l-144.5376 83.968a34.1504 34.1504 0 0 1-28.8256 2.56L314.5728 270.592l33.792 167.8848c1.4848 7.68 0.3584 15.5136-3.1744 22.3232l-3.072 4.9152-102.656 136.2944 166.4 19.1488c8.2944 0.9216 15.872 4.864 21.4016 10.9568l3.072 3.9424 93.8496 140.032 68.7104-154.7776a34.1504 34.1504 0 0 1 16.7936-17.0496l4.608-1.792 160.9216-48.4864-124.2624-116.0192a34.1504 34.1504 0 0 1-10.4448-20.0704l-0.3584-5.7856 4.6592-170.9056z',
+              fill: 'currentColor'
+            })
+          ]
+        )
+      ])
+    }
+  }
   // 'app-history-outlined': {
   //   iconReader: () => {
   //     return h('i', [

ui/src/components/markdown/EchartsRander.vue

@@ -7,12 +7,7 @@
 import { onMounted, nextTick, watch, onBeforeUnmount, ref } from 'vue'
 import * as echarts from 'echarts'
 const tmp = ref()
-const props = defineProps({
+const props = defineProps<{ option: string }>()
-  option: {
-    type: String,
-    required: true
-  }
-})
 const chartsRef = ref()
 
 const style = ref({

ui/src/components/markdown/FormRander.vue

@@ -20,7 +20,7 @@ import { computed, ref } from 'vue'
 import DynamicsForm from '@/components/dynamics-form/index.vue'
 const props = defineProps<{
   form_setting: string
-  sendMessage: (question: string, type: 'old' | 'new', other_params_data?: any) => void
+  sendMessage?: (question: string, type: 'old' | 'new', other_params_data?: any) => void
 }>()
 const form_setting_data = computed(() => {
   if (props.form_setting) {
@@ -68,11 +68,13 @@ const submit = () => {
   dynamicsFormRef.value?.validate().then(() => {
     _submit.value = true
     const setting = JSON.parse(props.form_setting)
-    props.sendMessage('', 'old', {
+    if (props.sendMessage) {
-      runtime_node_id: setting.runtime_node_id,
+      props.sendMessage('', 'old', {
-      chat_record_id: setting.chat_record_id,
+        runtime_node_id: setting.runtime_node_id,
-      node_data: form_data.value
+        chat_record_id: setting.chat_record_id,
-    })
+        node_data: form_data.value
+      })
+    }
   })
 }
 </script>

ui/src/components/markdown/MdRenderer.vue

@@ -15,7 +15,7 @@
       :option="item.content"
     ></EchartsRander>
     <FormRander
-      :sendMessage="sendMessage"
+      :send-message="sendMessage"
       v-else-if="item.type === 'form_rander'"
       :form_setting="item.content"
     ></FormRander>

ui/src/workflow/common/NodeContainer.vue

@@ -51,7 +51,7 @@
           </div>
         </div>
         <el-collapse-transition>
-          <div @mousedown.stop @keydown.stop @click.stop v-if="showNode" class="mt-16">
+          <div @mousedown.stop @keydown.stop @click.stop v-show="showNode" class="mt-16">
             <el-alert
               v-if="node_status != 200"
               class="mb-16"

ui/src/workflow/common/NodeControl.vue

@@ -11,6 +11,9 @@
     <el-button link @click="fitView">
       <AppIcon iconName="app-fitview" title="适应"></AppIcon>
     </el-button>
+    <el-button link @click="layout">
+      <AppIcon iconName="app-beautify" title="美化"></AppIcon>
+    </el-button>
   </el-card>
 </template>
 
@@ -30,5 +33,8 @@ function fitView() {
   props.lf?.resetTranslate()
   props.lf?.fitView()
 }
+const layout = () => {
+  props.lf?.extension.dagre.layout()
+}
 </script>
 <style scoped></style>

ui/src/workflow/common/app-node.ts

@@ -137,6 +137,26 @@ class AppNode extends HtmlResize.view {
 }
 
 class AppNodeModel extends HtmlResize.model {
+  refreshDeges() {
+    // 更新节点连接边的path
+    this.incoming.edges.forEach((edge: any) => {
+      // 调用自定义的更新方案
+      edge.updatePathByAnchor()
+    })
+    this.outgoing.edges.forEach((edge: any) => {
+      edge.updatePathByAnchor()
+    })
+  }
+  set_position(position: { x?: number; y?: number }) {
+    const { x, y } = position
+    if (x) {
+      this.x = x
+    }
+    if (y) {
+      this.y = y
+    }
+    this.refreshDeges()
+  }
   getResizeOutlineStyle() {
     const style = super.getResizeOutlineStyle()
     style.stroke = 'none'

ui/src/workflow/index.vue

@@ -5,13 +5,14 @@
 </template>
 <script setup lang="ts">
 import LogicFlow from '@logicflow/core'
-import { ref, onMounted, computed } from 'vue'
+import { ref, onMounted, computed, nextTick } from 'vue'
 import AppEdge from './common/edge'
 import Control from './common/NodeControl.vue'
 import { baseNodes } from '@/workflow/common/data'
 import '@logicflow/extension/lib/style/index.css'
 import '@logicflow/core/dist/style/index.css'
 import { initDefaultShortcut } from '@/workflow/common/shortcut'
+import Dagre from '@/workflow/plugins/dagre'
 const nodes: any = import.meta.glob('./nodes/**/index.ts', { eager: true })
 
 defineOptions({ name: 'WorkFlow' })
@@ -61,6 +62,7 @@ const renderGraphData = (data?: any) => {
   const container: any = document.querySelector('#container')
   if (container) {
     lf.value = new LogicFlow({
+      plugins: [Dagre],
       textEdit: false,
       adjustEdge: false,
       adjustEdgeStartAndEnd: false,

ui/src/workflow/nodes/document-extract-node/index.vue

@@ -10,7 +10,7 @@
         label-width="auto"
         ref="DatasetNodeFormRef"
       >
-        <el-form-item label="选择文件" :rules="{
+        <el-form-item label="选择文档" :rules="{
             type: 'array',
             required: true,
             message: '请选择文件',
@@ -21,8 +21,8 @@
             ref="nodeCascaderRef"
             :nodeModel="nodeModel"
             class="w-full"
-            placeholder="请选择文件"
+            placeholder="请选择文档"
-            v-model="form.file_list"
+            v-model="form.document_list"
           />
         </el-form-item>
       </el-form>
@@ -39,7 +39,7 @@ import NodeCascader from '@/workflow/common/NodeCascader.vue'
 const props = defineProps<{ nodeModel: any }>()
 
 const form = {
-  file_list: []
+  document_list: ["start-node", "document"]
 }
 
 

ui/src/workflow/plugins/dagre.ts

@@ -0,0 +1,69 @@
+import { DagreLayout, type DagreLayoutOptions } from '@antv/layout'
+
+export default class Dagre {
+  static pluginName = 'dagre'
+  lf: any
+  option: DagreLayoutOptions | any
+  render(lf: any) {
+    this.lf = lf
+  }
+
+  /**
+   * option: {
+   *   rankdir: "TB", // layout 方向, 可选 TB, BT, LR, RL
+   *   align: undefined, // 节点对齐方式，可选 UL, UR, DL, DR
+   *   nodeSize: undefined, // 节点大小
+   *   nodesepFunc: undefined, // 节点水平间距(px)
+   *   ranksepFunc: undefined, // 每一层节点之间间距
+   *   nodesep: 40, // 节点水平间距(px) 注意：如果有grid，需要保证nodesep为grid的偶数倍
+   *   ranksep: 40, // 每一层节点之间间距 注意：如果有grid，需要保证ranksep为grid的偶数倍
+   *   controlPoints: false, // 是否保留布局连线的控制点
+   *   radial: false, // 是否基于 dagre 进行辐射布局
+   *   focusNode: null, // radial 为 true 时生效，关注的节点
+   * };
+   */
+  layout(option = {}) {
+    const { nodes, edges, gridSize } = this.lf.graphModel
+    // 为了保证生成的节点在girdSize上，需要处理一下。
+    let nodesep = 40
+    let ranksep = 40
+    if (gridSize > 20) {
+      nodesep = gridSize * 2
+      ranksep = gridSize * 2
+    }
+    this.option = {
+      type: 'dagre',
+      rankdir: 'LR',
+      // align: 'UL',
+      // align: 'UR',
+      align: 'DR',
+      nodesep,
+      ranksep,
+      begin: [120, 120],
+      ...option
+    }
+    const layoutInstance = new DagreLayout(this.option)
+    const layoutData = layoutInstance.layout({
+      nodes: nodes.map((node: any) => ({
+        id: node.id,
+        size: {
+          width: node.width,
+          height: node.height
+        },
+        model: node
+      })),
+      edges: edges.map((edge: any) => ({
+        source: edge.sourceNodeId,
+        target: edge.targetNodeId,
+        model: edge
+      }))
+    })
+
+    layoutData.nodes?.forEach((node: any) => {
+      // @ts-ignore: pass node data
+      const { model } = node
+      model.set_position({ x: node.x, y: node.y })
+    })
+    this.lf.fitView()
+  }
+}