agent-swarm-kit 1.0.69 → 1.0.70

package/build/index.cjs

@@ -4416,6 +4416,7 @@ var ClientStorage = /** @class */ (function () {
      */
     function ClientStorage(params) {
         var _this = this;
+        var _a;
         this.params = params;
         this._itemMap = new Map();
         /**
@@ -4641,9 +4642,26 @@ var ClientStorage = /** @class */ (function () {
                 return [2 /*return*/, result];
             });
         }); };
+        /**
+         * Disposes of the state.
+         * @returns {Promise<void>}
+         */
+        this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
+            var _a;
+            return __generator(this, function (_b) {
+                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " dispose"));
+                if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onDispose) {
+                    this.params.callbacks.onDispose(this.params.clientId, this.params.storageName);
+                }
+                return [2 /*return*/];
+            });
+        }); };
         this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " CTOR"), {
             params: params,
         });
+        if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onInit) {
+            this.params.callbacks.onInit(this.params.clientId, this.params.storageName);
+        }
     }
     return ClientStorage;
 }());
@@ -4838,18 +4856,31 @@ var StorageConnectionService = /** @class */ (function () {
          * @returns {Promise<void>}
          */
         this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
-            var key;
+            var key, storage;
             return __generator(this, function (_a) {
-                this.loggerService.log("storageConnectionService dispose", {
-                    context: this.contextService.context,
-                });
-                key = "".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.storageName);
-                if (!this.getStorage.has(key)) {
-                    return [2 /*return*/];
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("storageConnectionService dispose", {
+                            context: this.contextService.context,
+                        });
+                        key = "".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.storageName);
+                        if (!this.getStorage.has(key)) {
+                            return [2 /*return*/];
+                        }
+                        if (!!this.getSharedStorage.has(this.contextService.context.storageName)) return [3 /*break*/, 3];
+                        storage = this.getSharedStorage(this.contextService.context.clientId, this.contextService.context.storageName);
+                        return [4 /*yield*/, storage.waitForInit()];
+                    case 1:
+                        _a.sent();
+                        return [4 /*yield*/, storage.dispose()];
+                    case 2:
+                        _a.sent();
+                        _a.label = 3;
+                    case 3:
+                        this.getStorage.clear(key);
+                        this.sessionValidationService.removeStorageUsage(this.contextService.context.clientId, this.contextService.context.storageName);
+                        return [2 /*return*/];
                 }
-                this.getStorage.clear(key);
-                this.sessionValidationService.removeStorageUsage(this.contextService.context.clientId, this.contextService.context.storageName);
-                return [2 /*return*/];
             });
         }); };
     }
@@ -5268,12 +5299,12 @@ var ClientState = /** @class */ (function () {
             return __generator(this, function (_c) {
                 switch (_c.label) {
                     case 0:
-                        this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " waitForInit"));
+                        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " waitForInit"));
                         _a = this;
                         return [4 /*yield*/, this.params.getState(this.params.clientId, this.params.stateName)];
                     case 1:
                         _a._state = _c.sent();
-                        this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " waitForInit output"), { initialState: this._state });
+                        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " waitForInit output"), { initialState: this._state });
                         if ((_b = this.params.callbacks) === null || _b === void 0 ? void 0 : _b.onLoad) {
                             this.params.callbacks.onLoad(this._state, this.params.clientId, this.params.stateName);
                         }
@@ -5292,7 +5323,7 @@ var ClientState = /** @class */ (function () {
             return __generator(this, function (_b) {
                 switch (_b.label) {
                     case 0:
-                        this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " setState"));
+                        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " setState"));
                         return [4 /*yield*/, this.dispatch("write", function (currentState) { return __awaiter(_this, void 0, void 0, function () {
                                 var _a, _b, middleware, e_1_1;
                                 var e_1, _c;
@@ -5330,7 +5361,7 @@ var ClientState = /** @class */ (function () {
                             }); })];
                     case 1:
                         _b.sent();
-                        this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " setState output"), { pendingState: this._state });
+                        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " setState output"), { pendingState: this._state });
                         this.params.setState &&
                             this.params.setState(this._state, this.params.clientId, this.params.stateName);
                         if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onWrite) {
@@ -5349,7 +5380,7 @@ var ClientState = /** @class */ (function () {
             return __generator(this, function (_b) {
                 switch (_b.label) {
                     case 0:
-                        this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " getState"));
+                        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " getState"));
                         return [4 /*yield*/, this.dispatch("read")];
                     case 1:
                         _b.sent();
@@ -5367,13 +5398,16 @@ var ClientState = /** @class */ (function () {
         this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
             var _a;
             return __generator(this, function (_b) {
-                this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " dispose"));
+                this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " dispose"));
                 if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onDispose) {
                     this.params.callbacks.onDispose(this.params.clientId, this.params.stateName);
                 }
                 return [2 /*return*/];
             });
         }); };
