From ed396894cbaaee2b6cddfc82850762d68d8aeaea Mon Sep 17 00:00:00 2001
From: "X1627315083@163.com" <1627315083@qq.com>
Date: Wed, 25 Feb 2026 10:14:10 +0800
Subject: [PATCH] =?UTF-8?q?=E7=94=BB=E5=B8=83=E9=A1=B5=E9=9D=A2=E4=BD=BF?=
 =?UTF-8?q?=E7=94=A8vueflow?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../tools/tools.js => utils/treeDiagram.js}   |  0
 .../canvas/components/node/InputNode.vue      | 26 +++++++++
 .../canvas/components/node/secondaryNode.vue  | 33 +++++++++++
 src/views/canvas/index.vue                    | 57 +++++++++++++++++--
 .../agent/components/versionTree/index.vue    |  2 +-
 .../versionTree/tree/view2/index.vue          |  2 +-
 6 files changed, 114 insertions(+), 6 deletions(-)
 rename src/{views/home/agent/components/versionTree/tools/tools.js => utils/treeDiagram.js} (100%)
 create mode 100644 src/views/canvas/components/node/InputNode.vue
 create mode 100644 src/views/canvas/components/node/secondaryNode.vue

diff --git a/src/views/home/agent/components/versionTree/tools/tools.js b/src/utils/treeDiagram.js
similarity index 100%
rename from src/views/home/agent/components/versionTree/tools/tools.js
rename to src/utils/treeDiagram.js
diff --git a/src/views/canvas/components/node/InputNode.vue b/src/views/canvas/components/node/InputNode.vue
new file mode 100644
index 0000000..cce43fc
--- /dev/null
+++ b/src/views/canvas/components/node/InputNode.vue
@@ -0,0 +1,26 @@
+<script lang="ts" setup>
+import { Handle, Position } from '@vue-flow/core'
+import { ref } from 'vue'
+const props = defineProps<{
+	data: {
+		type: Object,
+		default: () => ({
+			id: '',
+		})
+	}
+}>()
+
+</script>
+
+<template>
+	<div class="node start">
+		<Handle type="source" style="top: 40px;" id="Right" :position="Position.Right" />
+		<div class="item">
+			<slot name="content"></slot>
+		</div>
+	</div>
+</template>
+
+<style lang="less" scoped>
+
+</style>
diff --git a/src/views/canvas/components/node/secondaryNode.vue b/src/views/canvas/components/node/secondaryNode.vue
new file mode 100644
index 0000000..ca2d09e
--- /dev/null
+++ b/src/views/canvas/components/node/secondaryNode.vue
@@ -0,0 +1,33 @@
+<script lang="ts" setup>
+import { Handle, Position } from '@vue-flow/core'
+import { ref } from 'vue'
+const props = defineProps<{
+	data: {
+		type: Object,
+		default: () => ({
+			id: '',
+		})
+	},
+}>()
+
+</script>
+<!-- source输入，target输出 -->
+<template>
+	<div class="node">
+		<Handle type="target" style="top: 40px;" id="Left" :position="Position.Left" />
+		<Handle type="source" style="top: 40px;" id="Right" :position="Position.Right" />
+		<!-- <Handle type="source" id="Right" :position="Position.Right" />
+		<Handle type="target" id="Left" :position="Position.Left" /> -->
+		<!-- <div>{{ props.data.id }}</div> -->
+		<div class="item">
+			<slot name="content"></slot>
+		</div>
+	</div>
+</template>
+
+<style lang="less" scoped>
+.node{
+	.item{
+	}
+}
+</style>
diff --git a/src/views/canvas/index.vue b/src/views/canvas/index.vue
index 90479b4..04abcd3 100644
--- a/src/views/canvas/index.vue
+++ b/src/views/canvas/index.vue
@@ -1,6 +1,6 @@
 <template>
 	<div class="canvas">
-		<div class="canvas-main" ref="canvasMain" @mousedown="onMouseDown" @wheel="onMouseWheel">
+		<!-- <div class="canvas-main" ref="canvasMain" @mousedown="onMouseDown" @wheel="onMouseWheel">
 			<div
 				class="canvas-content"
 				:style="{
@@ -18,14 +18,35 @@
 				<card type="add-print" />
 				<card type="edit-material" />
 			</div>
