@openai/agents-core 0.3.7 → 0.3.8

package/dist/runner/conversation.js

@@ -0,0 +1,275 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ServerConversationTracker = exports.getTurnInput = void 0;
+exports.applyCallModelInputFilter = applyCallModelInputFilter;
+const errors_1 = require("../errors.js");
+const context_1 = require("../tracing/context.js");
+const items_1 = require("./items.js");
+var items_2 = require("./items.js");
+Object.defineProperty(exports, "getTurnInput", { enumerable: true, get: function () { return items_2.getTurnInput; } });
+/**
+ * Applies the optional callModelInputFilter and returns the filtered input alongside the original
+ * items so downstream tracking and session persistence stay in sync with what the model saw.
+ */
+async function applyCallModelInputFilter(agent, callModelInputFilter, context, inputItems, systemInstructions) {
+    const cloneInputItems = (items, map) => items.map((item) => {
+        const cloned = structuredClone(item);
+        if (map && cloned && typeof cloned === 'object') {
+            map.set(cloned, item);
+        }
+        return cloned;
+    });
+    // Record the relationship between the cloned array passed to filters and the original inputs.
+    const cloneMap = new WeakMap();
+    const originalPool = (0, items_1.buildAgentInputPool)(inputItems);
+    const fallbackOriginals = [];
+    // Track any original object inputs so filtered replacements can still mark them as delivered.
+    for (const item of inputItems) {
+        if (item && typeof item === 'object') {
+            fallbackOriginals.push(item);
+        }
+    }
+    const removeFromFallback = (candidate) => {
+        if (!candidate || typeof candidate !== 'object') {
+            return;
+        }
+        const index = fallbackOriginals.findIndex((original) => original === candidate);
+        if (index !== -1) {
+            fallbackOriginals.splice(index, 1);
+        }
+    };
+    const takeFallbackOriginal = () => {
+        const next = fallbackOriginals.shift();
+        if (next) {
+            (0, items_1.removeAgentInputFromPool)(originalPool, next);
+        }
+        return next;
+    };
+    // Always create a deep copy so downstream mutations inside filters cannot affect
+    // the cached turn state.
+    const clonedBaseInput = cloneInputItems(inputItems, cloneMap);
+    const base = {
+        input: clonedBaseInput,
+        instructions: systemInstructions,
+    };
+    if (!callModelInputFilter) {
+        return {
+            modelInput: base,
+            sourceItems: [...inputItems],
+            persistedItems: [],
+            filterApplied: false,
+        };
+    }
+    try {
+        const result = await callModelInputFilter({
+            modelData: base,
+            agent,
+            context: context.context,
+        });
+        if (!result || !Array.isArray(result.input)) {
+            throw new errors_1.UserError('callModelInputFilter must return a ModelInputData object with an input array.');
+        }
+        // Preserve a pointer to the original object backing each filtered clone so downstream
+        // trackers can keep their bookkeeping consistent even after redaction.
+        const sourceItems = result.input.map((item) => {
+            if (!item || typeof item !== 'object') {
+                return undefined;
+            }
+            const original = cloneMap.get(item);
+            if (original) {
+                removeFromFallback(original);
+                (0, items_1.removeAgentInputFromPool)(originalPool, original);
+                return original;
+            }
+            const key = (0, items_1.getAgentInputItemKey)(item);
+            const matchedByContent = (0, items_1.takeAgentInputFromPool)(originalPool, key);
+            if (matchedByContent) {
+                removeFromFallback(matchedByContent);
+                return matchedByContent;
+            }
+            const fallback = takeFallbackOriginal();
+            if (fallback) {
+                return fallback;
+            }
+            return undefined;
+        });
+        const clonedFilteredInput = cloneInputItems(result.input);
+        return {
+            modelInput: {
+                input: clonedFilteredInput,
+                instructions: typeof result.instructions === 'undefined'
+                    ? systemInstructions
+                    : result.instructions,
+            },
+            sourceItems,
+            persistedItems: clonedFilteredInput.map((item) => structuredClone(item)),
+            filterApplied: true,
+        };
+    }
+    catch (error) {
+        (0, context_1.addErrorToCurrentSpan)({
+            message: 'Error in callModelInputFilter',
+            data: { error: String(error) },
+        });
+        throw error;
+    }
+}
+/**
+ * Tracks which items have already been sent to or received from the Responses API when the caller
+ * supplies `conversationId`/`previousResponseId`. This ensures we only send the delta each turn.
+ */
+class ServerConversationTracker {
+    conversationId;
+    previousResponseId;
+    // Using this flag because WeakSet does not provide a way to check its size.
+    sentInitialInput = false;
+    // The items already sent to the model; using WeakSet for memory efficiency.
+    sentItems = new WeakSet();
+    // The items received from the server; using WeakSet for memory efficiency.
+    serverItems = new WeakSet();
+    // Track initial input items that have not yet been sent so they can be retried on later turns.
+    remainingInitialInput = null;
+    constructor({ conversationId, previousResponseId, }) {
+        this.conversationId = conversationId ?? undefined;
+        this.previousResponseId = previousResponseId ?? undefined;
+    }
+    /**
+     * Pre-populates tracker caches from an existing RunState when resuming server-managed runs.
+     */
+    primeFromState({ originalInput, generatedItems, modelResponses, }) {
+        if (this.sentInitialInput) {
+            return;
+        }
+        const originalItems = (0, items_1.toAgentInputList)(originalInput);
+        const hasResponses = modelResponses.length > 0;
+        const serverItemKeys = new Set();
+        for (const response of modelResponses) {
+            for (const item of response.output) {
+                if (item && typeof item === 'object') {
+                    this.serverItems.add(item);
+                    serverItemKeys.add((0, items_1.getAgentInputItemKey)(item));
+                }
+            }
+        }
+        if (hasResponses) {
+            for (const item of originalItems) {
+                if (item && typeof item === 'object') {
+                    this.sentItems.add(item);
+                }
+            }
+            this.sentInitialInput = true;
+            this.remainingInitialInput = null;
+        }
+        const latestResponse = modelResponses[modelResponses.length - 1];
+        if (!this.conversationId && latestResponse?.responseId) {
+            this.previousResponseId = latestResponse.responseId;
+        }
+        if (hasResponses) {
+            for (const item of generatedItems) {
+                const rawItem = item.rawItem;
+                if (!rawItem || typeof rawItem !== 'object') {
+                    continue;
+                }
+                const rawItemKey = (0, items_1.getAgentInputItemKey)(rawItem);
+                if (this.serverItems.has(rawItem) || serverItemKeys.has(rawItemKey)) {
+                    this.sentItems.add(rawItem);
+                }
+            }
+        }
+    }
+    /**
+     * Records the raw items returned by the server so future delta calculations skip them.
+     * Also captures the latest response identifier to chain follow-up calls when possible.
+     */
+    trackServerItems(modelResponse) {
+        if (!modelResponse) {
+            return;
+        }
+        for (const item of modelResponse.output) {
+            if (item && typeof item === 'object') {
+                this.serverItems.add(item);
+            }
+        }
+        if (!this.conversationId && modelResponse.responseId) {
+            this.previousResponseId = modelResponse.responseId;
+        }
+    }
+    /**
+     * Returns the minimum set of items that still need to be delivered to the server for the
+     * current turn. This includes the original turn inputs (until acknowledged) plus any
+     * newly generated items that have not yet been echoed back by the API.
+     */
+    prepareInput(originalInput, generatedItems) {
+        const inputItems = [];
+        if (!this.sentInitialInput) {
+            const initialItems = (0, items_1.toAgentInputList)(originalInput);
+            // Preserve the full initial payload so a filter can drop items without losing their originals.
+            inputItems.push(...initialItems);
+            this.remainingInitialInput = initialItems.filter((item) => Boolean(item) && typeof item === 'object');
+            this.sentInitialInput = true;
+        }
+        else if (this.remainingInitialInput &&
+            this.remainingInitialInput.length > 0) {
+            // Re-queue prior initial items until the tracker confirms they were delivered to the API.
+            inputItems.push(...this.remainingInitialInput);
+        }
+        for (const item of generatedItems) {
+            if (item.type === 'tool_approval_item') {
+                continue;
+            }
+            const rawItem = item.rawItem;
+            if (!rawItem || typeof rawItem !== 'object') {
+                continue;
+            }
+            if (this.sentItems.has(rawItem) || this.serverItems.has(rawItem)) {
+                continue;
+            }
+            inputItems.push(rawItem);
+        }
+        return inputItems;
+    }
+    /**
+     * Marks the provided originals as delivered so future turns do not resend them and any
+     * pending initial inputs can be dropped once the server acknowledges receipt.
+     */
+    markInputAsSent(items, options) {
+        const delivered = new Set();
+        const dropRemainingInitialInput = options?.filterApplied ?? false;
+        const markFilteredItemsAsSent = options?.filterApplied && Boolean(options.allTurnItems);
+        this.addDeliveredItems(delivered, items);
+        const allTurnItems = options?.allTurnItems;
+        if (markFilteredItemsAsSent && allTurnItems) {
+            this.addDeliveredItems(delivered, allTurnItems);
+        }
+        this.updateRemainingInitialInput(delivered, Boolean(dropRemainingInitialInput));
+    }
+    addDeliveredItems(delivered, items) {
+        for (const item of items) {
+            if (!item || typeof item !== 'object' || delivered.has(item)) {
+                continue;
+            }
+            // Some inputs may be repeated in the filtered list; only mark unique originals once.
+            delivered.add(item);
+            this.sentItems.add(item);
+        }
+    }
+    updateRemainingInitialInput(delivered, dropRemainingInitialInput) {
+        if (!this.remainingInitialInput ||
+            this.remainingInitialInput.length === 0 ||
+            delivered.size === 0) {
+            if (dropRemainingInitialInput && this.remainingInitialInput) {
+                this.remainingInitialInput = null;
+            }
+            return;
+        }
+        this.remainingInitialInput = this.remainingInitialInput.filter((item) => !delivered.has(item));
+        if (this.remainingInitialInput.length === 0) {
+            this.remainingInitialInput = null;
+        }
+        else if (dropRemainingInitialInput) {
+            this.remainingInitialInput = null;
+        }
+    }
+}
+exports.ServerConversationTracker = ServerConversationTracker;
+//# sourceMappingURL=conversation.js.map

