From 6789050c488dc2c41fa02f7b664df9f08dac8f7a Mon Sep 17 00:00:00 2001
From: wujingjing <gersonwu@qq.com>
Date: 星期二, 18 二月 2025 14:23:11 +0800
Subject: [PATCH] var

---
 src/components/vue-flow/MainCanvas.vue |   76 +++++++++++++++++++++++++++++--------
 1 files changed, 59 insertions(+), 17 deletions(-)

diff --git a/src/components/vue-flow/MainCanvas.vue b/src/components/vue-flow/MainCanvas.vue
index 7ad78d8..64db7c1 100644
--- a/src/components/vue-flow/MainCanvas.vue
+++ b/src/components/vue-flow/MainCanvas.vue
@@ -2,17 +2,17 @@
 	<div class="relative h-full w-full" id="main-canvas" @drop="handleOnDrop" @dragover="handleOnDragOver">
 		<VueFlow v-model="elements" :node-types="nodeTypes" :connection-mode="ConnectionMode.Strict">
 			<template #node-start="startNodeProps">
-				<StartNode v-bind="startNodeProps" :isViewMode="isViewMode" />
+				<StartNode ref="nodeRef" v-bind="startNodeProps" :isViewMode="isViewMode" @register="(ref) => registerNodeRef(startNodeProps.id, ref)" />
 			</template>
 			<template #node-condition="conditionNodeProps">
-				<ConditionNode v-bind="conditionNodeProps" :isViewMode="isViewMode" />
+				<ConditionNode ref="nodeRef" v-bind="conditionNodeProps" :isViewMode="isViewMode" @register="(ref) => registerNodeRef(conditionNodeProps.id, ref)" />
 			</template>
 			<template #node-output_msg="outputNodeProps">
-				<OutputNode v-bind="outputNodeProps" :isViewMode="isViewMode" />
+				<OutputNode ref="nodeRef" v-bind="outputNodeProps" :isViewMode="isViewMode" @register="(ref) => registerNodeRef(outputNodeProps.id, ref)" />
 			</template>
 
 			<template #node-end="endNodeProps">
-				<EndNode v-bind="endNodeProps" :isViewMode="isViewMode" />
+				<EndNode ref="nodeRef" v-bind="endNodeProps" :isViewMode="isViewMode" />
 			</template>
 
 			<template #edge-custom="customEdgeProps">
@@ -20,27 +20,35 @@
 			</template>
 
 			<template #node-agent="agentNodeProps">
-				<AgentNode v-bind="agentNodeProps" :agentNames="agentNames" :isViewMode="isViewMode" />
+				<AgentNode ref="nodeRef" v-bind="agentNodeProps" :agentNames="agentNames" :isViewMode="isViewMode" @register="(ref) => registerNodeRef(agentNodeProps.id, ref)" />
 			</template>
 			<template #node-func="funcNodeProps">
-				<FuncNode v-bind="funcNodeProps" :funcNames="funcNames" :isViewMode="isViewMode" />
+				<FuncNode ref="nodeRef" v-bind="funcNodeProps" :funcNames="funcNames" :isViewMode="isViewMode" />
 			</template>
 
 			<template #node-code="codeNodeProps">
-				<CodeNode v-bind="codeNodeProps" :isViewMode="isViewMode" />
+				<CodeNode ref="nodeRef" v-bind="codeNodeProps" :isViewMode="isViewMode" @register="(ref) => registerNodeRef(codeNodeProps.id, ref)" />
+			</template>
+			<template #node-python_code="pythonCodeNodeProps">
+				<PythonCodeNode ref="nodeRef" v-bind="pythonCodeNodeProps" :isViewMode="isViewMode" @register="(ref) => registerNodeRef(pythonCodeNodeProps.id, ref)" />
 			</template>
 			<template #node-text_resource="textResourceNodeProps">
-				<TextResourceNode v-bind="textResourceNodeProps" :isViewMode="isViewMode" />
+				<TextResourceNode ref="nodeRef" v-bind="textResourceNodeProps" :isViewMode="isViewMode" @register="(ref) => registerNodeRef(textResourceNodeProps.id, ref)" />
 			</template>
 
 			<template #node-analysis="analysisNodeProps">
