@positronic/core 0.0.2 → 0.0.4

package/dist/src/dsl/brain.test.js

@@ -1,4225 +0,0 @@
-function _array_like_to_array(arr, len) {
-    if (len == null || len > arr.length) len = arr.length;
-    for(var i = 0, arr2 = new Array(len); i < len; i++)arr2[i] = arr[i];
-    return arr2;
-}
-function _array_without_holes(arr) {
-    if (Array.isArray(arr)) return _array_like_to_array(arr);
-}
-function _async_iterator(iterable) {
-    var method, async, sync, retry = 2;
-    for("undefined" != typeof Symbol && (async = Symbol.asyncIterator, sync = Symbol.iterator); retry--;){
-        if (async && null != (method = iterable[async])) return method.call(iterable);
-        if (sync && null != (method = iterable[sync])) return new AsyncFromSyncIterator(method.call(iterable));
-        async = "@@asyncIterator", sync = "@@iterator";
-    }
-    throw new TypeError("Object is not async iterable");
-}
-function AsyncFromSyncIterator(s) {
-    function AsyncFromSyncIteratorContinuation(r) {
-        if (Object(r) !== r) return Promise.reject(new TypeError(r + " is not an object."));
-        var done = r.done;
-        return Promise.resolve(r.value).then(function(value) {
-            return {
-                value: value,
-                done: done
-            };
-        });
-    }
-    return AsyncFromSyncIterator = function(s) {
-        this.s = s, this.n = s.next;
-    }, AsyncFromSyncIterator.prototype = {
-        s: null,
-        n: null,
-        next: function() {
-            return AsyncFromSyncIteratorContinuation(this.n.apply(this.s, arguments));
-        },
-        return: function(value) {
-            var ret = this.s.return;
-            return void 0 === ret ? Promise.resolve({
-                value: value,
-                done: !0
-            }) : AsyncFromSyncIteratorContinuation(ret.apply(this.s, arguments));
-        },
-        throw: function(value) {
-            var thr = this.s.return;
-            return void 0 === thr ? Promise.reject(value) : AsyncFromSyncIteratorContinuation(thr.apply(this.s, arguments));
-        }
-    }, new AsyncFromSyncIterator(s);
-}
-function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) {
-    try {
-        var info = gen[key](arg);
-        var value = info.value;
-    } catch (error) {
-        reject(error);
-        return;
-    }
-    if (info.done) {
-        resolve(value);
-    } else {
-        Promise.resolve(value).then(_next, _throw);
-    }
-}
-function _async_to_generator(fn) {
-    return function() {
-        var self = this, args = arguments;
-        return new Promise(function(resolve, reject) {
-            var gen = fn.apply(self, args);
-            function _next(value) {
-                asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value);
-            }
-            function _throw(err) {
-                asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err);
-            }
-            _next(undefined);
-        });
-    };
-}
-function _define_property(obj, key, value) {
-    if (key in obj) {
-        Object.defineProperty(obj, key, {
-            value: value,
-            enumerable: true,
-            configurable: true,
-            writable: true
-        });
-    } else {
-        obj[key] = value;
-    }
-    return obj;
-}
-function _iterable_to_array(iter) {
-    if (typeof Symbol !== "undefined" && iter[Symbol.iterator] != null || iter["@@iterator"] != null) return Array.from(iter);
-}
-function _non_iterable_spread() {
-    throw new TypeError("Invalid attempt to spread non-iterable instance.\\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method.");
-}
-function _object_spread(target) {
-    for(var i = 1; i < arguments.length; i++){
-        var source = arguments[i] != null ? arguments[i] : {};
-        var ownKeys = Object.keys(source);
-        if (typeof Object.getOwnPropertySymbols === "function") {
-            ownKeys = ownKeys.concat(Object.getOwnPropertySymbols(source).filter(function(sym) {
-                return Object.getOwnPropertyDescriptor(source, sym).enumerable;
-            }));
-        }
-        ownKeys.forEach(function(key) {
-            _define_property(target, key, source[key]);
-        });
-    }
-    return target;
-}
-function ownKeys(object, enumerableOnly) {
-    var keys = Object.keys(object);
-    if (Object.getOwnPropertySymbols) {
-        var symbols = Object.getOwnPropertySymbols(object);
-        if (enumerableOnly) {
-            symbols = symbols.filter(function(sym) {
-                return Object.getOwnPropertyDescriptor(object, sym).enumerable;
-            });
-        }
-        keys.push.apply(keys, symbols);
-    }
-    return keys;
-}
-function _object_spread_props(target, source) {
-    source = source != null ? source : {};
-    if (Object.getOwnPropertyDescriptors) {
-        Object.defineProperties(target, Object.getOwnPropertyDescriptors(source));
-    } else {
-        ownKeys(Object(source)).forEach(function(key) {
-            Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key));
-        });
-    }
-    return target;
-}
-function _to_consumable_array(arr) {
-    return _array_without_holes(arr) || _iterable_to_array(arr) || _unsupported_iterable_to_array(arr) || _non_iterable_spread();
-}
-function _unsupported_iterable_to_array(o, minLen) {
-    if (!o) return;
-    if (typeof o === "string") return _array_like_to_array(o, minLen);
-    var n = Object.prototype.toString.call(o).slice(8, -1);
-    if (n === "Object" && o.constructor) n = o.constructor.name;
-    if (n === "Map" || n === "Set") return Array.from(n);
-    if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _array_like_to_array(o, minLen);
-}
-function _ts_generator(thisArg, body) {
-    var f, y, t, _ = {
-        label: 0,
-        sent: function() {
-            if (t[0] & 1) throw t[1];
-            return t[1];
-        },
-        trys: [],
-        ops: []
-    }, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
-    return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() {
-        return this;
-    }), g;
-    function verb(n) {
-        return function(v) {
-            return step([
-                n,
-                v
-            ]);
-        };
-    }
-    function step(op) {
-        if (f) throw new TypeError("Generator is already executing.");
-        while(g && (g = 0, op[0] && (_ = 0)), _)try {
-            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
-            if (y = 0, t) op = [
-                op[0] & 2,
-                t.value
-            ];
-            switch(op[0]){
-                case 0:
-                case 1:
-                    t = op;
-                    break;
-                case 4:
-                    _.label++;
-                    return {
-                        value: op[1],
-                        done: false
-                    };
-                case 5:
-                    _.label++;
-                    y = op[1];
-                    op = [
-                        0
-                    ];
-                    continue;
-                case 7:
-                    op = _.ops.pop();
-                    _.trys.pop();
-                    continue;
-                default:
-                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) {
-                        _ = 0;
-                        continue;
-                    }
-                    if (op[0] === 3 && (!t || op[1] > t[0] && op[1] < t[3])) {
-                        _.label = op[1];
-                        break;
-                    }
-                    if (op[0] === 6 && _.label < t[1]) {
-                        _.label = t[1];
-                        t = op;
-                        break;
-                    }
-                    if (t && _.label < t[2]) {
-                        _.label = t[2];
-                        _.ops.push(op);
-                        break;
-                    }
-                    if (t[2]) _.ops.pop();
-                    _.trys.pop();
-                    continue;
-            }
-            op = body.call(thisArg, _);
-        } catch (e) {
-            op = [
-                6,
-                e
-            ];
-            y = 0;
-        } finally{
-            f = t = 0;
-        }
-        if (op[0] & 5) throw op[1];
-        return {
-            value: op[0] ? op[1] : void 0,
-            done: true
-        };
-    }
-}
-import { BRAIN_EVENTS, STATUS } from './constants.js';
-import { applyPatches } from './json-patch.js';
-import { brain } from './brain.js';
-import { z } from 'zod';
-import { jest } from '@jest/globals';
-import { createResources } from '../resources/resources.js';
-// Helper function to get the next value from an AsyncIterator
-var nextStep = function(brainRun) {
-    return _async_to_generator(function() {
-        var result;
-        return _ts_generator(this, function(_state) {
-            switch(_state.label){
-                case 0:
-                    return [
-                        4,
-                        brainRun.next()
-                    ];
-                case 1:
-                    result = _state.sent();
-                    if (result.done) throw new Error('Iterator is done');
-                    return [
-                        2,
-                        result.value
-                    ];
-            }
-        });
-    })();
-};
-// Mock services for testing
-var testLogger = {
-    log: jest.fn()
-};
-// Mock ObjectGenerator for testing
-var mockGenerateObject = jest.fn();
-var mockClient = {
-    generateObject: mockGenerateObject
-};
-// Mock Resources for testing
-var mockResourceLoad = jest.fn(function(resourceName, type) {
-    return _async_to_generator(function() {
-        return _ts_generator(this, function(_state) {
-            if (type === 'binary') return [
-                2,
-                Buffer.from("mock ".concat(resourceName, " binary content"))
-            ];
-            return [
-                2,
-                "mock ".concat(resourceName, " text content")
-            ];
-        });
-    })();
-});
-var mockResourceLoader = {
-    load: mockResourceLoad
-};
-var testManifest = {
-    myFile: {
-        type: 'text',
-        key: 'myFile',
-        path: '/test/myFile.txt'
-    },
-    myBinaryFile: {
-        type: 'binary',
-        key: 'myBinaryFile',
-        path: '/test/myBinaryFile.bin'
-    },
-    nested: {
-        anotherFile: {
-            type: 'text',
-            key: 'anotherFile',
-            path: '/test/anotherFile.txt'
-        }
-    }
-};
-var mockResources = createResources(mockResourceLoader, testManifest);
-describe('brain creation', function() {
-    beforeEach(function() {
-        mockGenerateObject.mockClear();
-        mockResourceLoad.mockClear();
-    });
-    it('should create a brain with steps and run through them', function() {
-        return _async_to_generator(function() {
-            var testBrain, brainRun, startResult, firstStepStartResult, firstStepStatusRunning, firstStepResult, secondStepStartResult, secondStepStatusRunning, secondStepResult, stepStatusResult, completeResult;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        testBrain = brain('test brain').step('First step', function() {
-                            return {
-                                count: 1
-                            };
-                        }).step('Second step', function(param) {
-                            var state = param.state;
-                            return _object_spread_props(_object_spread({}, state), {
-                                doubled: state.count * 2
-                            });
-                        });
-                        brainRun = testBrain.run({
-                            client: mockClient
-                        });
-                        return [
-                            4,
-                            brainRun.next()
-                        ];
-                    case 1:
-                        startResult = _state.sent();
-                        expect(startResult.value).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.START,
-                            status: STATUS.RUNNING,
-                            brainTitle: 'test brain',
-                            brainDescription: undefined
-                        }));
-                        // Skip initial step status event
-                        return [
-                            4,
-                            nextStep(brainRun)
-                        ];
-                    case 2:
-                        _state.sent();
-                        return [
-                            4,
-                            nextStep(brainRun)
-                        ];
-                    case 3:
-                        firstStepStartResult = _state.sent();
-                        expect(firstStepStartResult).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.STEP_START,
-                            status: STATUS.RUNNING,
-                            stepTitle: 'First step',
-                            stepId: expect.any(String)
-                        }));
-                        return [
-                            4,
-                            nextStep(brainRun)
-                        ];
-                    case 4:
-                        firstStepStatusRunning = _state.sent();
-                        expect(firstStepStatusRunning).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.STEP_STATUS,
-                            steps: expect.any(Array)
-                        }));
-                        if (firstStepStatusRunning.type === BRAIN_EVENTS.STEP_STATUS) {
-                            expect(firstStepStatusRunning.steps[0].status).toBe(STATUS.RUNNING);
-                        }
-                        return [
-                            4,
-                            nextStep(brainRun)
-                        ];
-                    case 5:
-                        firstStepResult = _state.sent();
-                        expect(firstStepResult).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.STEP_COMPLETE,
-                            status: STATUS.RUNNING,
-                            stepTitle: 'First step',
-                            stepId: expect.any(String),
-                            patch: [
-                                {
-                                    op: 'add',
-                                    path: '/count',
-                                    value: 1
-                                }
-                            ]
-                        }));
-                        // Step Status Event
-                        return [
-                            4,
-                            nextStep(brainRun)
-                        ];
-                    case 6:
-                        _state.sent();
-                        return [
-                            4,
-                            nextStep(brainRun)
-                        ];
-                    case 7:
-                        secondStepStartResult = _state.sent();
-                        expect(secondStepStartResult).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.STEP_START,
-                            status: STATUS.RUNNING,
-                            stepTitle: 'Second step',
-                            stepId: expect.any(String)
-                        }));
-                        return [
-                            4,
-                            nextStep(brainRun)
-                        ];
-                    case 8:
-                        secondStepStatusRunning = _state.sent();
-                        expect(secondStepStatusRunning).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.STEP_STATUS,
-                            steps: expect.any(Array)
-                        }));
-                        if (secondStepStatusRunning.type === BRAIN_EVENTS.STEP_STATUS) {
-                            expect(secondStepStatusRunning.steps[1].status).toBe(STATUS.RUNNING);
-                        }
-                        return [
-                            4,
-                            nextStep(brainRun)
-                        ];
-                    case 9:
-                        secondStepResult = _state.sent();
-                        expect(secondStepResult).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.STEP_COMPLETE,
-                            stepTitle: 'Second step',
-                            stepId: expect.any(String),
-                            patch: [
-                                {
-                                    op: 'add',
-                                    path: '/doubled',
-                                    value: 2
-                                }
-                            ]
-                        }));
-                        return [
-                            4,
-                            nextStep(brainRun)
-                        ];
-                    case 10:
-                        stepStatusResult = _state.sent();
-                        expect(stepStatusResult).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.STEP_STATUS,
-                            steps: [
-                                expect.objectContaining({
-                                    title: 'First step',
-                                    status: STATUS.COMPLETE,
-                                    id: expect.any(String)
-                                }),
-                                expect.objectContaining({
-                                    title: 'Second step',
-                                    status: STATUS.COMPLETE,
-                                    id: expect.any(String)
-                                })
-                            ]
-                        }));
-                        return [
-                            4,
-                            nextStep(brainRun)
-                        ];
-                    case 11:
-                        completeResult = _state.sent();
-                        expect(completeResult).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.COMPLETE,
-                            status: STATUS.COMPLETE,
-                            brainTitle: 'test brain',
-                            brainDescription: undefined
-                        }));
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-    it('should create a brain with a name and description when passed an object', function() {
-        return _async_to_generator(function() {
-            var testBrain, brainRun, startResult;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        testBrain = brain({
-                            title: 'my named brain',
-                            description: 'some description'
-                        });
-                        brainRun = testBrain.run({
-                            client: mockClient
-                        });
-                        return [
-                            4,
-                            brainRun.next()
-                        ];
-                    case 1:
-                        startResult = _state.sent();
-                        expect(startResult.value).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.START,
-                            status: STATUS.RUNNING,
-                            brainTitle: 'my named brain',
-                            brainDescription: 'some description',
-                            options: {}
-                        }));
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-    it('should create a brain with just a name when passed a string', function() {
-        return _async_to_generator(function() {
-            var testBrain, brainRun, startResult;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        testBrain = brain('simple brain');
-                        brainRun = testBrain.run({
-                            client: mockClient
-                        });
-                        return [
-                            4,
-                            brainRun.next()
-                        ];
-                    case 1:
-                        startResult = _state.sent();
-                        expect(startResult.value).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.START,
-                            status: STATUS.RUNNING,
-                            brainTitle: 'simple brain',
-                            brainDescription: undefined,
-                            options: {}
-                        }));
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-    it('should allow overriding client per step', function() {
-        return _async_to_generator(function() {
-            var overrideClient, testBrain, events, finalState, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        overrideClient = {
-                            generateObject: jest.fn().mockResolvedValue({
-                                override: true
-                            })
-                        };
-                        // Make sure that for the default prompt the default client returns a known value.
-                        mockClient.generateObject.mockResolvedValueOnce({
-                            override: false
-                        });
-                        testBrain = brain('Client Override Test').prompt('Use default client', {
-                            template: function() {
-                                return 'prompt1';
-                            },
-                            outputSchema: {
-                                schema: z.object({
-                                    override: z.boolean()
-                                }),
-                                name: 'overrideResponse'
-                            }
-                        }).prompt('Use override client', {
-                            template: function() {
-                                return 'prompt2';
-                            },
-                            outputSchema: {
-                                schema: z.object({
-                                    override: z.boolean()
-                                }),
-                                name: 'overrideResponse'
-                            },
-                            client: overrideClient
-                        });
-                        // Run the brain and capture all events
-                        events = [];
-                        finalState = {};
-                        _iteratorAbruptCompletion = false, _didIteratorError = false;
-                        _state.label = 1;
-                    case 1:
-                        _state.trys.push([
-                            1,
-                            6,
-                            7,
-                            12
-                        ]);
-                        _iterator = _async_iterator(testBrain.run({
-                            client: mockClient
-                        }));
-                        _state.label = 2;
-                    case 2:
-                        return [
-                            4,
-                            _iterator.next()
-                        ];
-                    case 3:
-                        if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                            3,
-                            5
-                        ];
-                        _value = _step.value;
-                        event = _value;
-                        events.push(event);
-                        if (event.type === BRAIN_EVENTS.STEP_COMPLETE) {
-                            finalState = applyPatches(finalState, [
-                                event.patch
-                            ]);
-                        }
-                        _state.label = 4;
-                    case 4:
-                        _iteratorAbruptCompletion = false;
-                        return [
-                            3,
-                            2
-                        ];
-                    case 5:
-                        return [
-                            3,
-                            12
-                        ];
-                    case 6:
-                        err = _state.sent();
-                        _didIteratorError = true;
-                        _iteratorError = err;
-                        return [
-                            3,
-                            12
-                        ];
-                    case 7:
-                        _state.trys.push([
-                            7,
-                            ,
-                            10,
-                            11
-                        ]);
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                            3,
-                            9
-                        ];
-                        return [
-                            4,
-                            _iterator.return()
-                        ];
-                    case 8:
-                        _state.sent();
-                        _state.label = 9;
-                    case 9:
-                        return [
-                            3,
-                            11
-                        ];
-                    case 10:
-                        if (_didIteratorError) {
-                            throw _iteratorError;
-                        }
-                        return [
-                            7
-                        ];
-                    case 11:
-                        return [
-                            7
-                        ];
-                    case 12:
-                        // Final state should include both responses
-                        expect(finalState).toEqual({
-                            overrideResponse: {
-                                override: true
-                            }
-                        });
-                        // Verify that each client was used correctly based on the supplied prompt configuration.
-                        expect(mockClient.generateObject).toHaveBeenCalledWith({
-                            schema: expect.any(z.ZodObject),
-                            schemaName: 'overrideResponse',
-                            prompt: 'prompt1'
-                        });
-                        expect(overrideClient.generateObject).toHaveBeenCalledWith({
-                            schema: expect.any(z.ZodObject),
-                            schemaName: 'overrideResponse',
-                            prompt: 'prompt2'
-                        });
-                        return [
-                            2
-                        ];
-                }
-            });
-        // Verify that the state was updated correctly with values from both clients.
-        })();
-    });
-    it('should use the provided brainRunId for the initial run if supplied', function() {
-        return _async_to_generator(function() {
-            var testBrain, providedId, brainRun, startResult;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        testBrain = brain('Brain with Provided ID');
-                        providedId = 'my-custom-run-id-123';
-                        brainRun = testBrain.run({
-                            client: mockClient,
-                            brainRunId: providedId
-                        });
-                        return [
-                            4,
-                            brainRun.next()
-                        ];
-                    case 1:
-                        startResult = _state.sent();
-                        expect(startResult.value).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.START,
-                            status: STATUS.RUNNING,
-                            brainTitle: 'Brain with Provided ID',
-                            brainRunId: providedId
-                        }));
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-    describe('Resource Availability in Steps', function() {
-        it('should make resources available in a simple step action', function() {
-            return _async_to_generator(function() {
-                var loadedText, testBrain, run, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, _, err;
-                return _ts_generator(this, function(_state) {
-                    switch(_state.label){
-                        case 0:
-                            testBrain = brain('Resource Step Test').step('Load My File', function(param) {
-                                var resources = param.resources;
-                                return _async_to_generator(function() {
-                                    return _ts_generator(this, function(_state) {
-                                        switch(_state.label){
-                                            case 0:
-                                                return [
-                                                    4,
-                                                    resources.myFile.loadText()
-                                                ];
-                                            case 1:
-                                                loadedText = _state.sent();
-                                                return [
-                                                    2,
-                                                    {
-                                                        loadedText: loadedText
-                                                    }
-                                                ];
-                                        }
-                                    });
-                                })();
-                            });
-                            run = testBrain.run({
-                                client: mockClient,
-                                resources: mockResources
-                            });
-                            _iteratorAbruptCompletion = false, _didIteratorError = false;
-                            _state.label = 1;
-                        case 1:
-                            _state.trys.push([
-                                1,
-                                6,
-                                7,
-                                12
-                            ]);
-                            _iterator = _async_iterator(run);
-                            _state.label = 2;
-                        case 2:
-                            return [
-                                4,
-                                _iterator.next()
-                            ];
-                        case 3:
-                            if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                                3,
-                                5
-                            ];
-                            _value = _step.value;
-                            _ = _value;
-                            _state.label = 4;
-                        case 4:
-                            _iteratorAbruptCompletion = false;
-                            return [
-                                3,
-                                2
-                            ];
-                        case 5:
-                            return [
-                                3,
-                                12
-                            ];
-                        case 6:
-                            err = _state.sent();
-                            _didIteratorError = true;
-                            _iteratorError = err;
-                            return [
-                                3,
-                                12
-                            ];
-                        case 7:
-                            _state.trys.push([
-                                7,
-                                ,
-                                10,
-                                11
-                            ]);
-                            if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                                3,
-                                9
-                            ];
-                            return [
-                                4,
-                                _iterator.return()
-                            ];
-                        case 8:
-                            _state.sent();
-                            _state.label = 9;
-                        case 9:
-                            return [
-                                3,
-                                11
-                            ];
-                        case 10:
-                            if (_didIteratorError) {
-                                throw _iteratorError;
-                            }
-                            return [
-                                7
-                            ];
-                        case 11:
-                            return [
-                                7
-                            ];
-                        case 12:
-                            expect(mockResourceLoad).toHaveBeenCalledWith('myFile', 'text');
-                            expect(loadedText).toBe('mock myFile text content');
-                            return [
-                                2
-                            ];
-                    }
-                });
-            })();
-        });
-        it('should make resources available in a prompt action and reducer', function() {
-            return _async_to_generator(function() {
-                var promptTemplateString, reduceContent, testBrain, run, finalState, allPatches, initialStateForReconstruction, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err;
-                return _ts_generator(this, function(_state) {
-                    switch(_state.label){
-                        case 0:
-                            mockGenerateObject.mockResolvedValue({
-                                summary: 'Generated summary'
-                            });
-                            testBrain = brain('Resource Prompt Test').step('Load File For Prompt', function(param) {
-                                var resources = param.resources;
-                                return _async_to_generator(function() {
-                                    var fileContent;
-                                    return _ts_generator(this, function(_state) {
-                                        switch(_state.label){
-                                            case 0:
-                                                return [
-                                                    4,
-                                                    resources.myFile.loadText()
-                                                ];
-                                            case 1:
-                                                fileContent = _state.sent();
-                                                return [
-                                                    2,
-                                                    {
-                                                        fileContentForPrompt: fileContent
-                                                    }
-                                                ];
-                                        }
-                                    });
-                                })();
-                            }).prompt('Summarize File', {
-                                template: function(state) {
-                                    return 'Summarize: ' + state.fileContentForPrompt;
-                                },
-                                outputSchema: {
-                                    schema: z.object({
-                                        summary: z.string()
-                                    }),
-                                    name: 'promptResult'
-                                }
-                            }, function(param) {
-                                var state = param.state, response = param.response, resources = param.resources;
-                                return _async_to_generator(function() {
-                                    return _ts_generator(this, function(_state) {
-                                        switch(_state.label){
-                                            case 0:
-                                                return [
-                                                    4,
-                                                    resources.nested.anotherFile.loadText()
-                                                ];
-                                            case 1:
-                                                reduceContent = _state.sent();
-                                                return [
-                                                    2,
-                                                    _object_spread_props(_object_spread({}, state), {
-                                                        reducedResult: response.summary + ' based on ' + reduceContent
-                                                    })
-                                                ];
-                                        }
-                                    });
-                                })();
-                            });
-                            run = testBrain.run({
-                                client: mockClient,
-                                resources: mockResources,
-                                options: {
-                                    myOption: 'test'
-                                }
-                            });
-                            finalState = {};
-                            allPatches = [];
-                            initialStateForReconstruction = {};
-                            _iteratorAbruptCompletion = false, _didIteratorError = false;
-                            _state.label = 1;
-                        case 1:
-                            _state.trys.push([
-                                1,
-                                6,
-                                7,
-                                12
-                            ]);
-                            _iterator = _async_iterator(run);
-                            _state.label = 2;
-                        case 2:
-                            return [
-                                4,
-                                _iterator.next()
-                            ];
-                        case 3:
-                            if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                                3,
-                                5
-                            ];
-                            _value = _step.value;
-                            event = _value;
-                            if (event.type === BRAIN_EVENTS.STEP_COMPLETE && event.patch) {
-                                allPatches.push(event.patch);
-                            }
-                            _state.label = 4;
-                        case 4:
-                            _iteratorAbruptCompletion = false;
-                            return [
-                                3,
-                                2
-                            ];
-                        case 5:
-                            return [
-                                3,
-                                12
-                            ];
-                        case 6:
-                            err = _state.sent();
-                            _didIteratorError = true;
-                            _iteratorError = err;
-                            return [
-                                3,
-                                12
-                            ];
-                        case 7:
-                            _state.trys.push([
-                                7,
-                                ,
-                                10,
-                                11
-                            ]);
-                            if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                                3,
-                                9
-                            ];
-                            return [
-                                4,
-                                _iterator.return()
-                            ];
-                        case 8:
-                            _state.sent();
-                            _state.label = 9;
-                        case 9:
-                            return [
-                                3,
-                                11
-                            ];
-                        case 10:
-                            if (_didIteratorError) {
-                                throw _iteratorError;
-                            }
-                            return [
-                                7
-                            ];
-                        case 11:
-                            return [
-                                7
-                            ];
-                        case 12:
-                            finalState = applyPatches(initialStateForReconstruction, allPatches);
-                            expect(mockResourceLoad).toHaveBeenCalledWith('myFile', 'text');
-                            expect(mockResourceLoad).toHaveBeenCalledWith('anotherFile', 'text');
-                            expect(reduceContent).toBe('mock anotherFile text content');
-                            expect(finalState.reducedResult).toBe('Generated summary based on mock anotherFile text content');
-                            return [
-                                2
-                            ];
-                    }
-                });
-            })();
-        });
-        it('should pass resources to prompt template function', function() {
-            return _async_to_generator(function() {
-                var testBrain, run, finalState, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err;
-                return _ts_generator(this, function(_state) {
-                    switch(_state.label){
-                        case 0:
-                            testBrain = brain('Resource Prompt Template Test').prompt('Generate Summary', {
-                                template: function(state, resources) {
-                                    return _async_to_generator(function() {
-                                        var templateContent;
-                                        return _ts_generator(this, function(_state) {
-                                            switch(_state.label){
-                                                case 0:
-                                                    return [
-                                                        4,
-                                                        resources.myFile.loadText()
-                                                    ];
-                                                case 1:
-                                                    templateContent = _state.sent();
-                                                    return [
-                                                        2,
-                                                        "Generate a summary for: ".concat(templateContent)
-                                                    ];
-                                            }
-                                        });
-                                    })();
-                                },
-                                outputSchema: {
-                                    schema: z.object({
-                                        summary: z.string()
-                                    }),
-                                    name: 'promptResult'
-                                }
-                            });
-                            mockGenerateObject.mockResolvedValue({
-                                summary: 'Test summary'
-                            });
-                            run = testBrain.run({
-                                client: mockClient,
-                                resources: mockResources
-                            });
-                            finalState = {};
-                            _iteratorAbruptCompletion = false, _didIteratorError = false;
-                            _state.label = 1;
-                        case 1:
-                            _state.trys.push([
-                                1,
-                                6,
-                                7,
-                                12
-                            ]);
-                            _iterator = _async_iterator(run);
-                            _state.label = 2;
-                        case 2:
-                            return [
-                                4,
-                                _iterator.next()
-                            ];
-                        case 3:
-                            if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                                3,
-                                5
-                            ];
-                            _value = _step.value;
-                            event = _value;
-                            if (event.type === BRAIN_EVENTS.STEP_COMPLETE) {
-                                finalState = applyPatches(finalState, [
-                                    event.patch
-                                ]);
-                            }
-                            _state.label = 4;
-                        case 4:
-                            _iteratorAbruptCompletion = false;
-                            return [
-                                3,
-                                2
-                            ];
-                        case 5:
-                            return [
-                                3,
-                                12
-                            ];
-                        case 6:
-                            err = _state.sent();
-                            _didIteratorError = true;
-                            _iteratorError = err;
-                            return [
-                                3,
-                                12
-                            ];
-                        case 7:
-                            _state.trys.push([
-                                7,
-                                ,
-                                10,
-                                11
-                            ]);
-                            if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                                3,
-                                9
-                            ];
-                            return [
-                                4,
-                                _iterator.return()
-                            ];
-                        case 8:
-                            _state.sent();
-                            _state.label = 9;
-                        case 9:
-                            return [
-                                3,
-                                11
-                            ];
-                        case 10:
-                            if (_didIteratorError) {
-                                throw _iteratorError;
-                            }
-                            return [
-                                7
-                            ];
-                        case 11:
-                            return [
-                                7
-                            ];
-                        case 12:
-                            // Verify resource was loaded in template
-                            expect(mockResourceLoad).toHaveBeenCalledWith('myFile', 'text');
-                            // Verify the generated prompt included the resource content
-                            expect(mockGenerateObject).toHaveBeenCalledWith(expect.objectContaining({
-                                prompt: 'Generate a summary for: mock myFile text content'
-                            }));
-                            // Verify final state
-                            expect(finalState).toEqual({
-                                promptResult: {
-                                    summary: 'Test summary'
-                                }
-                            });
-                            return [
-                                2
-                            ];
-                    }
-                });
-            })();
-        });
-        it('templates can use state', function() {
-            return _async_to_generator(function() {
-                var testBrain, run, finalState, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err;
-                return _ts_generator(this, function(_state) {
-                    switch(_state.label){
-                        case 0:
-                            testBrain = brain('State Template Test').step('Set Data', function() {
-                                return {
-                                    existingData: 'legacy data'
-                                };
-                            }).prompt('Analyze Data', {
-                                template: function(state) {
-                                    return "Analyze this: ".concat(state.existingData);
-                                },
-                                outputSchema: {
-                                    schema: z.object({
-                                        analysis: z.string()
-                                    }),
-                                    name: 'promptResult'
-                                }
-                            });
-                            mockGenerateObject.mockResolvedValue({
-                                analysis: 'Analysis result'
-                            });
-                            run = testBrain.run({
-                                client: mockClient,
-                                resources: mockResources
-                            });
-                            finalState = {};
-                            _iteratorAbruptCompletion = false, _didIteratorError = false;
-                            _state.label = 1;
-                        case 1:
-                            _state.trys.push([
-                                1,
-                                6,
-                                7,
-                                12
-                            ]);
-                            _iterator = _async_iterator(run);
-                            _state.label = 2;
-                        case 2:
-                            return [
-                                4,
-                                _iterator.next()
-                            ];
-                        case 3:
-                            if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                                3,
-                                5
-                            ];
-                            _value = _step.value;
-                            event = _value;
-                            if (event.type === BRAIN_EVENTS.STEP_COMPLETE) {
-                                finalState = applyPatches(finalState, [
-                                    event.patch
-                                ]);
-                            }
-                            _state.label = 4;
-                        case 4:
-                            _iteratorAbruptCompletion = false;
-                            return [
-                                3,
-                                2
-                            ];
-                        case 5:
-                            return [
-                                3,
-                                12
-                            ];
-                        case 6:
-                            err = _state.sent();
-                            _didIteratorError = true;
-                            _iteratorError = err;
-                            return [
-                                3,
-                                12
-                            ];
-                        case 7:
-                            _state.trys.push([
-                                7,
-                                ,
-                                10,
-                                11
-                            ]);
-                            if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                                3,
-                                9
-                            ];
-                            return [
-                                4,
-                                _iterator.return()
-                            ];
-                        case 8:
-                            _state.sent();
-                            _state.label = 9;
-                        case 9:
-                            return [
-                                3,
-                                11
-                            ];
-                        case 10:
-                            if (_didIteratorError) {
-                                throw _iteratorError;
-                            }
-                            return [
-                                7
-                            ];
-                        case 11:
-                            return [
-                                7
-                            ];
-                        case 12:
-                            // Verify the prompt was generated correctly
-                            expect(mockGenerateObject).toHaveBeenCalledWith(expect.objectContaining({
-                                prompt: 'Analyze this: legacy data'
-                            }));
-                            // Verify final state
-                            expect(finalState).toEqual({
-                                existingData: 'legacy data',
-                                promptResult: {
-                                    analysis: 'Analysis result'
-                                }
-                            });
-                            // Verify no resources were loaded (since template didn't use them)
-                            expect(mockResourceLoad).not.toHaveBeenCalled();
-                            return [
-                                2
-                            ];
-                    }
-                });
-            })();
-        });
-        it('should make resources available in a nested brain step', function() {
-            return _async_to_generator(function() {
-                var nestedLoadedText, innerBrain, outerBrain, run, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, _, err;
-                return _ts_generator(this, function(_state) {
-                    switch(_state.label){
-                        case 0:
-                            innerBrain = brain('Inner Resource Brain').step('Inner Load Step', function(param) {
-                                var resources = param.resources;
-                                return _async_to_generator(function() {
-                                    return _ts_generator(this, function(_state) {
-                                        switch(_state.label){
-                                            case 0:
-                                                return [
-                                                    4,
-                                                    resources.myBinaryFile.loadBinary().then(function(b) {
-                                                        return b.toString();
-                                                    })
-                                                ];
-                                            case 1:
-                                                nestedLoadedText = _state.sent();
-                                                return [
-                                                    2,
-                                                    {
-                                                        nestedLoadedText: nestedLoadedText
-                                                    }
-                                                ];
-                                        }
-                                    });
-                                })();
-                            });
-                            outerBrain = brain('Outer Resource Brain').brain('Run Inner', innerBrain, function(param) {
-                                var state = param.state, brainState = param.brainState;
-                                return _object_spread({}, state, brainState);
-                            });
-                            run = outerBrain.run({
-                                client: mockClient,
-                                resources: mockResources
-                            });
-                            _iteratorAbruptCompletion = false, _didIteratorError = false;
-                            _state.label = 1;
-                        case 1:
-                            _state.trys.push([
-                                1,
-                                6,
-                                7,
-                                12
-                            ]);
-                            _iterator = _async_iterator(run);
-                            _state.label = 2;
-                        case 2:
-                            return [
-                                4,
-                                _iterator.next()
-                            ];
-                        case 3:
-                            if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                                3,
-                                5
-                            ];
-                            _value = _step.value;
-                            _ = _value;
-                            _state.label = 4;
-                        case 4:
-                            _iteratorAbruptCompletion = false;
-                            return [
-                                3,
-                                2
-                            ];
-                        case 5:
-                            return [
-                                3,
-                                12
-                            ];
-                        case 6:
-                            err = _state.sent();
-                            _didIteratorError = true;
-                            _iteratorError = err;
-                            return [
-                                3,
-                                12
-                            ];
-                        case 7:
-                            _state.trys.push([
-                                7,
-                                ,
-                                10,
-                                11
-                            ]);
-                            if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                                3,
-                                9
-                            ];
-                            return [
-                                4,
-                                _iterator.return()
-                            ];
-                        case 8:
-                            _state.sent();
-                            _state.label = 9;
-                        case 9:
-                            return [
-                                3,
-                                11
-                            ];
-                        case 10:
-                            if (_didIteratorError) {
-                                throw _iteratorError;
-                            }
-                            return [
-                                7
-                            ];
-                        case 11:
-                            return [
-                                7
-                            ];
-                        case 12:
-                            expect(mockResourceLoad).toHaveBeenCalledWith('myBinaryFile', 'binary');
-                            expect(nestedLoadedText).toBe('mock myBinaryFile binary content');
-                            return [
-                                2
-                            ];
-                    }
-                });
-            })();
-        });
-    });
-});
-describe('error handling', function() {
-    it('should handle errors in actions and maintain correct status', function() {
-        return _async_to_generator(function() {
-            var _errorEvent_error, errorBrain, errorEvent, finalStepStatusEvent, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err, error;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        errorBrain = brain('Error Brain')// Step 1: Normal step
-                        .step('First step', function() {
-                            return {
-                                value: 1
-                            };
-                        })// Step 2: Error step
-                        .step('Error step', function() {
-                            if (true) {
-                                throw new Error('Test error');
-                            }
-                            return {
-                                value: 1
-                            };
-                        })// Step 3: Should never execute
-                        .step('Never reached', function(param) {
-                            var state = param.state;
-                            return {
-                                value: state.value + 1
-                            };
-                        });
-                        _state.label = 1;
-                    case 1:
-                        _state.trys.push([
-                            1,
-                            14,
-                            ,
-                            15
-                        ]);
-                        _iteratorAbruptCompletion = false, _didIteratorError = false;
-                        _state.label = 2;
-                    case 2:
-                        _state.trys.push([
-                            2,
-                            7,
-                            8,
-                            13
-                        ]);
-                        _iterator = _async_iterator(errorBrain.run({
-                            client: mockClient
-                        }));
-                        _state.label = 3;
-                    case 3:
-                        return [
-                            4,
-                            _iterator.next()
-                        ];
-                    case 4:
-                        if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                            3,
-                            6
-                        ];
-                        _value = _step.value;
-                        event = _value;
-                        if (event.type === BRAIN_EVENTS.ERROR) {
-                            errorEvent = event;
-                        }
-                        if (event.type === BRAIN_EVENTS.STEP_STATUS) {
-                            finalStepStatusEvent = event;
-                        }
-                        _state.label = 5;
-                    case 5:
-                        _iteratorAbruptCompletion = false;
-                        return [
-                            3,
-                            3
-                        ];
-                    case 6:
-                        return [
-                            3,
-                            13
-                        ];
-                    case 7:
-                        err = _state.sent();
-                        _didIteratorError = true;
-                        _iteratorError = err;
-                        return [
-                            3,
-                            13
-                        ];
-                    case 8:
-                        _state.trys.push([
-                            8,
-                            ,
-                            11,
-                            12
-                        ]);
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                            3,
-                            10
-                        ];
-                        return [
-                            4,
-                            _iterator.return()
-                        ];
-                    case 9:
-                        _state.sent();
-                        _state.label = 10;
-                    case 10:
-                        return [
-                            3,
-                            12
-                        ];
-                    case 11:
-                        if (_didIteratorError) {
-                            throw _iteratorError;
-                        }
-                        return [
-                            7
-                        ];
-                    case 12:
-                        return [
-                            7
-                        ];
-                    case 13:
-                        return [
-                            3,
-                            15
-                        ];
-                    case 14:
-                        error = _state.sent();
-                        return [
-                            3,
-                            15
-                        ];
-                    case 15:
-                        // Verify final state
-                        expect(errorEvent === null || errorEvent === void 0 ? void 0 : errorEvent.status).toBe(STATUS.ERROR);
-                        expect(errorEvent === null || errorEvent === void 0 ? void 0 : (_errorEvent_error = errorEvent.error) === null || _errorEvent_error === void 0 ? void 0 : _errorEvent_error.message).toBe('Test error');
-                        // Verify steps status
-                        if (!(finalStepStatusEvent === null || finalStepStatusEvent === void 0 ? void 0 : finalStepStatusEvent.steps)) {
-                            throw new Error('Steps not found');
-                        }
-                        expect(finalStepStatusEvent.steps[0].status).toBe(STATUS.COMPLETE);
-                        expect(finalStepStatusEvent.steps[1].status).toBe(STATUS.ERROR);
-                        expect(finalStepStatusEvent.steps[2].status).toBe(STATUS.PENDING);
-                        // Verify error event structure
-                        expect(errorEvent).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.ERROR,
-                            status: STATUS.ERROR,
-                            brainTitle: 'Error Brain',
-                            error: expect.objectContaining({
-                                name: expect.any(String),
-                                message: expect.any(String)
-                            })
-                        }));
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-    it('should handle errors in nested brains and propagate them up', function() {
-        return _async_to_generator(function() {
-            var innerBrain, outerBrain, events, error, mainBrainId, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err, e, innerErrorEvent, outerErrorEvent;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        // Create an inner brain that will throw an error
-                        innerBrain = brain('Failing Inner Brain').step('Throw error', function() {
-                            throw new Error('Inner brain error');
-                        });
-                        // Create outer brain that uses the failing inner brain
-                        outerBrain = brain('Outer Brain').step('First step', function() {
-                            return {
-                                step: 'first'
-                            };
-                        }).brain('Run inner brain', innerBrain, function(param) {
-                            var state = param.state, brainState = param.brainState;
-                            return _object_spread_props(_object_spread({}, state), {
-                                step: 'second',
-                                innerResult: brainState.value
-                            });
-                        }, function() {
-                            return {
-                                value: 5
-                            };
-                        });
-                        events = [];
-                        _state.label = 1;
-                    case 1:
-                        _state.trys.push([
-                            1,
-                            14,
-                            ,
-                            15
-                        ]);
-                        _iteratorAbruptCompletion = false, _didIteratorError = false;
-                        _state.label = 2;
-                    case 2:
-                        _state.trys.push([
-                            2,
-                            7,
-                            8,
-                            13
-                        ]);
-                        _iterator = _async_iterator(outerBrain.run({
-                            client: mockClient
-                        }));
-                        _state.label = 3;
-                    case 3:
-                        return [
-                            4,
-                            _iterator.next()
-                        ];
-                    case 4:
-                        if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                            3,
-                            6
-                        ];
-                        _value = _step.value;
-                        event = _value;
-                        events.push(event);
-                        if (event.type === BRAIN_EVENTS.START && !mainBrainId) {
-                            mainBrainId = event.brainRunId;
-                        }
-                        _state.label = 5;
-                    case 5:
-                        _iteratorAbruptCompletion = false;
-                        return [
-                            3,
-                            3
-                        ];
-                    case 6:
-                        return [
-                            3,
-                            13
-                        ];
-                    case 7:
-                        err = _state.sent();
-                        _didIteratorError = true;
-                        _iteratorError = err;
-                        return [
-                            3,
-                            13
-                        ];
-                    case 8:
-                        _state.trys.push([
-                            8,
-                            ,
-                            11,
-                            12
-                        ]);
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                            3,
-                            10
-                        ];
-                        return [
-                            4,
-                            _iterator.return()
-                        ];
-                    case 9:
-                        _state.sent();
-                        _state.label = 10;
-                    case 10:
-                        return [
-                            3,
-                            12
-                        ];
-                    case 11:
-                        if (_didIteratorError) {
-                            throw _iteratorError;
-                        }
-                        return [
-                            7
-                        ];
-                    case 12:
-                        return [
-                            7
-                        ];
-                    case 13:
-                        return [
-                            3,
-                            15
-                        ];
-                    case 14:
-                        e = _state.sent();
-                        error = e;
-                        return [
-                            3,
-                            15
-                        ];
-                    case 15:
-                        // Verify error was thrown
-                        expect(error === null || error === void 0 ? void 0 : error.message).toBe('Inner brain error');
-                        // Verify event sequence including error
-                        expect(events).toEqual([
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.START,
-                                brainTitle: 'Outer Brain',
-                                status: STATUS.RUNNING,
-                                brainRunId: mainBrainId
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                steps: expect.any(Array)
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_START,
-                                status: STATUS.RUNNING,
-                                stepTitle: 'First step'
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                steps: expect.any(Array)
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_COMPLETE,
-                                status: STATUS.RUNNING,
-                                stepTitle: 'First step'
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                steps: expect.any(Array)
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_START,
-                                status: STATUS.RUNNING,
-                                stepTitle: 'Run inner brain'
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                steps: expect.any(Array)
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.START,
-                                brainTitle: 'Failing Inner Brain',
-                                status: STATUS.RUNNING
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                steps: expect.any(Array)
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_START,
-                                status: STATUS.RUNNING,
-                                stepTitle: 'Throw error'
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                steps: expect.any(Array)
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.ERROR,
-                                brainTitle: 'Failing Inner Brain',
-                                status: STATUS.ERROR,
-                                error: expect.objectContaining({
-                                    name: expect.any(String),
-                                    message: expect.any(String)
-                                })
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                steps: expect.arrayContaining([
-                                    expect.objectContaining({
-                                        title: 'Throw error',
-                                        status: STATUS.ERROR
-                                    })
-                                ])
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.ERROR,
-                                brainTitle: 'Outer Brain',
-                                status: STATUS.ERROR,
-                                error: expect.objectContaining({
-                                    name: expect.any(String),
-                                    message: expect.any(String)
-                                })
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                steps: expect.arrayContaining([
-                                    expect.objectContaining({
-                                        title: 'Run inner brain',
-                                        status: STATUS.ERROR
-                                    })
-                                ])
-                            })
-                        ]);
-                        // Find inner and outer error events by brainRunId
-                        innerErrorEvent = events.find(function(e) {
-                            return e.type === BRAIN_EVENTS.ERROR && e.brainRunId !== mainBrainId;
-                        });
-                        outerErrorEvent = events.find(function(e) {
-                            return e.type === BRAIN_EVENTS.ERROR && e.brainRunId === mainBrainId;
-                        });
-                        expect(innerErrorEvent.error).toEqual(expect.objectContaining({
-                            message: 'Inner brain error'
-                        }));
-                        expect(outerErrorEvent.error).toEqual(expect.objectContaining({
-                            message: 'Inner brain error'
-                        }));
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-});
-describe('step creation', function() {
-    it('should create a step that updates state', function() {
-        return _async_to_generator(function() {
-            var testBrain, events, finalState, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        testBrain = brain('Simple Brain').step('Simple step', function(param) {
-                            var state = param.state;
-                            return _object_spread_props(_object_spread({}, state), {
-                                count: 1,
-                                message: 'Count is now 1'
-                            });
-                        });
-                        events = [];
-                        finalState = {};
-                        _iteratorAbruptCompletion = false, _didIteratorError = false;
-                        _state.label = 1;
-                    case 1:
-                        _state.trys.push([
-                            1,
-                            6,
-                            7,
-                            12
-                        ]);
-                        _iterator = _async_iterator(testBrain.run({
-                            client: mockClient
-                        }));
-                        _state.label = 2;
-                    case 2:
-                        return [
-                            4,
-                            _iterator.next()
-                        ];
-                    case 3:
-                        if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                            3,
-                            5
-                        ];
-                        _value = _step.value;
-                        event = _value;
-                        events.push(event);
-                        if (event.type === BRAIN_EVENTS.STEP_COMPLETE) {
-                            finalState = applyPatches(finalState, event.patch);
-                        }
-                        _state.label = 4;
-                    case 4:
-                        _iteratorAbruptCompletion = false;
-                        return [
-                            3,
-                            2
-                        ];
-                    case 5:
-                        return [
-                            3,
-                            12
-                        ];
-                    case 6:
-                        err = _state.sent();
-                        _didIteratorError = true;
-                        _iteratorError = err;
-                        return [
-                            3,
-                            12
-                        ];
-                    case 7:
-                        _state.trys.push([
-                            7,
-                            ,
-                            10,
-                            11
-                        ]);
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                            3,
-                            9
-                        ];
-                        return [
-                            4,
-                            _iterator.return()
-                        ];
-                    case 8:
-                        _state.sent();
-                        _state.label = 9;
-                    case 9:
-                        return [
-                            3,
-                            11
-                        ];
-                    case 10:
-                        if (_didIteratorError) {
-                            throw _iteratorError;
-                        }
-                        return [
-                            7
-                        ];
-                    case 11:
-                        return [
-                            7
-                        ];
-                    case 12:
-                        // Skip checking events[0] (brain:start)
-                        // Skip checking events[1] (step:status)
-                        // Verify the step start event
-                        expect(events[2]).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.STEP_START,
-                            status: STATUS.RUNNING,
-                            stepTitle: 'Simple step',
-                            stepId: expect.any(String),
-                            options: {}
-                        }));
-                        // Verify the step status event (running)
-                        expect(events[3]).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.STEP_STATUS,
-                            steps: expect.any(Array),
-                            options: {}
-                        }));
-                        if (events[3].type === BRAIN_EVENTS.STEP_STATUS) {
-                            expect(events[3].steps[0].status).toBe(STATUS.RUNNING);
-                        }
-                        // Verify the step complete event
-                        expect(events[4]).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.STEP_COMPLETE,
-                            status: STATUS.RUNNING,
-                            stepTitle: 'Simple step',
-                            stepId: expect.any(String),
-                            patch: [
-                                {
-                                    op: 'add',
-                                    path: '/count',
-                                    value: 1
-                                },
-                                {
-                                    op: 'add',
-                                    path: '/message',
-                                    value: 'Count is now 1'
-                                }
-                            ],
-                            options: {}
-                        }));
-                        expect(events[5]).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.STEP_STATUS,
-                            steps: [
-                                expect.objectContaining({
-                                    title: 'Simple step',
-                                    status: STATUS.COMPLETE,
-                                    id: expect.any(String)
-                                })
-                            ],
-                            options: {}
-                        }));
-                        // Verify the brain complete event
-                        expect(events[6]).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.COMPLETE,
-                            status: STATUS.COMPLETE,
-                            brainTitle: 'Simple Brain',
-                            options: {}
-                        }));
-                        // Verify the final state
-                        expect(finalState).toEqual({
-                            count: 1,
-                            message: 'Count is now 1'
-                        });
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-    it('should maintain immutable results between steps', function() {
-        return _async_to_generator(function() {
-            var testBrain, finalState, patches, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, _patches, err;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        testBrain = brain('Immutable Steps Brain').step('First step', function() {
-                            return {
-                                value: 1
-                            };
-                        }).step('Second step', function(param) {
-                            var state = param.state;
-                            // Attempt to modify previous step's state
-                            state.value = 99;
-                            return {
-                                value: 2
-                            };
-                        });
-                        finalState = {};
-                        patches = [];
-                        _iteratorAbruptCompletion = false, _didIteratorError = false;
-                        _state.label = 1;
-                    case 1:
-                        _state.trys.push([
-                            1,
-                            6,
-                            7,
-                            12
-                        ]);
-                        _iterator = _async_iterator(testBrain.run({
-                            client: mockClient
-                        }));
-                        _state.label = 2;
-                    case 2:
-                        return [
-                            4,
-                            _iterator.next()
-                        ];
-                    case 3:
-                        if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                            3,
-                            5
-                        ];
-                        _value = _step.value;
-                        event = _value;
-                        if (event.type === BRAIN_EVENTS.STEP_COMPLETE) {
-                            ;
-                            (_patches = patches).push.apply(_patches, _to_consumable_array(event.patch));
-                        }
-                        _state.label = 4;
-                    case 4:
-                        _iteratorAbruptCompletion = false;
-                        return [
-                            3,
-                            2
-                        ];
-                    case 5:
-                        return [
-                            3,
-                            12
-                        ];
-                    case 6:
-                        err = _state.sent();
-                        _didIteratorError = true;
-                        _iteratorError = err;
-                        return [
-                            3,
-                            12
-                        ];
-                    case 7:
-                        _state.trys.push([
-                            7,
-                            ,
-                            10,
-                            11
-                        ]);
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                            3,
-                            9
-                        ];
-                        return [
-                            4,
-                            _iterator.return()
-                        ];
-                    case 8:
-                        _state.sent();
-                        _state.label = 9;
-                    case 9:
-                        return [
-                            3,
-                            11
-                        ];
-                    case 10:
-                        if (_didIteratorError) {
-                            throw _iteratorError;
-                        }
-                        return [
-                            7
-                        ];
-                    case 11:
-                        return [
-                            7
-                        ];
-                    case 12:
-                        // Apply all patches to the initial state
-                        finalState = applyPatches(finalState, patches);
-                        // Verify the final state
-                        expect(finalState).toEqual({
-                            value: 2
-                        });
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-});
-describe('brain resumption', function() {
-    var mockClient = {
-        generateObject: jest.fn()
-    };
-    it('should resume brain from the correct step when given initialCompletedSteps', function() {
-        return _async_to_generator(function() {
-            var executedSteps, threeStepBrain, initialCompletedSteps, initialState, firstStepState, allStepsInfo, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _loop, _iterator, _step, _ret, err, resumedState, _iteratorAbruptCompletion1, _didIteratorError1, _iteratorError1, _iterator1, _step1, _value, event, err1;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        executedSteps = [];
-                        threeStepBrain = brain('Three Step Brain').step('Step 1', function(param) {
-                            var state = param.state;
-                            executedSteps.push('Step 1');
-                            return _object_spread_props(_object_spread({}, state), {
-                                value: 2
-                            });
-                        }).step('Step 2', function(param) {
-                            var state = param.state;
-                            executedSteps.push('Step 2');
-                            return _object_spread_props(_object_spread({}, state), {
-                                value: state.value + 10
-                            });
-                        }).step('Step 3', function(param) {
-                            var state = param.state;
-                            executedSteps.push('Step 3');
-                            return _object_spread_props(_object_spread({}, state), {
-                                value: state.value * 3
-                            });
-                        });
-                        // First run to get the first step completed with initial state
-                        initialCompletedSteps = []; // Use the correct type
-                        initialState = {
-                            initialValue: true
-                        };
-                        firstStepState = initialState;
-                        allStepsInfo = []; // Explicit type annotation needed
-                        _iteratorAbruptCompletion = false, _didIteratorError = false;
-                        _state.label = 1;
-                    case 1:
-                        _state.trys.push([
-                            1,
-                            6,
-                            7,
-                            12
-                        ]);
-                        _loop = function() {
-                            var _value = _step.value;
-                            var event = _value;
-                            // Capture the full step list from the first status event
-                            if (event.type === BRAIN_EVENTS.STEP_STATUS) {
-                                allStepsInfo = event.steps; // Direct assignment, type is SerializedStepStatus[]
-                            }
-                            if (event.type === BRAIN_EVENTS.STEP_COMPLETE && event.stepTitle === 'Step 1') {
-                                firstStepState = applyPatches(firstStepState, [
-                                    event.patch
-                                ]);
-                                // Construct initialCompletedSteps with the full data for completed steps
-                                initialCompletedSteps = allStepsInfo.map(function(stepInfo, index) {
-                                    if (index === 0) {
-                                        // If it's Step 1
-                                        return _object_spread_props(_object_spread({}, stepInfo), {
-                                            status: STATUS.COMPLETE,
-                                            patch: event.patch
-                                        });
-                                    } else {
-                                        return _object_spread_props(_object_spread({}, stepInfo), {
-                                            status: STATUS.PENDING,
-                                            patch: undefined
-                                        });
-                                    }
-                                });
-                                return "break" // Stop after first step
-                                ;
-                            }
-                        };
-                        _iterator = _async_iterator(threeStepBrain.run({
-                            client: mockClient,
-                            initialState: initialState
-                        }));
-                        _state.label = 2;
-                    case 2:
-                        return [
-                            4,
-                            _iterator.next()
-                        ];
-                    case 3:
-                        if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                            3,
-                            5
-                        ];
-                        _ret = _loop();
-                        if (_ret === "break") return [
-                            3,
-                            5
-                        ];
-                        _state.label = 4;
-                    case 4:
-                        _iteratorAbruptCompletion = false;
-                        return [
-                            3,
-                            2
-                        ];
-                    case 5:
-                        return [
-                            3,
-                            12
-                        ];
-                    case 6:
-                        err = _state.sent();
-                        _didIteratorError = true;
-                        _iteratorError = err;
-                        return [
-                            3,
-                            12
-                        ];
-                    case 7:
-                        _state.trys.push([
-                            7,
-                            ,
-                            10,
-                            11
-                        ]);
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                            3,
-                            9
-                        ];
-                        return [
-                            4,
-                            _iterator.return()
-                        ];
-                    case 8:
-                        _state.sent();
-                        _state.label = 9;
-                    case 9:
-                        return [
-                            3,
-                            11
-                        ];
-                    case 10:
-                        if (_didIteratorError) {
-                            throw _iteratorError;
-                        }
-                        return [
-                            7
-                        ];
-                    case 11:
-                        return [
-                            7
-                        ];
-                    case 12:
-                        // Clear executed steps array
-                        executedSteps.length = 0;
-                        if (!initialCompletedSteps) throw new Error('Expected initialCompletedSteps');
-                        _iteratorAbruptCompletion1 = false, _didIteratorError1 = false;
-                        _state.label = 13;
-                    case 13:
-                        _state.trys.push([
-                            13,
-                            18,
-                            19,
-                            24
-                        ]);
-                        _iterator1 = _async_iterator(threeStepBrain.run({
-                            client: mockClient,
-                            initialState: initialState,
-                            initialCompletedSteps: initialCompletedSteps,
-                            brainRunId: 'test-run-id'
-                        }));
-                        _state.label = 14;
-                    case 14:
-                        return [
-                            4,
-                            _iterator1.next()
-                        ];
-                    case 15:
-                        if (!(_iteratorAbruptCompletion1 = !(_step1 = _state.sent()).done)) return [
-                            3,
-                            17
-                        ];
-                        _value = _step1.value;
-                        event = _value;
-                        if (event.type === BRAIN_EVENTS.RESTART) {
-                            resumedState = event.initialState;
-                        } else if (event.type === BRAIN_EVENTS.STEP_COMPLETE) {
-                            resumedState = applyPatches(resumedState, [
-                                event.patch
-                            ]);
-                        }
-                        _state.label = 16;
-                    case 16:
-                        _iteratorAbruptCompletion1 = false;
-                        return [
-                            3,
-                            14
-                        ];
-                    case 17:
-                        return [
-                            3,
-                            24
-                        ];
-                    case 18:
-                        err1 = _state.sent();
-                        _didIteratorError1 = true;
-                        _iteratorError1 = err1;
-                        return [
-                            3,
-                            24
-                        ];
-                    case 19:
-                        _state.trys.push([
-                            19,
-                            ,
-                            22,
-                            23
-                        ]);
-                        if (!(_iteratorAbruptCompletion1 && _iterator1.return != null)) return [
-                            3,
-                            21
-                        ];
-                        return [
-                            4,
-                            _iterator1.return()
-                        ];
-                    case 20:
-                        _state.sent();
-                        _state.label = 21;
-                    case 21:
-                        return [
-                            3,
-                            23
-                        ];
-                    case 22:
-                        if (_didIteratorError1) {
-                            throw _iteratorError1;
-                        }
-                        return [
-                            7
-                        ];
-                    case 23:
-                        return [
-                            7
-                        ];
-                    case 24:
-                        // Verify only steps 2 and 3 were executed
-                        expect(executedSteps).toEqual([
-                            'Step 2',
-                            'Step 3'
-                        ]);
-                        expect(executedSteps).not.toContain('Step 1');
-                        // Verify the final state after all steps complete
-                        expect(resumedState).toEqual({
-                            value: 36,
-                            initialValue: true
-                        });
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-});
-describe('nested brains', function() {
-    it('should execute nested brains and yield all inner brain events', function() {
-        return _async_to_generator(function() {
-            var innerBrain, outerBrain, events, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err, innerState, outerState, _iteratorNormalCompletion, _didIteratorError1, _iteratorError1, _iterator1, _step1, event1;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        // Create an inner brain that will be nested
-                        innerBrain = brain('Inner Brain').step('Double value', function(param) {
-                            var state = param.state;
-                            return {
-                                inner: true,
-                                value: state.value * 2
-                            };
-                        });
-                        // Create outer brain that uses the inner brain
-                        outerBrain = brain('Outer Brain').step('Set prefix', function() {
-                            return {
-                                prefix: 'test-'
-                            };
-                        }).brain('Run inner brain', innerBrain, function(param) {
-                            var state = param.state, brainState = param.brainState;
-                            return _object_spread_props(_object_spread({}, state), {
-                                innerResult: brainState.value
-                            });
-                        }, function() {
-                            return {
-                                value: 5
-                            };
-                        });
-                        events = [];
-                        _iteratorAbruptCompletion = false, _didIteratorError = false;
-                        _state.label = 1;
-                    case 1:
-                        _state.trys.push([
-                            1,
-                            6,
-                            7,
-                            12
-                        ]);
-                        _iterator = _async_iterator(outerBrain.run({
-                            client: mockClient
-                        }));
-                        _state.label = 2;
-                    case 2:
-                        return [
-                            4,
-                            _iterator.next()
-                        ];
-                    case 3:
-                        if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                            3,
-                            5
-                        ];
-                        _value = _step.value;
-                        event = _value;
-                        events.push(event);
-                        _state.label = 4;
-                    case 4:
-                        _iteratorAbruptCompletion = false;
-                        return [
-                            3,
-                            2
-                        ];
-                    case 5:
-                        return [
-                            3,
-                            12
-                        ];
-                    case 6:
-                        err = _state.sent();
-                        _didIteratorError = true;
-                        _iteratorError = err;
-                        return [
-                            3,
-                            12
-                        ];
-                    case 7:
-                        _state.trys.push([
-                            7,
-                            ,
-                            10,
-                            11
-                        ]);
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                            3,
-                            9
-                        ];
-                        return [
-                            4,
-                            _iterator.return()
-                        ];
-                    case 8:
-                        _state.sent();
-                        _state.label = 9;
-                    case 9:
-                        return [
-                            3,
-                            11
-                        ];
-                    case 10:
-                        if (_didIteratorError) {
-                            throw _iteratorError;
-                        }
-                        return [
-                            7
-                        ];
-                    case 11:
-                        return [
-                            7
-                        ];
-                    case 12:
-                        // Verify all events are yielded in correct order
-                        expect(events.map(function(e) {
-                            return {
-                                type: e.type,
-                                brainTitle: 'brainTitle' in e ? e.brainTitle : undefined,
-                                status: 'status' in e ? e.status : undefined,
-                                stepTitle: 'stepTitle' in e ? e.stepTitle : undefined
-                            };
-                        })).toEqual([
-                            // Outer brain start
-                            {
-                                type: BRAIN_EVENTS.START,
-                                brainTitle: 'Outer Brain',
-                                status: STATUS.RUNNING,
-                                stepTitle: undefined
-                            },
-                            // Initial step status for outer brain
-                            {
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                brainTitle: undefined,
-                                status: undefined,
-                                stepTitle: undefined
-                            },
-                            // First step of outer brain
-                            {
-                                type: BRAIN_EVENTS.STEP_START,
-                                brainTitle: undefined,
-                                status: STATUS.RUNNING,
-                                stepTitle: 'Set prefix'
-                            },
-                            // First step status (running)
-                            {
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                brainTitle: undefined,
-                                status: undefined,
-                                stepTitle: undefined
-                            },
-                            {
-                                type: BRAIN_EVENTS.STEP_COMPLETE,
-                                brainTitle: undefined,
-                                status: STATUS.RUNNING,
-                                stepTitle: 'Set prefix'
-                            },
-                            {
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                brainTitle: undefined,
-                                status: undefined,
-                                stepTitle: undefined
-                            },
-                            {
-                                type: BRAIN_EVENTS.STEP_START,
-                                brainTitle: undefined,
-                                status: STATUS.RUNNING,
-                                stepTitle: 'Run inner brain'
-                            },
-                            // Step status for inner brain (running)
-                            {
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                brainTitle: undefined,
-                                status: undefined,
-                                stepTitle: undefined
-                            },
-                            // Inner brain start
-                            {
-                                type: BRAIN_EVENTS.START,
-                                brainTitle: 'Inner Brain',
-                                status: STATUS.RUNNING,
-                                stepTitle: undefined
-                            },
-                            // Initial step status for inner brain
-                            {
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                brainTitle: undefined,
-                                status: undefined,
-                                stepTitle: undefined
-                            },
-                            // Inner brain step
-                            {
-                                type: BRAIN_EVENTS.STEP_START,
-                                brainTitle: undefined,
-                                status: STATUS.RUNNING,
-                                stepTitle: 'Double value'
-                            },
-                            // Inner brain step status (running)
-                            {
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                brainTitle: undefined,
-                                status: undefined,
-                                stepTitle: undefined
-                            },
-                            {
-                                type: BRAIN_EVENTS.STEP_COMPLETE,
-                                brainTitle: undefined,
-                                status: STATUS.RUNNING,
-                                stepTitle: 'Double value'
-                            },
-                            {
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                brainTitle: undefined,
-                                status: undefined,
-                                stepTitle: undefined
-                            },
-                            {
-                                type: BRAIN_EVENTS.COMPLETE,
-                                brainTitle: 'Inner Brain',
-                                status: STATUS.COMPLETE,
-                                stepTitle: undefined
-                            },
-                            // Outer brain nested step completion
-                            {
-                                type: BRAIN_EVENTS.STEP_COMPLETE,
-                                brainTitle: undefined,
-                                status: STATUS.RUNNING,
-                                stepTitle: 'Run inner brain'
-                            },
-                            {
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                brainTitle: undefined,
-                                status: undefined,
-                                stepTitle: undefined
-                            },
-                            // Outer brain completion
-                            {
-                                type: BRAIN_EVENTS.COMPLETE,
-                                brainTitle: 'Outer Brain',
-                                status: STATUS.COMPLETE,
-                                stepTitle: undefined
-                            }
-                        ]);
-                        // Verify states are passed correctly
-                        innerState = {
-                            value: 5
-                        }; // Match the initial state from the brain
-                        outerState = {};
-                        _iteratorNormalCompletion = true, _didIteratorError1 = false, _iteratorError1 = undefined;
-                        try {
-                            for(_iterator1 = events[Symbol.iterator](); !(_iteratorNormalCompletion = (_step1 = _iterator1.next()).done); _iteratorNormalCompletion = true){
-                                event1 = _step1.value;
-                                if (event1.type === BRAIN_EVENTS.STEP_COMPLETE) {
-                                    if (event1.stepTitle === 'Double value') {
-                                        innerState = applyPatches(innerState, [
-                                            event1.patch
-                                        ]);
-                                    } else {
-                                        outerState = applyPatches(outerState, [
-                                            event1.patch
-                                        ]);
-                                    }
-                                }
-                            }
-                        } catch (err) {
-                            _didIteratorError1 = true;
-                            _iteratorError1 = err;
-                        } finally{
-                            try {
-                                if (!_iteratorNormalCompletion && _iterator1.return != null) {
-                                    _iterator1.return();
-                                }
-                            } finally{
-                                if (_didIteratorError1) {
-                                    throw _iteratorError1;
-                                }
-                            }
-                        }
-                        // Verify final states
-                        expect(innerState).toEqual({
-                            inner: true,
-                            value: 10
-                        });
-                        expect(outerState).toEqual({
-                            prefix: 'test-',
-                            innerResult: 10
-                        });
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-    it('should handle errors in nested brains and propagate them up', function() {
-        return _async_to_generator(function() {
-            var innerBrain, outerBrain, events, error, mainBrainId, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err, e, innerErrorEvent, outerErrorEvent;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        // Create an inner brain that will throw an error
-                        innerBrain = brain('Failing Inner Brain').step('Throw error', function() {
-                            throw new Error('Inner brain error');
-                        });
-                        // Create outer brain that uses the failing inner brain
-                        outerBrain = brain('Outer Brain').step('First step', function() {
-                            return {
-                                step: 'first'
-                            };
-                        }).brain('Run inner brain', innerBrain, function(param) {
-                            var state = param.state, brainState = param.brainState;
-                            return _object_spread_props(_object_spread({}, state), {
-                                step: 'second',
-                                innerResult: brainState.value
-                            });
-                        }, function() {
-                            return {
-                                value: 5
-                            };
-                        });
-                        events = [];
-                        _state.label = 1;
-                    case 1:
-                        _state.trys.push([
-                            1,
-                            14,
-                            ,
-                            15
-                        ]);
-                        _iteratorAbruptCompletion = false, _didIteratorError = false;
-                        _state.label = 2;
-                    case 2:
-                        _state.trys.push([
-                            2,
-                            7,
-                            8,
-                            13
-                        ]);
-                        _iterator = _async_iterator(outerBrain.run({
-                            client: mockClient
-                        }));
-                        _state.label = 3;
-                    case 3:
-                        return [
-                            4,
-                            _iterator.next()
-                        ];
-                    case 4:
-                        if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                            3,
-                            6
-                        ];
-                        _value = _step.value;
-                        event = _value;
-                        events.push(event);
-                        if (event.type === BRAIN_EVENTS.START && !mainBrainId) {
-                            mainBrainId = event.brainRunId;
-                        }
-                        _state.label = 5;
-                    case 5:
-                        _iteratorAbruptCompletion = false;
-                        return [
-                            3,
-                            3
-                        ];
-                    case 6:
-                        return [
-                            3,
-                            13
-                        ];
-                    case 7:
-                        err = _state.sent();
-                        _didIteratorError = true;
-                        _iteratorError = err;
-                        return [
-                            3,
-                            13
-                        ];
-                    case 8:
-                        _state.trys.push([
-                            8,
-                            ,
-                            11,
-                            12
-                        ]);
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                            3,
-                            10
-                        ];
-                        return [
-                            4,
-                            _iterator.return()
-                        ];
-                    case 9:
-                        _state.sent();
-                        _state.label = 10;
-                    case 10:
-                        return [
-                            3,
-                            12
-                        ];
-                    case 11:
-                        if (_didIteratorError) {
-                            throw _iteratorError;
-                        }
-                        return [
-                            7
-                        ];
-                    case 12:
-                        return [
-                            7
-                        ];
-                    case 13:
-                        return [
-                            3,
-                            15
-                        ];
-                    case 14:
-                        e = _state.sent();
-                        error = e;
-                        return [
-                            3,
-                            15
-                        ];
-                    case 15:
-                        // Verify error was thrown
-                        expect(error === null || error === void 0 ? void 0 : error.message).toBe('Inner brain error');
-                        // Verify event sequence including error
-                        expect(events).toEqual([
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.START,
-                                brainTitle: 'Outer Brain',
-                                status: STATUS.RUNNING,
-                                brainRunId: mainBrainId
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                steps: expect.any(Array)
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_START,
-                                status: STATUS.RUNNING,
-                                stepTitle: 'First step'
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                steps: expect.any(Array)
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_COMPLETE,
-                                status: STATUS.RUNNING,
-                                stepTitle: 'First step'
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                steps: expect.any(Array)
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_START,
-                                status: STATUS.RUNNING,
-                                stepTitle: 'Run inner brain'
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                steps: expect.any(Array)
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.START,
-                                brainTitle: 'Failing Inner Brain',
-                                status: STATUS.RUNNING
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                steps: expect.any(Array)
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_START,
-                                status: STATUS.RUNNING,
-                                stepTitle: 'Throw error'
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                steps: expect.any(Array)
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.ERROR,
-                                brainTitle: 'Failing Inner Brain',
-                                status: STATUS.ERROR,
-                                error: expect.objectContaining({
-                                    name: expect.any(String),
-                                    message: expect.any(String)
-                                })
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                steps: expect.arrayContaining([
-                                    expect.objectContaining({
-                                        title: 'Throw error',
-                                        status: STATUS.ERROR
-                                    })
-                                ])
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.ERROR,
-                                brainTitle: 'Outer Brain',
-                                status: STATUS.ERROR,
-                                error: expect.objectContaining({
-                                    name: expect.any(String),
-                                    message: expect.any(String)
-                                })
-                            }),
-                            expect.objectContaining({
-                                type: BRAIN_EVENTS.STEP_STATUS,
-                                steps: expect.arrayContaining([
-                                    expect.objectContaining({
-                                        title: 'Run inner brain',
-                                        status: STATUS.ERROR
-                                    })
-                                ])
-                            })
-                        ]);
-                        // Find inner and outer error events by brainRunId
-                        innerErrorEvent = events.find(function(e) {
-                            return e.type === BRAIN_EVENTS.ERROR && e.brainRunId !== mainBrainId;
-                        });
-                        outerErrorEvent = events.find(function(e) {
-                            return e.type === BRAIN_EVENTS.ERROR && e.brainRunId === mainBrainId;
-                        });
-                        expect(innerErrorEvent.error).toEqual(expect.objectContaining({
-                            message: 'Inner brain error'
-                        }));
-                        expect(outerErrorEvent.error).toEqual(expect.objectContaining({
-                            message: 'Inner brain error'
-                        }));
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-    it('should include patches in step status events for inner brain steps', function() {
-        return _async_to_generator(function() {
-            var innerBrain, outerBrain, finalStepStatus, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        // Create an inner brain that modifies state
-                        innerBrain = brain('Inner Brain').step('Double value', function(param) {
-                            var state = param.state;
-                            return _object_spread_props(_object_spread({}, state), {
-                                value: state.value * 2
-                            });
-                        });
-                        // Create outer brain that uses the inner brain
-                        outerBrain = brain('Outer Brain').step('Set initial', function() {
-                            return {
-                                value: 5
-                            };
-                        }).brain('Run inner brain', innerBrain, function(param) {
-                            var state = param.state, brainState = param.brainState;
-                            return _object_spread_props(_object_spread({}, state), {
-                                result: brainState.value
-                            });
-                        }, function(state) {
-                            return {
-                                value: state.value
-                            };
-                        });
-                        _iteratorAbruptCompletion = false, _didIteratorError = false;
-                        _state.label = 1;
-                    case 1:
-                        _state.trys.push([
-                            1,
-                            6,
-                            7,
-                            12
-                        ]);
-                        _iterator = _async_iterator(outerBrain.run({
-                            client: mockClient
-                        }));
-                        _state.label = 2;
-                    case 2:
-                        return [
-                            4,
-                            _iterator.next()
-                        ];
-                    case 3:
-                        if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                            3,
-                            5
-                        ];
-                        _value = _step.value;
-                        event = _value;
-                        if (event.type === BRAIN_EVENTS.STEP_STATUS) {
-                            finalStepStatus = event;
-                        }
-                        _state.label = 4;
-                    case 4:
-                        _iteratorAbruptCompletion = false;
-                        return [
-                            3,
-                            2
-                        ];
-                    case 5:
-                        return [
-                            3,
-                            12
-                        ];
-                    case 6:
-                        err = _state.sent();
-                        _didIteratorError = true;
-                        _iteratorError = err;
-                        return [
-                            3,
-                            12
-                        ];
-                    case 7:
-                        _state.trys.push([
-                            7,
-                            ,
-                            10,
-                            11
-                        ]);
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                            3,
-                            9
-                        ];
-                        return [
-                            4,
-                            _iterator.return()
-                        ];
-                    case 8:
-                        _state.sent();
-                        _state.label = 9;
-                    case 9:
-                        return [
-                            3,
-                            11
-                        ];
-                    case 10:
-                        if (_didIteratorError) {
-                            throw _iteratorError;
-                        }
-                        return [
-                            7
-                        ];
-                    case 11:
-                        return [
-                            7
-                        ];
-                    case 12:
-                        // Verify step status contains patches for all steps including the inner brain step
-                        expect(finalStepStatus === null || finalStepStatus === void 0 ? void 0 : finalStepStatus.steps).toEqual([
-                            expect.objectContaining({
-                                title: 'Set initial',
-                                status: STATUS.COMPLETE
-                            }),
-                            expect.objectContaining({
-                                title: 'Run inner brain',
-                                status: STATUS.COMPLETE
-                            })
-                        ]);
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-});
-describe('brain options', function() {
-    it('should pass options through to brain events', function() {
-        return _async_to_generator(function() {
-            var testBrain, brainOptions, finalEvent, finalStepStatus, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        testBrain = brain('Options Brain').step('Simple step', function(param) {
-                            var state = param.state, options = param.options;
-                            return {
-                                value: 1,
-                                passedOption: options.testOption
-                            };
-                        });
-                        brainOptions = {
-                            testOption: 'test-value'
-                        };
-                        _iteratorAbruptCompletion = false, _didIteratorError = false;
-                        _state.label = 1;
-                    case 1:
-                        _state.trys.push([
-                            1,
-                            6,
-                            7,
-                            12
-                        ]);
-                        _iterator = _async_iterator(testBrain.run({
-                            client: mockClient,
-                            options: brainOptions
-                        }));
-                        _state.label = 2;
-                    case 2:
-                        return [
-                            4,
-                            _iterator.next()
-                        ];
-                    case 3:
-                        if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                            3,
-                            5
-                        ];
-                        _value = _step.value;
-                        event = _value;
-                        if (event.type === BRAIN_EVENTS.STEP_STATUS) {
-                            finalStepStatus = event;
-                        } else {
-                            finalEvent = event;
-                        }
-                        _state.label = 4;
-                    case 4:
-                        _iteratorAbruptCompletion = false;
-                        return [
-                            3,
-                            2
-                        ];
-                    case 5:
-                        return [
-                            3,
-                            12
-                        ];
-                    case 6:
-                        err = _state.sent();
-                        _didIteratorError = true;
-                        _iteratorError = err;
-                        return [
-                            3,
-                            12
-                        ];
-                    case 7:
-                        _state.trys.push([
-                            7,
-                            ,
-                            10,
-                            11
-                        ]);
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                            3,
-                            9
-                        ];
-                        return [
-                            4,
-                            _iterator.return()
-                        ];
-                    case 8:
-                        _state.sent();
-                        _state.label = 9;
-                    case 9:
-                        return [
-                            3,
-                            11
-                        ];
-                    case 10:
-                        if (_didIteratorError) {
-                            throw _iteratorError;
-                        }
-                        return [
-                            7
-                        ];
-                    case 11:
-                        return [
-                            7
-                        ];
-                    case 12:
-                        expect(finalEvent).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.COMPLETE,
-                            status: STATUS.COMPLETE,
-                            brainTitle: 'Options Brain',
-                            brainDescription: undefined,
-                            options: brainOptions
-                        }));
-                        expect(finalStepStatus).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.STEP_STATUS,
-                            steps: [
-                                expect.objectContaining({
-                                    title: 'Simple step',
-                                    status: STATUS.COMPLETE
-                                })
-                            ],
-                            options: brainOptions
-                        }));
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-    it('should provide empty object as default options', function() {
-        return _async_to_generator(function() {
-            var testBrain, brainRun, stepStartResult, stepStatusRunning, stepResult;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        testBrain = brain('Default Options Brain').step('Simple step', function(param) {
-                            var options = param.options;
-                            return {
-                                hasOptions: Object.keys(options).length === 0
-                            };
-                        });
-                        brainRun = testBrain.run({
-                            client: mockClient
-                        });
-                        // Skip start event
-                        return [
-                            4,
-                            brainRun.next()
-                        ];
-                    case 1:
-                        _state.sent();
-                        // Skip initial step status event
-                        return [
-                            4,
-                            brainRun.next()
-                        ];
-                    case 2:
-                        _state.sent();
-                        return [
-                            4,
-                            brainRun.next()
-                        ];
-                    case 3:
-                        stepStartResult = _state.sent();
-                        expect(stepStartResult.value).toEqual(expect.objectContaining({
-                            options: {},
-                            type: BRAIN_EVENTS.STEP_START
-                        }));
-                        return [
-                            4,
-                            brainRun.next()
-                        ];
-                    case 4:
-                        stepStatusRunning = _state.sent();
-                        expect(stepStatusRunning.value).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.STEP_STATUS,
-                            steps: expect.any(Array)
-                        }));
-                        if (stepStatusRunning.value.type === BRAIN_EVENTS.STEP_STATUS) {
-                            expect(stepStatusRunning.value.steps[0].status).toBe(STATUS.RUNNING);
-                        }
-                        return [
-                            4,
-                            brainRun.next()
-                        ];
-                    case 5:
-                        stepResult = _state.sent();
-                        expect(stepResult.value).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.STEP_COMPLETE,
-                            stepTitle: 'Simple step',
-                            options: {}
-                        }));
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-});
-describe('services support', function() {
-    it('should allow adding custom services to brains', function() {
-        return _async_to_generator(function() {
-            var testBrain, finalState, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        // Create a brain with services
-                        testBrain = brain('Services Test').withServices({
-                            logger: testLogger
-                        }).step('Use service', function(param) {
-                            var state = param.state, logger = param.logger;
-                            logger.log('Test service called');
-                            return {
-                                serviceUsed: true
-                            };
-                        });
-                        // Run the brain and collect events
-                        finalState = {};
-                        _iteratorAbruptCompletion = false, _didIteratorError = false;
-                        _state.label = 1;
-                    case 1:
-                        _state.trys.push([
-                            1,
-                            6,
-                            7,
-                            12
-                        ]);
-                        _iterator = _async_iterator(testBrain.run({
-                            client: mockClient
-                        }));
-                        _state.label = 2;
-                    case 2:
-                        return [
-                            4,
-                            _iterator.next()
-                        ];
-                    case 3:
-                        if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                            3,
-                            5
-                        ];
-                        _value = _step.value;
-                        event = _value;
-                        if (event.type === BRAIN_EVENTS.STEP_COMPLETE) {
-                            finalState = applyPatches(finalState, [
-                                event.patch
-                            ]);
-                        }
-                        _state.label = 4;
-                    case 4:
-                        _iteratorAbruptCompletion = false;
-                        return [
-                            3,
-                            2
-                        ];
-                    case 5:
-                        return [
-                            3,
-                            12
-                        ];
-                    case 6:
-                        err = _state.sent();
-                        _didIteratorError = true;
-                        _iteratorError = err;
-                        return [
-                            3,
-                            12
-                        ];
-                    case 7:
-                        _state.trys.push([
-                            7,
-                            ,
-                            10,
-                            11
-                        ]);
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                            3,
-                            9
-                        ];
-                        return [
-                            4,
-                            _iterator.return()
-                        ];
-                    case 8:
-                        _state.sent();
-                        _state.label = 9;
-                    case 9:
-                        return [
-                            3,
-                            11
-                        ];
-                    case 10:
-                        if (_didIteratorError) {
-                            throw _iteratorError;
-                        }
-                        return [
-                            7
-                        ];
-                    case 11:
-                        return [
-                            7
-                        ];
-                    case 12:
-                        // Verify the service was called
-                        expect(testLogger.log).toHaveBeenCalledWith('Test service called');
-                        // Verify the state was updated
-                        expect(finalState).toEqual({
-                            serviceUsed: true
-                        });
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-});
-describe('type inference', function() {
-    it('should correctly infer complex brain state types', function() {
-        return _async_to_generator(function() {
-            var innerBrain, complexBrain, _typeAssert, events, finalStepStatus, finalState, mainBrainId, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err, innerStartEvent, lastStep;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        // Create an inner brain that uses the shared options type
-                        innerBrain = brain('Inner Type Test').step('Process features', function(param) {
-                            var options = param.options;
-                            return {
-                                processedValue: options.features.includes('fast') ? 100 : 42,
-                                featureCount: options.features.length
-                            };
-                        });
-                        // Create a complex brain using multiple features
-                        complexBrain = brain('Complex Type Test').step('First step', function(param) {
-                            var options = param.options;
-                            return {
-                                initialFeatures: options.features,
-                                value: 42
-                            };
-                        }).brain('Nested brain', innerBrain, function(param) {
-                            var state = param.state, brainState = param.brainState;
-                            return _object_spread_props(_object_spread({}, state), {
-                                processedValue: brainState.processedValue,
-                                totalFeatures: brainState.featureCount
-                            });
-                        }, function() {
-                            return {
-                                processedValue: 0,
-                                featureCount: 0
-                            };
-                        }).step('Final step', function(param) {
-                            var state = param.state;
-                            return _object_spread_props(_object_spread({}, state), {
-                                completed: true
-                            });
-                        });
-                        _typeAssert = true;
-                        // Collect all events
-                        events = [];
-                        finalState = {};
-                        _iteratorAbruptCompletion = false, _didIteratorError = false;
-                        _state.label = 1;
-                    case 1:
-                        _state.trys.push([
-                            1,
-                            6,
-                            7,
-                            12
-                        ]);
-                        _iterator = _async_iterator(complexBrain.run({
-                            client: mockClient,
-                            options: {
-                                features: [
-                                    'fast',
-                                    'secure'
-                                ]
-                            }
-                        }));
-                        _state.label = 2;
-                    case 2:
-                        return [
-                            4,
-                            _iterator.next()
-                        ];
-                    case 3:
-                        if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                            3,
-                            5
-                        ];
-                        _value = _step.value;
-                        event = _value;
-                        events.push(event);
-                        // Capture the main brain's ID from its start event
-                        if (event.type === BRAIN_EVENTS.START && !mainBrainId) {
-                            mainBrainId = event.brainRunId;
-                        }
-                        if (event.type === BRAIN_EVENTS.STEP_STATUS) {
-                            finalStepStatus = event;
-                        } else if (event.type === BRAIN_EVENTS.STEP_COMPLETE && event.brainRunId === mainBrainId // Only process events from main brain
-                        ) {
-                            finalState = applyPatches(finalState, [
-                                event.patch
-                            ]);
-                        }
-                        _state.label = 4;
-                    case 4:
-                        _iteratorAbruptCompletion = false;
-                        return [
-                            3,
-                            2
-                        ];
-                    case 5:
-                        return [
-                            3,
-                            12
-                        ];
-                    case 6:
-                        err = _state.sent();
-                        _didIteratorError = true;
-                        _iteratorError = err;
-                        return [
-                            3,
-                            12
-                        ];
-                    case 7:
-                        _state.trys.push([
-                            7,
-                            ,
-                            10,
-                            11
-                        ]);
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                            3,
-                            9
-                        ];
-                        return [
-                            4,
-                            _iterator.return()
-                        ];
-                    case 8:
-                        _state.sent();
-                        _state.label = 9;
-                    case 9:
-                        return [
-                            3,
-                            11
-                        ];
-                    case 10:
-                        if (_didIteratorError) {
-                            throw _iteratorError;
-                        }
-                        return [
-                            7
-                        ];
-                    case 11:
-                        return [
-                            7
-                        ];
-                    case 12:
-                        // Verify brain start event
-                        expect(events[0]).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.START,
-                            status: STATUS.RUNNING,
-                            brainTitle: 'Complex Type Test',
-                            brainDescription: undefined,
-                            options: {
-                                features: [
-                                    'fast',
-                                    'secure'
-                                ]
-                            },
-                            brainRunId: mainBrainId
-                        }));
-                        // Verify inner brain events are included
-                        innerStartEvent = events.find(function(e) {
-                            return e.type === BRAIN_EVENTS.START && 'brainRunId' in e && e.brainRunId !== mainBrainId;
-                        });
-                        expect(innerStartEvent).toEqual(expect.objectContaining({
-                            type: BRAIN_EVENTS.START,
-                            status: STATUS.RUNNING,
-                            brainTitle: 'Inner Type Test',
-                            options: {
-                                features: [
-                                    'fast',
-                                    'secure'
-                                ]
-                            }
-                        }));
-                        // Verify the final step status
-                        if (!finalStepStatus) throw new Error('Expected final step status event');
-                        lastStep = finalStepStatus.steps[finalStepStatus.steps.length - 1];
-                        expect(lastStep.status).toBe(STATUS.COMPLETE);
-                        expect(lastStep.title).toBe('Final step');
-                        expect(finalState).toEqual({
-                            initialFeatures: [
-                                'fast',
-                                'secure'
-                            ],
-                            value: 42,
-                            processedValue: 100,
-                            totalFeatures: 2,
-                            completed: true
-                        });
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-    it('should correctly infer brain reducer state types', function() {
-        return _async_to_generator(function() {
-            var innerBrain, outerBrain, finalState, mainBrainId, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        // Create an inner brain with a specific state shape
-                        innerBrain = brain('Inner State Test').step('Inner step', function() {
-                            return {
-                                innerValue: 42,
-                                metadata: {
-                                    processed: true
-                                }
-                            };
-                        });
-                        // Create outer brain to test reducer type inference
-                        outerBrain = brain('Outer State Test').step('First step', function() {
-                            return {
-                                outerValue: 100,
-                                status: 'ready'
-                            };
-                        }).brain('Nested brain', innerBrain, function(param) {
-                            var state = param.state, brainState = param.brainState;
-                            var _outerAssert = true;
-                            var _innerAssert = true;
-                            return _object_spread_props(_object_spread({}, state), {
-                                innerResult: brainState.innerValue,
-                                processed: brainState.metadata.processed
-                            });
-                        }, function() {
-                            return {};
-                        });
-                        // Run the brain to verify runtime behavior
-                        finalState = {};
-                        _iteratorAbruptCompletion = false, _didIteratorError = false;
-                        _state.label = 1;
-                    case 1:
-                        _state.trys.push([
-                            1,
-                            6,
-                            7,
-                            12
-                        ]);
-                        _iterator = _async_iterator(outerBrain.run({
-                            client: mockClient
-                        }));
-                        _state.label = 2;
-                    case 2:
-                        return [
-                            4,
-                            _iterator.next()
-                        ];
-                    case 3:
-                        if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                            3,
-                            5
-                        ];
-                        _value = _step.value;
-                        event = _value;
-                        if (event.type === BRAIN_EVENTS.START && !mainBrainId) {
-                            mainBrainId = event.brainRunId;
-                        }
-                        if (event.type === BRAIN_EVENTS.STEP_COMPLETE && event.brainRunId === mainBrainId) {
-                            finalState = applyPatches(finalState, [
-                                event.patch
-                            ]);
-                        }
-                        _state.label = 4;
-                    case 4:
-                        _iteratorAbruptCompletion = false;
-                        return [
-                            3,
-                            2
-                        ];
-                    case 5:
-                        return [
-                            3,
-                            12
-                        ];
-                    case 6:
-                        err = _state.sent();
-                        _didIteratorError = true;
-                        _iteratorError = err;
-                        return [
-                            3,
-                            12
-                        ];
-                    case 7:
-                        _state.trys.push([
-                            7,
-                            ,
-                            10,
-                            11
-                        ]);
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                            3,
-                            9
-                        ];
-                        return [
-                            4,
-                            _iterator.return()
-                        ];
-                    case 8:
-                        _state.sent();
-                        _state.label = 9;
-                    case 9:
-                        return [
-                            3,
-                            11
-                        ];
-                    case 10:
-                        if (_didIteratorError) {
-                            throw _iteratorError;
-                        }
-                        return [
-                            7
-                        ];
-                    case 11:
-                        return [
-                            7
-                        ];
-                    case 12:
-                        expect(finalState).toEqual({
-                            outerValue: 100,
-                            status: 'ready',
-                            innerResult: 42,
-                            processed: true
-                        });
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-    it('should correctly infer step action state types', function() {
-        return _async_to_generator(function() {
-            var testBrain, finalState, mainBrainId, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        testBrain = brain('Action State Test').step('First step', function() {
-                            return {
-                                count: 1,
-                                metadata: {
-                                    created: new Date().toISOString()
-                                }
-                            };
-                        }).step('Second step', function(param) {
-                            var state = param.state;
-                            var _stateAssert = true;
-                            return _object_spread_props(_object_spread({}, state), {
-                                count: state.count + 1,
-                                metadata: _object_spread_props(_object_spread({}, state.metadata), {
-                                    updated: new Date().toISOString()
-                                })
-                            });
-                        });
-                        // Run the brain to verify runtime behavior
-                        finalState = {};
-                        _iteratorAbruptCompletion = false, _didIteratorError = false;
-                        _state.label = 1;
-                    case 1:
-                        _state.trys.push([
-                            1,
-                            6,
-                            7,
-                            12
-                        ]);
-                        _iterator = _async_iterator(testBrain.run({
-                            client: mockClient
-                        }));
-                        _state.label = 2;
-                    case 2:
-                        return [
-                            4,
-                            _iterator.next()
-                        ];
-                    case 3:
-                        if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                            3,
-                            5
-                        ];
-                        _value = _step.value;
-                        event = _value;
-                        if (event.type === BRAIN_EVENTS.START && !mainBrainId) {
-                            mainBrainId = event.brainRunId;
-                        }
-                        if (event.type === BRAIN_EVENTS.STEP_COMPLETE && event.brainRunId === mainBrainId) {
-                            finalState = applyPatches(finalState, [
-                                event.patch
-                            ]);
-                        }
-                        _state.label = 4;
-                    case 4:
-                        _iteratorAbruptCompletion = false;
-                        return [
-                            3,
-                            2
-                        ];
-                    case 5:
-                        return [
-                            3,
-                            12
-                        ];
-                    case 6:
-                        err = _state.sent();
-                        _didIteratorError = true;
-                        _iteratorError = err;
-                        return [
-                            3,
-                            12
-                        ];
-                    case 7:
-                        _state.trys.push([
-                            7,
-                            ,
-                            10,
-                            11
-                        ]);
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                            3,
-                            9
-                        ];
-                        return [
-                            4,
-                            _iterator.return()
-                        ];
-                    case 8:
-                        _state.sent();
-                        _state.label = 9;
-                    case 9:
-                        return [
-                            3,
-                            11
-                        ];
-                    case 10:
-                        if (_didIteratorError) {
-                            throw _iteratorError;
-                        }
-                        return [
-                            7
-                        ];
-                    case 11:
-                        return [
-                            7
-                        ];
-                    case 12:
-                        expect(finalState).toMatchObject({
-                            count: 2,
-                            metadata: {
-                                created: expect.any(String),
-                                updated: expect.any(String)
-                            }
-                        });
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-    it('should correctly infer prompt response types in subsequent steps', function() {
-        return _async_to_generator(function() {
-            var testBrain, finalState, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        testBrain = brain('Prompt Type Test').prompt('Get user info', {
-                            template: function() {
-                                return "What is the user's info?";
-                            },
-                            outputSchema: {
-                                schema: z.object({
-                                    name: z.string(),
-                                    age: z.number()
-                                }),
-                                name: 'userInfo'
-                            }
-                        }).step('Use response', function(param) {
-                            var state = param.state;
-                            var _stateAssert = true;
-                            return _object_spread_props(_object_spread({}, state), {
-                                greeting: "Hello ".concat(state.userInfo.name, ", you are ").concat(state.userInfo.age, " years old")
-                            });
-                        });
-                        // Mock the client response
-                        mockClient.generateObject.mockResolvedValueOnce({
-                            name: 'Test User',
-                            age: 30
-                        });
-                        // Run brain and collect final state
-                        finalState = {};
-                        _iteratorAbruptCompletion = false, _didIteratorError = false;
-                        _state.label = 1;
-                    case 1:
-                        _state.trys.push([
-                            1,
-                            6,
-                            7,
-                            12
-                        ]);
-                        _iterator = _async_iterator(testBrain.run({
-                            client: mockClient
-                        }));
-                        _state.label = 2;
-                    case 2:
-                        return [
-                            4,
-                            _iterator.next()
-                        ];
-                    case 3:
-                        if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                            3,
-                            5
-                        ];
-                        _value = _step.value;
-                        event = _value;
-                        if (event.type === BRAIN_EVENTS.STEP_COMPLETE) {
-                            finalState = applyPatches(finalState, [
-                                event.patch
-                            ]);
-                        }
-                        _state.label = 4;
-                    case 4:
-                        _iteratorAbruptCompletion = false;
-                        return [
-                            3,
-                            2
-                        ];
-                    case 5:
-                        return [
-                            3,
-                            12
-                        ];
-                    case 6:
-                        err = _state.sent();
-                        _didIteratorError = true;
-                        _iteratorError = err;
-                        return [
-                            3,
-                            12
-                        ];
-                    case 7:
-                        _state.trys.push([
-                            7,
-                            ,
-                            10,
-                            11
-                        ]);
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                            3,
-                            9
-                        ];
-                        return [
-                            4,
-                            _iterator.return()
-                        ];
-                    case 8:
-                        _state.sent();
-                        _state.label = 9;
-                    case 9:
-                        return [
-                            3,
-                            11
-                        ];
-                    case 10:
-                        if (_didIteratorError) {
-                            throw _iteratorError;
-                        }
-                        return [
-                            7
-                        ];
-                    case 11:
-                        return [
-                            7
-                        ];
-                    case 12:
-                        // Verify the brain executed correctly
-                        expect(finalState).toEqual({
-                            userInfo: {
-                                name: 'Test User',
-                                age: 30
-                            },
-                            greeting: 'Hello Test User, you are 30 years old'
-                        });
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-    it('should correctly handle prompt reduce function', function() {
-        return _async_to_generator(function() {
-            var testBrain, finalState, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err, _typeAssert;
-            return _ts_generator(this, function(_state) {
-                switch(_state.label){
-                    case 0:
-                        testBrain = brain('Prompt Reduce Test').prompt('Get numbers', {
-                            template: function() {
-                                return 'Give me some numbers';
-                            },
-                            outputSchema: {
-                                schema: z.object({
-                                    numbers: z.array(z.number())
-                                }),
-                                name: 'numbersResponse'
-                            }
-                        }, function(param) {
-                            var state = param.state, response = param.response, options = param.options;
-                            return _object_spread_props(_object_spread({}, state), {
-                                numbersResponse: response,
-                                sum: response.numbers.reduce(function(a, b) {
-                                    return a + b;
-                                }, 0),
-                                count: response.numbers.length
-                            });
-                        });
-                        // Mock the client response
-                        mockClient.generateObject.mockResolvedValueOnce({
-                            numbers: [
-                                1,
-                                2,
-                                3,
-                                4,
-                                5
-                            ]
-                        });
-                        // Run brain and collect final state
-                        finalState = {};
-                        _iteratorAbruptCompletion = false, _didIteratorError = false;
-                        _state.label = 1;
-                    case 1:
-                        _state.trys.push([
-                            1,
-                            6,
-                            7,
-                            12
-                        ]);
-                        _iterator = _async_iterator(testBrain.run({
-                            client: mockClient
-                        }));
-                        _state.label = 2;
-                    case 2:
-                        return [
-                            4,
-                            _iterator.next()
-                        ];
-                    case 3:
-                        if (!(_iteratorAbruptCompletion = !(_step = _state.sent()).done)) return [
-                            3,
-                            5
-                        ];
-                        _value = _step.value;
-                        event = _value;
-                        if (event.type === BRAIN_EVENTS.STEP_COMPLETE) {
-                            finalState = applyPatches(finalState, [
-                                event.patch
-                            ]);
-                        }
-                        _state.label = 4;
-                    case 4:
-                        _iteratorAbruptCompletion = false;
-                        return [
-                            3,
-                            2
-                        ];
-                    case 5:
-                        return [
-                            3,
-                            12
-                        ];
-                    case 6:
-                        err = _state.sent();
-                        _didIteratorError = true;
-                        _iteratorError = err;
-                        return [
-                            3,
-                            12
-                        ];
-                    case 7:
-                        _state.trys.push([
-                            7,
-                            ,
-                            10,
-                            11
-                        ]);
-                        if (!(_iteratorAbruptCompletion && _iterator.return != null)) return [
-                            3,
-                            9
-                        ];
-                        return [
-                            4,
-                            _iterator.return()
-                        ];
-                    case 8:
-                        _state.sent();
-                        _state.label = 9;
-                    case 9:
-                        return [
-                            3,
-                            11
-                        ];
-                    case 10:
-                        if (_didIteratorError) {
-                            throw _iteratorError;
-                        }
-                        return [
-                            7
-                        ];
-                    case 11:
-                        return [
-                            7
-                        ];
-                    case 12:
-                        // Verify the brain executed correctly with reduced state
-                        expect(finalState).toEqual({
-                            numbersResponse: {
-                                numbers: [
-                                    1,
-                                    2,
-                                    3,
-                                    4,
-                                    5
-                                ]
-                            },
-                            sum: 15,
-                            count: 5
-                        });
-                        _typeAssert = true;
-                        return [
-                            2
-                        ];
-                }
-            });
-        })();
-    });
-});
-describe('brain structure', function() {
-    it('should expose brain structure with steps', function() {
-        var testBrain = brain({
-            title: 'Test Brain',
-            description: 'A test brain description'
-        }).step('First step', function(param) {
-            var state = param.state;
-            return _object_spread_props(_object_spread({}, state), {
-                step1: true
-            });
-        }).step('Second step', function(param) {
-            var state = param.state;
-            return _object_spread_props(_object_spread({}, state), {
-                step2: true
-            });
-        }).step('Third step', function(param) {
-            var state = param.state;
-            return _object_spread_props(_object_spread({}, state), {
-                step3: true
-            });
-        });
-        var structure = testBrain.structure;
-        expect(structure).toEqual({
-            title: 'Test Brain',
-            description: 'A test brain description',
-            steps: [
-                {
-                    type: 'step',
-                    title: 'First step'
-                },
-                {
-                    type: 'step',
-                    title: 'Second step'
-                },
-                {
-                    type: 'step',
-                    title: 'Third step'
-                }
-            ]
-        });
-    });
-    it('should expose nested brain structure recursively', function() {
-        var innerBrain = brain({
-            title: 'Inner Brain',
-            description: 'An inner brain'
-        }).step('Inner step 1', function(param) {
-            var state = param.state;
-            return _object_spread_props(_object_spread({}, state), {
-                inner1: true
-            });
-        }).step('Inner step 2', function(param) {
-            var state = param.state;
-            return _object_spread_props(_object_spread({}, state), {
-                inner2: true
-            });
-        });
-        var outerBrain = brain({
-            title: 'Outer Brain',
-            description: 'An outer brain'
-        }).step('Outer step 1', function(param) {
-            var state = param.state;
-            return _object_spread_props(_object_spread({}, state), {
-                outer1: true
-            });
-        }).brain('Run inner brain', innerBrain, function(param) {
-            var brainState = param.brainState;
-            return {
-                result: brainState
-            };
-        }).step('Outer step 2', function(param) {
-            var state = param.state;
-            return _object_spread_props(_object_spread({}, state), {
-                outer2: true
-            });
-        });
-        var structure = outerBrain.structure;
-        expect(structure).toEqual({
-            title: 'Outer Brain',
-            description: 'An outer brain',
-            steps: [
-                {
-                    type: 'step',
-                    title: 'Outer step 1'
-                },
-                {
-                    type: 'brain',
-                    title: 'Run inner brain',
-                    innerBrain: {
-                        title: 'Inner Brain',
-                        description: 'An inner brain',
-                        steps: [
-                            {
-                                type: 'step',
-                                title: 'Inner step 1'
-                            },
-                            {
-                                type: 'step',
-                                title: 'Inner step 2'
-                            }
-                        ]
-                    }
-                },
-                {
-                    type: 'step',
-                    title: 'Outer step 2'
-                }
-            ]
-        });
-    });
-    it('should handle brain without description', function() {
-        var testBrain = brain('No Description Brain').step('Only step', function(param) {
-            var state = param.state;
-            return state;
-        });
-        var structure = testBrain.structure;
-        expect(structure).toEqual({
-            title: 'No Description Brain',
-            description: undefined,
-            steps: [
-                {
-                    type: 'step',
-                    title: 'Only step'
-                }
-            ]
-        });
-    });
-});