@positronic/core 0.0.52 → 0.0.53

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

@@ -0,0 +1,944 @@
+function _async_generator(gen) {
+    var front, back;
+    function send(key, arg) {
+        return new Promise(function(resolve, reject) {
+            var request = {
+                key: key,
+                arg: arg,
+                resolve: resolve,
+                reject: reject,
+                next: null
+            };
+            if (back) back = back.next = request;
+            else {
+                front = back = request;
+                resume(key, arg);
+            }
+        });
+    }
+    function resume(key, arg) {
+        try {
+            var result = gen[key](arg);
+            var value = result.value;
+            var overloaded = value instanceof _overload_yield;
+            Promise.resolve(overloaded ? value.v : value).then(function(arg) {
+                if (overloaded) {
+                    var nextKey = key === "return" ? "return" : "next";
+                    if (!value.k || arg.done) return resume(nextKey, arg);
+                    else arg = gen[nextKey](arg).value;
+                }
+                settle(result.done ? "return" : "normal", arg);
+            }, function(err) {
+                resume("throw", err);
+            });
+        } catch (err) {
+            settle("throw", err);
+        }
+    }
+    function settle(type, value) {
+        switch(type){
+            case "return":
+                front.resolve({
+                    value: value,
+                    done: true
+                });
+                break;
+            case "throw":
+                front.reject(value);
+                break;
+            default:
+                front.resolve({
+                    value: value,
+                    done: false
+                });
+                break;
+        }
+        front = front.next;
+        if (front) resume(front.key, front.arg);
+        else back = null;
+    }
+    this._invoke = send;
+    if (typeof gen.return !== "function") this.return = undefined;
+}
+_async_generator.prototype[typeof Symbol === "function" && Symbol.asyncIterator || "@@asyncIterator"] = function() {
+    return this;
+};
+_async_generator.prototype.next = function(arg) {
+    return this._invoke("next", arg);
+};
+_async_generator.prototype.throw = function(arg) {
+    return this._invoke("throw", arg);
+};
+_async_generator.prototype.return = function(arg) {
+    return this._invoke("return", arg);
+};
+function _async_generator_delegate(inner) {
+    var iter = {}, waiting = false;
+    function pump(key, value) {
+        waiting = true;
+        value = new Promise(function(resolve) {
+            resolve(inner[key](value));
+        });
+        return {
+            done: false,
+            value: new _overload_yield(value, 1)
+        };
+    }
+    iter[typeof Symbol !== "undefined" && Symbol.iterator || "@@iterator"] = function() {
+        return this;
+    };
+    iter.next = function(value) {
+        if (waiting) {
+            waiting = false;
+            return value;
+        }
+        return pump("next", value);
+    };
+    if (typeof inner.throw === "function") {
+        iter.throw = function(value) {
+            if (waiting) {
+                waiting = false;
+                throw value;
+            }
+            return pump("throw", value);
+        };
+    }
+    if (typeof inner.return === "function") {
+        iter.return = function(value) {
+            if (waiting) {
+                waiting = false;
+                return value;
+            }
+            return pump("return", value);
+        };
+    }
+    return iter;
+}
+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 _class_call_check(instance, Constructor) {
+    if (!(instance instanceof Constructor)) {
+        throw new TypeError("Cannot call a class as a function");
+    }
+}
+function _defineProperties(target, props) {
+    for(var i = 0; i < props.length; i++){
+        var descriptor = props[i];
+        descriptor.enumerable = descriptor.enumerable || false;
+        descriptor.configurable = true;
+        if ("value" in descriptor) descriptor.writable = true;
+        Object.defineProperty(target, descriptor.key, descriptor);
+    }
+}
+function _create_class(Constructor, protoProps, staticProps) {
+    if (protoProps) _defineProperties(Constructor.prototype, protoProps);
+    if (staticProps) _defineProperties(Constructor, staticProps);
+    return Constructor;
+}
+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 _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 _overload_yield(value, kind) {
+    this.v = value;
+    this.k = kind;
+}
+function _type_of(obj) {
+    "@swc/helpers - typeof";
+    return obj && typeof Symbol !== "undefined" && obj.constructor === Symbol ? "symbol" : typeof obj;
+}
+function _wrap_async_generator(fn) {
+    return function() {
+        return new _async_generator(fn.apply(this, arguments));
+    };
+}
+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
+        };
+    }
+}
+function _ts_values(o) {
+    var s = typeof Symbol === "function" && Symbol.iterator, m = s && o[s], i = 0;
+    if (m) return m.call(o);
+    if (o && typeof o.length === "number") return {
+        next: function() {
+            if (o && i >= o.length) o = void 0;
+            return {
+                value: o && o[i++],
+                done: !o
+            };
+        }
+    };
+    throw new TypeError(s ? "Object is not iterable." : "Symbol.iterator is not defined.");
+}
+import { z } from 'zod';
+import { BrainEventStream } from '../execution/event-stream.js';
+import { Semaphore, normalizeRetryConfig, executeWithRetry } from '../execution/retry.js';
+export var Brain = /*#__PURE__*/ function() {
+    "use strict";
+    function Brain(title, description) {
+        _class_call_check(this, Brain);
+        _define_property(this, "title", void 0);
+        _define_property(this, "description", void 0);
+        _define_property(this, "blocks", void 0);
+        _define_property(this, "type", void 0);
+        _define_property(this, "services", void 0);
+        _define_property(this, "optionsSchema", void 0);
+        _define_property(this, "components", void 0);
+        _define_property(this, "defaultTools", void 0);
+        this.title = title;
+        this.description = description;
+        this.blocks = [];
+        this.type = 'brain';
+        this.services = {};
+    }
+    _create_class(Brain, [
+        {
+            key: "structure",
+            get: function get() {
+                return {
+                    title: this.title,
+                    description: this.description,
+                    steps: this.blocks.map(function(block) {
+                        if (block.type === 'step') {
+                            return {
+                                type: 'step',
+                                title: block.title
+                            };
+                        } else if (block.type === 'agent') {
+                            return {
+                                type: 'agent',
+                                title: block.title
+                            };
+                        } else {
+                            // block.type === 'brain'
+                            return {
+                                type: 'brain',
+                                title: block.title,
+                                innerBrain: block.innerBrain.structure
+                            };
+                        }
+                    })
+                };
+            }
+        },
+        {
+            // New method to add services
+            key: "withServices",
+            value: function withServices(services) {
+                var nextBrain = new Brain(this.title, this.description).withBlocks(this.blocks);
+                // Set services
+                nextBrain.services = services;
+                // Copy optionsSchema to maintain it through the chain
+                nextBrain.optionsSchema = this.optionsSchema;
+                nextBrain.components = this.components;
+                nextBrain.defaultTools = this.defaultTools;
+                return nextBrain;
+            }
+        },
+        {
+            key: "withOptionsSchema",
+            value: function withOptionsSchema(schema) {
+                var nextBrain = new Brain(this.title, this.description).withBlocks(this.blocks);
+                nextBrain.optionsSchema = schema;
+                nextBrain.services = this.services;
+                nextBrain.components = this.components;
+                nextBrain.defaultTools = this.defaultTools;
+                return nextBrain;
+            }
+        },
+        {
+            /**
+   * Configure UI components for generative UI steps.
+   *
+   * @param components - Record of component definitions
+   *
+   * @example
+   * ```typescript
+   * import { components } from '@positronic/gen-ui-components';
+   *
+   * const myBrain = brain('my-brain')
+   *   .withComponents(components)
+   *   .ui('Show Form', formPrompt);
+   * ```
+   */ key: "withComponents",
+            value: function withComponents(components) {
+                var nextBrain = new Brain(this.title, this.description).withBlocks(this.blocks);
+                nextBrain.optionsSchema = this.optionsSchema;
+                nextBrain.services = this.services;
+                nextBrain.components = components;
+                nextBrain.defaultTools = this.defaultTools;
+                return nextBrain;
+            }
+        },
+        {
+            /**
+   * Configure default tools for agent steps.
+   * These tools will be automatically available in all agent steps and can be
+   * extended or overridden in individual step configurations.
+   *
+   * @param tools - Record of default tool definitions
+   *
+   * @example
+   * ```typescript
+   * import { defaultTools } from '@positronic/core';
+   *
+   * const myBrain = brain('my-brain')
+   *   .withTools(defaultTools)
+   *   .brain('agent', ({ tools }) => ({
+   *     system: 'You are helpful',
+   *     prompt: 'Do something',
+   *     tools  // uses defaults
+   *   }));
+   * ```
+   */ key: "withTools",
+            value: function withTools(tools) {
+                var next = this.nextBrain();
+                next.defaultTools = tools;
+                return next;
+            }
+        },
+        {
+            key: "step",
+            value: function step(title, action) {
+                var stepBlock = {
+                    type: 'step',
+                    title: title,
+                    action: action
+                };
+                this.blocks.push(stepBlock);
+                return this.nextBrain();
+            }
+        },
+        {
+            // Implementation
+            key: "brain",
+            value: function brain(title, innerBrainOrConfig, action, initialState) {
+                // Case 1: Nested brain instance
+                if (innerBrainOrConfig && (typeof innerBrainOrConfig === "undefined" ? "undefined" : _type_of(innerBrainOrConfig)) === 'object' && 'type' in innerBrainOrConfig && innerBrainOrConfig.type === 'brain') {
+                    var nestedBlock = {
+                        type: 'brain',
+                        title: title,
+                        innerBrain: innerBrainOrConfig,
+                        initialState: initialState || function() {
+                            return {};
+                        },
+                        action: function(outerState, innerState, services) {
+                            return action({
+                                state: outerState,
+                                brainState: innerState,
+                                services: services
+                            });
+                        }
+                    };
+                    this.blocks.push(nestedBlock);
+                    return this.nextBrain();
+                }
+                // Case 2 & 3: Agent config (object or function)
+                var configFn = typeof innerBrainOrConfig === 'function' ? innerBrainOrConfig : function() {
+                    return innerBrainOrConfig;
+                };
+                var agentBlock = {
+                    type: 'agent',
+                    title: title,
+                    configFn: configFn
+                };
+                this.blocks.push(agentBlock);
+                return this.nextBrain();
+            }
+        },
+        {
+            // Implementation
+            key: "prompt",
+            value: function prompt(title, config, batchConfig) {
+                // Schema-less prompt - returns text response for next step
+                if (!config.outputSchema) {
+                    var textSchema = z.object({
+                        text: z.string()
+                    });
+                    var promptBlock = {
+                        type: 'step',
+                        title: title,
+                        action: function(param) {
+                            var state = param.state, runClient = param.client, resources = param.resources;
+                            return _async_to_generator(function() {
+                                var template, stepClient, client, prompt, response;
+                                return _ts_generator(this, function(_state) {
+                                    switch(_state.label){
+                                        case 0:
+                                            template = config.template, stepClient = config.client;
+                                            client = stepClient !== null && stepClient !== void 0 ? stepClient : runClient;
+                                            return [
+                                                4,
+                                                template(state, resources)
+                                            ];
+                                        case 1:
+                                            prompt = _state.sent();
+                                            return [
+                                                4,
+                                                client.generateObject({
+                                                    schema: textSchema,
+                                                    schemaName: 'TextResponse',
+                                                    prompt: prompt
+                                                })
+                                            ];
+                                        case 2:
+                                            response = _state.sent();
+                                            return [
+                                                2,
+                                                {
+                                                    state: state,
+                                                    promptResponse: response
+                                                }
+                                            ];
+                                    }
+                                });
+                            })();
+                        }
+                    };
+                    this.blocks.push(promptBlock);
+                    return this.nextBrain();
+                }
+                // At this point, outputSchema is guaranteed to exist (schema-less case returned early)
+                var outputSchema = config.outputSchema;
+                if (batchConfig) {
+                    // Batch mode - run prompt for each item
+                    var promptBlock1 = {
+                        type: 'step',
+                        title: title,
+                        action: function(param) {
+                            var state = param.state, runClient = param.client, resources = param.resources;
+                            return _async_to_generator(function() {
+                                var template, stepClient, schema, schemaName, client, items, _batchConfig_concurrency, semaphore, _batchConfig_stagger, stagger, retryConfig, results, sleep, promises, finalResults;
+                                return _ts_generator(this, function(_state) {
+                                    switch(_state.label){
+                                        case 0:
+                                            template = config.template, stepClient = config.client;
+                                            schema = outputSchema.schema, schemaName = outputSchema.name;
+                                            client = stepClient !== null && stepClient !== void 0 ? stepClient : runClient;
+                                            items = batchConfig.over(state);
+                                            semaphore = new Semaphore((_batchConfig_concurrency = batchConfig.concurrency) !== null && _batchConfig_concurrency !== void 0 ? _batchConfig_concurrency : 10);
+                                            stagger = (_batchConfig_stagger = batchConfig.stagger) !== null && _batchConfig_stagger !== void 0 ? _batchConfig_stagger : 0;
+                                            retryConfig = normalizeRetryConfig(batchConfig.retry);
+                                            results = new Array(items.length);
+                                            sleep = function(ms) {
+                                                return new Promise(function(r) {
+                                                    return setTimeout(r, ms);
+                                                });
+                                            };
+                                            promises = items.map(function(item, index) {
+                                                return _async_to_generator(function() {
+                                                    var promptText, output, error, fallback;
+                                                    return _ts_generator(this, function(_state) {
+                                                        switch(_state.label){
+                                                            case 0:
+                                                                if (!(stagger > 0 && index > 0)) return [
+                                                                    3,
+                                                                    2
+                                                                ];
+                                                                return [
+                                                                    4,
+                                                                    sleep(stagger * index)
+                                                                ];
+                                                            case 1:
+                                                                _state.sent();
+                                                                _state.label = 2;
+                                                            case 2:
+                                                                return [
+                                                                    4,
+                                                                    semaphore.acquire()
+                                                                ];
+                                                            case 3:
+                                                                _state.sent();
+                                                                _state.label = 4;
+                                                            case 4:
+                                                                _state.trys.push([
+                                                                    4,
+                                                                    7,
+                                                                    8,
+                                                                    9
+                                                                ]);
+                                                                return [
+                                                                    4,
+                                                                    template(item, resources)
+                                                                ];
+                                                            case 5:
+                                                                promptText = _state.sent();
+                                                                return [
+                                                                    4,
+                                                                    executeWithRetry(function() {
+                                                                        return client.generateObject({
+                                                                            schema: schema,
+                                                                            schemaName: schemaName,
+                                                                            prompt: promptText
+                                                                        });
+                                                                    }, retryConfig)
+                                                                ];
+                                                            case 6:
+                                                                output = _state.sent();
+                                                                results[index] = [
+                                                                    item,
+                                                                    output
+                                                                ];
+                                                                return [
+                                                                    3,
+                                                                    9
+                                                                ];
+                                                            case 7:
+                                                                error = _state.sent();
+                                                                if (batchConfig.error) {
+                                                                    fallback = batchConfig.error(item, error);
+                                                                    if (fallback !== null) {
+                                                                        results[index] = [
+                                                                            item,
+                                                                            fallback
+                                                                        ];
+                                                                    }
+                                                                } else {
+                                                                    throw error;
+                                                                }
+                                                                return [
+                                                                    3,
+                                                                    9
+                                                                ];
+                                                            case 8:
+                                                                semaphore.release();
+                                                                return [
+                                                                    7
+                                                                ];
+                                                            case 9:
+                                                                return [
+                                                                    2
+                                                                ];
+                                                        }
+                                                    });
+                                                })();
+                                            });
+                                            return [
+                                                4,
+                                                Promise.all(promises)
+                                            ];
+                                        case 1:
+                                            _state.sent();
+                                            finalResults = results.filter(function(r) {
+                                                return r !== undefined;
+                                            });
+                                            return [
+                                                2,
+                                                _object_spread_props(_object_spread({}, state), _define_property({}, outputSchema.name, finalResults))
+                                            ];
+                                    }
+                                });
+                            })();
+                        }
+                    };
+                    this.blocks.push(promptBlock1);
+                    return this.nextBrain();
+                } else {
+                    // Single mode - run prompt once with current state
+                    var promptBlock2 = {
+                        type: 'step',
+                        title: title,
+                        action: function(param) {
+                            var state = param.state, runClient = param.client, resources = param.resources;
+                            return _async_to_generator(function() {
+                                var template, stepClient, schema, schemaName, client, prompt, response;
+                                return _ts_generator(this, function(_state) {
+                                    switch(_state.label){
+                                        case 0:
+                                            template = config.template, stepClient = config.client;
+                                            schema = outputSchema.schema, schemaName = outputSchema.name;
+                                            client = stepClient !== null && stepClient !== void 0 ? stepClient : runClient;
+                                            return [
+                                                4,
+                                                template(state, resources)
+                                            ];
+                                        case 1:
+                                            prompt = _state.sent();
+                                            return [
+                                                4,
+                                                client.generateObject({
+                                                    schema: schema,
+                                                    schemaName: schemaName,
+                                                    prompt: prompt
+                                                })
+                                            ];
+                                        case 2:
+                                            response = _state.sent();
+                                            return [
+                                                2,
+                                                _object_spread_props(_object_spread({}, state), _define_property({}, outputSchema.name, response))
+                                            ];
+                                    }
+                                });
+                            })();
+                        }
+                    };
+                    this.blocks.push(promptBlock2);
+                    return this.nextBrain();
+                }
+            }
+        },
+        {
+            /**
+   * Add a UI generation step that creates an interactive page.
+   *
+   * The step:
+   * 1. Calls an LLM agent to generate UI components based on the prompt
+   * 2. Renders the components to an HTML page
+   * 3. Stores the page and makes it available via URL
+   * 4. Creates a webhook for form submissions (typed based on responseSchema)
+   *
+   * The next step receives a `page` object with:
+   * - `url`: URL to the generated page
+   * - `webhook`: Pre-configured WebhookRegistration for form submissions
+   *
+   * The brain author is responsible for notifying users about the page (via Slack,
+   * email, etc.) and using `waitFor` to pause until the form is submitted.
+   * Form data arrives in the `response` parameter of the step after `waitFor`.
+   *
+   * @example
+   * ```typescript
+   * brain('feedback-form')
+   *   .step('Initialize', () => ({ userName: 'John' }))
+   *   .ui('Create Form', {
+   *     template: (state) => `Create a feedback form for ${state.userName}`,
+   *     responseSchema: z.object({
+   *       rating: z.number().min(1).max(5),
+   *       comments: z.string(),
+   *     }),
+   *   })
+   *   .step('Notify and Wait', async ({ state, page, slack }) => {
+   *     // Notify user however you want
+   *     await slack.post('#general', `Please fill out: ${page.url}`);
+   *     // Wait for form submission
+   *     return { state, waitFor: [page.webhook] };
+   *   })
+   *   .step('Process Feedback', ({ state, response }) => ({
+   *     ...state,
+   *     // response is typed: { rating: number, comments: string }
+   *     rating: response.rating,
+   *     comments: response.comments,
+   *   }))
+   * ```
+   */ key: "ui",
+            value: function ui(title, config) {
+                var uiBlock = {
+                    type: 'step',
+                    title: title,
+                    isUIStep: true,
+                    uiConfig: {
+                        template: config.template,
+                        responseSchema: config.responseSchema
+                    },
+                    action: function(params) {
+                        return _async_to_generator(function() {
+                            return _ts_generator(this, function(_state) {
+                                // The actual UI generation is handled by BrainRunner/BrainEventStream
+                                // This action is a placeholder that gets replaced during execution
+                                // when the runner detects `isUIStep: true` and has components configured
+                                throw new Error('UI step "'.concat(title, '" requires components to be configured via BrainRunner.withComponents(). ') + "The UI generation is handled by the runner, not the step action directly.");
+                            });
+                        })();
+                    }
+                };
+                this.blocks.push(uiBlock);
+                return this.nextBrain();
+            }
+        },
+        {
+            key: "run",
+            value: // Implementation signature
+            function run(params) {
+                return _wrap_async_generator(function() {
+                    var _this, title, description, blocks, validatedOptions, stream;
+                    return _ts_generator(this, function(_state) {
+                        switch(_state.label){
+                            case 0:
+                                _this = this, title = _this.title, description = _this.description, blocks = _this.blocks;
+                                if (this.optionsSchema) {
+                                    // Just call parse - Zod handles defaults automatically
+                                    validatedOptions = this.optionsSchema.parse(params.options || {});
+                                } else {
+                                    // If no schema is defined but options are provided, throw error
+                                    if (params.options && Object.keys(params.options).length > 0) {
+                                        throw new Error("Brain '".concat(this.title, "' received options but no schema was defined. Use withOptionsSchema() to define a schema for options."));
+                                    }
+                                    validatedOptions = {};
+                                }
+                                stream = new BrainEventStream(_object_spread_props(_object_spread({
+                                    title: title,
+                                    description: description,
+                                    blocks: blocks
+                                }, params), {
+                                    options: validatedOptions,
+                                    services: this.services,
+                                    components: this.components,
+                                    defaultTools: this.defaultTools
+                                }));
+                                return [
+                                    5,
+                                    _ts_values(_async_generator_delegate(_async_iterator(stream.next())))
+                                ];
+                            case 1:
+                                _state.sent();
+                                return [
+                                    2
+                                ];
+                        }
+                    });
+                }).call(this);
+            }
+        },
+        {
+            key: "withBlocks",
+            value: function withBlocks(blocks) {
+                this.blocks = blocks;
+                return this;
+            }
+        },
+        {
+            key: "nextBrain",
+            value: function nextBrain() {
+                // Pass default options to the next brain
+                var nextBrain = new Brain(this.title, this.description).withBlocks(this.blocks);
+                // Copy services to the next brain
+                nextBrain.services = this.services;
+                // Copy optionsSchema to the next brain
+                nextBrain.optionsSchema = this.optionsSchema;
+                // Copy components to the next brain
+                nextBrain.components = this.components;
+                // Copy defaultTools to the next brain
+                nextBrain.defaultTools = this.defaultTools;
+                return nextBrain;
+            }
+        }
+    ]);
+    return Brain;
+}();
+var brainNamesAreUnique = process.env.NODE_ENV !== 'test';
+var brainNames = new Set();
+/**
+ * Helper to register a brain name and check for uniqueness
+ */ function registerBrainName(title) {
+    if (brainNamesAreUnique && brainNames.has(title)) {
+        throw new Error('Brain with title "'.concat(title, '" already exists. Brain titles must be unique.'));
+    }
+    if (brainNamesAreUnique) {
+        brainNames.add(title);
+    }
+}
+// Implementation
+export function brain(titleOrConfig, agentConfig) {
+    var title = typeof titleOrConfig === 'string' ? titleOrConfig : titleOrConfig.title;
+    var description = typeof titleOrConfig === 'string' ? undefined : titleOrConfig.description;
+    registerBrainName(title);
+    var newBrain = new Brain(title, description);
+    // If agentConfig is provided, create a brain with a single 'main' agent step
+    if (agentConfig !== undefined) {
+        return newBrain.brain('main', agentConfig);
+    }
+    return newBrain;
+}