agent-swarm-kit 1.0.66 → 1.0.68

package/build/index.cjs

@@ -4319,7 +4319,7 @@ var ClientStorage = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " _createEmbedding"), {
+                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " _createEmbedding"), {
                             id: item.id,
                         });
                         return [4 /*yield*/, this.params.createIndex(item)];
@@ -4344,7 +4344,7 @@ var ClientStorage = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " waitForInit"));
+                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " waitForInit"));
                         if (!this.params.getData) {
                             return [2 /*return*/];
                         }
@@ -4382,7 +4382,7 @@ var ClientStorage = /** @class */ (function () {
                 return __generator(this, function (_c) {
                     switch (_c.label) {
                         case 0:
-                            this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " take"), {
+                            this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " take"), {
                                 search: search,
                                 total: total,
                             });
@@ -4416,7 +4416,7 @@ var ClientStorage = /** @class */ (function () {
                                 })))];
                         case 2:
                             _c.sent();
-                            this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " take indexed"), {
+                            this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " take indexed"), {
                                 indexed: indexed.getEntries(),
                             });
                             if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onSearch) {
@@ -4437,7 +4437,7 @@ var ClientStorage = /** @class */ (function () {
             return __generator(this, function (_c) {
                 switch (_c.label) {
                     case 0:
-                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " upsert"), {
+                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " upsert"), {
                             item: item,
                         });
                         this._itemMap.set(item.id, item);
@@ -4460,7 +4460,7 @@ var ClientStorage = /** @class */ (function () {
         this.remove = function (itemId) { return __awaiter(_this, void 0, void 0, function () {
             var _a, _b;
             return __generator(this, function (_c) {
-                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " remove"), {
+                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " remove"), {
                     id: itemId,
                 });
                 this._itemMap.delete(itemId);
@@ -4477,7 +4477,7 @@ var ClientStorage = /** @class */ (function () {
          */
         this.clear = function () { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
-                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " clear"));
+                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " clear"));
                 this._itemMap.clear();
                 this._createEmbedding.clear();
                 return [2 /*return*/];
@@ -4491,7 +4491,7 @@ var ClientStorage = /** @class */ (function () {
         this.get = function (itemId) { 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, " get"), {
+                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " get"), {
                     id: itemId,
                 });
                 return [2 /*return*/, (_a = this._itemMap.get(itemId)) !== null && _a !== void 0 ? _a : null];
@@ -4506,7 +4506,7 @@ var ClientStorage = /** @class */ (function () {
             var result, _a, _b, item;
             var e_1, _c;
             return __generator(this, function (_d) {
-                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " list"));
+                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " list"));
                 if (!filter) {
                     return [2 /*return*/, __spreadArray([], __read(this._itemMap.values()), false)];
                 }
@@ -4529,7 +4529,7 @@ var ClientStorage = /** @class */ (function () {
                 return [2 /*return*/, result];
             });
         }); };
-        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " CTOR"), {
+        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " CTOR"), {
             params: params,
         });
     }
@@ -4548,6 +4548,23 @@ var StorageConnectionService = /** @class */ (function () {
         this.storageSchemaService = inject(TYPES.storageSchemaService);
         this.sessionValidationService = inject(TYPES.sessionValidationService);
         this.embeddingSchemaService = inject(TYPES.embeddingSchemaService);
+        /**
+         * Retrieves a shared storage instance based on storage name.
+         * @param {string} clientId - The client ID.
+         * @param {string} storageName - The storage name.
+         * @returns {ClientStorage} The client storage instance.
+         */
+        this.getSharedStorage = functoolsKit.memoize(function (_a) {
+            var _b = __read(_a, 2), storageName = _b[1];
+            return "".concat(storageName);
+        }, function (clientId, storageName) {
+            var _a = _this.storageSchemaService.get(storageName), createIndex = _a.createIndex, getData = _a.getData, embeddingName = _a.embedding, shared = _a.shared, callbacks = _a.callbacks;
+            var _b = _this.embeddingSchemaService.get(embeddingName), calculateSimilarity = _b.calculateSimilarity, createEmbedding = _b.createEmbedding, embedding = _b.callbacks;
+            if (!shared) {
+                throw new Error("agent-swarm storage not shared storageName=".concat(storageName));
+            }
+            return new ClientStorage(__assign(__assign({ clientId: clientId, storageName: storageName, embedding: embeddingName, calculateSimilarity: calculateSimilarity, createEmbedding: createEmbedding, createIndex: createIndex, getData: getData, shared: shared, logger: _this.loggerService }, embedding), { callbacks: callbacks }));
+        });
         /**
          * Retrieves a storage instance based on client ID and storage name.
          * @param {string} clientId - The client ID.
@@ -4559,9 +4576,12 @@ var StorageConnectionService = /** @class */ (function () {
             return "".concat(clientId, "-").concat(storageName);
         }, function (clientId, storageName) {
             _this.sessionValidationService.addStorageUsage(clientId, storageName);
-            var _a = _this.storageSchemaService.get(storageName), createIndex = _a.createIndex, getData = _a.getData, embeddingName = _a.embedding, callbacks = _a.callbacks;
-            var _b = _this.embeddingSchemaService.get(embeddingName), calculateSimilarity = _b.calculateSimilarity, createEmbedding = _b.createEmbedding, embedding = _b.callbacks;
-            return new ClientStorage(__assign(__assign({ clientId: clientId, storageName: storageName, embedding: embeddingName, calculateSimilarity: calculateSimilarity, createEmbedding: createEmbedding, createIndex: createIndex, getData: getData, logger: _this.loggerService }, embedding), { callbacks: callbacks }));
+            var _a = _this.storageSchemaService.get(storageName), createIndex = _a.createIndex, getData = _a.getData, embeddingName = _a.embedding, _b = _a.shared, shared = _b === void 0 ? false : _b, callbacks = _a.callbacks;
+            if (shared) {
+                return _this.getSharedStorage(clientId, storageName);
+            }
+            var _c = _this.embeddingSchemaService.get(embeddingName), calculateSimilarity = _c.calculateSimilarity, createEmbedding = _c.createEmbedding, embedding = _c.callbacks;
+            return new ClientStorage(__assign(__assign({ clientId: clientId, storageName: storageName, embedding: embeddingName, calculateSimilarity: calculateSimilarity, createEmbedding: createEmbedding, createIndex: createIndex, getData: getData, shared: shared, logger: _this.loggerService }, embedding), { callbacks: callbacks }));
         });
         /**
          * Retrieves a list of storage data based on a search query and total number of items.
@@ -5221,11 +5241,16 @@ var addEmbedding = function (embeddingSchema) {
  * @returns {string} The name of the added storage.
  */
 var addStorage = function (storageSchema) {
-    swarm.loggerService.log('function addStorage', {
-        storageSchema: storageSchema
+    swarm.loggerService.log("function addStorage", {
+        storageSchema: storageSchema,
     });
     swarm.storageValidationService.addStorage(storageSchema.storageName, storageSchema);
     swarm.storageSchemaService.register(storageSchema.storageName, storageSchema);
+    if (storageSchema.shared) {
+        swarm.storageConnectionService
+            .getSharedStorage("shared", storageSchema.storageName)
+            .waitForInit();
+    }
     return storageSchema.storageName;
 };
 

package/build/index.mjs

@@ -4317,7 +4317,7 @@ var ClientStorage = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " _createEmbedding"), {
+                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " _createEmbedding"), {
                             id: item.id,
                         });
                         return [4 /*yield*/, this.params.createIndex(item)];
@@ -4342,7 +4342,7 @@ var ClientStorage = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " waitForInit"));
+                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " waitForInit"));
                         if (!this.params.getData) {
                             return [2 /*return*/];
                         }
@@ -4380,7 +4380,7 @@ var ClientStorage = /** @class */ (function () {
                 return __generator(this, function (_c) {
                     switch (_c.label) {
                         case 0:
-                            this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " take"), {
+                            this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " take"), {
                                 search: search,
                                 total: total,
                             });
@@ -4414,7 +4414,7 @@ var ClientStorage = /** @class */ (function () {
                                 })))];
                         case 2:
                             _c.sent();
-                            this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " take indexed"), {
+                            this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " take indexed"), {
                                 indexed: indexed.getEntries(),
                             });
                             if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onSearch) {
@@ -4435,7 +4435,7 @@ var ClientStorage = /** @class */ (function () {
             return __generator(this, function (_c) {
                 switch (_c.label) {
                     case 0:
-                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " upsert"), {
+                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " upsert"), {
                             item: item,
                         });
                         this._itemMap.set(item.id, item);
@@ -4458,7 +4458,7 @@ var ClientStorage = /** @class */ (function () {
         this.remove = function (itemId) { return __awaiter(_this, void 0, void 0, function () {
             var _a, _b;
             return __generator(this, function (_c) {
-                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " remove"), {
+                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " remove"), {
                     id: itemId,
                 });
                 this._itemMap.delete(itemId);
@@ -4475,7 +4475,7 @@ var ClientStorage = /** @class */ (function () {
          */
         this.clear = function () { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
-                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " clear"));
+                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " clear"));
                 this._itemMap.clear();
                 this._createEmbedding.clear();
                 return [2 /*return*/];
@@ -4489,7 +4489,7 @@ var ClientStorage = /** @class */ (function () {
         this.get = function (itemId) { 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, " get"), {
+                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " get"), {
                     id: itemId,
                 });
                 return [2 /*return*/, (_a = this._itemMap.get(itemId)) !== null && _a !== void 0 ? _a : null];
@@ -4504,7 +4504,7 @@ var ClientStorage = /** @class */ (function () {
             var result, _a, _b, item;
             var e_1, _c;
             return __generator(this, function (_d) {
-                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " list"));
+                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " list"));
                 if (!filter) {
                     return [2 /*return*/, __spreadArray([], __read(this._itemMap.values()), false)];
                 }
@@ -4527,7 +4527,7 @@ var ClientStorage = /** @class */ (function () {
                 return [2 /*return*/, result];
             });
         }); };
-        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " CTOR"), {
+        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " CTOR"), {
             params: params,
         });
     }
@@ -4546,6 +4546,23 @@ var StorageConnectionService = /** @class */ (function () {
         this.storageSchemaService = inject(TYPES.storageSchemaService);
         this.sessionValidationService = inject(TYPES.sessionValidationService);
         this.embeddingSchemaService = inject(TYPES.embeddingSchemaService);
+        /**
+         * Retrieves a shared storage instance based on storage name.
+         * @param {string} clientId - The client ID.
+         * @param {string} storageName - The storage name.
+         * @returns {ClientStorage} The client storage instance.
+         */
+        this.getSharedStorage = memoize(function (_a) {
+            var _b = __read(_a, 2), storageName = _b[1];
+            return "".concat(storageName);
+        }, function (clientId, storageName) {
+            var _a = _this.storageSchemaService.get(storageName), createIndex = _a.createIndex, getData = _a.getData, embeddingName = _a.embedding, shared = _a.shared, callbacks = _a.callbacks;
+            var _b = _this.embeddingSchemaService.get(embeddingName), calculateSimilarity = _b.calculateSimilarity, createEmbedding = _b.createEmbedding, embedding = _b.callbacks;
+            if (!shared) {
+                throw new Error("agent-swarm storage not shared storageName=".concat(storageName));
+            }
+            return new ClientStorage(__assign(__assign({ clientId: clientId, storageName: storageName, embedding: embeddingName, calculateSimilarity: calculateSimilarity, createEmbedding: createEmbedding, createIndex: createIndex, getData: getData, shared: shared, logger: _this.loggerService }, embedding), { callbacks: callbacks }));
+        });
         /**
          * Retrieves a storage instance based on client ID and storage name.
          * @param {string} clientId - The client ID.
@@ -4557,9 +4574,12 @@ var StorageConnectionService = /** @class */ (function () {
             return "".concat(clientId, "-").concat(storageName);
         }, function (clientId, storageName) {
             _this.sessionValidationService.addStorageUsage(clientId, storageName);
-            var _a = _this.storageSchemaService.get(storageName), createIndex = _a.createIndex, getData = _a.getData, embeddingName = _a.embedding, callbacks = _a.callbacks;
-            var _b = _this.embeddingSchemaService.get(embeddingName), calculateSimilarity = _b.calculateSimilarity, createEmbedding = _b.createEmbedding, embedding = _b.callbacks;
-            return new ClientStorage(__assign(__assign({ clientId: clientId, storageName: storageName, embedding: embeddingName, calculateSimilarity: calculateSimilarity, createEmbedding: createEmbedding, createIndex: createIndex, getData: getData, logger: _this.loggerService }, embedding), { callbacks: callbacks }));
+            var _a = _this.storageSchemaService.get(storageName), createIndex = _a.createIndex, getData = _a.getData, embeddingName = _a.embedding, _b = _a.shared, shared = _b === void 0 ? false : _b, callbacks = _a.callbacks;
+            if (shared) {
+                return _this.getSharedStorage(clientId, storageName);
+            }
+            var _c = _this.embeddingSchemaService.get(embeddingName), calculateSimilarity = _c.calculateSimilarity, createEmbedding = _c.createEmbedding, embedding = _c.callbacks;
+            return new ClientStorage(__assign(__assign({ clientId: clientId, storageName: storageName, embedding: embeddingName, calculateSimilarity: calculateSimilarity, createEmbedding: createEmbedding, createIndex: createIndex, getData: getData, shared: shared, logger: _this.loggerService }, embedding), { callbacks: callbacks }));
         });
         /**
          * Retrieves a list of storage data based on a search query and total number of items.
@@ -5219,11 +5239,16 @@ var addEmbedding = function (embeddingSchema) {
  * @returns {string} The name of the added storage.
  */
 var addStorage = function (storageSchema) {
-    swarm.loggerService.log('function addStorage', {
-        storageSchema: storageSchema
+    swarm.loggerService.log("function addStorage", {
+        storageSchema: storageSchema,
     });
     swarm.storageValidationService.addStorage(storageSchema.storageName, storageSchema);
     swarm.storageSchemaService.register(storageSchema.storageName, storageSchema);
+    if (storageSchema.shared) {
+        swarm.storageConnectionService
+            .getSharedStorage("shared", storageSchema.storageName)
+            .waitForInit();
+    }
     return storageSchema.storageName;
 };
 

package/package.json

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

package/types.d.ts

@@ -749,6 +749,10 @@ interface IStorageData {
  * @template T - Type of the storage data.
  */
 interface IStorageSchema<T extends IStorageData = IStorageData> {
+    /**
+     * All agents will share the same ClientStorage instance
+     */
+    shared?: boolean;
     /**
      * Function to get data from the storage.
      * @param clientId - The client ID.
@@ -2299,6 +2303,13 @@ declare class StorageConnectionService implements IStorage {
     private readonly storageSchemaService;
     private readonly sessionValidationService;
     private readonly embeddingSchemaService;
+    /**
+     * Retrieves a shared storage instance based on storage name.
+     * @param {string} clientId - The client ID.
+     * @param {string} storageName - The storage name.
+     * @returns {ClientStorage} The client storage instance.
+     */
+    getSharedStorage: ((clientId: string, storageName: StorageName) => ClientStorage<IStorageData>) & functools_kit.IClearableMemoize<string> & functools_kit.IControlMemoize<string, ClientStorage<IStorageData>>;
     /**
      * Retrieves a storage instance based on client ID and storage name.
      * @param {string} clientId - The client ID.
@@ -2353,6 +2364,7 @@ interface IStorageConnectionService extends StorageConnectionService {
 }
 type InternalKeys = keyof {
     getStorage: never;
+    getSharedStorage: never;
 };
 type TStorageConnectionService = {
     [key in Exclude<keyof IStorageConnectionService, InternalKeys>]: unknown;