modality-kit 0.9.1 → 0.9.2

package/dist/index.js

@@ -5656,7 +5656,7 @@ class WebSocketClient {
     initialReconnectDelay: 1000,
     maxReconnectDelay: 30000,
     maxReconnectAttempts: 10,
-    callTimeout: 5000,
+    lastReconnectDelay: 300000,
     heartbeatInterval: 30000,
     enableKeepAlive: true,
     handleMessage: (_validMessage, _ws) => {}
@@ -5674,6 +5674,7 @@ class WebSocketClient {
     this.config = { ...this.config, ...config };
     this.url = url;
     this.reconnectDelay = this.config.initialReconnectDelay;
+    this.send = this.send.bind(this);
   }
   isValidWebSocketUrl(url) {
     try {
@@ -5720,7 +5721,7 @@ class WebSocketClient {
         this.reconnectAttempts = 0;
         this.reconnectDelay = this.config.initialReconnectDelay;
         this.attemptReconnect();
-      }, 300000);
+      }, this.config.lastReconnectDelay);
     }
   }
   onOpen(event) {
@@ -5811,6 +5812,9 @@ class WebSocketClient {
         }
       };
       this.ws.onmessage = (event) => {
+        if (this.config.onReceiveMessage) {
+          event = this.config.onReceiveMessage(event);
+        }
         logger3.info("WebSocket message received:", event.data);
         try {
           const message = JSONRPCUtils.deserialize(event.data);

package/dist/types/websocket-client.d.ts

@@ -3,10 +3,11 @@ interface WebSocketConfig {
     maxReconnectAttempts: number;
     initialReconnectDelay: number;
     maxReconnectDelay: number;
-    callTimeout: number;
+    lastReconnectDelay: number;
     heartbeatInterval: number;
     enableKeepAlive: boolean;
     handleMessage: (validMessage: JSONRPCValidationResult, ws: WebSocketClient) => void;
+    onReceiveMessage?: (event: any) => any;
 }
 interface WebSocketInfo {
     url: string;
@@ -25,7 +26,7 @@ export declare class WebSocketClient {
     private reconnectAttempts;
     private reconnectDelay;
     private heartbeatInterval;
-    constructor(url: string, config?: Record<keyof WebSocketConfig, any>);
+    constructor(url: string, config?: Partial<WebSocketConfig>);
     private isValidWebSocketUrl;
     private stopCleanupInterval;
     private stopHeartbeat;

package/package.json

@@ -1,5 +1,5 @@
 {
-  "version": "0.9.1",
+  "version": "0.9.2",
   "name": "modality-kit",
   "repository": {
     "type": "git",