@cogitator-ai/core 0.2.0 → 0.3.0

package/dist/time-travel/forker.js

@@ -0,0 +1,118 @@
+import { nanoid } from 'nanoid';
+export class ExecutionForker {
+    checkpointStore;
+    replayer;
+    constructor(options) {
+        this.checkpointStore = options.checkpointStore;
+        this.replayer = options.replayer;
+    }
+    async fork(cogitator, agent, options) {
+        const checkpoint = await this.checkpointStore.load(options.checkpointId);
+        if (!checkpoint) {
+            throw new Error(`Checkpoint not found: ${options.checkpointId}`);
+        }
+        const forkedCheckpoint = this.createForkedCheckpoint(checkpoint, options);
+        await this.checkpointStore.save(forkedCheckpoint);
+        const modifiedMessages = options.additionalContext
+            ? this.injectContext(forkedCheckpoint.messages, options.additionalContext)
+            : options.input
+                ? this.replaceInput(forkedCheckpoint.messages, options.input)
+                : undefined;
+        const result = await this.replayer.replay(cogitator, agent, {
+            fromCheckpoint: forkedCheckpoint.id,
+            mode: 'live',
+            modifiedMessages,
+            modifiedToolResults: options.mockToolResults,
+        });
+        return {
+            forkId: forkedCheckpoint.id,
+            checkpoint: forkedCheckpoint,
+            result,
+        };
+    }
+    async forkMultiple(cogitator, agent, checkpointId, variants) {
+        const results = [];
+        for (const variant of variants) {
+            const result = await this.fork(cogitator, agent, {
+                checkpointId,
+                ...variant,
+            });
+            results.push(result);
+        }
+        return results;
+    }
+    async forkWithContext(cogitator, agent, checkpointId, additionalContext, label) {
+        return this.fork(cogitator, agent, {
+            checkpointId,
+            additionalContext,
+            label: label ?? `fork_context_${nanoid(6)}`,
+        });
+    }
+    async forkWithMockedTools(cogitator, agent, checkpointId, mockResults, label) {
+        return this.fork(cogitator, agent, {
+            checkpointId,
+            mockToolResults: mockResults,
+            label: label ?? `fork_mocked_${nanoid(6)}`,
+        });
+    }
+    async forkWithNewInput(cogitator, agent, checkpointId, newInput, label) {
+        return this.fork(cogitator, agent, {
+            checkpointId,
+            input: newInput,
+            label: label ?? `fork_input_${nanoid(6)}`,
+        });
+    }
+    createForkedCheckpoint(original, options) {
+        return {
+            id: `ckpt_fork_${nanoid(12)}`,
+            traceId: original.traceId,
+            runId: original.runId,
+            agentId: original.agentId,
+            stepIndex: original.stepIndex,
+            messages: [...original.messages],
+            toolResults: { ...original.toolResults },
+            pendingToolCalls: [...original.pendingToolCalls],
+            label: options.label ?? `fork_of_${original.id}`,
+            createdAt: new Date(),
+            metadata: {
+                ...original.metadata,
+                forkedFrom: original.id,
+                forkType: options.additionalContext
+                    ? 'context'
+                    : options.input
+                        ? 'input'
+                        : options.mockToolResults
+                            ? 'mocked'
+                            : 'plain',
+            },
+        };
+    }
+    injectContext(messages, context) {
+        const result = [...messages];
+        const systemIndex = result.findIndex((m) => m.role === 'system');
+        if (systemIndex >= 0) {
+            result[systemIndex] = {
+                ...result[systemIndex],
+                content: `${result[systemIndex].content}\n\n---\nAdditional Context:\n${context}`,
+            };
+        }
+        else {
+            result.unshift({
+                role: 'system',
+                content: `Context:\n${context}`,
+            });
+        }
+        return result;
+    }
+    replaceInput(messages, newInput) {
+        const result = [...messages];
+        for (let i = result.length - 1; i >= 0; i--) {
+            if (result[i].role === 'user') {
+                result[i] = { ...result[i], content: newInput };
+                break;
+            }
+        }
+        return result;
+    }
+}
+//# sourceMappingURL=forker.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"forker.js","sourceRoot":"","sources":["../../src/time-travel/forker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAiBhC,MAAM,OAAO,eAAe;IAClB,eAAe,CAA4B;IAC3C,QAAQ,CAAoB;IAEpC,YAAY,OAA+B;QACzC,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;QAC/C,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,SAAoB,EAAE,KAAY,EAAE,OAAoB;QACjE,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QACzE,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,yBAAyB,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,gBAAgB,GAAG,IAAI,CAAC,sBAAsB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAC1E,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAElD,MAAM,gBAAgB,GAAG,OAAO,CAAC,iBAAiB;YAChD,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,QAAQ,EAAE,OAAO,CAAC,iBAAiB,CAAC;YAC1E,CAAC,CAAC,OAAO,CAAC,KAAK;gBACb,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,gBAAgB,CAAC,QAAQ,EAAE,OAAO,CAAC,KAAK,CAAC;gBAC7D,CAAC,CAAC,SAAS,CAAC;QAEhB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,EAAE,KAAK,EAAE;YAC1D,cAAc,EAAE,gBAAgB,CAAC,EAAE;YACnC,IAAI,EAAE,MAAM;YACZ,gBAAgB;YAChB,mBAAmB,EAAE,OAAO,CAAC,eAAe;SAC7C,CAAC,CAAC;QAEH,OAAO;YACL,MAAM,EAAE,gBAAgB,CAAC,EAAE;YAC3B,UAAU,EAAE,gBAAgB;YAC5B,MAAM;SACP,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,SAAoB,EACpB,KAAY,EACZ,YAAoB,EACpB,QAAqC;QAErC,MAAM,OAAO,GAAiB,EAAE,CAAC;QAEjC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE;gBAC/C,YAAY;gBACZ,GAAG,OAAO;aACX,CAAC,CAAC;YACH,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvB,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,SAAoB,EACpB,KAAY,EACZ,YAAoB,EACpB,iBAAyB,EACzB,KAAc;QAEd,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE;YACjC,YAAY;YACZ,iBAAiB;YACjB,KAAK,EAAE,KAAK,IAAI,gBAAgB,MAAM,CAAC,CAAC,CAAC,EAAE;SAC5C,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,SAAoB,EACpB,KAAY,EACZ,YAAoB,EACpB,WAAoC,EACpC,KAAc;QAEd,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE;YACjC,YAAY;YACZ,eAAe,EAAE,WAAW;YAC5B,KAAK,EAAE,KAAK,IAAI,eAAe,MAAM,CAAC,CAAC,CAAC,EAAE;SAC3C,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,SAAoB,EACpB,KAAY,EACZ,YAAoB,EACpB,QAAgB,EAChB,KAAc;QAEd,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE;YACjC,YAAY;YACZ,KAAK,EAAE,QAAQ;YACf,KAAK,EAAE,KAAK,IAAI,cAAc,MAAM,CAAC,CAAC,CAAC,EAAE;SAC1C,CAAC,CAAC;IACL,CAAC;IAEO,sBAAsB,CAC5B,QAA6B,EAC7B,OAAoB;QAEpB,OAAO;YACL,EAAE,EAAE,aAAa,MAAM,CAAC,EAAE,CAAC,EAAE;YAC7B,OAAO,EAAE,QAAQ,CAAC,OAAO;YACzB,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,OAAO,EAAE,QAAQ,CAAC,OAAO;YACzB,SAAS,EAAE,QAAQ,CAAC,SAAS;YAC7B,QAAQ,EAAE,CAAC,GAAG,QAAQ,CAAC,QAAQ,CAAC;YAChC,WAAW,EAAE,EAAE,GAAG,QAAQ,CAAC,WAAW,EAAE;YACxC,gBAAgB,EAAE,CAAC,GAAG,QAAQ,CAAC,gBAAgB,CAAC;YAChD,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,WAAW,QAAQ,CAAC,EAAE,EAAE;YAChD,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,QAAQ,EAAE;gBACR,GAAG,QAAQ,CAAC,QAAQ;gBACpB,UAAU,EAAE,QAAQ,CAAC,EAAE;gBACvB,QAAQ,EAAE,OAAO,CAAC,iBAAiB;oBACjC,CAAC,CAAC,SAAS;oBACX,CAAC,CAAC,OAAO,CAAC,KAAK;wBACb,CAAC,CAAC,OAAO;wBACT,CAAC,CAAC,OAAO,CAAC,eAAe;4BACvB,CAAC,CAAC,QAAQ;4BACV,CAAC,CAAC,OAAO;aAChB;SACF,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,QAAmB,EAAE,OAAe;QACxD,MAAM,MAAM,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;QAC7B,MAAM,WAAW,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAEjE,IAAI,WAAW,IAAI,CAAC,EAAE,CAAC;YACrB,MAAM,CAAC,WAAW,CAAC,GAAG;gBACpB,GAAG,MAAM,CAAC,WAAW,CAAC;gBACtB,OAAO,EAAE,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC,OAAO,iCAAiC,OAAO,EAAE;aAClF,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,OAAO,CAAC;gBACb,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,aAAa,OAAO,EAAE;aAChC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,YAAY,CAAC,QAAmB,EAAE,QAAgB;QACxD,MAAM,MAAM,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;QAE7B,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5C,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC9B,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;gBAChD,MAAM;YACR,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;CACF"}

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

