ai 4.1.7 → 4.1.9

package/dist/index.mjs

@@ -1415,150 +1415,6 @@ import { InvalidPromptError } from "@ai-sdk/provider";
 import { safeValidateTypes } from "@ai-sdk/provider-utils";
 import { z as z7 } from "zod";
 
-// core/prompt/message.ts
-import { z as z6 } from "zod";
-
-// core/types/provider-metadata.ts
-import { z as z3 } from "zod";
-
-// core/types/json-value.ts
-import { z as z2 } from "zod";
-var jsonValueSchema = z2.lazy(
-  () => z2.union([
-    z2.null(),
-    z2.string(),
-    z2.number(),
-    z2.boolean(),
-    z2.record(z2.string(), jsonValueSchema),
-    z2.array(jsonValueSchema)
-  ])
-);
-
-// core/types/provider-metadata.ts
-var providerMetadataSchema = z3.record(
-  z3.string(),
-  z3.record(z3.string(), jsonValueSchema)
-);
-
-// core/prompt/content-part.ts
-import { z as z5 } from "zod";
-
-// core/prompt/tool-result-content.ts
-import { z as z4 } from "zod";
-var toolResultContentSchema = z4.array(
-  z4.union([
-    z4.object({ type: z4.literal("text"), text: z4.string() }),
-    z4.object({
-      type: z4.literal("image"),
-      data: z4.string(),
-      mimeType: z4.string().optional()
-    })
-  ])
-);
-
-// core/prompt/content-part.ts
-var textPartSchema = z5.object({
-  type: z5.literal("text"),
-  text: z5.string(),
-  experimental_providerMetadata: providerMetadataSchema.optional()
-});
-var imagePartSchema = z5.object({
-  type: z5.literal("image"),
-  image: z5.union([dataContentSchema, z5.instanceof(URL)]),
-  mimeType: z5.string().optional(),
-  experimental_providerMetadata: providerMetadataSchema.optional()
-});
-var filePartSchema = z5.object({
-  type: z5.literal("file"),
-  data: z5.union([dataContentSchema, z5.instanceof(URL)]),
-  mimeType: z5.string(),
-  experimental_providerMetadata: providerMetadataSchema.optional()
-});
-var toolCallPartSchema = z5.object({
-  type: z5.literal("tool-call"),
-  toolCallId: z5.string(),
-  toolName: z5.string(),
-  args: z5.unknown()
-});
-var toolResultPartSchema = z5.object({
-  type: z5.literal("tool-result"),
-  toolCallId: z5.string(),
-  toolName: z5.string(),
-  result: z5.unknown(),
-  content: toolResultContentSchema.optional(),
-  isError: z5.boolean().optional(),
-  experimental_providerMetadata: providerMetadataSchema.optional()
-});
-
-// core/prompt/message.ts
-var coreSystemMessageSchema = z6.object({
-  role: z6.literal("system"),
-  content: z6.string(),
-  experimental_providerMetadata: providerMetadataSchema.optional()
-});
-var coreUserMessageSchema = z6.object({
-  role: z6.literal("user"),
-  content: z6.union([
-    z6.string(),
-    z6.array(z6.union([textPartSchema, imagePartSchema, filePartSchema]))
-  ]),
-  experimental_providerMetadata: providerMetadataSchema.optional()
-});
-var coreAssistantMessageSchema = z6.object({
-  role: z6.literal("assistant"),
-  content: z6.union([
-    z6.string(),
-    z6.array(z6.union([textPartSchema, toolCallPartSchema]))
-  ]),
-  experimental_providerMetadata: providerMetadataSchema.optional()
-});
-var coreToolMessageSchema = z6.object({
-  role: z6.literal("tool"),
-  content: z6.array(toolResultPartSchema),
-  experimental_providerMetadata: providerMetadataSchema.optional()
-});
-var coreMessageSchema = z6.union([
-  coreSystemMessageSchema,
-  coreUserMessageSchema,
-  coreAssistantMessageSchema,
-  coreToolMessageSchema
-]);
-
-// core/prompt/detect-prompt-type.ts
-function detectPromptType(prompt) {
-  if (!Array.isArray(prompt)) {
-    return "other";
-  }
-  if (prompt.length === 0) {
-    return "messages";
-  }
-  const characteristics = prompt.map(detectSingleMessageCharacteristics);
-  if (characteristics.some((c) => c === "has-ui-specific-parts")) {
-    return "ui-messages";
-  } else if (characteristics.every(
-    (c) => c === "has-core-specific-parts" || c === "message"
-  )) {
-    return "messages";
-  } else {
-    return "other";
-  }
-}
-function detectSingleMessageCharacteristics(message) {
-  if (typeof message === "object" && message !== null && (message.role === "function" || // UI-only role
-  message.role === "data" || // UI-only role
-  "toolInvocations" in message || // UI-specific field
-  "experimental_attachments" in message)) {
-    return "has-ui-specific-parts";
-  } else if (typeof message === "object" && message !== null && "content" in message && (Array.isArray(message.content) || // Core messages can have array content
-  "experimental_providerMetadata" in message)) {
-    return "has-core-specific-parts";
-  } else if (typeof message === "object" && message !== null && "role" in message && "content" in message && typeof message.content === "string" && ["system", "user", "assistant", "tool"].includes(message.role)) {
-    return "message";
-  } else {
-    return "other";
-  }
-}
-
 // core/prompt/attachments-to-parts.ts
 function attachmentsToParts(attachments) {
   var _a15, _b, _c;
@@ -1743,6 +1599,150 @@ function convertToCoreMessages(messages, options) {
   return coreMessages;
 }
 
+// core/prompt/detect-prompt-type.ts
+function detectPromptType(prompt) {
+  if (!Array.isArray(prompt)) {
+    return "other";
+  }
+  if (prompt.length === 0) {
+    return "messages";
+  }
+  const characteristics = prompt.map(detectSingleMessageCharacteristics);
+  if (characteristics.some((c) => c === "has-ui-specific-parts")) {
+    return "ui-messages";
+  } else if (characteristics.every(
+    (c) => c === "has-core-specific-parts" || c === "message"
+  )) {
+    return "messages";
+  } else {
+    return "other";
+  }
+}
+function detectSingleMessageCharacteristics(message) {
+  if (typeof message === "object" && message !== null && (message.role === "function" || // UI-only role
+  message.role === "data" || // UI-only role
+  "toolInvocations" in message || // UI-specific field
+  "experimental_attachments" in message)) {
+    return "has-ui-specific-parts";
+  } else if (typeof message === "object" && message !== null && "content" in message && (Array.isArray(message.content) || // Core messages can have array content
+  "experimental_providerMetadata" in message)) {
+    return "has-core-specific-parts";
+  } else if (typeof message === "object" && message !== null && "role" in message && "content" in message && typeof message.content === "string" && ["system", "user", "assistant", "tool"].includes(message.role)) {
+    return "message";
+  } else {
+    return "other";
+  }
+}
+
+// core/prompt/message.ts
+import { z as z6 } from "zod";
+
+// core/types/provider-metadata.ts
+import { z as z3 } from "zod";
+
+// core/types/json-value.ts
+import { z as z2 } from "zod";
+var jsonValueSchema = z2.lazy(
+  () => z2.union([
+    z2.null(),
+    z2.string(),
+    z2.number(),
+    z2.boolean(),
+    z2.record(z2.string(), jsonValueSchema),
+    z2.array(jsonValueSchema)
+  ])
+);
+
+// core/types/provider-metadata.ts
+var providerMetadataSchema = z3.record(
+  z3.string(),
+  z3.record(z3.string(), jsonValueSchema)
+);
+
+// core/prompt/content-part.ts
+import { z as z5 } from "zod";
+
+// core/prompt/tool-result-content.ts
+import { z as z4 } from "zod";
+var toolResultContentSchema = z4.array(
+  z4.union([
+    z4.object({ type: z4.literal("text"), text: z4.string() }),
+    z4.object({
+      type: z4.literal("image"),
+      data: z4.string(),
+      mimeType: z4.string().optional()
+    })
+  ])
+);
+
+// core/prompt/content-part.ts
+var textPartSchema = z5.object({
+  type: z5.literal("text"),
+  text: z5.string(),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var imagePartSchema = z5.object({
+  type: z5.literal("image"),
+  image: z5.union([dataContentSchema, z5.instanceof(URL)]),
+  mimeType: z5.string().optional(),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var filePartSchema = z5.object({
+  type: z5.literal("file"),
+  data: z5.union([dataContentSchema, z5.instanceof(URL)]),
+  mimeType: z5.string(),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var toolCallPartSchema = z5.object({
+  type: z5.literal("tool-call"),
+  toolCallId: z5.string(),
+  toolName: z5.string(),
+  args: z5.unknown()
+});
+var toolResultPartSchema = z5.object({
+  type: z5.literal("tool-result"),
+  toolCallId: z5.string(),
+  toolName: z5.string(),
+  result: z5.unknown(),
+  content: toolResultContentSchema.optional(),
+  isError: z5.boolean().optional(),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+
+// core/prompt/message.ts
+var coreSystemMessageSchema = z6.object({
+  role: z6.literal("system"),
+  content: z6.string(),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var coreUserMessageSchema = z6.object({
+  role: z6.literal("user"),
+  content: z6.union([
+    z6.string(),
+    z6.array(z6.union([textPartSchema, imagePartSchema, filePartSchema]))
+  ]),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var coreAssistantMessageSchema = z6.object({
+  role: z6.literal("assistant"),
+  content: z6.union([
+    z6.string(),
+    z6.array(z6.union([textPartSchema, toolCallPartSchema]))
+  ]),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var coreToolMessageSchema = z6.object({
+  role: z6.literal("tool"),
+  content: z6.array(toolResultPartSchema),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var coreMessageSchema = z6.union([
+  coreSystemMessageSchema,
+  coreUserMessageSchema,
+  coreAssistantMessageSchema,
+  coreToolMessageSchema
+]);
+
 // core/prompt/standardize-prompt.ts
 function standardizePrompt({
   prompt,
@@ -5584,6 +5584,113 @@ var experimental_wrapLanguageModel = ({
   };
 };
 
+// core/util/get-potential-start-index.ts
+function getPotentialStartIndex(text2, searchedText) {
+  if (searchedText.length === 0) {
+    return null;
+  }
+  const directIndex = text2.indexOf(searchedText);
+  if (directIndex !== -1) {
+    return directIndex;
+  }
+  for (let i = text2.length - 1; i >= 0; i--) {
+    const suffix = text2.substring(i);
+    if (searchedText.startsWith(suffix)) {
+      return i;
+    }
+  }
+  return null;
+}
+
+// core/middleware/extract-reasoning-middleware.ts
+function extractReasoningMiddleware({
+  tagName,
+  separator = "\n"
+}) {
+  const openingTag = `<${tagName}>`;
+  const closingTag = `</${tagName}>`;
+  return {
+    wrapGenerate: async ({ doGenerate }) => {
+      const { text: text2, ...rest } = await doGenerate();
+      if (text2 == null) {
+        return { text: text2, ...rest };
+      }
+      const regexp = new RegExp(`${openingTag}(.*?)${closingTag}`, "gs");
+      const matches = Array.from(text2.matchAll(regexp));
+      if (!matches.length) {
+        return { text: text2, ...rest };
+      }
+      const reasoning = matches.map((match) => match[1]).join(separator);
+      let textWithoutReasoning = text2;
+      for (let i = matches.length - 1; i >= 0; i--) {
+        const match = matches[i];
+        const beforeMatch = textWithoutReasoning.slice(0, match.index);
+        const afterMatch = textWithoutReasoning.slice(
+          match.index + match[0].length
+        );
+        textWithoutReasoning = beforeMatch + (beforeMatch.length > 0 && afterMatch.length > 0 ? separator : "") + afterMatch;
+      }
+      return { text: textWithoutReasoning, reasoning, ...rest };
+    },
+    wrapStream: async ({ doStream }) => {
+      const { stream, ...rest } = await doStream();
+      let isFirstReasoning = true;
+      let isFirstText = true;
+      let afterSwitch = false;
+      let isReasoning = false;
+      let buffer = "";
+      return {
+        stream: stream.pipeThrough(
+          new TransformStream({
+            transform: (chunk, controller) => {
+              if (chunk.type !== "text-delta") {
+                controller.enqueue(chunk);
+                return;
+              }
+              buffer += chunk.textDelta;
+              function publish(text2) {
+                if (text2.length > 0) {
+                  const prefix = afterSwitch && (isReasoning ? !isFirstReasoning : !isFirstText) ? separator : "";
+                  controller.enqueue({
+                    type: isReasoning ? "reasoning" : "text-delta",
+                    textDelta: prefix + text2
+                  });
+                  afterSwitch = false;
+                  if (isReasoning) {
+                    isFirstReasoning = false;
+                  } else {
+                    isFirstText = false;
+                  }
+                }
+              }
+              do {
+                const nextTag = isReasoning ? closingTag : openingTag;
+                const startIndex = getPotentialStartIndex(buffer, nextTag);
+                if (startIndex == null) {
+                  publish(buffer);
+                  buffer = "";
+                  break;
+                }
+                publish(buffer.slice(0, startIndex));
+                const foundFullMatch = startIndex + nextTag.length <= buffer.length;
+                if (foundFullMatch) {
+                  buffer = buffer.slice(startIndex + nextTag.length);
+                  isReasoning = !isReasoning;
+                  afterSwitch = true;
+                } else {
+                  buffer = buffer.slice(startIndex);
+                  break;
+                }
+              } while (true);
+            }
+          })
+        ),
+        ...rest
+      };
+    }
+  };
+}
+
 // core/prompt/append-client-message.ts
 function appendClientMessage({
   messages,
@@ -6219,6 +6326,7 @@ export {
   experimental_customProvider,
   generateImage as experimental_generateImage,
   experimental_wrapLanguageModel,
+  extractReasoningMiddleware,
   formatAssistantStreamPart,
   formatDataStreamPart3 as formatDataStreamPart,
   generateId2 as generateId,