diff --git a/ui/src/workflow-canvas/component/NodeMenu.vue b/ui/src/workflow-canvas/component/NodeMenu.vue
index 2545376dab2..617e2f3bff1 100644
--- a/ui/src/workflow-canvas/component/NodeMenu.vue
+++ b/ui/src/workflow-canvas/component/NodeMenu.vue
@@ -29,6 +29,7 @@ const workflowComponentGroups = [
{ icon: aiChatIcon, iconClass: 'bg-primary-gradient', label: 'AI 对话', value: WorkflowNodeType.AiChat },
{ icon: aiChatIcon, iconClass: 'bg-primary-gradient', label: '意图识别', value: WorkflowNodeType.IntentNode },
{ icon: aiChatIcon, iconClass: 'bg-primary-gradient', label: '文本转语音', value: WorkflowNodeType.TextToSpeechNode },
+ { icon: aiChatIcon, iconClass: 'bg-primary-gradient', label: '图片生成', value: WorkflowNodeType.ImageGenerateNode},
{ icon: aiChatIcon, iconClass: 'bg-primary-gradient', label: '问题优化', value: WorkflowNodeType.Question},
],
},
diff --git a/ui/src/workflow-canvas/icons/image-generate-node-icon.vue b/ui/src/workflow-canvas/icons/image-generate-node-icon.vue
new file mode 100644
index 00000000000..7a220814144
--- /dev/null
+++ b/ui/src/workflow-canvas/icons/image-generate-node-icon.vue
@@ -0,0 +1,6 @@
+
+
+
+
+
+
diff --git a/ui/src/workflow-canvas/nodes/image-generate-node/index.ts b/ui/src/workflow-canvas/nodes/image-generate-node/index.ts
new file mode 100644
index 00000000000..74c71abd0c2
--- /dev/null
+++ b/ui/src/workflow-canvas/nodes/image-generate-node/index.ts
@@ -0,0 +1,15 @@
+import ImageGenerateNodeVue from './index.vue'
+import { WorkflowNodeModel, WorkflowNodeView } from '@/workflow-canvas/core/workflow-node'
+import { WorkflowNodeType } from '@/workflow-canvas/types.ts'
+
+class ImageGenerateNodeView extends WorkflowNodeView {
+ constructor(props: ConstructorParameters[0]) {
+ super(props, ImageGenerateNodeVue)
+ }
+}
+
+export default {
+ type: WorkflowNodeType.ImageGenerateNode,
+ model: WorkflowNodeModel,
+ view: ImageGenerateNodeView,
+}
diff --git a/ui/src/workflow-canvas/nodes/image-generate-node/index.vue b/ui/src/workflow-canvas/nodes/image-generate-node/index.vue
new file mode 100644
index 00000000000..60d4bedf093
--- /dev/null
+++ b/ui/src/workflow-canvas/nodes/image-generate-node/index.vue
@@ -0,0 +1,170 @@
+
+
+
+
+ 节点设置
+
+
+
+
+ 图片生成模型
+
+
+
+
+
+
+
+
+
+
+
+
+
+ 提示词(正向)
+
+
+
+
+
+
+
+
+
+
+
+ 提示词(负向)
+
+
+
+
+
+
+
+
+
+
+