@ai-sdk-tool/middleware 0.0.0

package/dist/chunk-Q2TVVB35.js

@@ -0,0 +1,210 @@
+// src/reasoning-parser/index.ts
+function getPotentialStartIndex(text, searchedText) {
+  if (searchedText.length === 0) {
+    return null;
+  }
+  const directIndex = text.indexOf(searchedText);
+  if (directIndex !== -1) {
+    return directIndex;
+  }
+  for (let i = text.length - 1; i >= 0; i -= 1) {
+    const suffix = text.substring(i);
+    if (searchedText.startsWith(suffix)) {
+      return i;
+    }
+  }
+  return null;
+}
+function extractReasoningMiddleware({
+  openingTag,
+  closingTag,
+  separator = "\n",
+  startWithReasoning = false
+}) {
+  function processTextPart(text, transformedContent) {
+    var _a;
+    const regexp = new RegExp(`${openingTag}(.*?)${closingTag}`, "gs");
+    const matches = Array.from(text.matchAll(regexp));
+    if (!matches.length) {
+      return;
+    }
+    const reasoningText = matches.map((match) => match[1]).join(separator);
+    let textWithoutReasoning = text;
+    for (let i = matches.length - 1; i >= 0; i -= 1) {
+      const match = matches[i];
+      const beforeMatch = textWithoutReasoning.slice(0, match.index);
+      const matchIndex = (_a = match.index) != null ? _a : 0;
+      const afterMatch = textWithoutReasoning.slice(
+        matchIndex + match[0].length
+      );
+      textWithoutReasoning = beforeMatch + (beforeMatch.length > 0 && afterMatch.length > 0 ? separator : "") + afterMatch;
+    }
+    transformedContent.push({
+      type: "reasoning",
+      text: reasoningText
+    });
+    transformedContent.push({
+      type: "text",
+      text: textWithoutReasoning
+    });
+  }
+  return {
+    specificationVersion: "v3",
+    wrapGenerate: async ({ doGenerate }) => {
+      const { content, ...rest } = await doGenerate();
+      const transformedContent = [];
+      for (const part of content) {
+        if (part.type !== "text") {
+          transformedContent.push(part);
+          continue;
+        }
+        const text = startWithReasoning ? openingTag + part.text : part.text;
+        const regexp = new RegExp(`${openingTag}(.*?)${closingTag}`, "gs");
+        const matches = Array.from(text.matchAll(regexp));
+        if (!matches.length) {
+          transformedContent.push(part);
+          continue;
+        }
+        processTextPart(text, transformedContent);
+      }
+      return { content: transformedContent, ...rest };
+    },
+    wrapStream: async ({ doStream }) => {
+      const { stream, ...rest } = await doStream();
+      const reasoningExtractions = {};
+      function createPublisher(activeExtraction, controller) {
+        return (text) => {
+          if (text.length === 0) {
+            return;
+          }
+          const prefix = getPrefix(activeExtraction);
+          enqueueReasoningStart(activeExtraction, controller);
+          enqueueDelta(activeExtraction, controller, prefix, text);
+          updateExtractionState(activeExtraction);
+        };
+      }
+      function getPrefix(activeExtraction) {
+        return activeExtraction.afterSwitch && (activeExtraction.isReasoning ? !activeExtraction.isFirstReasoning : !activeExtraction.isFirstText) ? separator : "";
+      }
+      function enqueueReasoningStart(activeExtraction, controller) {
+        if (activeExtraction.afterSwitch && activeExtraction.isReasoning || activeExtraction.isFirstReasoning) {
+          controller.enqueue({
+            type: "reasoning-start",
+            id: `reasoning-${activeExtraction.idCounter}`
+          });
+        }
+      }
+      function enqueueDelta(activeExtraction, controller, prefix, text) {
+        controller.enqueue(
+          activeExtraction.isReasoning ? {
+            type: "reasoning-delta",
+            delta: prefix + text,
+            id: `reasoning-${activeExtraction.idCounter}`
+          } : {
+            type: "text-delta",
+            delta: prefix + text,
+            id: activeExtraction.textId
+          }
+        );
+      }
+      function updateExtractionState(activeExtraction) {
+        activeExtraction.afterSwitch = false;
+        if (activeExtraction.isReasoning) {
+          activeExtraction.isFirstReasoning = false;
+        } else {
+          activeExtraction.isFirstText = false;
+        }
+      }
+      function handleFullMatch(activeExtraction, controller, startIndex, nextTag) {
+        activeExtraction.buffer = activeExtraction.buffer.slice(
+          startIndex + nextTag.length
+        );
+        if (activeExtraction.isReasoning) {
+          controller.enqueue({
+            type: "reasoning-end",
+            id: `reasoning-${activeExtraction.idCounter}`
+          });
+          activeExtraction.idCounter += 1;
+        }
+        activeExtraction.isReasoning = !activeExtraction.isReasoning;
+        activeExtraction.afterSwitch = true;
+      }
+      function processTagMatch({
+        activeExtraction,
+        controller,
+        publish,
+        startIndex,
+        nextTag
+      }) {
+        publish(activeExtraction.buffer.slice(0, startIndex));
+        const foundFullMatch = startIndex + nextTag.length <= activeExtraction.buffer.length;
+        if (foundFullMatch) {
+          handleFullMatch(activeExtraction, controller, startIndex, nextTag);
+          return true;
+        }
+        activeExtraction.buffer = activeExtraction.buffer.slice(startIndex);
+        return false;
+      }
+      function processBuffer(activeExtraction, controller) {
+        const publish = createPublisher(activeExtraction, controller);
+        let continueProcessing = true;
+        while (continueProcessing) {
+          const nextTag = activeExtraction.isReasoning ? closingTag : openingTag;
+          const startIndex = getPotentialStartIndex(
+            activeExtraction.buffer,
+            nextTag
+          );
+          if (startIndex == null) {
+            publish(activeExtraction.buffer);
+            activeExtraction.buffer = "";
+            break;
+          }
+          continueProcessing = processTagMatch({
+            activeExtraction,
+            controller,
+            publish,
+            startIndex,
+            nextTag
+          });
+        }
+      }
+      return {
+        stream: stream.pipeThrough(
+          new TransformStream({
+            transform: (chunk, controller) => {
+              if (chunk.type !== "text-delta") {
+                controller.enqueue(chunk);
+                return;
+              }
+              if (reasoningExtractions[chunk.id] == null) {
+                reasoningExtractions[chunk.id] = {
+                  isFirstReasoning: true,
+                  isFirstText: true,
+                  afterSwitch: false,
+                  isReasoning: startWithReasoning,
+                  buffer: "",
+                  idCounter: 0,
+                  textId: chunk.id
+                };
+              }
+              const activeExtraction = reasoningExtractions[chunk.id];
+              activeExtraction.buffer += chunk.delta;
+              processBuffer(activeExtraction, controller);
+            }
+          })
+        ),
+        ...rest
+      };
+    }
+  };
+}
+
+export {
+  getPotentialStartIndex,
+  extractReasoningMiddleware
+};
+/**
+ * @license
+ * Copyright (c) 2021-present, FriendliAI Inc. All rights reserved.
+ */
+//# sourceMappingURL=chunk-Q2TVVB35.js.map

