190proof 1.0.97 → 1.0.98

package/dist/index.mjs

@@ -843,6 +843,7 @@ async function prepareGoogleAIPayload(_identifier, payload) {
   const preparedPayload = {
     model: payload.model,
     messages: [],
+    requestTimeoutMs: payload.requestTimeoutMs,
     tools: payload.functions ? {
       functionDeclarations: payload.functions.map((fn) => ({
         name: fn.name,
@@ -928,7 +929,7 @@ async function prepareGoogleAIPayload(_identifier, payload) {
   return preparedPayload;
 }
 async function callGoogleAI(id, payload) {
-  var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r;
+  var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s;
   const contents = jigGoogleMessages(payload.messages);
   const requestBody = {
     contents,
@@ -951,47 +952,50 @@ async function callGoogleAI(id, payload) {
           "content-type": "application/json",
           "x-goog-api-key": process.env.GEMINI_API_KEY
         },
-        timeout: 6e4
+        // Per-attempt timeout. Defaults to 60s; callers pass a larger value via
+        // payload.requestTimeoutMs for slow, large generations (e.g. single-file
+        // app codegen) so a long-but-valid response isn't cut short.
+        timeout: (_a = payload.requestTimeoutMs) != null ? _a : 6e4
       }
     );
     response = httpResponse.data;
   } catch (err) {
-    const apiError = (_b = (_a = err == null ? void 0 : err.response) == null ? void 0 : _a.data) == null ? void 0 : _b.error;
+    const apiError = (_c = (_b = err == null ? void 0 : err.response) == null ? void 0 : _b.data) == null ? void 0 : _c.error;
     const wrapped = new Error(
       (apiError == null ? void 0 : apiError.message) || (err == null ? void 0 : err.message) || "Google AI API request failed"
     );
-    wrapped.status = (_d = apiError == null ? void 0 : apiError.status) != null ? _d : (_c = err == null ? void 0 : err.response) == null ? void 0 : _c.status;
+    wrapped.status = (_e = apiError == null ? void 0 : apiError.status) != null ? _e : (_d = err == null ? void 0 : err.response) == null ? void 0 : _d.status;
     wrapped.code = apiError == null ? void 0 : apiError.code;
     wrapped.details = apiError == null ? void 0 : apiError.details;
-    wrapped.promptFeedback = (_f = (_e = err == null ? void 0 : err.response) == null ? void 0 : _e.data) == null ? void 0 : _f.promptFeedback;
+    wrapped.promptFeedback = (_g = (_f = err == null ? void 0 : err.response) == null ? void 0 : _f.data) == null ? void 0 : _g.promptFeedback;
     throw wrapped;
   }
   let text = "";
   const files = [];
   const reasoningParts = [];
   const functionCalls = [];
-  for (const part of ((_i = (_h = (_g = response.candidates) == null ? void 0 : _g[0]) == null ? void 0 : _h.content) == null ? void 0 : _i.parts) || []) {
+  for (const part of ((_j = (_i = (_h = response.candidates) == null ? void 0 : _h[0]) == null ? void 0 : _i.content) == null ? void 0 : _j.parts) || []) {
     if (part.thought) {
       reasoningParts.push(part);
       continue;
     }
     if (part.functionCall) {
       functionCalls.push({
-        id: (_j = part.functionCall.id) != null ? _j : `call_${functionCalls.length}`,
-        name: (_k = part.functionCall.name) != null ? _k : "",
-        arguments: (_l = part.functionCall.args) != null ? _l : {},
+        id: (_k = part.functionCall.id) != null ? _k : `call_${functionCalls.length}`,
+        name: (_l = part.functionCall.name) != null ? _l : "",
+        arguments: (_m = part.functionCall.args) != null ? _m : {},
         thoughtSignature: part.thoughtSignature
       });
       continue;
     }
     if (part.text)
       text += part.text;
-    if ((_m = part.inlineData) == null ? void 0 : _m.data) {
+    if ((_n = part.inlineData) == null ? void 0 : _n.data) {
       files.push({ mimeType: "image/png", data: part.inlineData.data });
     }
   }
   if (!text && !functionCalls.length && !files.length) {
-    const candidate = (_n = response.candidates) == null ? void 0 : _n[0];
+    const candidate = (_o = response.candidates) == null ? void 0 : _o[0];
     const finishReason = candidate == null ? void 0 : candidate.finishReason;
     logger_default.error(id, "Missing text & functions in Google AI API response:", {
       finishReason,
@@ -1026,10 +1030,10 @@ async function callGoogleAI(id, payload) {
     function_calls: functionCalls,
     reasoningDetails: reasoningParts.length ? reasoningParts : void 0,
     usage: response.usageMetadata ? {
-      prompt_tokens: (_o = response.usageMetadata.promptTokenCount) != null ? _o : 0,
-      completion_tokens: (_p = response.usageMetadata.candidatesTokenCount) != null ? _p : 0,
-      total_tokens: (_q = response.usageMetadata.totalTokenCount) != null ? _q : 0,
-      cached_tokens: (_r = response.usageMetadata.cachedContentTokenCount) != null ? _r : 0
+      prompt_tokens: (_p = response.usageMetadata.promptTokenCount) != null ? _p : 0,
+      completion_tokens: (_q = response.usageMetadata.candidatesTokenCount) != null ? _q : 0,
+      total_tokens: (_r = response.usageMetadata.totalTokenCount) != null ? _r : 0,
+      cached_tokens: (_s = response.usageMetadata.cachedContentTokenCount) != null ? _s : 0
     } : null
   };
 }