@pitcher/js-api 1.27.1 → 1.27.3

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,29 +2474,41 @@ 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 }
-    })
-  });
-  if (!response.ok) {
-    throw new Error(`pia_search ${response.status}`);
+  const controller = new AbortController();
+  const timer = setTimeout(() => controller.abort(), ONLINE_TIMEOUT_MS);
+  try {
+    const 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 }
+      })
+    });
+    if (!response.ok) {
+      throw new Error(`pia_search ${response.status}`);
+    }
+    const data = await response.json();
+    const output = data?.pia_search ?? {};
+    return {
+      answer: typeof output.answer === "string" ? output.answer : "",
+      cited_file_ids: Array.isArray(output.cited_file_ids) ? output.cited_file_ids : [],
+      ...output.confidence && { confidence: output.confidence },
+      source: "online"
+    };
+  } 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);
   }
-  const data = await response.json();
-  const output = data?.pia_search ?? {};
-  return {
-    answer: typeof output.answer === "string" ? output.answer : "",
-    cited_file_ids: Array.isArray(output.cited_file_ids) ? output.cited_file_ids : [],
-    ...output.confidence && { confidence: output.confidence },
-    source: "online"
-  };
 }
 
 function open$2(payload = {}) {