From 29e5f3c382de4f40437a6500957a0f59967827bf Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Mon, 8 Dec 2025 17:51:16 +0800
Subject: [PATCH] fix: After one click beautification in the loop body, the
 workflow of the loop body becomes smaller and smaller. (#4464)

---
 ui/src/workflow/nodes/loop-body-node/LoopBodyContainer.vue | 6 ++++++
 ui/src/workflow/nodes/loop-body-node/index.vue             | 5 +++--
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/ui/src/workflow/nodes/loop-body-node/LoopBodyContainer.vue b/ui/src/workflow/nodes/loop-body-node/LoopBodyContainer.vue
index 88c0e42bd..8cf81bb8f 100644
--- a/ui/src/workflow/nodes/loop-body-node/LoopBodyContainer.vue
+++ b/ui/src/workflow/nodes/loop-body-node/LoopBodyContainer.vue
@@ -217,6 +217,12 @@ function enlargeHandle() {
     props.nodeModel.setHeight(height.value)
   }
 }
+const zoom = () => {
+  if (enlarge.value) {
+    enlargeHandle()
+  }
+}
+defineExpose({ close, zoom })
 </script>
 <style lang="scss" scoped>
 .workflow-node-container {
diff --git a/ui/src/workflow/nodes/loop-body-node/index.vue b/ui/src/workflow/nodes/loop-body-node/index.vue
index 3feabbaea..d010a5746 100644
--- a/ui/src/workflow/nodes/loop-body-node/index.vue
+++ b/ui/src/workflow/nodes/loop-body-node/index.vue
@@ -1,5 +1,5 @@
 <template>
-  <LoopBodyContainer :nodeModel="nodeModel">
+  <LoopBodyContainer :nodeModel="nodeModel" ref="LoopBodyContainerRef">
     <div ref="containerRef" @wheel.stop style="height: 100%; width: 100%"></div>
   </LoopBodyContainer>
 </template>
@@ -19,7 +19,7 @@ const loop_workflow_mode = inject('loopWorkflowMode') || WorkflowMode.Applicatio
 const nodes: any = import.meta.glob('@/workflow/nodes/**/index.ts', { eager: true })
 const props = defineProps<{ nodeModel: any }>()
 const containerRef = ref()
-
+const LoopBodyContainerRef = ref<InstanceType<typeof LoopBodyContainer>>()
 const validate = () => {
   const workflow = new WorkFlowInstance(lf.value.getGraphData(), WorkflowMode.ApplicationLoop)
   return Promise.all(lf.value.graphModel.nodes.map((element: any) => element?.validate?.()))
@@ -159,6 +159,7 @@ const renderGraphData = (data?: any) => {
 }
 
 const loopLayout = () => {
+  LoopBodyContainerRef.value?.zoom()
   lf.value?.extension?.dagre.layout()
 }
 onMounted(() => {