@axlsdk/axl 0.17.7 → 0.17.8

package/dist/index.cjs

@@ -1535,7 +1535,8 @@ var GEMINI_PRICING = {
   "gemini-3-flash-preview": [5e-7, 3e-6],
   "gemini-3.1-pro-preview": [2e-6, 12e-6],
   "gemini-3.1-flash-lite": [25e-8, 15e-7],
-  "gemini-3.1-flash-lite-preview": [25e-8, 15e-7]
+  "gemini-3.1-flash-lite-preview": [25e-8, 15e-7],
+  "gemini-3.5-flash": [15e-7, 9e-6]
 };
 var GEMINI_PRICING_KEYS_BY_LENGTH = Object.keys(GEMINI_PRICING).sort(
   (a, b) => b.length - a.length
@@ -1775,12 +1776,21 @@ var GeminiProvider = class {
    */
   mapMessages(messages) {
     const toolCallIdToName = /* @__PURE__ */ new Map();
+    const geminiNativeIds = /* @__PURE__ */ new Set();
     for (const msg of messages) {
       if (msg.role === "assistant" && msg.tool_calls) {
         for (const tc of msg.tool_calls) {
           toolCallIdToName.set(tc.id, tc.function.name);
         }
       }
+      if (msg.role === "assistant") {
+        const rawParts = msg.providerMetadata?.geminiParts;
+        if (rawParts) {
+          for (const p of rawParts) {
+            if (p.functionCall?.id) geminiNativeIds.add(p.functionCall.id);
+          }
+        }
+      }
     }
     const result = [];
     for (const msg of messages) {
@@ -1821,16 +1831,16 @@ var GeminiProvider = class {
         } catch {
           responseData = { result: msg.content };
         }
+        const functionResponse = {
+          name: functionName,
+          response: responseData
+        };
+        if (msg.tool_call_id && geminiNativeIds.has(msg.tool_call_id)) {
+          functionResponse.id = msg.tool_call_id;
+        }
         result.push({
           role: "user",
-          parts: [
-            {
-              functionResponse: {
-                name: functionName,
-                response: responseData
-              }
-            }
-          ]
+          parts: [{ functionResponse }]
         });
       } else if (msg.role === "user") {
         result.push({ role: "user", parts: [{ text: msg.content }] });
@@ -1898,7 +1908,7 @@ var GeminiProvider = class {
           content += part.text;
         } else if (part.functionCall) {
           toolCalls.push({
-            id: `call_${this.callCounter++}`,
+            id: part.functionCall.id || `call_${this.callCounter++}`,
             type: "function",
             function: {
               name: part.functionCall.name,
@@ -1978,7 +1988,7 @@ var GeminiProvider = class {
               } else if (part.functionCall) {
                 yield {
                   type: "tool_call_delta",
-                  id: `call_${this.callCounter++}`,
+                  id: part.functionCall.id || `call_${this.callCounter++}`,
                   name: part.functionCall.name,
                   arguments: JSON.stringify(part.functionCall.args)
                 };