package/dist/runner/conversation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"conversation.js","sourceRoot":"","sources":["../../src/runner/conversation.ts"],"names":[],"mappings":";;;AAqDA,8DA0HC;AA9KD,yCAAsC;AAKtC,mDAA2D;AAC3D,sCAMiB;AAEjB,oCAAuC;AAA9B,qGAAA,YAAY,OAAA;AAkCrB;;;GAGG;AACI,KAAK,UAAU,yBAAyB,CAC7C,KAAuC,EACvC,oBAA2D,EAC3D,OAA6B,EAC7B,UAA4B,EAC5B,kBAAsC;IAEtC,MAAM,eAAe,GAAG,CACtB,KAAuB,EACvB,GAAqC,EACrC,EAAE,CACF,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACjB,MAAM,MAAM,GAAG,eAAe,CAAC,IAAI,CAAmB,CAAC;QACvD,IAAI,GAAG,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAChD,GAAG,CAAC,GAAG,CAAC,MAAgB,EAAE,IAAI,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC,CAAC;IAEL,8FAA8F;IAC9F,MAAM,QAAQ,GAAG,IAAI,OAAO,EAA0B,CAAC;IACvD,MAAM,YAAY,GAAG,IAAA,2BAAmB,EAAC,UAAU,CAAC,CAAC;IACrD,MAAM,iBAAiB,GAAqB,EAAE,CAAC;IAC/C,8FAA8F;IAC9F,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAC9B,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YACrC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IACD,MAAM,kBAAkB,GAAG,CAAC,SAAqC,EAAE,EAAE;QACnE,IAAI,CAAC,SAAS,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;YAChD,OAAO;QACT,CAAC;QACD,MAAM,KAAK,GAAG,iBAAiB,CAAC,SAAS,CACvC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,KAAK,SAAS,CACrC,CAAC;QACF,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;YACjB,iBAAiB,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACrC,CAAC;IACH,CAAC,CAAC;IACF,MAAM,oBAAoB,GAAG,GAA+B,EAAE;QAC5D,MAAM,IAAI,GAAG,iBAAiB,CAAC,KAAK,EAAE,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,IAAA,gCAAwB,EAAC,YAAY,EAAE,IAAI,CAAC,CAAC;QAC/C,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC;IAEF,iFAAiF;IACjF,yBAAyB;IACzB,MAAM,eAAe,GAAG,eAAe,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;IAC9D,MAAM,IAAI,GAAG;QACX,KAAK,EAAE,eAAe;QACtB,YAAY,EAAE,kBAAkB;KACjC,CAAC;IACF,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC1B,OAAO;YACL,UAAU,EAAE,IAAI;YAChB,WAAW,EAAE,CAAC,GAAG,UAAU,CAAC;YAC5B,cAAc,EAAE,EAAE;YAClB,aAAa,EAAE,KAAK;SACrB,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,oBAAoB,CAAC;YACxC,SAAS,EAAE,IAAI;YACf,KAAK;YACL,OAAO,EAAE,OAAO,CAAC,OAAO;SACQ,CAAC,CAAC;QAEpC,IAAI,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;YAC5C,MAAM,IAAI,kBAAS,CACjB,+EAA+E,CAChF,CAAC;QACJ,CAAC;QAED,sFAAsF;QACtF,uEAAuE;QACvE,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YAC5C,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtC,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAc,CAAC,CAAC;YAC9C,IAAI,QAAQ,EAAE,CAAC;gBACb,kBAAkB,CAAC,QAAQ,CAAC,CAAC;gBAC7B,IAAA,gCAAwB,EAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;gBACjD,OAAO,QAAQ,CAAC;YAClB,CAAC;YACD,MAAM,GAAG,GAAG,IAAA,4BAAoB,EAAC,IAAsB,CAAC,CAAC;YACzD,MAAM,gBAAgB,GAAG,IAAA,8BAAsB,EAAC,YAAY,EAAE,GAAG,CAAC,CAAC;YACnE,IAAI,gBAAgB,EAAE,CAAC;gBACrB,kBAAkB,CAAC,gBAAgB,CAAC,CAAC;gBACrC,OAAO,gBAAgB,CAAC;YAC1B,CAAC;YACD,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;YACxC,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO,QAAQ,CAAC;YAClB,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,MAAM,mBAAmB,GAAG,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1D,OAAO;YACL,UAAU,EAAE;gBACV,KAAK,EAAE,mBAAmB;gBAC1B,YAAY,EACV,OAAO,MAAM,CAAC,YAAY,KAAK,WAAW;oBACxC,CAAC,CAAC,kBAAkB;oBACpB,CAAC,CAAC,MAAM,CAAC,YAAY;aAC1B;YACD,WAAW;YACX,cAAc,EAAE,mBAAmB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YACxE,aAAa,EAAE,IAAI;SACpB,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAA,+BAAqB,EAAC;YACpB,OAAO,EAAE,+BAA+B;YACxC,IAAI,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE;SAC/B,CAAC,CAAC;QACH,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAa,yBAAyB;IAC7B,cAAc,CAAU;IACxB,kBAAkB,CAAU;IAEnC,4EAA4E;IACpE,gBAAgB,GAAG,KAAK,CAAC;IACjC,4EAA4E;IACpE,SAAS,GAAG,IAAI,OAAO,EAAU,CAAC;IAC1C,2EAA2E;IACnE,WAAW,GAAG,IAAI,OAAO,EAAU,CAAC;IAC5C,+FAA+F;IACvF,qBAAqB,GAA4B,IAAI,CAAC;IAE9D,YAAY,EACV,cAAc,EACd,kBAAkB,GAInB;QACC,IAAI,CAAC,cAAc,GAAG,cAAc,IAAI,SAAS,CAAC;QAClD,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,IAAI,SAAS,CAAC;IAC5D,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,EACb,aAAa,EACb,cAAc,EACd,cAAc,GAKf;QACC,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,OAAO;QACT,CAAC;QAED,MAAM,aAAa,GAAG,IAAA,wBAAgB,EAAC,aAAa,CAAC,CAAC;QACtD,MAAM,YAAY,GAAG,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC;QAE/C,MAAM,cAAc,GAAG,IAAI,GAAG,EAAU,CAAC;QACzC,KAAK,MAAM,QAAQ,IAAI,cAAc,EAAE,CAAC;YACtC,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;gBACnC,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;oBACrC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;oBAC3B,cAAc,CAAC,GAAG,CAAC,IAAA,4BAAoB,EAAC,IAAsB,CAAC,CAAC,CAAC;gBACnE,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,YAAY,EAAE,CAAC;YACjB,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;gBACjC,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;oBACrC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBAC3B,CAAC;YACH,CAAC;YAED,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;YAC7B,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;QACpC,CAAC;QAED,MAAM,cAAc,GAAG,cAAc,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACjE,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,cAAc,EAAE,UAAU,EAAE,CAAC;YACvD,IAAI,CAAC,kBAAkB,GAAG,cAAc,CAAC,UAAU,CAAC;QACtD,CAAC;QAED,IAAI,YAAY,EAAE,CAAC;YACjB,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,CAAC;gBAClC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;gBAC7B,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;oBAC5C,SAAS;gBACX,CAAC;gBACD,MAAM,UAAU,GAAG,IAAA,4BAAoB,EAAC,OAAyB,CAAC,CAAC;gBACnE,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;oBACpE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,gBAAgB,CAAC,aAAwC;QACvD,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;YACxC,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACrC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,aAAa,CAAC,UAAU,EAAE,CAAC;YACrD,IAAI,CAAC,kBAAkB,GAAG,aAAa,CAAC,UAAU,CAAC;QACrD,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,YAAY,CACV,aAAwC,EACxC,cAAyB;QAEzB,MAAM,UAAU,GAAqB,EAAE,CAAC;QAExC,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC3B,MAAM,YAAY,GAAG,IAAA,wBAAgB,EAAC,aAAa,CAAC,CAAC;YACrD,+FAA+F;YAC/F,UAAU,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAC;YACjC,IAAI,CAAC,qBAAqB,GAAG,YAAY,CAAC,MAAM,CAC9C,CAAC,IAAI,EAA0B,EAAE,CAC/B,OAAO,CAAC,IAAI,CAAC,IAAI,OAAO,IAAI,KAAK,QAAQ,CAC5C,CAAC;YACF,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC/B,CAAC;aAAM,IACL,IAAI,CAAC,qBAAqB;YAC1B,IAAI,CAAC,qBAAqB,CAAC,MAAM,GAAG,CAAC,EACrC,CAAC;YACD,0FAA0F;YAC1F,UAAU,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACjD,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,CAAC;YAClC,IAAI,IAAI,CAAC,IAAI,KAAK,oBAAoB,EAAE,CAAC;gBACvC,SAAS;YACX,CAAC;YACD,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;YAC7B,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAC5C,SAAS;YACX,CAAC;YACD,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBACjE,SAAS;YACX,CAAC;YACD,UAAU,CAAC,IAAI,CAAC,OAAyB,CAAC,CAAC;QAC7C,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;;OAGG;IACH,eAAe,CACb,KAAqC,EACrC,OAAsE;QAEtE,MAAM,SAAS,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC5C,MAAM,yBAAyB,GAAG,OAAO,EAAE,aAAa,IAAI,KAAK,CAAC;QAClE,MAAM,uBAAuB,GAC3B,OAAO,EAAE,aAAa,IAAI,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QAE1D,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAEzC,MAAM,YAAY,GAAG,OAAO,EAAE,YAAY,CAAC;QAC3C,IAAI,uBAAuB,IAAI,YAAY,EAAE,CAAC;YAC5C,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;QAClD,CAAC;QAED,IAAI,CAAC,2BAA2B,CAC9B,SAAS,EACT,OAAO,CAAC,yBAAyB,CAAC,CACnC,CAAC;IACJ,CAAC;IAEO,iBAAiB,CACvB,SAA8B,EAC9B,KAAqC;QAErC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC7D,SAAS;YACX,CAAC;YACD,qFAAqF;YACrF,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACpB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IAEO,2BAA2B,CACjC,SAA8B,EAC9B,yBAAkC;QAElC,IACE,CAAC,IAAI,CAAC,qBAAqB;YAC3B,IAAI,CAAC,qBAAqB,CAAC,MAAM,KAAK,CAAC;YACvC,SAAS,CAAC,IAAI,KAAK,CAAC,EACpB,CAAC;YACD,IAAI,yBAAyB,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;gBAC5D,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;YACpC,CAAC;YACD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAC5D,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAC/B,CAAC;QACF,IAAI,IAAI,CAAC,qBAAqB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5C,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;QACpC,CAAC;aAAM,IAAI,yBAAyB,EAAE,CAAC;YACrC,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;QACpC,CAAC;IACH,CAAC;CACF;AAlND,8DAkNC"}