@@ -0,0 +1,6 @@
+export { InMemoryCheckpointStore } from './checkpoint-store';
+export { ExecutionReplayer, type ExecutionReplayerOptions } from './replayer';
+export { ExecutionForker, type ExecutionForkerOptions } from './forker';
+export { TraceComparator, type TraceComparatorOptions } from './comparator';
+export { TimeTravel, type TimeTravelOptions } from './time-travel';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/time-travel/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,KAAK,wBAAwB,EAAE,MAAM,YAAY,CAAC;AAC9E,OAAO,EAAE,eAAe,EAAE,KAAK,sBAAsB,EAAE,MAAM,UAAU,CAAC;AACxE,OAAO,EAAE,eAAe,EAAE,KAAK,sBAAsB,EAAE,MAAM,cAAc,CAAC;AAC5E,OAAO,EAAE,UAAU,EAAE,KAAK,iBAAiB,EAAE,MAAM,eAAe,CAAC"}

package/dist/time-travel/index.js

@@ -0,0 +1,6 @@
+export { InMemoryCheckpointStore } from './checkpoint-store';
+export { ExecutionReplayer } from './replayer';
+export { ExecutionForker } from './forker';
+export { TraceComparator } from './comparator';
+export { TimeTravel } from './time-travel';
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/time-travel/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAiC,MAAM,YAAY,CAAC;AAC9E,OAAO,EAAE,eAAe,EAA+B,MAAM,UAAU,CAAC;AACxE,OAAO,EAAE,eAAe,EAA+B,MAAM,cAAc,CAAC;AAC5E,OAAO,EAAE,UAAU,EAA0B,MAAM,eAAe,CAAC"}

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

