ai 3.0.18 → 3.0.20

package/dist/index.mjs

@@ -1,6 +1,3 @@
-// core/generate-object/generate-object.ts
-import zodToJsonSchema from "zod-to-json-schema";
-
 // spec/errors/api-call-error.ts
 var APICallError = class extends Error {
   constructor({
@@ -297,20 +294,26 @@ var NoTextGeneratedError = class extends Error {
 
 // spec/errors/no-such-tool-error.ts
 var NoSuchToolError = class extends Error {
-  constructor({ message, toolName }) {
+  constructor({
+    toolName,
+    availableTools = void 0,
+    message = `Model tried to call unavailable tool '${toolName}'. ${availableTools === void 0 ? "No tools are available." : `Available tools: ${availableTools.join(", ")}.`}`
+  }) {
     super(message);
     this.name = "AI_NoSuchToolError";
     this.toolName = toolName;
+    this.availableTools = availableTools;
   }
   static isNoSuchToolError(error) {
-    return error instanceof Error && error.name === "AI_NoSuchToolError" && typeof error.toolName === "string";
+    return error instanceof Error && error.name === "AI_NoSuchToolError" && "toolName" in error && error.toolName != void 0 && typeof error.name === "string";
   }
   toJSON() {
     return {
       name: this.name,
       message: this.message,
       stack: this.stack,
-      toolName: this.toolName
+      toolName: this.toolName,
+      availableTools: this.availableTools
     };
   }
 };
@@ -625,6 +628,12 @@ function prepareCallSettings({
   };
 }
 
+// core/util/convert-zod-to-json-schema.ts
+import zodToJsonSchema from "zod-to-json-schema";
+function convertZodToJSONSchema(zodSchema) {
+  return zodToJsonSchema(zodSchema);
+}
+
 // core/util/delay.ts
 async function delay(delayInMs) {
   return new Promise((resolve) => setTimeout(resolve, delayInMs));
@@ -716,7 +725,7 @@ async function experimental_generateObject({
 }) {
   var _a, _b;
   const retry = retryWithExponentialBackoff({ maxRetries });
-  const jsonSchema = zodToJsonSchema(schema);
+  const jsonSchema = convertZodToJSONSchema(schema);
   if (mode === "auto" || mode == null) {
     mode = model.defaultObjectGenerationMode;
   }
@@ -834,9 +843,6 @@ var GenerateObjectResult = class {
   }
 };
 
-// core/generate-object/stream-object.ts
-import zodToJsonSchema2 from "zod-to-json-schema";
-
 // core/util/async-iterable-stream.ts
 function createAsyncIterableStream(source, transformer) {
   const transformedStream = source.pipeThrough(
@@ -1239,7 +1245,7 @@ async function experimental_streamObject({
   ...settings
 }) {
   const retry = retryWithExponentialBackoff({ maxRetries });
-  const jsonSchema = zodToJsonSchema2(schema);
+  const jsonSchema = convertZodToJSONSchema(schema);
   if (mode === "auto" || mode == null) {
     mode = model.defaultObjectGenerationMode;
   }
@@ -1380,9 +1386,6 @@ var StreamObjectResult = class {
   }
 };
 
-// core/generate-text/generate-text.ts
-import zodToJsonSchema3 from "zod-to-json-schema";
-
 // core/generate-text/tool-call.ts
 function parseToolCall({
   toolCall,
@@ -1390,16 +1393,13 @@ function parseToolCall({
 }) {
   const toolName = toolCall.toolName;
   if (tools == null) {
-    throw new NoSuchToolError({
-      message: `Tool ${toolCall.toolName} not found (no tools provided).`,
-      toolName: toolCall.toolName
-    });
+    throw new NoSuchToolError({ toolName: toolCall.toolName });
   }
   const tool2 = tools[toolName];
   if (tool2 == null) {
     throw new NoSuchToolError({
-      message: `Tool ${toolCall.toolName} not found.`,
-      toolName: toolCall.toolName
+      toolName: toolCall.toolName,
+      availableTools: Object.keys(tools)
     });
   }
   const parseResult = safeParseJSON({
@@ -1414,6 +1414,7 @@ function parseToolCall({
     });
   }
   return {
+    type: "tool-call",
     toolCallId: toolCall.toolCallId,
     toolName,
     args: parseResult.value
@@ -1442,7 +1443,7 @@ async function experimental_generateText({
           type: "function",
           name,
           description: tool2.description,
-          parameters: zodToJsonSchema3(tool2.parameters)
+          parameters: convertZodToJSONSchema(tool2.parameters)
         }))
       },
       ...prepareCallSettings(settings),
@@ -1502,9 +1503,6 @@ var GenerateTextResult = class {
   }
 };
 
-// core/generate-text/stream-text.ts
-import zodToJsonSchema4 from "zod-to-json-schema";
-
 // shared/generate-id.ts
 import { customAlphabet } from "nanoid/non-secure";
 var generateId = customAlphabet(
@@ -1539,10 +1537,7 @@ function runToolsTransformation({
           if (tools == null) {
             toolResultsStreamController.enqueue({
               type: "error",
-              error: new NoSuchToolError({
-                message: `Tool ${chunk.toolName} not found (no tools provided).`,
-                toolName: chunk.toolName
-              })
+              error: new NoSuchToolError({ toolName: chunk.toolName })
             });
             break;
           }
@@ -1551,8 +1546,8 @@ function runToolsTransformation({
             toolResultsStreamController.enqueue({
               type: "error",
               error: new NoSuchToolError({
-                message: `Tool ${chunk.toolName} not found.`,
-                toolName: chunk.toolName
+                toolName: chunk.toolName,
+                availableTools: Object.keys(tools)
               })
             });
             break;
@@ -1562,18 +1557,15 @@ function runToolsTransformation({
               toolCall: chunk,
               tools
             });
-            controller.enqueue({
-              type: "tool-call",
-              ...toolCall
-            });
+            controller.enqueue(toolCall);
             if (tool2.execute != null) {
               const toolExecutionId = generateId();
               outstandingToolCalls.add(toolExecutionId);
               tool2.execute(toolCall.args).then(
                 (result) => {
                   toolResultsStreamController.enqueue({
-                    type: "tool-result",
                     ...toolCall,
+                    type: "tool-result",
                     result
                   });
                   outstandingToolCalls.delete(toolExecutionId);
@@ -1675,7 +1667,7 @@ async function experimental_streamText({
           type: "function",
           name,
           description: tool2.description,
-          parameters: zodToJsonSchema4(tool2.parameters)
+          parameters: convertZodToJSONSchema(tool2.parameters)
         }))
       },
       ...prepareCallSettings(settings),
@@ -1747,9 +1739,7 @@ var StreamTextResult = class {
   @returns an `AIStream` object.
      */
   toAIStream(callbacks) {
-    return readableFromAsyncIterable(this.textStream).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(
-      createStreamDataTransformer(callbacks == null ? void 0 : callbacks.experimental_streamData)
-    );
+    return readableFromAsyncIterable(this.textStream).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
   }
 };
 
@@ -1954,7 +1944,6 @@ function createChunkDecoder(complex) {
   };
 }
 var isStreamStringEqualToType = (type, value) => value.startsWith(`${StreamStringPrefixes[type]}:`) && value.endsWith("\n");
-var COMPLEX_HEADER = "X-Experimental-Stream-Data";
 
 // streams/ai-stream.ts
 import {
@@ -2081,7 +2070,7 @@ function readableFromAsyncIterable(iterable) {
 }
 
 // streams/stream-data.ts
-var experimental_StreamData = class {
+var StreamData = class {
   constructor() {
     this.encoder = new TextEncoder();
     this.controller = null;
@@ -2167,14 +2156,7 @@ var experimental_StreamData = class {
     this.messageAnnotations.push(value);
   }
 };
-function createStreamDataTransformer(experimental_streamData) {
-  if (!experimental_streamData) {
-    return new TransformStream({
-      transform: async (chunk, controller) => {
-        controller.enqueue(chunk);
-      }
-    });
-  }
+function createStreamDataTransformer() {
   const encoder = new TextEncoder();
   const decoder = new TextDecoder();
   return new TransformStream({
@@ -2184,6 +2166,8 @@ function createStreamDataTransformer(experimental_streamData) {
     }
   });
 }
+var experimental_StreamData = class extends StreamData {
+};
 
 // streams/anthropic-stream.ts
 function parseAnthropicStream() {
@@ -2223,16 +2207,16 @@ async function* streamable(stream) {
 }
 function AnthropicStream(res, cb) {
   if (Symbol.asyncIterator in res) {
-    return readableFromAsyncIterable(streamable(res)).pipeThrough(createCallbacksTransformer(cb)).pipeThrough(createStreamDataTransformer(cb == null ? void 0 : cb.experimental_streamData));
+    return readableFromAsyncIterable(streamable(res)).pipeThrough(createCallbacksTransformer(cb)).pipeThrough(createStreamDataTransformer());
   } else {
     return AIStream(res, parseAnthropicStream(), cb).pipeThrough(
-      createStreamDataTransformer(cb == null ? void 0 : cb.experimental_streamData)
+      createStreamDataTransformer()
     );
   }
 }
 
 // streams/assistant-response.ts
-function experimental_AssistantResponse({ threadId, messageId }, process2) {
+function AssistantResponse({ threadId, messageId }, process2) {
   const stream = new ReadableStream({
     async start(controller) {
       var _a;
@@ -2323,6 +2307,7 @@ function experimental_AssistantResponse({ threadId, messageId }, process2) {
     }
   });
 }
+var experimental_AssistantResponse = AssistantResponse;
 
 // streams/aws-bedrock-stream.ts
 async function* asDeltaIterable(response, extractTextDeltaFromChunk) {
@@ -2350,16 +2335,7 @@ function AWSBedrockAnthropicStream(response, callbacks) {
   return AWSBedrockStream(response, callbacks, (chunk) => chunk.completion);
 }
 function AWSBedrockCohereStream(response, callbacks) {
-  return AWSBedrockStream(
-    response,
-    callbacks,
-    // As of 2023-11-17, Bedrock does not support streaming for Cohere,
-    // so we take the full generation:
-    (chunk) => {
-      var _a, _b;
-      return (_b = (_a = chunk.generations) == null ? void 0 : _a[0]) == null ? void 0 : _b.text;
-    }
-  );
+  return AWSBedrockStream(response, callbacks, (chunk) => chunk == null ? void 0 : chunk.text);
 }
 function AWSBedrockLlama2Stream(response, callbacks) {
   return AWSBedrockStream(response, callbacks, (chunk) => chunk.generation);
@@ -2367,9 +2343,7 @@ function AWSBedrockLlama2Stream(response, callbacks) {
 function AWSBedrockStream(response, callbacks, extractTextDeltaFromChunk) {
   return readableFromAsyncIterable(
     asDeltaIterable(response, extractTextDeltaFromChunk)
-  ).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(
-    createStreamDataTransformer(callbacks == null ? void 0 : callbacks.experimental_streamData)
-  );
+  ).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
 }
 
 // streams/cohere-stream.ts
@@ -2424,13 +2398,9 @@ async function* streamable2(stream) {
 }
 function CohereStream(reader, callbacks) {
   if (Symbol.asyncIterator in reader) {
-    return readableFromAsyncIterable(streamable2(reader)).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(
-      createStreamDataTransformer(callbacks == null ? void 0 : callbacks.experimental_streamData)
-    );
+    return readableFromAsyncIterable(streamable2(reader)).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
   } else {
-    return createParser2(reader).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(
-      createStreamDataTransformer(callbacks == null ? void 0 : callbacks.experimental_streamData)
-    );
+    return createParser2(reader).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
   }
 }
 
@@ -2449,7 +2419,7 @@ async function* streamable3(response) {
   }
 }
 function GoogleGenerativeAIStream(response, cb) {
-  return readableFromAsyncIterable(streamable3(response)).pipeThrough(createCallbacksTransformer(cb)).pipeThrough(createStreamDataTransformer(cb == null ? void 0 : cb.experimental_streamData));
+  return readableFromAsyncIterable(streamable3(response)).pipeThrough(createCallbacksTransformer(cb)).pipeThrough(createStreamDataTransformer());
 }
 
 // streams/huggingface-stream.ts
@@ -2477,9 +2447,7 @@ function createParser3(res) {
   });
 }
 function HuggingFaceStream(res, callbacks) {
-  return createParser3(res).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(
-    createStreamDataTransformer(callbacks == null ? void 0 : callbacks.experimental_streamData)
-  );
+  return createParser3(res).pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
 }
 
 // streams/inkeep-stream.ts
@@ -2516,7 +2484,7 @@ function InkeepStream(res, callbacks) {
     }
   };
   return AIStream(res, inkeepEventParser, passThroughCallbacks).pipeThrough(
-    createStreamDataTransformer(passThroughCallbacks == null ? void 0 : passThroughCallbacks.experimental_streamData)
+    createStreamDataTransformer()
   );
 }
 
@@ -2541,9 +2509,7 @@ function LangChainStream(callbacks) {
     }
   };
   return {
-    stream: stream.readable.pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(
-      createStreamDataTransformer(callbacks == null ? void 0 : callbacks.experimental_streamData)
-    ),
+    stream: stream.readable.pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer()),
     writer,
     handlers: {
       handleLLMNewToken: async (token) => {
@@ -2594,9 +2560,7 @@ async function* streamable4(stream) {
 }
 function MistralStream(response, callbacks) {
   const stream = readableFromAsyncIterable(streamable4(response));
-  return stream.pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(
-    createStreamDataTransformer(callbacks == null ? void 0 : callbacks.experimental_streamData)
-  );
+  return stream.pipeThrough(createCallbacksTransformer(callbacks)).pipeThrough(createStreamDataTransformer());
 }
 
 // streams/openai-stream.ts
@@ -2740,9 +2704,7 @@ function OpenAIStream(res, callbacks) {
     const functionCallTransformer = createFunctionCallTransformer(cb);
     return stream.pipeThrough(functionCallTransformer);
   } else {
-    return stream.pipeThrough(
-      createStreamDataTransformer(cb == null ? void 0 : cb.experimental_streamData)
-    );
+    return stream.pipeThrough(createStreamDataTransformer());
   }
 }
 function createFunctionCallTransformer(callbacks) {
@@ -2752,7 +2714,6 @@ function createFunctionCallTransformer(callbacks) {
   let aggregatedFinalCompletionResponse = "";
   let isFunctionStreamingIn = false;
   let functionCallMessages = callbacks[__internal__OpenAIFnMessagesSymbol] || [];
-  const isComplexMode = callbacks == null ? void 0 : callbacks.experimental_streamData;
   const decode = createChunkDecoder();
   return new TransformStream({
     async transform(chunk, controller) {
@@ -2767,7 +2728,7 @@ function createFunctionCallTransformer(callbacks) {
       }
       if (!isFunctionStreamingIn) {
         controller.enqueue(
-          isComplexMode ? textEncoder.encode(formatStreamPart("text", message)) : chunk
+          textEncoder.encode(formatStreamPart("text", message))
         );
         return;
       } else {
@@ -2878,17 +2839,17 @@ function createFunctionCallTransformer(callbacks) {
           if (!functionResponse) {
             controller.enqueue(
               textEncoder.encode(
-                isComplexMode ? formatStreamPart(
+                formatStreamPart(
                   payload.function_call ? "function_call" : "tool_calls",
                   // parse to prevent double-encoding:
                   JSON.parse(aggregatedResponse)
-                ) : aggregatedResponse
+                )
               )
             );
             return;
           } else if (typeof functionResponse === "string") {
             controller.enqueue(
-              isComplexMode ? textEncoder.encode(formatStreamPart("text", functionResponse)) : textEncoder.encode(functionResponse)
+              textEncoder.encode(formatStreamPart("text", functionResponse))
             );
             aggregatedFinalCompletionResponse = functionResponse;
             return;
@@ -2938,7 +2899,7 @@ async function ReplicateStream(res, cb, options) {
     }
   });
   return AIStream(eventStream, void 0, cb).pipeThrough(
-    createStreamDataTransformer(cb == null ? void 0 : cb.experimental_streamData)
+    createStreamDataTransformer()
   );
 }
 
@@ -3099,74 +3060,41 @@ async function parseComplexResponse({
 // streams/streaming-react-response.ts
 var experimental_StreamingReactResponse = class {
   constructor(res, options) {
-    var _a;
+    var _a, _b;
     let resolveFunc = () => {
     };
     let next = new Promise((resolve) => {
       resolveFunc = resolve;
     });
-    if (options == null ? void 0 : options.data) {
-      const processedStream = res.pipeThrough(
-        options.data.stream
-      );
-      let lastPayload = void 0;
-      parseComplexResponse({
-        reader: processedStream.getReader(),
-        update: (merged, data) => {
-          var _a2, _b, _c;
-          const content2 = (_b = (_a2 = merged[0]) == null ? void 0 : _a2.content) != null ? _b : "";
-          const ui = ((_c = options == null ? void 0 : options.ui) == null ? void 0 : _c.call(options, { content: content2, data })) || content2;
-          const payload = { ui, content: content2 };
-          const resolvePrevious = resolveFunc;
-          const nextRow = new Promise((resolve) => {
-            resolveFunc = resolve;
-          });
-          resolvePrevious({
-            next: nextRow,
-            ...payload
+    const processedStream = (options == null ? void 0 : options.data) != null ? res.pipeThrough((_a = options == null ? void 0 : options.data) == null ? void 0 : _a.stream) : res;
+    let lastPayload = void 0;
+    parseComplexResponse({
+      reader: processedStream.getReader(),
+      update: (merged, data) => {
+        var _a2, _b2, _c;
+        const content = (_b2 = (_a2 = merged[0]) == null ? void 0 : _a2.content) != null ? _b2 : "";
+        const ui = ((_c = options == null ? void 0 : options.ui) == null ? void 0 : _c.call(options, { content, data })) || content;
+        const payload = { ui, content };
+        const resolvePrevious = resolveFunc;
+        const nextRow = new Promise((resolve) => {
+          resolveFunc = resolve;
+        });
+        resolvePrevious({
+          next: nextRow,
+          ...payload
+        });
+        lastPayload = payload;
+      },
+      generateId: (_b = options == null ? void 0 : options.generateId) != null ? _b : generateId,
+      onFinish: () => {
+        if (lastPayload !== void 0) {
+          resolveFunc({
+            next: null,
+            ...lastPayload
           });
-          lastPayload = payload;
-        },
-        generateId: (_a = options.generateId) != null ? _a : generateId,
-        onFinish: () => {
-          if (lastPayload !== void 0) {
-            resolveFunc({
-              next: null,
-              ...lastPayload
-            });
-          }
         }
-      });
-      return next;
-    }
-    let content = "";
-    const decode = createChunkDecoder();
-    const reader = res.getReader();
-    async function readChunk() {
-      var _a2;
-      const { done, value } = await reader.read();
-      if (!done) {
-        content += decode(value);
-      }
-      const ui = ((_a2 = options == null ? void 0 : options.ui) == null ? void 0 : _a2.call(options, { content })) || content;
-      const payload = {
-        ui,
-        content
-      };
-      const resolvePrevious = resolveFunc;
-      const nextRow = done ? null : new Promise((resolve) => {
-        resolveFunc = resolve;
-      });
-      resolvePrevious({
-        next: nextRow,
-        ...payload
-      });
-      if (done) {
-        return;
       }
-      await readChunk();
-    }
-    readChunk();
+    });
     return next;
   }
 };
@@ -3183,7 +3111,6 @@ var StreamingTextResponse = class extends Response {
       status: 200,
       headers: {
         "Content-Type": "text/plain; charset=utf-8",
-        [COMPLEX_HEADER]: data ? "true" : "false",
         ...init == null ? void 0 : init.headers
       }
     });
@@ -3215,7 +3142,7 @@ export {
   AWSBedrockLlama2Stream,
   AWSBedrockStream,
   AnthropicStream,
-  COMPLEX_HEADER,
+  AssistantResponse,
   CohereStream,
   GenerateObjectResult,
   GenerateTextResult,
@@ -3226,6 +3153,7 @@ export {
   MistralStream,
   OpenAIStream,
   ReplicateStream,
+  StreamData,
   StreamObjectResult,
   StreamTextResult,
   StreamingTextResponse,