ai 4.1.43 → 4.1.44

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # ai
 
+## 4.1.44
+
+### Patch Changes
+
+- f8e7df2: fix (ai/core): add `startWithReasoning` option to `extractReasoningMiddleware`
+
 ## 4.1.43
 
 ### Patch Changes

package/dist/index.d.mts

@@ -2665,10 +2665,12 @@ type Experimental_LanguageModelV1Middleware = LanguageModelV1Middleware;
  *
  * @param tagName - The name of the XML tag to extract reasoning from.
  * @param separator - The separator to use between reasoning and text sections.
+ * @param startWithReasoning - Whether to start with reasoning tokens.
  */
-declare function extractReasoningMiddleware({ tagName, separator, }: {
+declare function extractReasoningMiddleware({ tagName, separator, startWithReasoning, }: {
     tagName: string;
     separator?: string;
+    startWithReasoning?: boolean;
 }): LanguageModelV1Middleware;
 
 /**

package/dist/index.d.ts

@@ -2665,10 +2665,12 @@ type Experimental_LanguageModelV1Middleware = LanguageModelV1Middleware;
  *
  * @param tagName - The name of the XML tag to extract reasoning from.
  * @param separator - The separator to use between reasoning and text sections.
+ * @param startWithReasoning - Whether to start with reasoning tokens.
  */
-declare function extractReasoningMiddleware({ tagName, separator, }: {
+declare function extractReasoningMiddleware({ tagName, separator, startWithReasoning, }: {
     tagName: string;
     separator?: string;
+    startWithReasoning?: boolean;
 }): LanguageModelV1Middleware;
 
 /**

package/dist/index.js

@@ -5839,17 +5839,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) {
@@ -5872,7 +5874,7 @@ function extractReasoningMiddleware({
       let isFirstReasoning = true;
       let isFirstText = true;
       let afterSwitch = false;
-      let isReasoning = false;
+      let isReasoning = startWithReasoning;
       let buffer = "";
       return {
         stream: stream.pipeThrough(