+        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " CTOR"), {
+            params: params,
+        });
         if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onInit) {
             this.params.callbacks.onInit(this.params.clientId, this.params.stateName);
         }
@@ -5393,6 +5427,44 @@ var StateConnectionService = /** @class */ (function () {
         this.contextService = inject(TYPES.contextService);
         this.stateSchemaService = inject(TYPES.stateSchemaService);
         this.sessionValidationService = inject(TYPES.sessionValidationService);
+        /**
+         * Memoized function to get a shared state reference.
+         * @param {string} clientId - The client ID.
+         * @param {StateName} stateName - The state name.
+         * @returns {ClientState} The client state.
+         */
+        this.getSharedStateRef = functoolsKit.memoize(function (_a) {
+            var _b = __read(_a, 2), stateName = _b[1];
+            return "".concat(stateName);
+        }, function (clientId, stateName) {
+            _this.sessionValidationService.addStateUsage(clientId, stateName);
+            var _a = _this.stateSchemaService.get(stateName), getState = _a.getState, setState = _a.setState, _b = _a.middlewares, middlewares = _b === void 0 ? [] : _b, shared = _a.shared, callbacks = _a.callbacks;
+            if (!shared) {
+                throw new Error("agent-swarm state not shared stateName=".concat(stateName));
+            }
+            return new ClientState({
+                clientId: clientId,
+                stateName: stateName,
+                logger: _this.loggerService,
+                setState: setState
+                    ? functoolsKit.queued(function () {
+                        var args = [];
+                        for (var _i = 0; _i < arguments.length; _i++) {
+                            args[_i] = arguments[_i];
+                        }
+                        return __awaiter(_this, void 0, void 0, function () { return __generator(this, function (_a) {
+                            switch (_a.label) {
+                                case 0: return [4 /*yield*/, setState.apply(void 0, __spreadArray([], __read(args), false))];
+                                case 1: return [2 /*return*/, _a.sent()];
+                            }
+                        }); });
+                    })
+                    : setState,
+                getState: getState,
+                middlewares: middlewares,
+                callbacks: callbacks,
+            });
+        });
         /**
          * Memoized function to get a state reference.
          * @param {string} clientId - The client ID.
@@ -5404,7 +5476,10 @@ var StateConnectionService = /** @class */ (function () {
             return "".concat(clientId, "-").concat(stateName);
         }, function (clientId, stateName) {
             _this.sessionValidationService.addStateUsage(clientId, stateName);
-            var _a = _this.stateSchemaService.get(stateName), getState = _a.getState, setState = _a.setState, _b = _a.middlewares, middlewares = _b === void 0 ? [] : _b, callbacks = _a.callbacks;
+            var _a = _this.stateSchemaService.get(stateName), getState = _a.getState, setState = _a.setState, _b = _a.middlewares, middlewares = _b === void 0 ? [] : _b, callbacks = _a.callbacks, _c = _a.shared, shared = _c === void 0 ? false : _c;
+            if (shared) {
+                return _this.getSharedStateRef(clientId, stateName);
+            }
             return new ClientState({
                 clientId: clientId,
                 stateName: stateName,
@@ -5487,6 +5562,7 @@ var StateConnectionService = /** @class */ (function () {
                         if (!this.getStateRef.has(key)) {
                             return [2 /*return*/];
                         }
+                        if (!!this.getSharedStateRef.has(this.contextService.context.stateName)) return [3 /*break*/, 3];
                         state = this.getStateRef(this.contextService.context.clientId, this.contextService.context.stateName);
                         return [4 /*yield*/, state.waitForInit()];
                     case 1:
@@ -5494,6 +5570,8 @@ var StateConnectionService = /** @class */ (function () {
                         return [4 /*yield*/, state.dispose()];
                     case 2:
                         _a.sent();
+                        _a.label = 3;
+                    case 3:
                         this.getStateRef.clear(key);
                         this.sessionValidationService.removeStateUsage(this.contextService.context.clientId, this.contextService.context.stateName);
                         return [2 /*return*/];
@@ -5768,10 +5846,15 @@ var addTool = function (toolSchema) {
  * @returns {string} The name of the added state.
  */
 var addState = function (stateSchema) {
-    swarm.loggerService.log('function addState', {
-        stateSchema: stateSchema
+    swarm.loggerService.log("function addState", {
+        stateSchema: stateSchema,
     });
     swarm.stateSchemaService.register(stateSchema.stateName, stateSchema);
+    if (stateSchema.shared) {
+        swarm.stateConnectionService
+            .getSharedStateRef("shared", stateSchema.stateName)
+            .waitForInit();
+    }
     return stateSchema.stateName;
 };
 

package/build/index.mjs

@@ -4414,6 +4414,7 @@ var ClientStorage = /** @class */ (function () {
      */
     function ClientStorage(params) {
         var _this = this;
+        var _a;
         this.params = params;
         this._itemMap = new Map();
         /**
@@ -4639,9 +4640,26 @@ var ClientStorage = /** @class */ (function () {
                 return [2 /*return*/, result];
             });
         }); };
+        /**
+         * Disposes of the state.
+         * @returns {Promise<void>}
+         */
+        this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
+            var _a;
+            return __generator(this, function (_b) {
+                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " dispose"));
+                if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onDispose) {
+                    this.params.callbacks.onDispose(this.params.clientId, this.params.storageName);
+                }
+                return [2 /*return*/];
+            });
+        }); };
         this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " CTOR"), {
             params: params,
         });
+        if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onInit) {
+            this.params.callbacks.onInit(this.params.clientId, this.params.storageName);
+        }
     }
     return ClientStorage;
 }());
@@ -4836,18 +4854,31 @@ var StorageConnectionService = /** @class */ (function () {
          * @returns {Promise<void>}
          */
         this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
-            var key;
+            var key, storage;
             return __generator(this, function (_a) {
-                this.loggerService.log("storageConnectionService dispose", {
-                    context: this.contextService.context,
-                });
-                key = "".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.storageName);
-                if (!this.getStorage.has(key)) {
-                    return [2 /*return*/];
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("storageConnectionService dispose", {
+                            context: this.contextService.context,
+                        });
+                        key = "".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.storageName);
+                        if (!this.getStorage.has(key)) {
+                            return [2 /*return*/];
+                        }
+                        if (!!this.getSharedStorage.has(this.contextService.context.storageName)) return [3 /*break*/, 3];
+                        storage = this.getSharedStorage(this.contextService.context.clientId, this.contextService.context.storageName);
+                        return [4 /*yield*/, storage.waitForInit()];
+                    case 1:
+                        _a.sent();
+                        return [4 /*yield*/, storage.dispose()];
+                    case 2:
+                        _a.sent();
+                        _a.label = 3;
+                    case 3:
+                        this.getStorage.clear(key);
+                        this.sessionValidationService.removeStorageUsage(this.contextService.context.clientId, this.contextService.context.storageName);
+                        return [2 /*return*/];
                 }
-                this.getStorage.clear(key);
-                this.sessionValidationService.removeStorageUsage(this.contextService.context.clientId, this.contextService.context.storageName);
-                return [2 /*return*/];
             });
         }); };
     }