@@ -0,0 +1,20 @@
+import type { ReplayOptions, ReplayResult, TimeTravelCheckpointStore } from '@cogitator-ai/types';
+import type { Agent } from '../agent';
+import type { Cogitator } from '../cogitator';
+export interface ExecutionReplayerOptions {
+    checkpointStore: TimeTravelCheckpointStore;
+}
+export declare class ExecutionReplayer {
+    private checkpointStore;
+    constructor(options: ExecutionReplayerOptions);
+    replay(cogitator: Cogitator, agent: Agent, options: ReplayOptions): Promise<ReplayResult>;
+    private replayDeterministic;
+    private replayLive;
+    private buildMessagesForReplay;
+    private mergeToolResults;
+    private extractUserInput;
+    private createReplayAgent;
+    private countSteps;
+    private findDivergencePoint;
+}
+//# sourceMappingURL=replayer.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"replayer.d.ts","sourceRoot":"","sources":["../../src/time-travel/replayer.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAEV,aAAa,EACb,YAAY,EAGZ,yBAAyB,EAE1B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,UAAU,CAAC;AACtC,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAE9C,MAAM,WAAW,wBAAwB;IACvC,eAAe,EAAE,yBAAyB,CAAC;CAC5C;AAED,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,eAAe,CAA4B;gBAEvC,OAAO,EAAE,wBAAwB;IAIvC,MAAM,CAAC,SAAS,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,aAAa,GAAG,OAAO,CAAC,YAAY,CAAC;YAajF,mBAAmB;YAgDnB,UAAU;IA2CxB,OAAO,CAAC,sBAAsB;IAU9B,OAAO,CAAC,gBAAgB;IAOxB,OAAO,CAAC,gBAAgB;IAKxB,OAAO,CAAC,iBAAiB;IAkBzB,OAAO,CAAC,UAAU;IAclB,OAAO,CAAC,mBAAmB;CA4B5B"}

