valyu-js 2.8.0 → 2.9.0

package/dist/index.mjs

@@ -3,7 +3,8 @@ import { createHmac, timingSafeEqual } from "crypto";
 import http from "http";
 import https from "https";
 import axios from "axios";
-var SDK_VERSION = "2.8.0";
+var SDK_VERSION = "2.9.0";
+var TRANSIENT_STATUS_CODES = /* @__PURE__ */ new Set([408, 429, 500, 502, 503, 504]);
 function normalizeContentsJobResponse(api) {
   return {
     success: api.success ?? true,
@@ -788,19 +789,49 @@ var Valyu = class {
   /**
    * DeepResearch: Get task status
    */
-  async _deepresearchStatus(taskId) {
-    try {
-      const response = await this.client.get(
-        `${this.baseUrl}/deepresearch/tasks/${taskId}/status`,
-        { headers: this.headers }
-      );
-      return { success: true, ...response.data };
-    } catch (e) {
-      return {
-        success: false,
-        error: e.response?.data?.error || e.message
-      };
+  async _deepresearchStatus(taskId, maxAttempts = 5) {
+    const url = `${this.baseUrl}/deepresearch/tasks/${taskId}/status`;
+    let lastError = "status endpoint unreachable";
+    for (let attempt = 0; attempt < maxAttempts; attempt++) {
+      try {
+        const response = await this.client.get(url, {
+          headers: this.headers,
+          // Classify status codes ourselves rather than letting axios throw.
+          validateStatus: () => true
+        });
+        if (TRANSIENT_STATUS_CODES.has(response.status)) {
+          lastError = `HTTP ${response.status}`;
+        } else {
+          const contentType = String(
+            response.headers?.["content-type"] || ""
+          );
+          const body = response.data;
+          const isJsonObject = contentType.toLowerCase().includes("application/json") && typeof body === "object" && body !== null && !Array.isArray(body);
+          if (!isJsonObject) {
+            lastError = `non-JSON/empty status response (HTTP ${response.status}, content-type: ${contentType || "none"})`;
+          } else if (response.status >= 400) {
+            return {
+              success: false,
+              error: body.error || `HTTP Error: ${response.status}`
+            };
+          } else {
+            const { success: _ignored, ...rest } = body;
+            return { success: true, ...rest };
+          }
+        }
+      } catch (e) {
+        lastError = e?.message || String(e);
+      }
+      if (attempt < maxAttempts - 1) {
+        const delayMs = Math.min(2 ** attempt, 30) * 1e3 + Math.random() * 1e3;
+        await new Promise((resolve) => setTimeout(resolve, delayMs));
+      }
     }
+    return {
+      success: false,
+      unreachable: true,
+      error: `Status endpoint unreachable after ${maxAttempts} attempts: ${lastError}`
+    };
   }
   /**
    * DeepResearch: Wait for task completion with polling
@@ -812,6 +843,15 @@ var Valyu = class {
     while (true) {
       const status = await this._deepresearchStatus(taskId);
       if (!status.success) {
+        if (status.unreachable) {
+          if (Date.now() - startTime > maxWaitTime) {
+            throw new Error(
+              `Status endpoint unreachable for ${maxWaitTime}ms: ${status.error}`
+            );
+          }
+          await new Promise((resolve) => setTimeout(resolve, pollInterval));
+          continue;
+        }
         throw new Error(status.error);
       }
       if (options.onProgress) {