@ai-sdk-tool/parser 3.0.0-canary.0 → 3.0.0-canary.1

package/dist/index.cjs

@@ -20,7 +20,6 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var src_exports = {};
 __export(src_exports, {
-  RJSON: () => robust_json_exports,
   createDynamicIfThenElseSchema: () => createDynamicIfThenElseSchema,
   createToolMiddleware: () => createToolMiddleware,
   decodeOriginalTools: () => decodeOriginalTools,
@@ -37,6 +36,7 @@ __export(src_exports, {
   isToolChoiceActive: () => isToolChoiceActive,
   isToolResultPart: () => isToolResultPart,
   jsonMixProtocol: () => jsonMixProtocol,
+  logParseFailure: () => logParseFailure,
   logParsedChunk: () => logParsedChunk,
   logParsedSummary: () => logParsedSummary,
   logRawChunk: () => logRawChunk,
@@ -49,11 +49,8 @@ __export(src_exports, {
 });
 module.exports = __toCommonJS(src_exports);
 
-// src/protocols/dummy-protocol.ts
-var import_provider_utils = require("@ai-sdk/provider-utils");
-
 // src/protocols/json-mix-protocol.ts
-var import_provider_utils2 = require("@ai-sdk/provider-utils");
+var import_provider_utils = require("@ai-sdk/provider-utils");
 
 // src/utils/debug.ts
 var LINE_SPLIT_REGEX = /\r?\n/;
@@ -101,6 +98,7 @@ var cBgGreen = color(ANSI_BG_GREEN);
 var cInverse = color(ANSI_INVERSE);
 var cUnderline = color(ANSI_UNDERLINE);
 var cBold = color(ANSI_BOLD);
+var MAX_SNIPPET_LENGTH = 800;
 function safeStringify(value) {
   try {
     return `
@@ -109,6 +107,41 @@ ${typeof value === "string" ? value : JSON.stringify(value, null, 2)}`;
     return String(value);
   }
 }
+function formatError(error) {
+  if (error instanceof Error) {
+    const stack = error.stack ? `
+${error.stack}` : "";
+    return `
+${error.name}: ${error.message}${stack}`;
+  }
+  return safeStringify(error);
+}
+function truncateSnippet(snippet) {
+  if (snippet.length <= MAX_SNIPPET_LENGTH) {
+    return snippet;
+  }
+  return `${snippet.slice(0, MAX_SNIPPET_LENGTH)}
+\u2026[truncated ${snippet.length - MAX_SNIPPET_LENGTH} chars]`;
+}
+function logParseFailure({
+  phase,
+  reason,
+  snippet,
+  error
+}) {
+  if (getDebugLevel() !== "parse") {
+    return;
+  }
+  const label = cBgBlue(`[${phase}]`);
+  console.log(cGray("[debug:mw:fail]"), label, cYellow(reason));
+  if (snippet) {
+    const formatted = truncateSnippet(snippet);
+    console.log(cGray("[debug:mw:fail:snippet]"), formatted);
+  }
+  if (error) {
+    console.log(cGray("[debug:mw:fail:error]"), cCyan(formatError(error)));
+  }
+}
 function logRawChunk(part) {
   console.log(cGray("[debug:mw:raw]"), cYellow(safeStringify(part)));
 }
@@ -174,63 +207,6 @@ ${rendered}`);
   }
 }
 
-// src/utils/dynamic-tool-schema.ts
-function createDynamicIfThenElseSchema(tools) {
-  let currentSchema = {};
-  const toolNames = [];
-  for (let i = tools.length - 1; i >= 0; i -= 1) {
-    const tool = tools[i];
-    if (tool.type === "provider-defined") {
-      throw new Error(
-        "Provider-defined tools are not supported by this middleware. Please use custom tools."
-      );
-    }
-    toolNames.unshift(tool.name);
-    const toolCondition = {
-      if: {
-        properties: {
-          name: {
-            const: tool.name
-          }
-        },
-        required: ["name"]
-      },
-      // biome-ignore lint/suspicious/noThenProperty: JSON Schema uses 'then' as a keyword
-      then: {
-        properties: {
-          name: {
-            const: tool.name
-          },
-          arguments: tool.inputSchema
-        },
-        required: ["name", "arguments"]
-      }
-    };
-    if (Object.keys(currentSchema).length > 0) {
-      toolCondition.else = currentSchema;
-    }
-    currentSchema = toolCondition;
-  }
-  return {
-    type: "object",
-    // Explicitly specify type as "object"
-    properties: {
-      name: {
-        type: "string",
-        description: "Name of the tool to call",
-        enum: toolNames
-      },
-      arguments: {
-        type: "object",
-        // By default, arguments is also specified as object type
-        description: "Argument object to be passed to the tool"
-      }
-    },
-    required: ["name", "arguments"],
-    ...currentSchema
-  };
-}
-
 // src/utils/get-potential-start-index.ts
 function getPotentialStartIndex(text, searchedText) {
   if (searchedText.length === 0) {
@@ -249,60 +225,12 @@ function getPotentialStartIndex(text, searchedText) {
   return null;
 }
 
-// src/utils/on-error.ts
-function extractOnErrorOption(providerOptions) {
-  var _a;
-  if (providerOptions && typeof providerOptions === "object") {
-    const onError = (_a = providerOptions.toolCallMiddleware) == null ? void 0 : _a.onError;
-    return onError ? { onError } : void 0;
-  }
-  return;
-}
-
-// src/utils/provider-options.ts
-var originalToolsSchema = {
-  encode: encodeOriginalTools,
-  decode: decodeOriginalTools
-};
-function encodeOriginalTools(tools) {
-  return (tools == null ? void 0 : tools.map((t) => ({
-    name: t.name,
-    inputSchema: JSON.stringify(t.inputSchema)
-  }))) || [];
-}
-function decodeOriginalTools(originalTools) {
-  if (!originalTools) {
-    return [];
-  }
-  return originalTools.map(
-    (t) => ({
-      type: "function",
-      name: t.name,
-      inputSchema: JSON.parse(t.inputSchema)
-    })
-  );
-}
-function extractToolNamesFromOriginalTools(originalTools) {
-  return (originalTools == null ? void 0 : originalTools.map((t) => t.name)) || [];
-}
-function isToolChoiceActive(params) {
-  var _a, _b, _c;
-  const toolChoice = (_b = (_a = params.providerOptions) == null ? void 0 : _a.toolCallMiddleware) == null ? void 0 : _b.toolChoice;
-  return !!(typeof params.providerOptions === "object" && params.providerOptions !== null && typeof ((_c = params.providerOptions) == null ? void 0 : _c.toolCallMiddleware) === "object" && toolChoice && typeof toolChoice === "object" && (toolChoice.type === "tool" || toolChoice.type === "required"));
-}
-
 // src/utils/regex.ts
 function escapeRegExp(literal) {
   return literal.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
 }
 
 // src/utils/robust-json.ts
-var robust_json_exports = {};
-__export(robust_json_exports, {
-  parse: () => parse,
-  stringify: () => stringify,
-  transform: () => transform
-});
 var WHITESPACE_TEST_REGEX = /\s/;
 var WHITESPACE_REGEX = /^\s+/;
 var OBJECT_START_REGEX = /^\{/;
@@ -967,19 +895,6 @@ function stringify(obj) {
   return "null";
 }
 
-// src/utils/type-guards.ts
-function isToolCallContent(content) {
-  return content.type === "tool-call" && typeof content.toolName === "string" && // input may be a JSON string or an already-parsed object depending on provider/runtime
-  (typeof content.input === "string" || typeof content.input === "object");
-}
-function isToolResultPart(content) {
-  const c = content;
-  return !!c && c.type === "tool-result" && typeof c.toolName === "string" && typeof c.toolCallId === "string" && "output" in c;
-}
-function hasInputProperty(obj) {
-  return typeof obj === "object" && obj !== null && "input" in obj;
-}
-
 // src/protocols/json-mix-protocol.ts
 function processToolCallJson(toolCallJson, fullMatch, processedElements, options) {
   var _a;
@@ -987,11 +902,17 @@ function processToolCallJson(toolCallJson, fullMatch, processedElements, options
     const parsedToolCall = parse(toolCallJson);
     processedElements.push({
       type: "tool-call",
-      toolCallId: (0, import_provider_utils2.generateId)(),
+      toolCallId: (0, import_provider_utils.generateId)(),
       toolName: parsedToolCall.name,
       input: JSON.stringify((_a = parsedToolCall.arguments) != null ? _a : {})
     });
   } catch (error) {
+    logParseFailure({
+      phase: "generated-text",
+      reason: "Failed to parse tool call JSON segment",
+      snippet: fullMatch,
+      error
+    });
     if (options == null ? void 0 : options.onError) {
       options.onError(
         "Could not process JSON tool call, keeping original text.",
@@ -1024,7 +945,7 @@ function flushBuffer(state, controller, toolCallStart) {
     return;
   }
   if (!state.currentTextId) {
-    state.currentTextId = (0, import_provider_utils2.generateId)();
+    state.currentTextId = (0, import_provider_utils.generateId)();
     controller.enqueue({ type: "text-start", id: state.currentTextId });
     state.hasEmittedTextStart = true;
   }
@@ -1047,7 +968,12 @@ function emitIncompleteToolCall(state, controller, toolCallStart) {
   if (!state.currentToolCallJson) {
     return;
   }
-  const errorId = (0, import_provider_utils2.generateId)();
+  logParseFailure({
+    phase: "stream",
+    reason: "Incomplete streaming tool call segment emitted as text",
+    snippet: `${toolCallStart}${state.currentToolCallJson}`
+  });
+  const errorId = (0, import_provider_utils.generateId)();
   controller.enqueue({ type: "text-start", id: errorId });
   controller.enqueue({
     type: "text-delta",
@@ -1071,7 +997,7 @@ function publishText(text, state, controller) {
     state.currentToolCallJson += text;
   } else if (text.length > 0) {
     if (!state.currentTextId) {
-      state.currentTextId = (0, import_provider_utils2.generateId)();
+      state.currentTextId = (0, import_provider_utils.generateId)();
       controller.enqueue({ type: "text-start", id: state.currentTextId });
       state.hasEmittedTextStart = true;
     }
@@ -1086,16 +1012,22 @@ function emitToolCall(context) {
   var _a;
   const { state, controller, toolCallStart, toolCallEnd, options } = context;
   try {
-    const parsedToolCall = robust_json_exports.parse(state.currentToolCallJson);
+    const parsedToolCall = parse(state.currentToolCallJson);
     closeTextBlock(state, controller);
     controller.enqueue({
       type: "tool-call",
-      toolCallId: (0, import_provider_utils2.generateId)(),
+      toolCallId: (0, import_provider_utils.generateId)(),
       toolName: parsedToolCall.name,
       input: JSON.stringify((_a = parsedToolCall.arguments) != null ? _a : {})
     });
-  } catch (e) {
-    const errorId = (0, import_provider_utils2.generateId)();
+  } catch (error) {
+    logParseFailure({
+      phase: "stream",
+      reason: "Failed to parse streaming tool call JSON segment",
+      snippet: `${toolCallStart}${state.currentToolCallJson}${toolCallEnd}`,
+      error
+    });
+    const errorId = (0, import_provider_utils.generateId)();
     controller.enqueue({ type: "text-start", id: errorId });
     controller.enqueue({
       type: "text-delta",
@@ -1260,9 +1192,340 @@ var jsonMixProtocol = ({
 });
 
 // src/protocols/morph-xml-protocol.ts
-var import_provider_utils3 = require("@ai-sdk/provider-utils");
+var import_provider_utils2 = require("@ai-sdk/provider-utils");
 var import_rxml = require("@ai-sdk-tool/rxml");
+
+// src/utils/type-guards.ts
+function isToolCallContent(content) {
+  return content.type === "tool-call" && typeof content.toolName === "string" && // input may be a JSON string or an already-parsed object depending on provider/runtime
+  (typeof content.input === "string" || typeof content.input === "object");
+}
+function isToolResultPart(content) {
+  const c = content;
+  return !!c && c.type === "tool-result" && typeof c.toolName === "string" && typeof c.toolCallId === "string" && "output" in c;
+}
+function hasInputProperty(obj) {
+  return typeof obj === "object" && obj !== null && "input" in obj;
+}
+
+// src/protocols/morph-xml-protocol.ts
 var WHITESPACE_REGEX2 = /\s/;
+var MALFORMED_CLOSE_RE = /<\/\s+([A-Za-z0-9_:-]+)\s*>/;
+var MALFORMED_CLOSE_RE_G = /<\/\s+([A-Za-z0-9_:-]+)\s*>/g;
+var NAME_CHAR_RE = /[A-Za-z0-9_:-]/;
+var STATUS_TO_STEP_BOUNDARY_RE = /<\/status>\s*<step>/g;
+var STEP_TAG_RE = /<step>([\s\S]*?)<\/step>/i;
+var STATUS_TAG_RE = /<status>([\s\S]*?)<\/status>/i;
+function normalizeCloseTags(xml) {
+  return xml.replace(MALFORMED_CLOSE_RE_G, "</$1>");
+}
+function escapeInvalidLt(xml) {
+  const len = xml.length;
+  let out = "";
+  for (let i = 0; i < len; i += 1) {
+    const ch = xml[i];
+    if (ch === "<") {
+      const next = i + 1 < len ? xml[i + 1] : "";
+      if (!(NAME_CHAR_RE.test(next) || next === "/" || next === "!" || next === "?")) {
+        out += "&lt;";
+        continue;
+      }
+    }
+    out += ch;
+  }
+  return out;
+}
+function shouldDeduplicateStringTags(schema) {
+  const unwrapped = (0, import_rxml.unwrapJsonSchema)(schema);
+  if (!unwrapped || typeof unwrapped !== "object") {
+    return false;
+  }
+  const props = unwrapped.properties;
+  if (!props) {
+    return false;
+  }
+  const commandRaw = props.command;
+  if (!commandRaw) {
+    return false;
+  }
+  const command = (0, import_rxml.unwrapJsonSchema)(commandRaw);
+  return (command == null ? void 0 : command.type) === "array";
+}
+function tryParseSecondaryXml(content, toolSchema, options) {
+  const normalized = normalizeCloseTags(content);
+  const balanced = balanceTags(content);
+  const hasMalformedClose = MALFORMED_CLOSE_RE.test(content);
+  if (!hasMalformedClose && balanced.length > normalized.length) {
+    return null;
+  }
+  try {
+    let parsed = (0, import_rxml.parse)(balanced, toolSchema, {
+      onError: options == null ? void 0 : options.onError,
+      noChildNodes: []
+    });
+    parsed = repairParsedAgainstSchema(parsed, toolSchema, options);
+    return parsed;
+  } catch (_e) {
+    if (shouldDeduplicateStringTags(toolSchema)) {
+      const deduped = dedupeStringTagsAgainstSchema(balanced, toolSchema);
+      if (deduped !== balanced) {
+        try {
+          let reparsed = (0, import_rxml.parse)(deduped, toolSchema, {
+            onError: options == null ? void 0 : options.onError,
+            noChildNodes: []
+          });
+          reparsed = repairParsedAgainstSchema(reparsed, toolSchema, options);
+          return reparsed;
+        } catch (_) {
+          return null;
+        }
+      }
+    }
+    return null;
+  }
+}
+function balanceTags(xml) {
+  const src = normalizeCloseTags(xml).replace(
+    STATUS_TO_STEP_BOUNDARY_RE,
+    "</status></step><step>"
+  );
+  let i = 0;
+  const len = src.length;
+  const out = [];
+  const stack = [];
+  while (i < len) {
+    const lt = src.indexOf("<", i);
+    if (lt === -1) {
+      out.push(src.slice(i));
+      break;
+    }
+    out.push(src.slice(i, lt));
+    if (lt + 1 >= len) {
+      break;
+    }
+    const next = src[lt + 1];
+    if (next === "!" || next === "?") {
+      i = handleSpecialTagSegment(src, lt, out);
+      continue;
+    }
+    if (next === "/") {
+      i = handleClosingTagSegment(src, lt, out, stack);
+      continue;
+    }
+    i = handleOpeningTagSegment(src, lt, out, stack);
+  }
+  for (let k = stack.length - 1; k >= 0; k -= 1) {
+    out.push(`</${stack[k]}>`);
+  }
+  return out.join("");
+}
+function skipWs(s, p, len) {
+  let idx = p;
+  while (idx < len && WHITESPACE_REGEX2.test(s[idx])) {
+    idx += 1;
+  }
+  return idx;
+}
+function parseTagNameAt(s, p, len) {
+  let idx = p;
+  const start = idx;
+  while (idx < len && NAME_CHAR_RE.test(s[idx])) {
+    idx += 1;
+  }
+  return { name: s.slice(start, idx), pos: idx };
+}
+function handleSpecialTagSegment(src, lt, out) {
+  const gt = src.indexOf(">", lt + 1);
+  if (gt === -1) {
+    out.push(src.slice(lt));
+    return src.length;
+  }
+  out.push(src.slice(lt, gt + 1));
+  return gt + 1;
+}
+function handleClosingTagSegment(src, lt, out, stack) {
+  const len = src.length;
+  let p = skipWs(src, lt + 2, len);
+  const { name, pos } = parseTagNameAt(src, p, len);
+  p = pos;
+  const gt = src.indexOf(">", p);
+  const closingText = gt === -1 ? src.slice(lt) : src.slice(lt, gt + 1);
+  const idx = stack.lastIndexOf(name);
+  if (idx !== -1) {
+    for (let k = stack.length - 1; k > idx; k -= 1) {
+      out.push(`</${stack[k]}>`);
+      stack.pop();
+    }
+    out.push(closingText);
+    stack.pop();
+  }
+  return gt === -1 ? len : gt + 1;
+}
+function handleOpeningTagSegment(src, lt, out, stack) {
+  const len = src.length;
+  let p = skipWs(src, lt + 1, len);
+  const nameStart = p;
+  const parsed = parseTagNameAt(src, p, len);
+  p = parsed.pos;
+  const name = src.slice(nameStart, p);
+  const q = src.indexOf(">", p);
+  if (q === -1) {
+    out.push(src.slice(lt));
+    return len;
+  }
+  let r = q - 1;
+  while (r >= nameStart && WHITESPACE_REGEX2.test(src[r])) {
+    r -= 1;
+  }
+  const selfClosing = src[r] === "/";
+  out.push(src.slice(lt, q + 1));
+  if (!selfClosing && name) {
+    stack.push(name);
+  }
+  return q + 1;
+}
+function repairParsedAgainstSchema(input, schema, options) {
+  if (!input || typeof input !== "object") {
+    return input;
+  }
+  const unwrapped = (0, import_rxml.unwrapJsonSchema)(schema);
+  if (!unwrapped || typeof unwrapped !== "object") {
+    return input;
+  }
+  const properties = unwrapped.properties;
+  if (!properties) {
+    return input;
+  }
+  applySchemaProps(input, properties, options);
+  return input;
+}
+function applySchemaProps(obj, properties, options) {
+  for (const key of Object.keys(obj)) {
+    const propSchema = properties[key];
+    if (!propSchema) {
+      continue;
+    }
+    const prop = (0, import_rxml.unwrapJsonSchema)(propSchema);
+    const propType = prop.type;
+    if (propType === "array" && prop.items) {
+      const itemSchemaRaw = prop.items;
+      const itemSchema = (0, import_rxml.unwrapJsonSchema)(itemSchemaRaw);
+      obj[key] = coerceArrayItems(obj[key], itemSchema, options);
+      continue;
+    }
+    if (propType === "object") {
+      const val = obj[key];
+      if (val && typeof val === "object") {
+        obj[key] = repairParsedAgainstSchema(
+          val,
+          prop,
+          options
+        );
+      }
+    }
+  }
+}
+function coerceArrayItems(val, itemSchema, options) {
+  if (!Array.isArray(val)) {
+    return val;
+  }
+  return val.map((v) => coerceArrayItem(v, itemSchema, options));
+}
+function coerceArrayItem(v, itemSchema, options) {
+  const itemType = itemSchema == null ? void 0 : itemSchema.type;
+  if (typeof v === "string" && itemType === "object") {
+    const parsed = tryParseStringToSchemaObject(v, itemSchema, options);
+    if (parsed !== null) {
+      return parsed;
+    }
+    const fallback = extractStepStatusFromString(normalizeCloseTags(v));
+    if (fallback) {
+      return fallback;
+    }
+    return v;
+  }
+  if (v && typeof v === "object" && itemType === "object") {
+    return repairParsedAgainstSchema(
+      v,
+      itemSchema,
+      options
+    );
+  }
+  return v;
+}
+function getStringPropertyNames(schema) {
+  const unwrapped = (0, import_rxml.unwrapJsonSchema)(schema);
+  if (!unwrapped || typeof unwrapped !== "object") {
+    return [];
+  }
+  const props = unwrapped.properties;
+  if (!props) {
+    return [];
+  }
+  const names = [];
+  for (const key of Object.keys(props)) {
+    const prop = (0, import_rxml.unwrapJsonSchema)(
+      props[key]
+    );
+    const type = prop.type;
+    if (type === "string") {
+      names.push(key);
+    }
+  }
+  return names;
+}
+function escapeRegExp2(s) {
+  return s.replace(/[.*+?^${}()|[\\]\\]/g, "\\$&");
+}
+function dedupeStringTagsAgainstSchema(xml, schema) {
+  const names = getStringPropertyNames(schema);
+  let out = xml;
+  for (const key of names) {
+    out = dedupeSingleTag(out, key);
+  }
+  return out;
+}
+function dedupeSingleTag(xml, key) {
+  var _a, _b;
+  const escaped = escapeRegExp2(key);
+  const re = new RegExp(`<${escaped}>([\\s\\S]*?)<\\/${escaped}>`, "g");
+  const matches = Array.from(xml.matchAll(re));
+  if (matches.length <= 1) {
+    return xml;
+  }
+  const last = matches.at(-1);
+  let result = "";
+  let cursor = 0;
+  for (const m of matches) {
+    const idx = (_a = m.index) != null ? _a : 0;
+    result += xml.slice(cursor, idx);
+    if (last && idx === ((_b = last.index) != null ? _b : -1)) {
+      result += m[0];
+    }
+    cursor = idx + m[0].length;
+  }
+  result += xml.slice(cursor);
+  return result;
+}
+function tryParseStringToSchemaObject(xml, itemSchema, options) {
+  try {
+    const fixed = (0, import_rxml.parse)(normalizeCloseTags(xml), itemSchema, {
+      onError: options == null ? void 0 : options.onError,
+      noChildNodes: []
+    });
+    return typeof fixed === "string" ? null : fixed;
+  } catch (e) {
+    return null;
+  }
+}
+function extractStepStatusFromString(normXml) {
+  const stepMatch = normXml.match(STEP_TAG_RE);
+  const statusMatch = normXml.match(STATUS_TAG_RE);
+  if (stepMatch && statusMatch) {
+    return { step: stepMatch[1], status: statusMatch[1] };
+  }
+  return null;
+}
 function processTextBeforeToolCall(text, currentIndex, toolCallStartIndex, processedElements) {
   if (toolCallStartIndex > currentIndex) {
     const textSegment = text.substring(currentIndex, toolCallStartIndex);
@@ -1275,20 +1538,36 @@ function processTextBeforeToolCall(text, currentIndex, toolCallStartIndex, proce
 function processToolCall(params) {
   var _a;
   const { toolCall, tools, options, text, processedElements } = params;
+  const toolSchema = getToolSchema(tools, toolCall.toolName);
   try {
-    const toolSchema = getToolSchema(tools, toolCall.toolName);
-    const parsed = (0, import_rxml.parse)(toolCall.content, toolSchema, {
+    const primary = escapeInvalidLt(normalizeCloseTags(toolCall.content));
+    let parsed = (0, import_rxml.parse)(primary, toolSchema, {
       onError: options == null ? void 0 : options.onError,
       // Disable HTML self-closing tag behavior to allow base, meta, link etc. as regular tags
       noChildNodes: []
     });
+    parsed = repairParsedAgainstSchema(parsed, toolSchema, options);
     processedElements.push({
       type: "tool-call",
-      toolCallId: (0, import_provider_utils3.generateId)(),
+      toolCallId: (0, import_provider_utils2.generateId)(),
       toolName: toolCall.toolName,
       input: JSON.stringify(parsed)
     });
   } catch (error) {
+    const reparsed = tryParseSecondaryXml(
+      toolCall.content,
+      toolSchema,
+      options
+    );
+    if (reparsed !== null) {
+      processedElements.push({
+        type: "tool-call",
+        toolCallId: (0, import_provider_utils2.generateId)(),
+        toolName: toolCall.toolName,
+        input: JSON.stringify(reparsed)
+      });
+      return;
+    }
     const originalCallText = text.substring(
       toolCall.startIndex,
       toolCall.endIndex
@@ -1312,20 +1591,33 @@ function addRemainingText(text, currentIndex, processedElements) {
 }
 function handleStreamingToolCallEnd(params) {
   const { toolContent, currentToolCall, tools, options, ctrl, flushText } = params;
+  const toolSchema = getToolSchema(tools, currentToolCall.name);
   try {
-    const toolSchema = getToolSchema(tools, currentToolCall.name);
-    const parsed = (0, import_rxml.parse)(toolContent, toolSchema, {
+    const primary = escapeInvalidLt(normalizeCloseTags(toolContent));
+    let parsed = (0, import_rxml.parse)(primary, toolSchema, {
       onError: options == null ? void 0 : options.onError,
       noChildNodes: []
     });
+    parsed = repairParsedAgainstSchema(parsed, toolSchema, options);
     flushText(ctrl);
     ctrl.enqueue({
       type: "tool-call",
-      toolCallId: (0, import_provider_utils3.generateId)(),
+      toolCallId: (0, import_provider_utils2.generateId)(),
       toolName: currentToolCall.name,
       input: JSON.stringify(parsed)
     });
   } catch (error) {
+    const parsed = tryParseSecondaryXml(toolContent, toolSchema, options);
+    if (parsed !== null) {
+      flushText(ctrl);
+      ctrl.enqueue({
+        type: "tool-call",
+        toolCallId: (0, import_provider_utils2.generateId)(),
+        toolName: currentToolCall.name,
+        input: JSON.stringify(parsed)
+      });
+      return;
+    }
     handleStreamingToolCallError({
       error,
       currentToolCall,
@@ -1357,19 +1649,28 @@ function handleStreamingToolCallError(params) {
   flushText(ctrl, originalCallText);
 }
 function findEarliestToolTag(buffer, toolNames) {
-  let earliestStartTagIndex = -1;
-  let earliestToolName = "";
+  let bestIndex = -1;
+  let bestName = "";
+  let bestSelfClosing = false;
   if (toolNames.length > 0) {
     for (const name of toolNames) {
-      const startTag = `<${name}>`;
-      const index = buffer.indexOf(startTag);
-      if (index !== -1 && (earliestStartTagIndex === -1 || index < earliestStartTagIndex)) {
-        earliestStartTagIndex = index;
-        earliestToolName = name;
+      const openTag = `<${name}>`;
+      const selfTag = `<${name}/>`;
+      const idxOpen = buffer.indexOf(openTag);
+      const idxSelf = buffer.indexOf(selfTag);
+      if (idxOpen !== -1 && (bestIndex === -1 || idxOpen < bestIndex)) {
+        bestIndex = idxOpen;
+        bestName = name;
+        bestSelfClosing = false;
+      }
+      if (idxSelf !== -1 && (bestIndex === -1 || idxSelf < bestIndex)) {
+        bestIndex = idxSelf;
+        bestName = name;
+        bestSelfClosing = true;
       }
     }
   }
-  return { index: earliestStartTagIndex, name: earliestToolName };
+  return { index: bestIndex, name: bestName, selfClosing: bestSelfClosing };
 }
 function handleNoToolTagInBuffer(buffer, maxStartTagLen, controller, flushText) {
   const tail = Math.max(0, maxStartTagLen - 1);
@@ -1392,10 +1693,12 @@ function processToolCallInBuffer(params) {
     setBuffer
   } = params;
   const endTag = `</${currentToolCall.name}>`;
-  const endTagIndex = buffer.indexOf(endTag);
+  const normalized = normalizeCloseTags(buffer);
+  const effectiveBuffer = normalized;
+  const endTagIndex = effectiveBuffer.indexOf(endTag);
   if (endTagIndex !== -1) {
-    const toolContent = buffer.substring(0, endTagIndex);
-    const newBuffer = buffer.substring(endTagIndex + endTag.length);
+    const toolContent = effectiveBuffer.substring(0, endTagIndex);
+    const newBuffer = effectiveBuffer.substring(endTagIndex + endTag.length);
     setBuffer("");
     handleStreamingToolCallEnd({
       toolContent,
@@ -1408,14 +1711,46 @@ function processToolCallInBuffer(params) {
     setBuffer(newBuffer);
     return { buffer: newBuffer, currentToolCall: null, shouldBreak: false };
   }
-  return { buffer, currentToolCall, shouldBreak: true };
+  return { buffer: effectiveBuffer, currentToolCall, shouldBreak: true };
 }
 function processNoToolCallInBuffer(params) {
-  const { buffer, toolNames, maxStartTagLen, controller, flushText } = params;
-  const { index: earliestStartTagIndex, name: earliestToolName } = findEarliestToolTag(buffer, toolNames);
+  const {
+    buffer,
+    toolNames,
+    maxStartTagLen,
+    controller,
+    flushText,
+    tools,
+    options
+  } = params;
+  const {
+    index: earliestStartTagIndex,
+    name: earliestToolName,
+    selfClosing
+  } = findEarliestToolTag(buffer, toolNames);
   if (earliestStartTagIndex !== -1) {
     const textBeforeTag = buffer.substring(0, earliestStartTagIndex);
     flushText(controller, textBeforeTag);
+    if (selfClosing) {
+      const selfTag = `<${earliestToolName}/>`;
+      const newBuffer2 = buffer.substring(
+        earliestStartTagIndex + selfTag.length
+      );
+      handleStreamingToolCallEnd({
+        toolContent: "",
+        currentToolCall: { name: earliestToolName, content: "" },
+        tools,
+        options,
+        ctrl: controller,
+        flushText
+      });
+      return {
+        buffer: newBuffer2,
+        currentToolCall: null,
+        shouldBreak: false,
+        shouldContinue: false
+      };
+    }
     const startTag = `<${earliestToolName}>`;
     const newBuffer = buffer.substring(earliestStartTagIndex + startTag.length);
     return {
@@ -1444,7 +1779,7 @@ function createFlushTextHandler(getBuffer, setBuffer, getCurrentTextId, setCurre
     if (content) {
       const currentTextId2 = getCurrentTextId();
       if (!currentTextId2) {
-        const newId = (0, import_provider_utils3.generateId)();
+        const newId = (0, import_provider_utils2.generateId)();
         setCurrentTextId(newId);
         controller.enqueue({ type: "text-start", id: newId });
       }
@@ -1496,6 +1831,8 @@ function processBufferWithoutToolCall(params, controller) {
     getBuffer,
     setBuffer,
     setCurrentToolCall,
+    tools,
+    options,
     toolNames,
     maxStartTagLen,
     flushText
@@ -1505,7 +1842,9 @@ function processBufferWithoutToolCall(params, controller) {
     toolNames,
     maxStartTagLen,
     controller,
-    flushText
+    flushText,
+    tools,
+    options
   });
   setBuffer(result.buffer);
   setCurrentToolCall(result.currentToolCall);
@@ -1580,7 +1919,17 @@ var morphXmlProtocol = () => ({
     }
     const processedElements = [];
     let currentIndex = 0;
-    const toolCalls = findToolCalls(text, toolNames);
+    const toolCallsRaw = findToolCalls(text, toolNames);
+    const toolCallsNorm = collectToolCallsFromNormalizedText(text, toolNames);
+    const seen = /* @__PURE__ */ new Set();
+    const toolCalls = [...toolCallsRaw, ...toolCallsNorm].filter((tc) => {
+      const key = `${tc.toolName}:${tc.startIndex}:${tc.endIndex}`;
+      if (seen.has(key)) {
+        return false;
+      }
+      seen.add(key);
+      return true;
+    }).sort((a, b) => a.startIndex - b.startIndex);
     for (const toolCall of toolCalls) {
       currentIndex = processTextBeforeToolCall(
         text,
@@ -1668,53 +2017,225 @@ function getToolSchema(tools, toolName) {
   var _a;
   return (_a = tools.find((t) => t.name === toolName)) == null ? void 0 : _a.inputSchema;
 }
-function computeFullTagEnd(text, contentEnd, toolName) {
-  let fullTagEnd = contentEnd + `</${toolName}>`.length;
-  const closeHead = text.indexOf(`</${toolName}`, contentEnd);
-  if (closeHead === contentEnd) {
-    let p = closeHead + 2 + toolName.length;
+function findClosingTagEndFlexible(text, contentStart, toolName) {
+  let pos = contentStart;
+  let depth = 1;
+  while (pos < text.length) {
+    const tok = nextTagToken(text, pos);
+    if (tok.kind === "eof") {
+      break;
+    }
+    const result = updateDepthWithToken(tok, toolName, depth);
+    depth = result.depth;
+    if (result.closedAt !== void 0) {
+      return result.closedAt;
+    }
+    pos = tok.nextPos;
+  }
+  return -1;
+}
+function skipSpecialSegment(text, lt) {
+  const next = text[lt + 1];
+  if (next !== "!" && next !== "?") {
+    return null;
+  }
+  const gt = text.indexOf(">", lt + 1);
+  if (gt === -1) {
+    return null;
+  }
+  return gt + 1;
+}
+function consumeClosingTag(text, lt, toolName) {
+  let p = lt + 2;
+  while (p < text.length && WHITESPACE_REGEX2.test(text[p])) {
+    p += 1;
+  }
+  if (text.slice(p, p + toolName.length) === toolName) {
+    p += toolName.length;
     while (p < text.length && WHITESPACE_REGEX2.test(text[p])) {
       p += 1;
     }
     if (text[p] === ">") {
-      fullTagEnd = p + 1;
+      const endPos2 = p + 1;
+      return { matched: true, endPos: endPos2 };
     }
   }
-  return fullTagEnd;
+  const gt = text.indexOf(">", lt + 1);
+  const endPos = gt === -1 ? text.length : gt + 1;
+  return { matched: false, endPos };
 }
-function extractToolCallInfo(text, tagStart, toolName, range) {
-  var _a;
-  const startTag = `<${toolName}>`;
-  const contentStart = tagStart + startTag.length;
-  const contentEnd = contentStart + (range.end - range.start);
-  const fullTagEnd = computeFullTagEnd(text, contentEnd, toolName);
-  const segment = text.substring(tagStart, fullTagEnd);
-  const content = (_a = (0, import_rxml.extractRawInner)(segment, toolName)) != null ? _a : text.substring(contentStart, contentEnd);
+function consumeOpenTag(text, lt) {
+  let p = lt + 1;
+  while (p < text.length && WHITESPACE_REGEX2.test(text[p])) {
+    p += 1;
+  }
+  const nameStart = p;
+  while (p < text.length && NAME_CHAR_RE.test(text.charAt(p))) {
+    p += 1;
+  }
+  const name = text.slice(nameStart, p);
+  const q = text.indexOf(">", p);
+  if (q === -1) {
+    return null;
+  }
+  let r = q - 1;
+  while (r >= nameStart && WHITESPACE_REGEX2.test(text[r])) {
+    r -= 1;
+  }
+  const selfClosing = text[r] === "/";
+  return { name, selfClosing, nextPos: q + 1 };
+}
+function updateDepthWithToken(tok, toolName, depth) {
+  if (tok.kind === "close" && tok.name === toolName) {
+    const newDepth = depth - 1;
+    return newDepth === 0 ? { depth: newDepth, closedAt: tok.nextPos } : { depth: newDepth };
+  }
+  if (tok.kind === "open" && tok.name === toolName && !tok.selfClosing) {
+    return { depth: depth + 1 };
+  }
+  return { depth };
+}
+function nextTagToken(text, fromPos) {
+  const lt = text.indexOf("<", fromPos);
+  if (lt === -1 || lt + 1 >= text.length) {
+    return { kind: "eof", nextPos: text.length };
+  }
+  const next = text[lt + 1];
+  const specialEnd = skipSpecialSegment(text, lt);
+  if (specialEnd !== null) {
+    return { kind: "special", nextPos: specialEnd };
+  }
+  if (next === "/") {
+    const closing = consumeClosingTag(text, lt, "");
+    let p = lt + 2;
+    while (p < text.length && WHITESPACE_REGEX2.test(text[p])) {
+      p += 1;
+    }
+    const nameStart = p;
+    while (p < text.length && NAME_CHAR_RE.test(text.charAt(p))) {
+      p += 1;
+    }
+    const name = text.slice(nameStart, p);
+    return { kind: "close", name, nextPos: closing.endPos };
+  }
+  const open = consumeOpenTag(text, lt);
+  if (open === null) {
+    return { kind: "eof", nextPos: text.length };
+  }
   return {
-    toolName,
-    startIndex: tagStart,
-    endIndex: fullTagEnd,
-    content,
-    segment
+    kind: "open",
+    name: open.name,
+    selfClosing: open.selfClosing,
+    nextPos: open.nextPos
   };
 }
+function collectToolCallsFromNormalizedText(text, toolNames) {
+  var _a;
+  const normalizedText = normalizeCloseTags(text);
+  const collected = [];
+  for (const toolName of toolNames) {
+    const startTag = `<${toolName}>`;
+    let idx = 0;
+    let lastOrigIdx = 0;
+    while (idx < normalizedText.length) {
+      const tagStartNorm = normalizedText.indexOf(startTag, idx);
+      if (tagStartNorm === -1) {
+        break;
+      }
+      const contentStartNorm = tagStartNorm + startTag.length;
+      const endNorm = findClosingTagEndFlexible(
+        normalizedText,
+        contentStartNorm,
+        toolName
+      );
+      if (endNorm > contentStartNorm) {
+        const tagStartOrig = text.indexOf(startTag, lastOrigIdx);
+        const contentStartOrig = tagStartOrig + startTag.length;
+        let endOrig = findClosingTagEndFlexible(
+          text,
+          contentStartOrig,
+          toolName
+        );
+        if (endOrig === -1) {
+          const approxLen = endNorm - tagStartNorm;
+          endOrig = Math.min(text.length, tagStartOrig + approxLen);
+        }
+        const segment = text.substring(tagStartOrig, endOrig);
+        const inner = (_a = (0, import_rxml.extractRawInner)(segment, toolName)) != null ? _a : segment.substring(startTag.length, segment.lastIndexOf("<"));
+        collected.push({
+          toolName,
+          startIndex: tagStartOrig,
+          endIndex: endOrig,
+          content: inner,
+          segment
+        });
+        lastOrigIdx = endOrig;
+        idx = endNorm;
+      } else {
+        idx = contentStartNorm;
+      }
+    }
+  }
+  return collected.sort((a, b) => a.startIndex - b.startIndex);
+}
+function getNextTagInfo(text, toolName, fromIndex) {
+  const startTag = `<${toolName}>`;
+  const selfTag = `<${toolName}/>`;
+  const openIdx = text.indexOf(startTag, fromIndex);
+  const selfIdx = text.indexOf(selfTag, fromIndex);
+  const hasOpen = openIdx !== -1;
+  const hasSelf = selfIdx !== -1;
+  if (!(hasOpen || hasSelf)) {
+    return {
+      found: false,
+      tagStart: -1,
+      selfClosing: false,
+      startTag,
+      selfTag
+    };
+  }
+  const pickSelf = hasSelf && (!hasOpen || selfIdx < openIdx);
+  const tagStart = pickSelf ? selfIdx : openIdx;
+  return { found: true, tagStart, selfClosing: pickSelf, startTag, selfTag };
+}
 function findToolCallsForName(text, toolName) {
+  var _a;
   const toolCalls = [];
-  const startTag = `<${toolName}>`;
   let searchIndex = 0;
   while (searchIndex < text.length) {
-    const tagStart = text.indexOf(startTag, searchIndex);
-    if (tagStart === -1) {
+    const info = getNextTagInfo(text, toolName, searchIndex);
+    if (!info.found) {
       break;
     }
-    const remainingText = text.substring(tagStart);
-    const range = (0, import_rxml.findFirstTopLevelRange)(remainingText, toolName);
-    if (range) {
-      const toolCallInfo = extractToolCallInfo(text, tagStart, toolName, range);
-      toolCalls.push(toolCallInfo);
-      searchIndex = toolCallInfo.endIndex;
+    const { tagStart, selfClosing, startTag, selfTag } = info;
+    if (selfClosing) {
+      const endIndex = tagStart + selfTag.length;
+      const segment = text.substring(tagStart, endIndex);
+      toolCalls.push({
+        toolName,
+        startIndex: tagStart,
+        endIndex,
+        content: "",
+        segment
+      });
+      searchIndex = endIndex;
+      continue;
+    }
+    const contentStart = tagStart + startTag.length;
+    const fullTagEnd = findClosingTagEndFlexible(text, contentStart, toolName);
+    if (fullTagEnd !== -1 && fullTagEnd > contentStart) {
+      const segment = text.substring(tagStart, fullTagEnd);
+      const inner = (_a = (0, import_rxml.extractRawInner)(segment, toolName)) != null ? _a : segment.substring(startTag.length, segment.lastIndexOf("<"));
+      toolCalls.push({
+        toolName,
+        startIndex: tagStart,
+        endIndex: fullTagEnd,
+        content: inner,
+        segment
+      });
+      searchIndex = fullTagEnd;
     } else {
-      searchIndex = tagStart + startTag.length;
+      searchIndex = contentStart;
     }
   }
   return toolCalls;
@@ -1728,14 +2249,53 @@ function findToolCalls(text, toolNames) {
   return toolCalls.sort((a, b) => a.startIndex - b.startIndex);
 }
 
-// src/protocols/tool-call-protocol.ts
-function isProtocolFactory(protocol) {
-  return typeof protocol === "function";
+// src/generate-handler.ts
+var import_provider_utils3 = require("@ai-sdk/provider-utils");
+var import_rxml2 = require("@ai-sdk-tool/rxml");
+
+// src/utils/on-error.ts
+function extractOnErrorOption(providerOptions) {
+  var _a;
+  if (providerOptions && typeof providerOptions === "object") {
+    const onError = (_a = providerOptions.toolCallMiddleware) == null ? void 0 : _a.onError;
+    return onError ? { onError } : void 0;
+  }
+  return;
+}
+
+// src/utils/provider-options.ts
+var originalToolsSchema = {
+  encode: encodeOriginalTools,
+  decode: decodeOriginalTools
+};
+function encodeOriginalTools(tools) {
+  return (tools == null ? void 0 : tools.map((t) => ({
+    name: t.name,
+    inputSchema: JSON.stringify(t.inputSchema)
+  }))) || [];
+}
+function decodeOriginalTools(originalTools) {
+  if (!originalTools) {
+    return [];
+  }
+  return originalTools.map(
+    (t) => ({
+      type: "function",
+      name: t.name,
+      inputSchema: JSON.parse(t.inputSchema)
+    })
+  );
+}
+function extractToolNamesFromOriginalTools(originalTools) {
+  return (originalTools == null ? void 0 : originalTools.map((t) => t.name)) || [];
+}
+function isToolChoiceActive(params) {
+  var _a, _b, _c;
+  const toolChoice = (_b = (_a = params.providerOptions) == null ? void 0 : _a.toolCallMiddleware) == null ? void 0 : _b.toolChoice;
+  return !!(typeof params.providerOptions === "object" && params.providerOptions !== null && typeof ((_c = params.providerOptions) == null ? void 0 : _c.toolCallMiddleware) === "object" && toolChoice && typeof toolChoice === "object" && (toolChoice.type === "tool" || toolChoice.type === "required"));
 }
 
 // src/generate-handler.ts
-var import_provider_utils4 = require("@ai-sdk/provider-utils");
-var import_rxml2 = require("@ai-sdk-tool/rxml");
 function parseToolChoiceJson(text, providerOptions) {
   var _a;
   try {
@@ -1779,7 +2339,7 @@ async function handleToolChoice(doGenerate, params) {
   }
   const toolCall = {
     type: "tool-call",
-    toolCallId: (0, import_provider_utils4.generateId)(),
+    toolCallId: (0, import_provider_utils3.generateId)(),
     toolName: parsed.name || "unknown",
     input: JSON.stringify(parsed.arguments || {})
   };
@@ -1905,8 +2465,13 @@ function fixToolCallWithSchema(part, tools) {
   };
 }
 
+// src/protocols/tool-call-protocol.ts
+function isProtocolFactory(protocol) {
+  return typeof protocol === "function";
+}
+
 // src/stream-handler.ts
-var import_provider_utils5 = require("@ai-sdk/provider-utils");
+var import_provider_utils4 = require("@ai-sdk/provider-utils");
 function extractToolCallSegments(protocol, fullRawText, tools) {
   const segments = protocol.extractToolCallSegments ? protocol.extractToolCallSegments({
     text: fullRawText,
@@ -1943,7 +2508,7 @@ function handleDebugSummary(parsedToolCalls, origin, params) {
 }
 function createDebugSummaryTransform({
   protocol,
-  fullRawText,
+  getFullRawText,
   tools,
   params
 }) {
@@ -1956,11 +2521,9 @@ function createDebugSummaryTransform({
         }
         if (part.type === "finish") {
           try {
-            const origin = extractToolCallSegments(
-              protocol,
-              fullRawText,
-              tools
-            );
+            const raw = getFullRawText();
+            logRawChunk(raw);
+            const origin = extractToolCallSegments(protocol, raw, tools);
             handleDebugSummary(parsedToolCalls, origin, params);
           } catch (e) {
           }
@@ -2058,7 +2621,7 @@ async function wrapStream({
   const withSummary = parsed.pipeThrough(
     createDebugSummaryTransform({
       protocol,
-      fullRawText,
+      getFullRawText: () => fullRawText,
       tools,
       params
     })
@@ -2092,19 +2655,26 @@ async function toolChoiceStream({
   }
   const toolCallChunk = {
     type: "tool-call",
-    toolCallId: (0, import_provider_utils5.generateId)(),
+    toolCallId: (0, import_provider_utils4.generateId)(),
     toolName: toolJson.name || "unknown",
     input: JSON.stringify(toolJson.arguments || {})
   };
   const finishChunk = {
     type: "finish",
-    usage: (result == null ? void 0 : result.usage) || // TODO: If possible, try to return a certain amount of LLM usage.
-    {
-      inputTokens: 0,
-      outputTokens: 0,
-      totalTokens: 0
+    usage: (result == null ? void 0 : result.usage) || {
+      inputTokens: {
+        total: 0,
+        noCache: void 0,
+        cacheRead: void 0,
+        cacheWrite: void 0
+      },
+      outputTokens: {
+        total: 0,
+        text: void 0,
+        reasoning: void 0
+      }
     },
-    finishReason: "tool-calls"
+    finishReason: { unified: "tool-calls", raw: void 0 }
   };
   const stream = new ReadableStream({
     start(controller) {
@@ -2138,26 +2708,106 @@ async function toolChoiceStream({
   };
 }
 
+// src/utils/dynamic-tool-schema.ts
+function createDynamicIfThenElseSchema(tools) {
+  let currentSchema = {};
+  const toolNames = [];
+  for (let i = tools.length - 1; i >= 0; i -= 1) {
+    const tool = tools[i];
+    if (tool.type === "provider") {
+      throw new Error(
+        "Provider tools are not supported by this middleware. Please use function tools."
+      );
+    }
+    toolNames.unshift(tool.name);
+    const toolCondition = {
+      if: {
+        properties: {
+          name: {
+            const: tool.name
+          }
+        },
+        required: ["name"]
+      },
+      // biome-ignore lint/suspicious/noThenProperty: JSON Schema uses 'then' as a keyword
+      then: {
+        properties: {
+          name: {
+            const: tool.name
+          },
+          arguments: tool.inputSchema
+        },
+        required: ["name", "arguments"]
+      }
+    };
+    if (Object.keys(currentSchema).length > 0) {
+      toolCondition.else = currentSchema;
+    }
+    currentSchema = toolCondition;
+  }
+  return {
+    type: "object",
+    // Explicitly specify type as "object"
+    properties: {
+      name: {
+        type: "string",
+        description: "Name of the tool to call",
+        enum: toolNames
+      },
+      arguments: {
+        type: "object",
+        // By default, arguments is also specified as object type
+        description: "Argument object to be passed to the tool"
+      }
+    },
+    required: ["name", "arguments"],
+    ...currentSchema
+  };
+}
+
 // src/transform-handler.ts
-function buildFinalPrompt(systemPrompt, processedPrompt) {
-  var _a;
-  if (((_a = processedPrompt[0]) == null ? void 0 : _a.role) === "system") {
+function buildFinalPrompt(systemPrompt, processedPrompt, placement) {
+  const systemIndex = processedPrompt.findIndex((m) => m.role === "system");
+  if (systemIndex !== -1) {
+    const existing = processedPrompt[systemIndex].content;
+    let existingText = "";
+    if (typeof existing === "string") {
+      existingText = existing;
+    } else if (Array.isArray(existing)) {
+      existingText = existing.map((p) => {
+        var _a;
+        return (p == null ? void 0 : p.type) === "text" ? (_a = p.text) != null ? _a : "" : "";
+      }).filter(Boolean).join("\n");
+    } else {
+      existingText = String(existing != null ? existing : "");
+    }
+    const mergedContent = placement === "first" ? `${systemPrompt}
+
+${existingText}` : `${existingText}
+
+${systemPrompt}`;
+    return processedPrompt.map(
+      (m, idx) => idx === systemIndex ? {
+        ...m,
+        content: mergedContent
+      } : m
+    );
+  }
+  if (placement === "first") {
     return [
       {
         role: "system",
-        content: `${systemPrompt}
-
-${processedPrompt[0].content}`
+        content: systemPrompt
       },
-      ...processedPrompt.slice(1)
+      ...processedPrompt
     ];
   }
   return [
+    ...processedPrompt,
     {
       role: "system",
       content: systemPrompt
-    },
-    ...processedPrompt
+    }
   ];
 }
 function buildBaseReturnParams(params, finalPrompt, functionTools) {
@@ -2257,7 +2907,8 @@ function handleToolChoiceRequired(params, baseReturnParams, functionTools) {
 function transformParams({
   params,
   protocol,
-  toolSystemPromptTemplate
+  toolSystemPromptTemplate,
+  placement = "first"
 }) {
   var _a, _b, _c, _d, _e;
   const resolvedProtocol = isProtocolFactory(protocol) ? protocol() : protocol;
@@ -2273,7 +2924,11 @@ function transformParams({
     resolvedProtocol,
     extractOnErrorOption(params.providerOptions)
   );
-  const finalPrompt = buildFinalPrompt(systemPrompt, processedPrompt);
+  const finalPrompt = buildFinalPrompt(
+    systemPrompt,
+    processedPrompt,
+    placement
+  );
   const baseReturnParams = buildBaseReturnParams(
     params,
     finalPrompt,
@@ -2350,7 +3005,10 @@ function processMessage(message, resolvedProtocol, providerOptions) {
     };
   }
   if (message.role === "tool") {
-    return processToolMessage(message.content, resolvedProtocol);
+    const toolResultParts = message.content.filter(
+      (part) => part.type === "tool-result"
+    );
+    return processToolMessage(toolResultParts, resolvedProtocol);
   }
   return message;
 }
@@ -2425,7 +3083,8 @@ function convertToolPrompt(prompt, resolvedProtocol, providerOptions) {
 // src/tool-call-middleware.ts
 function createToolMiddleware({
   protocol,
-  toolSystemPromptTemplate
+  toolSystemPromptTemplate,
+  placement = "last"
 }) {
   const resolvedProtocol = isProtocolFactory(protocol) ? protocol() : protocol;
   return {
@@ -2452,6 +3111,7 @@ function createToolMiddleware({
     transformParams: async ({ params }) => transformParams({
       protocol: resolvedProtocol,
       toolSystemPromptTemplate,
+      placement,
       params
     })
   };
@@ -2497,6 +3157,7 @@ For each function call return a json object with function name and arguments wit
 });
 var morphXmlToolMiddleware = createToolMiddleware({
   protocol: morphXmlProtocol,
+  placement: "last",
   toolSystemPromptTemplate(tools) {
     return `You are a function calling AI model.
 
@@ -2520,7 +3181,6 @@ Available functions are listed inside <tools></tools>.
 });
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  RJSON,
   createDynamicIfThenElseSchema,
   createToolMiddleware,
   decodeOriginalTools,
@@ -2537,6 +3197,7 @@ Available functions are listed inside <tools></tools>.
   isToolChoiceActive,
   isToolResultPart,
   jsonMixProtocol,
+  logParseFailure,
   logParsedChunk,
   logParsedSummary,
   logRawChunk,