From 3fabf961e500a06a0bb95d15e94e1b05dc78a8da Mon Sep 17 00:00:00 2001
From: xuncha <1658671838@qq.com>
Date: Fri, 20 Mar 2026 14:57:45 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dhtml=E5=AF=BC=E5=87=BA?=
 =?UTF-8?q?=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 electron/services/config.ts        | 10 +++++++++-
 electron/services/exportService.ts |  2 ++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/electron/services/config.ts b/electron/services/config.ts
index 4b8324d..bb6b9f5 100644
--- a/electron/services/config.ts
+++ b/electron/services/config.ts
@@ -688,8 +688,16 @@ export class ConfigService {
     }
   }
 
+  private getUserDataPath(): string {
+    const workerUserDataPath = String(process.env.WEFLOW_USER_DATA_PATH || process.env.WEFLOW_CONFIG_CWD || '').trim()
+    if (workerUserDataPath) {
+      return workerUserDataPath
+    }
+    return app?.getPath?.('userData') || process.cwd()
+  }
+
   getCacheBasePath(): string {
-    return join(app.getPath('userData'), 'cache')
+    return join(this.getUserDataPath(), 'cache')
   }
 
   getAll(): Partial<ConfigSchema> {
diff --git a/electron/services/exportService.ts b/electron/services/exportService.ts
index 6929f59..6649e0c 100644
--- a/electron/services/exportService.ts
+++ b/electron/services/exportService.ts
@@ -6945,6 +6945,7 @@ class ExportService {
       if (collected.rows.length === 0) {
         return { success: false, error: '该会话在指定时间范围内没有消息' }
       }
+      const totalMessages = collected.rows.length
 
       const senderUsernames = new Set<string>()
       let senderScanIndex = 0
@@ -6987,6 +6988,7 @@ class ExportService {
         : []
 
       const mediaCache = new Map<string, MediaExportItem | null>()
+      const mediaDirCache = new Set<string>()
 
       if (mediaMessages.length > 0) {
         await this.preloadMediaLookupCaches(sessionId, mediaMessages, {