@positronic/core 0.0.1 → 0.0.3

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

@@ -0,0 +1,4225 @@
+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'
+                }
+            ]
+        });
+    });
+});