@albatrossai/albatross-sdk 1.1.40 → 1.1.42-alpha

package/dist/index.cjs

@@ -1,18 +1,14 @@
 'use strict';
 
-var sync = require('csv-stringify/sync');
 var zod = require('zod');
+var sync = require('csv-stringify/sync');
 
 // src/request.ts
 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,62 +90,6 @@ var makeRequest = async ({
   }
   throw lastError || new Error("Unexpected error in makeRequest");
 };
-function flattenNested(obj) {
-  const result = {};
-  Object.entries(obj).forEach(([key, value]) => {
-    if (typeof value === "object" && !Array.isArray(value) && value !== null) {
-      Object.entries(value).forEach(([nestedKey, nestedValue]) => {
-        if (Array.isArray(nestedValue)) {
-          result[`${key}_${nestedKey}`] = nestedValue.join(",");
-        } else if (typeof nestedValue === "object" && nestedValue !== null) {
-          Object.entries(nestedValue).forEach(([subKey, subValue]) => {
-            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) => {
-    const value = flattenedRow[header];
-    if (value === null || value === void 0) {
-      return "";
-    }
-    if (typeof value === "object") {
-      return JSON.stringify(value);
-    }
-    return String(value);
-  });
-};
-var bodyToCSV = (data) => {
-  if (data.length === 0) {
-    throw new Error("data is empty");
-  }
-  const row0 = data[0];
-  if (!row0) {
-    throw new Error("data is empty");
-  }
-  const rowHeaders = Object.keys(flattenNested(row0));
-  const csvData = [
-    rowHeaders,
-    ...data.map((row) => preFormatRow(rowHeaders, row))
-  ];
-  return sync.stringify(csvData, {
-    delimiter: ",",
-    quote: true,
-    quoted_string: true,
-    header: false
-  }).trim();
-};
 var EventSchema = zod.z.object({
   eventType: zod.z.string().min(1, "event type required"),
   payload: zod.z.record(zod.z.string(), zod.z.any()).optional(),
@@ -315,10 +258,71 @@ var FeedbackParamsSchema = zod.z.object({
   predictionUuid: zod.z.string().min(1, "prediction uuid required"),
   value: zod.z.record(zod.z.string(), zod.z.any())
 });
+var flattenNested = (obj) => {
+  const result = {};
+  for (const [key, value] of Object.entries(obj)) {
+    if (typeof value === "object" && !Array.isArray(value) && value !== null) {
+      for (const [nestedKey, nestedValue] of Object.entries(value)) {
+        if (Array.isArray(nestedValue)) {
+          result[`${key}_${nestedKey}`] = nestedValue.join(",");
+        } else if (typeof nestedValue === "object" && nestedValue !== null) {
+          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) => {
+    const value = flattenedRow[header];
+    if (value === null || value === void 0) {
+      return "";
+    }
+    if (typeof value === "object") {
+      return JSON.stringify(value);
+    }
+    return String(value);
+  });
+};
+var bodyToCSV = (data) => {
+  if (data.length === 0) {
+    throw new Error("data is empty");
+  }
+  const [row0] = data;
+  if (!row0) {
+    throw new Error("data is empty");
+  }
+  const rowHeaders = Object.keys(flattenNested(row0));
+  const csvData = [
+    rowHeaders,
+    ...data.map((row) => preFormatRow(rowHeaders, row))
+  ];
+  return sync.stringify(csvData, {
+    delimiter: ",",
+    quote: true,
+    quoted_string: true,
+    header: false
+  }).trim();
+};
 
 // 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;
   }
 };