@librechat/agents 3.1.70 → 3.1.71-dev.1

package/dist/esm/tools/toolOutputReferences.mjs

@@ -0,0 +1,649 @@
+import { ToolMessage } from '@langchain/core/messages';
+import { HARD_MAX_TOOL_RESULT_CHARS, HARD_MAX_TOTAL_TOOL_OUTPUT_SIZE, calculateMaxTotalToolOutputSize } from '../utils/truncation.mjs';
+
+/**
+ * Tool output reference registry.
+ *
+ * When enabled via `RunConfig.toolOutputReferences.enabled`, ToolNode
+ * stores each successful tool output under a stable key
+ * (`tool<idx>turn<turn>`) where `idx` is the tool's position within a
+ * ToolNode batch and `turn` is the batch index within the run
+ * (incremented once per ToolNode invocation).
+ *
+ * Subsequent tool calls can pipe a previous output into their args by
+ * embedding `{{tool<idx>turn<turn>}}` inside any string argument;
+ * {@link ToolOutputReferenceRegistry.resolve} walks the args and
+ * substitutes the placeholders immediately before invocation.
+ *
+ * The registry stores the *raw, untruncated* tool output so a later
+ * `{{…}}` substitution pipes the full payload into the next tool —
+ * even when the LLM only saw a head+tail-truncated preview in
+ * `ToolMessage.content`. Outputs are stored without any annotation
+ * (the `_ref` key or the `[ref: ...]` prefix seen by the LLM is
+ * strictly a UX signal attached to `ToolMessage.content`). Keeping the
+ * registry pristine means downstream bash/jq piping receives the
+ * complete, verbatim output with no injected fields.
+ */
+/** Object key used when a parsed-object output has `_ref` injected. */
+const TOOL_OUTPUT_REF_KEY = '_ref';
+/**
+ * Object key used to carry unresolved reference warnings on a parsed-
+ * object output. Using a dedicated field instead of a trailing text
+ * line keeps the annotated `ToolMessage.content` parseable as JSON for
+ * downstream consumers that rely on the object shape.
+ */
+const TOOL_OUTPUT_UNRESOLVED_KEY = '_unresolved_refs';
+/** Single-line prefix prepended to non-object tool outputs so the LLM sees the reference key. */
+function buildReferencePrefix(key) {
+    return `[ref: ${key}]`;
+}
+/** Stable registry key for a tool output. */
+function buildReferenceKey(toolIndex, turn) {
+    return `tool${toolIndex}turn${turn}`;
+}
+const EMPTY_ENTRIES = new Map();
+/**
+ * Per-run state bucket held inside the registry. Each distinct
+ * `run_id` gets its own bucket so overlapping concurrent runs on a
+ * shared registry cannot leak outputs, turn counters, or warn-memos
+ * into one another.
+ */
+class RunStateBucket {
+    entries = new Map();
+    totalSize = 0;
+    turnCounter = 0;
+    warnedNonStringTools = new Set();
+}
+/**
+ * Anonymous (`run_id` absent) bucket key. Anonymous batches are
+ * treated as fresh runs on every invocation — see `nextTurn`.
+ */
+const ANON_RUN_KEY = '\0anon';
+/**
+ * Default upper bound on the number of concurrently-tracked runs per
+ * registry. When exceeded, the oldest run's bucket (by insertion
+ * order) is evicted. Keeps memory bounded when a ToolNode is reused
+ * across many runs without explicit `releaseRun` calls.
+ */
+const DEFAULT_MAX_ACTIVE_RUNS = 32;
+/**
+ * Ordered map of reference-key → stored output, partitioned by run so
+ * concurrent / interleaved runs sharing one registry cannot leak
+ * outputs between each other.
+ *
+ * Each public method takes a `runId` which selects the run's bucket.
+ * Hosts typically get one registry per run via `Graph`, in which
+ * case only a single bucket is ever populated; the partitioning
+ * exists so the registry also behaves correctly when a single
+ * instance is reused directly.
+ */
+class ToolOutputReferenceRegistry {
+    runStates = new Map();
+    maxOutputSize;
+    maxTotalSize;
+    maxActiveRuns;
+    /**
+     * Local stateful matcher used only by `replaceInString`. Kept
+     * off-module so callers of the exported `TOOL_OUTPUT_REF_PATTERN`
+     * never see a stale `lastIndex`.
+     */
+    static PLACEHOLDER_MATCHER = /\{\{(tool\d+turn\d+)\}\}/g;
+    constructor(options = {}) {
+        /**
+         * Per-output default is the same ~400 KB budget as the standard
+         * tool-result truncation (`HARD_MAX_TOOL_RESULT_CHARS`). This
+         * keeps a single `{{…}}` substitution at a size that is safe to
+         * pass through typical shell `ARG_MAX` limits and matches what
+         * the LLM would otherwise have seen. Hosts that want larger per-
+         * output payloads (API consumers, long JSON streams) can raise
+         * the cap explicitly up to the 5 MB total budget.
+         */
+        const perOutput = options.maxOutputSize != null && options.maxOutputSize > 0
+            ? options.maxOutputSize
+            : HARD_MAX_TOOL_RESULT_CHARS;
+        /**
+         * Clamp a caller-supplied `maxTotalSize` to
+         * `HARD_MAX_TOTAL_TOOL_OUTPUT_SIZE` (5 MB) so the documented
+         * absolute cap is enforced regardless of host config —
+         * `calculateMaxTotalToolOutputSize` already applies the same
+         * upper bound on its computed default, but the user-provided
+         * branch was bypassing it.
+         */
+        const totalRaw = options.maxTotalSize != null && options.maxTotalSize > 0
+            ? Math.min(options.maxTotalSize, HARD_MAX_TOTAL_TOOL_OUTPUT_SIZE)
+            : calculateMaxTotalToolOutputSize(perOutput);
+        this.maxTotalSize = totalRaw;
+        /**
+         * The per-output cap can never exceed the per-run aggregate cap:
+         * if a single entry were allowed to be larger than `maxTotalSize`,
+         * the eviction loop would either blow the cap (to keep the entry)
+         * or self-evict a just-stored value. Clamping here turns
+         * `maxTotalSize` into a hard upper bound on *any* state the
+         * registry retains per run.
+         */
+        this.maxOutputSize = Math.min(perOutput, totalRaw);
+        this.maxActiveRuns =
+            options.maxActiveRuns != null && options.maxActiveRuns > 0
+                ? options.maxActiveRuns
+                : DEFAULT_MAX_ACTIVE_RUNS;
+    }
+    keyFor(runId) {
+        return runId ?? ANON_RUN_KEY;
+    }
+    getOrCreate(runId) {
+        const key = this.keyFor(runId);
+        let state = this.runStates.get(key);
+        if (state == null) {
+            state = new RunStateBucket();
+            this.runStates.set(key, state);
+            if (this.runStates.size > this.maxActiveRuns) {
+                const oldest = this.runStates.keys().next().value;
+                if (oldest != null && oldest !== key) {
+                    this.runStates.delete(oldest);
+                }
+            }
+        }
+        return state;
+    }
+    /** Registers (or replaces) the output stored under `key` for `runId`. */
+    set(runId, key, value) {
+        const bucket = this.getOrCreate(runId);
+        const clipped = value.length > this.maxOutputSize
+            ? value.slice(0, this.maxOutputSize)
+            : value;
+        const existing = bucket.entries.get(key);
+        if (existing != null) {
+            bucket.totalSize -= existing.length;
+            bucket.entries.delete(key);
+        }
+        bucket.entries.set(key, clipped);
+        bucket.totalSize += clipped.length;
+        this.evictWithinBucket(bucket);
+    }
+    /** Returns the stored value for `key` in `runId`'s bucket, or `undefined`. */
+    get(runId, key) {
+        return this.runStates.get(this.keyFor(runId))?.entries.get(key);
+    }
+    /**
+     * Returns `true` when `key` is currently stored in `runId`'s bucket.
+     * Used by {@link annotateMessagesForLLM} to gate transient annotation
+     * on whether the registry still owns the referenced output (a stale
+     * `_refKey` from a prior run silently no-ops here).
+     */
+    has(runId, key) {
+        return this.runStates.get(this.keyFor(runId))?.entries.has(key) ?? false;
+    }
+    /** Total number of registered outputs across every run bucket. */
+    get size() {
+        let n = 0;
+        for (const bucket of this.runStates.values()) {
+            n += bucket.entries.size;
+        }
+        return n;
+    }
+    /** Maximum characters retained per output (post-clip). */
+    get perOutputLimit() {
+        return this.maxOutputSize;
+    }
+    /** Maximum total characters retained *per run*. */
+    get totalLimit() {
+        return this.maxTotalSize;
+    }
+    /** Drops every run's state. */
+    clear() {
+        this.runStates.clear();
+    }
+    /**
+     * Explicitly release `runId`'s state. Safe to call when a run has
+     * finished. Hosts sharing one registry across runs should call this
+     * to reclaim memory deterministically; otherwise LRU eviction kicks
+     * in when `maxActiveRuns` runs accumulate.
+     */
+    releaseRun(runId) {
+        this.runStates.delete(this.keyFor(runId));
+    }
+    /**
+     * Claims the next batch turn synchronously from `runId`'s bucket.
+     *
+     * Must be called once at the start of each ToolNode batch before
+     * any `await`, so concurrent invocations within the same run see
+     * distinct turn values (reads are effectively atomic by JS's
+     * single-threaded execution of the sync prefix).
+     *
+     * If `runId` is missing the anonymous bucket is dropped and a
+     * fresh one created so each anonymous call behaves as its own run.
+     */
+    nextTurn(runId) {
+        if (runId == null) {
+            this.runStates.delete(ANON_RUN_KEY);
+        }
+        const bucket = this.getOrCreate(runId);
+        return bucket.turnCounter++;
+    }
+    /**
+     * Records that `toolName` has been warned about in `runId` (returns
+     * `true` on the first call per run, `false` after). Used by
+     * ToolNode to emit one log line per offending tool per run when a
+     * `ToolMessage.content` isn't a string.
+     */
+    claimWarnOnce(runId, toolName) {
+        const bucket = this.getOrCreate(runId);
+        if (bucket.warnedNonStringTools.has(toolName)) {
+            return false;
+        }
+        bucket.warnedNonStringTools.add(toolName);
+        return true;
+    }
+    /**
+     * Walks `args` and replaces every `{{tool<i>turn<n>}}` placeholder in
+     * string values with the stored output *from `runId`'s bucket*. Non-
+     * string values and object keys are left untouched. Unresolved
+     * references are left in-place and reported so the caller can
+     * surface them to the LLM. When no placeholder appears anywhere in
+     * the serialized args, the original input is returned without
+     * walking the tree.
+     */
+    resolve(runId, args) {
+        if (!hasAnyPlaceholder(args)) {
+            return { resolved: args, unresolved: [] };
+        }
+        const bucket = this.runStates.get(this.keyFor(runId));
+        return this.resolveAgainst(bucket?.entries ?? EMPTY_ENTRIES, args);
+    }
+    /**
+     * Captures a frozen snapshot of `runId`'s current entries and
+     * returns a view that resolves placeholders against *only* that
+     * snapshot. The snapshot is decoupled from the live registry, so
+     * subsequent `set()` calls (for example, same-turn direct outputs
+     * registering while an event branch is still in flight) are
+     * invisible to the snapshot's `resolve`. Used by the mixed
+     * direct+event dispatch path to preserve same-turn isolation when
+     * a `PreToolUse` hook rewrites event args after directs have
+     * completed.
+     */
+    snapshot(runId) {
+        const bucket = this.runStates.get(this.keyFor(runId));
+        const entries = bucket
+            ? new Map(bucket.entries)
+            : EMPTY_ENTRIES;
+        return {
+            resolve: (args) => this.resolveAgainst(entries, args),
+        };
+    }
+    resolveAgainst(entries, args) {
+        if (!hasAnyPlaceholder(args)) {
+            return { resolved: args, unresolved: [] };
+        }
+        const unresolved = new Set();
+        const resolved = this.transform(entries, args, unresolved);
+        return { resolved, unresolved: Array.from(unresolved) };
+    }
+    transform(entries, value, unresolved) {
+        if (typeof value === 'string') {
+            return this.replaceInString(entries, value, unresolved);
+        }
+        if (Array.isArray(value)) {
+            return value.map((item) => this.transform(entries, item, unresolved));
+        }
+        if (value !== null && typeof value === 'object') {
+            const source = value;
+            const next = {};
+            for (const [key, item] of Object.entries(source)) {
+                next[key] = this.transform(entries, item, unresolved);
+            }
+            return next;
+        }
+        return value;
+    }
+    replaceInString(entries, input, unresolved) {
+        if (input.indexOf('{{tool') === -1) {
+            return input;
+        }
+        return input.replace(ToolOutputReferenceRegistry.PLACEHOLDER_MATCHER, (match, key) => {
+            const stored = entries.get(key);
+            if (stored == null) {
+                unresolved.add(key);
+                return match;
+            }
+            return stored;
+        });
+    }
+    evictWithinBucket(bucket) {
+        if (bucket.totalSize <= this.maxTotalSize) {
+            return;
+        }
+        for (const key of bucket.entries.keys()) {
+            if (bucket.totalSize <= this.maxTotalSize) {
+                return;
+            }
+            const entry = bucket.entries.get(key);
+            if (entry == null) {
+                continue;
+            }
+            bucket.totalSize -= entry.length;
+            bucket.entries.delete(key);
+        }
+    }
+}
+/**
+ * Cheap pre-check: returns true if any string value in `args` contains
+ * the `{{tool` substring. Lets `resolve()` skip the deep tree walk (and
+ * its object allocations) for the common case of plain args.
+ */
+function hasAnyPlaceholder(value) {
+    if (typeof value === 'string') {
+        return value.indexOf('{{tool') !== -1;
+    }
+    if (Array.isArray(value)) {
+        for (const item of value) {
+            if (hasAnyPlaceholder(item)) {
+                return true;
+            }
+        }
+        return false;
+    }
+    if (value !== null && typeof value === 'object') {
+        for (const item of Object.values(value)) {
+            if (hasAnyPlaceholder(item)) {
+                return true;
+            }
+        }
+        return false;
+    }
+    return false;
+}
+/**
+ * Annotates `content` with a reference key and/or unresolved-ref
+ * warnings so the LLM sees both alongside the tool output.
+ *
+ * Behavior:
+ *  - If `content` parses as a plain (non-array, non-null) JSON object
+ *    and the object does not already have a conflicting `_ref` key,
+ *    the reference key and (when present) `_unresolved_refs` array
+ *    are injected as object fields, preserving JSON validity for
+ *    downstream consumers that parse the output.
+ *  - Otherwise (string output, JSON array/primitive, parse failure,
+ *    or `_ref` collision), a `[ref: <key>]\n` prefix line is
+ *    prepended and unresolved refs are appended as a trailing
+ *    `[unresolved refs: …]` line.
+ *
+ * The annotated string is what the LLM sees as `ToolMessage.content`.
+ * The *original* (un-annotated) value is what gets stored in the
+ * registry, so downstream piping remains pristine.
+ *
+ * @param content     Raw (post-truncation) tool output.
+ * @param key         Reference key for this output, or undefined when
+ *                    there is nothing to register (errors etc.).
+ * @param unresolved  Reference keys that failed to resolve during
+ *                    argument substitution. Surfaced so the LLM can
+ *                    self-correct its next tool call.
+ */
+function annotateToolOutputWithReference(content, key, unresolved = []) {
+    const hasRefKey = key != null;
+    const hasUnresolved = unresolved.length > 0;
+    if (!hasRefKey && !hasUnresolved) {
+        return content;
+    }
+    const trimmed = content.trimStart();
+    if (trimmed.startsWith('{')) {
+        const annotated = tryInjectRefIntoJsonObject(content, key, unresolved);
+        if (annotated != null) {
+            return annotated;
+        }
+    }
+    const prefix = hasRefKey ? `${buildReferencePrefix(key)}\n` : '';
+    const trailer = hasUnresolved
+        ? `\n[unresolved refs: ${unresolved.join(', ')}]`
+        : '';
+    return `${prefix}${content}${trailer}`;
+}
+function tryInjectRefIntoJsonObject(content, key, unresolved) {
+    let parsed;
+    try {
+        parsed = JSON.parse(content);
+    }
+    catch {
+        return null;
+    }
+    if (parsed === null || typeof parsed !== 'object' || Array.isArray(parsed)) {
+        return null;
+    }
+    const obj = parsed;
+    const injectingRef = key != null;
+    const injectingUnresolved = unresolved.length > 0;
+    /**
+     * Reject the JSON-injection path (fall back to prefix form) when
+     * either of our keys collides with real payload data:
+     *  - `_ref` collision: existing value is non-null and differs from
+     *    the key we're about to inject.
+     *  - `_unresolved_refs` collision: existing value is non-null and
+     *    is not a deep-equal match for the array we'd inject.
+     * This keeps us from silently overwriting legitimate tool output.
+     */
+    if (injectingRef &&
+        TOOL_OUTPUT_REF_KEY in obj &&
+        obj[TOOL_OUTPUT_REF_KEY] !== key &&
+        obj[TOOL_OUTPUT_REF_KEY] != null) {
+        return null;
+    }
+    if (injectingUnresolved &&
+        TOOL_OUTPUT_UNRESOLVED_KEY in obj &&
+        obj[TOOL_OUTPUT_UNRESOLVED_KEY] != null &&
+        !arraysShallowEqual(obj[TOOL_OUTPUT_UNRESOLVED_KEY], unresolved)) {
+        return null;
+    }
+    /**
+     * Only strip the framework-owned key we're actually injecting —
+     * leave everything else (including a pre-existing `_ref` on the
+     * unresolved-only path, or a pre-existing `_unresolved_refs` on a
+     * plain-annotation path) untouched so we annotate rather than
+     * mutate downstream payload data. Our injected keys land first in
+     * the serialized JSON so the LLM sees them before the body.
+     */
+    const omitKeys = new Set();
+    if (injectingRef)
+        omitKeys.add(TOOL_OUTPUT_REF_KEY);
+    if (injectingUnresolved)
+        omitKeys.add(TOOL_OUTPUT_UNRESOLVED_KEY);
+    const rest = {};
+    for (const [k, v] of Object.entries(obj)) {
+        if (!omitKeys.has(k)) {
+            rest[k] = v;
+        }
+    }
+    const injected = {};
+    if (injectingRef) {
+        injected[TOOL_OUTPUT_REF_KEY] = key;
+    }
+    if (injectingUnresolved) {
+        injected[TOOL_OUTPUT_UNRESOLVED_KEY] = unresolved;
+    }
+    Object.assign(injected, rest);
+    const pretty = /^\{\s*\n/.test(content);
+    return pretty ? JSON.stringify(injected, null, 2) : JSON.stringify(injected);
+}
+function arraysShallowEqual(a, b) {
+    if (!Array.isArray(a) || a.length !== b.length) {
+        return false;
+    }
+    for (let i = 0; i < a.length; i++) {
+        if (a[i] !== b[i]) {
+            return false;
+        }
+    }
+    return true;
+}
+/**
+ * Lazy projection that, given a registry and a runId, returns a new
+ * `messages` array where each `ToolMessage` carrying ref metadata is
+ * projected into a transient copy with annotated content (when the ref
+ * is live in the registry) and with the framework-owned `additional_
+ * kwargs` keys (`_refKey`, `_refScope`, `_unresolvedRefs`) stripped
+ * regardless of whether annotation applied. The original input array
+ * and its messages are never mutated.
+ *
+ * Annotation is gated on registry presence: a stale `_refKey` from a
+ * prior run (e.g. one that survived in persisted history) silently
+ * no-ops on the *content* side. The strip-metadata side still runs so
+ * stale framework keys never leak onto the wire under any custom or
+ * future provider serializer that might transmit `additional_kwargs`.
+ * `_unresolvedRefs` is always meaningful and is not gated.
+ *
+ * **Feature-disabled fast path:** when the host hasn't enabled the
+ * tool-output-reference feature, the registry is `undefined` and this
+ * function returns the input array reference-equal *without iterating
+ * a single message*. The loop is exclusive to the feature-enabled
+ * code path.
+ */
+function annotateMessagesForLLM(messages, registry, runId) {
+    if (registry == null)
+        return messages;
+    /**
+     * Lazy-allocate the output array so the common case (no ToolMessage
+     * carries framework metadata) returns the input reference-equal with
+     * zero allocations beyond the per-message predicate checks.
+     */
+    let out;
+    for (let i = 0; i < messages.length; i++) {
+        const m = messages[i];
+        if (m._getType() !== 'tool')
+            continue;
+        /**
+         * `additional_kwargs` is untyped at the LangChain layer
+         * (`Record<string, unknown>`), so persisted or client-supplied
+         * ToolMessages can carry arbitrary shapes under our framework
+         * keys. Treat them as untrusted input and coerce defensively
+         * before any array operation — a malformed field on a single
+         * hydrated message must not crash `attemptInvoke` before the
+         * provider call.
+         */
+        const meta = m.additional_kwargs;
+        const hasRefKey = meta != null && '_refKey' in meta;
+        const hasRefScope = meta != null && '_refScope' in meta;
+        const hasUnresolvedField = meta != null && '_unresolvedRefs' in meta;
+        if (!hasRefKey && !hasRefScope && !hasUnresolvedField)
+            continue;
+        const refKey = readRefKey(meta);
+        const unresolved = readUnresolvedRefs(meta);
+        /**
+         * Prefer the message-stamped `_refScope` for the registry lookup.
+         * For named runs it equals the current `runId`; for anonymous
+         * invocations it carries the per-batch synthetic scope minted by
+         * ToolNode (`\0anon-<n>`), which `runId` from config cannot
+         * recover. Falling back to `runId` keeps backward compatibility
+         * with messages stamped before this field existed.
+         */
+        const lookupScope = readRefScope(meta) ?? runId;
+        const liveRef = refKey != null && registry.has(lookupScope, refKey) ? refKey : undefined;
+        const annotates = liveRef != null || unresolved.length > 0;
+        const tm = m;
+        let nextContent = tm.content;
+        if (annotates && typeof tm.content === 'string') {
+            nextContent = annotateToolOutputWithReference(tm.content, liveRef, unresolved);
+        }
+        else if (annotates &&
+            Array.isArray(tm.content) &&
+            unresolved.length > 0) {
+            const warningBlock = {
+                type: 'text',
+                text: `[unresolved refs: ${unresolved.join(', ')}]`,
+            };
+            nextContent = [
+                warningBlock,
+                ...tm.content,
+            ];
+        }
+        /**
+         * Project unconditionally: even when no annotation applies (stale
+         * `_refKey` or non-annotatable content), `cloneToolMessageWithContent`
+         * runs `stripFrameworkRefMetadata` on `additional_kwargs` so the
+         * framework-owned keys never reach the wire.
+         */
+        out ??= messages.slice();
+        out[i] = cloneToolMessageWithContent(tm, nextContent);
+    }
+    return out ?? messages;
+}
+/**
+ * Reads `_refKey` defensively from untyped `additional_kwargs`. Returns
+ * undefined for non-string values so a malformed field cannot poison
+ * the registry lookup or downstream string operations.
+ */
+function readRefKey(meta) {
+    const v = meta?._refKey;
+    return typeof v === 'string' ? v : undefined;
+}
+/**
+ * Reads `_refScope` defensively from untyped `additional_kwargs`.
+ * Mirrors {@link readRefKey} — non-string scopes are dropped (the
+ * caller falls back to the run-derived scope) rather than passed into
+ * the registry as a malformed key.
+ */
+function readRefScope(meta) {
+    const v = meta?._refScope;
+    return typeof v === 'string' ? v : undefined;
+}
+/**
+ * Reads `_unresolvedRefs` defensively from untyped `additional_kwargs`.
+ * Returns an empty array for any non-array value, and filters out
+ * non-string entries from a real array. Without this guard, a hydrated
+ * ToolMessage carrying e.g. `_unresolvedRefs: 'tool0turn0'` would crash
+ * `attemptInvoke` on the eventual `.length` / `.join(...)` call.
+ */
+function readUnresolvedRefs(meta) {
+    const v = meta?._unresolvedRefs;
+    if (!Array.isArray(v))
+        return [];
+    const out = [];
+    for (const item of v) {
+        if (typeof item === 'string')
+            out.push(item);
+    }
+    return out;
+}
+/**
+ * Builds a fresh `ToolMessage` that mirrors `tm`'s identity fields with
+ * the supplied `content`. Every `ToolMessage` field but `content` is
+ * carried over so the projection is structurally identical to the
+ * original from a LangChain serializer's perspective.
+ *
+ * `additional_kwargs` is rebuilt with the framework-owned ref keys
+ * stripped. Defensive: LangChain's standard provider serializers do not
+ * transmit `additional_kwargs` to provider HTTP APIs, but a custom
+ * adapter or future LangChain change could. Stripping keeps the
+ * implementation correct under any serializer behavior at the cost of a
+ * shallow object spread per annotated message.
+ */
+function cloneToolMessageWithContent(tm, content) {
+    return new ToolMessage({
+        id: tm.id,
+        name: tm.name,
+        status: tm.status,
+        artifact: tm.artifact,
+        tool_call_id: tm.tool_call_id,
+        response_metadata: tm.response_metadata,
+        additional_kwargs: stripFrameworkRefMetadata(tm.additional_kwargs),
+        content,
+    });
+}
+/**
+ * Returns a copy of `kwargs` with `_refKey`, `_refScope`, and
+ * `_unresolvedRefs` removed. Returns the input reference-equal when
+ * none of those keys are present so the no-strip path stays cheap;
+ * returns `undefined` when stripping leaves the object empty so the
+ * caller can drop the field entirely.
+ */
+function stripFrameworkRefMetadata(kwargs) {
+    if (kwargs == null)
+        return undefined;
+    if (!('_refKey' in kwargs) &&
+        !('_refScope' in kwargs) &&
+        !('_unresolvedRefs' in kwargs)) {
+        return kwargs;
+    }
+    const { _refKey, _refScope, _unresolvedRefs, ...rest } = kwargs;
+    return Object.keys(rest).length === 0 ? undefined : rest;
+}
+
+export { TOOL_OUTPUT_REF_KEY, TOOL_OUTPUT_UNRESOLVED_KEY, ToolOutputReferenceRegistry, annotateMessagesForLLM, annotateToolOutputWithReference, buildReferenceKey, buildReferencePrefix };
+//# sourceMappingURL=toolOutputReferences.mjs.map