ai 4.1.43 → 4.1.45

package/dist/index.mjs

@@ -48,6 +48,9 @@ function createDataStream({
       writeMessageAnnotation(annotation) {
         safeEnqueue(formatDataStreamPart("message_annotations", [annotation]));
       },
+      writeSource(source) {
+        safeEnqueue(formatDataStreamPart("source", source));
+      },
       merge(streamArg) {
         ongoingStreamPromises.push(
           (async () => {
@@ -5793,17 +5796,19 @@ function getPotentialStartIndex(text2, searchedText) {
 // core/middleware/extract-reasoning-middleware.ts
 function extractReasoningMiddleware({
   tagName,
-  separator = "\n"
+  separator = "\n",
+  startWithReasoning = false
 }) {
   const openingTag = `<${tagName}>`;
   const closingTag = `</${tagName}>`;
   return {
     middlewareVersion: "v1",
     wrapGenerate: async ({ doGenerate }) => {
-      const { text: text2, ...rest } = await doGenerate();
-      if (text2 == null) {
-        return { text: text2, ...rest };
+      const { text: rawText, ...rest } = await doGenerate();
+      if (rawText == null) {
+        return { text: rawText, ...rest };
       }
+      const text2 = startWithReasoning ? openingTag + rawText : rawText;
       const regexp = new RegExp(`${openingTag}(.*?)${closingTag}`, "gs");
       const matches = Array.from(text2.matchAll(regexp));
       if (!matches.length) {
@@ -5826,7 +5831,7 @@ function extractReasoningMiddleware({
       let isFirstReasoning = true;
       let isFirstText = true;
       let afterSwitch = false;
-      let isReasoning = false;
+      let isReasoning = startWithReasoning;
       let buffer = "";
       return {
         stream: stream.pipeThrough(