@pitcher/js-api 1.27.1 → 1.27.2

package/js-api.esm.js

@@ -2388,6 +2388,7 @@ function resolvePiaSearchMode(config) {
 const PIA_SEARCH_ROUTE = "/core/api/protected/pia-prepare/pia-search";
 const CAPABILITIES_TIMEOUT_MS = 3e3;
 const ON_DEVICE_MAX_TOKENS = 512;
+const ONLINE_TIMEOUT_MS = 3e4;
 const WEB_CAPABILITIES = {
   available: true,
   offline: false,
@@ -2473,18 +2474,31 @@ async function piaSearchAnswer(payload) {
   if (!instanceId) {
     throw new Error("piaSearchAnswer: instance_id could not be resolved from the payload or env");
   }
-  const response = await fetch(`${apiOriginFromEnv(env)}${PIA_SEARCH_ROUTE}`, {
-    method: "POST",
-    headers: {
-      "Content-Type": "application/json",
-      Authorization: `Bearer ${env.pitcher.access_token}`
-    },
-    body: JSON.stringify({
-      query: payload.query,
-      instance_id: instanceId,
-      ...Array.isArray(payload.file_ids) && payload.file_ids.length > 0 && { file_ids: payload.file_ids }
-    })
-  });
+  const controller = new AbortController();
+  const timer = setTimeout(() => controller.abort(), ONLINE_TIMEOUT_MS);
+  let response;
+  try {
+    response = await fetch(`${apiOriginFromEnv(env)}${PIA_SEARCH_ROUTE}`, {
+      method: "POST",
+      signal: controller.signal,
+      headers: {
+        "Content-Type": "application/json",
+        Authorization: `Bearer ${env.pitcher.access_token}`
+      },
+      body: JSON.stringify({
+        query: payload.query,
+        instance_id: instanceId,
+        ...Array.isArray(payload.file_ids) && payload.file_ids.length > 0 && { file_ids: payload.file_ids }
+      })
+    });
+  } catch (error) {
+    if (error?.name === "AbortError") {
+      throw new Error(`pia_search online request timed out after ${ONLINE_TIMEOUT_MS}ms`);
+    }
+    throw error;
+  } finally {
+    clearTimeout(timer);
+  }
   if (!response.ok) {
     throw new Error(`pia_search ${response.status}`);
   }