exa-js 1.7.2 → 1.7.4

package/dist/index.mjs

@@ -242,10 +242,6 @@ var CreateEnrichmentParametersFormat = /* @__PURE__ */ ((CreateEnrichmentParamet
   CreateEnrichmentParametersFormat2["phone"] = "phone";
   return CreateEnrichmentParametersFormat2;
 })(CreateEnrichmentParametersFormat || {});
-var CreateWebsetSearchParametersBehaviour = /* @__PURE__ */ ((CreateWebsetSearchParametersBehaviour2) => {
-  CreateWebsetSearchParametersBehaviour2["override"] = "override";
-  return CreateWebsetSearchParametersBehaviour2;
-})(CreateWebsetSearchParametersBehaviour || {});
 var EventType = /* @__PURE__ */ ((EventType2) => {
   EventType2["webset_created"] = "webset.created";
   EventType2["webset_deleted"] = "webset.deleted";
@@ -261,6 +257,11 @@ var EventType = /* @__PURE__ */ ((EventType2) => {
   EventType2["webset_item_enriched"] = "webset.item.enriched";
   return EventType2;
 })(EventType || {});
+var UpdateStreamStatus = /* @__PURE__ */ ((UpdateStreamStatus2) => {
+  UpdateStreamStatus2["open"] = "open";
+  UpdateStreamStatus2["closed"] = "closed";
+  return UpdateStreamStatus2;
+})(UpdateStreamStatus || {});
 var WebhookStatus = /* @__PURE__ */ ((WebhookStatus2) => {
   WebhookStatus2["active"] = "active";
   WebhookStatus2["inactive"] = "inactive";
@@ -289,6 +290,7 @@ var WebsetEnrichmentFormat = /* @__PURE__ */ ((WebsetEnrichmentFormat2) => {
 })(WebsetEnrichmentFormat || {});
 var WebsetItemSource = /* @__PURE__ */ ((WebsetItemSource2) => {
   WebsetItemSource2["search"] = "search";
+  WebsetItemSource2["import"] = "import";
   return WebsetItemSource2;
 })(WebsetItemSource || {});
 var WebsetItemEvaluationSatisfied = /* @__PURE__ */ ((WebsetItemEvaluationSatisfied2) => {
@@ -297,11 +299,6 @@ var WebsetItemEvaluationSatisfied = /* @__PURE__ */ ((WebsetItemEvaluationSatisf
   WebsetItemEvaluationSatisfied2["unclear"] = "unclear";
   return WebsetItemEvaluationSatisfied2;
 })(WebsetItemEvaluationSatisfied || {});
-var WebsetSearchCanceledReason = /* @__PURE__ */ ((WebsetSearchCanceledReason2) => {
-  WebsetSearchCanceledReason2["webset_deleted"] = "webset_deleted";
-  WebsetSearchCanceledReason2["webset_canceled"] = "webset_canceled";
-  return WebsetSearchCanceledReason2;
-})(WebsetSearchCanceledReason || {});
 var WebsetSearchStatus = /* @__PURE__ */ ((WebsetSearchStatus2) => {
   WebsetSearchStatus2["created"] = "created";
   WebsetSearchStatus2["running"] = "running";
@@ -309,6 +306,16 @@ var WebsetSearchStatus = /* @__PURE__ */ ((WebsetSearchStatus2) => {
   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 {
@@ -351,6 +358,94 @@ var WebsetSearchesClient = class extends WebsetsBaseClient {
   }
 };
 
+// src/websets/streams.ts
+var WebsetStreamRunsClient = class extends WebsetsBaseClient {
+  /**
+   * List all runs for a Stream
+   * @param streamId The ID of the Stream
+   * @param options Pagination options
+   * @returns The list of Stream runs
+   */
+  async list(streamId, options) {
+    const params = this.buildPaginationParams(options);
+    return this.request(
+      `/v0/streams/${streamId}/runs`,
+      "GET",
+      void 0,
+      params
+    );
+  }
+  /**
+   * Get a specific Stream run
+   * @param streamId The ID of the Stream
+   * @param runId The ID of the Stream run
+   * @returns The Stream run
+   */
+  async get(streamId, runId) {
+    return this.request(
+      `/v0/streams/${streamId}/runs/${runId}`,
+      "GET"
+    );
+  }
+};
+var WebsetStreamsClient = class extends WebsetsBaseClient {
+  constructor(client) {
+    super(client);
+    this.runs = new WebsetStreamRunsClient(client);
+  }
+  /**
+   * Create a Stream
+   * @param params The stream parameters
+   * @returns The created Stream
+   */
+  async create(params) {
+    return this.request("/v0/streams", "POST", params);
+  }
+  /**
+   * Get a Stream by ID
+   * @param id The ID of the Stream
+   * @returns The Stream
+   */
+  async get(id) {
+    return this.request(`/v0/streams/${id}`, "GET");
+  }
+  /**
+   * List all Streams
+   * @param options Pagination and filtering options
+   * @returns The list of Streams
+   */
+  async list(options) {
+    const params = {
+      cursor: options?.cursor,
+      limit: options?.limit,
+      websetId: options?.websetId
+    };
+    return this.request(
+      "/v0/streams",
+      "GET",
+      void 0,
+      params
+    );
+  }
+  /**
+   * Update a Stream
+   * @param id The ID of the Stream
+   * @param params The stream update parameters (status, metadata)
+   * @returns The updated Stream
+   */
+  async update(id, params) {
+    return this.request(`/v0/streams/${id}`, "PATCH", params);
+  }
+  /**
+   * Delete a Stream
+   * @param id The ID of the Stream
+   * @returns The deleted Stream
+   */
+  async delete(id) {
+    return this.request(`/v0/streams/${id}`, "DELETE");
+  }
+};
+
 // src/websets/webhooks.ts
 var WebsetWebhooksClient = class extends WebsetsBaseClient {
   /**
@@ -499,6 +594,7 @@ var WebsetsClient = class extends WebsetsBaseClient {
     this.items = new WebsetItemsClient(client);
     this.searches = new WebsetSearchesClient(client);
     this.enrichments = new WebsetEnrichmentsClient(client);
+    this.streams = new WebsetStreamsClient(client);
     this.webhooks = new WebsetWebhooksClient(client);
   }
   /**
@@ -654,7 +750,12 @@ var ResearchBaseClient = class {
    * @returns The parsed JSON response
    */
   async request(endpoint, method = "POST", data, params) {
-    return this.client.request(`/research${endpoint}`, method, data, params);
+    return this.client.request(
+      `/v0/research${endpoint}`,
+      method,
+      data,
+      params
+    );
   }
 };
 
@@ -664,7 +765,7 @@ var ResearchClient = class extends ResearchBaseClient {
     super(client);
   }
   /**
-   * Create and run a research task (blocking call).
+   * Create a research task.
    *
    * Both parameters are required and have fixed shapes:
    * 1. `input`
@@ -687,17 +788,36 @@ var ResearchClient = class extends ResearchBaseClient {
   }
   /**
    * Retrieve a research task by ID.
-   *
-   * Not yet implemented server-side. Calling this will throw until the API is
-   * available.
    */
-  async getTask() {
-    throw new Error("getTask is not implemented yet.");
+  async getTask(id) {
+    return this.request(`/tasks/${id}`, "GET");
+  }
+  /**
+   * Poll a research task until completion or failure.
+   * Polls every 1 second with a maximum timeout of 10 minutes.
+   */
+  async pollTask(id) {
+    const pollingInterval = 1e3;
+    const maxPollingTime = 10 * 60 * 1e3;
+    const startTime = Date.now();
+    while (true) {
+      const task = await this.request(`/tasks/${id}`, "GET");
+      if (task.status === "completed" || task.status === "failed") {
+        return task;
+      }
+      if (Date.now() - startTime > maxPollingTime) {
+        throw new Error(
+          `Polling timeout: Task ${id} did not complete within 10 minutes`
+        );
+      }
+      await new Promise((resolve) => setTimeout(resolve, pollingInterval));
+    }
   }
 };
 
 // src/research/types.ts
 var ResearchStatus = /* @__PURE__ */ ((ResearchStatus2) => {
+  ResearchStatus2["in_progress"] = "in_progress";
   ResearchStatus2["completed"] = "completed";
   ResearchStatus2["failed"] = "failed";
   return ResearchStatus2;
@@ -1115,13 +1235,13 @@ var Exa2 = class {
 var index_default = Exa2;
 export {
   CreateEnrichmentParametersFormat,
-  CreateWebsetSearchParametersBehaviour,
   EventType,
   Exa2 as Exa,
   ExaError,
   HttpStatusCode,
   ResearchClient,
   ResearchStatus,
+  UpdateStreamStatus,
   WebhookStatus,
   WebsetEnrichmentFormat,
   WebsetEnrichmentStatus,
@@ -1129,10 +1249,12 @@ export {
   WebsetItemEvaluationSatisfied,
   WebsetItemSource,
   WebsetItemsClient,
+  WebsetSearchBehavior,
   WebsetSearchCanceledReason,
   WebsetSearchStatus,
   WebsetSearchesClient,
   WebsetStatus,
+  WebsetStreamsClient,
   WebsetWebhooksClient,
   WebsetsClient,
   index_default as default