agent-swarm-kit 1.0.59 → 1.0.61

package/build/index.cjs

@@ -3869,11 +3869,12 @@ var ClientStorage = /** @class */ (function () {
                 args_1[_i - 2] = arguments[_i];
             }
             return __awaiter(_this, __spreadArray([search_1, total_1], __read(args_1), false), void 0, function (search, total, score) {
-                var indexed, searchEmbeddings, filtered;
+                var indexed, searchEmbeddings;
                 var _this = this;
+                var _a, _b;
                 if (score === void 0) { score = GLOBAL_CONFIG.CC_STORAGE_SEARCH_SIMILARITY; }
-                return __generator(this, function (_a) {
-                    switch (_a.label) {
+                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"), {
                                 search: search,
@@ -3882,7 +3883,7 @@ var ClientStorage = /** @class */ (function () {
                             indexed = new functoolsKit.SortedArray();
                             return [4 /*yield*/, this.params.createEmbedding(search)];
                         case 1:
-                            searchEmbeddings = _a.sent();
+                            searchEmbeddings = _c.sent();
                             if (this.params.onCreate) {
                                 this.params.onCreate(search, searchEmbeddings, this.params.clientId, this.params.embedding);
                             }
@@ -3899,7 +3900,7 @@ var ClientStorage = /** @class */ (function () {
                                                 if (this.params.onCompare) {
                                                     this.params.onCompare(search, index, score, this.params.clientId, this.params.embedding);
                                                 }
-                                                indexed.push({ id: item.id }, score);
+                                                indexed.push(item, score);
                                                 return [2 /*return*/];
                                         }
                                     });
@@ -3908,15 +3909,14 @@ var ClientStorage = /** @class */ (function () {
                                     maxExec: GLOBAL_CONFIG.CC_STORAGE_SEARCH_POOL,
                                 })))];
                         case 2:
-                            _a.sent();
+                            _c.sent();
                             this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " take indexed"), {
                                 indexed: indexed.getEntries(),
                             });
-                            filtered = indexed.take(total, score);
-                            return [2 /*return*/, filtered.map(function (_a) {
-                                    var id = _a.id;
-                                    return _this._itemMap.get(id);
-                                })];
+                            if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onSearch) {
+                                (_b = this.params.callbacks) === null || _b === void 0 ? void 0 : _b.onSearch(search, indexed, this.params.clientId, this.params.storageName);
+                            }
+                            return [2 /*return*/, indexed.take(total, score)];
                     }
                 });
             });
