@cogitator-ai/core 0.1.0 → 0.3.0

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

@@ -0,0 +1,240 @@
+import { nanoid } from 'nanoid';
+export class InMemoryCheckpointStore {
+    checkpoints = new Map();
+    traceIndex = new Map();
+    agentIndex = new Map();
+    runIdIndex = new Map();
+    labelIndex = new Map();
+    async save(checkpoint) {
+        this.checkpoints.set(checkpoint.id, checkpoint);
+        this.addToIndex(this.traceIndex, checkpoint.traceId, checkpoint.id);
+        this.addToIndex(this.agentIndex, checkpoint.agentId, checkpoint.id);
+        this.addToIndex(this.runIdIndex, checkpoint.runId, checkpoint.id);
+        if (checkpoint.label) {
+            this.addToIndex(this.labelIndex, checkpoint.label, checkpoint.id);
+        }
+    }
+    async load(id) {
+        return this.checkpoints.get(id) ?? null;
+    }
+    async list(query) {
+        let candidates = [];
+        if (query.traceId) {
+            const traceCheckpoints = this.traceIndex.get(query.traceId);
+            if (!traceCheckpoints)
+                return [];
+            for (const id of traceCheckpoints) {
+                const cp = this.checkpoints.get(id);
+                if (cp)
+                    candidates.push(cp);
+            }
+        }
+        else if (query.agentId) {
+            const agentCheckpoints = this.agentIndex.get(query.agentId);
+            if (!agentCheckpoints)
+                return [];
+            for (const id of agentCheckpoints) {
+                const cp = this.checkpoints.get(id);
+                if (cp)
+                    candidates.push(cp);
+            }
+        }
+        else if (query.runId) {
+            const runCheckpoints = this.runIdIndex.get(query.runId);
+            if (!runCheckpoints)
+                return [];
+            for (const id of runCheckpoints) {
+                const cp = this.checkpoints.get(id);
+                if (cp)
+                    candidates.push(cp);
+            }
+        }
+        else if (query.label) {
+            const labelCheckpoints = this.labelIndex.get(query.label);
+            if (!labelCheckpoints)
+                return [];
+            for (const id of labelCheckpoints) {
+                const cp = this.checkpoints.get(id);
+                if (cp)
+                    candidates.push(cp);
+            }
+        }
+        else {
+            candidates = Array.from(this.checkpoints.values());
+        }
+        if (query.before) {
+            candidates = candidates.filter((cp) => cp.createdAt < query.before);
+        }
+        if (query.after) {
+            candidates = candidates.filter((cp) => cp.createdAt > query.after);
+        }
+        candidates.sort((a, b) => a.stepIndex - b.stepIndex);
+        if (query.limit) {
+            candidates = candidates.slice(0, query.limit);
+        }
+        return candidates;
+    }
+    async delete(id) {
+        const checkpoint = this.checkpoints.get(id);
+        if (!checkpoint)
+            return false;
+        this.checkpoints.delete(id);
+        this.removeFromIndex(this.traceIndex, checkpoint.traceId, id);
+        this.removeFromIndex(this.agentIndex, checkpoint.agentId, id);
+        this.removeFromIndex(this.runIdIndex, checkpoint.runId, id);
+        if (checkpoint.label) {
+            this.removeFromIndex(this.labelIndex, checkpoint.label, id);
+        }
+        return true;
+    }
+    async getByTrace(traceId) {
+        return this.list({ traceId });
+    }
+    async getByAgent(agentId, limit) {
+        return this.list({ agentId, limit });
+    }
+    async getByLabel(label) {
+        return this.list({ label });
+    }
+    async clear(agentId) {
+        if (agentId) {
+            const agentCheckpoints = this.agentIndex.get(agentId);
+            if (!agentCheckpoints)
+                return;
+            for (const id of Array.from(agentCheckpoints)) {
+                await this.delete(id);
+            }
+        }
+        else {
+            this.checkpoints.clear();
+            this.traceIndex.clear();
+            this.agentIndex.clear();
+            this.runIdIndex.clear();
+            this.labelIndex.clear();
+        }
+    }
+    createFromRunResult(result, stepIndex, options) {
+        const messagesUpToStep = this.extractMessagesUpToStep(result.messages, stepIndex);
+        const toolResultsUpToStep = this.extractToolResultsUpToStep(result, stepIndex);
+        const pendingToolCalls = this.extractPendingToolCalls(result, stepIndex);
+        const checkpoint = {
+            id: `ckpt_${nanoid(12)}`,
+            traceId: result.trace.traceId,
+            runId: result.runId,
+            agentId: result.agentId,
+            stepIndex,
+            messages: messagesUpToStep,
+            toolResults: toolResultsUpToStep,
+            pendingToolCalls,
+            label: options?.label,
+            createdAt: new Date(),
+            metadata: options?.metadata,
+        };
+        return checkpoint;
+    }
+    createFromTrace(trace, stepIndex, options) {
+        const stepsUpToIndex = trace.steps.slice(0, stepIndex + 1);
+        const messages = [];
+        messages.push({ role: 'user', content: trace.input });
+        const toolResults = {};
+        for (const step of stepsUpToIndex) {
+            if (step.type === 'tool_call' && step.toolResult) {
+                toolResults[step.toolResult.callId] = step.toolResult.result;
+            }
+        }
+        const pendingStep = trace.steps[stepIndex];
+        const pendingToolCalls = pendingStep?.type === 'tool_call' && pendingStep.toolCall ? [pendingStep.toolCall] : [];
+        const checkpoint = {
+            id: `ckpt_${nanoid(12)}`,
+            traceId: trace.id,
+            runId: trace.runId,
+            agentId: trace.agentId,
+            stepIndex,
+            messages,
+            toolResults,
+            pendingToolCalls,
+            label: options?.label,
+            createdAt: new Date(),
+            metadata: options?.metadata,
+        };
+        return checkpoint;
+    }
+    async createAllFromRunResult(result, options) {
+        const checkpoints = [];
+        const stepCount = this.countStepsFromSpans(result);
+        for (let i = 0; i < stepCount; i++) {
+            const checkpoint = this.createFromRunResult(result, i, {
+                label: options?.labelPrefix ? `${options.labelPrefix}_step_${i}` : undefined,
+            });
+            await this.save(checkpoint);
+            checkpoints.push(checkpoint);
+        }
+        return checkpoints;
+    }
+    addToIndex(index, key, id) {
+        let set = index.get(key);
+        if (!set) {
+            set = new Set();
+            index.set(key, set);
+        }
+        set.add(id);
+    }
+    removeFromIndex(index, key, id) {
+        index.get(key)?.delete(id);
+    }
+    extractMessagesUpToStep(messages, stepIndex) {
+        let toolCallCount = 0;
+        const result = [];
+        for (const msg of messages) {
+            result.push(msg);
+            if (msg.role === 'tool') {
+                toolCallCount++;
+                if (toolCallCount > stepIndex)
+                    break;
+            }
+        }
+        return result;
+    }
+    extractToolResultsUpToStep(result, stepIndex) {
+        const toolResults = {};
+        let count = 0;
+        for (const span of result.trace.spans) {
+            if (span.name.startsWith('tool.')) {
+                if (count < stepIndex && span.attributes?.result !== undefined) {
+                    const callId = span.attributes?.call_id;
+                    if (callId) {
+                        toolResults[callId] = span.attributes.result;
+                    }
+                }
+                count++;
+            }
+        }
+        return toolResults;
+    }
+    extractPendingToolCalls(result, stepIndex) {
+        let count = 0;
+        for (const span of result.trace.spans) {
+            if (span.name.startsWith('tool.')) {
+                if (count === stepIndex) {
+                    const toolName = span.name.replace('tool.', '');
+                    const toolCall = result.toolCalls.find((tc) => tc.name === toolName);
+                    return toolCall ? [toolCall] : [];
+                }
+                count++;
+            }
+        }
+        return [];
+    }
+    countStepsFromSpans(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=checkpoint-store.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"checkpoint-store.js","sourceRoot":"","sources":["../../src/time-travel/checkpoint-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAUhC,MAAM,OAAO,uBAAuB;IAC1B,WAAW,GAAG,IAAI,GAAG,EAA+B,CAAC;IACrD,UAAU,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC5C,UAAU,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC5C,UAAU,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC5C,UAAU,GAAG,IAAI,GAAG,EAAuB,CAAC;IAEpD,KAAK,CAAC,IAAI,CAAC,UAA+B;QACxC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QAChD,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,OAAO,EAAE,UAAU,CAAC,EAAE,CAAC,CAAC;QACpE,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,OAAO,EAAE,UAAU,CAAC,EAAE,CAAC,CAAC;QACpE,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,KAAK,EAAE,UAAU,CAAC,EAAE,CAAC,CAAC;QAElE,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;YACrB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,KAAK,EAAE,UAAU,CAAC,EAAE,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,EAAU;QACnB,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,KAAsB;QAC/B,IAAI,UAAU,GAA0B,EAAE,CAAC;QAE3C,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,MAAM,gBAAgB,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC5D,IAAI,CAAC,gBAAgB;gBAAE,OAAO,EAAE,CAAC;YACjC,KAAK,MAAM,EAAE,IAAI,gBAAgB,EAAE,CAAC;gBAClC,MAAM,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACpC,IAAI,EAAE;oBAAE,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;aAAM,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YACzB,MAAM,gBAAgB,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC5D,IAAI,CAAC,gBAAgB;gBAAE,OAAO,EAAE,CAAC;YACjC,KAAK,MAAM,EAAE,IAAI,gBAAgB,EAAE,CAAC;gBAClC,MAAM,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACpC,IAAI,EAAE;oBAAE,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;aAAM,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YACvB,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACxD,IAAI,CAAC,cAAc;gBAAE,OAAO,EAAE,CAAC;YAC/B,KAAK,MAAM,EAAE,IAAI,cAAc,EAAE,CAAC;gBAChC,MAAM,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACpC,IAAI,EAAE;oBAAE,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;aAAM,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YACvB,MAAM,gBAAgB,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC1D,IAAI,CAAC,gBAAgB;gBAAE,OAAO,EAAE,CAAC;YACjC,KAAK,MAAM,EAAE,IAAI,gBAAgB,EAAE,CAAC;gBAClC,MAAM,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACpC,IAAI,EAAE;oBAAE,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;aAAM,CAAC;YACN,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACjB,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,SAAS,GAAG,KAAK,CAAC,MAAO,CAAC,CAAC;QACvE,CAAC;QAED,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YAChB,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,SAAS,GAAG,KAAK,CAAC,KAAM,CAAC,CAAC;QACtE,CAAC;QAED,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;QAErD,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YAChB,UAAU,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;QAChD,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,EAAU;QACrB,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC5C,IAAI,CAAC,UAAU;YAAE,OAAO,KAAK,CAAC;QAE9B,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC5B,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAC9D,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAC9D,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAE5D,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;YACrB,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAAe;QAC9B,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAAe,EAAE,KAAc;QAC9C,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAa;QAC5B,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,OAAgB;QAC1B,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,gBAAgB,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACtD,IAAI,CAAC,gBAAgB;gBAAE,OAAO;YAE9B,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBAC9C,MAAM,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YACxB,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YACxB,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YACxB,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,mBAAmB,CACjB,MAAiB,EACjB,SAAiB,EACjB,OAAgE;QAEhE,MAAM,gBAAgB,GAAG,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,QAAqB,EAAE,SAAS,CAAC,CAAC;QAC/F,MAAM,mBAAmB,GAAG,IAAI,CAAC,0BAA0B,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAC/E,MAAM,gBAAgB,GAAG,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAEzE,MAAM,UAAU,GAAwB;YACtC,EAAE,EAAE,QAAQ,MAAM,CAAC,EAAE,CAAC,EAAE;YACxB,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,OAAO;YAC7B,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,SAAS;YACT,QAAQ,EAAE,gBAAgB;YAC1B,WAAW,EAAE,mBAAmB;YAChC,gBAAgB;YAChB,KAAK,EAAE,OAAO,EAAE,KAAK;YACrB,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,QAAQ,EAAE,OAAO,EAAE,QAAQ;SAC5B,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,eAAe,CACb,KAAqB,EACrB,SAAiB,EACjB,OAAgE;QAEhE,MAAM,cAAc,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;QAE3D,MAAM,QAAQ,GAAc,EAAE,CAAC;QAC/B,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;QAEtD,MAAM,WAAW,GAA4B,EAAE,CAAC;QAEhD,KAAK,MAAM,IAAI,IAAI,cAAc,EAAE,CAAC;YAClC,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBACjD,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;YAC/D,CAAC;QACH,CAAC;QAED,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAC3C,MAAM,gBAAgB,GACpB,WAAW,EAAE,IAAI,KAAK,WAAW,IAAI,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAE1F,MAAM,UAAU,GAAwB;YACtC,EAAE,EAAE,QAAQ,MAAM,CAAC,EAAE,CAAC,EAAE;YACxB,OAAO,EAAE,KAAK,CAAC,EAAE;YACjB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,SAAS;YACT,QAAQ;YACR,WAAW;YACX,gBAAgB;YAChB,KAAK,EAAE,OAAO,EAAE,KAAK;YACrB,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,QAAQ,EAAE,OAAO,EAAE,QAAQ;SAC5B,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,sBAAsB,CAC1B,MAAiB,EACjB,OAAkC;QAElC,MAAM,WAAW,GAA0B,EAAE,CAAC;QAC9C,MAAM,SAAS,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;QAEnD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;YACnC,MAAM,UAAU,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,CAAC,EAAE;gBACrD,KAAK,EAAE,OAAO,EAAE,WAAW,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,WAAW,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;aAC7E,CAAC,CAAC;YACH,MAAM,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC5B,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC/B,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAEO,UAAU,CAAC,KAA+B,EAAE,GAAW,EAAE,EAAU;QACzE,IAAI,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACzB,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,GAAG,GAAG,IAAI,GAAG,EAAE,CAAC;YAChB,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACtB,CAAC;QACD,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IACd,CAAC;IAEO,eAAe,CAAC,KAA+B,EAAE,GAAW,EAAE,EAAU;QAC9E,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC;IAEO,uBAAuB,CAAC,QAAmB,EAAE,SAAiB;QACpE,IAAI,aAAa,GAAG,CAAC,CAAC;QACtB,MAAM,MAAM,GAAc,EAAE,CAAC;QAE7B,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACjB,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACxB,aAAa,EAAE,CAAC;gBAChB,IAAI,aAAa,GAAG,SAAS;oBAAE,MAAM;YACvC,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,0BAA0B,CAChC,MAAiB,EACjB,SAAiB;QAEjB,MAAM,WAAW,GAA4B,EAAE,CAAC;QAChD,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YACtC,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;gBAClC,IAAI,KAAK,GAAG,SAAS,IAAI,IAAI,CAAC,UAAU,EAAE,MAAM,KAAK,SAAS,EAAE,CAAC;oBAC/D,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,OAAiB,CAAC;oBAClD,IAAI,MAAM,EAAE,CAAC;wBACX,WAAW,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;oBAC/C,CAAC;gBACH,CAAC;gBACD,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAEO,uBAAuB,CAC7B,MAAiB,EACjB,SAAiB;QAEjB,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YACtC,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;gBAClC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;oBACxB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;oBAChD,MAAM,QAAQ,GAAG,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;oBACrE,OAAO,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACpC,CAAC;gBACD,KAAK,EAAE,CAAC;YACV,CAAC;QACH,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAEO,mBAAmB,CAAC,MAAiB;QAC3C,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/dist/time-travel/comparator.d.ts

@@ -0,0 +1,26 @@
+import type { TraceDiff, StepDiff, ExecutionStep, ExecutionTrace, TraceMetrics, TraceStore, ToolResult } from '@cogitator-ai/types';
+export interface TraceComparatorOptions {
+    traceStore: TraceStore;
+}
+export declare class TraceComparator {
+    private traceStore;
+    constructor(options: TraceComparatorOptions);
+    compare(traceId1: string, traceId2: string): Promise<TraceDiff>;
+    computeDiff(trace1: ExecutionTrace, trace2: ExecutionTrace): TraceDiff;
+    findDivergencePoint(steps1: ExecutionStep[], steps2: ExecutionStep[]): number | undefined;
+    compareSteps(step1: ExecutionStep, step2: ExecutionStep): StepDiff;
+    compareToolResults(result1: ToolResult, result2: ToolResult): {
+        similar: boolean;
+        differences: string[];
+    };
+    compareMetrics(metrics1: TraceMetrics, metrics2: TraceMetrics): Record<string, {
+        value1: number;
+        value2: number;
+        delta: number;
+    }>;
+    formatDiff(diff: TraceDiff): string;
+    private compareStepSequences;
+    private formatDelta;
+    private getStatusIcon;
+}
+//# sourceMappingURL=comparator.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"comparator.d.ts","sourceRoot":"","sources":["../../src/time-travel/comparator.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,SAAS,EACT,QAAQ,EAER,aAAa,EACb,cAAc,EACd,YAAY,EACZ,UAAU,EACV,UAAU,EACX,MAAM,qBAAqB,CAAC;AAE7B,MAAM,WAAW,sBAAsB;IACrC,UAAU,EAAE,UAAU,CAAC;CACxB;AAED,qBAAa,eAAe;IAC1B,OAAO,CAAC,UAAU,CAAa;gBAEnB,OAAO,EAAE,sBAAsB;IAIrC,OAAO,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC;IAcrE,WAAW,CAAC,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,cAAc,GAAG,SAAS;IA8CtE,mBAAmB,CAAC,MAAM,EAAE,aAAa,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE,GAAG,MAAM,GAAG,SAAS;IAiBzF,YAAY,CAAC,KAAK,EAAE,aAAa,EAAE,KAAK,EAAE,aAAa,GAAG,QAAQ;IAoDlE,kBAAkB,CAChB,OAAO,EAAE,UAAU,EACnB,OAAO,EAAE,UAAU,GAClB;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,WAAW,EAAE,MAAM,EAAE,CAAA;KAAE;IA6B9C,cAAc,CACZ,QAAQ,EAAE,YAAY,EACtB,QAAQ,EAAE,YAAY,GACrB,MAAM,CAAC,MAAM,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IAoBpE,UAAU,CAAC,IAAI,EAAE,SAAS,GAAG,MAAM;IA2DnC,OAAO,CAAC,oBAAoB;IA4B5B,OAAO,CAAC,WAAW;IAKnB,OAAO,CAAC,aAAa;CAgBtB"}

package/dist/time-travel/comparator.js

@@ -0,0 +1,253 @@
+export class TraceComparator {
+    traceStore;
+    constructor(options) {
+        this.traceStore = options.traceStore;
+    }
+    async compare(traceId1, traceId2) {
+        const trace1 = await this.traceStore.get(traceId1);
+        const trace2 = await this.traceStore.get(traceId2);
+        if (!trace1) {
+            throw new Error(`Trace not found: ${traceId1}`);
+        }
+        if (!trace2) {
+            throw new Error(`Trace not found: ${traceId2}`);
+        }
+        return this.computeDiff(trace1, trace2);
+    }
+    computeDiff(trace1, trace2) {
+        const stepDiffs = this.compareStepSequences(trace1.steps, trace2.steps);
+        const divergencePoint = this.findDivergencePoint(trace1.steps, trace2.steps);
+        const commonSteps = stepDiffs.filter((d) => d.status === 'identical' || d.status === 'similar').length;
+        const trace1OnlySteps = stepDiffs.filter((d) => d.status === 'only_in_1').length;
+        const trace2OnlySteps = stepDiffs.filter((d) => d.status === 'only_in_2').length;
+        const tokens1 = trace1.usage.inputTokens + trace1.usage.outputTokens;
+        const tokens2 = trace2.usage.inputTokens + trace2.usage.outputTokens;
+        return {
+            trace1Id: trace1.id,
+            trace2Id: trace2.id,
+            stepDiffs,
+            commonSteps,
+            divergencePoint,
+            trace1OnlySteps,
+            trace2OnlySteps,
+            metricsDiff: {
+                success: {
+                    trace1: trace1.metrics.success,
+                    trace2: trace2.metrics.success,
+                },
+                score: {
+                    trace1: trace1.score,
+                    trace2: trace2.score,
+                    delta: trace2.score - trace1.score,
+                },
+                tokens: {
+                    trace1: tokens1,
+                    trace2: tokens2,
+                    delta: tokens2 - tokens1,
+                },
+                duration: {
+                    trace1: trace1.duration,
+                    trace2: trace2.duration,
+                    delta: trace2.duration - trace1.duration,
+                },
+            },
+        };
+    }
+    findDivergencePoint(steps1, steps2) {
+        const minLength = Math.min(steps1.length, steps2.length);
+        for (let i = 0; i < minLength; i++) {
+            const comparison = this.compareSteps(steps1[i], steps2[i]);
+            if (comparison.status === 'different') {
+                return i;
+            }
+        }
+        if (steps1.length !== steps2.length) {
+            return minLength;
+        }
+        return undefined;
+    }
+    compareSteps(step1, step2) {
+        const differences = [];
+        if (step1.type !== step2.type) {
+            differences.push(`Type: ${step1.type} → ${step2.type}`);
+        }
+        if (step1.type === 'tool_call' && step2.type === 'tool_call') {
+            if (step1.toolCall?.name !== step2.toolCall?.name) {
+                differences.push(`Tool: ${step1.toolCall?.name} → ${step2.toolCall?.name}`);
+            }
+            if (step1.toolCall && step2.toolCall) {
+                const args1 = JSON.stringify(step1.toolCall.arguments);
+                const args2 = JSON.stringify(step2.toolCall.arguments);
+                if (args1 !== args2) {
+                    differences.push('Tool arguments differ');
+                }
+            }
+            if (step1.toolResult && step2.toolResult) {
+                const resultComparison = this.compareToolResults(step1.toolResult, step2.toolResult);
+                if (!resultComparison.similar) {
+                    differences.push(...resultComparison.differences);
+                }
+            }
+        }
+        if (step1.type === 'llm_call' && step2.type === 'llm_call') {
+            if (step1.response !== step2.response) {
+                differences.push('LLM response differs');
+            }
+        }
+        let status;
+        if (differences.length === 0) {
+            status = 'identical';
+        }
+        else if (differences.length === 1 && differences[0] === 'LLM response differs') {
+            status = 'similar';
+        }
+        else {
+            status = 'different';
+        }
+        return {
+            index: step1.index,
+            status,
+            step1,
+            step2,
+            differences: differences.length > 0 ? differences : undefined,
+        };
+    }
+    compareToolResults(result1, result2) {
+        const differences = [];
+        if (result1.name !== result2.name) {
+            differences.push(`Tool name: ${result1.name} → ${result2.name}`);
+        }
+        if (result1.error !== result2.error) {
+            if (result1.error && !result2.error) {
+                differences.push(`Error removed: ${result1.error}`);
+            }
+            else if (!result1.error && result2.error) {
+                differences.push(`Error added: ${result2.error}`);
+            }
+            else {
+                differences.push(`Error changed: ${result1.error} → ${result2.error}`);
+            }
+        }
+        const resultStr1 = JSON.stringify(result1.result);
+        const resultStr2 = JSON.stringify(result2.result);
+        if (resultStr1 !== resultStr2) {
+            differences.push('Result data differs');
+        }
+        return {
+            similar: differences.length === 0,
+            differences,
+        };
+    }
+    compareMetrics(metrics1, metrics2) {
+        return {
+            toolAccuracy: {
+                value1: metrics1.toolAccuracy,
+                value2: metrics2.toolAccuracy,
+                delta: metrics2.toolAccuracy - metrics1.toolAccuracy,
+            },
+            efficiency: {
+                value1: metrics1.efficiency,
+                value2: metrics2.efficiency,
+                delta: metrics2.efficiency - metrics1.efficiency,
+            },
+            completeness: {
+                value1: metrics1.completeness,
+                value2: metrics2.completeness,
+                delta: metrics2.completeness - metrics1.completeness,
+            },
+        };
+    }
+    formatDiff(diff) {
+        const lines = [];
+        lines.push('═══════════════════════════════════════════');
+        lines.push('              TRACE COMPARISON              ');
+        lines.push('═══════════════════════════════════════════');
+        lines.push('');
+        lines.push(`Trace 1: ${diff.trace1Id}`);
+        lines.push(`Trace 2: ${diff.trace2Id}`);
+        lines.push('');
+        if (diff.divergencePoint !== undefined) {
+            lines.push(`⚠ Traces diverged at step ${diff.divergencePoint}`);
+        }
+        else {
+            lines.push('✓ Traces are identical');
+        }
+        lines.push('');
+        lines.push('─── Summary ───');
+        lines.push(`Common steps:     ${diff.commonSteps}`);
+        lines.push(`Only in trace 1:  ${diff.trace1OnlySteps}`);
+        lines.push(`Only in trace 2:  ${diff.trace2OnlySteps}`);
+        lines.push('');
+        lines.push('─── Metrics ───');
+        lines.push(`Success:  ${diff.metricsDiff.success.trace1} → ${diff.metricsDiff.success.trace2}`);
+        lines.push(`Score:    ${diff.metricsDiff.score.trace1.toFixed(3)} → ${diff.metricsDiff.score.trace2.toFixed(3)} (${this.formatDelta(diff.metricsDiff.score.delta)})`);
+        lines.push(`Tokens:   ${diff.metricsDiff.tokens.trace1} → ${diff.metricsDiff.tokens.trace2} (${this.formatDelta(diff.metricsDiff.tokens.delta)})`);
+        lines.push(`Duration: ${diff.metricsDiff.duration.trace1}ms → ${diff.metricsDiff.duration.trace2}ms (${this.formatDelta(diff.metricsDiff.duration.delta)}ms)`);
+        lines.push('');
+        if (diff.stepDiffs.some((d) => d.status !== 'identical')) {
+            lines.push('─── Step Differences ───');
+            for (const stepDiff of diff.stepDiffs) {
+                if (stepDiff.status !== 'identical') {
+                    const icon = this.getStatusIcon(stepDiff.status);
+                    lines.push(`${icon} Step ${stepDiff.index}: ${stepDiff.status}`);
+                    if (stepDiff.differences) {
+                        for (const d of stepDiff.differences) {
+                            lines.push(`   └─ ${d}`);
+                        }
+                    }
+                }
+            }
+        }
+        lines.push('');
+        lines.push('═══════════════════════════════════════════');
+        return lines.join('\n');
+    }
+    compareStepSequences(steps1, steps2) {
+        const diffs = [];
+        const maxLength = Math.max(steps1.length, steps2.length);
+        for (let i = 0; i < maxLength; i++) {
+            const step1 = steps1[i];
+            const step2 = steps2[i];
+            if (!step1) {
+                diffs.push({
+                    index: i,
+                    status: 'only_in_2',
+                    step2,
+                });
+            }
+            else if (!step2) {
+                diffs.push({
+                    index: i,
+                    status: 'only_in_1',
+                    step1,
+                });
+            }
+            else {
+                diffs.push(this.compareSteps(step1, step2));
+            }
+        }
+        return diffs;
+    }
+    formatDelta(delta) {
+        if (delta > 0)
+            return `+${delta}`;
+        return String(delta);
+    }
+    getStatusIcon(status) {
+        switch (status) {
+            case 'identical':
+                return '✓';
+            case 'similar':
+                return '≈';
+            case 'different':
+                return '✗';
+            case 'only_in_1':
+                return '◀';
+            case 'only_in_2':
+                return '▶';
+            default:
+                return '?';
+        }
+    }
+}
+//# sourceMappingURL=comparator.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"comparator.js","sourceRoot":"","sources":["../../src/time-travel/comparator.ts"],"names":[],"mappings":"AAeA,MAAM,OAAO,eAAe;IAClB,UAAU,CAAa;IAE/B,YAAY,OAA+B;QACzC,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,QAAgB,EAAE,QAAgB;QAC9C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACnD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAEnD,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,oBAAoB,QAAQ,EAAE,CAAC,CAAC;QAClD,CAAC;QACD,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,oBAAoB,QAAQ,EAAE,CAAC,CAAC;QAClD,CAAC;QAED,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC1C,CAAC;IAED,WAAW,CAAC,MAAsB,EAAE,MAAsB;QACxD,MAAM,SAAS,GAAG,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QACxE,MAAM,eAAe,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QAE7E,MAAM,WAAW,GAAG,SAAS,CAAC,MAAM,CAClC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,WAAW,IAAI,CAAC,CAAC,MAAM,KAAK,SAAS,CAC1D,CAAC,MAAM,CAAC;QAET,MAAM,eAAe,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC,MAAM,CAAC;QACjF,MAAM,eAAe,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,WAAW,CAAC,CAAC,MAAM,CAAC;QAEjF,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC;QACrE,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC;QAErE,OAAO;YACL,QAAQ,EAAE,MAAM,CAAC,EAAE;YACnB,QAAQ,EAAE,MAAM,CAAC,EAAE;YACnB,SAAS;YACT,WAAW;YACX,eAAe;YACf,eAAe;YACf,eAAe;YACf,WAAW,EAAE;gBACX,OAAO,EAAE;oBACP,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO;oBAC9B,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO;iBAC/B;gBACD,KAAK,EAAE;oBACL,MAAM,EAAE,MAAM,CAAC,KAAK;oBACpB,MAAM,EAAE,MAAM,CAAC,KAAK;oBACpB,KAAK,EAAE,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK;iBACnC;gBACD,MAAM,EAAE;oBACN,MAAM,EAAE,OAAO;oBACf,MAAM,EAAE,OAAO;oBACf,KAAK,EAAE,OAAO,GAAG,OAAO;iBACzB;gBACD,QAAQ,EAAE;oBACR,MAAM,EAAE,MAAM,CAAC,QAAQ;oBACvB,MAAM,EAAE,MAAM,CAAC,QAAQ;oBACvB,KAAK,EAAE,MAAM,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ;iBACzC;aACF;SACF,CAAC;IACJ,CAAC;IAED,mBAAmB,CAAC,MAAuB,EAAE,MAAuB;QAClE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QAEzD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;YACnC,MAAM,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,IAAI,UAAU,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBACtC,OAAO,CAAC,CAAC;YACX,CAAC;QACH,CAAC;QAED,IAAI,MAAM,CAAC,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;YACpC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,YAAY,CAAC,KAAoB,EAAE,KAAoB;QACrD,MAAM,WAAW,GAAa,EAAE,CAAC;QAEjC,IAAI,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,IAAI,EAAE,CAAC;YAC9B,WAAW,CAAC,IAAI,CAAC,SAAS,KAAK,CAAC,IAAI,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QAC1D,CAAC;QAED,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC7D,IAAI,KAAK,CAAC,QAAQ,EAAE,IAAI,KAAK,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;gBAClD,WAAW,CAAC,IAAI,CAAC,SAAS,KAAK,CAAC,QAAQ,EAAE,IAAI,MAAM,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;YAC9E,CAAC;YAED,IAAI,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACrC,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;gBACvD,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;gBACvD,IAAI,KAAK,KAAK,KAAK,EAAE,CAAC;oBACpB,WAAW,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;gBAC5C,CAAC;YACH,CAAC;YAED,IAAI,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;gBACzC,MAAM,gBAAgB,GAAG,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;gBACrF,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC;oBAC9B,WAAW,CAAC,IAAI,CAAC,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAC;gBACpD,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC3D,IAAI,KAAK,CAAC,QAAQ,KAAK,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACtC,WAAW,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC;QAED,IAAI,MAAsB,CAAC;QAC3B,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,MAAM,GAAG,WAAW,CAAC;QACvB,CAAC;aAAM,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,IAAI,WAAW,CAAC,CAAC,CAAC,KAAK,sBAAsB,EAAE,CAAC;YACjF,MAAM,GAAG,SAAS,CAAC;QACrB,CAAC;aAAM,CAAC;YACN,MAAM,GAAG,WAAW,CAAC;QACvB,CAAC;QAED,OAAO;YACL,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,MAAM;YACN,KAAK;YACL,KAAK;YACL,WAAW,EAAE,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;SAC9D,CAAC;IACJ,CAAC;IAED,kBAAkB,CAChB,OAAmB,EACnB,OAAmB;QAEnB,MAAM,WAAW,GAAa,EAAE,CAAC;QAEjC,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,CAAC,IAAI,EAAE,CAAC;YAClC,WAAW,CAAC,IAAI,CAAC,cAAc,OAAO,CAAC,IAAI,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,OAAO,CAAC,KAAK,KAAK,OAAO,CAAC,KAAK,EAAE,CAAC;YACpC,IAAI,OAAO,CAAC,KAAK,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBACpC,WAAW,CAAC,IAAI,CAAC,kBAAkB,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;YACtD,CAAC;iBAAM,IAAI,CAAC,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBAC3C,WAAW,CAAC,IAAI,CAAC,gBAAgB,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;YACpD,CAAC;iBAAM,CAAC;gBACN,WAAW,CAAC,IAAI,CAAC,kBAAkB,OAAO,CAAC,KAAK,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;YACzE,CAAC;QACH,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAClD,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAClD,IAAI,UAAU,KAAK,UAAU,EAAE,CAAC;YAC9B,WAAW,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC1C,CAAC;QAED,OAAO;YACL,OAAO,EAAE,WAAW,CAAC,MAAM,KAAK,CAAC;YACjC,WAAW;SACZ,CAAC;IACJ,CAAC;IAED,cAAc,CACZ,QAAsB,EACtB,QAAsB;QAEtB,OAAO;YACL,YAAY,EAAE;gBACZ,MAAM,EAAE,QAAQ,CAAC,YAAY;gBAC7B,MAAM,EAAE,QAAQ,CAAC,YAAY;gBAC7B,KAAK,EAAE,QAAQ,CAAC,YAAY,GAAG,QAAQ,CAAC,YAAY;aACrD;YACD,UAAU,EAAE;gBACV,MAAM,EAAE,QAAQ,CAAC,UAAU;gBAC3B,MAAM,EAAE,QAAQ,CAAC,UAAU;gBAC3B,KAAK,EAAE,QAAQ,CAAC,UAAU,GAAG,QAAQ,CAAC,UAAU;aACjD;YACD,YAAY,EAAE;gBACZ,MAAM,EAAE,QAAQ,CAAC,YAAY;gBAC7B,MAAM,EAAE,QAAQ,CAAC,YAAY;gBAC7B,KAAK,EAAE,QAAQ,CAAC,YAAY,GAAG,QAAQ,CAAC,YAAY;aACrD;SACF,CAAC;IACJ,CAAC;IAED,UAAU,CAAC,IAAe;QACxB,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,KAAK,CAAC,IAAI,CAAC,6CAA6C,CAAC,CAAC;QAC1D,KAAK,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC;QAC3D,KAAK,CAAC,IAAI,CAAC,6CAA6C,CAAC,CAAC;QAC1D,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEf,KAAK,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QACxC,KAAK,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QACxC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEf,IAAI,IAAI,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;YACvC,KAAK,CAAC,IAAI,CAAC,6BAA6B,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC;QAClE,CAAC;aAAM,CAAC;YACN,KAAK,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACvC,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEf,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC9B,KAAK,CAAC,IAAI,CAAC,qBAAqB,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QACpD,KAAK,CAAC,IAAI,CAAC,qBAAqB,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC;QACxD,KAAK,CAAC,IAAI,CAAC,qBAAqB,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC;QACxD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEf,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC9B,KAAK,CAAC,IAAI,CAAC,aAAa,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAChG,KAAK,CAAC,IAAI,CACR,aAAa,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAC1J,CAAC;QACF,KAAK,CAAC,IAAI,CACR,aAAa,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,KAAK,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CACvI,CAAC;QACF,KAAK,CAAC,IAAI,CACR,aAAa,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,MAAM,QAAQ,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,MAAM,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CACnJ,CAAC;QACF,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAEf,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,WAAW,CAAC,EAAE,CAAC;YACzD,KAAK,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YACvC,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACtC,IAAI,QAAQ,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;oBACpC,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;oBACjD,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,SAAS,QAAQ,CAAC,KAAK,KAAK,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;oBACjE,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC;wBACzB,KAAK,MAAM,CAAC,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC;4BACrC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;wBAC3B,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,KAAK,CAAC,IAAI,CAAC,6CAA6C,CAAC,CAAC;QAE1D,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAEO,oBAAoB,CAAC,MAAuB,EAAE,MAAuB;QAC3E,MAAM,KAAK,GAAe,EAAE,CAAC;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QAEzD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;YACnC,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YACxB,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YAExB,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,KAAK,CAAC,IAAI,CAAC;oBACT,KAAK,EAAE,CAAC;oBACR,MAAM,EAAE,WAAW;oBACnB,KAAK;iBACN,CAAC,CAAC;YACL,CAAC;iBAAM,IAAI,CAAC,KAAK,EAAE,CAAC;gBAClB,KAAK,CAAC,IAAI,CAAC;oBACT,KAAK,EAAE,CAAC;oBACR,MAAM,EAAE,WAAW;oBACnB,KAAK;iBACN,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,WAAW,CAAC,KAAa;QAC/B,IAAI,KAAK,GAAG,CAAC;YAAE,OAAO,IAAI,KAAK,EAAE,CAAC;QAClC,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;IAEO,aAAa,CAAC,MAAsB;QAC1C,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,WAAW;gBACd,OAAO,GAAG,CAAC;YACb,KAAK,SAAS;gBACZ,OAAO,GAAG,CAAC;YACb,KAAK,WAAW;gBACd,OAAO,GAAG,CAAC;YACb,KAAK,WAAW;gBACd,OAAO,GAAG,CAAC;YACb,KAAK,WAAW;gBACd,OAAO,GAAG,CAAC;YACb;gBACE,OAAO,GAAG,CAAC;QACf,CAAC;IACH,CAAC;CACF"}

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

@@ -0,0 +1,22 @@
+import type { ForkOptions, ForkResult, TimeTravelCheckpointStore } from '@cogitator-ai/types';
+import type { Agent } from '../agent';
+import type { Cogitator } from '../cogitator';
+import { ExecutionReplayer } from './replayer';
+export interface ExecutionForkerOptions {
+    checkpointStore: TimeTravelCheckpointStore;
+    replayer: ExecutionReplayer;
+}
+export declare class ExecutionForker {
+    private checkpointStore;
+    private replayer;
+    constructor(options: ExecutionForkerOptions);
+    fork(cogitator: Cogitator, agent: Agent, options: ForkOptions): Promise<ForkResult>;
+    forkMultiple(cogitator: Cogitator, agent: Agent, checkpointId: string, variants: Array<Partial<ForkOptions>>): Promise<ForkResult[]>;
+    forkWithContext(cogitator: Cogitator, agent: Agent, checkpointId: string, additionalContext: string, label?: string): Promise<ForkResult>;
+    forkWithMockedTools(cogitator: Cogitator, agent: Agent, checkpointId: string, mockResults: Record<string, unknown>, label?: string): Promise<ForkResult>;
+    forkWithNewInput(cogitator: Cogitator, agent: Agent, checkpointId: string, newInput: string, label?: string): Promise<ForkResult>;
+    private createForkedCheckpoint;
+    private injectContext;
+    private replaceInput;
+}
+//# sourceMappingURL=forker.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"forker.d.ts","sourceRoot":"","sources":["../../src/time-travel/forker.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAEV,WAAW,EACX,UAAU,EAEV,yBAAyB,EAC1B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,UAAU,CAAC;AACtC,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAE/C,MAAM,WAAW,sBAAsB;IACrC,eAAe,EAAE,yBAAyB,CAAC;IAC3C,QAAQ,EAAE,iBAAiB,CAAC;CAC7B;AAED,qBAAa,eAAe;IAC1B,OAAO,CAAC,eAAe,CAA4B;IACnD,OAAO,CAAC,QAAQ,CAAoB;gBAExB,OAAO,EAAE,sBAAsB;IAKrC,IAAI,CAAC,SAAS,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC;IA6BnF,YAAY,CAChB,SAAS,EAAE,SAAS,EACpB,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,GACpC,OAAO,CAAC,UAAU,EAAE,CAAC;IAclB,eAAe,CACnB,SAAS,EAAE,SAAS,EACpB,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,iBAAiB,EAAE,MAAM,EACzB,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,UAAU,CAAC;IAQhB,mBAAmB,CACvB,SAAS,EAAE,SAAS,EACpB,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;IAQhB,gBAAgB,CACpB,SAAS,EAAE,SAAS,EACpB,KAAK,EAAE,KAAK,EACZ,YAAY,EAAE,MAAM,EACpB,QAAQ,EAAE,MAAM,EAChB,KAAK,CAAC,EAAE,MAAM,GACb,OAAO,CAAC,UAAU,CAAC;IAQtB,OAAO,CAAC,sBAAsB;IA6B9B,OAAO,CAAC,aAAa;IAmBrB,OAAO,CAAC,YAAY;CAYrB"}