@runware/sdk-js 1.1.15 → 1.1.16

package/dist/index.d.mts

@@ -20,6 +20,7 @@ declare enum ETaskType {
 type RunwareBaseType = {
     apiKey: string;
     url?: string;
+    shouldReconnect?: boolean;
 };
 type IOutputType = "base64Data" | "dataURI" | "URL";
 type IOutputFormat = "JPG" | "PNG" | "WEBP";
@@ -253,7 +254,8 @@ declare class RunwareBase {
     _connectionSessionUUID: string | undefined;
     _invalidAPIkey: string | undefined;
     _sdkType: SdkType;
-    constructor({ apiKey, url }: RunwareBaseType);
+    _shouldReconnect: boolean;
+    constructor({ apiKey, url, shouldReconnect, }: RunwareBaseType);
     protected isWebsocketReadyState: () => boolean;
     protected addListener({ lis, groupKey, taskUUID, }: {
         lis: (v: any) => any;
@@ -292,10 +294,11 @@ declare class RunwareServer extends RunwareBase {
     _reconnectingIntervalId: null | any;
     _pingTimeout: any;
     _pongListener: any;
-    constructor({ apiKey, url }: RunwareBaseType);
+    constructor({ apiKey, url, shouldReconnect }: RunwareBaseType);
     protected connect(): Promise<void>;
     protected send: (msg: Object) => void;
     protected handleClose(): void;
+    protected resetConnection: () => void;
     protected heartBeat(): void;
 }
 

package/dist/index.d.ts

@@ -20,6 +20,7 @@ declare enum ETaskType {
 type RunwareBaseType = {
     apiKey: string;
     url?: string;
+    shouldReconnect?: boolean;
 };
 type IOutputType = "base64Data" | "dataURI" | "URL";
 type IOutputFormat = "JPG" | "PNG" | "WEBP";
@@ -253,7 +254,8 @@ declare class RunwareBase {
     _connectionSessionUUID: string | undefined;
     _invalidAPIkey: string | undefined;
     _sdkType: SdkType;
-    constructor({ apiKey, url }: RunwareBaseType);
+    _shouldReconnect: boolean;
+    constructor({ apiKey, url, shouldReconnect, }: RunwareBaseType);
     protected isWebsocketReadyState: () => boolean;
     protected addListener({ lis, groupKey, taskUUID, }: {
         lis: (v: any) => any;
@@ -292,10 +294,11 @@ declare class RunwareServer extends RunwareBase {
     _reconnectingIntervalId: null | any;
     _pingTimeout: any;
     _pongListener: any;
-    constructor({ apiKey, url }: RunwareBaseType);
+    constructor({ apiKey, url, shouldReconnect }: RunwareBaseType);
     protected connect(): Promise<void>;
     protected send: (msg: Object) => void;
     protected handleClose(): void;
+    protected resetConnection: () => void;
     protected heartBeat(): void;
 }
 

package/dist/index.js

@@ -436,7 +436,11 @@ var asyncRetry = async (apiCall, options = {}) => {
 
 // Runware/Runware-base.ts
 var RunwareBase = class {
-  constructor({ apiKey, url = BASE_RUNWARE_URLS.PRODUCTION }) {
+  constructor({
+    apiKey,
+    url = BASE_RUNWARE_URLS.PRODUCTION,
+    shouldReconnect = true
+  }) {
     this._listeners = [];
     // _globalMessages: any[] = [];
     this._globalMessages = {};
@@ -748,6 +752,7 @@ var RunwareBase = class {
     this._apiKey = apiKey;
     this._url = url;
     this._sdkType = "CLIENT" /* CLIENT */;
+    this._shouldReconnect = shouldReconnect;
   }
   // protected addListener({
   //   lis,
@@ -1186,14 +1191,27 @@ var RunwareClient = class extends RunwareBase {
 // Runware/Runware-server.ts
 var import_ws = __toESM(require("ws"));
 var RunwareServer = class extends RunwareBase {
-  constructor({ apiKey, url }) {
-    super({ apiKey, url });
+  constructor({ apiKey, url, shouldReconnect }) {
+    super({ apiKey, url, shouldReconnect });
     this._instantiated = false;
     this._listeners = [];
     this._reconnectingIntervalId = null;
     this.send = (msg) => {
       this._ws.send(JSON.stringify([msg]));
     };
+    this.resetConnection = () => {
+      if (this._ws) {
+        this._listeners.forEach((list) => {
+          var _a;
+          (_a = list == null ? void 0 : list.destroy) == null ? void 0 : _a.call(list);
+        });
+        this._ws.removeAllListeners();
+        this._ws.terminate();
+        this._ws.close();
+        this._ws = null;
+        this._listeners = [];
+      }
+    };
     this._sdkType = "SERVER" /* SERVER */;
     if (apiKey) {
       this.connect();
@@ -1227,6 +1245,7 @@ var RunwareServer = class extends RunwareBase {
   async connect() {
     if (!this._url)
       return;
+    this.resetConnection();
     this._ws = new import_ws.default(this._url, {
       perMessageDeflate: false
     });
@@ -1287,6 +1306,11 @@ var RunwareServer = class extends RunwareBase {
     if (this._reconnectingIntervalId) {
       clearInterval(this._reconnectingIntervalId);
     }
+    console.log("should reconnect", this._shouldReconnect);
+    if (this._shouldReconnect) {
+      console.log("reconnecting");
+      setTimeout(() => this.connect(), 1e3);
+    }
   }
   heartBeat() {
     clearTimeout(this._pingTimeout);

package/dist/index.mjs

@@ -414,7 +414,11 @@ var asyncRetry = async (apiCall, options = {}) => {
 
 // Runware/Runware-base.ts
 var RunwareBase = class {
-  constructor({ apiKey, url = BASE_RUNWARE_URLS.PRODUCTION }) {
+  constructor({
+    apiKey,
+    url = BASE_RUNWARE_URLS.PRODUCTION,
+    shouldReconnect = true
+  }) {
     this._listeners = [];
     // _globalMessages: any[] = [];
     this._globalMessages = {};
@@ -726,6 +730,7 @@ var RunwareBase = class {
     this._apiKey = apiKey;
     this._url = url;
     this._sdkType = "CLIENT" /* CLIENT */;
+    this._shouldReconnect = shouldReconnect;
   }
   // protected addListener({
   //   lis,
@@ -1164,14 +1169,27 @@ var RunwareClient = class extends RunwareBase {
 // Runware/Runware-server.ts
 import WebSocket2 from "ws";
 var RunwareServer = class extends RunwareBase {
-  constructor({ apiKey, url }) {
-    super({ apiKey, url });
+  constructor({ apiKey, url, shouldReconnect }) {
+    super({ apiKey, url, shouldReconnect });
     this._instantiated = false;
     this._listeners = [];
     this._reconnectingIntervalId = null;
     this.send = (msg) => {
       this._ws.send(JSON.stringify([msg]));
     };
+    this.resetConnection = () => {
+      if (this._ws) {
+        this._listeners.forEach((list) => {
+          var _a;
+          (_a = list == null ? void 0 : list.destroy) == null ? void 0 : _a.call(list);
+        });
+        this._ws.removeAllListeners();
+        this._ws.terminate();
+        this._ws.close();
+        this._ws = null;
+        this._listeners = [];
+      }
+    };
     this._sdkType = "SERVER" /* SERVER */;
     if (apiKey) {
       this.connect();
@@ -1205,6 +1223,7 @@ var RunwareServer = class extends RunwareBase {
   async connect() {
     if (!this._url)
       return;
+    this.resetConnection();
     this._ws = new WebSocket2(this._url, {
       perMessageDeflate: false
     });
@@ -1265,6 +1284,11 @@ var RunwareServer = class extends RunwareBase {
     if (this._reconnectingIntervalId) {
       clearInterval(this._reconnectingIntervalId);
     }
+    console.log("should reconnect", this._shouldReconnect);
+    if (this._shouldReconnect) {
+      console.log("reconnecting");
+      setTimeout(() => this.connect(), 1e3);
+    }
   }
   heartBeat() {
     clearTimeout(this._pingTimeout);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@runware/sdk-js",
-  "version": "1.1.15",
+  "version": "1.1.16",
   "description": "The SDK is used to run image inference with the Runware API, powered by the RunWare inference platform. It can be used to generate imaged with text-to-image and image-to-image. It also allows the use of an existing gallery of models or selecting any model or LoRA from the CivitAI gallery. The API also supports upscaling, background removal, inpainting and outpainting, and a series of other ControlNet models.",
   "main": "dist/index.js",
   "module": "dist/index.mjs",

package/readme.md

@@ -360,6 +360,12 @@ return interface IControlNetImage {
 
 ## Changelog
 
+### - v1.1.16
+
+**Added or Changed**
+
+- Add reconnection to server on close
+
 ### - v1.1.14/v1.1.15
 
 **Added or Changed**