diff --git a/src/views/ai/image/ImageTask.vue b/src/views/ai/image/ImageTask.vue
index 36eb4595..9ddd7461 100644
--- a/src/views/ai/image/ImageTask.vue
+++ b/src/views/ai/image/ImageTask.vue
@@ -130,6 +130,10 @@ const handlerImageBtnClick = async (type: string, imageDetail: ImageVO) => {
message.success('删除成功!')
} else if (type === 'download') {
await downloadImage(imageDetail.picUrl)
+ } else if (type === 'regeneration') {
+ // Midjourney 平台
+ console.log('regeneration', imageDetail.id)
+ await emits('onRegeneration', imageDetail)
}
}
@@ -175,6 +179,9 @@ const handlerPageChange = async (page) => {
/** 暴露组件方法 */
defineExpose({ getImageList })
+// emits
+const emits = defineEmits(['onRegeneration'])
+
/** 组件挂在的时候 */
onMounted(async () => {
// 获取 image 列表
diff --git a/src/views/ai/image/ImageTaskCard.vue b/src/views/ai/image/ImageTaskCard.vue
index 62ba222e..4f8525f6 100644
--- a/src/views/ai/image/ImageTaskCard.vue
+++ b/src/views/ai/image/ImageTaskCard.vue
@@ -25,6 +25,12 @@
:icon="Download"
@click="handlerBtnClick('download', imageDetail)"
/>
+