diff --git a/src/components/composables/fragment.ts b/src/components/composables/fragment.ts
index 2bbcf8f..f9ceea3 100644
--- a/src/components/composables/fragment.ts
+++ b/src/components/composables/fragment.ts
@@ -25,34 +25,37 @@ export const useStudyFragment = () => {
             });
     };
 
-    const confirmGenerateFragment = async (sessionNum: string) => {
+    const confirmGenerateFragment = async (sessionNum: string): Promise<boolean> => {
         if (!fragmentContent.value.trim()) {
             ElMessage.warning('请输入学习内容！');
-            return;
+            return false;
         }
 
-        ElMessageBox.confirm('确定要生成学习残片吗？', '提示', {
-            confirmButtonText: '确定',
-            cancelButtonText: '取消',
-            type: 'info',
-        })
-            .then(async () => {
-                try {
-                    const res = await createFragments(sessionNum, fragmentContent.value);
-
-                    if (res.code === 200) {
-                        ElMessage.success('学习残片生成成功！');
-                        fragmentsDialogVisible.value = false;
-                    } else {
-                        ElMessage.error(res.message || '生成学习残片失败');
-                    }
-                } catch (error: any) {
-                    ElMessage.error(error.message || '请求失败');
-                }
-            })
-            .catch(() => {
-                ElMessage.info('已取消生成');
+        try {
+            await ElMessageBox.confirm('确定要生成学习残片吗？', '提示', {
+                confirmButtonText: '确定',
+                cancelButtonText: '取消',
+                type: 'info',
             });
+        } catch {
+            ElMessage.info('已取消生成');
+            return false;
+        }
+
+        try {
+            const res = await createFragments(sessionNum, fragmentContent.value);
+            if (res.code === 200) {
+                ElMessage.success('学习残片生成成功！');
+                fragmentsDialogVisible.value = false;
+                return true;
+            } else {
+                ElMessage.error(res.message || '生成学习残片失败');
+                return false;
+            }
+        } catch (error: any) {
+            ElMessage.error(error.message || '请求失败');
+            return false;
+        }
     };
 
     return {