ai 4.1.7 → 4.1.9

package/dist/index.js

@@ -68,6 +68,7 @@ __export(streams_exports, {
   experimental_customProvider: () => experimental_customProvider,
   experimental_generateImage: () => generateImage,
   experimental_wrapLanguageModel: () => experimental_wrapLanguageModel,
+  extractReasoningMiddleware: () => extractReasoningMiddleware,
   formatAssistantStreamPart: () => import_ui_utils9.formatAssistantStreamPart,
   formatDataStreamPart: () => import_ui_utils9.formatDataStreamPart,
   generateId: () => import_provider_utils14.generateId,
@@ -1484,150 +1485,6 @@ var import_provider10 = require("@ai-sdk/provider");
 var import_provider_utils4 = require("@ai-sdk/provider-utils");
 var import_zod7 = require("zod");
 
-// core/prompt/message.ts
-var import_zod6 = require("zod");
-
-// core/types/provider-metadata.ts
-var import_zod3 = require("zod");
-
-// core/types/json-value.ts
-var import_zod2 = require("zod");
-var jsonValueSchema = import_zod2.z.lazy(
-  () => import_zod2.z.union([
-    import_zod2.z.null(),
-    import_zod2.z.string(),
-    import_zod2.z.number(),
-    import_zod2.z.boolean(),
-    import_zod2.z.record(import_zod2.z.string(), jsonValueSchema),
-    import_zod2.z.array(jsonValueSchema)
-  ])
-);
-
-// core/types/provider-metadata.ts
-var providerMetadataSchema = import_zod3.z.record(
-  import_zod3.z.string(),
-  import_zod3.z.record(import_zod3.z.string(), jsonValueSchema)
-);
-
-// core/prompt/content-part.ts
-var import_zod5 = require("zod");
-
-// core/prompt/tool-result-content.ts
-var import_zod4 = require("zod");
-var toolResultContentSchema = import_zod4.z.array(
-  import_zod4.z.union([
-    import_zod4.z.object({ type: import_zod4.z.literal("text"), text: import_zod4.z.string() }),
-    import_zod4.z.object({
-      type: import_zod4.z.literal("image"),
-      data: import_zod4.z.string(),
-      mimeType: import_zod4.z.string().optional()
-    })
-  ])
-);
-
-// core/prompt/content-part.ts
-var textPartSchema = import_zod5.z.object({
-  type: import_zod5.z.literal("text"),
-  text: import_zod5.z.string(),
-  experimental_providerMetadata: providerMetadataSchema.optional()
-});
-var imagePartSchema = import_zod5.z.object({
-  type: import_zod5.z.literal("image"),
-  image: import_zod5.z.union([dataContentSchema, import_zod5.z.instanceof(URL)]),
-  mimeType: import_zod5.z.string().optional(),
-  experimental_providerMetadata: providerMetadataSchema.optional()
-});
-var filePartSchema = import_zod5.z.object({
-  type: import_zod5.z.literal("file"),
-  data: import_zod5.z.union([dataContentSchema, import_zod5.z.instanceof(URL)]),
-  mimeType: import_zod5.z.string(),
-  experimental_providerMetadata: providerMetadataSchema.optional()
-});
-var toolCallPartSchema = import_zod5.z.object({
-  type: import_zod5.z.literal("tool-call"),
-  toolCallId: import_zod5.z.string(),
-  toolName: import_zod5.z.string(),
-  args: import_zod5.z.unknown()
-});
-var toolResultPartSchema = import_zod5.z.object({
-  type: import_zod5.z.literal("tool-result"),
-  toolCallId: import_zod5.z.string(),
-  toolName: import_zod5.z.string(),
-  result: import_zod5.z.unknown(),
-  content: toolResultContentSchema.optional(),
-  isError: import_zod5.z.boolean().optional(),
-  experimental_providerMetadata: providerMetadataSchema.optional()
-});
-
-// core/prompt/message.ts
-var coreSystemMessageSchema = import_zod6.z.object({
-  role: import_zod6.z.literal("system"),
-  content: import_zod6.z.string(),
-  experimental_providerMetadata: providerMetadataSchema.optional()
-});
-var coreUserMessageSchema = import_zod6.z.object({
-  role: import_zod6.z.literal("user"),
-  content: import_zod6.z.union([
-    import_zod6.z.string(),
-    import_zod6.z.array(import_zod6.z.union([textPartSchema, imagePartSchema, filePartSchema]))
-  ]),
-  experimental_providerMetadata: providerMetadataSchema.optional()
-});
-var coreAssistantMessageSchema = import_zod6.z.object({
-  role: import_zod6.z.literal("assistant"),
-  content: import_zod6.z.union([
-    import_zod6.z.string(),
-    import_zod6.z.array(import_zod6.z.union([textPartSchema, toolCallPartSchema]))
-  ]),
-  experimental_providerMetadata: providerMetadataSchema.optional()
-});
-var coreToolMessageSchema = import_zod6.z.object({
-  role: import_zod6.z.literal("tool"),
-  content: import_zod6.z.array(toolResultPartSchema),
-  experimental_providerMetadata: providerMetadataSchema.optional()
-});
-var coreMessageSchema = import_zod6.z.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;
@@ -1812,6 +1669,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
+var import_zod6 = require("zod");
+
+// core/types/provider-metadata.ts
+var import_zod3 = require("zod");
+
+// core/types/json-value.ts
+var import_zod2 = require("zod");
+var jsonValueSchema = import_zod2.z.lazy(
+  () => import_zod2.z.union([
+    import_zod2.z.null(),
+    import_zod2.z.string(),
+    import_zod2.z.number(),
+    import_zod2.z.boolean(),
+    import_zod2.z.record(import_zod2.z.string(), jsonValueSchema),
+    import_zod2.z.array(jsonValueSchema)
+  ])
+);
+
+// core/types/provider-metadata.ts
+var providerMetadataSchema = import_zod3.z.record(
+  import_zod3.z.string(),
+  import_zod3.z.record(import_zod3.z.string(), jsonValueSchema)
+);
+
+// core/prompt/content-part.ts
+var import_zod5 = require("zod");
+
+// core/prompt/tool-result-content.ts
+var import_zod4 = require("zod");
+var toolResultContentSchema = import_zod4.z.array(
+  import_zod4.z.union([
+    import_zod4.z.object({ type: import_zod4.z.literal("text"), text: import_zod4.z.string() }),
+    import_zod4.z.object({
+      type: import_zod4.z.literal("image"),
+      data: import_zod4.z.string(),
+      mimeType: import_zod4.z.string().optional()
+    })
+  ])
+);
+
+// core/prompt/content-part.ts
+var textPartSchema = import_zod5.z.object({
+  type: import_zod5.z.literal("text"),
+  text: import_zod5.z.string(),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var imagePartSchema = import_zod5.z.object({
+  type: import_zod5.z.literal("image"),
+  image: import_zod5.z.union([dataContentSchema, import_zod5.z.instanceof(URL)]),
+  mimeType: import_zod5.z.string().optional(),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var filePartSchema = import_zod5.z.object({
+  type: import_zod5.z.literal("file"),
+  data: import_zod5.z.union([dataContentSchema, import_zod5.z.instanceof(URL)]),
+  mimeType: import_zod5.z.string(),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var toolCallPartSchema = import_zod5.z.object({
+  type: import_zod5.z.literal("tool-call"),
+  toolCallId: import_zod5.z.string(),
+  toolName: import_zod5.z.string(),
+  args: import_zod5.z.unknown()
+});
+var toolResultPartSchema = import_zod5.z.object({
+  type: import_zod5.z.literal("tool-result"),
+  toolCallId: import_zod5.z.string(),
+  toolName: import_zod5.z.string(),
+  result: import_zod5.z.unknown(),
+  content: toolResultContentSchema.optional(),
+  isError: import_zod5.z.boolean().optional(),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+
+// core/prompt/message.ts
+var coreSystemMessageSchema = import_zod6.z.object({
+  role: import_zod6.z.literal("system"),
+  content: import_zod6.z.string(),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var coreUserMessageSchema = import_zod6.z.object({
+  role: import_zod6.z.literal("user"),
+  content: import_zod6.z.union([
+    import_zod6.z.string(),
+    import_zod6.z.array(import_zod6.z.union([textPartSchema, imagePartSchema, filePartSchema]))
+  ]),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var coreAssistantMessageSchema = import_zod6.z.object({
+  role: import_zod6.z.literal("assistant"),
+  content: import_zod6.z.union([
+    import_zod6.z.string(),
+    import_zod6.z.array(import_zod6.z.union([textPartSchema, toolCallPartSchema]))
+  ]),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var coreToolMessageSchema = import_zod6.z.object({
+  role: import_zod6.z.literal("tool"),
+  content: import_zod6.z.array(toolResultPartSchema),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
+var coreMessageSchema = import_zod6.z.union([
+  coreSystemMessageSchema,
+  coreUserMessageSchema,
+  coreAssistantMessageSchema,
+  coreToolMessageSchema
+]);
+
 // core/prompt/standardize-prompt.ts
 function standardizePrompt({
   prompt,
@@ -5630,6 +5631,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,
@@ -6264,6 +6372,7 @@ var StreamData = class {
   experimental_customProvider,
   experimental_generateImage,
   experimental_wrapLanguageModel,
+  extractReasoningMiddleware,
   formatAssistantStreamPart,
   formatDataStreamPart,
   generateId,