package/dist/chunk-Q2TVVB35.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/reasoning-parser/index.ts"],"sourcesContent":["/**\n * @license\n * Copyright (c) 2021-present, FriendliAI Inc. All rights reserved.\n */\n\nimport type {\n  LanguageModelV3Content,\n  LanguageModelV3Middleware,\n  LanguageModelV3StreamPart,\n} from \"@ai-sdk/provider\";\n\n/**\n * All code below is forked from the following link:\n * https://github.com/vercel/ai/blob/v5/packages/ai/core/middleware/extract-reasoning-middleware.ts\n */\n\n/**\n * Returns the index of the start of the searchedText in the text, or null if it\n * is not found.\n */\nexport function getPotentialStartIndex(\n  text: string,\n  searchedText: string\n): number | null {\n  // Return null immediately if searchedText is empty.\n  if (searchedText.length === 0) {\n    return null;\n  }\n\n  // Check if the searchedText exists as a direct substring of text.\n  const directIndex = text.indexOf(searchedText);\n  if (directIndex !== -1) {\n    return directIndex;\n  }\n\n  // Otherwise, look for the largest suffix of \"text\" that matches\n  // a prefix of \"searchedText\". We go from the end of text inward.\n  for (let i = text.length - 1; i >= 0; i -= 1) {\n    const suffix = text.substring(i);\n    if (searchedText.startsWith(suffix)) {\n      return i;\n    }\n  }\n\n  return null;\n}\n\n/**\n * Extract an XML-tagged reasoning section from the generated text and exposes it\n * as a `reasoning` property on the result.\n *\n * @param openingTag - The opening XML tag to extract reasoning from.\n * @param closingTag - The closing XML tag to extract reasoning from.\n * @param separator - The separator to use between reasoning and text sections.\n * @param startWithReasoning - Whether to start with reasoning tokens.\n */\nexport function extractReasoningMiddleware({\n  openingTag,\n  closingTag,\n  separator = \"\\n\",\n  startWithReasoning = false,\n}: {\n  openingTag: string;\n  closingTag: string;\n  separator?: string;\n  startWithReasoning?: boolean;\n}): LanguageModelV3Middleware {\n  function processTextPart(\n    text: string,\n    transformedContent: LanguageModelV3Content[]\n  ) {\n    const regexp = new RegExp(`${openingTag}(.*?)${closingTag}`, \"gs\");\n    const matches = Array.from(text.matchAll(regexp));\n\n    if (!matches.length) {\n      return;\n    }\n\n    const reasoningText = matches.map((match) => match[1]).join(separator);\n\n    let textWithoutReasoning = text;\n    for (let i = matches.length - 1; i >= 0; i -= 1) {\n      const match = matches[i];\n\n      const beforeMatch = textWithoutReasoning.slice(0, match.index);\n      const matchIndex = match.index ?? 0;\n      const afterMatch = textWithoutReasoning.slice(\n        matchIndex + match[0].length\n      );\n\n      textWithoutReasoning =\n        beforeMatch +\n        (beforeMatch.length > 0 && afterMatch.length > 0 ? separator : \"\") +\n        afterMatch;\n    }\n\n    transformedContent.push({\n      type: \"reasoning\",\n      text: reasoningText,\n    });\n\n    transformedContent.push({\n      type: \"text\",\n      text: textWithoutReasoning,\n    });\n  }\n\n  return {\n    specificationVersion: \"v3\",\n    wrapGenerate: async ({ doGenerate }) => {\n      const { content, ...rest } = await doGenerate();\n\n      const transformedContent: LanguageModelV3Content[] = [];\n      for (const part of content) {\n        if (part.type !== \"text\") {\n          transformedContent.push(part);\n          continue;\n        }\n\n        const text = startWithReasoning ? openingTag + part.text : part.text;\n        const regexp = new RegExp(`${openingTag}(.*?)${closingTag}`, \"gs\");\n        const matches = Array.from(text.matchAll(regexp));\n\n        if (!matches.length) {\n          transformedContent.push(part);\n          continue;\n        }\n\n        processTextPart(text, transformedContent);\n      }\n\n      return { content: transformedContent, ...rest };\n    },\n\n    wrapStream: async ({ doStream }) => {\n      const { stream, ...rest } = await doStream();\n\n      interface ExtractionState {\n        isFirstReasoning: boolean;\n        isFirstText: boolean;\n        afterSwitch: boolean;\n        isReasoning: boolean;\n        buffer: string;\n        idCounter: number;\n        textId: string;\n      }\n\n      const reasoningExtractions: Record<string, ExtractionState> = {};\n\n      function createPublisher(\n        activeExtraction: ExtractionState,\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>\n      ) {\n        return (text: string) => {\n          if (text.length === 0) {\n            return;\n          }\n\n          const prefix = getPrefix(activeExtraction);\n          enqueueReasoningStart(activeExtraction, controller);\n          enqueueDelta(activeExtraction, controller, prefix, text);\n          updateExtractionState(activeExtraction);\n        };\n      }\n\n      function getPrefix(activeExtraction: ExtractionState): string {\n        return activeExtraction.afterSwitch &&\n          (activeExtraction.isReasoning\n            ? !activeExtraction.isFirstReasoning\n            : !activeExtraction.isFirstText)\n          ? separator\n          : \"\";\n      }\n\n      function enqueueReasoningStart(\n        activeExtraction: ExtractionState,\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>\n      ) {\n        if (\n          (activeExtraction.afterSwitch && activeExtraction.isReasoning) ||\n          activeExtraction.isFirstReasoning\n        ) {\n          controller.enqueue({\n            type: \"reasoning-start\",\n            id: `reasoning-${activeExtraction.idCounter}`,\n          });\n        }\n      }\n\n      function enqueueDelta(\n        activeExtraction: ExtractionState,\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>,\n        prefix: string,\n        text: string\n      ) {\n        controller.enqueue(\n          activeExtraction.isReasoning\n            ? {\n                type: \"reasoning-delta\",\n                delta: prefix + text,\n                id: `reasoning-${activeExtraction.idCounter}`,\n              }\n            : {\n                type: \"text-delta\",\n                delta: prefix + text,\n                id: activeExtraction.textId,\n              }\n        );\n      }\n\n      function updateExtractionState(activeExtraction: ExtractionState) {\n        activeExtraction.afterSwitch = false;\n        if (activeExtraction.isReasoning) {\n          activeExtraction.isFirstReasoning = false;\n        } else {\n          activeExtraction.isFirstText = false;\n        }\n      }\n\n      function handleFullMatch(\n        activeExtraction: ExtractionState,\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>,\n        startIndex: number,\n        nextTag: string\n      ) {\n        activeExtraction.buffer = activeExtraction.buffer.slice(\n          startIndex + nextTag.length\n        );\n\n        if (activeExtraction.isReasoning) {\n          controller.enqueue({\n            type: \"reasoning-end\",\n            id: `reasoning-${activeExtraction.idCounter}`,\n          });\n          activeExtraction.idCounter += 1;\n        }\n\n        activeExtraction.isReasoning = !activeExtraction.isReasoning;\n        activeExtraction.afterSwitch = true;\n      }\n\n      function processTagMatch({\n        activeExtraction,\n        controller,\n        publish,\n        startIndex,\n        nextTag,\n      }: {\n        activeExtraction: ExtractionState;\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>;\n        publish: (text: string) => void;\n        startIndex: number;\n        nextTag: string;\n      }): boolean {\n        publish(activeExtraction.buffer.slice(0, startIndex));\n\n        const foundFullMatch =\n          startIndex + nextTag.length <= activeExtraction.buffer.length;\n\n        if (foundFullMatch) {\n          handleFullMatch(activeExtraction, controller, startIndex, nextTag);\n          return true;\n        }\n\n        activeExtraction.buffer = activeExtraction.buffer.slice(startIndex);\n        return false;\n      }\n\n      function processBuffer(\n        activeExtraction: ExtractionState,\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>\n      ) {\n        const publish = createPublisher(activeExtraction, controller);\n        let continueProcessing = true;\n\n        while (continueProcessing) {\n          const nextTag = activeExtraction.isReasoning\n            ? closingTag\n            : openingTag;\n          const startIndex = getPotentialStartIndex(\n            activeExtraction.buffer,\n            nextTag\n          );\n\n          if (startIndex == null) {\n            publish(activeExtraction.buffer);\n            activeExtraction.buffer = \"\";\n            break;\n          }\n\n          continueProcessing = processTagMatch({\n            activeExtraction,\n            controller,\n            publish,\n            startIndex,\n            nextTag,\n          });\n        }\n      }\n\n      return {\n        stream: stream.pipeThrough(\n          new TransformStream<\n            LanguageModelV3StreamPart,\n            LanguageModelV3StreamPart\n          >({\n            transform: (chunk, controller) => {\n              if (chunk.type !== \"text-delta\") {\n                controller.enqueue(chunk);\n                return;\n              }\n\n              if (reasoningExtractions[chunk.id] == null) {\n                reasoningExtractions[chunk.id] = {\n                  isFirstReasoning: true,\n                  isFirstText: true,\n                  afterSwitch: false,\n                  isReasoning: startWithReasoning,\n                  buffer: \"\",\n                  idCounter: 0,\n                  textId: chunk.id,\n                };\n              }\n\n              const activeExtraction = reasoningExtractions[chunk.id];\n              activeExtraction.buffer += chunk.delta;\n              processBuffer(activeExtraction, controller);\n            },\n          })\n        ),\n        ...rest,\n      };\n    },\n  };\n}\n"],"mappings":";AAoBO,SAAS,uBACd,MACA,cACe;AAEf,MAAI,aAAa,WAAW,GAAG;AAC7B,WAAO;AAAA,EACT;AAGA,QAAM,cAAc,KAAK,QAAQ,YAAY;AAC7C,MAAI,gBAAgB,IAAI;AACtB,WAAO;AAAA,EACT;AAIA,WAAS,IAAI,KAAK,SAAS,GAAG,KAAK,GAAG,KAAK,GAAG;AAC5C,UAAM,SAAS,KAAK,UAAU,CAAC;AAC/B,QAAI,aAAa,WAAW,MAAM,GAAG;AACnC,aAAO;AAAA,IACT;AAAA,EACF;AAEA,SAAO;AACT;AAWO,SAAS,2BAA2B;AAAA,EACzC;AAAA,EACA;AAAA,EACA,YAAY;AAAA,EACZ,qBAAqB;AACvB,GAK8B;AAC5B,WAAS,gBACP,MACA,oBACA;AAtEJ;AAuEI,UAAM,SAAS,IAAI,OAAO,GAAG,UAAU,QAAQ,UAAU,IAAI,IAAI;AACjE,UAAM,UAAU,MAAM,KAAK,KAAK,SAAS,MAAM,CAAC;AAEhD,QAAI,CAAC,QAAQ,QAAQ;AACnB;AAAA,IACF;AAEA,UAAM,gBAAgB,QAAQ,IAAI,CAAC,UAAU,MAAM,CAAC,CAAC,EAAE,KAAK,SAAS;AAErE,QAAI,uBAAuB;AAC3B,aAAS,IAAI,QAAQ,SAAS,GAAG,KAAK,GAAG,KAAK,GAAG;AAC/C,YAAM,QAAQ,QAAQ,CAAC;AAEvB,YAAM,cAAc,qBAAqB,MAAM,GAAG,MAAM,KAAK;AAC7D,YAAM,cAAa,WAAM,UAAN,YAAe;AAClC,YAAM,aAAa,qBAAqB;AAAA,QACtC,aAAa,MAAM,CAAC,EAAE;AAAA,MACxB;AAEA,6BACE,eACC,YAAY,SAAS,KAAK,WAAW,SAAS,IAAI,YAAY,MAC/D;AAAA,IACJ;AAEA,uBAAmB,KAAK;AAAA,MACtB,MAAM;AAAA,MACN,MAAM;AAAA,IACR,CAAC;AAED,uBAAmB,KAAK;AAAA,MACtB,MAAM;AAAA,MACN,MAAM;AAAA,IACR,CAAC;AAAA,EACH;AAEA,SAAO;AAAA,IACL,sBAAsB;AAAA,IACtB,cAAc,OAAO,EAAE,WAAW,MAAM;AACtC,YAAM,EAAE,SAAS,GAAG,KAAK,IAAI,MAAM,WAAW;AAE9C,YAAM,qBAA+C,CAAC;AACtD,iBAAW,QAAQ,SAAS;AAC1B,YAAI,KAAK,SAAS,QAAQ;AACxB,6BAAmB,KAAK,IAAI;AAC5B;AAAA,QACF;AAEA,cAAM,OAAO,qBAAqB,aAAa,KAAK,OAAO,KAAK;AAChE,cAAM,SAAS,IAAI,OAAO,GAAG,UAAU,QAAQ,UAAU,IAAI,IAAI;AACjE,cAAM,UAAU,MAAM,KAAK,KAAK,SAAS,MAAM,CAAC;AAEhD,YAAI,CAAC,QAAQ,QAAQ;AACnB,6BAAmB,KAAK,IAAI;AAC5B;AAAA,QACF;AAEA,wBAAgB,MAAM,kBAAkB;AAAA,MAC1C;AAEA,aAAO,EAAE,SAAS,oBAAoB,GAAG,KAAK;AAAA,IAChD;AAAA,IAEA,YAAY,OAAO,EAAE,SAAS,MAAM;AAClC,YAAM,EAAE,QAAQ,GAAG,KAAK,IAAI,MAAM,SAAS;AAY3C,YAAM,uBAAwD,CAAC;AAE/D,eAAS,gBACP,kBACA,YACA;AACA,eAAO,CAAC,SAAiB;AACvB,cAAI,KAAK,WAAW,GAAG;AACrB;AAAA,UACF;AAEA,gBAAM,SAAS,UAAU,gBAAgB;AACzC,gCAAsB,kBAAkB,UAAU;AAClD,uBAAa,kBAAkB,YAAY,QAAQ,IAAI;AACvD,gCAAsB,gBAAgB;AAAA,QACxC;AAAA,MACF;AAEA,eAAS,UAAU,kBAA2C;AAC5D,eAAO,iBAAiB,gBACrB,iBAAiB,cACd,CAAC,iBAAiB,mBAClB,CAAC,iBAAiB,eACpB,YACA;AAAA,MACN;AAEA,eAAS,sBACP,kBACA,YACA;AACA,YACG,iBAAiB,eAAe,iBAAiB,eAClD,iBAAiB,kBACjB;AACA,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN,IAAI,aAAa,iBAAiB,SAAS;AAAA,UAC7C,CAAC;AAAA,QACH;AAAA,MACF;AAEA,eAAS,aACP,kBACA,YACA,QACA,MACA;AACA,mBAAW;AAAA,UACT,iBAAiB,cACb;AAAA,YACE,MAAM;AAAA,YACN,OAAO,SAAS;AAAA,YAChB,IAAI,aAAa,iBAAiB,SAAS;AAAA,UAC7C,IACA;AAAA,YACE,MAAM;AAAA,YACN,OAAO,SAAS;AAAA,YAChB,IAAI,iBAAiB;AAAA,UACvB;AAAA,QACN;AAAA,MACF;AAEA,eAAS,sBAAsB,kBAAmC;AAChE,yBAAiB,cAAc;AAC/B,YAAI,iBAAiB,aAAa;AAChC,2BAAiB,mBAAmB;AAAA,QACtC,OAAO;AACL,2BAAiB,cAAc;AAAA,QACjC;AAAA,MACF;AAEA,eAAS,gBACP,kBACA,YACA,YACA,SACA;AACA,yBAAiB,SAAS,iBAAiB,OAAO;AAAA,UAChD,aAAa,QAAQ;AAAA,QACvB;AAEA,YAAI,iBAAiB,aAAa;AAChC,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN,IAAI,aAAa,iBAAiB,SAAS;AAAA,UAC7C,CAAC;AACD,2BAAiB,aAAa;AAAA,QAChC;AAEA,yBAAiB,cAAc,CAAC,iBAAiB;AACjD,yBAAiB,cAAc;AAAA,MACjC;AAEA,eAAS,gBAAgB;AAAA,QACvB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF,GAMY;AACV,gBAAQ,iBAAiB,OAAO,MAAM,GAAG,UAAU,CAAC;AAEpD,cAAM,iBACJ,aAAa,QAAQ,UAAU,iBAAiB,OAAO;AAEzD,YAAI,gBAAgB;AAClB,0BAAgB,kBAAkB,YAAY,YAAY,OAAO;AACjE,iBAAO;AAAA,QACT;AAEA,yBAAiB,SAAS,iBAAiB,OAAO,MAAM,UAAU;AAClE,eAAO;AAAA,MACT;AAEA,eAAS,cACP,kBACA,YACA;AACA,cAAM,UAAU,gBAAgB,kBAAkB,UAAU;AAC5D,YAAI,qBAAqB;AAEzB,eAAO,oBAAoB;AACzB,gBAAM,UAAU,iBAAiB,cAC7B,aACA;AACJ,gBAAM,aAAa;AAAA,YACjB,iBAAiB;AAAA,YACjB;AAAA,UACF;AAEA,cAAI,cAAc,MAAM;AACtB,oBAAQ,iBAAiB,MAAM;AAC/B,6BAAiB,SAAS;AAC1B;AAAA,UACF;AAEA,+BAAqB,gBAAgB;AAAA,YACnC;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AAEA,aAAO;AAAA,QACL,QAAQ,OAAO;AAAA,UACb,IAAI,gBAGF;AAAA,YACA,WAAW,CAAC,OAAO,eAAe;AAChC,kBAAI,MAAM,SAAS,cAAc;AAC/B,2BAAW,QAAQ,KAAK;AACxB;AAAA,cACF;AAEA,kBAAI,qBAAqB,MAAM,EAAE,KAAK,MAAM;AAC1C,qCAAqB,MAAM,EAAE,IAAI;AAAA,kBAC/B,kBAAkB;AAAA,kBAClB,aAAa;AAAA,kBACb,aAAa;AAAA,kBACb,aAAa;AAAA,kBACb,QAAQ;AAAA,kBACR,WAAW;AAAA,kBACX,QAAQ,MAAM;AAAA,gBAChB;AAAA,cACF;AAEA,oBAAM,mBAAmB,qBAAqB,MAAM,EAAE;AACtD,+BAAiB,UAAU,MAAM;AACjC,4BAAc,kBAAkB,UAAU;AAAA,YAC5C;AAAA,UACF,CAAC;AAAA,QACH;AAAA,QACA,GAAG;AAAA,MACL;AAAA,IACF;AAAA,EACF;AACF;","names":[]}

package/dist/index.cjs