-				<AnalysisNode v-bind="analysisNodeProps" :llmInfoList="llmInfoList" :isViewMode="isViewMode" />
+				<AnalysisNode ref="nodeRef" v-bind="analysisNodeProps" :llmInfoList="llmInfoList" :isViewMode="isViewMode" />
 			</template>
 
 			<template #node-LLM="llmNodeProps">
-				<LLMNode v-bind="llmNodeProps" :llmInfoList="llmInfoList" :isViewMode="isViewMode" />
+				<LLMNode
+					v-bind="llmNodeProps"
+					:llmInfoList="llmInfoList"
+					:isViewMode="isViewMode"
+					@register="(ref) => registerNodeRef(llmNodeProps.id, ref)"
+				/>
 			</template>
-			<Controls :showInteractive="false"/>
+			<Controls :showInteractive="false" />
 			<Background />
 			<MiniMap pannable zoomable />
 		</VueFlow>
@@ -52,7 +60,7 @@
 import { Controls } from '@vue-flow/controls';
 import { MiniMap } from '@vue-flow/minimap';
 
-import type { Dimensions, Elements, FlowOptions, FlowProps } from '@vue-flow/core';
+import type { Dimensions, Elements, FlowProps } from '@vue-flow/core';
 import { ConnectionMode, MarkerType, useVueFlow, VueFlow } from '@vue-flow/core';
 import { nextTick, onMounted, ref, watch } from 'vue';
 import { Test_data } from './testData';
@@ -65,6 +73,7 @@
 import FuncNode from './ui/nodes/FuncNode.vue';
 import LLMNode from './ui/nodes/LLMNode.vue';
 import OutputNode from './ui/nodes/OutputNode.vue';
+import PythonCodeNode from './ui/nodes/PythonCodeNode.vue';
 import StartNode from './ui/nodes/StartNode.vue';
 import TextResourceNode from './ui/nodes/TextResourceNode.vue';
 import { NodeType, nodeTypeMap } from './vueFlowEnum';
@@ -74,7 +83,20 @@
 
 const nodeTypes = {};
 const elements = ref<Elements>();
-const { findNode, nodes, addNodes, addEdges, project, vueFlowRef, onConnect, setNodes, setEdges, setViewport, fitView,setInteractive } = useVueFlow(
+const {
+	findNode,
+	nodes,
+	addNodes,
+	addEdges,
+	project,
+	vueFlowRef,
+	onConnect,
+	setNodes,
+	setEdges,
+	setViewport,
+	fitView,
+	setInteractive,
+} = useVueFlow(
 	Object.keys(props.flowJson).length > 0
 		? props.flowJson
 		: ({
@@ -88,8 +110,6 @@
 					},
 				],
 				edges: [],
-				
-				
 		  } as FlowProps)
 );
 
@@ -111,11 +131,11 @@
 
 	// addEdges());
 });
-const initFlowStatus = () =>{
+const initFlowStatus = () => {
 	fitView();
 
 	setInteractive(!props.isViewMode);
-}
+};
 onMounted(() => {
 	setTimeout(() => {
 		initFlowStatus();
@@ -174,6 +194,28 @@
 		event.dataTransfer.dropEffect = 'move';
 	}
 }
+
+const nodeRefs = ref(new Map());
+
+const registerNodeRef = (nodeId: string, ref: any) => {
+	nodeRefs.value.set(nodeId, ref);
+};
+
+const validateForm = async () => {
+	const validPromises = [];
+	for (const nodeRef of nodeRefs.value.values()) {
+		if (nodeRef?.validateForm) {
+			validPromises.push(nodeRef.validateForm());
+		}
+	}
+
+	const results = await Promise.all(validPromises);
+	return results;
+};
+
+defineExpose({
+	validateForm,
+});
 </script>
 <style lang="scss">
 @import '@vue-flow/core/dist/style.css';

--
Gitblit v1.9.3