package/dist/runner/conversation.mjs

@@ -0,0 +1,269 @@
+import { UserError } from "../errors.mjs";
+import { addErrorToCurrentSpan } from "../tracing/context.mjs";
+import { buildAgentInputPool, getAgentInputItemKey, removeAgentInputFromPool, takeAgentInputFromPool, toAgentInputList, } from "./items.mjs";
+export { getTurnInput } from "./items.mjs";
+/**
+ * Applies the optional callModelInputFilter and returns the filtered input alongside the original
+ * items so downstream tracking and session persistence stay in sync with what the model saw.
+ */
+export async function applyCallModelInputFilter(agent, callModelInputFilter, context, inputItems, systemInstructions) {
+    const cloneInputItems = (items, map) => items.map((item) => {
+        const cloned = structuredClone(item);
+        if (map && cloned && typeof cloned === 'object') {
+            map.set(cloned, item);
+        }
+        return cloned;
+    });
+    // Record the relationship between the cloned array passed to filters and the original inputs.
+    const cloneMap = new WeakMap();
+    const originalPool = buildAgentInputPool(inputItems);
+    const fallbackOriginals = [];
+    // Track any original object inputs so filtered replacements can still mark them as delivered.
+    for (const item of inputItems) {
+        if (item && typeof item === 'object') {
+            fallbackOriginals.push(item);
+        }
+    }
+    const removeFromFallback = (candidate) => {
+        if (!candidate || typeof candidate !== 'object') {
+            return;
+        }
+        const index = fallbackOriginals.findIndex((original) => original === candidate);
+        if (index !== -1) {
+            fallbackOriginals.splice(index, 1);
+        }
+    };
+    const takeFallbackOriginal = () => {
+        const next = fallbackOriginals.shift();
+        if (next) {
+            removeAgentInputFromPool(originalPool, next);
+        }
+        return next;
+    };
+    // Always create a deep copy so downstream mutations inside filters cannot affect
+    // the cached turn state.
+    const clonedBaseInput = cloneInputItems(inputItems, cloneMap);
+    const base = {
+        input: clonedBaseInput,
+        instructions: systemInstructions,
+    };
+    if (!callModelInputFilter) {
+        return {
+            modelInput: base,
+            sourceItems: [...inputItems],
+            persistedItems: [],
+            filterApplied: false,
+        };
+    }
+    try {
+        const result = await callModelInputFilter({
+            modelData: base,
+            agent,
+            context: context.context,
+        });
+        if (!result || !Array.isArray(result.input)) {
+            throw new UserError('callModelInputFilter must return a ModelInputData object with an input array.');
+        }
+        // Preserve a pointer to the original object backing each filtered clone so downstream
+        // trackers can keep their bookkeeping consistent even after redaction.
+        const sourceItems = result.input.map((item) => {
+            if (!item || typeof item !== 'object') {
+                return undefined;
+            }
+            const original = cloneMap.get(item);
+            if (original) {
+                removeFromFallback(original);
+                removeAgentInputFromPool(originalPool, original);
+                return original;
+            }
+            const key = getAgentInputItemKey(item);
+            const matchedByContent = takeAgentInputFromPool(originalPool, key);
+            if (matchedByContent) {
+                removeFromFallback(matchedByContent);
+                return matchedByContent;
+            }
+            const fallback = takeFallbackOriginal();
+            if (fallback) {
+                return fallback;
+            }
+            return undefined;
+        });
+        const clonedFilteredInput = cloneInputItems(result.input);
+        return {
+            modelInput: {
+                input: clonedFilteredInput,
+                instructions: typeof result.instructions === 'undefined'
+                    ? systemInstructions
+                    : result.instructions,
+            },
+            sourceItems,
+            persistedItems: clonedFilteredInput.map((item) => structuredClone(item)),
+            filterApplied: true,
+        };
+    }
+    catch (error) {
+        addErrorToCurrentSpan({
+            message: 'Error in callModelInputFilter',
+            data: { error: String(error) },
+        });
+        throw error;
+    }
+}
+/**
+ * Tracks which items have already been sent to or received from the Responses API when the caller
+ * supplies `conversationId`/`previousResponseId`. This ensures we only send the delta each turn.
+ */
+export class ServerConversationTracker {
+    conversationId;
+    previousResponseId;
+    // Using this flag because WeakSet does not provide a way to check its size.
+    sentInitialInput = false;
+    // The items already sent to the model; using WeakSet for memory efficiency.
+    sentItems = new WeakSet();
+    // The items received from the server; using WeakSet for memory efficiency.
+    serverItems = new WeakSet();
+    // Track initial input items that have not yet been sent so they can be retried on later turns.
+    remainingInitialInput = null;
+    constructor({ conversationId, previousResponseId, }) {
+        this.conversationId = conversationId ?? undefined;
+        this.previousResponseId = previousResponseId ?? undefined;
+    }
+    /**
+     * Pre-populates tracker caches from an existing RunState when resuming server-managed runs.
+     */
+    primeFromState({ originalInput, generatedItems, modelResponses, }) {
+        if (this.sentInitialInput) {
+            return;
+        }
+        const originalItems = toAgentInputList(originalInput);
+        const hasResponses = modelResponses.length > 0;
+        const serverItemKeys = new Set();
+        for (const response of modelResponses) {
+            for (const item of response.output) {
+                if (item && typeof item === 'object') {
+                    this.serverItems.add(item);
+                    serverItemKeys.add(getAgentInputItemKey(item));
+                }
+            }
+        }
+        if (hasResponses) {
+            for (const item of originalItems) {
+                if (item && typeof item === 'object') {
+                    this.sentItems.add(item);
+                }
+            }
+            this.sentInitialInput = true;
+            this.remainingInitialInput = null;
+        }
+        const latestResponse = modelResponses[modelResponses.length - 1];
+        if (!this.conversationId && latestResponse?.responseId) {
+            this.previousResponseId = latestResponse.responseId;
+        }
+        if (hasResponses) {
+            for (const item of generatedItems) {
+                const rawItem = item.rawItem;
+                if (!rawItem || typeof rawItem !== 'object') {
+                    continue;
+                }
+                const rawItemKey = getAgentInputItemKey(rawItem);
+                if (this.serverItems.has(rawItem) || serverItemKeys.has(rawItemKey)) {
+                    this.sentItems.add(rawItem);
+                }
+            }
+        }
+    }
+    /**
+     * Records the raw items returned by the server so future delta calculations skip them.
+     * Also captures the latest response identifier to chain follow-up calls when possible.
+     */
+    trackServerItems(modelResponse) {
+        if (!modelResponse) {
+            return;
+        }
+        for (const item of modelResponse.output) {
+            if (item && typeof item === 'object') {
+                this.serverItems.add(item);
+            }
+        }
+        if (!this.conversationId && modelResponse.responseId) {
+            this.previousResponseId = modelResponse.responseId;
+        }
+    }
+    /**
+     * Returns the minimum set of items that still need to be delivered to the server for the
+     * current turn. This includes the original turn inputs (until acknowledged) plus any
+     * newly generated items that have not yet been echoed back by the API.
+     */
+    prepareInput(originalInput, generatedItems) {
+        const inputItems = [];
+        if (!this.sentInitialInput) {
+            const initialItems = toAgentInputList(originalInput);
+            // Preserve the full initial payload so a filter can drop items without losing their originals.
+            inputItems.push(...initialItems);
+            this.remainingInitialInput = initialItems.filter((item) => Boolean(item) && typeof item === 'object');
+            this.sentInitialInput = true;
+        }
+        else if (this.remainingInitialInput &&
+            this.remainingInitialInput.length > 0) {
+            // Re-queue prior initial items until the tracker confirms they were delivered to the API.
+            inputItems.push(...this.remainingInitialInput);
+        }
+        for (const item of generatedItems) {
+            if (item.type === 'tool_approval_item') {
+                continue;
+            }
+            const rawItem = item.rawItem;
+            if (!rawItem || typeof rawItem !== 'object') {
+                continue;
+            }
+            if (this.sentItems.has(rawItem) || this.serverItems.has(rawItem)) {
+                continue;
+            }
+            inputItems.push(rawItem);
+        }
+        return inputItems;
+    }
+    /**
+     * Marks the provided originals as delivered so future turns do not resend them and any
+     * pending initial inputs can be dropped once the server acknowledges receipt.
+     */
+    markInputAsSent(items, options) {
+        const delivered = new Set();
+        const dropRemainingInitialInput = options?.filterApplied ?? false;
+        const markFilteredItemsAsSent = options?.filterApplied && Boolean(options.allTurnItems);
+        this.addDeliveredItems(delivered, items);
+        const allTurnItems = options?.allTurnItems;
+        if (markFilteredItemsAsSent && allTurnItems) {
+            this.addDeliveredItems(delivered, allTurnItems);
+        }
+        this.updateRemainingInitialInput(delivered, Boolean(dropRemainingInitialInput));
+    }
+    addDeliveredItems(delivered, items) {
+        for (const item of items) {
+            if (!item || typeof item !== 'object' || delivered.has(item)) {
+                continue;
+            }
+            // Some inputs may be repeated in the filtered list; only mark unique originals once.
+            delivered.add(item);
+            this.sentItems.add(item);
+        }
+    }
+    updateRemainingInitialInput(delivered, dropRemainingInitialInput) {
+        if (!this.remainingInitialInput ||
+            this.remainingInitialInput.length === 0 ||
+            delivered.size === 0) {
+            if (dropRemainingInitialInput && this.remainingInitialInput) {
+                this.remainingInitialInput = null;
+            }
+            return;
+        }
+        this.remainingInitialInput = this.remainingInitialInput.filter((item) => !delivered.has(item));
+        if (this.remainingInitialInput.length === 0) {
+            this.remainingInitialInput = null;
+        }
+        else if (dropRemainingInitialInput) {
+            this.remainingInitialInput = null;
+        }
+    }
+}
+//# sourceMappingURL=conversation.mjs.map

