exa-js 1.8.27 → 1.9.0

package/dist/index.mjs

@@ -1019,50 +1019,25 @@ function zodToJsonSchema(schema) {
 
 // src/research/base.ts
 var ResearchBaseClient = class {
-  /**
-   * Initialize a new Research base client
-   * @param client The Exa client instance
-   */
   constructor(client) {
     this.client = client;
   }
-  /**
-   * Make a request to the Research API (prefixes all paths with `/research`).
-   * @param endpoint The endpoint path, beginning with a slash (e.g. "/tasks").
-   * @param method The HTTP method. Defaults to "POST".
-   * @param data Optional request body
-   * @param params Optional query parameters
-   * @returns The parsed JSON response
-   */
   async request(endpoint, method = "POST", data, params) {
     return this.client.request(
-      `/research/v0${endpoint}`,
+      `/research/v1${endpoint}`,
       method,
       data,
       params
     );
   }
-  /**
-   * Make a request to the Research API (prefixes all paths with `/research`).
-   * @param endpoint The endpoint path, beginning with a slash (e.g. "/tasks").
-   * @param method The HTTP method. Defaults to "POST".
-   * @param data Optional request body
-   * @param params Optional query parameters
-   * @returns The parsed JSON response
-   */
   async rawRequest(endpoint, method = "POST", data, params) {
     return this.client.rawRequest(
-      `/research/v0${endpoint}`,
+      `/research/v1${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;
@@ -1077,26 +1052,34 @@ var ResearchClient = class extends ResearchBaseClient {
   constructor(client) {
     super(client);
   }
-  async createTask(params) {
-    const { instructions, model, output } = params;
-    let schema = output?.schema;
+  async create(params) {
+    const { instructions, model, outputSchema } = params;
+    let schema = outputSchema;
     if (schema && isZodSchema(schema)) {
       schema = zodToJsonSchema(schema);
     }
     const payload = {
       instructions,
-      model: model ?? "exa-research",
-      output: output ? {
-        schema,
-        inferSchema: output.inferSchema ?? true
-      } : { inferSchema: true }
+      model: model ?? "exa-research"
     };
-    return this.request("/tasks", "POST", payload);
+    if (schema) {
+      payload.outputSchema = schema;
+    }
+    return this.request("", "POST", payload);
   }
-  getTask(id, options) {
+  get(researchId, options) {
     if (options?.stream) {
       const promise = async () => {
-        const resp = await this.rawRequest(`/tasks/${id}?stream=true`, "GET");
+        const params = { stream: "true" };
+        if (options.events !== void 0) {
+          params.events = options.events.toString();
+        }
+        const resp = await this.rawRequest(
+          `/${researchId}`,
+          "GET",
+          void 0,
+          params
+        );
         if (!resp.body) {
           throw new Error("No response body for SSE stream");
         }
@@ -1140,59 +1123,53 @@ var ResearchClient = class extends ResearchBaseClient {
       };
       return promise();
     } else {
-      return this.request(`/tasks/${id}`, "GET");
+      const params = { stream: "false" };
+      if (options?.events !== void 0) {
+        params.events = options.events.toString();
+      }
+      return this.request(
+        `/${researchId}`,
+        "GET",
+        void 0,
+        params
+      );
     }
   }
-  /**
-   * @deprecated This method is deprecated and may be removed in a future release.
-   * @see getTask(id, {stream: true})
-   * 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;
+  async list(options) {
+    const params = this.buildPaginationParams(options);
+    return this.request("", "GET", void 0, params);
+  }
+  async pollUntilFinished(researchId, options) {
+    const pollInterval = options?.pollInterval ?? 1e3;
+    const timeoutMs = options?.timeoutMs ?? 10 * 60 * 1e3;
+    const maxConsecutiveFailures = 5;
     const startTime = Date.now();
     let consecutiveFailures = 0;
     while (true) {
       try {
-        const task = await this.request(`/tasks/${id}`, "GET");
+        const research = await this.get(researchId, {
+          events: options?.events
+        });
         consecutiveFailures = 0;
-        if (task.status === "completed" || task.status === "failed") {
-          return task;
+        if (research.status === "completed" || research.status === "failed" || research.status === "canceled") {
+          return research;
         }
       } catch (err) {
         consecutiveFailures += 1;
         if (consecutiveFailures >= maxConsecutiveFailures) {
           throw new Error(
-            `Polling failed ${maxConsecutiveFailures} times in a row for task ${id}: ${err}`
+            `Polling failed ${maxConsecutiveFailures} times in a row for research ${researchId}: ${err}`
           );
         }
       }
-      if (Date.now() - startTime > maxPollingTime) {
+      if (Date.now() - startTime > timeoutMs) {
         throw new Error(
-          `Polling timeout: Task ${id} did not complete within 10 minutes`
+          `Polling timeout: Research ${researchId} did not complete within ${timeoutMs}ms`
         );
       }
-      await new Promise((resolve) => setTimeout(resolve, pollingInterval));
+      await new Promise((resolve) => setTimeout(resolve, pollInterval));
     }
   }
-  /**
-   * 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/index.ts
@@ -1642,7 +1619,6 @@ export {
   MonitorRunType,
   MonitorStatus,
   PreviewWebsetResponseEnrichmentsFormat,
-  ResearchClient,
   ScopeSourceType,
   UpdateMonitorStatus,
   WebhookStatus,