package/dist/time-travel/replayer.js

@@ -0,0 +1,147 @@
+import { nanoid } from 'nanoid';
+export class ExecutionReplayer {
+    checkpointStore;
+    constructor(options) {
+        this.checkpointStore = options.checkpointStore;
+    }
+    async replay(cogitator, agent, options) {
+        const checkpoint = await this.checkpointStore.load(options.fromCheckpoint);
+        if (!checkpoint) {
+            throw new Error(`Checkpoint not found: ${options.fromCheckpoint}`);
+        }
+        if (options.mode === 'deterministic') {
+            return this.replayDeterministic(cogitator, agent, checkpoint, options);
+        }
+        else {
+            return this.replayLive(cogitator, agent, checkpoint, options);
+        }
+    }
+    async replayDeterministic(_cogitator, agent, checkpoint, options) {
+        const messages = this.buildMessagesForReplay(checkpoint, options.modifiedMessages);
+        this.mergeToolResults(checkpoint.toolResults, options.modifiedToolResults);
+        const stepsReplayed = checkpoint.stepIndex + 1;
+        const stepsExecuted = 0;
+        const divergedAt = undefined;
+        const runId = `replay_${nanoid(12)}`;
+        const traceId = `trace_${nanoid(16)}`;
+        const startTime = Date.now();
+        const spans = [];
+        const output = messages.filter((m) => m.role === 'assistant').pop()?.content ?? '';
+        const result = {
+            output,
+            runId,
+            agentId: agent.id,
+            threadId: checkpoint.runId,
+            usage: {
+                inputTokens: 0,
+                outputTokens: 0,
+                totalTokens: 0,
+                cost: 0,
+                duration: Date.now() - startTime,
+            },
+            toolCalls: checkpoint.pendingToolCalls,
+            messages,
+            trace: {
+                traceId,
+                spans,
+            },
+            replayedFrom: checkpoint.id,
+            originalTraceId: checkpoint.traceId,
+            divergedAt,
+            stepsReplayed,
+            stepsExecuted,
+        };
+        return result;
+    }
+    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 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);
+        const divergedAt = this.findDivergencePoint(checkpoint, runResult);
+        const replayResult = {
+            ...runResult,
+            replayedFrom: checkpoint.id,
+            originalTraceId: checkpoint.traceId,
+            divergedAt,
+            stepsReplayed,
+            stepsExecuted,
+        };
+        return replayResult;
+    }
+    buildMessagesForReplay(checkpoint, modifications) {
+        if (modifications && modifications.length > 0) {
+            return modifications;
+        }
+        return [...checkpoint.messages];
+    }
+    mergeToolResults(cached, modified) {
+        return { ...cached, ...modified };
+    }
+    extractUserInput(messages) {
+        const userMessages = messages.filter((m) => m.role === 'user');
+        return userMessages[userMessages.length - 1]?.content ?? '';
+    }
+    createReplayAgent(agent, preloadedMessages) {
+        const systemMessage = preloadedMessages.find((m) => m.role === 'system');
+        const contextFromHistory = preloadedMessages
+            .filter((m) => m.role === 'assistant' || m.role === 'tool')
+            .map((m) => `[${m.role}]: ${m.content}`)
+            .join('\n');
+        const newInstructions = systemMessage
+            ? `${systemMessage.content}\n\n---\nReplay Context (conversation history up to checkpoint):\n${contextFromHistory}`
+            : agent.instructions;
+        return new agent.constructor({
+            ...agent.config,
+            name: `${agent.name}_replay`,
+            instructions: newInstructions,
+        });
+    }
+    countSteps(result) {
+        let count = 0;
+        for (const span of result.trace.spans) {
+            if (span.name.startsWith('tool.') ||
+                span.name.includes('llm') ||
+                span.name.includes('chat')) {
+                count++;
+            }
+        }
+        return count;
+    }
+    findDivergencePoint(checkpoint, result) {
+        const originalToolCalls = checkpoint.pendingToolCalls;
+        const newToolCalls = result.toolCalls;
+        for (let i = 0; i < Math.min(originalToolCalls.length, newToolCalls.length); i++) {
+            const orig = originalToolCalls[i];
+            const curr = newToolCalls[i];
+            if (orig.name !== curr.name) {
+                return checkpoint.stepIndex + i + 1;
+            }
+            const origArgs = JSON.stringify(orig.arguments);
+            const currArgs = JSON.stringify(curr.arguments);
+            if (origArgs !== currArgs) {
+                return checkpoint.stepIndex + i + 1;
+            }
+        }
+        if (originalToolCalls.length !== newToolCalls.length) {
+            return checkpoint.stepIndex + Math.min(originalToolCalls.length, newToolCalls.length) + 1;
+        }
+        return undefined;
+    }
+}
+//# sourceMappingURL=replayer.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"replayer.js","sourceRoot":"","sources":["../../src/time-travel/replayer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAiBhC,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,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,WAAW,EAAE,OAAO,CAAC,mBAAmB,CAAC,CAAC;QAE3E,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,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE,OAAO,IAAI,EAAE,CAAC;QAEnF,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;YACR,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,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,OAAO,YAAY,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,OAAO,IAAI,EAAE,CAAC;IAC9D,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,CAAC,CAAC,OAAO,EAAE,CAAC;aACvC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,MAAM,eAAe,GAAG,aAAa;YACnC,CAAC,CAAC,GAAG,aAAa,CAAC,OAAO,qEAAqE,kBAAkB,EAAE;YACnH,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;CACF"}

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