package/dist/runner/conversation.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"conversation.mjs","sourceRoot":"","sources":["../../src/runner/conversation.ts"],"names":[],"mappings":"OACO,EAAE,SAAS,EAAE;OAKb,EAAE,qBAAqB,EAAE;OACzB,EACL,mBAAmB,EACnB,oBAAoB,EACpB,wBAAwB,EACxB,sBAAsB,EACtB,gBAAgB,GACjB;OAEM,EAAE,YAAY,EAAE;AAkCvB;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,KAAuC,EACvC,oBAA2D,EAC3D,OAA6B,EAC7B,UAA4B,EAC5B,kBAAsC;IAEtC,MAAM,eAAe,GAAG,CACtB,KAAuB,EACvB,GAAqC,EACrC,EAAE,CACF,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACjB,MAAM,MAAM,GAAG,eAAe,CAAC,IAAI,CAAmB,CAAC;QACvD,IAAI,GAAG,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;YAChD,GAAG,CAAC,GAAG,CAAC,MAAgB,EAAE,IAAI,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC,CAAC;IAEL,8FAA8F;IAC9F,MAAM,QAAQ,GAAG,IAAI,OAAO,EAA0B,CAAC;IACvD,MAAM,YAAY,GAAG,mBAAmB,CAAC,UAAU,CAAC,CAAC;IACrD,MAAM,iBAAiB,GAAqB,EAAE,CAAC;IAC/C,8FAA8F;IAC9F,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAC9B,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YACrC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IACD,MAAM,kBAAkB,GAAG,CAAC,SAAqC,EAAE,EAAE;QACnE,IAAI,CAAC,SAAS,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;YAChD,OAAO;QACT,CAAC;QACD,MAAM,KAAK,GAAG,iBAAiB,CAAC,SAAS,CACvC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,KAAK,SAAS,CACrC,CAAC;QACF,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;YACjB,iBAAiB,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACrC,CAAC;IACH,CAAC,CAAC;IACF,MAAM,oBAAoB,GAAG,GAA+B,EAAE;QAC5D,MAAM,IAAI,GAAG,iBAAiB,CAAC,KAAK,EAAE,CAAC;QACvC,IAAI,IAAI,EAAE,CAAC;YACT,wBAAwB,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;QAC/C,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC;IAEF,iFAAiF;IACjF,yBAAyB;IACzB,MAAM,eAAe,GAAG,eAAe,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;IAC9D,MAAM,IAAI,GAAG;QACX,KAAK,EAAE,eAAe;QACtB,YAAY,EAAE,kBAAkB;KACjC,CAAC;IACF,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC1B,OAAO;YACL,UAAU,EAAE,IAAI;YAChB,WAAW,EAAE,CAAC,GAAG,UAAU,CAAC;YAC5B,cAAc,EAAE,EAAE;YAClB,aAAa,EAAE,KAAK;SACrB,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,oBAAoB,CAAC;YACxC,SAAS,EAAE,IAAI;YACf,KAAK;YACL,OAAO,EAAE,OAAO,CAAC,OAAO;SACQ,CAAC,CAAC;QAEpC,IAAI,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;YAC5C,MAAM,IAAI,SAAS,CACjB,+EAA+E,CAChF,CAAC;QACJ,CAAC;QAED,sFAAsF;QACtF,uEAAuE;QACvE,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YAC5C,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtC,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAc,CAAC,CAAC;YAC9C,IAAI,QAAQ,EAAE,CAAC;gBACb,kBAAkB,CAAC,QAAQ,CAAC,CAAC;gBAC7B,wBAAwB,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;gBACjD,OAAO,QAAQ,CAAC;YAClB,CAAC;YACD,MAAM,GAAG,GAAG,oBAAoB,CAAC,IAAsB,CAAC,CAAC;YACzD,MAAM,gBAAgB,GAAG,sBAAsB,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;YACnE,IAAI,gBAAgB,EAAE,CAAC;gBACrB,kBAAkB,CAAC,gBAAgB,CAAC,CAAC;gBACrC,OAAO,gBAAgB,CAAC;YAC1B,CAAC;YACD,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;YACxC,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO,QAAQ,CAAC;YAClB,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,MAAM,mBAAmB,GAAG,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1D,OAAO;YACL,UAAU,EAAE;gBACV,KAAK,EAAE,mBAAmB;gBAC1B,YAAY,EACV,OAAO,MAAM,CAAC,YAAY,KAAK,WAAW;oBACxC,CAAC,CAAC,kBAAkB;oBACpB,CAAC,CAAC,MAAM,CAAC,YAAY;aAC1B;YACD,WAAW;YACX,cAAc,EAAE,mBAAmB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;YACxE,aAAa,EAAE,IAAI;SACpB,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,qBAAqB,CAAC;YACpB,OAAO,EAAE,+BAA+B;YACxC,IAAI,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE;SAC/B,CAAC,CAAC;QACH,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,OAAO,yBAAyB;IAC7B,cAAc,CAAU;IACxB,kBAAkB,CAAU;IAEnC,4EAA4E;IACpE,gBAAgB,GAAG,KAAK,CAAC;IACjC,4EAA4E;IACpE,SAAS,GAAG,IAAI,OAAO,EAAU,CAAC;IAC1C,2EAA2E;IACnE,WAAW,GAAG,IAAI,OAAO,EAAU,CAAC;IAC5C,+FAA+F;IACvF,qBAAqB,GAA4B,IAAI,CAAC;IAE9D,YAAY,EACV,cAAc,EACd,kBAAkB,GAInB;QACC,IAAI,CAAC,cAAc,GAAG,cAAc,IAAI,SAAS,CAAC;QAClD,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,IAAI,SAAS,CAAC;IAC5D,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,EACb,aAAa,EACb,cAAc,EACd,cAAc,GAKf;QACC,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC1B,OAAO;QACT,CAAC;QAED,MAAM,aAAa,GAAG,gBAAgB,CAAC,aAAa,CAAC,CAAC;QACtD,MAAM,YAAY,GAAG,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC;QAE/C,MAAM,cAAc,GAAG,IAAI,GAAG,EAAU,CAAC;QACzC,KAAK,MAAM,QAAQ,IAAI,cAAc,EAAE,CAAC;YACtC,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;gBACnC,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;oBACrC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;oBAC3B,cAAc,CAAC,GAAG,CAAC,oBAAoB,CAAC,IAAsB,CAAC,CAAC,CAAC;gBACnE,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,YAAY,EAAE,CAAC;YACjB,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;gBACjC,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;oBACrC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBAC3B,CAAC;YACH,CAAC;YAED,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;YAC7B,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;QACpC,CAAC;QAED,MAAM,cAAc,GAAG,cAAc,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACjE,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,cAAc,EAAE,UAAU,EAAE,CAAC;YACvD,IAAI,CAAC,kBAAkB,GAAG,cAAc,CAAC,UAAU,CAAC;QACtD,CAAC;QAED,IAAI,YAAY,EAAE,CAAC;YACjB,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,CAAC;gBAClC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;gBAC7B,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;oBAC5C,SAAS;gBACX,CAAC;gBACD,MAAM,UAAU,GAAG,oBAAoB,CAAC,OAAyB,CAAC,CAAC;gBACnE,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;oBACpE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC9B,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,gBAAgB,CAAC,aAAwC;QACvD,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;YACxC,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACrC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,aAAa,CAAC,UAAU,EAAE,CAAC;YACrD,IAAI,CAAC,kBAAkB,GAAG,aAAa,CAAC,UAAU,CAAC;QACrD,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,YAAY,CACV,aAAwC,EACxC,cAAyB;QAEzB,MAAM,UAAU,GAAqB,EAAE,CAAC;QAExC,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC3B,MAAM,YAAY,GAAG,gBAAgB,CAAC,aAAa,CAAC,CAAC;YACrD,+FAA+F;YAC/F,UAAU,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAC;YACjC,IAAI,CAAC,qBAAqB,GAAG,YAAY,CAAC,MAAM,CAC9C,CAAC,IAAI,EAA0B,EAAE,CAC/B,OAAO,CAAC,IAAI,CAAC,IAAI,OAAO,IAAI,KAAK,QAAQ,CAC5C,CAAC;YACF,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC/B,CAAC;aAAM,IACL,IAAI,CAAC,qBAAqB;YAC1B,IAAI,CAAC,qBAAqB,CAAC,MAAM,GAAG,CAAC,EACrC,CAAC;YACD,0FAA0F;YAC1F,UAAU,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACjD,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,CAAC;YAClC,IAAI,IAAI,CAAC,IAAI,KAAK,oBAAoB,EAAE,CAAC;gBACvC,SAAS;YACX,CAAC;YACD,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;YAC7B,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAC5C,SAAS;YACX,CAAC;YACD,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBACjE,SAAS;YACX,CAAC;YACD,UAAU,CAAC,IAAI,CAAC,OAAyB,CAAC,CAAC;QAC7C,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;;OAGG;IACH,eAAe,CACb,KAAqC,EACrC,OAAsE;QAEtE,MAAM,SAAS,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC5C,MAAM,yBAAyB,GAAG,OAAO,EAAE,aAAa,IAAI,KAAK,CAAC;QAClE,MAAM,uBAAuB,GAC3B,OAAO,EAAE,aAAa,IAAI,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QAE1D,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAEzC,MAAM,YAAY,GAAG,OAAO,EAAE,YAAY,CAAC;QAC3C,IAAI,uBAAuB,IAAI,YAAY,EAAE,CAAC;YAC5C,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;QAClD,CAAC;QAED,IAAI,CAAC,2BAA2B,CAC9B,SAAS,EACT,OAAO,CAAC,yBAAyB,CAAC,CACnC,CAAC;IACJ,CAAC;IAEO,iBAAiB,CACvB,SAA8B,EAC9B,KAAqC;QAErC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC7D,SAAS;YACX,CAAC;YACD,qFAAqF;YACrF,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACpB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC;IAEO,2BAA2B,CACjC,SAA8B,EAC9B,yBAAkC;QAElC,IACE,CAAC,IAAI,CAAC,qBAAqB;YAC3B,IAAI,CAAC,qBAAqB,CAAC,MAAM,KAAK,CAAC;YACvC,SAAS,CAAC,IAAI,KAAK,CAAC,EACpB,CAAC;YACD,IAAI,yBAAyB,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;gBAC5D,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;YACpC,CAAC;YACD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAC5D,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAC/B,CAAC;QACF,IAAI,IAAI,CAAC,qBAAqB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5C,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;QACpC,CAAC;aAAM,IAAI,yBAAyB,EAAE,CAAC;YACrC,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;QACpC,CAAC;IACH,CAAC;CACF"}