-		</div>
+		</div> -->
+
+		<VueFlow :nodes="nodes" @nodes-initialized="layoutGraph('LR')" :edges="edges" @node-click="" :nodes-draggable="true">
+			<template #node-InputNode="nodeProps">
+				<inputNode v-bind="nodeProps" >
+					<template v-slot:content>{{ nodeProps.type }}
+						<card type="to-real-style" />
+					</template> 	
+				</inputNode>
+			</template>
+			<template #node-SecondaryNode="nodeProps">
+				<secondaryNode v-bind="nodeProps" >
+					<template v-slot:content>
+						<card type="scene-composition" />
+					</template> 
+				</secondaryNode>
+			</template>
+		</VueFlow>
 	</div>
 </template>
 
 <script setup lang="ts">
 	import card from './components/cards/index.vue'
-	import { computed, ref, markRaw, onMounted, reactive } from 'vue'
+	import { useLayout } from '@/utils/treeDiagram'
+	import { VueFlow, useVueFlow } from '@vue-flow/core'
+	import { computed, ref, markRaw, onMounted, reactive, nextTick } from 'vue'
 	import { useGlobalStore } from '@/stores'
+	import secondaryNode from './components/node/secondaryNode.vue'
+	import inputNode from './components/node/InputNode.vue'
 	const globalStore = useGlobalStore()
 	const data = reactive({
 		x: 100,
@@ -59,11 +80,35 @@
 		if (scale > 10) scale = 10
 		data.scale = scale
 	}
+	const position = { x: 0, y: 0 }
+
+	const nodes = ref<Node[]>([
+		  { id: '1', type: 'InputNode', label: 'Node 1', class: 'custom-node start', position },
+		  { id: '2', type: 'SecondaryNode', class: 'custom-node', data: { id: '主 1' }, position },
+	])
+	const edges = ref<Edge[]>([
+		  { id: 'e1-3', source: '1', target: '2', type: 'smoothstep', },
+	])
+	const { fitView } = useVueFlow()
+	const { layout } = useLayout()
+	async function layoutGraph(direction) {
+		setTimeout(() => {
+			nodes.value = layout(nodes.value, edges.value, direction)
+			console.log(nodes.value)
+			nextTick(() => {
+				fitView()
+			})
+		}, 0)
+	}
+
 	onMounted(() => {
 		globalStore.setHomeLeftNavCollapse(true)
 	})
 </script>
-
+<style lang="less">
+@import "@vue-flow/core/dist/style.css";
+@import "@vue-flow/core/dist/theme-default.css";
+</style>
 <style lang="less" scoped>
 	.canvas {
 		// overflow-y: auto;
@@ -90,5 +135,9 @@
 				transform-origin: top left;
 			}
 		}
+		> .vue-flow{
+			width: 100%;
+			height: 100%;
+		}
 	}
 </style>
diff --git a/src/views/home/agent/components/versionTree/index.vue b/src/views/home/agent/components/versionTree/index.vue
index 3e2b9ff..63921cb 100644
--- a/src/views/home/agent/components/versionTree/index.vue
+++ b/src/views/home/agent/components/versionTree/index.vue
@@ -3,7 +3,7 @@ import { ref, onMounted, onUnmounted, reactive, toRefs, watch } from 'vue'
 import Tree from './tree/index.vue'
 import Detail from './detail/index.vue'
 // import { versionsList } from './tools/versionsData'
-import { findAndAddChild, findAndRemoveChild } from './tools/tools'
+import { findAndAddChild, findAndRemoveChild } from '../../../../../utils/treeDiagram'
 import { useProjectStore } from '@/stores'
 import { versionTree, getChatNodeDetail } from '@/api/versitonTree'
 
diff --git a/src/views/home/agent/components/versionTree/tree/view2/index.vue b/src/views/home/agent/components/versionTree/tree/view2/index.vue
index 5971502..1c49695 100644
--- a/src/views/home/agent/components/versionTree/tree/view2/index.vue
+++ b/src/views/home/agent/components/versionTree/tree/view2/index.vue
@@ -5,7 +5,7 @@ import { VueFlow, useVueFlow } from '@vue-flow/core'
 import SpecialEdge from './speciaiEdge.vue'
 import InputNode from './InputNode.vue'//主
 import SecondaryNode from './secondaryNode.vue'//分支
-import { useLayout } from '../../tools/tools'
+import { useLayout } from '@/utils/treeDiagram'
 import dialogVue from "../../components/dialog.vue";
 const props = defineProps({
 	selectItem: {