ai 3.1.5 → 3.1.7

package/rsc/dist/rsc-server.mjs

@@ -204,7 +204,15 @@ function convertDataContentToUint8Array(content) {
     return content;
   }
   if (typeof content === "string") {
-    return convertBase64ToUint8Array(content);
+    try {
+      return convertBase64ToUint8Array(content);
+    } catch (error) {
+      throw new InvalidDataContentError({
+        message: "Invalid data content. Content string is not a base64-encoded image.",
+        content,
+        cause: error
+      });
+    }
   }
   if (content instanceof ArrayBuffer) {
     return new Uint8Array(content);
@@ -588,7 +596,7 @@ var dataMessageStreamPart = {
     };
   }
 };
-var toolCallStreamPart = {
+var toolCallsStreamPart = {
   code: "7",
   name: "tool_calls",
   parse: (value) => {
@@ -615,6 +623,36 @@ var messageAnnotationsStreamPart = {
     return { type: "message_annotations", value };
   }
 };
+var toolCallStreamPart = {
+  code: "9",
+  name: "tool_call",
+  parse: (value) => {
+    if (value == null || typeof value !== "object" || !("toolCallId" in value) || typeof value.toolCallId !== "string" || !("toolName" in value) || typeof value.toolName !== "string" || !("args" in value) || typeof value.args !== "object") {
+      throw new Error(
+        '"tool_call" parts expect an object with a "toolCallId", "toolName", and "args" property.'
+      );
+    }
+    return {
+      type: "tool_call",
+      value
+    };
+  }
+};
+var toolResultStreamPart = {
+  code: "a",
+  name: "tool_result",
+  parse: (value) => {
+    if (value == null || typeof value !== "object" || !("toolCallId" in value) || typeof value.toolCallId !== "string" || !("toolName" in value) || typeof value.toolName !== "string" || !("args" in value) || typeof value.args !== "object" || !("result" in value)) {
+      throw new Error(
+        '"tool_result" parts expect an object with a "toolCallId", "toolName", "args", and "result" property.'
+      );
+    }
+    return {
+      type: "tool_result",
+      value
+    };
+  }
+};
 var streamParts = [
   textStreamPart,
   functionCallStreamPart,
@@ -623,8 +661,10 @@ var streamParts = [
   assistantMessageStreamPart,
   assistantControlDataStreamPart,
   dataMessageStreamPart,
+  toolCallsStreamPart,
+  messageAnnotationsStreamPart,
   toolCallStreamPart,
-  messageAnnotationsStreamPart
+  toolResultStreamPart
 ];
 var streamPartsByCode = {
   [textStreamPart.code]: textStreamPart,
@@ -634,8 +674,10 @@ var streamPartsByCode = {
   [assistantMessageStreamPart.code]: assistantMessageStreamPart,
   [assistantControlDataStreamPart.code]: assistantControlDataStreamPart,
   [dataMessageStreamPart.code]: dataMessageStreamPart,
+  [toolCallsStreamPart.code]: toolCallsStreamPart,
+  [messageAnnotationsStreamPart.code]: messageAnnotationsStreamPart,
   [toolCallStreamPart.code]: toolCallStreamPart,
-  [messageAnnotationsStreamPart.code]: messageAnnotationsStreamPart
+  [toolResultStreamPart.code]: toolResultStreamPart
 };
 var StreamStringPrefixes = {
   [textStreamPart.name]: textStreamPart.code,
@@ -645,8 +687,10 @@ var StreamStringPrefixes = {
   [assistantMessageStreamPart.name]: assistantMessageStreamPart.code,
   [assistantControlDataStreamPart.name]: assistantControlDataStreamPart.code,
   [dataMessageStreamPart.name]: dataMessageStreamPart.code,
+  [toolCallsStreamPart.name]: toolCallsStreamPart.code,
+  [messageAnnotationsStreamPart.name]: messageAnnotationsStreamPart.code,
   [toolCallStreamPart.name]: toolCallStreamPart.code,
-  [messageAnnotationsStreamPart.name]: messageAnnotationsStreamPart.code
+  [toolResultStreamPart.name]: toolResultStreamPart.code
 };
 var validCodes = streamParts.map((part) => part.code);
 var parseStreamPart = (line) => {
@@ -1170,7 +1214,7 @@ function createStreamableUI(initialValue) {
     }
   }
   warnUnclosedStream();
-  return {
+  const streamable2 = {
     /**
      * The value of the streamable UI. This can be returned from a Server Action and received by the client.
      */
@@ -1182,7 +1226,7 @@ function createStreamableUI(initialValue) {
       assertStream(".update()");
       if (value === currentValue) {
         warnUnclosedStream();
-        return;
+        return streamable2;
       }
       const resolvable = createResolvablePromise();
       currentValue = value;
@@ -1190,6 +1234,7 @@ function createStreamableUI(initialValue) {
       resolve = resolvable.resolve;
       reject = resolvable.reject;
       warnUnclosedStream();
+      return streamable2;
     },
     /**
      * This method is used to append a new UI node to the end of the old one.
@@ -1215,6 +1260,7 @@ function createStreamableUI(initialValue) {
       resolve = resolvable.resolve;
       reject = resolvable.reject;
       warnUnclosedStream();
+      return streamable2;
     },
     /**
      * This method is used to signal that there is an error in the UI stream.
@@ -1227,6 +1273,7 @@ function createStreamableUI(initialValue) {
       }
       closed = true;
       reject(error);
+      return streamable2;
     },
     /**
      * This method marks the UI node as finalized. You can either call it without any parameters or with a new UI node as the final state.
@@ -1242,11 +1289,13 @@ function createStreamableUI(initialValue) {
       closed = true;
       if (args.length) {
         resolve({ value: args[0], done: true });
-        return;
+        return streamable2;
       }
       resolve({ value: currentValue, done: true });
+      return streamable2;
     }
   };
+  return streamable2;
 }
 var STREAMABLE_VALUE_INTERNAL_LOCK = Symbol("streamable.value.lock");
 function createStreamableValue(initialValue) {
@@ -1343,7 +1392,7 @@ function createStreamableValueImpl(initialValue) {
     }
     currentValue = value;
   }
-  return {
+  const streamable2 = {
     /**
      * @internal This is an internal lock to prevent the value from being
      * updated by the user.
@@ -1370,6 +1419,7 @@ function createStreamableValueImpl(initialValue) {
       currentPromise = resolvable.promise;
       resolvePrevious(createWrapped());
       warnUnclosedStream();
+      return streamable2;
     },
     /**
      * This method is used to append a delta string to the current value. It
@@ -1407,6 +1457,7 @@ function createStreamableValueImpl(initialValue) {
       currentPromise = resolvable.promise;
       resolvePrevious(createWrapped());
       warnUnclosedStream();
+      return streamable2;
     },
     /**
      * This method is used to signal that there is an error in the value stream.
@@ -1422,6 +1473,7 @@ function createStreamableValueImpl(initialValue) {
       currentError = error;
       currentPromise = void 0;
       resolvable.resolve({ error });
+      return streamable2;
     },
     /**
      * This method marks the value as finalized. You can either call it without
@@ -1441,11 +1493,13 @@ function createStreamableValueImpl(initialValue) {
       if (args.length) {
         updateValueStates(args[0]);
         resolvable.resolve(createWrapped());
-        return;
+        return streamable2;
       }
       resolvable.resolve({});
+      return streamable2;
     }
   };
+  return streamable2;
 }
 function render(options) {
   const ui = createStreamableUI(options.initial);