@@ -0,0 +1,339 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var src_exports = {};
+__export(src_exports, {
+  defaultSystemPromptMiddleware: () => defaultSystemPromptMiddleware,
+  extractReasoningMiddleware: () => extractReasoningMiddleware,
+  getPotentialStartIndex: () => getPotentialStartIndex
+});
+module.exports = __toCommonJS(src_exports);
+
+// src/default-system-prompt.ts
+function extractSystemText(content) {
+  if (typeof content === "string") {
+    return content;
+  }
+  if (!Array.isArray(content)) {
+    if (content == null) {
+      return;
+    }
+    return String(content);
+  }
+  const parts = content.map((part) => {
+    var _a;
+    if ((part == null ? void 0 : part.type) === "text" && "text" in part) {
+      return String((_a = part.text) != null ? _a : "");
+    }
+    return JSON.stringify(part);
+  });
+  const textParts = parts.filter((value) => value.length > 0);
+  if (textParts.length === 0) {
+    return;
+  }
+  return textParts.join("\n");
+}
+function mergeSystemPrompts({
+  base,
+  addition,
+  placement
+}) {
+  if (!base) {
+    return addition;
+  }
+  if (addition.length === 0) {
+    return base;
+  }
+  return placement === "first" ? `${addition}
+
+${base}` : `${base}
+
+${addition}`;
+}
+function ensurePromptArray(prompt) {
+  if (!prompt) {
+    return [];
+  }
+  return [...prompt];
+}
+function defaultSystemPromptMiddleware({
+  systemPrompt,
+  placement = "first"
+}) {
+  return {
+    specificationVersion: "v3",
+    transformParams: ({ params }) => {
+      const prompt = ensurePromptArray(params.prompt);
+      const systemIndex = prompt.findIndex(
+        (message) => message.role === "system"
+      );
+      if (systemIndex === -1) {
+        const promptWithSystem = placement === "first" ? [
+          {
+            role: "system",
+            content: systemPrompt
+          },
+          ...prompt
+        ] : [
+          ...prompt,
+          {
+            role: "system",
+            content: systemPrompt
+          }
+        ];
+        const nextParams2 = {
+          ...params,
+          prompt: promptWithSystem
+        };
+        return Promise.resolve(nextParams2);
+      }
+      const systemMessage = prompt[systemIndex];
+      const baseText = extractSystemText(systemMessage.content);
+      const mergedContent = mergeSystemPrompts({
+        base: baseText,
+        addition: systemPrompt,
+        placement
+      });
+      const updatedPrompt = prompt.map(
+        (message, index) => index === systemIndex ? {
+          ...message,
+          content: mergedContent
+        } : message
+      );
+      const nextParams = {
+        ...params,
+        prompt: updatedPrompt
+      };
+      return Promise.resolve(nextParams);
+    }
+  };
+}
+
+// src/reasoning-parser/index.ts
+function getPotentialStartIndex(text, searchedText) {
+  if (searchedText.length === 0) {
+    return null;
+  }
+  const directIndex = text.indexOf(searchedText);
+  if (directIndex !== -1) {
+    return directIndex;
+  }
+  for (let i = text.length - 1; i >= 0; i -= 1) {
+    const suffix = text.substring(i);
+    if (searchedText.startsWith(suffix)) {
+      return i;
+    }
+  }
+  return null;
+}
+function extractReasoningMiddleware({
+  openingTag,
+  closingTag,
+  separator = "\n",
+  startWithReasoning = false
+}) {
+  function processTextPart(text, transformedContent) {
+    var _a;
+    const regexp = new RegExp(`${openingTag}(.*?)${closingTag}`, "gs");
+    const matches = Array.from(text.matchAll(regexp));
+    if (!matches.length) {
+      return;
+    }
+    const reasoningText = matches.map((match) => match[1]).join(separator);
+    let textWithoutReasoning = text;
+    for (let i = matches.length - 1; i >= 0; i -= 1) {
+      const match = matches[i];
+      const beforeMatch = textWithoutReasoning.slice(0, match.index);
+      const matchIndex = (_a = match.index) != null ? _a : 0;
+      const afterMatch = textWithoutReasoning.slice(
+        matchIndex + match[0].length
+      );
+      textWithoutReasoning = beforeMatch + (beforeMatch.length > 0 && afterMatch.length > 0 ? separator : "") + afterMatch;
+    }
+    transformedContent.push({
+      type: "reasoning",
+      text: reasoningText
+    });
+    transformedContent.push({
+      type: "text",
+      text: textWithoutReasoning
+    });
+  }
+  return {
+    specificationVersion: "v3",
+    wrapGenerate: async ({ doGenerate }) => {
+      const { content, ...rest } = await doGenerate();
+      const transformedContent = [];
+      for (const part of content) {
+        if (part.type !== "text") {
+          transformedContent.push(part);
+          continue;
+        }
+        const text = startWithReasoning ? openingTag + part.text : part.text;
+        const regexp = new RegExp(`${openingTag}(.*?)${closingTag}`, "gs");
+        const matches = Array.from(text.matchAll(regexp));
+        if (!matches.length) {
+          transformedContent.push(part);
+          continue;
+        }
+        processTextPart(text, transformedContent);
+      }
+      return { content: transformedContent, ...rest };
+    },
+    wrapStream: async ({ doStream }) => {
+      const { stream, ...rest } = await doStream();
+      const reasoningExtractions = {};
+      function createPublisher(activeExtraction, controller) {
+        return (text) => {
+          if (text.length === 0) {
+            return;
+          }
+          const prefix = getPrefix(activeExtraction);
+          enqueueReasoningStart(activeExtraction, controller);
+          enqueueDelta(activeExtraction, controller, prefix, text);
+          updateExtractionState(activeExtraction);
+        };
+      }
+      function getPrefix(activeExtraction) {
+        return activeExtraction.afterSwitch && (activeExtraction.isReasoning ? !activeExtraction.isFirstReasoning : !activeExtraction.isFirstText) ? separator : "";
+      }
+      function enqueueReasoningStart(activeExtraction, controller) {
+        if (activeExtraction.afterSwitch && activeExtraction.isReasoning || activeExtraction.isFirstReasoning) {
+          controller.enqueue({
+            type: "reasoning-start",
+            id: `reasoning-${activeExtraction.idCounter}`
+          });
+        }
+      }
+      function enqueueDelta(activeExtraction, controller, prefix, text) {
+        controller.enqueue(
+          activeExtraction.isReasoning ? {
+            type: "reasoning-delta",
+            delta: prefix + text,
+            id: `reasoning-${activeExtraction.idCounter}`
+          } : {
+            type: "text-delta",
+            delta: prefix + text,
+            id: activeExtraction.textId
+          }
+        );
+      }
+      function updateExtractionState(activeExtraction) {
+        activeExtraction.afterSwitch = false;
+        if (activeExtraction.isReasoning) {
+          activeExtraction.isFirstReasoning = false;
+        } else {
+          activeExtraction.isFirstText = false;
+        }
+      }
+      function handleFullMatch(activeExtraction, controller, startIndex, nextTag) {
+        activeExtraction.buffer = activeExtraction.buffer.slice(
+          startIndex + nextTag.length
+        );
+        if (activeExtraction.isReasoning) {
+          controller.enqueue({
+            type: "reasoning-end",
+            id: `reasoning-${activeExtraction.idCounter}`
+          });
+          activeExtraction.idCounter += 1;
+        }
+        activeExtraction.isReasoning = !activeExtraction.isReasoning;
+        activeExtraction.afterSwitch = true;
+      }
+      function processTagMatch({
+        activeExtraction,
+        controller,
+        publish,
+        startIndex,
+        nextTag
+      }) {
+        publish(activeExtraction.buffer.slice(0, startIndex));
+        const foundFullMatch = startIndex + nextTag.length <= activeExtraction.buffer.length;
+        if (foundFullMatch) {
+          handleFullMatch(activeExtraction, controller, startIndex, nextTag);
+          return true;
+        }
+        activeExtraction.buffer = activeExtraction.buffer.slice(startIndex);
+        return false;
+      }
+      function processBuffer(activeExtraction, controller) {
+        const publish = createPublisher(activeExtraction, controller);
+        let continueProcessing = true;
+        while (continueProcessing) {
+          const nextTag = activeExtraction.isReasoning ? closingTag : openingTag;
+          const startIndex = getPotentialStartIndex(
+            activeExtraction.buffer,
+            nextTag
+          );
+          if (startIndex == null) {
+            publish(activeExtraction.buffer);
+            activeExtraction.buffer = "";
+            break;
+          }
+          continueProcessing = processTagMatch({
+            activeExtraction,
+            controller,
+            publish,
+            startIndex,
+            nextTag
+          });
+        }
+      }
+      return {
+        stream: stream.pipeThrough(
+          new TransformStream({
+            transform: (chunk, controller) => {
+              if (chunk.type !== "text-delta") {
+                controller.enqueue(chunk);
+                return;
+              }
+              if (reasoningExtractions[chunk.id] == null) {
+                reasoningExtractions[chunk.id] = {
+                  isFirstReasoning: true,
+                  isFirstText: true,
+                  afterSwitch: false,
+                  isReasoning: startWithReasoning,
+                  buffer: "",
+                  idCounter: 0,
+                  textId: chunk.id
+                };
+              }
+              const activeExtraction = reasoningExtractions[chunk.id];
+              activeExtraction.buffer += chunk.delta;
+              processBuffer(activeExtraction, controller);
+            }
+          })
+        ),
+        ...rest
+      };
+    }
+  };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  defaultSystemPromptMiddleware,
+  extractReasoningMiddleware,
+  getPotentialStartIndex
+});
+/**
+ * @license
+ * Copyright (c) 2021-present, FriendliAI Inc. All rights reserved.
+ */
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts","../src/default-system-prompt.ts","../src/reasoning-parser/index.ts"],"sourcesContent":["// biome-ignore lint/performance/noBarrelFile: Package entrypoint - must re-export for public API\nexport * from \"./default-system-prompt\";\nexport * from \"./reasoning-parser\";\n","import type {\n  LanguageModelV3CallOptions,\n  LanguageModelV3Content,\n  LanguageModelV3Middleware,\n  LanguageModelV3Prompt,\n} from \"@ai-sdk/provider\";\n\ntype SystemPromptPlacement = \"first\" | \"last\";\n\ninterface DefaultSystemPromptMiddlewareOptions {\n  systemPrompt: string;\n  placement?: SystemPromptPlacement;\n}\n\nfunction extractSystemText(content: unknown): string | undefined {\n  if (typeof content === \"string\") {\n    return content;\n  }\n\n  if (!Array.isArray(content)) {\n    if (content == null) {\n      return;\n    }\n    return String(content);\n  }\n\n  const parts = (content as LanguageModelV3Content[]).map((part) => {\n    if (part?.type === \"text\" && \"text\" in part) {\n      return String(part.text ?? \"\");\n    }\n\n    return JSON.stringify(part);\n  });\n\n  const textParts = parts.filter((value) => value.length > 0);\n  if (textParts.length === 0) {\n    return;\n  }\n\n  return textParts.join(\"\\n\");\n}\n\nfunction mergeSystemPrompts({\n  base,\n  addition,\n  placement,\n}: {\n  base?: string;\n  addition: string;\n  placement: SystemPromptPlacement;\n}): string {\n  if (!base) {\n    return addition;\n  }\n\n  if (addition.length === 0) {\n    return base;\n  }\n\n  return placement === \"first\"\n    ? `${addition}\\n\\n${base}`\n    : `${base}\\n\\n${addition}`;\n}\n\nfunction ensurePromptArray(\n  prompt?: LanguageModelV3Prompt\n): LanguageModelV3Prompt {\n  if (!prompt) {\n    return [];\n  }\n\n  return [...prompt];\n}\n\nexport function defaultSystemPromptMiddleware({\n  systemPrompt,\n  placement = \"first\",\n}: DefaultSystemPromptMiddlewareOptions): LanguageModelV3Middleware {\n  return {\n    specificationVersion: \"v3\",\n    transformParams: ({ params }) => {\n      const prompt = ensurePromptArray(params.prompt);\n      const systemIndex = prompt.findIndex(\n        (message) => message.role === \"system\"\n      );\n\n      if (systemIndex === -1) {\n        const promptWithSystem =\n          placement === \"first\"\n            ? ([\n                {\n                  role: \"system\" as const,\n                  content: systemPrompt,\n                },\n                ...prompt,\n              ] as LanguageModelV3Prompt)\n            : ([\n                ...prompt,\n                {\n                  role: \"system\" as const,\n                  content: systemPrompt,\n                },\n              ] as LanguageModelV3Prompt);\n\n        const nextParams: LanguageModelV3CallOptions = {\n          ...params,\n          prompt: promptWithSystem,\n        };\n\n        return Promise.resolve<LanguageModelV3CallOptions>(nextParams);\n      }\n\n      const systemMessage = prompt[systemIndex];\n      const baseText = extractSystemText(systemMessage.content);\n      const mergedContent = mergeSystemPrompts({\n        base: baseText,\n        addition: systemPrompt,\n        placement,\n      });\n\n      const updatedPrompt = prompt.map((message, index) =>\n        index === systemIndex\n          ? {\n              ...message,\n              content: mergedContent,\n            }\n          : message\n      ) as LanguageModelV3Prompt;\n\n      const nextParams: LanguageModelV3CallOptions = {\n        ...params,\n        prompt: updatedPrompt,\n      };\n\n      return Promise.resolve<LanguageModelV3CallOptions>(nextParams);\n    },\n  };\n}\n","/**\n * @license\n * Copyright (c) 2021-present, FriendliAI Inc. All rights reserved.\n */\n\nimport type {\n  LanguageModelV3Content,\n  LanguageModelV3Middleware,\n  LanguageModelV3StreamPart,\n} from \"@ai-sdk/provider\";\n\n/**\n * All code below is forked from the following link:\n * https://github.com/vercel/ai/blob/v5/packages/ai/core/middleware/extract-reasoning-middleware.ts\n */\n\n/**\n * Returns the index of the start of the searchedText in the text, or null if it\n * is not found.\n */\nexport function getPotentialStartIndex(\n  text: string,\n  searchedText: string\n): number | null {\n  // Return null immediately if searchedText is empty.\n  if (searchedText.length === 0) {\n    return null;\n  }\n\n  // Check if the searchedText exists as a direct substring of text.\n  const directIndex = text.indexOf(searchedText);\n  if (directIndex !== -1) {\n    return directIndex;\n  }\n\n  // Otherwise, look for the largest suffix of \"text\" that matches\n  // a prefix of \"searchedText\". We go from the end of text inward.\n  for (let i = text.length - 1; i >= 0; i -= 1) {\n    const suffix = text.substring(i);\n    if (searchedText.startsWith(suffix)) {\n      return i;\n    }\n  }\n\n  return null;\n}\n\n/**\n * Extract an XML-tagged reasoning section from the generated text and exposes it\n * as a `reasoning` property on the result.\n *\n * @param openingTag - The opening XML tag to extract reasoning from.\n * @param closingTag - The closing XML tag to extract reasoning from.\n * @param separator - The separator to use between reasoning and text sections.\n * @param startWithReasoning - Whether to start with reasoning tokens.\n */\nexport function extractReasoningMiddleware({\n  openingTag,\n  closingTag,\n  separator = \"\\n\",\n  startWithReasoning = false,\n}: {\n  openingTag: string;\n  closingTag: string;\n  separator?: string;\n  startWithReasoning?: boolean;\n}): LanguageModelV3Middleware {\n  function processTextPart(\n    text: string,\n    transformedContent: LanguageModelV3Content[]\n  ) {\n    const regexp = new RegExp(`${openingTag}(.*?)${closingTag}`, \"gs\");\n    const matches = Array.from(text.matchAll(regexp));\n\n    if (!matches.length) {\n      return;\n    }\n\n    const reasoningText = matches.map((match) => match[1]).join(separator);\n\n    let textWithoutReasoning = text;\n    for (let i = matches.length - 1; i >= 0; i -= 1) {\n      const match = matches[i];\n\n      const beforeMatch = textWithoutReasoning.slice(0, match.index);\n      const matchIndex = match.index ?? 0;\n      const afterMatch = textWithoutReasoning.slice(\n        matchIndex + match[0].length\n      );\n\n      textWithoutReasoning =\n        beforeMatch +\n        (beforeMatch.length > 0 && afterMatch.length > 0 ? separator : \"\") +\n        afterMatch;\n    }\n\n    transformedContent.push({\n      type: \"reasoning\",\n      text: reasoningText,\n    });\n\n    transformedContent.push({\n      type: \"text\",\n      text: textWithoutReasoning,\n    });\n  }\n\n  return {\n    specificationVersion: \"v3\",\n    wrapGenerate: async ({ doGenerate }) => {\n      const { content, ...rest } = await doGenerate();\n\n      const transformedContent: LanguageModelV3Content[] = [];\n      for (const part of content) {\n        if (part.type !== \"text\") {\n          transformedContent.push(part);\n          continue;\n        }\n\n        const text = startWithReasoning ? openingTag + part.text : part.text;\n        const regexp = new RegExp(`${openingTag}(.*?)${closingTag}`, \"gs\");\n        const matches = Array.from(text.matchAll(regexp));\n\n        if (!matches.length) {\n          transformedContent.push(part);\n          continue;\n        }\n\n        processTextPart(text, transformedContent);\n      }\n\n      return { content: transformedContent, ...rest };\n    },\n\n    wrapStream: async ({ doStream }) => {\n      const { stream, ...rest } = await doStream();\n\n      interface ExtractionState {\n        isFirstReasoning: boolean;\n        isFirstText: boolean;\n        afterSwitch: boolean;\n        isReasoning: boolean;\n        buffer: string;\n        idCounter: number;\n        textId: string;\n      }\n\n      const reasoningExtractions: Record<string, ExtractionState> = {};\n\n      function createPublisher(\n        activeExtraction: ExtractionState,\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>\n      ) {\n        return (text: string) => {\n          if (text.length === 0) {\n            return;\n          }\n\n          const prefix = getPrefix(activeExtraction);\n          enqueueReasoningStart(activeExtraction, controller);\n          enqueueDelta(activeExtraction, controller, prefix, text);\n          updateExtractionState(activeExtraction);\n        };\n      }\n\n      function getPrefix(activeExtraction: ExtractionState): string {\n        return activeExtraction.afterSwitch &&\n          (activeExtraction.isReasoning\n            ? !activeExtraction.isFirstReasoning\n            : !activeExtraction.isFirstText)\n          ? separator\n          : \"\";\n      }\n\n      function enqueueReasoningStart(\n        activeExtraction: ExtractionState,\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>\n      ) {\n        if (\n          (activeExtraction.afterSwitch && activeExtraction.isReasoning) ||\n          activeExtraction.isFirstReasoning\n        ) {\n          controller.enqueue({\n            type: \"reasoning-start\",\n            id: `reasoning-${activeExtraction.idCounter}`,\n          });\n        }\n      }\n\n      function enqueueDelta(\n        activeExtraction: ExtractionState,\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>,\n        prefix: string,\n        text: string\n      ) {\n        controller.enqueue(\n          activeExtraction.isReasoning\n            ? {\n                type: \"reasoning-delta\",\n                delta: prefix + text,\n                id: `reasoning-${activeExtraction.idCounter}`,\n              }\n            : {\n                type: \"text-delta\",\n                delta: prefix + text,\n                id: activeExtraction.textId,\n              }\n        );\n      }\n\n      function updateExtractionState(activeExtraction: ExtractionState) {\n        activeExtraction.afterSwitch = false;\n        if (activeExtraction.isReasoning) {\n          activeExtraction.isFirstReasoning = false;\n        } else {\n          activeExtraction.isFirstText = false;\n        }\n      }\n\n      function handleFullMatch(\n        activeExtraction: ExtractionState,\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>,\n        startIndex: number,\n        nextTag: string\n      ) {\n        activeExtraction.buffer = activeExtraction.buffer.slice(\n          startIndex + nextTag.length\n        );\n\n        if (activeExtraction.isReasoning) {\n          controller.enqueue({\n            type: \"reasoning-end\",\n            id: `reasoning-${activeExtraction.idCounter}`,\n          });\n          activeExtraction.idCounter += 1;\n        }\n\n        activeExtraction.isReasoning = !activeExtraction.isReasoning;\n        activeExtraction.afterSwitch = true;\n      }\n\n      function processTagMatch({\n        activeExtraction,\n        controller,\n        publish,\n        startIndex,\n        nextTag,\n      }: {\n        activeExtraction: ExtractionState;\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>;\n        publish: (text: string) => void;\n        startIndex: number;\n        nextTag: string;\n      }): boolean {\n        publish(activeExtraction.buffer.slice(0, startIndex));\n\n        const foundFullMatch =\n          startIndex + nextTag.length <= activeExtraction.buffer.length;\n\n        if (foundFullMatch) {\n          handleFullMatch(activeExtraction, controller, startIndex, nextTag);\n          return true;\n        }\n\n        activeExtraction.buffer = activeExtraction.buffer.slice(startIndex);\n        return false;\n      }\n\n      function processBuffer(\n        activeExtraction: ExtractionState,\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>\n      ) {\n        const publish = createPublisher(activeExtraction, controller);\n        let continueProcessing = true;\n\n        while (continueProcessing) {\n          const nextTag = activeExtraction.isReasoning\n            ? closingTag\n            : openingTag;\n          const startIndex = getPotentialStartIndex(\n            activeExtraction.buffer,\n            nextTag\n          );\n\n          if (startIndex == null) {\n            publish(activeExtraction.buffer);\n            activeExtraction.buffer = \"\";\n            break;\n          }\n\n          continueProcessing = processTagMatch({\n            activeExtraction,\n            controller,\n            publish,\n            startIndex,\n            nextTag,\n          });\n        }\n      }\n\n      return {\n        stream: stream.pipeThrough(\n          new TransformStream<\n            LanguageModelV3StreamPart,\n            LanguageModelV3StreamPart\n          >({\n            transform: (chunk, controller) => {\n              if (chunk.type !== \"text-delta\") {\n                controller.enqueue(chunk);\n                return;\n              }\n\n              if (reasoningExtractions[chunk.id] == null) {\n                reasoningExtractions[chunk.id] = {\n                  isFirstReasoning: true,\n                  isFirstText: true,\n                  afterSwitch: false,\n                  isReasoning: startWithReasoning,\n                  buffer: \"\",\n                  idCounter: 0,\n                  textId: chunk.id,\n                };\n              }\n\n              const activeExtraction = reasoningExtractions[chunk.id];\n              activeExtraction.buffer += chunk.delta;\n              processBuffer(activeExtraction, controller);\n            },\n          })\n        ),\n        ...rest,\n      };\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACcA,SAAS,kBAAkB,SAAsC;AAC/D,MAAI,OAAO,YAAY,UAAU;AAC/B,WAAO;AAAA,EACT;AAEA,MAAI,CAAC,MAAM,QAAQ,OAAO,GAAG;AAC3B,QAAI,WAAW,MAAM;AACnB;AAAA,IACF;AACA,WAAO,OAAO,OAAO;AAAA,EACvB;AAEA,QAAM,QAAS,QAAqC,IAAI,CAAC,SAAS;AA1BpE;AA2BI,SAAI,6BAAM,UAAS,UAAU,UAAU,MAAM;AAC3C,aAAO,QAAO,UAAK,SAAL,YAAa,EAAE;AAAA,IAC/B;AAEA,WAAO,KAAK,UAAU,IAAI;AAAA,EAC5B,CAAC;AAED,QAAM,YAAY,MAAM,OAAO,CAAC,UAAU,MAAM,SAAS,CAAC;AAC1D,MAAI,UAAU,WAAW,GAAG;AAC1B;AAAA,EACF;AAEA,SAAO,UAAU,KAAK,IAAI;AAC5B;AAEA,SAAS,mBAAmB;AAAA,EAC1B;AAAA,EACA;AAAA,EACA;AACF,GAIW;AACT,MAAI,CAAC,MAAM;AACT,WAAO;AAAA,EACT;AAEA,MAAI,SAAS,WAAW,GAAG;AACzB,WAAO;AAAA,EACT;AAEA,SAAO,cAAc,UACjB,GAAG,QAAQ;AAAA;AAAA,EAAO,IAAI,KACtB,GAAG,IAAI;AAAA;AAAA,EAAO,QAAQ;AAC5B;AAEA,SAAS,kBACP,QACuB;AACvB,MAAI,CAAC,QAAQ;AACX,WAAO,CAAC;AAAA,EACV;AAEA,SAAO,CAAC,GAAG,MAAM;AACnB;AAEO,SAAS,8BAA8B;AAAA,EAC5C;AAAA,EACA,YAAY;AACd,GAAoE;AAClE,SAAO;AAAA,IACL,sBAAsB;AAAA,IACtB,iBAAiB,CAAC,EAAE,OAAO,MAAM;AAC/B,YAAM,SAAS,kBAAkB,OAAO,MAAM;AAC9C,YAAM,cAAc,OAAO;AAAA,QACzB,CAAC,YAAY,QAAQ,SAAS;AAAA,MAChC;AAEA,UAAI,gBAAgB,IAAI;AACtB,cAAM,mBACJ,cAAc,UACT;AAAA,UACC;AAAA,YACE,MAAM;AAAA,YACN,SAAS;AAAA,UACX;AAAA,UACA,GAAG;AAAA,QACL,IACC;AAAA,UACC,GAAG;AAAA,UACH;AAAA,YACE,MAAM;AAAA,YACN,SAAS;AAAA,UACX;AAAA,QACF;AAEN,cAAMA,cAAyC;AAAA,UAC7C,GAAG;AAAA,UACH,QAAQ;AAAA,QACV;AAEA,eAAO,QAAQ,QAAoCA,WAAU;AAAA,MAC/D;AAEA,YAAM,gBAAgB,OAAO,WAAW;AACxC,YAAM,WAAW,kBAAkB,cAAc,OAAO;AACxD,YAAM,gBAAgB,mBAAmB;AAAA,QACvC,MAAM;AAAA,QACN,UAAU;AAAA,QACV;AAAA,MACF,CAAC;AAED,YAAM,gBAAgB,OAAO;AAAA,QAAI,CAAC,SAAS,UACzC,UAAU,cACN;AAAA,UACE,GAAG;AAAA,UACH,SAAS;AAAA,QACX,IACA;AAAA,MACN;AAEA,YAAM,aAAyC;AAAA,QAC7C,GAAG;AAAA,QACH,QAAQ;AAAA,MACV;AAEA,aAAO,QAAQ,QAAoC,UAAU;AAAA,IAC/D;AAAA,EACF;AACF;;;ACrHO,SAAS,uBACd,MACA,cACe;AAEf,MAAI,aAAa,WAAW,GAAG;AAC7B,WAAO;AAAA,EACT;AAGA,QAAM,cAAc,KAAK,QAAQ,YAAY;AAC7C,MAAI,gBAAgB,IAAI;AACtB,WAAO;AAAA,EACT;AAIA,WAAS,IAAI,KAAK,SAAS,GAAG,KAAK,GAAG,KAAK,GAAG;AAC5C,UAAM,SAAS,KAAK,UAAU,CAAC;AAC/B,QAAI,aAAa,WAAW,MAAM,GAAG;AACnC,aAAO;AAAA,IACT;AAAA,EACF;AAEA,SAAO;AACT;AAWO,SAAS,2BAA2B;AAAA,EACzC;AAAA,EACA;AAAA,EACA,YAAY;AAAA,EACZ,qBAAqB;AACvB,GAK8B;AAC5B,WAAS,gBACP,MACA,oBACA;AAtEJ;AAuEI,UAAM,SAAS,IAAI,OAAO,GAAG,UAAU,QAAQ,UAAU,IAAI,IAAI;AACjE,UAAM,UAAU,MAAM,KAAK,KAAK,SAAS,MAAM,CAAC;AAEhD,QAAI,CAAC,QAAQ,QAAQ;AACnB;AAAA,IACF;AAEA,UAAM,gBAAgB,QAAQ,IAAI,CAAC,UAAU,MAAM,CAAC,CAAC,EAAE,KAAK,SAAS;AAErE,QAAI,uBAAuB;AAC3B,aAAS,IAAI,QAAQ,SAAS,GAAG,KAAK,GAAG,KAAK,GAAG;AAC/C,YAAM,QAAQ,QAAQ,CAAC;AAEvB,YAAM,cAAc,qBAAqB,MAAM,GAAG,MAAM,KAAK;AAC7D,YAAM,cAAa,WAAM,UAAN,YAAe;AAClC,YAAM,aAAa,qBAAqB;AAAA,QACtC,aAAa,MAAM,CAAC,EAAE;AAAA,MACxB;AAEA,6BACE,eACC,YAAY,SAAS,KAAK,WAAW,SAAS,IAAI,YAAY,MAC/D;AAAA,IACJ;AAEA,uBAAmB,KAAK;AAAA,MACtB,MAAM;AAAA,MACN,MAAM;AAAA,IACR,CAAC;AAED,uBAAmB,KAAK;AAAA,MACtB,MAAM;AAAA,MACN,MAAM;AAAA,IACR,CAAC;AAAA,EACH;AAEA,SAAO;AAAA,IACL,sBAAsB;AAAA,IACtB,cAAc,OAAO,EAAE,WAAW,MAAM;AACtC,YAAM,EAAE,SAAS,GAAG,KAAK,IAAI,MAAM,WAAW;AAE9C,YAAM,qBAA+C,CAAC;AACtD,iBAAW,QAAQ,SAAS;AAC1B,YAAI,KAAK,SAAS,QAAQ;AACxB,6BAAmB,KAAK,IAAI;AAC5B;AAAA,QACF;AAEA,cAAM,OAAO,qBAAqB,aAAa,KAAK,OAAO,KAAK;AAChE,cAAM,SAAS,IAAI,OAAO,GAAG,UAAU,QAAQ,UAAU,IAAI,IAAI;AACjE,cAAM,UAAU,MAAM,KAAK,KAAK,SAAS,MAAM,CAAC;AAEhD,YAAI,CAAC,QAAQ,QAAQ;AACnB,6BAAmB,KAAK,IAAI;AAC5B;AAAA,QACF;AAEA,wBAAgB,MAAM,kBAAkB;AAAA,MAC1C;AAEA,aAAO,EAAE,SAAS,oBAAoB,GAAG,KAAK;AAAA,IAChD;AAAA,IAEA,YAAY,OAAO,EAAE,SAAS,MAAM;AAClC,YAAM,EAAE,QAAQ,GAAG,KAAK,IAAI,MAAM,SAAS;AAY3C,YAAM,uBAAwD,CAAC;AAE/D,eAAS,gBACP,kBACA,YACA;AACA,eAAO,CAAC,SAAiB;AACvB,cAAI,KAAK,WAAW,GAAG;AACrB;AAAA,UACF;AAEA,gBAAM,SAAS,UAAU,gBAAgB;AACzC,gCAAsB,kBAAkB,UAAU;AAClD,uBAAa,kBAAkB,YAAY,QAAQ,IAAI;AACvD,gCAAsB,gBAAgB;AAAA,QACxC;AAAA,MACF;AAEA,eAAS,UAAU,kBAA2C;AAC5D,eAAO,iBAAiB,gBACrB,iBAAiB,cACd,CAAC,iBAAiB,mBAClB,CAAC,iBAAiB,eACpB,YACA;AAAA,MACN;AAEA,eAAS,sBACP,kBACA,YACA;AACA,YACG,iBAAiB,eAAe,iBAAiB,eAClD,iBAAiB,kBACjB;AACA,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN,IAAI,aAAa,iBAAiB,SAAS;AAAA,UAC7C,CAAC;AAAA,QACH;AAAA,MACF;AAEA,eAAS,aACP,kBACA,YACA,QACA,MACA;AACA,mBAAW;AAAA,UACT,iBAAiB,cACb;AAAA,YACE,MAAM;AAAA,YACN,OAAO,SAAS;AAAA,YAChB,IAAI,aAAa,iBAAiB,SAAS;AAAA,UAC7C,IACA;AAAA,YACE,MAAM;AAAA,YACN,OAAO,SAAS;AAAA,YAChB,IAAI,iBAAiB;AAAA,UACvB;AAAA,QACN;AAAA,MACF;AAEA,eAAS,sBAAsB,kBAAmC;AAChE,yBAAiB,cAAc;AAC/B,YAAI,iBAAiB,aAAa;AAChC,2BAAiB,mBAAmB;AAAA,QACtC,OAAO;AACL,2BAAiB,cAAc;AAAA,QACjC;AAAA,MACF;AAEA,eAAS,gBACP,kBACA,YACA,YACA,SACA;AACA,yBAAiB,SAAS,iBAAiB,OAAO;AAAA,UAChD,aAAa,QAAQ;AAAA,QACvB;AAEA,YAAI,iBAAiB,aAAa;AAChC,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN,IAAI,aAAa,iBAAiB,SAAS;AAAA,UAC7C,CAAC;AACD,2BAAiB,aAAa;AAAA,QAChC;AAEA,yBAAiB,cAAc,CAAC,iBAAiB;AACjD,yBAAiB,cAAc;AAAA,MACjC;AAEA,eAAS,gBAAgB;AAAA,QACvB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF,GAMY;AACV,gBAAQ,iBAAiB,OAAO,MAAM,GAAG,UAAU,CAAC;AAEpD,cAAM,iBACJ,aAAa,QAAQ,UAAU,iBAAiB,OAAO;AAEzD,YAAI,gBAAgB;AAClB,0BAAgB,kBAAkB,YAAY,YAAY,OAAO;AACjE,iBAAO;AAAA,QACT;AAEA,yBAAiB,SAAS,iBAAiB,OAAO,MAAM,UAAU;AAClE,eAAO;AAAA,MACT;AAEA,eAAS,cACP,kBACA,YACA;AACA,cAAM,UAAU,gBAAgB,kBAAkB,UAAU;AAC5D,YAAI,qBAAqB;AAEzB,eAAO,oBAAoB;AACzB,gBAAM,UAAU,iBAAiB,cAC7B,aACA;AACJ,gBAAM,aAAa;AAAA,YACjB,iBAAiB;AAAA,YACjB;AAAA,UACF;AAEA,cAAI,cAAc,MAAM;AACtB,oBAAQ,iBAAiB,MAAM;AAC/B,6BAAiB,SAAS;AAC1B;AAAA,UACF;AAEA,+BAAqB,gBAAgB;AAAA,YACnC;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AAEA,aAAO;AAAA,QACL,QAAQ,OAAO;AAAA,UACb,IAAI,gBAGF;AAAA,YACA,WAAW,CAAC,OAAO,eAAe;AAChC,kBAAI,MAAM,SAAS,cAAc;AAC/B,2BAAW,QAAQ,KAAK;AACxB;AAAA,cACF;AAEA,kBAAI,qBAAqB,MAAM,EAAE,KAAK,MAAM;AAC1C,qCAAqB,MAAM,EAAE,IAAI;AAAA,kBAC/B,kBAAkB;AAAA,kBAClB,aAAa;AAAA,kBACb,aAAa;AAAA,kBACb,aAAa;AAAA,kBACb,QAAQ;AAAA,kBACR,WAAW;AAAA,kBACX,QAAQ,MAAM;AAAA,gBAChB;AAAA,cACF;AAEA,oBAAM,mBAAmB,qBAAqB,MAAM,EAAE;AACtD,+BAAiB,UAAU,MAAM;AACjC,4BAAc,kBAAkB,UAAU;AAAA,YAC5C;AAAA,UACF,CAAC;AAAA,QACH;AAAA,QACA,GAAG;AAAA,MACL;AAAA,IACF;AAAA,EACF;AACF;","names":["nextParams"]}

