bizydraft 0.2.76__py3-none-any.whl → 0.2.76.dev20251024063111__py3-none-any.whl

bizydraft/static/js/postEvent.js

@@ -103,7 +103,7 @@ app.registerExtension({
         this.isConnecting = true;
 
         // 先关闭现有连接
-        this.closeSocket();
+        // this.closeSocket();
 
         const url = customUrl || app.api.socket.url;
         console.log('创建WebSocket连接:', url);
@@ -127,6 +127,18 @@ app.registerExtension({
 
             socket.onmessage = function (event) {
                 try {
+                    // 从 WebSocket URL 中提取 taskId
+                    let taskIdFromUrl = null;
+                    try {
+                        const urlParams = new URLSearchParams(socket.url.split('?')[1]);
+                        taskIdFromUrl = urlParams.get('taskId');
+                        if (taskIdFromUrl) {
+                            taskIdFromUrl = parseInt(taskIdFromUrl, 10);
+                        }
+                        console.log('taskIdFromUrl:', taskIdFromUrl);
+                    } catch (e) {
+                        console.warn('无法从 WebSocket URL 中提取 taskId:', e);
+                    }
                     // 处理心跳响应
                     if (event.data === 'pong') {
                         // 标记收到pong响应
@@ -207,21 +219,33 @@ app.registerExtension({
                         return;
                     }
                       const msg = JSON.parse(event.data)
-                        window.parent.postMessage({
-                            type: 'functionResult',
-                            method: 'progress_info_change',
-                            result: msg.progress_info
-                        }, '*');
+                        // 发送进度信息，添加从 URL 中提取的 taskId
+                        if (msg.progress_info) {
+                            const progressData = { ...msg.progress_info };
+                            if (taskIdFromUrl && !progressData.task_id) {
+                                progressData.task_id = taskIdFromUrl;
+                            }
+                            window.parent.postMessage({
+                                type: 'functionResult',
+                                method: 'progress_info_change',
+                                result: progressData
+                            }, '*');
+                        }
 
                       switch (msg.type) {
                         case 'load_start':
                         case 'load_end':
                         case 'prompt_id':
-                          // 发送准备状态信息
+                          // 发送准备状态信息，添加从 URL 中提取的 taskId
+                          const preparingData = { ...msg };
+                          if (taskIdFromUrl) {
+                            preparingData.task_id = taskIdFromUrl;
+                            console.log(`🔗 [WebSocket] 添加 task_id=${taskIdFromUrl} 到 ${msg.type} 消息`);
+                          }
                           window.parent.postMessage({
                             type: 'functionResult',
                             method: 'preparingStatus',
-                            result: msg
+                            result: preparingData
                           }, '*')
                           break
                         case 'status':

{bizydraft-0.2.76.dist-info → bizydraft-0.2.76.dev20251024063111.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bizydraft
-Version: 0.2.76
+Version: 0.2.76.dev20251024063111
 Summary: bizydraft
 Requires-Dist: loguru
 Requires-Dist: aiohttp

{bizydraft-0.2.76.dist-info → bizydraft-0.2.76.dev20251024063111.dist-info}/RECORD

@@ -20,14 +20,14 @@ bizydraft/static/js/hookLoadModel.js,sha256=TWVmfKp45ta-nE6U5rY3Bv9wiaEBp0QMNt2x
 bizydraft/static/js/main.js,sha256=PRe4LdsquEQWrZDnVd4ubpVQuD1eDIedAXFFazKdoKQ,188
 bizydraft/static/js/nodeFocusHandler.js,sha256=24xXbS4Q-GjJdRqf11i-1pBo8MkOJ24F7MHFV44EG6Q,4683
 bizydraft/static/js/nodeParamsFilter.js,sha256=H7lBB0G8HNqoGhOCH1hNXqPU-rPlrFyTxg_f_JgLEMk,4168
-bizydraft/static/js/postEvent.js,sha256=dnstkIU_898KvVka1wP8d1TJDLlAphDYghgpQpkTmLM,39864
+bizydraft/static/js/postEvent.js,sha256=BasbT9vCdTI9mjJwrp2uXwnEc_wraSI7bUk8WIB1-y8,41292
 bizydraft/static/js/socket.js,sha256=VE3fTAgEfM0FZhL526Skt7OCRokOa3mzTCAjAomI_tE,2432
 bizydraft/static/js/tool.js,sha256=VupamUuh7tYiDnBTrL5Z_yLmhJinskhzRXwE3zfsKZM,2901
 bizydraft/static/js/uploadFile.js,sha256=WvglKzHMeOzDhOH3P-fLcPHxCLbKOJpo4DntoRxeJtI,4908
 bizydraft/static/js/workflow_io.js,sha256=FWAjncvWhvy-3nN_legD2fpRwgnIncpRLHU5X016a-U,5236
 bizydraft/static/js/hookLoad/media.js,sha256=BD9RJxTMGO8wagOI9IZNYzKfsOnpyl0JtEUghpURpaI,2620
 bizydraft/static/js/hookLoad/model.js,sha256=C7Hi6BkdtEAgNCwpT71ZRgv9BaYKBsyNu8wt9h5hYvk,10576
-bizydraft-0.2.76.dist-info/METADATA,sha256=N7GBtzZA1p8JWeLncuBsNfjBblcEL7So9Zx9Gh0EH08,162
-bizydraft-0.2.76.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizydraft-0.2.76.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
-bizydraft-0.2.76.dist-info/RECORD,,
+bizydraft-0.2.76.dev20251024063111.dist-info/METADATA,sha256=jGzpy8mdiG0Nph8CxuDJFVKvAnI8VhqZqiFGzb6WLes,180
+bizydraft-0.2.76.dev20251024063111.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizydraft-0.2.76.dev20251024063111.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
+bizydraft-0.2.76.dev20251024063111.dist-info/RECORD,,