@@ -5922,22 +5922,22 @@ var StorageUtils = /** @class */ (function () {
          * @returns {Promise<T[]>} - A promise that resolves to an array of items.
          * @template T
          */
-        this.take = function (search, total, clientId, agentName, storageName, score) { return __awaiter(_this, void 0, void 0, function () {
+        this.take = function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         swarm.loggerService.log("StorageStatic take", {
-                            search: search,
-                            total: total,
-                            clientId: clientId,
-                            storageName: storageName,
-                            score: score,
+                            search: payload.search,
+                            total: payload.total,
+                            clientId: payload.clientId,
+                            storageName: payload.storageName,
+                            score: payload.score,
                         });
-                        swarm.storageValidationService.validate(storageName, "StorageStatic");
-                        if (!swarm.agentValidationService.hasStorage(agentName, storageName)) {
-                            throw new Error("agent-swarm StorageUtils ".concat(storageName, " not registered in ").concat(agentName, " (take)"));
+                        swarm.storageValidationService.validate(payload.storageName, "StorageStatic");
+                        if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
+                            throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (take)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.take(search, total, clientId, storageName, score)];
+                        return [4 /*yield*/, swarm.storagePublicService.take(payload.search, payload.total, payload.clientId, payload.storageName, payload.score)];
                     case 1: return [2 /*return*/, (_a.sent())];
                 }
             });
@@ -5951,20 +5951,20 @@ var StorageUtils = /** @class */ (function () {
          * @returns {Promise<void>} - A promise that resolves when the operation is complete.
          * @template T
          */
-        this.upsert = function (item, clientId, agentName, storageName) { return __awaiter(_this, void 0, void 0, function () {
+        this.upsert = function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         swarm.loggerService.log("StorageStatic upsert", {
-                            item: item,
-                            clientId: clientId,
-                            storageName: storageName,
+                            item: payload.item,
+                            clientId: payload.clientId,
+                            storageName: payload.storageName,
                         });
-                        swarm.storageValidationService.validate(storageName, "StorageStatic");
-                        if (!swarm.agentValidationService.hasStorage(agentName, storageName)) {
-                            throw new Error("agent-swarm StorageUtils ".concat(storageName, " not registered in ").concat(agentName, " (upsert)"));
+                        swarm.storageValidationService.validate(payload.storageName, "StorageStatic");
+                        if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
+                            throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (upsert)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.upsert(item, clientId, storageName)];
+                        return [4 /*yield*/, swarm.storagePublicService.upsert(payload.item, payload.clientId, payload.storageName)];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
@@ -5977,20 +5977,20 @@ var StorageUtils = /** @class */ (function () {
          * @param {StorageName} storageName - The storage name.
          * @returns {Promise<void>} - A promise that resolves when the operation is complete.
          */
-        this.remove = function (itemId, clientId, agentName, storageName) { return __awaiter(_this, void 0, void 0, function () {
+        this.remove = function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         swarm.loggerService.log("StorageStatic remove", {
-                            itemId: itemId,
-                            clientId: clientId,
-                            storageName: storageName,
+                            itemId: payload.itemId,
+                            clientId: payload.clientId,
+                            storageName: payload.storageName,
                         });
-                        swarm.storageValidationService.validate(storageName, "StorageStatic");
-                        if (!swarm.agentValidationService.hasStorage(agentName, storageName)) {
-                            throw new Error("agent-swarm StorageUtils ".concat(storageName, " not registered in ").concat(agentName, " (remove)"));
+                        swarm.storageValidationService.validate(payload.storageName, "StorageStatic");
+                        if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
+                            throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (remove)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.remove(itemId, clientId, storageName)];
+                        return [4 /*yield*/, swarm.storagePublicService.remove(payload.itemId, payload.clientId, payload.storageName)];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
@@ -6004,20 +6004,20 @@ var StorageUtils = /** @class */ (function () {
          * @returns {Promise<T | null>} - A promise that resolves to the item or null if not found.
          * @template T
          */
-        this.get = function (itemId, clientId, agentName, storageName) { return __awaiter(_this, void 0, void 0, function () {
+        this.get = function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         swarm.loggerService.log("StorageStatic get", {
-                            itemId: itemId,
-                            clientId: clientId,
-                            storageName: storageName,
+                            itemId: payload.itemId,
+                            clientId: payload.clientId,
+                            storageName: payload.storageName,
                         });
-                        swarm.storageValidationService.validate(storageName, "StorageStatic");
-                        if (!swarm.agentValidationService.hasStorage(agentName, storageName)) {
-                            throw new Error("agent-swarm StorageUtils ".concat(storageName, " not registered in ").concat(agentName, " (get)"));
+                        swarm.storageValidationService.validate(payload.storageName, "StorageStatic");
+                        if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
+                            throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (get)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.get(itemId, clientId, storageName)];
+                        return [4 /*yield*/, swarm.storagePublicService.get(payload.itemId, payload.clientId, payload.storageName)];
                     case 1: return [2 /*return*/, (_a.sent())];
                 }
             });
@@ -6031,19 +6031,19 @@ var StorageUtils = /** @class */ (function () {
          * @returns {Promise<T[]>} - A promise that resolves to an array of items.
          * @template T
          */
-        this.list = function (clientId, agentName, storageName, filter) { return __awaiter(_this, void 0, void 0, function () {
+        this.list = function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         swarm.loggerService.log("StorageStatic list", {
-                            clientId: clientId,
-                            storageName: storageName,
+                            clientId: payload.clientId,
+                            storageName: payload.storageName,
                         });
-                        swarm.storageValidationService.validate(storageName, "StorageStatic");
-                        if (!swarm.agentValidationService.hasStorage(agentName, storageName)) {
-                            throw new Error("agent-swarm StorageUtils ".concat(storageName, " not registered in ").concat(agentName, " (list)"));
+                        swarm.storageValidationService.validate(payload.storageName, "StorageStatic");
+                        if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
+                            throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (list)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.list(clientId, storageName, filter)];
+                        return [4 /*yield*/, swarm.storagePublicService.list(payload.clientId, payload.storageName, payload.filter)];
                     case 1: return [2 /*return*/, (_a.sent())];
                 }
             });
@@ -6055,19 +6055,19 @@ var StorageUtils = /** @class */ (function () {
          * @param {StorageName} storageName - The storage name.
          * @returns {Promise<void>} - A promise that resolves when the operation is complete.
          */
-        this.clear = function (clientId, agentName, storageName) { return __awaiter(_this, void 0, void 0, function () {
+        this.clear = function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         swarm.loggerService.log("StorageStatic clear", {
-                            clientId: clientId,
-                            storageName: storageName,
+                            clientId: payload.clientId,
+                            storageName: payload.storageName,
                         });
-                        swarm.storageValidationService.validate(storageName, "StorageStatic");
-                        if (!swarm.agentValidationService.hasStorage(agentName, storageName)) {
-                            throw new Error("agent-swarm StorageUtils ".concat(storageName, " not registered in ").concat(agentName, " (clear)"));
+                        swarm.storageValidationService.validate(payload.storageName, "StorageStatic");
+                        if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
+                            throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (clear)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.clear(clientId, storageName)];
+                        return [4 /*yield*/, swarm.storagePublicService.clear(payload.clientId, payload.storageName)];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });

package/build/index.mjs

@@ -3867,11 +3867,12 @@ var ClientStorage = /** @class */ (function () {
                 args_1[_i - 2] = arguments[_i];
             }
             return __awaiter(_this, __spreadArray([search_1, total_1], __read(args_1), false), void 0, function (search, total, score) {
-                var indexed, searchEmbeddings, filtered;
+                var indexed, searchEmbeddings;
                 var _this = this;
+                var _a, _b;
                 if (score === void 0) { score = GLOBAL_CONFIG.CC_STORAGE_SEARCH_SIMILARITY; }
-                return __generator(this, function (_a) {
-                    switch (_a.label) {
+                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"), {
                                 search: search,
@@ -3880,7 +3881,7 @@ var ClientStorage = /** @class */ (function () {
                             indexed = new SortedArray();
                             return [4 /*yield*/, this.params.createEmbedding(search)];
                         case 1:
-                            searchEmbeddings = _a.sent();
+                            searchEmbeddings = _c.sent();
                             if (this.params.onCreate) {
                                 this.params.onCreate(search, searchEmbeddings, this.params.clientId, this.params.embedding);
                             }
@@ -3897,7 +3898,7 @@ var ClientStorage = /** @class */ (function () {
                                                 if (this.params.onCompare) {
                                                     this.params.onCompare(search, index, score, this.params.clientId, this.params.embedding);
                                                 }
-                                                indexed.push({ id: item.id }, score);
+                                                indexed.push(item, score);
                                                 return [2 /*return*/];
                                         }
                                     });
@@ -3906,15 +3907,14 @@ var ClientStorage = /** @class */ (function () {
                                     maxExec: GLOBAL_CONFIG.CC_STORAGE_SEARCH_POOL,
                                 })))];
                         case 2:
-                            _a.sent();
+                            _c.sent();
                             this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " take indexed"), {
                                 indexed: indexed.getEntries(),
                             });
-                            filtered = indexed.take(total, score);
-                            return [2 /*return*/, filtered.map(function (_a) {
-                                    var id = _a.id;
-                                    return _this._itemMap.get(id);
-                                })];
+                            if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onSearch) {
+                                (_b = this.params.callbacks) === null || _b === void 0 ? void 0 : _b.onSearch(search, indexed, this.params.clientId, this.params.storageName);
+                            }
+                            return [2 /*return*/, indexed.take(total, score)];
                     }
                 });
             });
@@ -5920,22 +5920,22 @@ var StorageUtils = /** @class */ (function () {
          * @returns {Promise<T[]>} - A promise that resolves to an array of items.
          * @template T
          */
-        this.take = function (search, total, clientId, agentName, storageName, score) { return __awaiter(_this, void 0, void 0, function () {
+        this.take = function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         swarm.loggerService.log("StorageStatic take", {
-                            search: search,
-                            total: total,
-                            clientId: clientId,
-                            storageName: storageName,
-                            score: score,
+                            search: payload.search,
+                            total: payload.total,
+                            clientId: payload.clientId,
+                            storageName: payload.storageName,
+                            score: payload.score,
                         });
-                        swarm.storageValidationService.validate(storageName, "StorageStatic");
-                        if (!swarm.agentValidationService.hasStorage(agentName, storageName)) {
-                            throw new Error("agent-swarm StorageUtils ".concat(storageName, " not registered in ").concat(agentName, " (take)"));
+                        swarm.storageValidationService.validate(payload.storageName, "StorageStatic");
+                        if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
+                            throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (take)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.take(search, total, clientId, storageName, score)];
+                        return [4 /*yield*/, swarm.storagePublicService.take(payload.search, payload.total, payload.clientId, payload.storageName, payload.score)];
                     case 1: return [2 /*return*/, (_a.sent())];
                 }
             });
@@ -5949,20 +5949,20 @@ var StorageUtils = /** @class */ (function () {
          * @returns {Promise<void>} - A promise that resolves when the operation is complete.
          * @template T
          */
-        this.upsert = function (item, clientId, agentName, storageName) { return __awaiter(_this, void 0, void 0, function () {
+        this.upsert = function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         swarm.loggerService.log("StorageStatic upsert", {
-                            item: item,
-                            clientId: clientId,
-                            storageName: storageName,
+                            item: payload.item,
+                            clientId: payload.clientId,
+                            storageName: payload.storageName,
                         });
-                        swarm.storageValidationService.validate(storageName, "StorageStatic");
-                        if (!swarm.agentValidationService.hasStorage(agentName, storageName)) {
-                            throw new Error("agent-swarm StorageUtils ".concat(storageName, " not registered in ").concat(agentName, " (upsert)"));
+                        swarm.storageValidationService.validate(payload.storageName, "StorageStatic");
+                        if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
+                            throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (upsert)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.upsert(item, clientId, storageName)];
+                        return [4 /*yield*/, swarm.storagePublicService.upsert(payload.item, payload.clientId, payload.storageName)];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
@@ -5975,20 +5975,20 @@ var StorageUtils = /** @class */ (function () {
          * @param {StorageName} storageName - The storage name.
          * @returns {Promise<void>} - A promise that resolves when the operation is complete.
          */
-        this.remove = function (itemId, clientId, agentName, storageName) { return __awaiter(_this, void 0, void 0, function () {
+        this.remove = function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         swarm.loggerService.log("StorageStatic remove", {
-                            itemId: itemId,
-                            clientId: clientId,
-                            storageName: storageName,
+                            itemId: payload.itemId,
+                            clientId: payload.clientId,
+                            storageName: payload.storageName,
                         });
-                        swarm.storageValidationService.validate(storageName, "StorageStatic");
-                        if (!swarm.agentValidationService.hasStorage(agentName, storageName)) {
-                            throw new Error("agent-swarm StorageUtils ".concat(storageName, " not registered in ").concat(agentName, " (remove)"));
+                        swarm.storageValidationService.validate(payload.storageName, "StorageStatic");
+                        if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
+                            throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (remove)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.remove(itemId, clientId, storageName)];
+                        return [4 /*yield*/, swarm.storagePublicService.remove(payload.itemId, payload.clientId, payload.storageName)];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
@@ -6002,20 +6002,20 @@ var StorageUtils = /** @class */ (function () {
          * @returns {Promise<T | null>} - A promise that resolves to the item or null if not found.
          * @template T
          */
-        this.get = function (itemId, clientId, agentName, storageName) { return __awaiter(_this, void 0, void 0, function () {
+        this.get = function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         swarm.loggerService.log("StorageStatic get", {
-                            itemId: itemId,
-                            clientId: clientId,
-                            storageName: storageName,
+                            itemId: payload.itemId,
+                            clientId: payload.clientId,
+                            storageName: payload.storageName,
                         });
-                        swarm.storageValidationService.validate(storageName, "StorageStatic");
-                        if (!swarm.agentValidationService.hasStorage(agentName, storageName)) {
-                            throw new Error("agent-swarm StorageUtils ".concat(storageName, " not registered in ").concat(agentName, " (get)"));
+                        swarm.storageValidationService.validate(payload.storageName, "StorageStatic");
+                        if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
+                            throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (get)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.get(itemId, clientId, storageName)];
+                        return [4 /*yield*/, swarm.storagePublicService.get(payload.itemId, payload.clientId, payload.storageName)];
                     case 1: return [2 /*return*/, (_a.sent())];
                 }
             });
@@ -6029,19 +6029,19 @@ var StorageUtils = /** @class */ (function () {
          * @returns {Promise<T[]>} - A promise that resolves to an array of items.
          * @template T
          */
-        this.list = function (clientId, agentName, storageName, filter) { return __awaiter(_this, void 0, void 0, function () {
+        this.list = function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         swarm.loggerService.log("StorageStatic list", {
-                            clientId: clientId,
-                            storageName: storageName,
+                            clientId: payload.clientId,
+                            storageName: payload.storageName,
                         });
-                        swarm.storageValidationService.validate(storageName, "StorageStatic");
-                        if (!swarm.agentValidationService.hasStorage(agentName, storageName)) {
-                            throw new Error("agent-swarm StorageUtils ".concat(storageName, " not registered in ").concat(agentName, " (list)"));
+                        swarm.storageValidationService.validate(payload.storageName, "StorageStatic");
+                        if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
+                            throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (list)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.list(clientId, storageName, filter)];
+                        return [4 /*yield*/, swarm.storagePublicService.list(payload.clientId, payload.storageName, payload.filter)];
                     case 1: return [2 /*return*/, (_a.sent())];
                 }
             });
@@ -6053,19 +6053,19 @@ var StorageUtils = /** @class */ (function () {
          * @param {StorageName} storageName - The storage name.
          * @returns {Promise<void>} - A promise that resolves when the operation is complete.
          */
-        this.clear = function (clientId, agentName, storageName) { return __awaiter(_this, void 0, void 0, function () {
+        this.clear = function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         swarm.loggerService.log("StorageStatic clear", {
-                            clientId: clientId,
-                            storageName: storageName,
+                            clientId: payload.clientId,
+                            storageName: payload.storageName,
                         });
-                        swarm.storageValidationService.validate(storageName, "StorageStatic");
-                        if (!swarm.agentValidationService.hasStorage(agentName, storageName)) {
-                            throw new Error("agent-swarm StorageUtils ".concat(storageName, " not registered in ").concat(agentName, " (clear)"));
+                        swarm.storageValidationService.validate(payload.storageName, "StorageStatic");
+                        if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
+                            throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (clear)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.clear(clientId, storageName)];
+                        return [4 /*yield*/, swarm.storagePublicService.clear(payload.clientId, payload.storageName)];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });

package/package.json

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

package/types.d.ts

@@ -1,6 +1,6 @@
 import * as di_scoped from 'di-scoped';
 import * as functools_kit from 'functools-kit';
-import { IPubsubArray, Subject } from 'functools-kit';
+import { IPubsubArray, SortedArray, Subject } from 'functools-kit';
 
 /**
  * Interface representing an incoming message.
@@ -517,6 +517,7 @@ interface IStorageSchema<T extends IStorageData = IStorageData> {
 }
 interface IStorageCallbacks<T extends IStorageData = IStorageData> {
     onUpdate: (items: T[], clientId: string, storageName: StorageName) => void;
+    onSearch: (search: string, index: SortedArray<T>, clientId: string, storageName: StorageName) => void;
 }
 interface IStorageParams<T extends IStorageData = IStorageData> extends IStorageSchema<T>, Partial<IEmbeddingCallbacks> {
     clientId: string;
@@ -2570,7 +2571,14 @@ declare class StorageUtils implements TStorage {
      * @returns {Promise<T[]>} - A promise that resolves to an array of items.
      * @template T
      */
-    take: <T extends IStorageData = IStorageData>(search: string, total: number, clientId: string, agentName: AgentName, storageName: StorageName, score?: number) => Promise<T[]>;
+    take: <T extends IStorageData = IStorageData>(payload: {
+        search: string;
+        total: number;
+        clientId: string;
+        agentName: AgentName;
+        storageName: StorageName;
+        score?: number;
+    }) => Promise<T[]>;
     /**
      * Upserts an item in the storage.
      * @param {T} item - The item to upsert.
@@ -2580,7 +2588,12 @@ declare class StorageUtils implements TStorage {
      * @returns {Promise<void>} - A promise that resolves when the operation is complete.
      * @template T
      */
-    upsert: <T extends IStorageData = IStorageData>(item: T, clientId: string, agentName: AgentName, storageName: StorageName) => Promise<void>;
+    upsert: <T extends IStorageData = IStorageData>(payload: {
+        item: T;
+        clientId: string;
+        agentName: AgentName;
+        storageName: StorageName;
+    }) => Promise<void>;
     /**
      * Removes an item from the storage.
      * @param {IStorageData["id"]} itemId - The ID of the item to remove.
@@ -2589,7 +2602,12 @@ declare class StorageUtils implements TStorage {
      * @param {StorageName} storageName - The storage name.
      * @returns {Promise<void>} - A promise that resolves when the operation is complete.
      */
-    remove: (itemId: IStorageData["id"], clientId: string, agentName: AgentName, storageName: StorageName) => Promise<void>;
+    remove: (payload: {
+        itemId: IStorageData["id"];
+        clientId: string;
+        agentName: AgentName;
+        storageName: StorageName;
+    }) => Promise<void>;
     /**
      * Gets an item from the storage.
      * @param {IStorageData["id"]} itemId - The ID of the item to get.
@@ -2599,7 +2617,12 @@ declare class StorageUtils implements TStorage {
      * @returns {Promise<T | null>} - A promise that resolves to the item or null if not found.
      * @template T
      */
-    get: <T extends IStorageData = IStorageData>(itemId: IStorageData["id"], clientId: string, agentName: AgentName, storageName: StorageName) => Promise<T | null>;
+    get: <T extends IStorageData = IStorageData>(payload: {
+        itemId: IStorageData["id"];
+        clientId: string;
+        agentName: AgentName;
+        storageName: StorageName;
+    }) => Promise<T | null>;
     /**
      * Lists items from the storage.
      * @param {string} clientId - The client ID.
@@ -2609,7 +2632,12 @@ declare class StorageUtils implements TStorage {
      * @returns {Promise<T[]>} - A promise that resolves to an array of items.
      * @template T
      */
-    list: <T extends IStorageData = IStorageData>(clientId: string, agentName: AgentName, storageName: StorageName, filter?: (item: T) => boolean) => Promise<T[]>;
+    list: <T extends IStorageData = IStorageData>(payload: {
+        clientId: string;
+        agentName: AgentName;
+        storageName: StorageName;
+        filter?: (item: T) => boolean;
+    }) => Promise<T[]>;
     /**
      * Clears the storage.
      * @param {string} clientId - The client ID.
@@ -2617,7 +2645,11 @@ declare class StorageUtils implements TStorage {
      * @param {StorageName} storageName - The storage name.
      * @returns {Promise<void>} - A promise that resolves when the operation is complete.
      */
-    clear: (clientId: string, agentName: AgentName, storageName: StorageName) => Promise<void>;
+    clear: (payload: {
+        clientId: string;
+        agentName: AgentName;
+        storageName: StorageName;
+    }) => Promise<void>;
 }
 declare const Storage: StorageUtils;