package/dist/index.d.cts

@@ -0,0 +1,11 @@
+import { LanguageModelV3Middleware } from '@ai-sdk/provider';
+export { extractReasoningMiddleware, getPotentialStartIndex } from './reasoning-parser.cjs';
+
+type SystemPromptPlacement = "first" | "last";
+interface DefaultSystemPromptMiddlewareOptions {
+    systemPrompt: string;
+    placement?: SystemPromptPlacement;
+}
+declare function defaultSystemPromptMiddleware({ systemPrompt, placement, }: DefaultSystemPromptMiddlewareOptions): LanguageModelV3Middleware;
+
+export { defaultSystemPromptMiddleware };

package/dist/index.d.ts

@@ -0,0 +1,11 @@
+import { LanguageModelV3Middleware } from '@ai-sdk/provider';
+export { extractReasoningMiddleware, getPotentialStartIndex } from './reasoning-parser.js';
+
+type SystemPromptPlacement = "first" | "last";
+interface DefaultSystemPromptMiddlewareOptions {
+    systemPrompt: string;
+    placement?: SystemPromptPlacement;
+}
+declare function defaultSystemPromptMiddleware({ systemPrompt, placement, }: DefaultSystemPromptMiddlewareOptions): LanguageModelV3Middleware;
+
+export { defaultSystemPromptMiddleware };

