From d2da078b40578cf72901442c7a2b878dfc34cae5 Mon Sep 17 00:00:00 2001
From: wujingjing <gersonwu@qq.com>
Date: 星期五, 13 十二月 2024 12:10:00 +0800
Subject: [PATCH] feat(flow): 新增执行功能节点并优化流程画布

---
 src/components/vue-flow/ui/nodes/OutputNode.vue |  115 ++++++++++++++++++++++++++++++++++++++++++++++++++++-----
 1 files changed, 105 insertions(+), 10 deletions(-)

diff --git a/src/components/vue-flow/ui/nodes/OutputNode.vue b/src/components/vue-flow/ui/nodes/OutputNode.vue
index faf8da3..2d45c89 100644
--- a/src/components/vue-flow/ui/nodes/OutputNode.vue
+++ b/src/components/vue-flow/ui/nodes/OutputNode.vue
@@ -2,7 +2,7 @@
 	<div
 		class="min-w-[320px] w-max-[520px] border-2 rounded-lg border-solid border-gray-100 bg-white p-3 shadow-md relative hover:border-blue-500 group"
 	>
-		<Handle :id="targetHandleId" type="source" :position="Position.Left" />
+		<Handle :id="targetHandleId" type="target" :position="Position.Left" />
 		<div
 			class="group-hover:visible invisible flex absolute divide-y-[1.5px] divide-solid divide-gray-100 rounded-lg right-0 -top-0.5 translate-y-[-100%]"
 			style="box-shadow: 0 0 15px #dbdee6"
@@ -27,10 +27,16 @@
 		<div class="flex flex-col gap-y-2">
 			<div class="flex justify-between flex-0">
 				<div class="flex items-center gap-x-2">
-					<img src="/@/components/vue-flow/ui/assets/images/icon_Start.png" class="h-4 w-4" alt="Start icon" />
+					<YWIcon
+						:name="VueFlowConfig.nodeStyleMap.get(NodeType.Output).icon"
+						:fontSize="VueFlowConfig.nodeStyleMap.get(NodeType.Output).fontSize"
+						:color="VueFlowConfig.nodeStyleMap.get(NodeType.Output).color"
+						class="rounded-lg p-1.5"
+						:class="VueFlowConfig.nodeStyleMap.get(NodeType.Output).class"
+					/>
 					<div class="flex flex-col gap-y-1">
 						<p v-if="!titleIsEdit" class="text-xl font-bold text-gray-500" @click="titleIsEdit = true">{{ data.title }}</p>
-						<el-input v-else v-model="data.title" @blur="() => (titleIsEdit = false)"></el-input>
+						<el-input v-elInputFocus="false" v-else v-model="data.title" @blur="() => (titleIsEdit = false)"></el-input>
 					</div>
 				</div>
 			</div>
@@ -52,17 +58,56 @@
 				<div class="flex-column gap-y-1.5 w-full">
 					<span class="self-start text-gray-500">{{ VueFlowHelper.getParams(outputParams, 'output_result').label }}</span>
 
-					<el-radio-group v-model="VueFlowHelper.getParams(outputParams, 'output_result').value.type">
+					<el-radio-group v-model="VueFlowHelper.getParams(outputParams, 'output_result').value.type" @change="interactionTypeChange">
 						<el-radio v-for="item in Object.keys(interactionTypeMap)" :key="item" :value="item">{{
 							interactionTypeMap[item]
 						}}</el-radio>
 					</el-radio-group>
+					<el-input
+						v-model="VueFlowHelper.getParams(outputParams, 'output_result').value.value"
+						type="textarea"
+						:rows="4"
+						v-if="VueFlowHelper.getParams(outputParams, 'output_result').value.type === InteractionType.Input"
+					>
+					</el-input>
 