@@ -0,0 +1,41 @@
+import type { ExecutionCheckpoint, ReplayOptions, ReplayResult, ForkOptions, ForkResult, TraceDiff, TraceStore, TimeTravelCheckpointStore, RunResult, TimeTravelConfig } from '@cogitator-ai/types';
+import type { Agent } from '../agent';
+import type { Cogitator } from '../cogitator';
+export interface TimeTravelOptions {
+    checkpointStore?: TimeTravelCheckpointStore;
+    traceStore?: TraceStore;
+    config?: Partial<TimeTravelConfig>;
+}
+export declare class TimeTravel {
+    private cogitator;
+    private checkpointStore;
+    private traceStore;
+    private replayer;
+    private forker;
+    private comparator;
+    private config;
+    constructor(cogitator: Cogitator, options?: TimeTravelOptions);
+    checkpoint(result: RunResult, stepIndex: number, label?: string): Promise<ExecutionCheckpoint>;
+    checkpointAll(result: RunResult, labelPrefix?: string): Promise<ExecutionCheckpoint[]>;
+    checkpointEvery(result: RunResult, interval: number, labelPrefix?: string): Promise<ExecutionCheckpoint[]>;
+    getCheckpoints(traceId: string): Promise<ExecutionCheckpoint[]>;
+    getCheckpoint(checkpointId: string): Promise<ExecutionCheckpoint | null>;
+    deleteCheckpoint(checkpointId: string): Promise<boolean>;
+    replay(agent: Agent, checkpointId: string, options?: Partial<ReplayOptions>): Promise<ReplayResult>;
+    replayDeterministic(agent: Agent, checkpointId: string): Promise<ReplayResult>;
+    replayLive(agent: Agent, checkpointId: string, options?: Omit<Partial<ReplayOptions>, 'mode'>): Promise<ReplayResult>;
+    fork(agent: Agent, checkpointId: string, options?: Partial<ForkOptions>): Promise<ForkResult>;
+    forkWithContext(agent: Agent, checkpointId: string, additionalContext: string, label?: string): Promise<ForkResult>;
+    forkWithMockedTool(agent: Agent, checkpointId: string, toolName: string, mockResult: unknown, label?: string): Promise<ForkResult>;
+    forkWithMockedTools(agent: Agent, checkpointId: string, mockResults: Record<string, unknown>, label?: string): Promise<ForkResult>;
+    forkWithNewInput(agent: Agent, checkpointId: string, newInput: string, label?: string): Promise<ForkResult>;
+    forkMultiple(agent: Agent, checkpointId: string, variants: Array<Partial<ForkOptions>>): Promise<ForkResult[]>;
+    compare(traceId1: string, traceId2: string): Promise<TraceDiff>;
+    compareWithOriginal(replayResult: ReplayResult): Promise<TraceDiff>;
+    formatDiff(diff: TraceDiff): string;
+    getCheckpointStore(): TimeTravelCheckpointStore;
+    getTraceStore(): TraceStore;
+    getConfig(): TimeTravelConfig;
+    private countSteps;
+}
+//# sourceMappingURL=time-travel.d.ts.map

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

