@hautechai/sdk 2.30.2 → 2.31.0

package/dist/index.mjs

@@ -10298,11 +10298,15 @@ import { decodeJwt } from "jose";
 var getConfig = (options) => ({
   ...options,
   baseUrl: options.baseUrl ?? "https://api.hautech.ai",
-  baseWsUrl: options.baseWsUrl ?? options.baseUrl ?? "https://api.hautech.ai"
+  baseWsUrl: options.baseWsUrl ?? options.baseUrl ?? "https://api.hautech.ai",
+  onRequestError: options.onRequestError,
+  invalidateAuthToken: () => void 0
 });
 var getDirectoryConfig = (options) => ({
   baseUrl: options.baseUrl ?? "https://api-directory.hautech.ai",
-  authToken: options.authToken
+  authToken: options.authToken,
+  onRequestError: options.onRequestError,
+  invalidateAuthToken: () => void 0
 });
 
 // src/api-utils.ts
@@ -10315,49 +10319,125 @@ var isWrappedFn = (fn) => {
 var isCustomWrappedFn = (fn) => {
   return typeof fn === "function" && fn.__customMethodWrapper === true;
 };
+var retryLocks = /* @__PURE__ */ new WeakMap();
+var delay = (ms) => ms > 0 ? new Promise((resolve) => setTimeout(resolve, ms)) : Promise.resolve();
+var cloneRequestConfig = (request) => ({
+  ...request,
+  headers: request.headers ? { ...request.headers } : request.headers
+});
+var toResponseInfo = (response) => {
+  if (!response) return void 0;
+  return {
+    status: response.status,
+    headers: response.headers,
+    data: response.data
+  };
+};
+var enhanceAxiosError = (error) => {
+  let responseData;
+  try {
+    const messageFromBody = error.response?.data?.message;
+    if (typeof messageFromBody === "string" && messageFromBody.length > 0) {
+      responseData = messageFromBody;
+    } else if (typeof error.response?.data === "string") {
+      responseData = error.response.data;
+    } else if (error.response?.data !== void 0) {
+      responseData = JSON.stringify(error.response.data);
+    } else {
+      responseData = "[no response data]";
+    }
+  } catch {
+    responseData = "[unserializable data]";
+  }
+  error.message = `Request error: ${error.message || "Unknown error"}.
+${responseData}`;
+};
+var runOnRequestError = async (config, info, context) => {
+  if (!config.onRequestError) {
+    return void 0;
+  }
+  const status = info.error.response?.status;
+  if (status === 401) {
+    const existing = retryLocks.get(config);
+    if (existing) {
+      return existing;
+    }
+    const pending = (async () => config.onRequestError(info, context))();
+    retryLocks.set(config, pending);
+    try {
+      return await pending;
+    } finally {
+      retryLocks.delete(config);
+    }
+  }
+  return await config.onRequestError(info, context);
+};
+var MAX_ATTEMPTS = 2;
 var createApiCallWrapper = (getBaseURL) => {
   return (fn, config) => {
     const wrapped = async function(...args) {
-      const token = await config.authToken();
-      const baseOptions = {
-        baseURL: getBaseURL(config),
-        headers: {
-          Authorization: `Bearer ${token}`
-        }
-      };
-      const finalArgs = [...args];
-      const lastParamIndex = fn.length - 1;
-      if (looksLikeAxiosRequestOptions(args[lastParamIndex])) {
-        finalArgs[lastParamIndex] = {
-          ...args[lastParamIndex],
-          ...baseOptions,
+      const originalArgs = [...args];
+      for (let attempt = 1; attempt <= MAX_ATTEMPTS; attempt++) {
+        const token = await config.authToken();
+        const baseOptions = {
+          baseURL: getBaseURL(config),
           headers: {
-            ...args[lastParamIndex]?.headers ?? {},
             Authorization: `Bearer ${token}`
           }
         };
-      } else {
-        while (finalArgs.length < fn.length - 1) {
-          finalArgs.push(void 0);
+        const finalArgs = [...originalArgs];
+        const lastParamIndex = fn.length - 1;
+        if (looksLikeAxiosRequestOptions(originalArgs[lastParamIndex])) {
+          finalArgs[lastParamIndex] = {
+            ...originalArgs[lastParamIndex],
+            ...baseOptions,
+            headers: {
+              ...originalArgs[lastParamIndex]?.headers ?? {},
+              Authorization: `Bearer ${token}`
+            }
+          };
+        } else {
+          while (finalArgs.length < fn.length - 1) {
+            finalArgs.push(void 0);
+          }
+          finalArgs.push(baseOptions);
         }
-        finalArgs.push(baseOptions);
-      }
-      try {
-        const res = await fn.call(this, ...finalArgs);
-        return typeof res === "object" && "headers" in res ? res.data : res;
-      } catch (err) {
-        if (axios.isAxiosError(err)) {
-          let responseData;
-          try {
-            responseData = err.response?.data?.message ?? (typeof err.response?.data === "string" ? err.response.data : JSON.stringify(err.response?.data));
-          } catch {
-            responseData = "[unserializable data]";
+        const requestConfig = finalArgs[finalArgs.length - 1] ?? {};
+        try {
+          const res = await fn.call(this, ...finalArgs);
+          return typeof res === "object" && "headers" in res ? res.data : res;
+        } catch (err) {
+          if (!axios.isAxiosError(err)) {
+            throw err;
           }
-          err.message = `Request error: ${err.message || "Unknown error"}.
-${responseData}`;
+          enhanceAxiosError(err);
+          const context = {
+            attempt,
+            request: cloneRequestConfig(requestConfig)
+          };
+          const handlerResult = await runOnRequestError(
+            config,
+            {
+              error: err,
+              response: toResponseInfo(err.response)
+            },
+            context
+          );
+          if (handlerResult?.invalidateToken) {
+            config.invalidateAuthToken();
+          }
+          const hasAttemptsRemaining = attempt < MAX_ATTEMPTS;
+          if (handlerResult?.retry && hasAttemptsRemaining) {
+            const backoffMs = handlerResult.backoffMs ?? 0;
+            if (backoffMs > 0) {
+              await delay(backoffMs);
+            }
+            continue;
+          }
+          throw err;
         }
-        throw err;
       }
+      throw new Error("Request error handling exceeded maximum attempts");
     };
     wrapped.__isWrapped = true;
     return wrapped;
@@ -11870,7 +11950,7 @@ var getOperations = () => {
 };
 
 // src/sdk/api-definitions/operations.ts
-var waitOperation = wrapCustomMethod(async function(operation, timeoutMs = 6e4, delay = 3e3) {
+var waitOperation = wrapCustomMethod(async function(operation, timeoutMs = 6e4, delay2 = 3e3) {
   const deadline = Date.now() + timeoutMs;
   const sdk = this;
   const sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
@@ -11883,7 +11963,7 @@ var waitOperation = wrapCustomMethod(async function(operation, timeoutMs = 6e4,
   while (Date.now() < deadline) {
     const result = await poll(operation.id);
     if (result) return result;
-    await sleep(delay);
+    await sleep(delay2);
   }
   throw new Error("Operation timed out");
 });
@@ -16667,7 +16747,7 @@ var usePipelinesApi = () => {
       const sdk = this;
       return builder(buildPipeline(sdk));
     }),
-    wait: wrapCustomMethod(async function(pipeline, timeoutMs = 6e4, delay = 3e3) {
+    wait: wrapCustomMethod(async function(pipeline, timeoutMs = 6e4, delay2 = 3e3) {
       const sdk = this;
       const deadline = Date.now() + timeoutMs;
       const sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
@@ -16682,7 +16762,7 @@ var usePipelinesApi = () => {
       while (Date.now() < deadline) {
         const polled = await poll(pipeline.id);
         if (polled) return polled;
-        await sleep(delay);
+        await sleep(delay2);
       }
       throw new Error("Operation timed out");
     })
@@ -17007,6 +17087,12 @@ var WsClient = class {
       this.getSocket().off("server_error");
     }
   }
+  disconnect() {
+    if (this.socket) {
+      this.socket.disconnect();
+      this.socket = void 0;
+    }
+  }
 };
 
 // src/sdk/api.ts
@@ -17041,6 +17127,9 @@ var getWsClientDefinitions = (config) => ({
 var createSDK = (options) => {
   let token = void 0;
   const config = getConfig(options);
+  config.invalidateAuthToken = () => {
+    token = void 0;
+  };
   const authToken = config.authToken;
   const getAuthToken = async () => {
     if (token) {
@@ -17062,6 +17151,9 @@ var createSDK = (options) => {
 var createDirectorySDK = (options) => {
   let token = void 0;
   const config = getDirectoryConfig(options);
+  config.invalidateAuthToken = () => {
+    token = void 0;
+  };
   const authToken = config.authToken;
   const getAuthToken = async () => {
     if (token) {