@animalabs/agent-framework 0.27.0 → 0.28.0

package/dist/src/cut-off-round.js

@@ -0,0 +1,87 @@
+import { randomUUID } from 'node:crypto';
+/**
+ * The last round of a response that ended at the output limit, when it holds
+ * calls that never ran, or, in XML tool mode, a tool-call block cut off before
+ * any call in it was named (`unnamed`). Membrane yields a round's calls only
+ * at its tool_use stop, or in XML tool mode at its `</function_calls>` stop,
+ * and the round begins after the last result that answers one of the
+ * response's calls.
+ */
+export function cutOffRound(response) {
+    if (response.stopReason !== 'max_tokens')
+        return undefined;
+    const callIds = new Set();
+    let start = 0;
+    response.content.forEach((block, index) => {
+        if (block.type === 'tool_use')
+            callIds.add(block.id);
+        else if (block.type === 'tool_result' && callIds.has(block.toolUseId))
+            start = index + 1;
+    });
+    const cut = response.details?.stop?.unclosedToolBlock ? cutToolBlock(response.content, start) : undefined;
+    const content = cut ?? response.content;
+    const uses = content.slice(start).filter((block) => block.type === 'tool_use');
+    const calls = uses.map(({ id, name, input, unparseableInput }) => ({
+        id, name, input,
+        ...(unparseableInput !== undefined ? { unparseableInput } : {}),
+    }));
+    const unnamed = !!cut && calls.length === 0;
+    const xml = !!cut || uses.some((block) => block.rawXml !== undefined);
+    return calls.length > 0 || unnamed ? { content, start, calls, unnamed, xml } : undefined;
+}
+/**
+ * Membrane 0.8.0's grammar (unchanged from 0.7.2) for the anthropic-xml
+ * formatter's XML tool mode. Membrane exports two of its predicates,
+ * hasUnclosedToolBlock and endsWithPartialToolBlock, but not the grammar
+ * itself (dist/utils/tool-parser.js, INVOKE_REGEX): a block opens
+ * with `<function_calls>` and each call with `<invoke name="…">`, either tag
+ * optionally `antml:` prefixed, and a closer ends the innermost block still
+ * open. Its parser takes spans out of an unclosed block's text, which leaves
+ * that text in several text blocks: thinking, which stays with the round, a
+ * whole inner block its stop sequence didn't end, whose calls are the round's
+ * too, and results, which don't stay. Only a block's opener
+ * and each call's name head are read, nothing after a name, and a name that
+ * isn't a tool name's characters isn't taken as one: the formatter writes a
+ * stored call's name into the next request as it is.
+ */
+const OPENER = /<(antml:)?function_calls(>|[^>]*$)/g;
+const CLOSER = /<\/(antml:)?function_calls>/g;
+const NAME_HEAD = /<(antml:)?invoke\s+name=(["'])((?:(?!\2).)+)\2\s*\/?>/g;
+const TOOL_NAME = /^[A-Za-z0-9_.:-]{1,128}$/;
+function cutToolBlock(content, start) {
+    const texts = content.map((block, index) => ({ block, index }))
+        .filter((entry) => entry.index >= start && entry.block.type === 'text');
+    const joined = texts.map(({ block }) => block.text).join('');
+    const tags = [
+        ...[...joined.matchAll(OPENER)].map((match) => ({ index: match.index, opens: true })),
+        ...[...joined.matchAll(CLOSER)].map((match) => ({ index: match.index, opens: false })),
+    ].sort((a, b) => a.index - b.index);
+    const open = [];
+    for (const tag of tags) {
+        if (tag.opens)
+            open.push(tag.index);
+        else
+            open.pop();
+    }
+    if (open.length === 0)
+        return undefined;
+    const opener = open[0];
+    let at = opener;
+    const holder = texts.find(({ block }) => {
+        if (at < block.text.length)
+            return true;
+        at -= block.text.length;
+        return false;
+    });
+    const blockText = joined.slice(opener);
+    const heads = [...blockText.matchAll(NAME_HEAD)].filter((head) => TOOL_NAME.test(head[3]));
+    const calls = heads.map((head, index) => ({
+        type: 'tool_use', id: `tool_${randomUUID()}`, name: head[3], input: {},
+        unparseableInput: blockText.slice(head.index, heads[index + 1]?.index),
+    }));
+    const prose = at > 0 ? [{ type: 'text', text: holder.block.text.slice(0, at) }] : [];
+    const kept = content.slice(holder.index + 1)
+        .filter((block) => block.type === 'thinking' || block.type === 'redacted_thinking' || block.type === 'tool_use');
+    return [...content.slice(0, holder.index), ...prose, ...kept, ...calls];
+}
+//# sourceMappingURL=cut-off-round.js.map

package/dist/src/cut-off-round.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cut-off-round.js","sourceRoot":"","sources":["../../src/cut-off-round.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAkBzC;;;;;;;GAOG;AACH,MAAM,UAAU,WAAW,CAAC,QAA4B;IACtD,IAAI,QAAQ,CAAC,UAAU,KAAK,YAAY;QAAE,OAAO,SAAS,CAAC;IAC3D,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;IAClC,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;QACxC,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU;YAAE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;aAChD,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,IAAI,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC;YAAE,KAAK,GAAG,KAAK,GAAG,CAAC,CAAC;IAC3F,CAAC,CAAC,CAAC;IACH,MAAM,GAAG,GAAG,QAAQ,CAAC,OAAO,EAAE,IAAI,EAAE,iBAAiB,CAAC,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAC1G,MAAM,OAAO,GAAG,GAAG,IAAI,QAAQ,CAAC,OAAO,CAAC;IACxC,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAgD,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;IAC7H,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAAE,EAAE,CAAC,CAAC;QACjE,EAAE,EAAE,IAAI,EAAE,KAAK;QACf,GAAG,CAAC,gBAAgB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAChE,CAAC,CAAC,CAAC;IACJ,MAAM,OAAO,GAAG,CAAC,CAAC,GAAG,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,CAAC;IAC5C,MAAM,GAAG,GAAG,CAAC,CAAC,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAE,KAA6B,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC;IAC/F,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;AAC3F,CAAC;AAED;;;;;;;;;;;;;;GAcG;AACH,MAAM,MAAM,GAAG,qCAAqC,CAAC;AACrD,MAAM,MAAM,GAAG,8BAA8B,CAAC;AAC9C,MAAM,SAAS,GAAG,wDAAwD,CAAC;AAC3E,MAAM,SAAS,GAAG,0BAA0B,CAAC;AAE7C,SAAS,YAAY,CAAC,OAAuB,EAAE,KAAa;IAC1D,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;SAC5D,MAAM,CAAC,CAAC,KAAK,EAAsE,EAAE,CAAC,KAAK,CAAC,KAAK,IAAI,KAAK,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;IAC9I,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC7D,MAAM,IAAI,GAAG;QACX,GAAG,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACrF,GAAG,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;KACvF,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;IACpC,MAAM,IAAI,GAAa,EAAE,CAAC;IAC1B,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,GAAG,CAAC,KAAK;YAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;;YAC/B,IAAI,CAAC,GAAG,EAAE,CAAC;IAClB,CAAC;IACD,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACxC,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAE,CAAC;IACxB,IAAI,EAAE,GAAG,MAAM,CAAC;IAChB,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE;QACtC,IAAI,EAAE,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC;QACxC,EAAE,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC;QACxB,OAAO,KAAK,CAAC;IACf,CAAC,CAAE,CAAC;IACJ,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IACvC,MAAM,KAAK,GAAG,CAAC,GAAG,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC;IAC5F,MAAM,KAAK,GAAmB,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;QACxD,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ,UAAU,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAE,EAAE,KAAK,EAAE,EAAE;QACvE,gBAAgB,EAAE,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC;KACvE,CAAC,CAAC,CAAC;IACJ,MAAM,KAAK,GAAmB,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACrG,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC;SACzC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,CAAC,IAAI,KAAK,mBAAmB,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;IACnH,OAAO,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,GAAG,KAAK,EAAE,GAAG,IAAI,EAAE,GAAG,KAAK,CAAC,CAAC;AAC1E,CAAC"}