@@ -0,0 +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,cAAc,CAAC;AAO9C,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;IAOzB,aAAa,CAAC,MAAM,EAAE,SAAS,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAKtF,eAAe,CACnB,MAAM,EAAE,SAAS,EACjB,QAAQ,EAAE,MAAM,EAChB,WAAW,CAAC,EAAE,MAAM,GACnB,OAAO,CAAC,mBAAmB,EAAE,CAAC;IAgB3B,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"}

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

@@ -0,0 +1,127 @@
+import { DEFAULT_TIME_TRAVEL_CONFIG } from '@cogitator-ai/types';
+import { InMemoryCheckpointStore } from './checkpoint-store';
+import { ExecutionReplayer } from './replayer';
+import { ExecutionForker } from './forker';
+import { TraceComparator } from './comparator';
+import { InMemoryTraceStore } from '../learning/trace-store';
+export class TimeTravel {
+    cogitator;
+    checkpointStore;
+    traceStore;
+    replayer;
+    forker;
+    comparator;
+    config;
+    constructor(cogitator, options) {
+        this.cogitator = cogitator;
+        this.config = { ...DEFAULT_TIME_TRAVEL_CONFIG, ...options?.config };
+        this.checkpointStore = options?.checkpointStore ?? new InMemoryCheckpointStore();
+        this.traceStore = options?.traceStore ?? new InMemoryTraceStore();
+        this.replayer = new ExecutionReplayer({
+            checkpointStore: this.checkpointStore,
+        });
+        this.forker = new ExecutionForker({
+            checkpointStore: this.checkpointStore,
+            replayer: this.replayer,
+        });
+        this.comparator = new TraceComparator({
+            traceStore: this.traceStore,
+        });
+    }
+    async checkpoint(result, stepIndex, label) {
+        const store = this.checkpointStore;
+        const checkpoint = store.createFromRunResult(result, stepIndex, { label });
+        await store.save(checkpoint);
+        return checkpoint;
+    }
+    async checkpointAll(result, labelPrefix) {
+        const store = this.checkpointStore;
+        return store.createAllFromRunResult(result, { labelPrefix });
+    }
+    async checkpointEvery(result, interval, labelPrefix) {
+        const store = this.checkpointStore;
+        const stepCount = this.countSteps(result);
+        const checkpoints = [];
+        for (let i = 0; i < stepCount; i += interval) {
+            const checkpoint = store.createFromRunResult(result, i, {
+                label: labelPrefix ? `${labelPrefix}_step_${i}` : undefined,
+            });
+            await store.save(checkpoint);
+            checkpoints.push(checkpoint);
+        }
+        return checkpoints;
+    }
+    async getCheckpoints(traceId) {
+        return this.checkpointStore.getByTrace(traceId);
+    }
+    async getCheckpoint(checkpointId) {
+        return this.checkpointStore.load(checkpointId);
+    }
+    async deleteCheckpoint(checkpointId) {
+        return this.checkpointStore.delete(checkpointId);
+    }
+    async replay(agent, checkpointId, options) {
+        return this.replayer.replay(this.cogitator, agent, {
+            fromCheckpoint: checkpointId,
+            mode: options?.mode ?? 'live',
+            ...options,
+        });
+    }
+    async replayDeterministic(agent, checkpointId) {
+        return this.replay(agent, checkpointId, { mode: 'deterministic' });
+    }
+    async replayLive(agent, checkpointId, options) {
+        return this.replay(agent, checkpointId, { mode: 'live', ...options });
+    }
+    async fork(agent, checkpointId, options) {
+        return this.forker.fork(this.cogitator, agent, {
+            checkpointId,
+            ...options,
+        });
+    }
+    async forkWithContext(agent, checkpointId, additionalContext, label) {
+        return 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);
+    }
+    async forkWithMockedTools(agent, checkpointId, mockResults, label) {
+        return 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);
+    }
+    async forkMultiple(agent, checkpointId, variants) {
+        return this.forker.forkMultiple(this.cogitator, agent, checkpointId, variants);
+    }
+    async compare(traceId1, traceId2) {
+        return this.comparator.compare(traceId1, traceId2);
+    }
+    async compareWithOriginal(replayResult) {
+        return this.comparator.compare(replayResult.originalTraceId, replayResult.trace.traceId);
+    }
+    formatDiff(diff) {
+        return this.comparator.formatDiff(diff);
+    }
+    getCheckpointStore() {
+        return this.checkpointStore;
+    }
+    getTraceStore() {
+        return this.traceStore;
+    }
+    getConfig() {
+        return { ...this.config };
+    }
+    countSteps(result) {
+        let count = 0;
+        for (const span of result.trace.spans) {
+            if (span.name.startsWith('tool.') ||
+                span.name.includes('llm') ||
+                span.name.includes('chat')) {
+                count++;
+            }
+        }
+        return count;
+    }
+}
+//# sourceMappingURL=time-travel.js.map

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