package/dist/index.js

@@ -0,0 +1,110 @@
+import {
+  extractReasoningMiddleware,
+  getPotentialStartIndex
+} from "./chunk-Q2TVVB35.js";
+
+// src/default-system-prompt.ts
+function extractSystemText(content) {
+  if (typeof content === "string") {
+    return content;
+  }
+  if (!Array.isArray(content)) {
+    if (content == null) {
+      return;
+    }
+    return String(content);
+  }
+  const parts = content.map((part) => {
+    var _a;
+    if ((part == null ? void 0 : part.type) === "text" && "text" in part) {
+      return String((_a = part.text) != null ? _a : "");
+    }
+    return JSON.stringify(part);
+  });
+  const textParts = parts.filter((value) => value.length > 0);
+  if (textParts.length === 0) {
+    return;
+  }
+  return textParts.join("\n");
+}
+function mergeSystemPrompts({
+  base,
+  addition,
+  placement
+}) {
+  if (!base) {
+    return addition;
+  }
+  if (addition.length === 0) {
+    return base;
+  }
+  return placement === "first" ? `${addition}
+
+${base}` : `${base}
+
+${addition}`;
+}
+function ensurePromptArray(prompt) {
+  if (!prompt) {
+    return [];
+  }
+  return [...prompt];
+}
+function defaultSystemPromptMiddleware({
+  systemPrompt,
+  placement = "first"
+}) {
+  return {
+    specificationVersion: "v3",
+    transformParams: ({ params }) => {
+      const prompt = ensurePromptArray(params.prompt);
+      const systemIndex = prompt.findIndex(
+        (message) => message.role === "system"
+      );
+      if (systemIndex === -1) {
+        const promptWithSystem = placement === "first" ? [
+          {
+            role: "system",
+            content: systemPrompt
+          },
+          ...prompt
+        ] : [
+          ...prompt,
+          {
+            role: "system",
+            content: systemPrompt
+          }
+        ];
+        const nextParams2 = {
+          ...params,
+          prompt: promptWithSystem
+        };
+        return Promise.resolve(nextParams2);
+      }
+      const systemMessage = prompt[systemIndex];
+      const baseText = extractSystemText(systemMessage.content);
+      const mergedContent = mergeSystemPrompts({
+        base: baseText,
+        addition: systemPrompt,
+        placement
+      });
+      const updatedPrompt = prompt.map(
+        (message, index) => index === systemIndex ? {
+          ...message,
+          content: mergedContent
+        } : message
+      );
+      const nextParams = {
+        ...params,
+        prompt: updatedPrompt
+      };
+      return Promise.resolve(nextParams);
+    }
+  };
+}
+export {
+  defaultSystemPromptMiddleware,
+  extractReasoningMiddleware,
+  getPotentialStartIndex
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/default-system-prompt.ts"],"sourcesContent":["import type {\n  LanguageModelV3CallOptions,\n  LanguageModelV3Content,\n  LanguageModelV3Middleware,\n  LanguageModelV3Prompt,\n} from \"@ai-sdk/provider\";\n\ntype SystemPromptPlacement = \"first\" | \"last\";\n\ninterface DefaultSystemPromptMiddlewareOptions {\n  systemPrompt: string;\n  placement?: SystemPromptPlacement;\n}\n\nfunction extractSystemText(content: unknown): string | undefined {\n  if (typeof content === \"string\") {\n    return content;\n  }\n\n  if (!Array.isArray(content)) {\n    if (content == null) {\n      return;\n    }\n    return String(content);\n  }\n\n  const parts = (content as LanguageModelV3Content[]).map((part) => {\n    if (part?.type === \"text\" && \"text\" in part) {\n      return String(part.text ?? \"\");\n    }\n\n    return JSON.stringify(part);\n  });\n\n  const textParts = parts.filter((value) => value.length > 0);\n  if (textParts.length === 0) {\n    return;\n  }\n\n  return textParts.join(\"\\n\");\n}\n\nfunction mergeSystemPrompts({\n  base,\n  addition,\n  placement,\n}: {\n  base?: string;\n  addition: string;\n  placement: SystemPromptPlacement;\n}): string {\n  if (!base) {\n    return addition;\n  }\n\n  if (addition.length === 0) {\n    return base;\n  }\n\n  return placement === \"first\"\n    ? `${addition}\\n\\n${base}`\n    : `${base}\\n\\n${addition}`;\n}\n\nfunction ensurePromptArray(\n  prompt?: LanguageModelV3Prompt\n): LanguageModelV3Prompt {\n  if (!prompt) {\n    return [];\n  }\n\n  return [...prompt];\n}\n\nexport function defaultSystemPromptMiddleware({\n  systemPrompt,\n  placement = \"first\",\n}: DefaultSystemPromptMiddlewareOptions): LanguageModelV3Middleware {\n  return {\n    specificationVersion: \"v3\",\n    transformParams: ({ params }) => {\n      const prompt = ensurePromptArray(params.prompt);\n      const systemIndex = prompt.findIndex(\n        (message) => message.role === \"system\"\n      );\n\n      if (systemIndex === -1) {\n        const promptWithSystem =\n          placement === \"first\"\n            ? ([\n                {\n                  role: \"system\" as const,\n                  content: systemPrompt,\n                },\n                ...prompt,\n              ] as LanguageModelV3Prompt)\n            : ([\n                ...prompt,\n                {\n                  role: \"system\" as const,\n                  content: systemPrompt,\n                },\n              ] as LanguageModelV3Prompt);\n\n        const nextParams: LanguageModelV3CallOptions = {\n          ...params,\n          prompt: promptWithSystem,\n        };\n\n        return Promise.resolve<LanguageModelV3CallOptions>(nextParams);\n      }\n\n      const systemMessage = prompt[systemIndex];\n      const baseText = extractSystemText(systemMessage.content);\n      const mergedContent = mergeSystemPrompts({\n        base: baseText,\n        addition: systemPrompt,\n        placement,\n      });\n\n      const updatedPrompt = prompt.map((message, index) =>\n        index === systemIndex\n          ? {\n              ...message,\n              content: mergedContent,\n            }\n          : message\n      ) as LanguageModelV3Prompt;\n\n      const nextParams: LanguageModelV3CallOptions = {\n        ...params,\n        prompt: updatedPrompt,\n      };\n\n      return Promise.resolve<LanguageModelV3CallOptions>(nextParams);\n    },\n  };\n}\n"],"mappings":";;;;;;AAcA,SAAS,kBAAkB,SAAsC;AAC/D,MAAI,OAAO,YAAY,UAAU;AAC/B,WAAO;AAAA,EACT;AAEA,MAAI,CAAC,MAAM,QAAQ,OAAO,GAAG;AAC3B,QAAI,WAAW,MAAM;AACnB;AAAA,IACF;AACA,WAAO,OAAO,OAAO;AAAA,EACvB;AAEA,QAAM,QAAS,QAAqC,IAAI,CAAC,SAAS;AA1BpE;AA2BI,SAAI,6BAAM,UAAS,UAAU,UAAU,MAAM;AAC3C,aAAO,QAAO,UAAK,SAAL,YAAa,EAAE;AAAA,IAC/B;AAEA,WAAO,KAAK,UAAU,IAAI;AAAA,EAC5B,CAAC;AAED,QAAM,YAAY,MAAM,OAAO,CAAC,UAAU,MAAM,SAAS,CAAC;AAC1D,MAAI,UAAU,WAAW,GAAG;AAC1B;AAAA,EACF;AAEA,SAAO,UAAU,KAAK,IAAI;AAC5B;AAEA,SAAS,mBAAmB;AAAA,EAC1B;AAAA,EACA;AAAA,EACA;AACF,GAIW;AACT,MAAI,CAAC,MAAM;AACT,WAAO;AAAA,EACT;AAEA,MAAI,SAAS,WAAW,GAAG;AACzB,WAAO;AAAA,EACT;AAEA,SAAO,cAAc,UACjB,GAAG,QAAQ;AAAA;AAAA,EAAO,IAAI,KACtB,GAAG,IAAI;AAAA;AAAA,EAAO,QAAQ;AAC5B;AAEA,SAAS,kBACP,QACuB;AACvB,MAAI,CAAC,QAAQ;AACX,WAAO,CAAC;AAAA,EACV;AAEA,SAAO,CAAC,GAAG,MAAM;AACnB;AAEO,SAAS,8BAA8B;AAAA,EAC5C;AAAA,EACA,YAAY;AACd,GAAoE;AAClE,SAAO;AAAA,IACL,sBAAsB;AAAA,IACtB,iBAAiB,CAAC,EAAE,OAAO,MAAM;AAC/B,YAAM,SAAS,kBAAkB,OAAO,MAAM;AAC9C,YAAM,cAAc,OAAO;AAAA,QACzB,CAAC,YAAY,QAAQ,SAAS;AAAA,MAChC;AAEA,UAAI,gBAAgB,IAAI;AACtB,cAAM,mBACJ,cAAc,UACT;AAAA,UACC;AAAA,YACE,MAAM;AAAA,YACN,SAAS;AAAA,UACX;AAAA,UACA,GAAG;AAAA,QACL,IACC;AAAA,UACC,GAAG;AAAA,UACH;AAAA,YACE,MAAM;AAAA,YACN,SAAS;AAAA,UACX;AAAA,QACF;AAEN,cAAMA,cAAyC;AAAA,UAC7C,GAAG;AAAA,UACH,QAAQ;AAAA,QACV;AAEA,eAAO,QAAQ,QAAoCA,WAAU;AAAA,MAC/D;AAEA,YAAM,gBAAgB,OAAO,WAAW;AACxC,YAAM,WAAW,kBAAkB,cAAc,OAAO;AACxD,YAAM,gBAAgB,mBAAmB;AAAA,QACvC,MAAM;AAAA,QACN,UAAU;AAAA,QACV;AAAA,MACF,CAAC;AAED,YAAM,gBAAgB,OAAO;AAAA,QAAI,CAAC,SAAS,UACzC,UAAU,cACN;AAAA,UACE,GAAG;AAAA,UACH,SAAS;AAAA,QACX,IACA;AAAA,MACN;AAEA,YAAM,aAAyC;AAAA,QAC7C,GAAG;AAAA,QACH,QAAQ;AAAA,MACV;AAEA,aAAO,QAAQ,QAAoC,UAAU;AAAA,IAC/D;AAAA,EACF;AACF;","names":["nextParams"]}

