dify

dify/web/app/components/rag-pipeline/hooks/use-pipeline-template.ts

@@ -0,0 +1,30 @@
+import { useTranslation } from 'react-i18next'
+import { generateNewNode } from '@/app/components/workflow/utils'
+import {
+  START_INITIAL_POSITION,
+} from '@/app/components/workflow/constants'
+import type { KnowledgeBaseNodeType } from '@/app/components/workflow/nodes/knowledge-base/types'
+import knowledgeBaseDefault from '@/app/components/workflow/nodes/knowledge-base/default'
+
+export const usePipelineTemplate = () => {
+  const { t } = useTranslation()
+
+  const { newNode: knowledgeBaseNode } = generateNewNode({
+    id: 'knowledgeBase',
+    data: {
+      ...knowledgeBaseDefault.defaultValue as KnowledgeBaseNodeType,
+      type: knowledgeBaseDefault.metaData.type,
+      title: t(`workflow.blocks.${knowledgeBaseDefault.metaData.type}`),
+      selected: true,
+    },
+    position: {
+      x: START_INITIAL_POSITION.x + 500,
+      y: START_INITIAL_POSITION.y,
+    },
+  })
+
+  return {
+    nodes: [knowledgeBaseNode],
+    edges: [],
+  }
+}