@albatrossai/albatross-sdk 1.1.41 → 1.1.42

package/dist/index.cjs

@@ -7,12 +7,8 @@ var zod = require('zod');
 var MAX_RETRIES = 3;
 var BASE_DELAY_MS = 1e3;
 var REQUEST_TIMEOUT_MS = 3e4;
-var isRetryableError = (status) => {
-  return status >= 500 && status <= 599;
-};
-var getRetryDelay = (attempt) => {
-  return Math.pow(2, attempt) * BASE_DELAY_MS;
-};
+var isRetryableError = (status) => status >= 500 && status <= 599;
+var getRetryDelay = (attempt) => 2 ** attempt * BASE_DELAY_MS;
 var makeRequest = async ({
   url,
   method = "GET",
@@ -21,7 +17,7 @@ var makeRequest = async ({
 }) => {
   const body = data ? JSON.stringify(data) : void 0;
   let lastError = null;
-  for (let attempt = 0; attempt <= MAX_RETRIES; attempt++) {
+  for (let attempt = 0; attempt <= MAX_RETRIES; attempt += 1) {
     const controller = new AbortController();
     const timeoutId = setTimeout(
       () => controller.abort(),
@@ -51,7 +47,8 @@ var makeRequest = async ({
           continue;
         }
         throw new Error(
-          `Request failed after ${MAX_RETRIES} retries: ${response.status} ${response.statusText} ${errorText}`
+          `Request failed after ${MAX_RETRIES} retries: ${response.status} ${response.statusText} ${errorText}`,
+          { cause: error }
         );
       }
       return response.json();
@@ -70,7 +67,8 @@ var makeRequest = async ({
             continue;
           }
           throw new Error(
-            `Request timeout after ${REQUEST_TIMEOUT_MS}ms (${MAX_RETRIES} retries)`
+            `Request timeout after ${REQUEST_TIMEOUT_MS}ms (${MAX_RETRIES} retries)`,
+            { cause: error }
           );
         }
         if (error.message.includes("fetch") || error.message.includes("network") || error.message.includes("Failed to fetch")) {
@@ -82,7 +80,8 @@ var makeRequest = async ({
             continue;
           }
           throw new Error(
-            `Network error after ${MAX_RETRIES} retries: ${error.message}`
+            `Network error after ${MAX_RETRIES} retries: ${error.message}`,
+            { cause: error }
           );
         }
       }
@@ -91,29 +90,29 @@ var makeRequest = async ({
   }
   throw lastError || new Error("Unexpected error in makeRequest");
 };
-function flattenNested(obj) {
+var flattenNested = (obj) => {
   const result = {};
-  Object.entries(obj).forEach(([key, value]) => {
+  for (const [key, value] of Object.entries(obj)) {
     if (typeof value === "object" && !Array.isArray(value) && value !== null) {
-      Object.entries(value).forEach(([nestedKey, nestedValue]) => {
+      for (const [nestedKey, nestedValue] of Object.entries(value)) {
         if (Array.isArray(nestedValue)) {
           result[`${key}_${nestedKey}`] = nestedValue.join(",");
         } else if (typeof nestedValue === "object" && nestedValue !== null) {
-          Object.entries(nestedValue).forEach(([subKey, subValue]) => {
+          for (const [subKey, subValue] of Object.entries(nestedValue)) {
             result[`${nestedKey}_${subKey}`] = subValue;
-          });
+          }
         } else {
           result[`${key}_${nestedKey}`] = nestedValue;
         }
-      });
+      }
     } else if (Array.isArray(value)) {
       result[key] = value.join(",");
     } else {
       result[key] = value;
     }
-  });
+  }
   return result;
-}
+};
 var preFormatRow = (rowHeaders, row) => {
   const flattenedRow = flattenNested(row);
   return rowHeaders.map((header) => {
@@ -131,7 +130,7 @@ var bodyToCSV = (data) => {
   if (data.length === 0) {
     throw new Error("data is empty");
   }
-  const row0 = data[0];
+  const [row0] = data;
   if (!row0) {
     throw new Error("data is empty");
   }
@@ -318,7 +317,12 @@ var FeedbackParamsSchema = zod.z.object({
 
 // src/client.ts
 var hostDefault = "https://app.usealbatross.ai/api";
-var Client = class {
+var Client = class _Client {
+  token;
+  tenantId;
+  baseUrl;
+  makeRequest;
+  predictionPreProcessing;
   /**
    * create new albatross sdk client instance
    *
@@ -356,11 +360,6 @@ var Client = class {
     this.makeRequest = options.makeRequest || makeRequest;
     this.predictionPreProcessing = options.predictionPreProcessing || ((data) => data);
   }
-  token;
-  tenantId;
-  baseUrl;
-  makeRequest;
-  predictionPreProcessing;
   /**
    * generate http headers based on endpoint type
    *
@@ -374,7 +373,7 @@ var Client = class {
   getHeaders(endpointType) {
     const headers = {
       "content-type": "application/json",
-      Authorization: "Bearer " + this.token
+      Authorization: `Bearer ${this.token}`
     };
     if (endpointType === "robin") {
       headers["X-Tenant-Id"] = this.tenantId;
@@ -393,7 +392,7 @@ var Client = class {
    * @throws {Error} if image exceeds 50mb (52428800 bytes)
    * @private
    */
-  validateImageSize(base64Image) {
+  static validateImageSize(base64Image) {
     const MAX_SIZE = 52428800;
     if (base64Image.length > MAX_SIZE) {
       throw new Error(
@@ -436,7 +435,7 @@ var Client = class {
   async contentSelection(request) {
     ContentSelectionRequestSchema.parse(request);
     if (request.options?.image) {
-      this.validateImageSize(request.options.image);
+      _Client.validateImageSize(request.options.image);
     }
     const url = `${this.baseUrl}/prediction/content-selection`;
     return this.makeRequest({
@@ -615,7 +614,7 @@ var Client = class {
    */
   async searchByImage(image, useCase, options) {
     SearchByImageParamsSchema.parse({ image, useCase, options });
-    this.validateImageSize(image);
+    _Client.validateImageSize(image);
     return this.contentSelection({
       use_case: useCase,
       search: options?.query,
@@ -691,7 +690,7 @@ var Client = class {
    * ```
    */
   async getVersion() {
-    const url = this.baseUrl + "/version";
+    const url = `${this.baseUrl}/version`;
     return this.makeRequest({ url, headers: this.getHeaders("catalog") });
   }
   /**
@@ -732,7 +731,7 @@ var Client = class {
     CatalogAddPropsSchema.parse({ entity, data, mainUnit, formatData });
     const formattedData = formatData === void 0 || formatData === true ? data.map(flattenNested) : data;
     return this.makeRequest({
-      url: this.baseUrl + "/catalog",
+      url: `${this.baseUrl}/catalog`,
       method: "PUT",
       data: { data: formattedData, entity, mainUnit },
       headers: this.getHeaders("catalog")
@@ -775,7 +774,7 @@ var Client = class {
     };
     const queryParamsString = Object.entries(queryParams).filter((entry) => !!entry[1]).map(([k, v]) => `${k}=${encodeURIComponent(v)}`).join("&");
     const path = "/catalog/csv";
-    const url = this.baseUrl + path + "?" + queryParamsString;
+    const url = `${this.baseUrl}${path}?${queryParamsString}`;
     const body = bodyToCSV(data);
     const r = await fetch(url, { headers: this.getHeaders("catalog"), method: "PUT", body });
     return r.json();
@@ -977,9 +976,13 @@ var FilterBuilder = class {
       field,
       condition: { type: "Match", value }
     };
-    if (type === "must") this.must.push(condition);
-    else if (type === "should") this.should.push(condition);
-    else this.must_not.push(condition);
+    if (type === "must") {
+      this.must.push(condition);
+    } else if (type === "should") {
+      this.should.push(condition);
+    } else {
+      this.must_not.push(condition);
+    }
     return this;
   }
   /**
@@ -1015,9 +1018,13 @@ var FilterBuilder = class {
       field,
       condition: { type: "Range", value }
     };
-    if (type === "must") this.must.push(condition);
-    else if (type === "should") this.should.push(condition);
-    else this.must_not.push(condition);
+    if (type === "must") {
+      this.must.push(condition);
+    } else if (type === "should") {
+      this.should.push(condition);
+    } else {
+      this.must_not.push(condition);
+    }
     return this;
   }
   /**
@@ -1053,9 +1060,13 @@ var FilterBuilder = class {
       field,
       condition: { type: "DatetimeRange", value }
     };
-    if (type === "must") this.must.push(condition);
-    else if (type === "should") this.should.push(condition);
-    else this.must_not.push(condition);
+    if (type === "must") {
+      this.must.push(condition);
+    } else if (type === "should") {
+      this.should.push(condition);
+    } else {
+      this.must_not.push(condition);
+    }
     return this;
   }
   /**
@@ -1086,9 +1097,13 @@ var FilterBuilder = class {
       field,
       condition: { type: "GeoRadius", value: { lat, lon, radius } }
     };
-    if (type === "must") this.must.push(condition);
-    else if (type === "should") this.should.push(condition);
-    else this.must_not.push(condition);
+    if (type === "must") {
+      this.must.push(condition);
+    } else if (type === "should") {
+      this.should.push(condition);
+    } else {
+      this.must_not.push(condition);
+    }
     return this;
   }
   /**
@@ -1115,9 +1130,15 @@ var FilterBuilder = class {
    */
   build() {
     const filter = {};
-    if (this.must.length > 0) filter.must = this.must;
-    if (this.should.length > 0) filter.should = this.should;
-    if (this.must_not.length > 0) filter.must_not = this.must_not;
+    if (this.must.length > 0) {
+      filter.must = this.must;
+    }
+    if (this.should.length > 0) {
+      filter.should = this.should;
+    }
+    if (this.must_not.length > 0) {
+      filter.must_not = this.must_not;
+    }
     return filter;
   }
 };