package/dist/reasoning-parser.cjs

@@ -0,0 +1,235 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/reasoning-parser/index.ts
+var reasoning_parser_exports = {};
+__export(reasoning_parser_exports, {
+  extractReasoningMiddleware: () => extractReasoningMiddleware,
+  getPotentialStartIndex: () => getPotentialStartIndex
+});
+module.exports = __toCommonJS(reasoning_parser_exports);
+function getPotentialStartIndex(text, searchedText) {
+  if (searchedText.length === 0) {
+    return null;
+  }
+  const directIndex = text.indexOf(searchedText);
+  if (directIndex !== -1) {
+    return directIndex;
+  }
+  for (let i = text.length - 1; i >= 0; i -= 1) {
+    const suffix = text.substring(i);
+    if (searchedText.startsWith(suffix)) {
+      return i;
+    }
+  }
+  return null;
+}
+function extractReasoningMiddleware({
+  openingTag,
+  closingTag,
+  separator = "\n",
+  startWithReasoning = false
+}) {
+  function processTextPart(text, transformedContent) {
+    var _a;
+    const regexp = new RegExp(`${openingTag}(.*?)${closingTag}`, "gs");
+    const matches = Array.from(text.matchAll(regexp));
+    if (!matches.length) {
+      return;
+    }
+    const reasoningText = matches.map((match) => match[1]).join(separator);
+    let textWithoutReasoning = text;
+    for (let i = matches.length - 1; i >= 0; i -= 1) {
+      const match = matches[i];
+      const beforeMatch = textWithoutReasoning.slice(0, match.index);
+      const matchIndex = (_a = match.index) != null ? _a : 0;
+      const afterMatch = textWithoutReasoning.slice(
+        matchIndex + match[0].length
+      );
+      textWithoutReasoning = beforeMatch + (beforeMatch.length > 0 && afterMatch.length > 0 ? separator : "") + afterMatch;
+    }
+    transformedContent.push({
+      type: "reasoning",
+      text: reasoningText
+    });
+    transformedContent.push({
+      type: "text",
+      text: textWithoutReasoning
+    });
+  }
+  return {
+    specificationVersion: "v3",
+    wrapGenerate: async ({ doGenerate }) => {
+      const { content, ...rest } = await doGenerate();
+      const transformedContent = [];
+      for (const part of content) {
+        if (part.type !== "text") {
+          transformedContent.push(part);
+          continue;
+        }
+        const text = startWithReasoning ? openingTag + part.text : part.text;
+        const regexp = new RegExp(`${openingTag}(.*?)${closingTag}`, "gs");
+        const matches = Array.from(text.matchAll(regexp));
+        if (!matches.length) {
+          transformedContent.push(part);
+          continue;
+        }
+        processTextPart(text, transformedContent);
+      }
+      return { content: transformedContent, ...rest };
+    },
+    wrapStream: async ({ doStream }) => {
+      const { stream, ...rest } = await doStream();
+      const reasoningExtractions = {};
+      function createPublisher(activeExtraction, controller) {
+        return (text) => {
+          if (text.length === 0) {
+            return;
+          }
+          const prefix = getPrefix(activeExtraction);
+          enqueueReasoningStart(activeExtraction, controller);
+          enqueueDelta(activeExtraction, controller, prefix, text);
+          updateExtractionState(activeExtraction);
+        };
+      }
+      function getPrefix(activeExtraction) {
+        return activeExtraction.afterSwitch && (activeExtraction.isReasoning ? !activeExtraction.isFirstReasoning : !activeExtraction.isFirstText) ? separator : "";
+      }
+      function enqueueReasoningStart(activeExtraction, controller) {
+        if (activeExtraction.afterSwitch && activeExtraction.isReasoning || activeExtraction.isFirstReasoning) {
+          controller.enqueue({
+            type: "reasoning-start",
+            id: `reasoning-${activeExtraction.idCounter}`
+          });
+        }
+      }
+      function enqueueDelta(activeExtraction, controller, prefix, text) {
+        controller.enqueue(
+          activeExtraction.isReasoning ? {
+            type: "reasoning-delta",
+            delta: prefix + text,
+            id: `reasoning-${activeExtraction.idCounter}`
+          } : {
+            type: "text-delta",
+            delta: prefix + text,
+            id: activeExtraction.textId
+          }
+        );
+      }
+      function updateExtractionState(activeExtraction) {
+        activeExtraction.afterSwitch = false;
+        if (activeExtraction.isReasoning) {
+          activeExtraction.isFirstReasoning = false;
+        } else {
+          activeExtraction.isFirstText = false;
+        }
+      }
+      function handleFullMatch(activeExtraction, controller, startIndex, nextTag) {
+        activeExtraction.buffer = activeExtraction.buffer.slice(
+          startIndex + nextTag.length
+        );
+        if (activeExtraction.isReasoning) {
+          controller.enqueue({
+            type: "reasoning-end",
+            id: `reasoning-${activeExtraction.idCounter}`
+          });
+          activeExtraction.idCounter += 1;
+        }
+        activeExtraction.isReasoning = !activeExtraction.isReasoning;
+        activeExtraction.afterSwitch = true;
+      }
+      function processTagMatch({
+        activeExtraction,
+        controller,
+        publish,
+        startIndex,
+        nextTag
+      }) {
+        publish(activeExtraction.buffer.slice(0, startIndex));
+        const foundFullMatch = startIndex + nextTag.length <= activeExtraction.buffer.length;
+        if (foundFullMatch) {
+          handleFullMatch(activeExtraction, controller, startIndex, nextTag);
+          return true;
+        }
+        activeExtraction.buffer = activeExtraction.buffer.slice(startIndex);
+        return false;
+      }
+      function processBuffer(activeExtraction, controller) {
+        const publish = createPublisher(activeExtraction, controller);
+        let continueProcessing = true;
+        while (continueProcessing) {
+          const nextTag = activeExtraction.isReasoning ? closingTag : openingTag;
+          const startIndex = getPotentialStartIndex(
+            activeExtraction.buffer,
+            nextTag
+          );
+          if (startIndex == null) {
+            publish(activeExtraction.buffer);
+            activeExtraction.buffer = "";
+            break;
+          }
+          continueProcessing = processTagMatch({
+            activeExtraction,
+            controller,
+            publish,
+            startIndex,
+            nextTag
+          });
+        }
+      }
+      return {
+        stream: stream.pipeThrough(
+          new TransformStream({
+            transform: (chunk, controller) => {
+              if (chunk.type !== "text-delta") {
+                controller.enqueue(chunk);
+                return;
+              }
+              if (reasoningExtractions[chunk.id] == null) {
+                reasoningExtractions[chunk.id] = {
+                  isFirstReasoning: true,
+                  isFirstText: true,
+                  afterSwitch: false,
+                  isReasoning: startWithReasoning,
+                  buffer: "",
+                  idCounter: 0,
+                  textId: chunk.id
+                };
+              }
+              const activeExtraction = reasoningExtractions[chunk.id];
+              activeExtraction.buffer += chunk.delta;
+              processBuffer(activeExtraction, controller);
+            }
+          })
+        ),
+        ...rest
+      };
+    }
+  };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  extractReasoningMiddleware,
+  getPotentialStartIndex
+});
+/**
+ * @license
+ * Copyright (c) 2021-present, FriendliAI Inc. All rights reserved.
+ */
+//# sourceMappingURL=reasoning-parser.cjs.map

