@superatomai/sdk-node 0.0.20-mds → 0.0.21-mds

package/dist/index.js

@@ -5133,17 +5133,14 @@ async function withProgressHeartbeat(operation, progressMessage, streamBuffer, i
   const startTime = Date.now();
   await streamDelay(30);
   streamBuffer.write(`\u23F3 ${progressMessage}`);
-  const heartbeatInterval = setInterval(() => {
-    const elapsedSeconds = Math.floor((Date.now() - startTime) / 1e3);
-    if (elapsedSeconds >= 1) {
-      streamBuffer.write(` (${elapsedSeconds}s)`);
-    }
-  }, intervalMs);
   try {
     const result = await operation();
     return result;
   } finally {
-    clearInterval(heartbeatInterval);
+    const totalSeconds = Math.floor((Date.now() - startTime) / 1e3);
+    if (totalSeconds >= 1) {
+      streamBuffer.write(` (${totalSeconds}s)`);
+    }
     streamBuffer.write("\n\n");
   }
 }
@@ -7120,11 +7117,17 @@ ${queryDisplay}
           await streamDelay();
         }
         try {
-          const result = await withProgressHeartbeat(
-            () => this.tool.fn(cappedInput),
-            `Executing query`,
-            this.streamBuffer
-          );
+          if (this.streamBuffer.hasCallback()) {
+            this.streamBuffer.write(`__QUERY_TIMER_START_Executing query__`);
+          }
+          const queryStartTime = Date.now();
+          const result = await this.tool.fn(cappedInput);
+          if (this.streamBuffer.hasCallback()) {
+            const querySeconds = ((Date.now() - queryStartTime) / 1e3).toFixed(1);
+            this.streamBuffer.write(`__QUERY_TIMER_DONE_${querySeconds}__
+
+`);
+          }
           if (result && result.error) {
             const errorMsg = typeof result.error === "string" ? result.error : JSON.stringify(result.error);
             logger.warn(`[SourceAgent:${this.tool.name}] Tool returned error (attempt ${this.attempts}/${this.config.maxRetries}): ${errorMsg}`);