ai 4.1.7 → 4.1.8

package/dist/index.mjs

@@ -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,