-					
+					<div
+						class="self-start w-full flex-col flex items-start gap-2"
+						v-else-if="VueFlowHelper.getParams(outputParams, 'output_result').value.type === InteractionType.Select"
+					>
+						<div class="flex-column gap-2">
+							<div
+								class="text-gray-400 cursor-not-allowed border border-solid rounded-lg border-gray-300 py-3 pl-3 pr-2 items-center justify-between flex group/option relative mr-6"
+								:key="item.id"
+								v-for="(item, index) in VueFlowHelper.getParams(outputParams, 'output_result').options"
+							>
+								<span>{{ item.label }}</span>
+								<span
+									@click="delOption(index)"
+									class="cursor-pointer group-hover/option:visible invisible ywifont ywicon-shanchu text-red-500"
+								></span>
+								<Handle class="absolute !-right-4" :id="item.id" type="source" :position="Position.Right" />
+							</div>
+						</div>
+
+						<div v-if="isEditStatus" class="flex-items-center w-full">
+							<el-input class="flex-auto" v-model="tmpEditValue"></el-input>
+							<el-button class="flex-0 ml-2" type="success" @click="confirmClick">纭畾 </el-button>
+						</div>
+						<el-button v-else type="primary" @click="addOptionClick">娣诲姞閫夐」</el-button>
+					</div>
 				</div>
 			</div>
 		</div>
-		<Handle :id="sourceHandleId" type="source" :position="Position.Right" />
+		<Handle
+			v-if="
+				VueFlowHelper.getParams(outputParams, 'output_result').value.type === InteractionType.Input ||
+				VueFlowHelper.getParams(outputParams, 'output_result').value.type === InteractionType.None
+			"
+			:id="sourceHandleId"
+			type="source"
+			:position="Position.Right"
+		/>
 	</div>
 
 	<!-- <div>
@@ -77,11 +122,13 @@
 import { ref } from 'vue';
 
 import type { NodeProps } from '@vue-flow/core';
+import { nextTick } from 'vue';
 import { VueFlowHelper } from '../../VueFlowHelper';
+import { InteractionType, interactionTypeMap } from '../../vueFlowEnum';
 import { LLMNodeData, LLMNodeEvents } from './index';
 import { deepClone } from '/@/utils/other';
-import { interactionTypeMap } from '../../vueFlowEnum';
-defineProps<NodeProps<LLMNodeData, LLMNodeEvents>>();
+import { VueFlowConfig } from '../VueFlowConfig';
+import { NodeType } from '../../vueFlowEnum';defineProps<NodeProps<LLMNodeData, LLMNodeEvents>>();
 
 const node = useNode();
 const sourceHandleId = ref(VueFlowHelper.getHandleId(node.node, 'source'));
@@ -89,12 +136,11 @@
 
 const data = ref(node.node.data);
 
-
 const titleIsEdit = ref(false);
 
 const outputParams = ref(VueFlowHelper.getGroupParam(data.value, 0));
 
-const { removeNodes, nodes, addNodes } = useVueFlow();
+const { removeNodes, nodes, addNodes, removeEdges } = useVueFlow();
 
 function handleClickDuplicateBtn() {
 	const { type, position, data } = node.node;
@@ -112,4 +158,53 @@
 const clickDeleteBtn = () => {
 	removeNodes(node.id);
 };
+const addOptionClick = () => {
+	isEditStatus.value = true;
+};
+
+const confirmClick = () => {
+	isEditStatus.value = false;
+
+	addOption(tmpEditValue.value);
+	nextTick(() => {
+		tmpEditValue.value = '';
+	});
+};
+const addOption = (val) => {
+	const options = VueFlowHelper.getParams(outputParams.value, 'output_result').options;
+	options.push({
+		id: VueFlowHelper.genId(),
+		label: val,
+		value: '',
+	});
+};
+
+const removeRelativeHandleEdge = (handleId: string) => {
+	const edges = node.connectedEdges.value.filter((item) => item.sourceHandle === handleId || item.targetHandle === handleId);
+	removeEdges(edges);
+};
+
+const delOption = (index) => {
+	const options = VueFlowHelper.getParams(outputParams.value, 'output_result').options;
+	const option = options[index];
+
+	options.splice(index);
+	removeRelativeHandleEdge(option.id);
+};
+
+let preInteractionType = VueFlowHelper.getParams(outputParams.value, 'output_result').value.type;
+const interactionTypeChange = (type) => {
+	removeRelativeHandleEdge(sourceHandleId.value);
+	if (preInteractionType === InteractionType.Select) {
+		const options = VueFlowHelper.getParams(outputParams.value, 'output_result').options;
+		for (const item of options) {
+			removeRelativeHandleEdge(item.id);
+		}
+	}
+	preInteractionType = type;
+
+};
+
+const isEditStatus = ref(false);
+const tmpEditValue = ref('');
 </script>

--
Gitblit v1.9.3