@cogitator-ai/core 0.21.1 → 0.23.0

package/dist/time-travel/replayer.js

@@ -18,7 +18,6 @@ export class ExecutionReplayer {
     }
     async replayDeterministic(_cogitator, agent, checkpoint, options) {
         const messages = this.buildMessagesForReplay(checkpoint, options.modifiedMessages);
-        const mergedToolResults = this.mergeToolResults(checkpoint.toolResults, options.modifiedToolResults);
         const stepsReplayed = checkpoint.stepIndex + 1;
         const stepsExecuted = 0;
         const divergedAt = undefined;
@@ -41,7 +40,7 @@ export class ExecutionReplayer {
                 duration: Date.now() - startTime,
             },
             toolCalls: checkpoint.pendingToolCalls,
-            messages: this.applyToolResultsToMessages(messages, mergedToolResults),
+            messages: this.applyToolResultsToMessages(messages, checkpoint.toolResults, options.modifiedToolResults ?? {}),
             trace: {
                 traceId,
                 spans,
@@ -57,20 +56,10 @@ export class ExecutionReplayer {
     async replayLive(cogitator, agent, checkpoint, options) {
         const initialMessages = this.buildMessagesForReplay(checkpoint, options.modifiedMessages);
         const input = this.extractUserInput(initialMessages);
-        const modifiedAgent = this.createReplayAgent(agent, initialMessages);
+        const modifiedAgent = this.createReplayAgent(agent, initialMessages, options);
         const runResult = await cogitator.run(modifiedAgent, {
             input,
             threadId: `replay_${checkpoint.runId}`,
-            onToolCall: (toolCall) => {
-                if (options.skipTools?.includes(toolCall.name)) {
-                    return;
-                }
-            },
-            onToolResult: (result) => {
-                if (options.modifiedToolResults?.[result.callId]) {
-                    result.result = options.modifiedToolResults[result.callId];
-                }
-            },
         });
         const stepsReplayed = checkpoint.stepIndex + 1;
         const stepsExecuted = this.countSteps(runResult);
@@ -91,17 +80,21 @@ export class ExecutionReplayer {
         }
         return [...checkpoint.messages];
     }
-    mergeToolResults(cached, modified) {
-        return { ...cached, ...modified };
-    }
-    applyToolResultsToMessages(messages, toolResults) {
-        if (Object.keys(toolResults).length === 0)
-            return messages;
+    /**
+     * Puts tool results into the replayed tool messages: a modified result, keyed
+     * by call id or tool name, wins over the result cached in the checkpoint.
+     */
+    applyToolResultsToMessages(messages, cached, modified) {
+        const pick = (record, keys) => keys.find((key) => key !== undefined && Object.hasOwn(record, key));
         return messages.map((msg) => {
-            if (msg.role === 'tool' && msg.toolCallId && msg.toolCallId in toolResults) {
-                return { ...msg, content: String(toolResults[msg.toolCallId]) };
+            if (msg.role !== 'tool')
+                return msg;
+            const modifiedKey = pick(modified, [msg.toolCallId, msg.name]);
+            if (modifiedKey !== undefined) {
+                return { ...msg, content: toolContent(modified[modifiedKey]) };
             }
-            return msg;
+            const cachedKey = pick(cached, [msg.toolCallId]);
+            return cachedKey === undefined ? msg : { ...msg, content: toolContent(cached[cachedKey]) };
         });
     }
     extractUserInput(messages) {
@@ -109,7 +102,12 @@ export class ExecutionReplayer {
         const lastUserMessage = userMessages[userMessages.length - 1];
         return lastUserMessage ? this.getTextContent(lastUserMessage.content) : '';
     }
-    createReplayAgent(agent, preloadedMessages) {
+    /**
+     * The agent a live replay runs: the original with the checkpoint's history in
+     * its instructions, tools in `skipTools` removed, and tools named in
+     * `modifiedToolResults` answering with the given value instead of running.
+     */
+    createReplayAgent(agent, preloadedMessages, options) {
         const systemMessage = preloadedMessages.find((m) => m.role === 'system');
         const contextFromHistory = preloadedMessages
             .filter((m) => m.role === 'assistant' || m.role === 'tool')
@@ -122,6 +120,7 @@ export class ExecutionReplayer {
             ...agent.config,
             name: `${agent.name}_replay`,
             instructions: newInstructions,
+            tools: replayTools(agent.tools, options),
         });
     }
     countSteps(result) {
@@ -165,4 +164,22 @@ export class ExecutionReplayer {
             .join(' ');
     }
 }
+function replayTools(tools, options) {
+    const skipped = new Set(options.skipTools ?? []);
+    const mocks = options.modifiedToolResults ?? {};
+    return tools
+        .filter((tool) => !skipped.has(tool.name))
+        .map((tool) => (Object.hasOwn(mocks, tool.name) ? mockedTool(tool, mocks[tool.name]) : tool));
+}
+function mockedTool(tool, result) {
+    const mocked = Object.create(Object.getPrototypeOf(tool));
+    return Object.assign(mocked, tool, {
+        execute: async () => result,
+        sandbox: undefined,
+        requiresApproval: undefined,
+    });
+}
+function toolContent(value) {
+    return typeof value === 'string' ? value : JSON.stringify(value);
+}
 //# sourceMappingURL=replayer.js.map

package/dist/time-travel/replayer.js.map

@@ -1 +1 @@
-{"version":3,"file":"replayer.js","sourceRoot":"","sources":["../../src/time-travel/replayer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAkBhC,MAAM,OAAO,iBAAiB;IACpB,eAAe,CAA4B;IAEnD,YAAY,OAAiC;QAC3C,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAoB,EAAE,KAAY,EAAE,OAAsB;QACrE,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QAC3E,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,yBAAyB,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;QACrE,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;YACrC,OAAO,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QACzE,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC/B,UAAqB,EACrB,KAAY,EACZ,UAA+B,EAC/B,OAAsB;QAEtB,MAAM,QAAQ,GAAG,IAAI,CAAC,sBAAsB,CAAC,UAAU,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;QACnF,MAAM,iBAAiB,GAAG,IAAI,CAAC,gBAAgB,CAC7C,UAAU,CAAC,WAAW,EACtB,OAAO,CAAC,mBAAmB,CAC5B,CAAC;QAEF,MAAM,aAAa,GAAG,UAAU,CAAC,SAAS,GAAG,CAAC,CAAC;QAC/C,MAAM,aAAa,GAAG,CAAC,CAAC;QACxB,MAAM,UAAU,GAAuB,SAAS,CAAC;QAEjD,MAAM,KAAK,GAAG,UAAU,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC;QACrC,MAAM,OAAO,GAAG,SAAS,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC;QACtC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,KAAK,GAAW,EAAE,CAAC;QAEzB,MAAM,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,GAAG,EAAE,CAAC;QAC3E,MAAM,MAAM,GAAG,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAE/E,MAAM,MAAM,GAAiB;YAC3B,MAAM;YACN,KAAK;YACL,OAAO,EAAE,KAAK,CAAC,EAAE;YACjB,QAAQ,EAAE,UAAU,CAAC,KAAK;YAC1B,KAAK,EAAE;gBACL,WAAW,EAAE,CAAC;gBACd,YAAY,EAAE,CAAC;gBACf,WAAW,EAAE,CAAC;gBACd,IAAI,EAAE,CAAC;gBACP,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aACjC;YACD,SAAS,EAAE,UAAU,CAAC,gBAAgB;YACtC,QAAQ,EAAE,IAAI,CAAC,0BAA0B,CAAC,QAAQ,EAAE,iBAAiB,CAAC;YACtE,KAAK,EAAE;gBACL,OAAO;gBACP,KAAK;aACN;YACD,YAAY,EAAE,UAAU,CAAC,EAAE;YAC3B,eAAe,EAAE,UAAU,CAAC,OAAO;YACnC,UAAU;YACV,aAAa;YACb,aAAa;SACd,CAAC;QAEF,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,UAAU,CACtB,SAAoB,EACpB,KAAY,EACZ,UAA+B,EAC/B,OAAsB;QAEtB,MAAM,eAAe,GAAG,IAAI,CAAC,sBAAsB,CAAC,UAAU,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;QAE1F,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;QAErD,MAAM,aAAa,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;QAErE,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,aAAa,EAAE;YACnD,KAAK;YACL,QAAQ,EAAE,UAAU,UAAU,CAAC,KAAK,EAAE;YACtC,UAAU,EAAE,CAAC,QAAQ,EAAE,EAAE;gBACvB,IAAI,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC/C,OAAO;gBACT,CAAC;YACH,CAAC;YACD,YAAY,EAAE,CAAC,MAAM,EAAE,EAAE;gBACvB,IAAI,OAAO,CAAC,mBAAmB,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;oBAChD,MAA8B,CAAC,MAAM,GAAG,OAAO,CAAC,mBAAmB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACtF,CAAC;YACH,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,UAAU,CAAC,SAAS,GAAG,CAAC,CAAC;QAC/C,MAAM,aAAa,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,UAAU,GAAG,IAAI,CAAC,mBAAmB,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;QAEnE,MAAM,YAAY,GAAiB;YACjC,GAAG,SAAS;YACZ,YAAY,EAAE,UAAU,CAAC,EAAE;YAC3B,eAAe,EAAE,UAAU,CAAC,OAAO;YACnC,UAAU;YACV,aAAa;YACb,aAAa;SACd,CAAC;QAEF,OAAO,YAAY,CAAC;IACtB,CAAC;IAEO,sBAAsB,CAC5B,UAA+B,EAC/B,aAAyB;QAEzB,IAAI,aAAa,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9C,OAAO,aAAa,CAAC;QACvB,CAAC;QACD,OAAO,CAAC,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;IAClC,CAAC;IAEO,gBAAgB,CACtB,MAA+B,EAC/B,QAAkC;QAElC,OAAO,EAAE,GAAG,MAAM,EAAE,GAAG,QAAQ,EAAE,CAAC;IACpC,CAAC;IAEO,0BAA0B,CAChC,QAAmB,EACnB,WAAoC;QAEpC,IAAI,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,QAAQ,CAAC;QAE3D,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YAC1B,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,UAAU,IAAI,WAAW,EAAE,CAAC;gBAC3E,OAAO,EAAE,GAAG,GAAG,EAAE,OAAO,EAAE,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC;YAClE,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,gBAAgB,CAAC,QAAmB;QAC1C,MAAM,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QAC/D,MAAM,eAAe,GAAG,YAAY,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC9D,OAAO,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC7E,CAAC;IAEO,iBAAiB,CAAC,KAAY,EAAE,iBAA4B;QAClE,MAAM,aAAa,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QACzE,MAAM,kBAAkB,GAAG,iBAAiB;aACzC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;aAC1D,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,MAAM,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;aAC5D,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,MAAM,eAAe,GAAG,aAAa;YACnC,CAAC,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,OAAO,CAAC,qEAAqE,kBAAkB,EAAE;YACxI,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC;QAEvB,OAAO,IAAK,KAAK,CAAC,WAA4B,CAAC;YAC7C,GAAG,KAAK,CAAC,MAAM;YACf,IAAI,EAAE,GAAG,KAAK,CAAC,IAAI,SAAS;YAC5B,YAAY,EAAE,eAAe;SAC9B,CAAC,CAAC;IACL,CAAC;IAEO,UAAU,CAAC,MAAiB;QAClC,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YACtC,IACE,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;gBAC7B,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACzB,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAC1B,CAAC;gBACD,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,mBAAmB,CACzB,UAA+B,EAC/B,MAAiB;QAEjB,MAAM,iBAAiB,GAAG,UAAU,CAAC,gBAAgB,CAAC;QACtD,MAAM,YAAY,GAAG,MAAM,CAAC,SAAS,CAAC;QAEtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,iBAAiB,CAAC,MAAM,EAAE,YAAY,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACjF,MAAM,IAAI,GAAG,iBAAiB,CAAC,CAAC,CAAC,CAAC;YAClC,MAAM,IAAI,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;YAE7B,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC5B,OAAO,UAAU,CAAC,SAAS,GAAG,CAAC,GAAG,CAAC,CAAC;YACtC,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAChD,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBAC1B,OAAO,UAAU,CAAC,SAAS,GAAG,CAAC,GAAG,CAAC,CAAC;YACtC,CAAC;QACH,CAAC;QAED,IAAI,iBAAiB,CAAC,MAAM,KAAK,YAAY,CAAC,MAAM,EAAE,CAAC;YACrD,OAAO,UAAU,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,iBAAiB,CAAC,MAAM,EAAE,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC5F,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,cAAc,CAAC,OAAuB;QAC5C,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAChC,OAAO,OAAO,CAAC;QACjB,CAAC;QACD,OAAO,OAAO;aACX,MAAM,CAAC,CAAC,IAAI,EAA0C,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;aAC9E,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACxB,IAAI,CAAC,GAAG,CAAC,CAAC;IACf,CAAC;CACF"}
+{"version":3,"file":"replayer.js","sourceRoot":"","sources":["../../src/time-travel/replayer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAmBhC,MAAM,OAAO,iBAAiB;IACpB,eAAe,CAA4B;IAEnD,YAAY,OAAiC;QAC3C,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAoB,EAAE,KAAY,EAAE,OAAsB;QACrE,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QAC3E,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,yBAAyB,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;QACrE,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;YACrC,OAAO,IAAI,CAAC,mBAAmB,CAAC,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QACzE,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC/B,UAAqB,EACrB,KAAY,EACZ,UAA+B,EAC/B,OAAsB;QAEtB,MAAM,QAAQ,GAAG,IAAI,CAAC,sBAAsB,CAAC,UAAU,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;QAEnF,MAAM,aAAa,GAAG,UAAU,CAAC,SAAS,GAAG,CAAC,CAAC;QAC/C,MAAM,aAAa,GAAG,CAAC,CAAC;QACxB,MAAM,UAAU,GAAuB,SAAS,CAAC;QAEjD,MAAM,KAAK,GAAG,UAAU,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC;QACrC,MAAM,OAAO,GAAG,SAAS,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC;QACtC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,KAAK,GAAW,EAAE,CAAC;QAEzB,MAAM,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,GAAG,EAAE,CAAC;QAC3E,MAAM,MAAM,GAAG,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAE/E,MAAM,MAAM,GAAiB;YAC3B,MAAM;YACN,KAAK;YACL,OAAO,EAAE,KAAK,CAAC,EAAE;YACjB,QAAQ,EAAE,UAAU,CAAC,KAAK;YAC1B,KAAK,EAAE;gBACL,WAAW,EAAE,CAAC;gBACd,YAAY,EAAE,CAAC;gBACf,WAAW,EAAE,CAAC;gBACd,IAAI,EAAE,CAAC;gBACP,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aACjC;YACD,SAAS,EAAE,UAAU,CAAC,gBAAgB;YACtC,QAAQ,EAAE,IAAI,CAAC,0BAA0B,CACvC,QAAQ,EACR,UAAU,CAAC,WAAW,EACtB,OAAO,CAAC,mBAAmB,IAAI,EAAE,CAClC;YACD,KAAK,EAAE;gBACL,OAAO;gBACP,KAAK;aACN;YACD,YAAY,EAAE,UAAU,CAAC,EAAE;YAC3B,eAAe,EAAE,UAAU,CAAC,OAAO;YACnC,UAAU;YACV,aAAa;YACb,aAAa;SACd,CAAC;QAEF,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,KAAK,CAAC,UAAU,CACtB,SAAoB,EACpB,KAAY,EACZ,UAA+B,EAC/B,OAAsB;QAEtB,MAAM,eAAe,GAAG,IAAI,CAAC,sBAAsB,CAAC,UAAU,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;QAE1F,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;QAErD,MAAM,aAAa,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;QAE9E,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,aAAa,EAAE;YACnD,KAAK;YACL,QAAQ,EAAE,UAAU,UAAU,CAAC,KAAK,EAAE;SACvC,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,UAAU,CAAC,SAAS,GAAG,CAAC,CAAC;QAC/C,MAAM,aAAa,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACjD,MAAM,UAAU,GAAG,IAAI,CAAC,mBAAmB,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;QAEnE,MAAM,YAAY,GAAiB;YACjC,GAAG,SAAS;YACZ,YAAY,EAAE,UAAU,CAAC,EAAE;YAC3B,eAAe,EAAE,UAAU,CAAC,OAAO;YACnC,UAAU;YACV,aAAa;YACb,aAAa;SACd,CAAC;QAEF,OAAO,YAAY,CAAC;IACtB,CAAC;IAEO,sBAAsB,CAC5B,UAA+B,EAC/B,aAAyB;QAEzB,IAAI,aAAa,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9C,OAAO,aAAa,CAAC;QACvB,CAAC;QACD,OAAO,CAAC,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;IAClC,CAAC;IAED;;;OAGG;IACK,0BAA0B,CAChC,QAAmB,EACnB,MAA+B,EAC/B,QAAiC;QAEjC,MAAM,IAAI,GAAG,CAAC,MAA+B,EAAE,IAA+B,EAAE,EAAE,CAChF,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,EAAiB,EAAE,CAAC,GAAG,KAAK,SAAS,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC;QAErF,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YAC1B,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM;gBAAE,OAAO,GAAG,CAAC;YACpC,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;YAC/D,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;gBAC9B,OAAO,EAAE,GAAG,GAAG,EAAE,OAAO,EAAE,WAAW,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC;YACjE,CAAC;YACD,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;YACjD,OAAO,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,GAAG,EAAE,OAAO,EAAE,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC;QAC7F,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,gBAAgB,CAAC,QAAmB;QAC1C,MAAM,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QAC/D,MAAM,eAAe,GAAG,YAAY,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC9D,OAAO,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC7E,CAAC;IAED;;;;OAIG;IACK,iBAAiB,CACvB,KAAY,EACZ,iBAA4B,EAC5B,OAAsB;QAEtB,MAAM,aAAa,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QACzE,MAAM,kBAAkB,GAAG,iBAAiB;aACzC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;aAC1D,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,MAAM,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;aAC5D,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,MAAM,eAAe,GAAG,aAAa;YACnC,CAAC,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,OAAO,CAAC,qEAAqE,kBAAkB,EAAE;YACxI,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC;QAEvB,OAAO,IAAK,KAAK,CAAC,WAA4B,CAAC;YAC7C,GAAG,KAAK,CAAC,MAAM;YACf,IAAI,EAAE,GAAG,KAAK,CAAC,IAAI,SAAS;YAC5B,YAAY,EAAE,eAAe;YAC7B,KAAK,EAAE,WAAW,CAAC,KAAK,CAAC,KAAK,EAAE,OAAO,CAAC;SACzC,CAAC,CAAC;IACL,CAAC;IAEO,UAAU,CAAC,MAAiB;QAClC,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YACtC,IACE,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;gBAC7B,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACzB,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAC1B,CAAC;gBACD,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,mBAAmB,CACzB,UAA+B,EAC/B,MAAiB;QAEjB,MAAM,iBAAiB,GAAG,UAAU,CAAC,gBAAgB,CAAC;QACtD,MAAM,YAAY,GAAG,MAAM,CAAC,SAAS,CAAC;QAEtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,iBAAiB,CAAC,MAAM,EAAE,YAAY,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACjF,MAAM,IAAI,GAAG,iBAAiB,CAAC,CAAC,CAAC,CAAC;YAClC,MAAM,IAAI,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;YAE7B,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC5B,OAAO,UAAU,CAAC,SAAS,GAAG,CAAC,GAAG,CAAC,CAAC;YACtC,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAChD,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBAC1B,OAAO,UAAU,CAAC,SAAS,GAAG,CAAC,GAAG,CAAC,CAAC;YACtC,CAAC;QACH,CAAC;QAED,IAAI,iBAAiB,CAAC,MAAM,KAAK,YAAY,CAAC,MAAM,EAAE,CAAC;YACrD,OAAO,UAAU,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,iBAAiB,CAAC,MAAM,EAAE,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC5F,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,cAAc,CAAC,OAAuB;QAC5C,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAChC,OAAO,OAAO,CAAC;QACjB,CAAC;QACD,OAAO,OAAO;aACX,MAAM,CAAC,CAAC,IAAI,EAA0C,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;aAC9E,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACxB,IAAI,CAAC,GAAG,CAAC,CAAC;IACf,CAAC;CACF;AAED,SAAS,WAAW,CAAC,KAAsB,EAAE,OAAsB;IACjE,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC;IACjD,MAAM,KAAK,GAAG,OAAO,CAAC,mBAAmB,IAAI,EAAE,CAAC;IAChD,OAAO,KAAK;SACT,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;SACzC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;AAClG,CAAC;AAED,SAAS,UAAU,CAAC,IAAU,EAAE,MAAe;IAC7C,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,IAAI,CAAkB,CAAS,CAAC;IACnF,OAAO,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,EAAE;QACjC,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM;QAC3B,OAAO,EAAE,SAAS;QAClB,gBAAgB,EAAE,SAAS;KAC5B,CAAC,CAAC;AACL,CAAC;AAED,SAAS,WAAW,CAAC,KAAc;IACjC,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;AACnE,CAAC"}

package/dist/time-travel/time-travel.d.ts

@@ -36,6 +36,19 @@ export declare class TimeTravel {
     getCheckpointStore(): TimeTravelCheckpointStore;
     getTraceStore(): TraceStore;
     getConfig(): TimeTravelConfig;
+    /**
+     * Stores a run's trace under its trace id, so `compare()` can read it. The
+     * trace of an original run is stored once; a replay's is always new.
+     */
+    private recordTrace;
+    /**
+     * A deterministic replay runs no model or tool calls, so its trace is the
+     * original's up to the replayed step, ending in the replayed output.
+     */
+    private recordDeterministicTrace;
+    /** Deletes expired checkpoints of the run's agent and the oldest beyond the per-trace cap. */
+    private pruneCheckpoints;
+    private recordFork;
     private countSteps;
 }
 //# sourceMappingURL=time-travel.d.ts.map

package/dist/time-travel/time-travel.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"time-travel.d.ts","sourceRoot":"","sources":["../../src/time-travel/time-travel.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,mBAAmB,EACnB,aAAa,EACb,YAAY,EACZ,WAAW,EACX,UAAU,EACV,SAAS,EACT,UAAU,EACV,yBAAyB,EACzB,SAAS,EACT,gBAAgB,EACjB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,UAAU,CAAC;AACtC,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAO5C,MAAM,WAAW,iBAAiB;IAChC,eAAe,CAAC,EAAE,yBAAyB,CAAC;IAC5C,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,MAAM,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;CACpC;AAED,qBAAa,UAAU;IACrB,OAAO,CAAC,SAAS,CAAY;IAC7B,OAAO,CAAC,eAAe,CAA4B;IACnD,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,QAAQ,CAAoB;IACpC,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,UAAU,CAAkB;IACpC,OAAO,CAAC,MAAM,CAAmB;gBAErB,SAAS,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,iBAAiB;IAqBvD,UAAU,CACd,MAAM,EAAE,SAAS,EACjB,SAAS,EAAE,MAAM,EACjB,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,mBAAmB,CAAC;IAMzB,aAAa,CAAC,MAAM,EAAE,SAAS,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAItF,eAAe,CACnB,MAAM,EAAE,SAAS,EACjB,QAAQ,EAAE,MAAM,EAChB,WAAW,CAAC,EAAE,MAAM,GACnB,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAmB3B,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAI/D,aAAa,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IAIxE,gBAAgB,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAIxD,MAAM,CACV,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,OAAO,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,GAC/B,OAAO,CAAC,YAAY,CAAC;IAQlB,mBAAmB,CAAC,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAI9E,UAAU,CACd,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,OAAO,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,MAAM,CAAC,GAC7C,OAAO,CAAC,YAAY,CAAC;IAIlB,IAAI,CACR,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,OAAO,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GAC7B,OAAO,CAAC,UAAU,CAAC;IAOhB,eAAe,CACnB,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,iBAAiB,EAAE,MAAM,EACzB,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,UAAU,CAAC;IAUhB,kBAAkB,CACtB,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,OAAO,EACnB,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,UAAU,CAAC;IAUhB,mBAAmB,CACvB,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACpC,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,UAAU,CAAC;IAIhB,gBAAgB,CACpB,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,MAAM,EAChB,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,UAAU,CAAC;IAIhB,YAAY,CAChB,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,GACpC,OAAO,CAAC,UAAU,EAAE,CAAC;IAIlB,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC;IAI/D,mBAAmB,CAAC,YAAY,EAAE,YAAY,GAAG,OAAO,CAAC,SAAS,CAAC;IAIzE,UAAU,CAAC,IAAI,EAAE,SAAS,GAAG,MAAM;IAInC,kBAAkB,IAAI,yBAAyB;IAI/C,aAAa,IAAI,UAAU;IAI3B,SAAS,IAAI,gBAAgB;IAI7B,OAAO,CAAC,UAAU;CAanB"}
+{"version":3,"file":"time-travel.d.ts","sourceRoot":"","sources":["../../src/time-travel/time-travel.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,mBAAmB,EACnB,aAAa,EACb,YAAY,EACZ,WAAW,EACX,UAAU,EACV,SAAS,EACT,UAAU,EACV,yBAAyB,EACzB,SAAS,EACT,gBAAgB,EACjB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,UAAU,CAAC;AACtC,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAQ5C,MAAM,WAAW,iBAAiB;IAChC,eAAe,CAAC,EAAE,yBAAyB,CAAC;IAC5C,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,MAAM,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;CACpC;AAED,qBAAa,UAAU;IACrB,OAAO,CAAC,SAAS,CAAY;IAC7B,OAAO,CAAC,eAAe,CAA4B;IACnD,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,QAAQ,CAAoB;IACpC,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,UAAU,CAAkB;IACpC,OAAO,CAAC,MAAM,CAAmB;gBAErB,SAAS,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,iBAAiB;IAqBvD,UAAU,CACd,MAAM,EAAE,SAAS,EACjB,SAAS,EAAE,MAAM,EACjB,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,mBAAmB,CAAC;IAQzB,aAAa,CAAC,MAAM,EAAE,SAAS,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAOtF,eAAe,CACnB,MAAM,EAAE,SAAS,EACjB,QAAQ,EAAE,MAAM,EAChB,WAAW,CAAC,EAAE,MAAM,GACnB,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAqB3B,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAI/D,aAAa,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,GAAG,IAAI,CAAC;IAIxE,gBAAgB,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAIxD,MAAM,CACV,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,OAAO,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,GAC/B,OAAO,CAAC,YAAY,CAAC;IAclB,mBAAmB,CAAC,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAI9E,UAAU,CACd,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,OAAO,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,MAAM,CAAC,GAC7C,OAAO,CAAC,YAAY,CAAC;IAIlB,IAAI,CACR,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,OAAO,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GAC7B,OAAO,CAAC,UAAU,CAAC;IAShB,eAAe,CACnB,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,iBAAiB,EAAE,MAAM,EACzB,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,UAAU,CAAC;IAYhB,kBAAkB,CACtB,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,OAAO,EACnB,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,UAAU,CAAC;IAYhB,mBAAmB,CACvB,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACpC,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,UAAU,CAAC;IAMhB,gBAAgB,CACpB,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,MAAM,EAChB,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,UAAU,CAAC;IAMhB,YAAY,CAChB,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,GACpC,OAAO,CAAC,UAAU,EAAE,CAAC;IAMlB,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC;IAI/D,mBAAmB,CAAC,YAAY,EAAE,YAAY,GAAG,OAAO,CAAC,SAAS,CAAC;IAIzE,UAAU,CAAC,IAAI,EAAE,SAAS,GAAG,MAAM;IAInC,kBAAkB,IAAI,yBAAyB;IAI/C,aAAa,IAAI,UAAU;IAI3B,SAAS,IAAI,gBAAgB;IAI7B;;;OAGG;YACW,WAAW;IAMzB;;;OAGG;YACW,wBAAwB;IAiBtC,8FAA8F;YAChF,gBAAgB;YAsBhB,UAAU;IAKxB,OAAO,CAAC,UAAU;CAanB"}

package/dist/time-travel/time-travel.js

@@ -4,6 +4,7 @@ import { ExecutionReplayer } from './replayer.js';
 import { ExecutionForker } from './forker.js';
 import { TraceComparator } from './comparator.js';
 import { InMemoryTraceStore } from '../learning/trace-store.js';
+import { buildExecutionTrace, runInput } from '../learning/trace-builder.js';
 export class TimeTravel {
     cogitator;
     checkpointStore;
@@ -31,15 +32,21 @@ export class TimeTravel {
     async checkpoint(result, stepIndex, label) {
         const checkpoint = this.checkpointStore.createFromRunResult(result, stepIndex, { label });
         await this.checkpointStore.save(checkpoint);
+        await this.recordTrace(result);
+        await this.pruneCheckpoints(result);
         return checkpoint;
     }
     async checkpointAll(result, labelPrefix) {
-        return this.checkpointStore.createAllFromRunResult(result, { labelPrefix });
+        await this.recordTrace(result);
+        const checkpoints = await this.checkpointStore.createAllFromRunResult(result, { labelPrefix });
+        await this.pruneCheckpoints(result);
+        return checkpoints;
     }
     async checkpointEvery(result, interval, labelPrefix) {
         if (!Number.isInteger(interval) || interval <= 0) {
             throw new Error('Checkpoint interval must be a positive integer');
         }
+        await this.recordTrace(result);
         const stepCount = this.countSteps(result);
         const checkpoints = [];
         for (let i = 0; i < stepCount; i += interval) {
@@ -49,6 +56,7 @@ export class TimeTravel {
             await this.checkpointStore.save(checkpoint);
             checkpoints.push(checkpoint);
         }
+        await this.pruneCheckpoints(result);
         return checkpoints;
     }
     async getCheckpoints(traceId) {
@@ -61,11 +69,18 @@ export class TimeTravel {
         return this.checkpointStore.delete(checkpointId);
     }
     async replay(agent, checkpointId, options) {
-        return this.replayer.replay(this.cogitator, agent, {
+        const result = await this.replayer.replay(this.cogitator, agent, {
             fromCheckpoint: checkpointId,
             mode: options?.mode ?? 'live',
             ...options,
         });
+        if (options?.mode === 'deterministic') {
+            await this.recordDeterministicTrace(result);
+        }
+        else {
+            await this.recordTrace(result, true);
+        }
+        return result;
     }
     async replayDeterministic(agent, checkpointId) {
         return this.replay(agent, checkpointId, { mode: 'deterministic' });
@@ -74,25 +89,28 @@ export class TimeTravel {
         return this.replay(agent, checkpointId, { mode: 'live', ...options });
     }
     async fork(agent, checkpointId, options) {
-        return this.forker.fork(this.cogitator, agent, {
+        return this.recordFork(await this.forker.fork(this.cogitator, agent, {
             checkpointId,
             ...options,
-        });
+        }));
     }
     async forkWithContext(agent, checkpointId, additionalContext, label) {
-        return this.forker.forkWithContext(this.cogitator, agent, checkpointId, additionalContext, label);
+        return this.recordFork(await this.forker.forkWithContext(this.cogitator, agent, checkpointId, additionalContext, label));
     }
     async forkWithMockedTool(agent, checkpointId, toolName, mockResult, label) {
-        return this.forker.forkWithMockedTools(this.cogitator, agent, checkpointId, { [toolName]: mockResult }, label);
+        return this.recordFork(await this.forker.forkWithMockedTools(this.cogitator, agent, checkpointId, { [toolName]: mockResult }, label));
     }
     async forkWithMockedTools(agent, checkpointId, mockResults, label) {
-        return this.forker.forkWithMockedTools(this.cogitator, agent, checkpointId, mockResults, label);
+        return this.recordFork(await this.forker.forkWithMockedTools(this.cogitator, agent, checkpointId, mockResults, label));
     }
     async forkWithNewInput(agent, checkpointId, newInput, label) {
-        return this.forker.forkWithNewInput(this.cogitator, agent, checkpointId, newInput, label);
+        return this.recordFork(await this.forker.forkWithNewInput(this.cogitator, agent, checkpointId, newInput, label));
     }
     async forkMultiple(agent, checkpointId, variants) {
-        return this.forker.forkMultiple(this.cogitator, agent, checkpointId, variants);
+        const forks = await this.forker.forkMultiple(this.cogitator, agent, checkpointId, variants);
+        for (const fork of forks)
+            await this.recordFork(fork);
+        return forks;
     }
     async compare(traceId1, traceId2) {
         return this.comparator.compare(traceId1, traceId2);
@@ -112,6 +130,58 @@ export class TimeTravel {
     getConfig() {
         return { ...this.config };
     }
+    /**
+     * Stores a run's trace under its trace id, so `compare()` can read it. The
+     * trace of an original run is stored once; a replay's is always new.
+     */
+    async recordTrace(result, replay = false) {
+        const id = result.trace.traceId;
+        if (!replay && (await this.traceStore.get(id)))
+            return;
+        await this.traceStore.store(buildExecutionTrace(result, runInput(result), { id }));
+    }
+    /**
+     * A deterministic replay runs no model or tool calls, so its trace is the
+     * original's up to the replayed step, ending in the replayed output.
+     */
+    async recordDeterministicTrace(result) {
+        const original = await this.traceStore.get(result.originalTraceId);
+        if (!original) {
+            await this.recordTrace(result, true);
+            return;
+        }
+        await this.traceStore.store({
+            ...original,
+            id: result.trace.traceId,
+            runId: result.runId,
+            output: result.output,
+            steps: original.steps.slice(0, result.stepsReplayed),
+            createdAt: new Date(),
+            isDemo: false,
+        });
+    }
+    /** Deletes expired checkpoints of the run's agent and the oldest beyond the per-trace cap. */
+    async pruneCheckpoints(result) {
+        const { maxCheckpointsPerTrace, checkpointRetention } = this.config;
+        if (checkpointRetention !== undefined && checkpointRetention > 0) {
+            const expiredBefore = Date.now() - checkpointRetention;
+            for (const checkpoint of await this.checkpointStore.getByAgent(result.agentId)) {
+                if (checkpoint.createdAt.getTime() < expiredBefore) {
+                    await this.checkpointStore.delete(checkpoint.id);
+                }
+            }
+        }
+        if (maxCheckpointsPerTrace !== undefined && maxCheckpointsPerTrace > 0) {
+            const forTrace = (await this.checkpointStore.getByTrace(result.trace.traceId)).sort((a, b) => a.createdAt.getTime() - b.createdAt.getTime());
+            for (const checkpoint of forTrace.slice(0, forTrace.length - maxCheckpointsPerTrace)) {
+                await this.checkpointStore.delete(checkpoint.id);
+            }
+        }
+    }
+    async recordFork(fork) {
+        await this.recordTrace(fork.result, true);
+        return fork;
+    }
     countSteps(result) {
         let count = 0;
         for (const span of result.trace.spans) {

package/dist/time-travel/time-travel.js.map

@@ -1 +1 @@
-{"version":3,"file":"time-travel.js","sourceRoot":"","sources":["../../src/time-travel/time-travel.ts"],"names":[],"mappings":"AAYA,OAAO,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AAGjE,OAAO,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAC/C,OAAO,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAC3C,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAQ7D,MAAM,OAAO,UAAU;IACb,SAAS,CAAY;IACrB,eAAe,CAA4B;IAC3C,UAAU,CAAa;IACvB,QAAQ,CAAoB;IAC5B,MAAM,CAAkB;IACxB,UAAU,CAAkB;IAC5B,MAAM,CAAmB;IAEjC,YAAY,SAAoB,EAAE,OAA2B;QAC3D,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,0BAA0B,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,CAAC;QAEpE,IAAI,CAAC,eAAe,GAAG,OAAO,EAAE,eAAe,IAAI,IAAI,uBAAuB,EAAE,CAAC;QACjF,IAAI,CAAC,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,IAAI,kBAAkB,EAAE,CAAC;QAElE,IAAI,CAAC,QAAQ,GAAG,IAAI,iBAAiB,CAAC;YACpC,eAAe,EAAE,IAAI,CAAC,eAAe;SACtC,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,GAAG,IAAI,eAAe,CAAC;YAChC,eAAe,EAAE,IAAI,CAAC,eAAe;YACrC,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC,CAAC;QAEH,IAAI,CAAC,UAAU,GAAG,IAAI,eAAe,CAAC;YACpC,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,UAAU,CACd,MAAiB,EACjB,SAAiB,EACjB,KAAc;QAEd,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC1F,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC5C,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAiB,EAAE,WAAoB;QACzD,OAAO,IAAI,CAAC,eAAe,CAAC,sBAAsB,CAAC,MAAM,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;IAC9E,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,MAAiB,EACjB,QAAgB,EAChB,WAAoB;QAEpB,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,QAAQ,IAAI,CAAC,EAAE,CAAC;YACjD,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;QACpE,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAC1C,MAAM,WAAW,GAA0B,EAAE,CAAC;QAE9C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,IAAI,QAAQ,EAAE,CAAC;YAC7C,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,MAAM,EAAE,CAAC,EAAE;gBACrE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,GAAG,WAAW,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;aAC5D,CAAC,CAAC;YACH,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC5C,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC/B,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,OAAe;QAClC,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,YAAoB;QACtC,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,YAAoB;QACzC,OAAO,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,MAAM,CACV,KAAY,EACZ,YAAoB,EACpB,OAAgC;QAEhC,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE;YACjD,cAAc,EAAE,YAAY;YAC5B,IAAI,EAAE,OAAO,EAAE,IAAI,IAAI,MAAM;YAC7B,GAAG,OAAO;SACX,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,KAAY,EAAE,YAAoB;QAC1D,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,CAAC;IACrE,CAAC;IAED,KAAK,CAAC,UAAU,CACd,KAAY,EACZ,YAAoB,EACpB,OAA8C;QAE9C,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,IAAI,CACR,KAAY,EACZ,YAAoB,EACpB,OAA8B;QAE9B,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE;YAC7C,YAAY;YACZ,GAAG,OAAO;SACX,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,KAAY,EACZ,YAAoB,EACpB,iBAAyB,EACzB,KAAc;QAEd,OAAO,IAAI,CAAC,MAAM,CAAC,eAAe,CAChC,IAAI,CAAC,SAAS,EACd,KAAK,EACL,YAAY,EACZ,iBAAiB,EACjB,KAAK,CACN,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,kBAAkB,CACtB,KAAY,EACZ,YAAoB,EACpB,QAAgB,EAChB,UAAmB,EACnB,KAAc;QAEd,OAAO,IAAI,CAAC,MAAM,CAAC,mBAAmB,CACpC,IAAI,CAAC,SAAS,EACd,KAAK,EACL,YAAY,EACZ,EAAE,CAAC,QAAQ,CAAC,EAAE,UAAU,EAAE,EAC1B,KAAK,CACN,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,KAAY,EACZ,YAAoB,EACpB,WAAoC,EACpC,KAAc;QAEd,OAAO,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC;IAClG,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,KAAY,EACZ,YAAoB,EACpB,QAAgB,EAChB,KAAc;QAEd,OAAO,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;IAC5F,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,KAAY,EACZ,YAAoB,EACpB,QAAqC;QAErC,OAAO,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC;IACjF,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,QAAgB,EAAE,QAAgB;QAC9C,OAAO,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,YAA0B;QAClD,OAAO,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,YAAY,CAAC,eAAe,EAAE,YAAY,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC3F,CAAC;IAED,UAAU,CAAC,IAAe;QACxB,OAAO,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IAC1C,CAAC;IAED,kBAAkB;QAChB,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,SAAS;QACP,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;IAEO,UAAU,CAAC,MAAiB;QAClC,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YACtC,IACE,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;gBAC7B,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACzB,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAC1B,CAAC;gBACD,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;CACF"}
+{"version":3,"file":"time-travel.js","sourceRoot":"","sources":["../../src/time-travel/time-travel.ts"],"names":[],"mappings":"AAYA,OAAO,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AAGjE,OAAO,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAC/C,OAAO,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAC3C,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,mBAAmB,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAC;AAQ1E,MAAM,OAAO,UAAU;IACb,SAAS,CAAY;IACrB,eAAe,CAA4B;IAC3C,UAAU,CAAa;IACvB,QAAQ,CAAoB;IAC5B,MAAM,CAAkB;IACxB,UAAU,CAAkB;IAC5B,MAAM,CAAmB;IAEjC,YAAY,SAAoB,EAAE,OAA2B;QAC3D,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,0BAA0B,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,CAAC;QAEpE,IAAI,CAAC,eAAe,GAAG,OAAO,EAAE,eAAe,IAAI,IAAI,uBAAuB,EAAE,CAAC;QACjF,IAAI,CAAC,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,IAAI,kBAAkB,EAAE,CAAC;QAElE,IAAI,CAAC,QAAQ,GAAG,IAAI,iBAAiB,CAAC;YACpC,eAAe,EAAE,IAAI,CAAC,eAAe;SACtC,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,GAAG,IAAI,eAAe,CAAC;YAChC,eAAe,EAAE,IAAI,CAAC,eAAe;YACrC,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC,CAAC;QAEH,IAAI,CAAC,UAAU,GAAG,IAAI,eAAe,CAAC;YACpC,UAAU,EAAE,IAAI,CAAC,UAAU;SAC5B,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,UAAU,CACd,MAAiB,EACjB,SAAiB,EACjB,KAAc;QAEd,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC1F,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC5C,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAC/B,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACpC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAiB,EAAE,WAAoB;QACzD,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAC/B,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,sBAAsB,CAAC,MAAM,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;QAC/F,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACpC,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,MAAiB,EACjB,QAAgB,EAChB,WAAoB;QAEpB,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,QAAQ,IAAI,CAAC,EAAE,CAAC;YACjD,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;QACpE,CAAC;QAED,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAC/B,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAC1C,MAAM,WAAW,GAA0B,EAAE,CAAC;QAE9C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,IAAI,QAAQ,EAAE,CAAC;YAC7C,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,MAAM,EAAE,CAAC,EAAE;gBACrE,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,GAAG,WAAW,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;aAC5D,CAAC,CAAC;YACH,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC5C,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC/B,CAAC;QAED,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACpC,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,OAAe;QAClC,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,YAAoB;QACtC,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,YAAoB;QACzC,OAAO,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,MAAM,CACV,KAAY,EACZ,YAAoB,EACpB,OAAgC;QAEhC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE;YAC/D,cAAc,EAAE,YAAY;YAC5B,IAAI,EAAE,OAAO,EAAE,IAAI,IAAI,MAAM;YAC7B,GAAG,OAAO;SACX,CAAC,CAAC;QACH,IAAI,OAAO,EAAE,IAAI,KAAK,eAAe,EAAE,CAAC;YACtC,MAAM,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC;QAC9C,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QACvC,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,KAAY,EAAE,YAAoB;QAC1D,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,CAAC;IACrE,CAAC;IAED,KAAK,CAAC,UAAU,CACd,KAAY,EACZ,YAAoB,EACpB,OAA8C;QAE9C,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,IAAI,CACR,KAAY,EACZ,YAAoB,EACpB,OAA8B;QAE9B,OAAO,IAAI,CAAC,UAAU,CACpB,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE;YAC5C,YAAY;YACZ,GAAG,OAAO;SACX,CAAC,CACH,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,KAAY,EACZ,YAAoB,EACpB,iBAAyB,EACzB,KAAc;QAEd,OAAO,IAAI,CAAC,UAAU,CACpB,MAAM,IAAI,CAAC,MAAM,CAAC,eAAe,CAC/B,IAAI,CAAC,SAAS,EACd,KAAK,EACL,YAAY,EACZ,iBAAiB,EACjB,KAAK,CACN,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,kBAAkB,CACtB,KAAY,EACZ,YAAoB,EACpB,QAAgB,EAChB,UAAmB,EACnB,KAAc;QAEd,OAAO,IAAI,CAAC,UAAU,CACpB,MAAM,IAAI,CAAC,MAAM,CAAC,mBAAmB,CACnC,IAAI,CAAC,SAAS,EACd,KAAK,EACL,YAAY,EACZ,EAAE,CAAC,QAAQ,CAAC,EAAE,UAAU,EAAE,EAC1B,KAAK,CACN,CACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,KAAY,EACZ,YAAoB,EACpB,WAAoC,EACpC,KAAc;QAEd,OAAO,IAAI,CAAC,UAAU,CACpB,MAAM,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,KAAK,CAAC,CAC/F,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,KAAY,EACZ,YAAoB,EACpB,QAAgB,EAChB,KAAc;QAEd,OAAO,IAAI,CAAC,UAAU,CACpB,MAAM,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,QAAQ,EAAE,KAAK,CAAC,CACzF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,KAAY,EACZ,YAAoB,EACpB,QAAqC;QAErC,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC;QAC5F,KAAK,MAAM,IAAI,IAAI,KAAK;YAAE,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QACtD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,QAAgB,EAAE,QAAgB;QAC9C,OAAO,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,YAA0B;QAClD,OAAO,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,YAAY,CAAC,eAAe,EAAE,YAAY,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC3F,CAAC;IAED,UAAU,CAAC,IAAe;QACxB,OAAO,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IAC1C,CAAC;IAED,kBAAkB;QAChB,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,SAAS;QACP,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,WAAW,CAAC,MAAiB,EAAE,MAAM,GAAG,KAAK;QACzD,MAAM,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC;QAChC,IAAI,CAAC,MAAM,IAAI,CAAC,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAAE,OAAO;QACvD,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,mBAAmB,CAAC,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;IACrF,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,wBAAwB,CAAC,MAAoB;QACzD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;QACnE,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YACrC,OAAO;QACT,CAAC;QACD,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;YAC1B,GAAG,QAAQ;YACX,EAAE,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO;YACxB,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,aAAa,CAAC;YACpD,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,MAAM,EAAE,KAAK;SACd,CAAC,CAAC;IACL,CAAC;IAED,8FAA8F;IACtF,KAAK,CAAC,gBAAgB,CAAC,MAAiB;QAC9C,MAAM,EAAE,sBAAsB,EAAE,mBAAmB,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QAEpE,IAAI,mBAAmB,KAAK,SAAS,IAAI,mBAAmB,GAAG,CAAC,EAAE,CAAC;YACjE,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,mBAAmB,CAAC;YACvD,KAAK,MAAM,UAAU,IAAI,MAAM,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC/E,IAAI,UAAU,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,aAAa,EAAE,CAAC;oBACnD,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;gBACnD,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,sBAAsB,KAAK,SAAS,IAAI,sBAAsB,GAAG,CAAC,EAAE,CAAC;YACvE,MAAM,QAAQ,GAAG,CAAC,MAAM,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CACjF,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,CACxD,CAAC;YACF,KAAK,MAAM,UAAU,IAAI,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,MAAM,GAAG,sBAAsB,CAAC,EAAE,CAAC;gBACrF,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YACnD,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,IAAgB;QACvC,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC1C,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,UAAU,CAAC,MAAiB;QAClC,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YACtC,IACE,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;gBAC7B,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACzB,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAC1B,CAAC;gBACD,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;CACF"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cogitator-ai/core",
-  "version": "0.21.1",
+  "version": "0.23.0",
   "description": "Core runtime for Cogitator AI agents",
   "type": "module",
   "main": "./dist/index.js",
@@ -19,9 +19,9 @@
   },
   "dependencies": {
     "@anthropic-ai/sdk": "^0.131.0",
-    "@cogitator-ai/memory": "0.8.0",
+    "@cogitator-ai/memory": "0.9.0",
     "@cogitator-ai/models": "18.0.0",
-    "@cogitator-ai/types": "0.24.0",
+    "@cogitator-ai/types": "0.26.0",
     "nanoid": "^6.0.1",
     "openai": "^7.25.0",
     "zod": "^4.6.5"
@@ -31,7 +31,7 @@
   },
   "devDependencies": {
     "@aws-sdk/client-bedrock-runtime": "^3.1145.0",
-    "@cogitator-ai/sandbox": "0.4.0",
+    "@cogitator-ai/sandbox": "0.4.2",
     "@types/node": "^22.20.4",
     "@types/pg": "^8.23.1",
     "typescript": "^6.0.3",
@@ -39,7 +39,7 @@
   },
   "peerDependencies": {
     "@aws-sdk/client-bedrock-runtime": "^3.0.0",
-    "@cogitator-ai/sandbox": ">=0.4.0",
+    "@cogitator-ai/sandbox": ">=0.4.2",
     "better-sqlite3": "^11.0.0 || ^12.0.0 || ^13.0.0",
     "langfuse": "^3.0.0",
     "nodemailer": "^6.0.0",