@hautechai/sdk 2.1.0 → 2.2.0

package/dist/index.d.mts

@@ -2,6 +2,7 @@ import * as axios from 'axios';
 import { AxiosResponse } from 'axios';
 import { Pipeline, Methods } from '@hautechai/pipelines';
 import { ChangeMethodSignaturesInObject, WrapRefOrValue, TaskOutput, WrapRef, UnwrapRef } from '@hautechai/pipelines/dist/pipeline';
+import { SocketOptions } from 'socket.io-client';
 
 /**
  * Generated by orval v7.10.0 🍺
@@ -6841,9 +6842,7 @@ interface Config {
     baseUrl: string;
     baseWsUrl: string;
     authToken: () => string | Promise<string>;
-    wsConfig?: {
-        transports?: ('polling' | 'websocket' | 'webtransport')[];
-    };
+    wsConfig?: SocketOptions;
 }
 
 interface BaseWsEvent<T> {
@@ -7304,6 +7303,7 @@ interface SDKOptions {
     authToken: () => string | Promise<string>;
     baseUrl?: string;
     baseWsUrl?: string;
+    wsConfig?: SocketOptions;
 }
 type ApiDefinitionTree<T> = {
     [K in keyof T]: T[K] extends (...args: any[]) => any ? T[K] : T[K] extends object ? ApiDefinitionTree<T[K]> : never;

package/dist/index.d.ts

@@ -2,6 +2,7 @@ import * as axios from 'axios';
 import { AxiosResponse } from 'axios';
 import { Pipeline, Methods } from '@hautechai/pipelines';
 import { ChangeMethodSignaturesInObject, WrapRefOrValue, TaskOutput, WrapRef, UnwrapRef } from '@hautechai/pipelines/dist/pipeline';
+import { SocketOptions } from 'socket.io-client';
 
 /**
  * Generated by orval v7.10.0 🍺
@@ -6841,9 +6842,7 @@ interface Config {
     baseUrl: string;
     baseWsUrl: string;
     authToken: () => string | Promise<string>;
-    wsConfig?: {
-        transports?: ('polling' | 'websocket' | 'webtransport')[];
-    };
+    wsConfig?: SocketOptions;
 }
 
 interface BaseWsEvent<T> {
@@ -7304,6 +7303,7 @@ interface SDKOptions {
     authToken: () => string | Promise<string>;
     baseUrl?: string;
     baseWsUrl?: string;
+    wsConfig?: SocketOptions;
 }
 type ApiDefinitionTree<T> = {
     [K in keyof T]: T[K] extends (...args: any[]) => any ? T[K] : T[K] extends object ? ApiDefinitionTree<T[K]> : never;

package/dist/index.js

@@ -10473,9 +10473,9 @@ var import_jose = require("jose");
 
 // src/config.ts
 var getConfig = (options) => ({
+  ...options,
   baseUrl: options.baseUrl ?? "https://api.hautech.ai",
-  baseWsUrl: options.baseWsUrl ?? options.baseUrl ?? "https://api.hautech.ai",
-  authToken: options.authToken
+  baseWsUrl: options.baseWsUrl ?? options.baseUrl ?? "https://api.hautech.ai"
 });
 
 // src/api-utils.ts
@@ -14219,7 +14219,7 @@ var WsClient = class {
   getSocket() {
     if (!this.socket) {
       this.socket = (0, import_socket.io)(this.config.baseWsUrl, {
-        transports: this.config.wsConfig?.transports || ["websocket"],
+        ...this.config.wsConfig || {},
         auth: (cb) => {
           Promise.resolve(this.config.authToken()).then((token) => {
             cb({