ai 3.1.6 → 3.1.7

package/svelte/dist/index.js

@@ -525,6 +525,13 @@ var F2 = (t, e) => c.useSWR(t, e);
 // svelte/use-chat.ts
 var import_store = require("svelte/store");
 
+// shared/generate-id.ts
+var import_non_secure = require("nanoid/non-secure");
+var generateId = (0, import_non_secure.customAlphabet)(
+  "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
+  7
+);
+
 // shared/stream-parts.ts
 var textStreamPart = {
   code: "0",
@@ -621,7 +628,7 @@ var dataMessageStreamPart = {
     };
   }
 };
-var toolCallStreamPart = {
+var toolCallsStreamPart = {
   code: "7",
   name: "tool_calls",
   parse: (value) => {
@@ -648,6 +655,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,
@@ -656,8 +693,10 @@ var streamParts = [
   assistantMessageStreamPart,
   assistantControlDataStreamPart,
   dataMessageStreamPart,
+  toolCallsStreamPart,
+  messageAnnotationsStreamPart,
   toolCallStreamPart,
-  messageAnnotationsStreamPart
+  toolResultStreamPart
 ];
 var streamPartsByCode = {
   [textStreamPart.code]: textStreamPart,
@@ -667,8 +706,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,
@@ -678,8 +719,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) => {
@@ -740,13 +783,6 @@ async function* readDataStream(reader, {
   }
 }
 
-// shared/generate-id.ts
-var import_non_secure = require("nanoid/non-secure");
-var generateId = (0, import_non_secure.customAlphabet)(
-  "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
-  7
-);
-
 // shared/parse-complex-response.ts
 function assignAnnotationsToMessage(message, annotations) {
   if (!message || !annotations || !annotations.length)
@@ -784,6 +820,40 @@ async function parseComplexResponse({
         };
       }
     }
+    if (type === "tool_call") {
+      if (prefixMap.text == null) {
+        prefixMap.text = {
+          id: generateId2(),
+          role: "assistant",
+          content: "",
+          createdAt
+        };
+      }
+      if (prefixMap.text.toolInvocations == null) {
+        prefixMap.text.toolInvocations = [];
+      }
+      prefixMap.text.toolInvocations.push(value);
+    } else if (type === "tool_result") {
+      if (prefixMap.text == null) {
+        prefixMap.text = {
+          id: generateId2(),
+          role: "assistant",
+          content: "",
+          createdAt
+        };
+      }
+      if (prefixMap.text.toolInvocations == null) {
+        prefixMap.text.toolInvocations = [];
+      }
+      const toolInvocationIndex = prefixMap.text.toolInvocations.findIndex(
+        (invocation) => invocation.toolCallId === value.toolCallId
+      );
+      if (toolInvocationIndex !== -1) {
+        prefixMap.text.toolInvocations[toolInvocationIndex] = value;
+      } else {
+        prefixMap.text.toolInvocations.push(value);
+      }
+    }
     let functionCallMessage = null;
     if (type === "function_call") {
       prefixMap["function_call"] = {