@positronic/core 0.0.67 → 0.0.69

package/dist/src/dsl/brain-runner.js

@@ -377,15 +377,31 @@ export var BrainRunner = /*#__PURE__*/ function() {
                 }));
             }
         },
+        {
+            key: "withGovernor",
+            value: function withGovernor(governor) {
+                return new BrainRunner(_object_spread_props(_object_spread({}, this.options), {
+                    governor: governor
+                }));
+            }
+        },
+        {
+            key: "withStoreProvider",
+            value: function withStoreProvider(storeProvider) {
+                return new BrainRunner(_object_spread_props(_object_spread({}, this.options), {
+                    storeProvider: storeProvider
+                }));
+            }
+        },
         {
             key: "run",
             value: /**
    * Run a brain from the beginning with fresh state.
    */ function run(brain, options) {
                 return _async_to_generator(function() {
-                    var _ref, _ref_initialState, initialState, brainOptions, brainRunId, endAfter, signal;
+                    var _options_initialState, initialState, brainOptions, brainRunId, endAfter, signal, currentUser;
                     return _ts_generator(this, function(_state) {
-                        _ref = options !== null && options !== void 0 ? options : {}, _ref_initialState = _ref.initialState, initialState = _ref_initialState === void 0 ? {} : _ref_initialState, brainOptions = _ref.options, brainRunId = _ref.brainRunId, endAfter = _ref.endAfter, signal = _ref.signal;
+                        _options_initialState = options.initialState, initialState = _options_initialState === void 0 ? {} : _options_initialState, brainOptions = options.options, brainRunId = options.brainRunId, endAfter = options.endAfter, signal = options.signal, currentUser = options.currentUser;
                         return [
                             2,
                             this.execute(brain, {
@@ -394,6 +410,7 @@ export var BrainRunner = /*#__PURE__*/ function() {
                                 brainRunId: brainRunId,
                                 endAfter: endAfter,
                                 signal: signal,
+                                currentUser: currentUser,
                                 initialStepCount: 0
                             })
                         ];
@@ -410,9 +427,9 @@ export var BrainRunner = /*#__PURE__*/ function() {
    * Webhook response data comes from signals, not as a parameter.
    */ function resume(brain, options) {
                 return _async_to_generator(function() {
-                    var machine, brainRunId, brainOptions, endAfter, signal, _machine_context, executionStack, agentContext, batchContext, resumeContext;
+                    var machine, brainRunId, brainOptions, endAfter, signal, currentUser, _machine_context, executionStack, agentContext, batchContext, resumeContext;
                     return _ts_generator(this, function(_state) {
-                        machine = options.machine, brainRunId = options.brainRunId, brainOptions = options.options, endAfter = options.endAfter, signal = options.signal;
+                        machine = options.machine, brainRunId = options.brainRunId, brainOptions = options.options, endAfter = options.endAfter, signal = options.signal, currentUser = options.currentUser;
                         // Build ResumeContext from machine's execution stack
                         // Webhook response comes from signals during execution, not from resume parameters
                         _machine_context = machine.context, executionStack = _machine_context.executionStack, agentContext = _machine_context.agentContext, batchContext = _machine_context.batchContext;
@@ -426,6 +443,7 @@ export var BrainRunner = /*#__PURE__*/ function() {
                                 brainRunId: brainRunId,
                                 endAfter: endAfter,
                                 signal: signal,
+                                currentUser: currentUser,
                                 initialStepCount: resumeContext.stepIndex
                             })
                         ];
@@ -439,13 +457,14 @@ export var BrainRunner = /*#__PURE__*/ function() {
    * Internal execution method shared by run() and resume().
    */ function execute(brain, params) {
                 return _async_to_generator(function() {
-                    var _this_options, adapters, client, resources, pages, env, signalProvider, resolvedEnv, initialState, resumeContext, providedMachine, options, brainRunId, endAfter, signal, initialStepCount, _resumeContext_state, _ref, machine, brainRun, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _loop, _iterator, _step, _ret, err, error, _machine_context_brainRunId, cancelledEvent;
+                    var _this_options, adapters, rawClient, resources, pages, env, signalProvider, governor, storeProvider, client, resolvedEnv, initialState, resumeContext, providedMachine, options, brainRunId, endAfter, signal, currentUser, initialStepCount, _resumeContext_state, _ref, machine, brainRun, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _loop, _iterator, _step, _ret, err, error, _machine_context_brainRunId, cancelledEvent;
                     return _ts_generator(this, function(_state) {
                         switch(_state.label){
                             case 0:
-                                _this_options = this.options, adapters = _this_options.adapters, client = _this_options.client, resources = _this_options.resources, pages = _this_options.pages, env = _this_options.env, signalProvider = _this_options.signalProvider;
+                                _this_options = this.options, adapters = _this_options.adapters, rawClient = _this_options.client, resources = _this_options.resources, pages = _this_options.pages, env = _this_options.env, signalProvider = _this_options.signalProvider, governor = _this_options.governor, storeProvider = _this_options.storeProvider;
+                                client = governor ? governor(rawClient) : rawClient;
                                 resolvedEnv = env !== null && env !== void 0 ? env : DEFAULT_ENV;
-                                initialState = params.initialState, resumeContext = params.resumeContext, providedMachine = params.machine, options = params.options, brainRunId = params.brainRunId, endAfter = params.endAfter, signal = params.signal, initialStepCount = params.initialStepCount;
+                                initialState = params.initialState, resumeContext = params.resumeContext, providedMachine = params.machine, options = params.options, brainRunId = params.brainRunId, endAfter = params.endAfter, signal = params.signal, currentUser = params.currentUser, initialStepCount = params.initialStepCount;
                                 // Use provided state machine if available (for resumes with historical events),
                                 // otherwise create a new one
                                 machine = providedMachine !== null && providedMachine !== void 0 ? providedMachine : createBrainExecutionMachine({
@@ -459,7 +478,10 @@ export var BrainRunner = /*#__PURE__*/ function() {
                                     resources: resources !== null && resources !== void 0 ? resources : {},
                                     pages: pages,
                                     env: resolvedEnv,
-                                    signalProvider: signalProvider
+                                    signalProvider: signalProvider,
+                                    governor: governor,
+                                    storeProvider: storeProvider,
+                                    currentUser: currentUser
                                 }) : brain.run({
                                     initialState: initialState !== null && initialState !== void 0 ? initialState : {},
                                     options: options,
@@ -468,7 +490,10 @@ export var BrainRunner = /*#__PURE__*/ function() {
                                     resources: resources !== null && resources !== void 0 ? resources : {},
                                     pages: pages,
                                     env: resolvedEnv,
-                                    signalProvider: signalProvider
+                                    signalProvider: signalProvider,
+                                    governor: governor,
+                                    storeProvider: storeProvider,
+                                    currentUser: currentUser
                                 });
                                 _state.label = 1;
                             case 1:

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

@@ -389,6 +389,7 @@ export var Brain = /*#__PURE__*/ function() {
         _define_property(this, "components", void 0);
         _define_property(this, "defaultTools", void 0);
         _define_property(this, "memoryProvider", void 0);
+        _define_property(this, "storeSchema", void 0);
         this.title = title;
         this.description = description;
         this.blocks = [];
@@ -447,6 +448,7 @@ export var Brain = /*#__PURE__*/ function() {
                 nextBrain.components = this.components;
                 nextBrain.defaultTools = this.defaultTools;
                 nextBrain.memoryProvider = this.memoryProvider;
+                nextBrain.storeSchema = this.storeSchema;
                 return nextBrain;
             }
         },
@@ -459,6 +461,7 @@ export var Brain = /*#__PURE__*/ function() {
                 nextBrain.components = this.components;
                 nextBrain.defaultTools = this.defaultTools;
                 nextBrain.memoryProvider = this.memoryProvider;
+                nextBrain.storeSchema = this.storeSchema;
                 return nextBrain;
             }
         },
@@ -484,6 +487,7 @@ export var Brain = /*#__PURE__*/ function() {
                 nextBrain.components = components;
                 nextBrain.defaultTools = this.defaultTools;
                 nextBrain.memoryProvider = this.memoryProvider;
+                nextBrain.storeSchema = this.storeSchema;
                 return nextBrain;
             }
         },
@@ -541,6 +545,39 @@ export var Brain = /*#__PURE__*/ function() {
                 return next;
             }
         },
+        {
+            /**
+   * Configure a typed key-value store for this brain.
+   * When configured, steps receive a `store` object in their context with typed get/set/delete/has.
+   *
+   * @param fields - Store field definitions. Plain values are shared; `{ value, perUser: true }` are per-user scoped.
+   *
+   * @example
+   * ```typescript
+   * const myBrain = brain('email-digest')
+   *   .withStore({
+   *     deselectedThreads: [] as string[],
+   *     lastDigestDate: '',
+   *     theme: { value: 'light', perUser: true },
+   *   })
+   *   .step('Process', async ({ store }) => {
+   *     const deselected = await store.get('deselectedThreads');
+   *     await store.set('deselectedThreads', [...deselected, 'new-id']);
+   *     return { processed: true };
+   *   });
+   * ```
+   */ key: "withStore",
+            value: function withStore(storeSchema) {
+                var nextBrain = new Brain(this.title, this.description).withBlocks(this.blocks);
+                nextBrain.optionsSchema = this.optionsSchema;
+                nextBrain.services = this.services;
+                nextBrain.components = this.components;
+                nextBrain.defaultTools = this.defaultTools;
+                nextBrain.memoryProvider = this.memoryProvider;
+                nextBrain.storeSchema = storeSchema;
+                return nextBrain;
+            }
+        },
         {
             key: "step",
             value: function step(title, action) {
@@ -628,18 +665,17 @@ export var Brain = /*#__PURE__*/ function() {
                     var promptBlock = {
                         type: 'step',
                         title: title,
+                        client: config.client,
                         action: function(param) {
-                            var state = param.state, runClient = param.client, resources = param.resources;
+                            var state = param.state, client = param.client, resources = param.resources;
                             return _async_to_generator(function() {
-                                var template, stepClient, client, prompt, response;
+                                var prompt, result;
                                 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)
+                                                config.template(state, resources)
                                             ];
                                         case 1:
                                             prompt = _state.sent();
@@ -652,12 +688,12 @@ export var Brain = /*#__PURE__*/ function() {
                                                 })
                                             ];
                                         case 2:
-                                            response = _state.sent();
+                                            result = _state.sent();
                                             return [
                                                 2,
                                                 {
                                                     state: state,
-                                                    promptResponse: response
+                                                    promptResponse: result.object
                                                 }
                                             ];
                                     }
@@ -703,19 +739,18 @@ export var Brain = /*#__PURE__*/ function() {
                     var promptBlock2 = {
                         type: 'step',
                         title: title,
+                        client: config.client,
                         action: function(param) {
-                            var state = param.state, runClient = param.client, resources = param.resources;
+                            var state = param.state, client = param.client, resources = param.resources;
                             return _async_to_generator(function() {
-                                var template, stepClient, schema, schemaName, client, prompt, response;
+                                var schema, schemaName, prompt, result;
                                 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)
+                                                config.template(state, resources)
                                             ];
                                         case 1:
                                             prompt = _state.sent();
@@ -728,10 +763,10 @@ export var Brain = /*#__PURE__*/ function() {
                                                 })
                                             ];
                                         case 2:
-                                            response = _state.sent();
+                                            result = _state.sent();
                                             return [
                                                 2,
-                                                _object_spread_props(_object_spread({}, state), _define_property({}, outputSchema.name, response))
+                                                _object_spread_props(_object_spread({}, state), _define_property({}, outputSchema.name, result.object))
                                             ];
                                     }
                                 });
@@ -814,7 +849,7 @@ export var Brain = /*#__PURE__*/ function() {
             value: // Implementation signature
             function run(params) {
                 return _wrap_async_generator(function() {
-                    var _this, title, description, blocks, validatedOptions, stream;
+                    var _this, title, description, blocks, validatedOptions, store, stream;
                     return _ts_generator(this, function(_state) {
                         switch(_state.label){
                             case 0:
@@ -829,6 +864,12 @@ export var Brain = /*#__PURE__*/ function() {
                                     }
                                     validatedOptions = {};
                                 }
+                                // Build store if withStore() was called and a store provider is given
+                                store = this.storeSchema && params.storeProvider ? params.storeProvider({
+                                    schema: this.storeSchema,
+                                    brainTitle: this.title,
+                                    currentUser: params.currentUser
+                                }) : undefined;
                                 stream = new BrainEventStream(_object_spread_props(_object_spread({
                                     title: title,
                                     description: description,
@@ -838,7 +879,8 @@ export var Brain = /*#__PURE__*/ function() {
                                     services: this.services,
                                     components: this.components,
                                     defaultTools: this.defaultTools,
-                                    memoryProvider: this.memoryProvider
+                                    memoryProvider: this.memoryProvider,
+                                    store: store
                                 }));
                                 return [
                                     5,
@@ -876,6 +918,8 @@ export var Brain = /*#__PURE__*/ function() {
                 nextBrain.defaultTools = this.defaultTools;
                 // Copy memoryProvider to the next brain
                 nextBrain.memoryProvider = this.memoryProvider;
+                // Copy store schema to the next brain
+                nextBrain.storeSchema = this.storeSchema;
                 return nextBrain;
             }
         }

package/dist/src/dsl/create-brain.js

@@ -41,7 +41,7 @@ import { brain as coreBrain } from './builder/brain.js';
  * }));
  * ```
  */ export function createBrain(config) {
-    var services = config.services, components = config.components, defaultTools = config.defaultTools, memory = config.memory;
+    var services = config.services, components = config.components, defaultTools = config.defaultTools, memory = config.memory, store = config.store;
     // Implementation
     function brain(titleOrConfig, agentConfig) {
         var base = coreBrain(titleOrConfig);
@@ -54,6 +54,9 @@ import { brain as coreBrain } from './builder/brain.js';
         if (memory) {
             base = base.withMemory(memory);
         }
+        if (store) {
+            base = base.withStore(store);
+        }
         if (services) {
             base = base.withServices(services);
         }

package/dist/src/dsl/execution/event-stream.js

@@ -537,10 +537,16 @@ export var BrainEventStream = /*#__PURE__*/ function() {
         _define_property(this, "signalProvider", void 0);
         _define_property(this, "memoryProvider", void 0);
         _define_property(this, "scopedMemory", void 0);
+        _define_property(this, "store", void 0);
+        _define_property(this, "governor", void 0);
+        _define_property(this, "currentUser", void 0);
         _define_property(this, "guards", new Map());
         _define_property(this, "waits", new Map());
         _define_property(this, "stopped", false);
-        var blocks = params.blocks, title = params.title, description = params.description, providedBrainRunId = params.brainRunId, _params_options = params.options, options = _params_options === void 0 ? {} : _params_options, client = params.client, services = params.services, _params_resources = params.resources, resources = _params_resources === void 0 ? {} : _params_resources, pages = params.pages, env = params.env, components = params.components, defaultTools = params.defaultTools, signalProvider = params.signalProvider, memoryProvider = params.memoryProvider;
+        var blocks = params.blocks, title = params.title, description = params.description, providedBrainRunId = params.brainRunId, _params_options = params.options, options = _params_options === void 0 ? {} : _params_options, client = params.client, services = params.services, _params_resources = params.resources, resources = _params_resources === void 0 ? {} : _params_resources, pages = params.pages, env = params.env, components = params.components, defaultTools = params.defaultTools, signalProvider = params.signalProvider, memoryProvider = params.memoryProvider, store = params.store, currentUser = params.currentUser;
+        // Store governor for per-step client resolution
+        this.governor = params.governor;
+        this.currentUser = currentUser;
         // Check if this is a resume run or fresh start
         var resumeParams = params;
         var initialParams = params;
@@ -558,9 +564,10 @@ export var BrainEventStream = /*#__PURE__*/ function() {
         this.defaultTools = defaultTools;
         this.signalProvider = signalProvider;
         this.memoryProvider = memoryProvider;
+        this.store = store;
         // Create scoped memory if provider is configured
         if (memoryProvider) {
-            this.scopedMemory = createScopedMemory(memoryProvider, title);
+            this.scopedMemory = createScopedMemory(memoryProvider, title, this.currentUser.id);
         }
         // Initialize steps - track guard and wait blocks by index
         this.steps = [];
@@ -655,7 +662,8 @@ export var BrainEventStream = /*#__PURE__*/ function() {
                                         brainDescription: brainDescription,
                                         initialState: currentState,
                                         options: options,
-                                        brainRunId: brainRunId
+                                        brainRunId: brainRunId,
+                                        currentUser: this.currentUser
                                     }
                                 ];
                             case 2:
@@ -1048,7 +1056,7 @@ export var BrainEventStream = /*#__PURE__*/ function() {
             key: "executeStep",
             value: function executeStep(step) {
                 return _wrap_async_generator(function() {
-                    var block, stepBlock, _this_resumeContext, brainBlock, initialState, innerResumeContext, patches, innerBrainPaused, _this_options, _this_options1, innerRun, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err, _innerResumeContext_state, baseState, innerState, prevState, _, prevState1, stepBlock1, _this_options2, result;
+                    var block, stepBlock, _this_resumeContext, brainBlock, initialState, innerResumeContext, patches, innerBrainPaused, _this_options, _this_options1, innerRun, _iteratorAbruptCompletion, _didIteratorError, _iteratorError, _iterator, _step, _value, event, err, _innerResumeContext_state, baseState, innerState, prevState, _, prevState1, stepBlock1, stepClient, _this_options2, result;
                     return _ts_generator(this, function(_state) {
                         switch(_state.label){
                             case 0:
@@ -1101,19 +1109,23 @@ export var BrainEventStream = /*#__PURE__*/ function() {
                                 innerRun = innerResumeContext ? brainBlock.innerBrain.run({
                                     resources: this.resources,
                                     client: this.client,
+                                    currentUser: this.currentUser,
                                     resumeContext: innerResumeContext,
                                     options: (_this_options = this.options) !== null && _this_options !== void 0 ? _this_options : {},
                                     pages: this.pages,
                                     env: this.env,
-                                    brainRunId: this.brainRunId
+                                    brainRunId: this.brainRunId,
+                                    governor: this.governor
                                 }) : brainBlock.innerBrain.run({
                                     resources: this.resources,
                                     client: this.client,
+                                    currentUser: this.currentUser,
                                     initialState: initialState,
                                     options: (_this_options1 = this.options) !== null && _this_options1 !== void 0 ? _this_options1 : {},
                                     pages: this.pages,
                                     env: this.env,
-                                    brainRunId: this.brainRunId
+                                    brainRunId: this.brainRunId,
+                                    governor: this.governor
                                 });
                                 _iteratorAbruptCompletion = false, _didIteratorError = false;
                                 _state.label = 5;
@@ -1263,18 +1275,23 @@ export var BrainEventStream = /*#__PURE__*/ function() {
                                 // Get previous state before action
                                 prevState1 = this.currentState;
                                 stepBlock1 = block;
+                                // Resolve per-step client: if the step has an override, apply governor to it;
+                                // otherwise use the default (already-governed) client
+                                stepClient = stepBlock1.client ? this.governor ? this.governor(stepBlock1.client) : stepBlock1.client : this.client;
                                 return [
                                     4,
                                     _await_async_generator(Promise.resolve(stepBlock1.action(_object_spread({
                                         state: this.currentState,
                                         options: (_this_options2 = this.options) !== null && _this_options2 !== void 0 ? _this_options2 : {},
-                                        client: this.client,
+                                        client: stepClient,
                                         resources: this.resources,
                                         response: this.currentResponse,
                                         page: this.currentPage,
                                         pages: this.pages,
                                         env: this.env,
-                                        memory: this.scopedMemory
+                                        memory: this.scopedMemory,
+                                        store: this.store,
+                                        currentUser: this.currentUser
                                     }, this.services))))
                                 ];
                             case 23:
@@ -1333,7 +1350,9 @@ export var BrainEventStream = /*#__PURE__*/ function() {
                                         page: this.currentPage,
                                         pages: this.pages,
                                         env: this.env,
-                                        memory: this.scopedMemory
+                                        memory: this.scopedMemory,
+                                        store: this.store,
+                                        currentUser: this.currentUser
                                     }, this.services)))
                                 ];
                             case 1:
@@ -1945,7 +1964,9 @@ export var BrainEventStream = /*#__PURE__*/ function() {
                                     components: this.components,
                                     brainRunId: this.brainRunId,
                                     stepId: effectiveStepId,
-                                    memory: this.scopedMemory
+                                    memory: this.scopedMemory,
+                                    store: this.store,
+                                    currentUser: this.currentUser
                                 };
                                 return [
                                     4,
@@ -2148,7 +2169,7 @@ export var BrainEventStream = /*#__PURE__*/ function() {
    * Cloudflare backends can restart the DO to reclaim memory.
    */ function executeBatchPrompt(step) {
                 return _wrap_async_generator(function() {
-                    var _this, _this_resumeContext, block, batchConfig, prevState, _batchConfig_client, client, items, totalItems, _batchConfig_concurrency, concurrency, semaphore, batchProgress, _batchProgress_processedCount, startIndex, results, chunkStart, signals, _iteratorNormalCompletion, _didIteratorError, _iteratorError, _iterator, _step, signal, _this_options, err, chunkEnd, chunk, chunkResults, i, _this_options1, finalResults;
+                    var _this, _this_resumeContext, block, batchConfig, prevState, rawClient, client, items, totalItems, _batchConfig_concurrency, concurrency, semaphore, batchProgress, _batchProgress_processedCount, startIndex, results, chunkStart, signals, _iteratorNormalCompletion, _didIteratorError, _iteratorError, _iterator, _step, signal, _this_options, err, chunkEnd, chunk, chunkResults, i, _this_options1, finalResults;
                     return _ts_generator(this, function(_state) {
                         switch(_state.label){
                             case 0:
@@ -2156,7 +2177,8 @@ export var BrainEventStream = /*#__PURE__*/ function() {
                                 block = step.block;
                                 batchConfig = block.batchConfig;
                                 prevState = this.currentState;
-                                client = (_batchConfig_client = batchConfig.client) !== null && _batchConfig_client !== void 0 ? _batchConfig_client : this.client;
+                                rawClient = batchConfig.client;
+                                client = rawClient ? this.governor ? this.governor(rawClient) : rawClient : this.client;
                                 items = batchConfig.over(this.currentState);
                                 totalItems = items.length;
                                 concurrency = (_batchConfig_concurrency = batchConfig.concurrency) !== null && _batchConfig_concurrency !== void 0 ? _batchConfig_concurrency : 10;
@@ -2273,7 +2295,7 @@ export var BrainEventStream = /*#__PURE__*/ function() {
                                     4,
                                     _await_async_generator(Promise.all(chunk.map(function(item) {
                                         return _async_to_generator(function() {
-                                            var release, promptText, output, error, fallback;
+                                            var release, promptText, result, error, fallback;
                                             return _ts_generator(this, function(_state) {
                                                 switch(_state.label){
                                                     case 0:
@@ -2306,12 +2328,12 @@ export var BrainEventStream = /*#__PURE__*/ function() {
                                                             })
                                                         ];
                                                     case 4:
-                                                        output = _state.sent();
+                                                        result = _state.sent();
                                                         return [
                                                             2,
                                                             [
                                                                 item,
-                                                                output
+                                                                result.object
                                                             ]
                                                         ];
                                                     case 5:
@@ -2544,7 +2566,9 @@ export var BrainEventStream = /*#__PURE__*/ function() {
                                         resources: this.resources,
                                         page: this.currentPage,
                                         pages: this.pages,
-                                        env: this.env
+                                        env: this.env,
+                                        store: this.store,
+                                        currentUser: this.currentUser
                                     }, this.services)))
                                 ];
                             case 3:

package/dist/src/memory/scoped-memory.js

@@ -119,24 +119,25 @@ function _ts_generator(thisArg, body) {
     }
 }
 /**
- * Creates a scoped memory instance with the agentId pre-bound.
+ * Creates a scoped memory instance with agentId and userId pre-bound.
  *
  * This wraps a MemoryProvider and automatically includes the agentId
- * in all calls, so brain steps don't need to pass it explicitly.
+ * and userId in all calls, so brain steps don't need to pass them explicitly.
  *
  * @param provider - The underlying memory provider
  * @param agentId - The agent/brain ID to scope memories to
+ * @param userId - The user ID to scope memories to (from currentUser.id)
  * @returns A ScopedMemory instance
  *
  * @example
  * ```typescript
  * const provider = createMem0Provider({ apiKey: '...' });
- * const scopedMemory = createScopedMemory(provider, 'my-brain');
+ * const scopedMemory = createScopedMemory(provider, 'my-brain', 'user-123');
  *
- * // Now search without passing agentId
+ * // Now search without passing agentId or userId
  * const memories = await scopedMemory.search('user preferences');
  * ```
- */ export function createScopedMemory(provider, agentId) {
+ */ export function createScopedMemory(provider, agentId, userId) {
     return {
         search: function search(query, options) {
             return _async_to_generator(function() {
@@ -144,7 +145,7 @@ function _ts_generator(thisArg, body) {
                 return _ts_generator(this, function(_state) {
                     scope = {
                         agentId: agentId,
-                        userId: options === null || options === void 0 ? void 0 : options.userId
+                        userId: userId
                     };
                     return [
                         2,
@@ -161,7 +162,7 @@ function _ts_generator(thisArg, body) {
                 return _ts_generator(this, function(_state) {
                     scope = {
                         agentId: agentId,
-                        userId: options === null || options === void 0 ? void 0 : options.userId
+                        userId: userId
                     };
                     return [
                         2,

package/dist/src/memory/types.js

@@ -7,6 +7,6 @@
  */ /**
  * A single memory entry returned from search operations.
  */ /**
- * Scoped memory interface with agentId pre-bound.
- * This is what brain steps receive - they don't need to pass agentId.
+ * Scoped memory interface with agentId and userId pre-bound.
+ * This is what brain steps receive - they don't need to pass agentId or userId.
  */ export { };

package/dist/src/store/index.js

@@ -0,0 +1 @@
+export { };

package/dist/src/store/types.js

@@ -0,0 +1,5 @@
+/**
+ * Conditional type that adds `store` to step params only when withStore() is used.
+ * When TStore is `never` (default), this resolves to `{}` — no store in params.
+ * When TStore is a record type, this adds `{ store: Store<TStore> }`.
+ */ export { };

package/dist/types/clients/types.d.ts

@@ -47,6 +47,12 @@ export type ResponseMessage = unknown;
  * and potentially other types of content in the future.
  */
 export interface ObjectGenerator {
+    /** SHA-256 hash of `model:apiKey`, used for rate-limit bucket identification. */
+    identity?: string;
+    /** The model identifier string (e.g. 'gemini-3-pro-preview'). */
+    modelId?: string;
+    /** The API key used by this client, for rate-limit bucket identification. */
+    apiKey?: string;
     /**
      * Generates a structured JSON object that conforms to the provided Zod schema.
      *
@@ -81,7 +87,13 @@ export interface ObjectGenerator {
          * prepend this as a `system` role message to the full message list.
          */
         system?: string;
-    }): Promise<z.infer<T>>;
+    }): Promise<{
+        object: z.infer<T>;
+        usage?: {
+            totalTokens: number;
+        };
+        responseHeaders?: Record<string, string>;
+    }>;
     /**
      * Creates a tool result message in the SDK-native format.
      * Use this to append tool results to responseMessages before the next generateText call.
@@ -135,6 +147,8 @@ export interface ObjectGenerator {
          * Pass these back in the next call via responseMessages parameter.
          */
         responseMessages: ResponseMessage[];
+        /** Raw response headers from the provider, used for rate limit tracking. */
+        responseHeaders?: Record<string, string>;
     }>;
     /**
      * Generates text with multi-step tool calling support.
@@ -185,6 +199,8 @@ export interface ObjectGenerator {
         usage: {
             totalTokens: number;
         };
+        /** Raw response headers from the provider, used for rate limit tracking. */
+        responseHeaders?: Record<string, string>;
     }>;
 }
 //# sourceMappingURL=types.d.ts.map

package/dist/types/clients/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/clients/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB;;;;;GAKG;AACH,MAAM,MAAM,UAAU,GAAG,MAAM,GAAG,UAAU,GAAG,MAAM,CAAC;AAEtD;;GAEG;AACH,MAAM,MAAM,OAAO,GAAG;IACpB,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,QAAQ,GAAG;IACrB,IAAI,EAAE,UAAU,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,OAAO,CAAC;CACf,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,WAAW,GAAG;IACxB,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,GAAG,MAAM,CAAC;IAC/C,OAAO,EAAE,MAAM,CAAC;IAChB,oEAAoE;IACpE,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,4CAA4C;IAC5C,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,6DAA6D;IAC7D,SAAS,CAAC,EAAE,QAAQ,EAAE,CAAC;CACxB,CAAC;AAEF;;;;GAIG;AACH,MAAM,MAAM,eAAe,GAAG,OAAO,CAAC;AAEtC;;;GAGG;AACH,MAAM,WAAW,eAAe;IAC9B;;;;;OAKG;IACH,cAAc,CAAC,CAAC,SAAS,CAAC,CAAC,YAAY,EAAE,MAAM,EAAE;QAC/C;;;WAGG;QACH,MAAM,EAAE,CAAC,CAAC;QACV,UAAU,EAAE,MAAM,CAAC;QACnB,iBAAiB,CAAC,EAAE,MAAM,CAAC;QAE3B;;;;;WAKG;QACH,MAAM,CAAC,EAAE,MAAM,CAAC;QAEhB;;;;;WAKG;QACH,QAAQ,CAAC,EAAE,OAAO,EAAE,CAAC;QAErB;;;;WAIG;QACH,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAExB;;;OAGG;IACH,uBAAuB,CAAC,CACtB,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,OAAO,GACd,eAAe,CAAC;IAEnB;;;OAGG;IACH,YAAY,CAAC,CAAC,MAAM,EAAE;QACpB,gCAAgC;QAChC,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB;;WAEG;QACH,QAAQ,EAAE,WAAW,EAAE,CAAC;QACxB;;;;WAIG;QACH,gBAAgB,CAAC,EAAE,eAAe,EAAE,CAAC;QACrC,0CAA0C;QAC1C,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE;YAAE,WAAW,EAAE,MAAM,CAAC;YAAC,WAAW,EAAE,CAAC,CAAC,SAAS,CAAA;SAAE,CAAC,CAAC;QACzE;;;;;;WAMG;QACH,UAAU,CAAC,EAAE,UAAU,CAAC;KACzB,GAAG,OAAO,CAAC;QACV,iCAAiC;QACjC,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,iCAAiC;QACjC,SAAS,CAAC,EAAE,KAAK,CAAC;YAChB,UAAU,EAAE,MAAM,CAAC;YACnB,QAAQ,EAAE,MAAM,CAAC;YACjB,IAAI,EAAE,OAAO,CAAC;SACf,CAAC,CAAC;QACH,8BAA8B;QAC9B,KAAK,EAAE;YAAE,WAAW,EAAE,MAAM,CAAA;SAAE,CAAC;QAC/B;;;WAGG;QACH,gBAAgB,EAAE,eAAe,EAAE,CAAC;KACrC,CAAC,CAAC;IAEH;;;;;;;;;;;OAWG;IACH,UAAU,CAAC,MAAM,EAAE;QACjB,gCAAgC;QAChC,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,0BAA0B;QAC1B,MAAM,EAAE,MAAM,CAAC;QACf,oDAAoD;QACpD,QAAQ,CAAC,EAAE,WAAW,EAAE,CAAC;QACzB,0CAA0C;QAC1C,KAAK,EAAE,MAAM,CACX,MAAM,EACN;YACE,WAAW,EAAE,MAAM,CAAC;YACpB,WAAW,EAAE,CAAC,CAAC,SAAS,CAAC;YACzB,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,OAAO,KAAK,OAAO,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC;SACzD,CACF,CAAC;QACF,qDAAqD;QACrD,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB;;;;;;WAMG;QACH,UAAU,CAAC,EAAE,UAAU,CAAC;KACzB,GAAG,OAAO,CAAC;QACV,+DAA+D;QAC/D,SAAS,EAAE,KAAK,CAAC;YACf,UAAU,EAAE,MAAM,CAAC;YACnB,QAAQ,EAAE,MAAM,CAAC;YACjB,IAAI,EAAE,OAAO,CAAC;YACd,MAAM,EAAE,OAAO,CAAC;SACjB,CAAC,CAAC;QACH,mCAAmC;QACnC,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,mCAAmC;QACnC,KAAK,EAAE;YAAE,WAAW,EAAE,MAAM,CAAA;SAAE,CAAC;KAChC,CAAC,CAAC;CACJ"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/clients/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB;;;;;GAKG;AACH,MAAM,MAAM,UAAU,GAAG,MAAM,GAAG,UAAU,GAAG,MAAM,CAAC;AAEtD;;GAEG;AACH,MAAM,MAAM,OAAO,GAAG;IACpB,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;CACjB,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,QAAQ,GAAG;IACrB,IAAI,EAAE,UAAU,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,OAAO,CAAC;CACf,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,WAAW,GAAG;IACxB,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,GAAG,MAAM,CAAC;IAC/C,OAAO,EAAE,MAAM,CAAC;IAChB,oEAAoE;IACpE,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,4CAA4C;IAC5C,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,6DAA6D;IAC7D,SAAS,CAAC,EAAE,QAAQ,EAAE,CAAC;CACxB,CAAC;AAEF;;;;GAIG;AACH,MAAM,MAAM,eAAe,GAAG,OAAO,CAAC;AAEtC;;;GAGG;AACH,MAAM,WAAW,eAAe;IAC9B,iFAAiF;IACjF,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB,iEAAiE;IACjE,OAAO,CAAC,EAAE,MAAM,CAAC;IAEjB,6EAA6E;IAC7E,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB;;;;;OAKG;IACH,cAAc,CAAC,CAAC,SAAS,CAAC,CAAC,YAAY,EAAE,MAAM,EAAE;QAC/C;;;WAGG;QACH,MAAM,EAAE,CAAC,CAAC;QACV,UAAU,EAAE,MAAM,CAAC;QACnB,iBAAiB,CAAC,EAAE,MAAM,CAAC;QAE3B;;;;;WAKG;QACH,MAAM,CAAC,EAAE,MAAM,CAAC;QAEhB;;;;;WAKG;QACH,QAAQ,CAAC,EAAE,OAAO,EAAE,CAAC;QAErB;;;;WAIG;QACH,MAAM,CAAC,EAAE,MAAM,CAAC;KACjB,GAAG,OAAO,CAAC;QACV,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACnB,KAAK,CAAC,EAAE;YAAE,WAAW,EAAE,MAAM,CAAA;SAAE,CAAC;QAChC,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KAC1C,CAAC,CAAC;IAEH;;;OAGG;IACH,uBAAuB,CAAC,CACtB,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,OAAO,GACd,eAAe,CAAC;IAEnB;;;OAGG;IACH,YAAY,CAAC,CAAC,MAAM,EAAE;QACpB,gCAAgC;QAChC,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB;;WAEG;QACH,QAAQ,EAAE,WAAW,EAAE,CAAC;QACxB;;;;WAIG;QACH,gBAAgB,CAAC,EAAE,eAAe,EAAE,CAAC;QACrC,0CAA0C;QAC1C,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE;YAAE,WAAW,EAAE,MAAM,CAAC;YAAC,WAAW,EAAE,CAAC,CAAC,SAAS,CAAA;SAAE,CAAC,CAAC;QACzE;;;;;;WAMG;QACH,UAAU,CAAC,EAAE,UAAU,CAAC;KACzB,GAAG,OAAO,CAAC;QACV,iCAAiC;QACjC,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,iCAAiC;QACjC,SAAS,CAAC,EAAE,KAAK,CAAC;YAChB,UAAU,EAAE,MAAM,CAAC;YACnB,QAAQ,EAAE,MAAM,CAAC;YACjB,IAAI,EAAE,OAAO,CAAC;SACf,CAAC,CAAC;QACH,8BAA8B;QAC9B,KAAK,EAAE;YAAE,WAAW,EAAE,MAAM,CAAA;SAAE,CAAC;QAC/B;;;WAGG;QACH,gBAAgB,EAAE,eAAe,EAAE,CAAC;QACpC,4EAA4E;QAC5E,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KAC1C,CAAC,CAAC;IAEH;;;;;;;;;;;OAWG;IACH,UAAU,CAAC,MAAM,EAAE;QACjB,gCAAgC;QAChC,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,0BAA0B;QAC1B,MAAM,EAAE,MAAM,CAAC;QACf,oDAAoD;QACpD,QAAQ,CAAC,EAAE,WAAW,EAAE,CAAC;QACzB,0CAA0C;QAC1C,KAAK,EAAE,MAAM,CACX,MAAM,EACN;YACE,WAAW,EAAE,MAAM,CAAC;YACpB,WAAW,EAAE,CAAC,CAAC,SAAS,CAAC;YACzB,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,OAAO,KAAK,OAAO,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC;SACzD,CACF,CAAC;QACF,qDAAqD;QACrD,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB;;;;;;WAMG;QACH,UAAU,CAAC,EAAE,UAAU,CAAC;KACzB,GAAG,OAAO,CAAC;QACV,+DAA+D;QAC/D,SAAS,EAAE,KAAK,CAAC;YACf,UAAU,EAAE,MAAM,CAAC;YACnB,QAAQ,EAAE,MAAM,CAAC;YACjB,IAAI,EAAE,OAAO,CAAC;YACd,MAAM,EAAE,OAAO,CAAC;SACjB,CAAC,CAAC;QACH,mCAAmC;QACnC,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,mCAAmC;QACnC,KAAK,EAAE;YAAE,WAAW,EAAE,MAAM,CAAA;SAAE,CAAC;QAC/B,4EAA4E;QAC5E,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;KAC1C,CAAC,CAAC;CACJ"}