@superatomai/sdk-node 0.0.29 → 0.0.30

package/dist/index.d.mts

@@ -1650,6 +1650,10 @@ declare class SuperatomSDK {
     private userManager;
     private dashboardManager;
     private reportManager;
+    private pingInterval;
+    private lastPong;
+    private readonly PING_INTERVAL_MS;
+    private readonly PONG_TIMEOUT_MS;
     constructor(config: SuperatomSDKConfig);
     /**
      * Initialize PromptLoader and load prompts into memory
@@ -1717,6 +1721,19 @@ declare class SuperatomSDK {
      */
     addCollection<TParams = any, TResult = any>(collectionName: string, operation: CollectionOperation | string, handler: CollectionHandler<TParams, TResult>): void;
     private handleReconnect;
+    /**
+     * Start heartbeat to keep WebSocket connection alive
+     * Sends PING every 3 minutes to prevent idle timeout from cloud infrastructure
+     */
+    private startHeartbeat;
+    /**
+     * Stop the heartbeat interval
+     */
+    private stopHeartbeat;
+    /**
+     * Handle PONG response from server
+     */
+    private handlePong;
     private storeComponents;
     /**
      * Set tools for the SDK instance

package/dist/index.d.ts

@@ -1650,6 +1650,10 @@ declare class SuperatomSDK {
     private userManager;
     private dashboardManager;
     private reportManager;
+    private pingInterval;
+    private lastPong;
+    private readonly PING_INTERVAL_MS;
+    private readonly PONG_TIMEOUT_MS;
     constructor(config: SuperatomSDKConfig);
     /**
      * Initialize PromptLoader and load prompts into memory
@@ -1717,6 +1721,19 @@ declare class SuperatomSDK {
      */
     addCollection<TParams = any, TResult = any>(collectionName: string, operation: CollectionOperation | string, handler: CollectionHandler<TParams, TResult>): void;
     private handleReconnect;
+    /**
+     * Start heartbeat to keep WebSocket connection alive
+     * Sends PING every 3 minutes to prevent idle timeout from cloud infrastructure
+     */
+    private startHeartbeat;
+    /**
+     * Stop the heartbeat interval
+     */
+    private stopHeartbeat;
+    /**
+     * Handle PONG response from server
+     */
+    private handlePong;
     private storeComponents;
     /**
      * Set tools for the SDK instance

package/dist/index.js

@@ -9781,6 +9781,7 @@ init_logger();
 var SDK_VERSION = "0.0.8";
 var DEFAULT_WS_URL = "wss://ws.superatom.ai/websocket";
 var SuperatomSDK = class {
+  // 3.5 minutes (PING_INTERVAL + 30s grace)
   constructor(config) {
     this.ws = null;
     this.messageHandlers = /* @__PURE__ */ new Map();
@@ -9791,6 +9792,12 @@ var SuperatomSDK = class {
     this.collections = {};
     this.components = [];
     this.tools = [];
+    // Heartbeat properties for keeping WebSocket connection alive
+    this.pingInterval = null;
+    this.lastPong = Date.now();
+    this.PING_INTERVAL_MS = 18e4;
+    // 3 minutes
+    this.PONG_TIMEOUT_MS = 21e4;
     if (config.logLevel) {
       logger.setLogLevel(config.logLevel);
     }
@@ -9904,6 +9911,7 @@ var SuperatomSDK = class {
           this.connected = true;
           this.reconnectAttempts = 0;
           logger.info("WebSocket connected successfully");
+          this.startHeartbeat();
           resolve();
         });
         this.ws.addEventListener("message", (event) => {
@@ -9932,6 +9940,9 @@ var SuperatomSDK = class {
       const message = IncomingMessageSchema.parse(parsed);
       logger.debug("Received message:", message.type);
       switch (message.type) {
+        case "PONG":
+          this.handlePong();
+          break;
         case "DATA_REQ":
           handleDataRequest(parsed, this.collections, (msg) => this.send(msg)).catch((error) => {
             logger.error("Failed to handle data request:", error);
@@ -10054,6 +10065,7 @@ var SuperatomSDK = class {
    * Disconnect from the WebSocket service
    */
   disconnect() {
+    this.stopHeartbeat();
     if (this.ws) {
       this.ws.close();
       this.ws = null;
@@ -10064,6 +10076,7 @@ var SuperatomSDK = class {
    * Cleanup and disconnect - stops reconnection attempts and closes the connection
    */
   async destroy() {
+    this.stopHeartbeat();
     this.maxReconnectAttempts = 0;
     this.reconnectAttempts = 0;
     this.messageHandlers.clear();
@@ -10105,6 +10118,59 @@ var SuperatomSDK = class {
       logger.error("Max reconnection attempts reached");
     }
   }
+  /**
+   * Start heartbeat to keep WebSocket connection alive
+   * Sends PING every 3 minutes to prevent idle timeout from cloud infrastructure
+   */
+  startHeartbeat() {
+    this.stopHeartbeat();
+    this.lastPong = Date.now();
+    this.pingInterval = setInterval(() => {
+      if (!this.ws || this.ws.readyState !== this.ws.OPEN) {
+        logger.warn("WebSocket not open during heartbeat check, stopping heartbeat");
+        this.stopHeartbeat();
+        return;
+      }
+      const timeSinceLastPong = Date.now() - this.lastPong;
+      if (timeSinceLastPong > this.PONG_TIMEOUT_MS) {
+        logger.warn(`No PONG received for ${timeSinceLastPong}ms, connection may be dead. Reconnecting...`);
+        this.stopHeartbeat();
+        this.ws.close();
+        return;
+      }
+      try {
+        const pingMessage = {
+          type: "PING",
+          from: { type: this.type },
+          payload: { timestamp: Date.now() }
+        };
+        this.ws.send(JSON.stringify(pingMessage));
+        logger.debug("Heartbeat PING sent");
+      } catch (error) {
+        logger.error("Failed to send PING:", error);
+        this.stopHeartbeat();
+        this.ws.close();
+      }
+    }, this.PING_INTERVAL_MS);
+    logger.info(`Heartbeat started (PING every ${this.PING_INTERVAL_MS / 1e3}s)`);
+  }
+  /**
+   * Stop the heartbeat interval
+   */
+  stopHeartbeat() {
+    if (this.pingInterval) {
+      clearInterval(this.pingInterval);
+      this.pingInterval = null;
+      logger.debug("Heartbeat stopped");
+    }
+  }
+  /**
+   * Handle PONG response from server
+   */
+  handlePong() {
+    this.lastPong = Date.now();
+    logger.debug("Heartbeat PONG received");
+  }
   storeComponents(components) {
     this.components = components;
   }