@hautechai/sdk 0.3.6 → 0.3.7

package/dist/sdk/index.js

@@ -41,6 +41,7 @@ export const createSDK = (options) => {
             : null,
         // TODO: Refactor the API initialization
         operations: () => getAPI(OperationsApi, optionsWithTokenRefresher),
+        allowPollingFallback: options.allowPollingFallback ?? true,
     });
     operationsListener.subscribe();
     return {

package/dist/sdk/listeners/index.d.ts

@@ -7,12 +7,14 @@ export declare class OperationsListener {
     } | null;
     ws: WebSocketClient | null;
     operations: () => Promise<OperationsApi>;
-    constructor({ ws, operations, }: {
+    allowPollingFallback: boolean;
+    constructor({ ws, operations, allowPollingFallback, }: {
         ws: {
             endpoint: string;
             token: () => string | Promise<string>;
         } | null;
         operations: () => Promise<OperationsApi>;
+        allowPollingFallback: boolean;
     });
     operationsStore: Record<string, OperationEntity>;
     getOperation(id: string): Promise<OperationEntity | null>;

package/dist/sdk/listeners/index.js

@@ -2,10 +2,12 @@ import { w3cwebsocket as WebSocketClient } from 'websocket';
 import { HautechError } from '../errors';
 // This is pretty much a dirty solution until we need to rework this part.
 export class OperationsListener {
-    constructor({ ws, operations, }) {
+    constructor({ ws, operations, allowPollingFallback = false, }) {
         this.useWebsocket = null;
         this.ws = null;
         this.operationsStore = {};
+        if (!ws)
+            allowPollingFallback = true;
         if (ws) {
             this.useWebsocket = {
                 endpoint: ws?.endpoint,
@@ -13,10 +15,11 @@ export class OperationsListener {
             };
         }
         this.operations = operations;
+        this.allowPollingFallback = allowPollingFallback;
     }
     async getOperation(id) {
-        const isWsReady = this.ws?.readyState === WebSocket.OPEN;
-        if (!this.operationsStore[id] || !isWsReady) {
+        const fallbackToPolling = this.allowPollingFallback && !(this.ws?.readyState === WebSocket.OPEN);
+        if (!this.operationsStore[id] || fallbackToPolling) {
             const api = await this.operations();
             const operation = await api.operationsControllerGetOperationV1(id);
             if (operation.status == 200)
@@ -34,7 +37,8 @@ export class OperationsListener {
         if (!this.useWebsocket)
             return;
         try {
-            this.ws = new WebSocketClient(this.useWebsocket.endpoint, ['1', await this.useWebsocket.token()]);
+            const token = await this.useWebsocket.token();
+            this.ws = new WebSocketClient(this.useWebsocket.endpoint, ['1', token]);
             this.ws.onopen = () => {
                 this.onOpen();
             };

package/dist/types.d.ts

@@ -2,6 +2,7 @@ export interface SDKOptions {
     authToken: () => string | Promise<string>;
     endpoint?: string;
     useWebsocket?: boolean;
+    allowPollingFallback?: boolean;
 }
 export type ListProps = {
     cursor?: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hautechai/sdk",
-  "version": "0.3.6",
+  "version": "0.3.7",
   "license": "MIT",
   "keywords": [],
   "repository": {