exa-js 1.7.4 → 1.8.0

package/dist/index.js

@@ -36,7 +36,8 @@ __export(index_exports, {
   ExaError: () => ExaError,
   HttpStatusCode: () => HttpStatusCode,
   ResearchClient: () => ResearchClient,
-  ResearchStatus: () => ResearchStatus,
+  ResearchModel: () => ResearchCreateTaskRequestDtoModel,
+  ResearchStatus: () => ResearchTaskDtoStatus,
   UpdateStreamStatus: () => UpdateStreamStatus,
   WebhookStatus: () => WebhookStatus,
   WebsetEnrichmentFormat: () => WebsetEnrichmentFormat,
@@ -808,40 +809,65 @@ var ResearchBaseClient = class {
    */
   async request(endpoint, method = "POST", data, params) {
     return this.client.request(
-      `/v0/research${endpoint}`,
+      `/research/v0${endpoint}`,
       method,
       data,
       params
     );
   }
+  /**
+   * Helper to build pagination parameters.
+   * @param pagination The pagination parameters
+   * @returns QueryParams object with pagination parameters
+   */
+  buildPaginationParams(pagination) {
+    const params = {};
+    if (!pagination) return params;
+    if (pagination.cursor) params.cursor = pagination.cursor;
+    if (pagination.limit) params.limit = pagination.limit;
+    return params;
+  }
 };
 
+// src/research/openapi.ts
+var ResearchCreateTaskRequestDtoModel = /* @__PURE__ */ ((ResearchCreateTaskRequestDtoModel2) => {
+  ResearchCreateTaskRequestDtoModel2["exa_research"] = "exa-research";
+  ResearchCreateTaskRequestDtoModel2["exa_ultra"] = "exa-ultra";
+  return ResearchCreateTaskRequestDtoModel2;
+})(ResearchCreateTaskRequestDtoModel || {});
+var ResearchTaskDtoStatus = /* @__PURE__ */ ((ResearchTaskDtoStatus2) => {
+  ResearchTaskDtoStatus2["running"] = "running";
+  ResearchTaskDtoStatus2["completed"] = "completed";
+  ResearchTaskDtoStatus2["failed"] = "failed";
+  return ResearchTaskDtoStatus2;
+})(ResearchTaskDtoStatus || {});
+
 // src/research/client.ts
 var ResearchClient = class extends ResearchBaseClient {
   constructor(client) {
     super(client);
   }
   /**
-   * Create a research task.
+   * Create a new research task.
    *
-   * Both parameters are required and have fixed shapes:
-   * 1. `input`
-   *      `{ instructions: string }`
-   *     • `instructions` – High-level guidance that tells the research agent what to do.
-   * 2. `output`
-   *    defines the exact structure you expect back, and guides the research conducted by the agent.
-   *      `{ schema: JSONSchema }`.
-   *    The agent's response will be validated against this schema.
+   * @param params Object containing:
+   *   - model: The research model to use (e.g., ResearchModel.ExaResearch).
+   *   - instructions: High-level guidance for the research agent.
+   *   - output: An object with a `schema` property (JSONSchema) that defines the expected output structure.
    *
-   * @param input  Object containing high-level research instructions.
-   * @param output Object containing the expected output schema.
-   * @returns The ResearchTaskResponse returned by the API.
+   * @returns An object containing the unique ID of the created research task.
    */
-  async createTask(input, output) {
-    return this.request("/tasks", "POST", {
-      input,
-      output
-    });
+  async createTask(params) {
+    const { model, ...rest } = params;
+    const payload = {
+      model: model ?? "exa-research" /* exa_research */,
+      ...rest
+    };
+    return this.request(
+      "/tasks",
+      "POST",
+      payload
+    );
   }
   /**
    * Retrieve a research task by ID.
@@ -852,15 +878,28 @@ var ResearchClient = class extends ResearchBaseClient {
   /**
    * Poll a research task until completion or failure.
    * Polls every 1 second with a maximum timeout of 10 minutes.
+   * Resilient to up to 10 consecutive polling failures.
    */
   async pollTask(id) {
     const pollingInterval = 1e3;
     const maxPollingTime = 10 * 60 * 1e3;
+    const maxConsecutiveFailures = 10;
     const startTime = Date.now();
+    let consecutiveFailures = 0;
     while (true) {
-      const task = await this.request(`/tasks/${id}`, "GET");
-      if (task.status === "completed" || task.status === "failed") {
-        return task;
+      try {
+        const task = await this.request(`/tasks/${id}`, "GET");
+        consecutiveFailures = 0;
+        if (task.status === "completed" || task.status === "failed") {
+          return task;
+        }
+      } catch (err) {
+        consecutiveFailures += 1;
+        if (consecutiveFailures >= maxConsecutiveFailures) {
+          throw new Error(
+            `Polling failed ${maxConsecutiveFailures} times in a row for task ${id}: ${err}`
+          );
+        }
       }
       if (Date.now() - startTime > maxPollingTime) {
         throw new Error(
@@ -870,16 +909,22 @@ var ResearchClient = class extends ResearchBaseClient {
       await new Promise((resolve) => setTimeout(resolve, pollingInterval));
     }
   }
+  /**
+   * List research tasks
+   * @param options Pagination options
+   * @returns The paginated list of research tasks
+   */
+  async listTasks(options) {
+    const params = this.buildPaginationParams(options);
+    return this.request(
+      "/tasks",
+      "GET",
+      void 0,
+      params
+    );
+  }
 };
 
-// src/research/types.ts
-var ResearchStatus = /* @__PURE__ */ ((ResearchStatus2) => {
-  ResearchStatus2["in_progress"] = "in_progress";
-  ResearchStatus2["completed"] = "completed";
-  ResearchStatus2["failed"] = "failed";
-  return ResearchStatus2;
-})(ResearchStatus || {});
-
 // src/index.ts
 var fetchImpl = typeof global !== "undefined" && global.fetch ? global.fetch : import_cross_fetch.default;
 var HeadersImpl = typeof global !== "undefined" && global.Headers ? global.Headers : import_cross_fetch.Headers;
@@ -1298,6 +1343,7 @@ var index_default = Exa2;
   ExaError,
   HttpStatusCode,
   ResearchClient,
+  ResearchModel,
   ResearchStatus,
   UpdateStreamStatus,
   WebhookStatus,