ai 3.4.24 → 3.4.26

package/dist/index.mjs

@@ -3593,8 +3593,6 @@ function runToolsTransformation({
   telemetry,
   abortSignal
 }) {
-  let canClose = false;
-  const outstandingToolCalls = /* @__PURE__ */ new Set();
   let toolResultsStreamController = null;
   const toolResultsStream = new ReadableStream({
     start(controller) {
@@ -3602,6 +3600,17 @@ function runToolsTransformation({
     }
   });
   const activeToolCalls = {};
+  const outstandingToolResults = /* @__PURE__ */ new Set();
+  let canClose = false;
+  let finishChunk = void 0;
+  function attemptClose() {
+    if (canClose && outstandingToolResults.size === 0) {
+      if (finishChunk != null) {
+        toolResultsStreamController.enqueue(finishChunk);
+      }
+      toolResultsStreamController.close();
+    }
+  }
   const forwardStream = new TransformStream({
     transform(chunk, controller) {
       const chunkType = chunk.type;
@@ -3659,7 +3668,7 @@ function runToolsTransformation({
             controller.enqueue(toolCall);
             if (tool2.execute != null) {
               const toolExecutionId = generateId();
-              outstandingToolCalls.add(toolExecutionId);
+              outstandingToolResults.add(toolExecutionId);
               recordSpan({
                 name: "ai.toolCall",
                 attributes: selectTelemetryAttributes({
@@ -3684,10 +3693,8 @@ function runToolsTransformation({
                       type: "tool-result",
                       result
                     });
-                    outstandingToolCalls.delete(toolExecutionId);
-                    if (canClose && outstandingToolCalls.size === 0) {
-                      toolResultsStreamController.close();
-                    }
+                    outstandingToolResults.delete(toolExecutionId);
+                    attemptClose();
                     try {
                       span.setAttributes(
                         selectTelemetryAttributes({
@@ -3707,10 +3714,8 @@ function runToolsTransformation({
                       type: "error",
                       error
                     });
-                    outstandingToolCalls.delete(toolExecutionId);
-                    if (canClose && outstandingToolCalls.size === 0) {
-                      toolResultsStreamController.close();
-                    }
+                    outstandingToolResults.delete(toolExecutionId);
+                    attemptClose();
                   }
                 )
               });
@@ -3724,13 +3729,13 @@ function runToolsTransformation({
           break;
         }
         case "finish": {
-          controller.enqueue({
+          finishChunk = {
             type: "finish",
             finishReason: chunk.finishReason,
             logprobs: chunk.logprobs,
             usage: calculateLanguageModelUsage(chunk.usage),
             experimental_providerMetadata: chunk.providerMetadata
-          });
+          };
           break;
         }
         default: {
@@ -3741,9 +3746,7 @@ function runToolsTransformation({
     },
     flush() {
       canClose = true;
-      if (outstandingToolCalls.size === 0) {
-        toolResultsStreamController.close();
-      }
+      attemptClose();
     }
   });
   return new ReadableStream({
@@ -4718,7 +4721,9 @@ function attachmentsToParts(attachments) {
 }
 
 // core/prompt/convert-to-core-messages.ts
-function convertToCoreMessages(messages) {
+function convertToCoreMessages(messages, options) {
+  var _a11;
+  const tools = (_a11 = options == null ? void 0 : options.tools) != null ? _a11 : {};
   const coreMessages = [];
   for (const message of messages) {
     const { role, content, toolInvocations, experimental_attachments } = message;
@@ -4749,29 +4754,37 @@ function convertToCoreMessages(messages) {
           role: "assistant",
           content: [
             { type: "text", text: content },
-            ...toolInvocations.map(({ toolCallId, toolName, args }) => ({
-              type: "tool-call",
-              toolCallId,
-              toolName,
-              args
-            }))
+            ...toolInvocations.map(
+              ({ toolCallId, toolName, args }) => ({
+                type: "tool-call",
+                toolCallId,
+                toolName,
+                args
+              })
+            )
           ]
         });
         coreMessages.push({
           role: "tool",
-          content: toolInvocations.map((ToolInvocation) => {
-            if (!("result" in ToolInvocation)) {
+          content: toolInvocations.map((toolInvocation) => {
+            if (!("result" in toolInvocation)) {
               throw new MessageConversionError({
                 originalMessage: message,
-                message: "ToolInvocation must have a result: " + JSON.stringify(ToolInvocation)
+                message: "ToolInvocation must have a result: " + JSON.stringify(toolInvocation)
               });
             }
-            const { toolCallId, toolName, args, result } = ToolInvocation;
-            return {
+            const { toolCallId, toolName, result } = toolInvocation;
+            const tool2 = tools[toolName];
+            return (tool2 == null ? void 0 : tool2.experimental_toToolResultContent) != null ? {
+              type: "tool-result",
+              toolCallId,
+              toolName,
+              result: tool2.experimental_toToolResultContent(result),
+              experimental_content: tool2.experimental_toToolResultContent(result)
+            } : {
               type: "tool-result",
               toolCallId,
               toolName,
-              args,
               result
             };
           })