190proof 1.0.106 → 1.0.107

package/dist/index.js

@@ -146,6 +146,19 @@ function anySignal(signals) {
     signals
   );
 }
+async function withRequestDeadline(apiName, requestTimeoutMs, signal, fn) {
+  const deadline = AbortSignal.timeout(requestTimeoutMs);
+  try {
+    return await fn(signal ? anySignal([signal, deadline]) : deadline);
+  } catch (error2) {
+    if (deadline.aborted && !(signal == null ? void 0 : signal.aborted)) {
+      throw new Error(
+        `${apiName} request exceeded hard deadline of ${requestTimeoutMs}ms`
+      );
+    }
+    throw error2;
+  }
+}
 async function withRetries(identifier, apiName, fn, options = {}) {
   var _a, _b, _c, _d, _e;
   const { retries = 5, baseDelayMs = 125, onError } = options;
@@ -786,28 +799,33 @@ async function callAnthropic(id, payload, config, requestTimeoutMs = 12e4, signa
       ...tools.slice(0, -1),
       { ...tools[tools.length - 1], cache_control: { type: "ephemeral" } }
     ] : tools;
-    const response = await import_axios.default.post(
-      "https://api.anthropic.com/v1/messages",
-      {
-        model: payload.model,
-        messages: anthropicMessages,
-        tools: cachedTools,
-        // tool_choice requires tools in the request; drop it otherwise.
-        tool_choice: (cachedTools == null ? void 0 : cachedTools.length) ? payload.tool_choice : void 0,
-        temperature: payload.temperature,
-        system: payload.system,
-        max_tokens: 4096
-      },
-      {
-        headers: {
-          "content-type": "application/json",
-          "x-api-key": process.env.ANTHROPIC_API_KEY,
-          "anthropic-version": "2023-06-01",
-          "anthropic-beta": "tools-2024-04-04"
+    const response = await withRequestDeadline(
+      "Anthropic",
+      requestTimeoutMs,
+      signal,
+      (mergedSignal) => import_axios.default.post(
+        "https://api.anthropic.com/v1/messages",
+        {
+          model: payload.model,
+          messages: anthropicMessages,
+          tools: cachedTools,
+          // tool_choice requires tools in the request; drop it otherwise.
+          tool_choice: (cachedTools == null ? void 0 : cachedTools.length) ? payload.tool_choice : void 0,
+          temperature: payload.temperature,
+          system: payload.system,
+          max_tokens: 4096
         },
-        timeout: requestTimeoutMs,
-        signal
-      }
+        {
+          headers: {
+            "content-type": "application/json",
+            "x-api-key": process.env.ANTHROPIC_API_KEY,
+            "anthropic-version": "2023-06-01",
+            "anthropic-beta": "tools-2024-04-04"
+          },
+          timeout: requestTimeoutMs,
+          signal: mergedSignal
+        }
+      )
     );
     data = response.data;
   }
@@ -1024,17 +1042,22 @@ async function callGoogleAI(id, payload, requestTimeoutMs = 12e4, signal) {
   }
   let response;
   try {
-    const httpResponse = await import_axios.default.post(
-      `https://generativelanguage.googleapis.com/v1beta/models/${payload.model}:generateContent`,
-      requestBody,
-      {
-        headers: {
-          "content-type": "application/json",
-          "x-goog-api-key": process.env.GEMINI_API_KEY
-        },
-        timeout: requestTimeoutMs,
-        signal
-      }
+    const httpResponse = await withRequestDeadline(
+      "Google AI",
+      requestTimeoutMs,
+      signal,
+      (mergedSignal) => import_axios.default.post(
+        `https://generativelanguage.googleapis.com/v1beta/models/${payload.model}:generateContent`,
+        requestBody,
+        {
+          headers: {
+            "content-type": "application/json",
+            "x-goog-api-key": process.env.GEMINI_API_KEY
+          },
+          timeout: requestTimeoutMs,
+          signal: mergedSignal
+        }
+      )
     );
     response = httpResponse.data;
   } catch (err) {
@@ -1257,17 +1280,22 @@ function prepareGroqPayload(payload) {
 }
 async function callGroq(id, payload, requestTimeoutMs = 12e4, signal) {
   var _a, _b, _c, _d, _e, _f, _g;
-  const response = await import_axios.default.post(
-    "https://api.groq.com/openai/v1/chat/completions",
-    payload,
-    {
-      headers: {
-        "content-type": "application/json",
-        Authorization: `Bearer ${process.env.GROQ_API_KEY}`
-      },
-      timeout: requestTimeoutMs,
-      signal
-    }
+  const response = await withRequestDeadline(
+    "Groq",
+    requestTimeoutMs,
+    signal,
+    (mergedSignal) => import_axios.default.post(
+      "https://api.groq.com/openai/v1/chat/completions",
+      payload,
+      {
+        headers: {
+          "content-type": "application/json",
+          Authorization: `Bearer ${process.env.GROQ_API_KEY}`
+        },
+        timeout: requestTimeoutMs,
+        signal: mergedSignal
+      }
+    )
   );
   if (response.data.error) {
     logger_default.error(id, "Groq error:", response.data.error);
@@ -1383,17 +1411,23 @@ function parseDsmlToolCalls(content) {
 }
 async function callOpenRouter(id, payload, requestTimeoutMs = 12e4, signal) {
   var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
-  const response = await import_axios.default.post(
-    "https://openrouter.ai/api/v1/chat/completions",
-    payload,
-    {
-      headers: {
-        "content-type": "application/json",
-        Authorization: `Bearer ${process.env.OPENROUTER_API_KEY}`
-      },
-      timeout: requestTimeoutMs,
-      signal
-    }
+  const response = await withRequestDeadline(
+    "OpenRouter",
+    requestTimeoutMs,
+    signal,
+    (mergedSignal) => import_axios.default.post(
+      // Override point for tests (deadline behavior needs a local server).
+      `${process.env.OPENROUTER_BASE_URL || "https://openrouter.ai"}/api/v1/chat/completions`,
+      payload,
+      {
+        headers: {
+          "content-type": "application/json",
+          Authorization: `Bearer ${process.env.OPENROUTER_API_KEY}`
+        },
+        timeout: requestTimeoutMs,
+        signal: mergedSignal
+      }
+    )
   );
   if (response.data.error) {
     logger_default.error(id, "OpenRouter error:", response.data.error);