@superatomai/sdk-node 0.0.55 → 0.0.56

package/dist/index.mjs

@@ -6448,6 +6448,32 @@ ${executedToolsText}`);
         fullStreamedText += chunk;
         streamCallback(chunk);
       } : void 0;
+      const withProgressHeartbeat = async (operation, progressMessage, intervalMs = 1e3) => {
+        if (!wrappedStreamCallback) {
+          return operation();
+        }
+        const startTime = Date.now();
+        let dotCount = 0;
+        const maxDots = 3;
+        wrappedStreamCallback(`\u23F3 ${progressMessage}`);
+        const heartbeatInterval = setInterval(() => {
+          const elapsedSeconds = Math.floor((Date.now() - startTime) / 1e3);
+          dotCount = (dotCount + 1) % (maxDots + 1);
+          const dots = ".".repeat(dotCount || 1);
+          if (elapsedSeconds >= 2) {
+            wrappedStreamCallback(`${dots} (${elapsedSeconds}s)`);
+          } else {
+            wrappedStreamCallback(dots);
+          }
+        }, intervalMs);
+        try {
+          const result2 = await operation();
+          return result2;
+        } finally {
+          clearInterval(heartbeatInterval);
+          wrappedStreamCallback("\n\n");
+        }
+      };
       const toolHandler = async (toolName, toolInput) => {
         if (toolName === "execute_query") {
           let sql = toolInput.sql;
@@ -6501,9 +6527,6 @@ Please try rephrasing your question or simplifying your request.
 ${sql}
 \`\`\`${paramsDisplay}
 
-`);
-                wrappedStreamCallback(`\u26A1 **Executing query...**
-
 `);
               } else {
                 wrappedStreamCallback(`
@@ -6521,9 +6544,6 @@ ${sql}
 ${sql}
 \`\`\`${paramsDisplay}
 
-`);
-                wrappedStreamCallback(`\u26A1 **Executing query...**
-
 `);
               }
             }
@@ -6536,13 +6556,19 @@ ${sql}
               throw new Error("Database collection not registered. Please register database.execute collection to execute queries.");
             }
             const queryPayload = Object.keys(params).length > 0 ? { sql: JSON.stringify({ sql, values: params }) } : { sql };
-            const result2 = await collections["database"]["execute"](queryPayload);
+            const result2 = await withProgressHeartbeat(
+              () => collections["database"]["execute"](queryPayload),
+              "Executing database query",
+              800
+              // Send heartbeat every 800ms for responsive feedback
+            );
             const data = result2?.data || result2;
             const rowCount = result2?.count ?? (Array.isArray(data) ? data.length : "N/A");
             logger.info(`[${this.getProviderName()}] Query executed successfully, rows returned: ${rowCount}`);
             logCollector?.info(`Query successful, returned ${rowCount} rows`);
             if (wrappedStreamCallback) {
-              wrappedStreamCallback(`\u2705 **Query executed successfully!**
+              wrappedStreamCallback(`
+\u2705 **Query executed successfully!**
 
 `);
               if (Array.isArray(data) && data.length > 0) {
@@ -6629,7 +6655,12 @@ Please try rephrasing your request or contact support.
 `);
                 }
               }
-              const result2 = await externalTool.fn(toolInput);
+              const result2 = await withProgressHeartbeat(
+                () => externalTool.fn(toolInput),
+                `Running ${externalTool.name}`,
+                800
+                // Send heartbeat every 800ms
+              );
               logger.info(`[${this.getProviderName()}] External tool ${externalTool.name} executed successfully`);
               logCollector?.info(`\u2713 ${externalTool.name} executed successfully`);
               if (!executedToolsList.find((t) => t.id === externalTool.id)) {
@@ -6731,6 +6762,7 @@ ${errorMsg}
         }
       );
       if (wrappedStreamCallback && components && components.length > 0 && category !== "general") {
+        wrappedStreamCallback("\n\n\u{1F4CA} **Generating visualization components...**\n\n");
         wrappedStreamCallback("__TEXT_COMPLETE__COMPONENT_GENERATION_START__");
       }
       let matchedComponents = [];