@@ -5266,12 +5297,12 @@ var ClientState = /** @class */ (function () {
             return __generator(this, function (_c) {
                 switch (_c.label) {
                     case 0:
-                        this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " waitForInit"));
+                        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " waitForInit"));
                         _a = this;
                         return [4 /*yield*/, this.params.getState(this.params.clientId, this.params.stateName)];
                     case 1:
                         _a._state = _c.sent();
-                        this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " waitForInit output"), { initialState: this._state });
+                        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " waitForInit output"), { initialState: this._state });
                         if ((_b = this.params.callbacks) === null || _b === void 0 ? void 0 : _b.onLoad) {
                             this.params.callbacks.onLoad(this._state, this.params.clientId, this.params.stateName);
                         }
@@ -5290,7 +5321,7 @@ var ClientState = /** @class */ (function () {
             return __generator(this, function (_b) {
                 switch (_b.label) {
                     case 0:
-                        this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " setState"));
+                        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " setState"));
                         return [4 /*yield*/, this.dispatch("write", function (currentState) { return __awaiter(_this, void 0, void 0, function () {
                                 var _a, _b, middleware, e_1_1;
                                 var e_1, _c;
@@ -5328,7 +5359,7 @@ var ClientState = /** @class */ (function () {
                             }); })];
                     case 1:
                         _b.sent();
-                        this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " setState output"), { pendingState: this._state });
+                        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " setState output"), { pendingState: this._state });
                         this.params.setState &&
                             this.params.setState(this._state, this.params.clientId, this.params.stateName);
                         if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onWrite) {
@@ -5347,7 +5378,7 @@ var ClientState = /** @class */ (function () {
             return __generator(this, function (_b) {
                 switch (_b.label) {
                     case 0:
-                        this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " getState"));
+                        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " getState"));
                         return [4 /*yield*/, this.dispatch("read")];
                     case 1:
                         _b.sent();
@@ -5365,13 +5396,16 @@ var ClientState = /** @class */ (function () {
         this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
             var _a;
             return __generator(this, function (_b) {
-                this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " dispose"));
+                this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " dispose"));
                 if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onDispose) {
                     this.params.callbacks.onDispose(this.params.clientId, this.params.stateName);
                 }
                 return [2 /*return*/];
             });
         }); };