@@ -0,0 +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,KAAK,GAAG,IAAI,CAAC,eAA0C,CAAC;QAC9D,MAAM,UAAU,GAAG,KAAK,CAAC,mBAAmB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC3E,MAAM,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC7B,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,MAAiB,EAAE,WAAoB;QACzD,MAAM,KAAK,GAAG,IAAI,CAAC,eAA0C,CAAC;QAC9D,OAAO,KAAK,CAAC,sBAAsB,CAAC,MAAM,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,MAAiB,EACjB,QAAgB,EAChB,WAAoB;QAEpB,MAAM,KAAK,GAAG,IAAI,CAAC,eAA0C,CAAC;QAC9D,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,KAAK,CAAC,mBAAmB,CAAC,MAAM,EAAE,CAAC,EAAE;gBACtD,KAAK,EAAE,WAAW,CAAC,CAAC,CAAC,GAAG,WAAW,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;aAC5D,CAAC,CAAC;YACH,MAAM,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC7B,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"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cogitator-ai/core",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "description": "Core runtime for Cogitator AI agents",
   "type": "module",
   "main": "./dist/index.js",
@@ -17,24 +17,32 @@
   "dependencies": {
     "@anthropic-ai/sdk": "^0.71.2",
     "@types/node": "^25.0.0",
+    "@types/pg": "^8.10.9",
     "nanoid": "^5.0.4",
     "openai": "^4.24.0",
     "zod": "^3.22.4",
     "zod-to-json-schema": "^3.22.4",
-    "@cogitator-ai/memory": "0.2.0",
-    "@cogitator-ai/models": "1.1.0",
-    "@cogitator-ai/types": "0.3.1"
+    "@cogitator-ai/memory": "0.3.0",
+    "@cogitator-ai/types": "0.4.0",
+    "@cogitator-ai/models": "2.0.0"
+  },
+  "optionalDependencies": {
+    "pg": "^8.11.3"
   },
   "devDependencies": {
     "typescript": "^5.3.0",
     "vitest": "^1.0.0"
   },
   "peerDependencies": {
-    "@cogitator-ai/sandbox": "0.2.0"
+    "pg": "^8.0.0",
+    "@cogitator-ai/sandbox": "0.2.1"
   },
   "peerDependenciesMeta": {
     "@cogitator-ai/sandbox": {
       "optional": true
+    },
+    "pg": {
+      "optional": true
     }
   },
   "repository": {