exa-js 1.8.6 → 1.8.9

package/dist/index.mjs

@@ -782,6 +782,22 @@ var ResearchBaseClient = class {
       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}`,
+      method,
+      data,
+      params
+    );
+  }
   /**
    * Helper to build pagination parameters.
    * @param pagination The pagination parameters
@@ -796,19 +812,6 @@ var ResearchBaseClient = class {
   }
 };
 
-// src/research/openapi.ts
-var ResearchCreateTaskRequestDtoModel = /* @__PURE__ */ ((ResearchCreateTaskRequestDtoModel2) => {
-  ResearchCreateTaskRequestDtoModel2["exa_research"] = "exa-research";
-  ResearchCreateTaskRequestDtoModel2["exa_research_pro"] = "exa-research-pro";
-  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) {
@@ -834,19 +837,61 @@ var ResearchClient = class extends ResearchBaseClient {
         inferSchema: output.inferSchema ?? true
       } : { inferSchema: true }
     };
-    return this.request(
-      "/tasks",
-      "POST",
-      payload
-    );
+    return this.request("/tasks", "POST", payload);
   }
-  /**
-   * Retrieve a research task by ID.
-   */
-  async getTask(id) {
-    return this.request(`/tasks/${id}`, "GET");
+  getTask(id, options) {
+    if (options?.stream) {
+      const promise = async () => {
+        const resp = await this.rawRequest(`/tasks/${id}?stream=true`, "GET");
+        if (!resp.body) {
+          throw new Error("No response body for SSE stream");
+        }
+        const reader = resp.body.getReader();
+        const decoder = new TextDecoder();
+        let buffer = "";
+        function processPart(part) {
+          const lines = part.split("\n");
+          let data = lines.slice(1).join("\n");
+          if (data.startsWith("data:")) {
+            data = data.slice(5).trimStart();
+          }
+          try {
+            return JSON.parse(data);
+          } catch (e) {
+            return null;
+          }
+        }
+        async function* streamEvents() {
+          while (true) {
+            const { done, value } = await reader.read();
+            if (done) break;
+            buffer += decoder.decode(value, { stream: true });
+            let parts = buffer.split("\n\n");
+            buffer = parts.pop() ?? "";
+            for (const part of parts) {
+              const processed = processPart(part);
+              if (processed) {
+                yield processed;
+              }
+            }
+          }
+          if (buffer.trim()) {
+            const processed = processPart(buffer.trim());
+            if (processed) {
+              yield processed;
+            }
+          }
+        }
+        return streamEvents();
+      };
+      return promise();
+    } else {
+      return this.request(`/tasks/${id}`, "GET");
+    }
   }
   /**
+   * @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.
@@ -896,6 +941,19 @@ var ResearchClient = class extends ResearchBaseClient {
   }
 };
 
+// src/research/openapi.ts
+var ResearchCreateOpenAIResponseDtoModel = /* @__PURE__ */ ((ResearchCreateOpenAIResponseDtoModel2) => {
+  ResearchCreateOpenAIResponseDtoModel2["exa_research"] = "exa-research";
+  ResearchCreateOpenAIResponseDtoModel2["exa_research_pro"] = "exa-research-pro";
+  return ResearchCreateOpenAIResponseDtoModel2;
+})(ResearchCreateOpenAIResponseDtoModel || {});
+var ResearchTaskDtoStatus = /* @__PURE__ */ ((ResearchTaskDtoStatus2) => {
+  ResearchTaskDtoStatus2["running"] = "running";
+  ResearchTaskDtoStatus2["completed"] = "completed";
+  ResearchTaskDtoStatus2["failed"] = "failed";
+  return ResearchTaskDtoStatus2;
+})(ResearchTaskDtoStatus || {});
+
 // src/index.ts
 var fetchImpl = typeof global !== "undefined" && global.fetch ? global.fetch : fetch;
 var HeadersImpl = typeof global !== "undefined" && global.Headers ? global.Headers : Headers;
@@ -1017,6 +1075,28 @@ var Exa2 = class {
     }
     return await response.json();
   }
+  async rawRequest(endpoint, method = "POST", body, queryParams) {
+    let url = this.baseURL + endpoint;
+    if (queryParams) {
+      const searchParams = new URLSearchParams();
+      for (const [key, value] of Object.entries(queryParams)) {
+        if (Array.isArray(value)) {
+          for (const item of value) {
+            searchParams.append(key, String(item));
+          }
+        } else if (value !== void 0) {
+          searchParams.append(key, String(value));
+        }
+      }
+      url += `?${searchParams.toString()}`;
+    }
+    const response = await fetchImpl(url, {
+      method,
+      headers: this.headers,
+      body: body ? JSON.stringify(body) : void 0
+    });
+    return response;
+  }
   /**
    * Performs a search with an Exa prompt-engineered query.
    *
@@ -1324,7 +1404,7 @@ export {
   MonitorRunType,
   MonitorStatus,
   ResearchClient,
-  ResearchCreateTaskRequestDtoModel as ResearchModel,
+  ResearchCreateOpenAIResponseDtoModel as ResearchModel,
   ResearchTaskDtoStatus as ResearchStatus,
   UpdateMonitorStatus,
   WebhookStatus,