+        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " CTOR"), {
+            params: params,
+        });
         if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onInit) {
             this.params.callbacks.onInit(this.params.clientId, this.params.stateName);
         }
@@ -5391,6 +5425,44 @@ var StateConnectionService = /** @class */ (function () {
         this.contextService = inject(TYPES.contextService);
         this.stateSchemaService = inject(TYPES.stateSchemaService);
         this.sessionValidationService = inject(TYPES.sessionValidationService);
+        /**
+         * Memoized function to get a shared state reference.
+         * @param {string} clientId - The client ID.
+         * @param {StateName} stateName - The state name.
+         * @returns {ClientState} The client state.
+         */
+        this.getSharedStateRef = memoize(function (_a) {
+            var _b = __read(_a, 2), stateName = _b[1];
+            return "".concat(stateName);
+        }, function (clientId, stateName) {
+            _this.sessionValidationService.addStateUsage(clientId, stateName);
+            var _a = _this.stateSchemaService.get(stateName), getState = _a.getState, setState = _a.setState, _b = _a.middlewares, middlewares = _b === void 0 ? [] : _b, shared = _a.shared, callbacks = _a.callbacks;
+            if (!shared) {
+                throw new Error("agent-swarm state not shared stateName=".concat(stateName));
+            }
+            return new ClientState({
+                clientId: clientId,
+                stateName: stateName,
+                logger: _this.loggerService,
+                setState: setState
+                    ? queued(function () {
+                        var args = [];
+                        for (var _i = 0; _i < arguments.length; _i++) {
+                            args[_i] = arguments[_i];
+                        }
+                        return __awaiter(_this, void 0, void 0, function () { return __generator(this, function (_a) {
+                            switch (_a.label) {
+                                case 0: return [4 /*yield*/, setState.apply(void 0, __spreadArray([], __read(args), false))];
+                                case 1: return [2 /*return*/, _a.sent()];
+                            }
+                        }); });
+                    })
+                    : setState,
+                getState: getState,
+                middlewares: middlewares,
+                callbacks: callbacks,
+            });
+        });
         /**
          * Memoized function to get a state reference.
          * @param {string} clientId - The client ID.
@@ -5402,7 +5474,10 @@ var StateConnectionService = /** @class */ (function () {
             return "".concat(clientId, "-").concat(stateName);
         }, function (clientId, stateName) {
             _this.sessionValidationService.addStateUsage(clientId, stateName);
-            var _a = _this.stateSchemaService.get(stateName), getState = _a.getState, setState = _a.setState, _b = _a.middlewares, middlewares = _b === void 0 ? [] : _b, callbacks = _a.callbacks;
+            var _a = _this.stateSchemaService.get(stateName), getState = _a.getState, setState = _a.setState, _b = _a.middlewares, middlewares = _b === void 0 ? [] : _b, callbacks = _a.callbacks, _c = _a.shared, shared = _c === void 0 ? false : _c;
+            if (shared) {
+                return _this.getSharedStateRef(clientId, stateName);
+            }
             return new ClientState({
                 clientId: clientId,
                 stateName: stateName,
@@ -5485,6 +5560,7 @@ var StateConnectionService = /** @class */ (function () {
                         if (!this.getStateRef.has(key)) {
                             return [2 /*return*/];
                         }
+                        if (!!this.getSharedStateRef.has(this.contextService.context.stateName)) return [3 /*break*/, 3];
                         state = this.getStateRef(this.contextService.context.clientId, this.contextService.context.stateName);
                         return [4 /*yield*/, state.waitForInit()];
                     case 1:
@@ -5492,6 +5568,8 @@ var StateConnectionService = /** @class */ (function () {
                         return [4 /*yield*/, state.dispose()];
                     case 2:
                         _a.sent();
+                        _a.label = 3;
+                    case 3:
                         this.getStateRef.clear(key);
                         this.sessionValidationService.removeStateUsage(this.contextService.context.clientId, this.contextService.context.stateName);
                         return [2 /*return*/];
@@ -5766,10 +5844,15 @@ var addTool = function (toolSchema) {
  * @returns {string} The name of the added state.
  */
 var addState = function (stateSchema) {
-    swarm.loggerService.log('function addState', {
-        stateSchema: stateSchema
+    swarm.loggerService.log("function addState", {
+        stateSchema: stateSchema,
     });
     swarm.stateSchemaService.register(stateSchema.stateName, stateSchema);
+    if (stateSchema.shared) {
+        swarm.stateConnectionService
+            .getSharedStateRef("shared", stateSchema.stateName)
+            .waitForInit();
+    }
     return stateSchema.stateName;
 };
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-swarm-kit",
-  "version": "1.0.69",
+  "version": "1.0.70",
   "description": "A TypeScript library for building orchestrated framework-agnostic multi-agent AI systems",
   "author": {
     "name": "Petr Tripolsky",

package/types.d.ts

@@ -842,6 +842,18 @@ interface IStorageCallbacks<T extends IStorageData = IStorageData> {
      * @param storageName - The name of the storage.
      */
     onSearch: (search: string, index: SortedArray<T>, clientId: string, storageName: StorageName) => void;
+    /**
+     * Callback function for init
+     * @param clientId - The client ID.
+     * @param storageName - The name of the storage.
+     */
+    onInit: (clientId: string, storageName: StorageName) => void;
+    /**
+     * Callback function for dispose
+     * @param clientId - The client ID.
+     * @param storageName - The name of the storage.
+     */
+    onDispose: (clientId: string, storageName: StorageName) => void;
 }
 /**
  * Interface representing the parameters for storage.
@@ -973,6 +985,10 @@ interface IStateCallbacks<T extends IStateData = IStateData> {
  * @template T - The type of the state data.
  */
 interface IStateSchema<T extends IStateData = IStateData> {
+    /**
+     * The agents can share the state
+     */
+    shared?: boolean;
     /**
      * The name of the state.
      */
@@ -2482,6 +2498,11 @@ declare class ClientStorage<T extends IStorageData = IStorageData> implements IS
      * @returns {Promise<T[]>} - The list of items.
      */
     list: (filter?: (item: T) => boolean) => Promise<T[]>;
+    /**
+     * Disposes of the state.
+     * @returns {Promise<void>}
+     */
+    dispose: () => Promise<void>;
 }
 
 /**
@@ -2704,6 +2725,13 @@ declare class StateConnectionService<T extends IStateData = IStateData> implemen
     private readonly contextService;
     private readonly stateSchemaService;
     private readonly sessionValidationService;
+    /**
+     * Memoized function to get a shared state reference.
+     * @param {string} clientId - The client ID.
+     * @param {StateName} stateName - The state name.
+     * @returns {ClientState} The client state.
+     */
+    getSharedStateRef: ((clientId: string, stateName: StateName) => ClientState<any>) & functools_kit.IClearableMemoize<string> & functools_kit.IControlMemoize<string, ClientState<any>>;
     /**
      * Memoized function to get a state reference.
      * @param {string} clientId - The client ID.
@@ -2733,6 +2761,7 @@ interface IStateConnectionService extends StateConnectionService {
 }
 type InternalKeys = keyof {
     getStateRef: never;
+    getSharedStateRef: never;
 };
 type TStateConnectionService = {
     [key in Exclude<keyof IStateConnectionService, InternalKeys>]: unknown;