package/dist/runner/guardrails.d.ts

@@ -0,0 +1,23 @@
+import { Agent, AgentOutputType } from '../agent';
+import { InputGuardrailDefinition, InputGuardrailResult, OutputGuardrailDefinition, OutputGuardrailMetadata } from '../guardrail';
+import { RunState } from '../runState';
+export type GuardrailTracker = {
+    readonly pending: boolean;
+    readonly failed: boolean;
+    readonly error: unknown;
+    markPending: () => void;
+    setPromise: (promise?: Promise<InputGuardrailResult[]>) => void;
+    setError: (err: unknown) => void;
+    throwIfError: () => void;
+    awaitCompletion: (options?: {
+        suppressErrors?: boolean;
+    }) => Promise<void>;
+};
+export declare const createGuardrailTracker: () => GuardrailTracker;
+export declare function buildInputGuardrailDefinitions<TContext, TAgent extends Agent<TContext, AgentOutputType>>(state: RunState<TContext, TAgent>, runnerGuardrails: InputGuardrailDefinition[]): InputGuardrailDefinition[];
+export declare function splitInputGuardrails(guardrails: InputGuardrailDefinition[]): {
+    blocking: InputGuardrailDefinition[];
+    parallel: InputGuardrailDefinition[];
+};
+export declare function runInputGuardrails<TContext, TAgent extends Agent<TContext, AgentOutputType>>(state: RunState<TContext, TAgent>, guardrails: InputGuardrailDefinition[]): Promise<InputGuardrailResult[]>;
+export declare function runOutputGuardrails<TContext, TOutput extends AgentOutputType, TAgent extends Agent<TContext, TOutput>>(state: RunState<TContext, TAgent>, runnerOutputGuardrails: OutputGuardrailDefinition<OutputGuardrailMetadata, AgentOutputType<unknown>>[], output: string): Promise<void>;