exa-js 1.8.12 → 1.8.13

package/dist/index.js

@@ -31,11 +31,20 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 var index_exports = {};
 __export(index_exports, {
   CreateEnrichmentParametersFormat: () => CreateEnrichmentParametersFormat,
+  CreateImportParametersFormat: () => CreateImportParametersFormat,
+  CreateWebsetParametersImportSource: () => CreateWebsetParametersImportSource,
+  CreateWebsetParametersSearchExcludeSource: () => CreateWebsetParametersSearchExcludeSource,
+  CreateWebsetSearchParametersExcludeSource: () => CreateWebsetSearchParametersExcludeSource,
   EventType: () => EventType,
   EventsClient: () => EventsClient,
   Exa: () => Exa2,
   ExaError: () => ExaError,
   HttpStatusCode: () => HttpStatusCode,
+  ImportFailedReason: () => ImportFailedReason,
+  ImportFormat: () => ImportFormat,
+  ImportObject: () => ImportObject,
+  ImportStatus: () => ImportStatus,
+  ImportsClient: () => ImportsClient,
   MonitorObject: () => MonitorObject,
   MonitorRunObject: () => MonitorRunObject,
   MonitorRunStatus: () => MonitorRunStatus,
@@ -217,6 +226,311 @@ var EventsClient = class extends WebsetsBaseClient {
   }
 };
 
+// src/websets/openapi.ts
+var CreateEnrichmentParametersFormat = /* @__PURE__ */ ((CreateEnrichmentParametersFormat2) => {
+  CreateEnrichmentParametersFormat2["text"] = "text";
+  CreateEnrichmentParametersFormat2["date"] = "date";
+  CreateEnrichmentParametersFormat2["number"] = "number";
+  CreateEnrichmentParametersFormat2["options"] = "options";
+  CreateEnrichmentParametersFormat2["email"] = "email";
+  CreateEnrichmentParametersFormat2["phone"] = "phone";
+  return CreateEnrichmentParametersFormat2;
+})(CreateEnrichmentParametersFormat || {});
+var CreateImportParametersFormat = /* @__PURE__ */ ((CreateImportParametersFormat2) => {
+  CreateImportParametersFormat2["csv"] = "csv";
+  return CreateImportParametersFormat2;
+})(CreateImportParametersFormat || {});
+var CreateWebsetParametersImportSource = /* @__PURE__ */ ((CreateWebsetParametersImportSource2) => {
+  CreateWebsetParametersImportSource2["import"] = "import";
+  CreateWebsetParametersImportSource2["webset"] = "webset";
+  return CreateWebsetParametersImportSource2;
+})(CreateWebsetParametersImportSource || {});
+var CreateWebsetParametersSearchExcludeSource = /* @__PURE__ */ ((CreateWebsetParametersSearchExcludeSource2) => {
+  CreateWebsetParametersSearchExcludeSource2["import"] = "import";
+  CreateWebsetParametersSearchExcludeSource2["webset"] = "webset";
+  return CreateWebsetParametersSearchExcludeSource2;
+})(CreateWebsetParametersSearchExcludeSource || {});
+var CreateWebsetSearchParametersExcludeSource = /* @__PURE__ */ ((CreateWebsetSearchParametersExcludeSource2) => {
+  CreateWebsetSearchParametersExcludeSource2["import"] = "import";
+  CreateWebsetSearchParametersExcludeSource2["webset"] = "webset";
+  return CreateWebsetSearchParametersExcludeSource2;
+})(CreateWebsetSearchParametersExcludeSource || {});
+var EventType = /* @__PURE__ */ ((EventType2) => {
+  EventType2["webset_created"] = "webset.created";
+  EventType2["webset_deleted"] = "webset.deleted";
+  EventType2["webset_paused"] = "webset.paused";
+  EventType2["webset_idle"] = "webset.idle";
+  EventType2["webset_search_created"] = "webset.search.created";
+  EventType2["webset_search_canceled"] = "webset.search.canceled";
+  EventType2["webset_search_completed"] = "webset.search.completed";
+  EventType2["webset_search_updated"] = "webset.search.updated";
+  EventType2["import_created"] = "import.created";
+  EventType2["import_completed"] = "import.completed";
+  EventType2["import_processing"] = "import.processing";
+  EventType2["webset_item_created"] = "webset.item.created";
+  EventType2["webset_item_enriched"] = "webset.item.enriched";
+  EventType2["webset_export_created"] = "webset.export.created";
+  EventType2["webset_export_completed"] = "webset.export.completed";
+  return EventType2;
+})(EventType || {});
+var ImportFailedReason = /* @__PURE__ */ ((ImportFailedReason2) => {
+  ImportFailedReason2["invalid_format"] = "invalid_format";
+  ImportFailedReason2["invalid_file_content"] = "invalid_file_content";
+  ImportFailedReason2["missing_identifier"] = "missing_identifier";
+  return ImportFailedReason2;
+})(ImportFailedReason || {});
+var ImportFormat = /* @__PURE__ */ ((ImportFormat2) => {
+  ImportFormat2["csv"] = "csv";
+  ImportFormat2["webset"] = "webset";
+  return ImportFormat2;
+})(ImportFormat || {});
+var ImportObject = /* @__PURE__ */ ((ImportObject2) => {
+  ImportObject2["import"] = "import";
+  return ImportObject2;
+})(ImportObject || {});
+var ImportStatus = /* @__PURE__ */ ((ImportStatus2) => {
+  ImportStatus2["pending"] = "pending";
+  ImportStatus2["processing"] = "processing";
+  ImportStatus2["completed"] = "completed";
+  ImportStatus2["failed"] = "failed";
+  return ImportStatus2;
+})(ImportStatus || {});
+var MonitorObject = /* @__PURE__ */ ((MonitorObject2) => {
+  MonitorObject2["monitor"] = "monitor";
+  return MonitorObject2;
+})(MonitorObject || {});
+var MonitorStatus = /* @__PURE__ */ ((MonitorStatus2) => {
+  MonitorStatus2["enabled"] = "enabled";
+  MonitorStatus2["disabled"] = "disabled";
+  return MonitorStatus2;
+})(MonitorStatus || {});
+var MonitorRunObject = /* @__PURE__ */ ((MonitorRunObject2) => {
+  MonitorRunObject2["monitor_run"] = "monitor_run";
+  return MonitorRunObject2;
+})(MonitorRunObject || {});
+var MonitorRunStatus = /* @__PURE__ */ ((MonitorRunStatus2) => {
+  MonitorRunStatus2["created"] = "created";
+  MonitorRunStatus2["running"] = "running";
+  MonitorRunStatus2["completed"] = "completed";
+  MonitorRunStatus2["canceled"] = "canceled";
+  return MonitorRunStatus2;
+})(MonitorRunStatus || {});
+var MonitorRunType = /* @__PURE__ */ ((MonitorRunType2) => {
+  MonitorRunType2["search"] = "search";
+  MonitorRunType2["refresh"] = "refresh";
+  return MonitorRunType2;
+})(MonitorRunType || {});
+var UpdateMonitorStatus = /* @__PURE__ */ ((UpdateMonitorStatus2) => {
+  UpdateMonitorStatus2["enabled"] = "enabled";
+  UpdateMonitorStatus2["disabled"] = "disabled";
+  return UpdateMonitorStatus2;
+})(UpdateMonitorStatus || {});
+var WebhookStatus = /* @__PURE__ */ ((WebhookStatus2) => {
+  WebhookStatus2["active"] = "active";
+  WebhookStatus2["inactive"] = "inactive";
+  return WebhookStatus2;
+})(WebhookStatus || {});
+var WebsetStatus = /* @__PURE__ */ ((WebsetStatus2) => {
+  WebsetStatus2["idle"] = "idle";
+  WebsetStatus2["running"] = "running";
+  WebsetStatus2["paused"] = "paused";
+  return WebsetStatus2;
+})(WebsetStatus || {});
+var WebsetEnrichmentStatus = /* @__PURE__ */ ((WebsetEnrichmentStatus2) => {
+  WebsetEnrichmentStatus2["pending"] = "pending";
+  WebsetEnrichmentStatus2["canceled"] = "canceled";
+  WebsetEnrichmentStatus2["completed"] = "completed";
+  return WebsetEnrichmentStatus2;
+})(WebsetEnrichmentStatus || {});
+var WebsetEnrichmentFormat = /* @__PURE__ */ ((WebsetEnrichmentFormat2) => {
+  WebsetEnrichmentFormat2["text"] = "text";
+  WebsetEnrichmentFormat2["date"] = "date";
+  WebsetEnrichmentFormat2["number"] = "number";
+  WebsetEnrichmentFormat2["options"] = "options";
+  WebsetEnrichmentFormat2["email"] = "email";
+  WebsetEnrichmentFormat2["phone"] = "phone";
+  return WebsetEnrichmentFormat2;
+})(WebsetEnrichmentFormat || {});
+var WebsetItemSource = /* @__PURE__ */ ((WebsetItemSource2) => {
+  WebsetItemSource2["search"] = "search";
+  WebsetItemSource2["import"] = "import";
+  return WebsetItemSource2;
+})(WebsetItemSource || {});
+var WebsetItemEvaluationSatisfied = /* @__PURE__ */ ((WebsetItemEvaluationSatisfied2) => {
+  WebsetItemEvaluationSatisfied2["yes"] = "yes";
+  WebsetItemEvaluationSatisfied2["no"] = "no";
+  WebsetItemEvaluationSatisfied2["unclear"] = "unclear";
+  return WebsetItemEvaluationSatisfied2;
+})(WebsetItemEvaluationSatisfied || {});
+var WebsetSearchStatus = /* @__PURE__ */ ((WebsetSearchStatus2) => {
+  WebsetSearchStatus2["created"] = "created";
+  WebsetSearchStatus2["running"] = "running";
+  WebsetSearchStatus2["completed"] = "completed";
+  WebsetSearchStatus2["canceled"] = "canceled";
+  return WebsetSearchStatus2;
+})(WebsetSearchStatus || {});
+var WebsetSearchBehavior = /* @__PURE__ */ ((WebsetSearchBehavior2) => {
+  WebsetSearchBehavior2["override"] = "override";
+  WebsetSearchBehavior2["append"] = "append";
+  return WebsetSearchBehavior2;
+})(WebsetSearchBehavior || {});
+var WebsetSearchCanceledReason = /* @__PURE__ */ ((WebsetSearchCanceledReason2) => {
+  WebsetSearchCanceledReason2["webset_deleted"] = "webset_deleted";
+  WebsetSearchCanceledReason2["webset_canceled"] = "webset_canceled";
+  return WebsetSearchCanceledReason2;
+})(WebsetSearchCanceledReason || {});
+
+// src/websets/imports.ts
+var ImportsClient = class extends WebsetsBaseClient {
+  async create(params, csv) {
+    if (csv === void 0) {
+      return this.request(
+        "/v0/imports",
+        "POST",
+        params
+      );
+    }
+    let csvBuffer;
+    if (typeof csv === "string") {
+      csvBuffer = Buffer.from(csv, "utf8");
+    } else if (Buffer.isBuffer(csv)) {
+      csvBuffer = csv;
+    } else {
+      throw new ExaError(
+        "Invalid CSV data input. Must be string or Buffer",
+        400 /* BadRequest */
+      );
+    }
+    const sizeInBytes = csvBuffer.length;
+    const sizeInMB = Math.max(1, Math.ceil(sizeInBytes / (1024 * 1024)));
+    const csvText = csvBuffer.toString("utf8");
+    const lines = csvText.split("\n").filter((line) => line.trim().length > 0);
+    const recordCount = Math.max(0, lines.length - 1);
+    if (sizeInMB > 50) {
+      throw new ExaError(
+        `CSV file too large: ${sizeInMB}MB. Maximum size is 50MB.`,
+        400 /* BadRequest */
+      );
+    }
+    if (recordCount === 0) {
+      throw new ExaError(
+        "CSV file appears to have no data records (only header or empty)",
+        400 /* BadRequest */
+      );
+    }
+    const createParams = {
+      title: params.title,
+      format: "csv" /* csv */,
+      entity: params.entity,
+      size: sizeInBytes,
+      count: recordCount,
+      metadata: params.metadata,
+      csv: params.csv
+    };
+    const importResponse = await this.request(
+      "/v0/imports",
+      "POST",
+      createParams
+    );
+    try {
+      const uploadResponse = await fetch(importResponse.uploadUrl, {
+        method: "PUT",
+        body: csvBuffer
+      });
+      if (!uploadResponse.ok) {
+        const errorText = await uploadResponse.text();
+        throw new ExaError(
+          `Upload failed: ${uploadResponse.status} ${uploadResponse.statusText}. ${errorText}`,
+          400 /* BadRequest */
+        );
+      }
+    } catch (error) {
+      if (error instanceof ExaError) {
+        throw error;
+      }
+      throw new ExaError(
+        `Failed to upload CSV data: ${error.message}`,
+        400 /* BadRequest */
+      );
+    }
+    return importResponse;
+  }
+  /**
+   * Get an Import by ID
+   * @param id The ID of the Import
+   * @returns The Import
+   */
+  async get(id) {
+    return this.request(`/v0/imports/${id}`, "GET");
+  }
+  /**
+   * List all Imports
+   * @param options Pagination options
+   * @returns The list of Imports
+   */
+  async list(options) {
+    const params = this.buildPaginationParams(options);
+    return this.request(
+      "/v0/imports",
+      "GET",
+      void 0,
+      params
+    );
+  }
+  /**
+   * Update an Import
+   * @param id The ID of the Import
+   * @param params The import update parameters
+   * @returns The updated Import
+   */
+  async update(id, params) {
+    return this.request(`/v0/imports/${id}`, "PATCH", params);
+  }
+  /**
+   * Delete an Import
+   * @param id The ID of the Import
+   * @returns The deleted Import
+   */
+  async delete(id) {
+    return this.request(`/v0/imports/${id}`, "DELETE");
+  }
+  /**
+   * Wait until an Import is completed or failed
+   * @param id The ID of the Import
+   * @param options Configuration options for timeout and polling
+   * @returns The Import once it reaches a final state (completed or failed)
+   * @throws Error if the Import does not complete within the timeout or fails
+   */
+  async waitUntilCompleted(id, options) {
+    const timeout = options?.timeout ?? 30 * 60 * 1e3;
+    const pollInterval = options?.pollInterval ?? 2e3;
+    const onPoll = options?.onPoll;
+    const startTime = Date.now();
+    while (true) {
+      const importItem = await this.get(id);
+      if (onPoll) {
+        onPoll(importItem.status);
+      }
+      if (importItem.status === "completed" /* completed */) {
+        return importItem;
+      }
+      if (importItem.status === "failed" /* failed */) {
+        throw new ExaError(
+          `Import ${id} failed: ${importItem.failedMessage || "Unknown error"}`,
+          400 /* BadRequest */
+        );
+      }
+      if (Date.now() - startTime > timeout) {
+        throw new ExaError(
+          `Import ${id} did not complete within ${timeout}ms. Current status: ${importItem.status}`,
+          408 /* RequestTimeout */
+        );
+      }
+      await new Promise((resolve) => setTimeout(resolve, pollInterval));
+    }
+  }
+};
+
 // src/websets/items.ts
 var WebsetItemsClient = class extends WebsetsBaseClient {
   /**
@@ -382,116 +696,6 @@ var WebsetMonitorsClient = class extends WebsetsBaseClient {
   }
 };
 
-// src/websets/openapi.ts
-var CreateEnrichmentParametersFormat = /* @__PURE__ */ ((CreateEnrichmentParametersFormat2) => {
-  CreateEnrichmentParametersFormat2["text"] = "text";
-  CreateEnrichmentParametersFormat2["date"] = "date";
-  CreateEnrichmentParametersFormat2["number"] = "number";
-  CreateEnrichmentParametersFormat2["options"] = "options";
-  CreateEnrichmentParametersFormat2["email"] = "email";
-  CreateEnrichmentParametersFormat2["phone"] = "phone";
-  return CreateEnrichmentParametersFormat2;
-})(CreateEnrichmentParametersFormat || {});
-var EventType = /* @__PURE__ */ ((EventType2) => {
-  EventType2["webset_created"] = "webset.created";
-  EventType2["webset_deleted"] = "webset.deleted";
-  EventType2["webset_paused"] = "webset.paused";
-  EventType2["webset_idle"] = "webset.idle";
-  EventType2["webset_search_created"] = "webset.search.created";
-  EventType2["webset_search_canceled"] = "webset.search.canceled";
-  EventType2["webset_search_completed"] = "webset.search.completed";
-  EventType2["webset_search_updated"] = "webset.search.updated";
-  EventType2["webset_export_created"] = "webset.export.created";
-  EventType2["webset_export_completed"] = "webset.export.completed";
-  EventType2["webset_item_created"] = "webset.item.created";
-  EventType2["webset_item_enriched"] = "webset.item.enriched";
-  return EventType2;
-})(EventType || {});
-var MonitorObject = /* @__PURE__ */ ((MonitorObject2) => {
-  MonitorObject2["monitor"] = "monitor";
-  return MonitorObject2;
-})(MonitorObject || {});
-var MonitorStatus = /* @__PURE__ */ ((MonitorStatus2) => {
-  MonitorStatus2["enabled"] = "enabled";
-  MonitorStatus2["disabled"] = "disabled";
-  return MonitorStatus2;
-})(MonitorStatus || {});
-var MonitorRunObject = /* @__PURE__ */ ((MonitorRunObject2) => {
-  MonitorRunObject2["monitor_run"] = "monitor_run";
-  return MonitorRunObject2;
-})(MonitorRunObject || {});
-var MonitorRunStatus = /* @__PURE__ */ ((MonitorRunStatus2) => {
-  MonitorRunStatus2["created"] = "created";
-  MonitorRunStatus2["running"] = "running";
-  MonitorRunStatus2["completed"] = "completed";
-  MonitorRunStatus2["canceled"] = "canceled";
-  return MonitorRunStatus2;
-})(MonitorRunStatus || {});
-var MonitorRunType = /* @__PURE__ */ ((MonitorRunType2) => {
-  MonitorRunType2["search"] = "search";
-  MonitorRunType2["refresh"] = "refresh";
-  return MonitorRunType2;
-})(MonitorRunType || {});
-var UpdateMonitorStatus = /* @__PURE__ */ ((UpdateMonitorStatus2) => {
-  UpdateMonitorStatus2["enabled"] = "enabled";
-  UpdateMonitorStatus2["disabled"] = "disabled";
-  return UpdateMonitorStatus2;
-})(UpdateMonitorStatus || {});
-var WebhookStatus = /* @__PURE__ */ ((WebhookStatus2) => {
-  WebhookStatus2["active"] = "active";
-  WebhookStatus2["inactive"] = "inactive";
-  return WebhookStatus2;
-})(WebhookStatus || {});
-var WebsetStatus = /* @__PURE__ */ ((WebsetStatus2) => {
-  WebsetStatus2["idle"] = "idle";
-  WebsetStatus2["running"] = "running";
-  WebsetStatus2["paused"] = "paused";
-  return WebsetStatus2;
-})(WebsetStatus || {});
-var WebsetEnrichmentStatus = /* @__PURE__ */ ((WebsetEnrichmentStatus2) => {
-  WebsetEnrichmentStatus2["pending"] = "pending";
-  WebsetEnrichmentStatus2["canceled"] = "canceled";
-  WebsetEnrichmentStatus2["completed"] = "completed";
-  return WebsetEnrichmentStatus2;
-})(WebsetEnrichmentStatus || {});
-var WebsetEnrichmentFormat = /* @__PURE__ */ ((WebsetEnrichmentFormat2) => {
-  WebsetEnrichmentFormat2["text"] = "text";
-  WebsetEnrichmentFormat2["date"] = "date";
-  WebsetEnrichmentFormat2["number"] = "number";
-  WebsetEnrichmentFormat2["options"] = "options";
-  WebsetEnrichmentFormat2["email"] = "email";
-  WebsetEnrichmentFormat2["phone"] = "phone";
-  return WebsetEnrichmentFormat2;
-})(WebsetEnrichmentFormat || {});
-var WebsetItemSource = /* @__PURE__ */ ((WebsetItemSource2) => {
-  WebsetItemSource2["search"] = "search";
-  WebsetItemSource2["import"] = "import";
-  return WebsetItemSource2;
-})(WebsetItemSource || {});
-var WebsetItemEvaluationSatisfied = /* @__PURE__ */ ((WebsetItemEvaluationSatisfied2) => {
-  WebsetItemEvaluationSatisfied2["yes"] = "yes";
-  WebsetItemEvaluationSatisfied2["no"] = "no";
-  WebsetItemEvaluationSatisfied2["unclear"] = "unclear";
-  return WebsetItemEvaluationSatisfied2;
-})(WebsetItemEvaluationSatisfied || {});
-var WebsetSearchStatus = /* @__PURE__ */ ((WebsetSearchStatus2) => {
-  WebsetSearchStatus2["created"] = "created";
-  WebsetSearchStatus2["running"] = "running";
-  WebsetSearchStatus2["completed"] = "completed";
-  WebsetSearchStatus2["canceled"] = "canceled";
-  return WebsetSearchStatus2;
-})(WebsetSearchStatus || {});
-var WebsetSearchBehavior = /* @__PURE__ */ ((WebsetSearchBehavior2) => {
-  WebsetSearchBehavior2["override"] = "override";
-  WebsetSearchBehavior2["append"] = "append";
-  return WebsetSearchBehavior2;
-})(WebsetSearchBehavior || {});
-var WebsetSearchCanceledReason = /* @__PURE__ */ ((WebsetSearchCanceledReason2) => {
-  WebsetSearchCanceledReason2["webset_deleted"] = "webset_deleted";
-  WebsetSearchCanceledReason2["webset_canceled"] = "webset_canceled";
-  return WebsetSearchCanceledReason2;
-})(WebsetSearchCanceledReason || {});
-
 // src/websets/searches.ts
 var WebsetSearchesClient = class extends WebsetsBaseClient {
   /**
@@ -678,6 +882,7 @@ var WebsetsClient = class extends WebsetsBaseClient {
   constructor(client) {
     super(client);
     this.events = new EventsClient(client);
+    this.imports = new ImportsClient(client);
     this.items = new WebsetItemsClient(client);
     this.searches = new WebsetSearchesClient(client);
     this.enrichments = new WebsetEnrichmentsClient(client);
@@ -1443,11 +1648,20 @@ var index_default = Exa2;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   CreateEnrichmentParametersFormat,
+  CreateImportParametersFormat,
+  CreateWebsetParametersImportSource,
+  CreateWebsetParametersSearchExcludeSource,
+  CreateWebsetSearchParametersExcludeSource,
   EventType,
   EventsClient,
   Exa,
   ExaError,
   HttpStatusCode,
+  ImportFailedReason,
+  ImportFormat,
+  ImportObject,
+  ImportStatus,
+  ImportsClient,
   MonitorObject,
   MonitorRunObject,
   MonitorRunStatus,