package/dist/reasoning-parser.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/reasoning-parser/index.ts"],"sourcesContent":["/**\n * @license\n * Copyright (c) 2021-present, FriendliAI Inc. All rights reserved.\n */\n\nimport type {\n  LanguageModelV3Content,\n  LanguageModelV3Middleware,\n  LanguageModelV3StreamPart,\n} from \"@ai-sdk/provider\";\n\n/**\n * All code below is forked from the following link:\n * https://github.com/vercel/ai/blob/v5/packages/ai/core/middleware/extract-reasoning-middleware.ts\n */\n\n/**\n * Returns the index of the start of the searchedText in the text, or null if it\n * is not found.\n */\nexport function getPotentialStartIndex(\n  text: string,\n  searchedText: string\n): number | null {\n  // Return null immediately if searchedText is empty.\n  if (searchedText.length === 0) {\n    return null;\n  }\n\n  // Check if the searchedText exists as a direct substring of text.\n  const directIndex = text.indexOf(searchedText);\n  if (directIndex !== -1) {\n    return directIndex;\n  }\n\n  // Otherwise, look for the largest suffix of \"text\" that matches\n  // a prefix of \"searchedText\". We go from the end of text inward.\n  for (let i = text.length - 1; i >= 0; i -= 1) {\n    const suffix = text.substring(i);\n    if (searchedText.startsWith(suffix)) {\n      return i;\n    }\n  }\n\n  return null;\n}\n\n/**\n * Extract an XML-tagged reasoning section from the generated text and exposes it\n * as a `reasoning` property on the result.\n *\n * @param openingTag - The opening XML tag to extract reasoning from.\n * @param closingTag - The closing XML tag to extract reasoning from.\n * @param separator - The separator to use between reasoning and text sections.\n * @param startWithReasoning - Whether to start with reasoning tokens.\n */\nexport function extractReasoningMiddleware({\n  openingTag,\n  closingTag,\n  separator = \"\\n\",\n  startWithReasoning = false,\n}: {\n  openingTag: string;\n  closingTag: string;\n  separator?: string;\n  startWithReasoning?: boolean;\n}): LanguageModelV3Middleware {\n  function processTextPart(\n    text: string,\n    transformedContent: LanguageModelV3Content[]\n  ) {\n    const regexp = new RegExp(`${openingTag}(.*?)${closingTag}`, \"gs\");\n    const matches = Array.from(text.matchAll(regexp));\n\n    if (!matches.length) {\n      return;\n    }\n\n    const reasoningText = matches.map((match) => match[1]).join(separator);\n\n    let textWithoutReasoning = text;\n    for (let i = matches.length - 1; i >= 0; i -= 1) {\n      const match = matches[i];\n\n      const beforeMatch = textWithoutReasoning.slice(0, match.index);\n      const matchIndex = match.index ?? 0;\n      const afterMatch = textWithoutReasoning.slice(\n        matchIndex + match[0].length\n      );\n\n      textWithoutReasoning =\n        beforeMatch +\n        (beforeMatch.length > 0 && afterMatch.length > 0 ? separator : \"\") +\n        afterMatch;\n    }\n\n    transformedContent.push({\n      type: \"reasoning\",\n      text: reasoningText,\n    });\n\n    transformedContent.push({\n      type: \"text\",\n      text: textWithoutReasoning,\n    });\n  }\n\n  return {\n    specificationVersion: \"v3\",\n    wrapGenerate: async ({ doGenerate }) => {\n      const { content, ...rest } = await doGenerate();\n\n      const transformedContent: LanguageModelV3Content[] = [];\n      for (const part of content) {\n        if (part.type !== \"text\") {\n          transformedContent.push(part);\n          continue;\n        }\n\n        const text = startWithReasoning ? openingTag + part.text : part.text;\n        const regexp = new RegExp(`${openingTag}(.*?)${closingTag}`, \"gs\");\n        const matches = Array.from(text.matchAll(regexp));\n\n        if (!matches.length) {\n          transformedContent.push(part);\n          continue;\n        }\n\n        processTextPart(text, transformedContent);\n      }\n\n      return { content: transformedContent, ...rest };\n    },\n\n    wrapStream: async ({ doStream }) => {\n      const { stream, ...rest } = await doStream();\n\n      interface ExtractionState {\n        isFirstReasoning: boolean;\n        isFirstText: boolean;\n        afterSwitch: boolean;\n        isReasoning: boolean;\n        buffer: string;\n        idCounter: number;\n        textId: string;\n      }\n\n      const reasoningExtractions: Record<string, ExtractionState> = {};\n\n      function createPublisher(\n        activeExtraction: ExtractionState,\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>\n      ) {\n        return (text: string) => {\n          if (text.length === 0) {\n            return;\n          }\n\n          const prefix = getPrefix(activeExtraction);\n          enqueueReasoningStart(activeExtraction, controller);\n          enqueueDelta(activeExtraction, controller, prefix, text);\n          updateExtractionState(activeExtraction);\n        };\n      }\n\n      function getPrefix(activeExtraction: ExtractionState): string {\n        return activeExtraction.afterSwitch &&\n          (activeExtraction.isReasoning\n            ? !activeExtraction.isFirstReasoning\n            : !activeExtraction.isFirstText)\n          ? separator\n          : \"\";\n      }\n\n      function enqueueReasoningStart(\n        activeExtraction: ExtractionState,\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>\n      ) {\n        if (\n          (activeExtraction.afterSwitch && activeExtraction.isReasoning) ||\n          activeExtraction.isFirstReasoning\n        ) {\n          controller.enqueue({\n            type: \"reasoning-start\",\n            id: `reasoning-${activeExtraction.idCounter}`,\n          });\n        }\n      }\n\n      function enqueueDelta(\n        activeExtraction: ExtractionState,\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>,\n        prefix: string,\n        text: string\n      ) {\n        controller.enqueue(\n          activeExtraction.isReasoning\n            ? {\n                type: \"reasoning-delta\",\n                delta: prefix + text,\n                id: `reasoning-${activeExtraction.idCounter}`,\n              }\n            : {\n                type: \"text-delta\",\n                delta: prefix + text,\n                id: activeExtraction.textId,\n              }\n        );\n      }\n\n      function updateExtractionState(activeExtraction: ExtractionState) {\n        activeExtraction.afterSwitch = false;\n        if (activeExtraction.isReasoning) {\n          activeExtraction.isFirstReasoning = false;\n        } else {\n          activeExtraction.isFirstText = false;\n        }\n      }\n\n      function handleFullMatch(\n        activeExtraction: ExtractionState,\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>,\n        startIndex: number,\n        nextTag: string\n      ) {\n        activeExtraction.buffer = activeExtraction.buffer.slice(\n          startIndex + nextTag.length\n        );\n\n        if (activeExtraction.isReasoning) {\n          controller.enqueue({\n            type: \"reasoning-end\",\n            id: `reasoning-${activeExtraction.idCounter}`,\n          });\n          activeExtraction.idCounter += 1;\n        }\n\n        activeExtraction.isReasoning = !activeExtraction.isReasoning;\n        activeExtraction.afterSwitch = true;\n      }\n\n      function processTagMatch({\n        activeExtraction,\n        controller,\n        publish,\n        startIndex,\n        nextTag,\n      }: {\n        activeExtraction: ExtractionState;\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>;\n        publish: (text: string) => void;\n        startIndex: number;\n        nextTag: string;\n      }): boolean {\n        publish(activeExtraction.buffer.slice(0, startIndex));\n\n        const foundFullMatch =\n          startIndex + nextTag.length <= activeExtraction.buffer.length;\n\n        if (foundFullMatch) {\n          handleFullMatch(activeExtraction, controller, startIndex, nextTag);\n          return true;\n        }\n\n        activeExtraction.buffer = activeExtraction.buffer.slice(startIndex);\n        return false;\n      }\n\n      function processBuffer(\n        activeExtraction: ExtractionState,\n        controller: TransformStreamDefaultController<LanguageModelV3StreamPart>\n      ) {\n        const publish = createPublisher(activeExtraction, controller);\n        let continueProcessing = true;\n\n        while (continueProcessing) {\n          const nextTag = activeExtraction.isReasoning\n            ? closingTag\n            : openingTag;\n          const startIndex = getPotentialStartIndex(\n            activeExtraction.buffer,\n            nextTag\n          );\n\n          if (startIndex == null) {\n            publish(activeExtraction.buffer);\n            activeExtraction.buffer = \"\";\n            break;\n          }\n\n          continueProcessing = processTagMatch({\n            activeExtraction,\n            controller,\n            publish,\n            startIndex,\n            nextTag,\n          });\n        }\n      }\n\n      return {\n        stream: stream.pipeThrough(\n          new TransformStream<\n            LanguageModelV3StreamPart,\n            LanguageModelV3StreamPart\n          >({\n            transform: (chunk, controller) => {\n              if (chunk.type !== \"text-delta\") {\n                controller.enqueue(chunk);\n                return;\n              }\n\n              if (reasoningExtractions[chunk.id] == null) {\n                reasoningExtractions[chunk.id] = {\n                  isFirstReasoning: true,\n                  isFirstText: true,\n                  afterSwitch: false,\n                  isReasoning: startWithReasoning,\n                  buffer: \"\",\n                  idCounter: 0,\n                  textId: chunk.id,\n                };\n              }\n\n              const activeExtraction = reasoningExtractions[chunk.id];\n              activeExtraction.buffer += chunk.delta;\n              processBuffer(activeExtraction, controller);\n            },\n          })\n        ),\n        ...rest,\n      };\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAoBO,SAAS,uBACd,MACA,cACe;AAEf,MAAI,aAAa,WAAW,GAAG;AAC7B,WAAO;AAAA,EACT;AAGA,QAAM,cAAc,KAAK,QAAQ,YAAY;AAC7C,MAAI,gBAAgB,IAAI;AACtB,WAAO;AAAA,EACT;AAIA,WAAS,IAAI,KAAK,SAAS,GAAG,KAAK,GAAG,KAAK,GAAG;AAC5C,UAAM,SAAS,KAAK,UAAU,CAAC;AAC/B,QAAI,aAAa,WAAW,MAAM,GAAG;AACnC,aAAO;AAAA,IACT;AAAA,EACF;AAEA,SAAO;AACT;AAWO,SAAS,2BAA2B;AAAA,EACzC;AAAA,EACA;AAAA,EACA,YAAY;AAAA,EACZ,qBAAqB;AACvB,GAK8B;AAC5B,WAAS,gBACP,MACA,oBACA;AAtEJ;AAuEI,UAAM,SAAS,IAAI,OAAO,GAAG,UAAU,QAAQ,UAAU,IAAI,IAAI;AACjE,UAAM,UAAU,MAAM,KAAK,KAAK,SAAS,MAAM,CAAC;AAEhD,QAAI,CAAC,QAAQ,QAAQ;AACnB;AAAA,IACF;AAEA,UAAM,gBAAgB,QAAQ,IAAI,CAAC,UAAU,MAAM,CAAC,CAAC,EAAE,KAAK,SAAS;AAErE,QAAI,uBAAuB;AAC3B,aAAS,IAAI,QAAQ,SAAS,GAAG,KAAK,GAAG,KAAK,GAAG;AAC/C,YAAM,QAAQ,QAAQ,CAAC;AAEvB,YAAM,cAAc,qBAAqB,MAAM,GAAG,MAAM,KAAK;AAC7D,YAAM,cAAa,WAAM,UAAN,YAAe;AAClC,YAAM,aAAa,qBAAqB;AAAA,QACtC,aAAa,MAAM,CAAC,EAAE;AAAA,MACxB;AAEA,6BACE,eACC,YAAY,SAAS,KAAK,WAAW,SAAS,IAAI,YAAY,MAC/D;AAAA,IACJ;AAEA,uBAAmB,KAAK;AAAA,MACtB,MAAM;AAAA,MACN,MAAM;AAAA,IACR,CAAC;AAED,uBAAmB,KAAK;AAAA,MACtB,MAAM;AAAA,MACN,MAAM;AAAA,IACR,CAAC;AAAA,EACH;AAEA,SAAO;AAAA,IACL,sBAAsB;AAAA,IACtB,cAAc,OAAO,EAAE,WAAW,MAAM;AACtC,YAAM,EAAE,SAAS,GAAG,KAAK,IAAI,MAAM,WAAW;AAE9C,YAAM,qBAA+C,CAAC;AACtD,iBAAW,QAAQ,SAAS;AAC1B,YAAI,KAAK,SAAS,QAAQ;AACxB,6BAAmB,KAAK,IAAI;AAC5B;AAAA,QACF;AAEA,cAAM,OAAO,qBAAqB,aAAa,KAAK,OAAO,KAAK;AAChE,cAAM,SAAS,IAAI,OAAO,GAAG,UAAU,QAAQ,UAAU,IAAI,IAAI;AACjE,cAAM,UAAU,MAAM,KAAK,KAAK,SAAS,MAAM,CAAC;AAEhD,YAAI,CAAC,QAAQ,QAAQ;AACnB,6BAAmB,KAAK,IAAI;AAC5B;AAAA,QACF;AAEA,wBAAgB,MAAM,kBAAkB;AAAA,MAC1C;AAEA,aAAO,EAAE,SAAS,oBAAoB,GAAG,KAAK;AAAA,IAChD;AAAA,IAEA,YAAY,OAAO,EAAE,SAAS,MAAM;AAClC,YAAM,EAAE,QAAQ,GAAG,KAAK,IAAI,MAAM,SAAS;AAY3C,YAAM,uBAAwD,CAAC;AAE/D,eAAS,gBACP,kBACA,YACA;AACA,eAAO,CAAC,SAAiB;AACvB,cAAI,KAAK,WAAW,GAAG;AACrB;AAAA,UACF;AAEA,gBAAM,SAAS,UAAU,gBAAgB;AACzC,gCAAsB,kBAAkB,UAAU;AAClD,uBAAa,kBAAkB,YAAY,QAAQ,IAAI;AACvD,gCAAsB,gBAAgB;AAAA,QACxC;AAAA,MACF;AAEA,eAAS,UAAU,kBAA2C;AAC5D,eAAO,iBAAiB,gBACrB,iBAAiB,cACd,CAAC,iBAAiB,mBAClB,CAAC,iBAAiB,eACpB,YACA;AAAA,MACN;AAEA,eAAS,sBACP,kBACA,YACA;AACA,YACG,iBAAiB,eAAe,iBAAiB,eAClD,iBAAiB,kBACjB;AACA,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN,IAAI,aAAa,iBAAiB,SAAS;AAAA,UAC7C,CAAC;AAAA,QACH;AAAA,MACF;AAEA,eAAS,aACP,kBACA,YACA,QACA,MACA;AACA,mBAAW;AAAA,UACT,iBAAiB,cACb;AAAA,YACE,MAAM;AAAA,YACN,OAAO,SAAS;AAAA,YAChB,IAAI,aAAa,iBAAiB,SAAS;AAAA,UAC7C,IACA;AAAA,YACE,MAAM;AAAA,YACN,OAAO,SAAS;AAAA,YAChB,IAAI,iBAAiB;AAAA,UACvB;AAAA,QACN;AAAA,MACF;AAEA,eAAS,sBAAsB,kBAAmC;AAChE,yBAAiB,cAAc;AAC/B,YAAI,iBAAiB,aAAa;AAChC,2BAAiB,mBAAmB;AAAA,QACtC,OAAO;AACL,2BAAiB,cAAc;AAAA,QACjC;AAAA,MACF;AAEA,eAAS,gBACP,kBACA,YACA,YACA,SACA;AACA,yBAAiB,SAAS,iBAAiB,OAAO;AAAA,UAChD,aAAa,QAAQ;AAAA,QACvB;AAEA,YAAI,iBAAiB,aAAa;AAChC,qBAAW,QAAQ;AAAA,YACjB,MAAM;AAAA,YACN,IAAI,aAAa,iBAAiB,SAAS;AAAA,UAC7C,CAAC;AACD,2BAAiB,aAAa;AAAA,QAChC;AAEA,yBAAiB,cAAc,CAAC,iBAAiB;AACjD,yBAAiB,cAAc;AAAA,MACjC;AAEA,eAAS,gBAAgB;AAAA,QACvB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF,GAMY;AACV,gBAAQ,iBAAiB,OAAO,MAAM,GAAG,UAAU,CAAC;AAEpD,cAAM,iBACJ,aAAa,QAAQ,UAAU,iBAAiB,OAAO;AAEzD,YAAI,gBAAgB;AAClB,0BAAgB,kBAAkB,YAAY,YAAY,OAAO;AACjE,iBAAO;AAAA,QACT;AAEA,yBAAiB,SAAS,iBAAiB,OAAO,MAAM,UAAU;AAClE,eAAO;AAAA,MACT;AAEA,eAAS,cACP,kBACA,YACA;AACA,cAAM,UAAU,gBAAgB,kBAAkB,UAAU;AAC5D,YAAI,qBAAqB;AAEzB,eAAO,oBAAoB;AACzB,gBAAM,UAAU,iBAAiB,cAC7B,aACA;AACJ,gBAAM,aAAa;AAAA,YACjB,iBAAiB;AAAA,YACjB;AAAA,UACF;AAEA,cAAI,cAAc,MAAM;AACtB,oBAAQ,iBAAiB,MAAM;AAC/B,6BAAiB,SAAS;AAC1B;AAAA,UACF;AAEA,+BAAqB,gBAAgB;AAAA,YACnC;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAAA,QACH;AAAA,MACF;AAEA,aAAO;AAAA,QACL,QAAQ,OAAO;AAAA,UACb,IAAI,gBAGF;AAAA,YACA,WAAW,CAAC,OAAO,eAAe;AAChC,kBAAI,MAAM,SAAS,cAAc;AAC/B,2BAAW,QAAQ,KAAK;AACxB;AAAA,cACF;AAEA,kBAAI,qBAAqB,MAAM,EAAE,KAAK,MAAM;AAC1C,qCAAqB,MAAM,EAAE,IAAI;AAAA,kBAC/B,kBAAkB;AAAA,kBAClB,aAAa;AAAA,kBACb,aAAa;AAAA,kBACb,aAAa;AAAA,kBACb,QAAQ;AAAA,kBACR,WAAW;AAAA,kBACX,QAAQ,MAAM;AAAA,gBAChB;AAAA,cACF;AAEA,oBAAM,mBAAmB,qBAAqB,MAAM,EAAE;AACtD,+BAAiB,UAAU,MAAM;AACjC,4BAAc,kBAAkB,UAAU;AAAA,YAC5C;AAAA,UACF,CAAC;AAAA,QACH;AAAA,QACA,GAAG;AAAA,MACL;AAAA,IACF;AAAA,EACF;AACF;","names":[]}

package/dist/reasoning-parser.d.cts

@@ -0,0 +1,33 @@
+import { LanguageModelV3Middleware } from '@ai-sdk/provider';
+
+/**
+ * @license
+ * Copyright (c) 2021-present, FriendliAI Inc. All rights reserved.
+ */
+
+/**
+ * All code below is forked from the following link:
+ * https://github.com/vercel/ai/blob/v5/packages/ai/core/middleware/extract-reasoning-middleware.ts
+ */
+/**
+ * Returns the index of the start of the searchedText in the text, or null if it
+ * is not found.
+ */
+declare function getPotentialStartIndex(text: string, searchedText: string): number | null;
+/**
+ * Extract an XML-tagged reasoning section from the generated text and exposes it
+ * as a `reasoning` property on the result.
+ *
+ * @param openingTag - The opening XML tag to extract reasoning from.
+ * @param closingTag - The closing 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({ openingTag, closingTag, separator, startWithReasoning, }: {
+    openingTag: string;
+    closingTag: string;
+    separator?: string;
+    startWithReasoning?: boolean;
+}): LanguageModelV3Middleware;
+
+export { extractReasoningMiddleware, getPotentialStartIndex };

package/dist/reasoning-parser.d.ts

@@ -0,0 +1,33 @@
+import { LanguageModelV3Middleware } from '@ai-sdk/provider';
+
+/**
+ * @license
+ * Copyright (c) 2021-present, FriendliAI Inc. All rights reserved.
+ */
+
+/**
+ * All code below is forked from the following link:
+ * https://github.com/vercel/ai/blob/v5/packages/ai/core/middleware/extract-reasoning-middleware.ts
+ */
+/**
+ * Returns the index of the start of the searchedText in the text, or null if it
+ * is not found.
+ */
+declare function getPotentialStartIndex(text: string, searchedText: string): number | null;
+/**
+ * Extract an XML-tagged reasoning section from the generated text and exposes it
+ * as a `reasoning` property on the result.
+ *
+ * @param openingTag - The opening XML tag to extract reasoning from.
+ * @param closingTag - The closing 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({ openingTag, closingTag, separator, startWithReasoning, }: {
+    openingTag: string;
+    closingTag: string;
+    separator?: string;
+    startWithReasoning?: boolean;
+}): LanguageModelV3Middleware;
+
+export { extractReasoningMiddleware, getPotentialStartIndex };

package/dist/reasoning-parser.js

@@ -0,0 +1,9 @@
+import {
+  extractReasoningMiddleware,
+  getPotentialStartIndex
+} from "./chunk-Q2TVVB35.js";
+export {
+  extractReasoningMiddleware,
+  getPotentialStartIndex
+};
+//# sourceMappingURL=reasoning-parser.js.map

package/dist/reasoning-parser.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/package.json

@@ -0,0 +1,54 @@
+{
+  "name": "@ai-sdk-tool/middleware",
+  "version": "0.0.0",
+  "description": "Collection of reusable AI SDK middlewares",
+  "type": "module",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/minpeter/ai-sdk-tool-call-middleware",
+    "directory": "packages/middleware"
+  },
+  "main": "./dist/index.js",
+  "module": "./dist/index.mjs",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "require": "./dist/index.cjs"
+    },
+    "./reasoning-parser": {
+      "types": "./dist/reasoning-parser.d.ts",
+      "import": "./dist/reasoning-parser.js",
+      "require": "./dist/reasoning-parser.cjs"
+    }
+  },
+  "files": [
+    "dist/**/*"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "pnpm clean && tsup --tsconfig tsconfig.build.json",
+    "build:watch": "pnpm clean && tsup --watch --tsconfig tsconfig.build.json",
+    "clean": "rm -rf dist *.tsbuildinfo",
+    "typecheck": "tsc --noEmit",
+    "test": "vitest run"
+  },
+  "dependencies": {
+    "@ai-sdk/provider": "3.0.0"
+  },
+  "devDependencies": {
+    "@ai-sdkx/tsconfig": "workspace:*",
+    "@types/node": "^25.0.3",
+    "tsup": "^8.5.1"
+  },
+  "keywords": [
+    "ai",
+    "sdk",
+    "middleware"
+  ],
+  "author": "",
+  "license": "Apache-2.0"
+}