agent-swarm-kit 1.0.50 → 1.0.52

package/build/index.cjs

@@ -218,18 +218,22 @@ var connectionServices$1 = {
     historyConnectionService: Symbol('historyConnectionService'),
     swarmConnectionService: Symbol('swarmConnectionService'),
     sessionConnectionService: Symbol('sessionConnectionService'),
+    storageConnectionService: Symbol('storageConnectionService'),
 };
-var specServices$1 = {
+var schemaServices$1 = {
     completionSchemaService: Symbol('completionSchemaService'),
     agentSchemaService: Symbol('agentSchemaService'),
     swarmSchemaService: Symbol('swarmSchemaService'),
     toolSchemaService: Symbol('toolSchemaService'),
+    embeddingSchemaService: Symbol('embeddingSchemaService'),
+    storageSchemaService: Symbol('storageSchemaService'),
 };
 var publicServices$1 = {
     agentPublicService: Symbol('agentPublicService'),
     historyPublicService: Symbol('historyPublicService'),
     sessionPublicService: Symbol('sessionPublicService'),
     swarmPublicService: Symbol('swarmPublicService'),
+    storagePublicService: Symbol('storagePublicService'),
 };
 var validationServices$1 = {
     agentValidationService: Symbol('agentValidationService'),
@@ -237,8 +241,10 @@ var validationServices$1 = {
     sessionValidationService: Symbol('sessionValidationService'),
     swarmValidationService: Symbol('swarmValidationService'),
     completionValidationService: Symbol('completionValidationService'),
+    embeddingValidationService: Symbol('embeddingValidationService'),
+    storageValidationService: Symbol('storageValidationService'),
 };
-var TYPES = __assign(__assign(__assign(__assign(__assign({}, baseServices$1), specServices$1), connectionServices$1), publicServices$1), validationServices$1);
+var TYPES = __assign(__assign(__assign(__assign(__assign({}, baseServices$1), schemaServices$1), connectionServices$1), publicServices$1), validationServices$1);
 
 /**
  * Service for managing agent schemas.
@@ -430,6 +436,8 @@ var CC_GET_AGENT_HISTORY = function () { return new functoolsKit.PubsubArrayAdap
 var CC_AGENT_OUTPUT_MAP = function (message) { return message; };
 var CC_AGENT_SEPARATE_HISTORY = false;
 var CC_AGENT_SYSTEM_PROMPT = undefined;
+var CC_STORAGE_SEARCH_SIMILARITY = 0.65;
+var CC_STORAGE_SEARCH_POOL = 5;
 var GLOBAL_CONFIG = {
     CC_TOOL_CALL_EXCEPTION_PROMPT: CC_TOOL_CALL_EXCEPTION_PROMPT,
     CC_EMPTY_OUTPUT_PLACEHOLDERS: CC_EMPTY_OUTPUT_PLACEHOLDERS,
@@ -445,6 +453,8 @@ var GLOBAL_CONFIG = {
     CC_AGENT_SEPARATE_HISTORY: CC_AGENT_SEPARATE_HISTORY,
     CC_AGENT_DISALLOWED_TAGS: CC_AGENT_DISALLOWED_TAGS,
     CC_AGENT_DISALLOWED_SYMBOLS: CC_AGENT_DISALLOWED_SYMBOLS,
+    CC_STORAGE_SEARCH_SIMILARITY: CC_STORAGE_SEARCH_SIMILARITY,
+    CC_STORAGE_SEARCH_POOL: CC_STORAGE_SEARCH_POOL,
 };
 var setConfig = function (config) {
     Object.assign(GLOBAL_CONFIG, config);
@@ -465,6 +475,7 @@ var ClientAgent = /** @class */ (function () {
     function ClientAgent(params) {
         var _this = this;
         this.params = params;
+        this._toolErrorSubject = new functoolsKit.Subject();
         this._toolCommitSubject = new functoolsKit.Subject();
         this._outputSubject = new functoolsKit.Subject();
         /**
@@ -810,14 +821,22 @@ var ClientAgent = /** @class */ (function () {
                                             params: tool.function.arguments,
                                             isLast: idx === toolCalls.length - 1,
                                             toolCalls: toolCalls,
-                                        })).then(function () {
+                                        }))
+                                            .then(function () {
                                             var _a, _b;
                                             ((_a = targetFn.callbacks) === null || _a === void 0 ? void 0 : _a.onAfterCall) &&
                                                 ((_b = targetFn.callbacks) === null || _b === void 0 ? void 0 : _b.onAfterCall(tool.id, _this.params.clientId, _this.params.agentName, tool.function.arguments));
+                                        })
+                                            .catch(function (error) {
+                                            var _a, _b;
+                                            ((_a = targetFn.callbacks) === null || _a === void 0 ? void 0 : _a.onCallError) &&
+                                                ((_b = targetFn.callbacks) === null || _b === void 0 ? void 0 : _b.onCallError(tool.id, _this.params.clientId, _this.params.agentName, tool.function.arguments, error));
+                                            _this._toolErrorSubject.next();
                                         });
                                         this_1.params.logger.debug("ClientAgent agentName=".concat(this_1.params.agentName, " clientId=").concat(this_1.params.clientId, " functionName=").concat(tool.function.name, " tool call executing"));
                                         return [4 /*yield*/, Promise.race([
                                                 this_1._toolCommitSubject.toPromise(),
+                                                this_1._toolErrorSubject.toPromise(),
                                                 this_1._outputSubject.toPromise(),
                                             ])];
                                     case 8:
@@ -1037,16 +1056,21 @@ var AgentConnectionService = /** @class */ (function () {
          * @returns {Promise<void>} The dispose result.
          */
         this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
+            var key;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("agentConnectionService dispose", {
                             context: this.contextService.context,
                         });
+                        key = "".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.agentName);
+                        if (!this.getAgent.has(key)) {
+                            return [2 /*return*/];
+                        }
                         return [4 /*yield*/, this.getAgent(this.contextService.context.clientId, this.contextService.context.agentName).dispose()];
                     case 1:
                         _a.sent();
-                        this.getAgent.clear("".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.agentName));
+                        this.getAgent.clear(key);
                         this.sessionValidationService.removeAgentUsage(this.contextService.context.clientId, this.contextService.context.agentName);
                         return [2 /*return*/];
                 }
@@ -1141,7 +1165,7 @@ var ClientHistory = /** @class */ (function () {
          * @returns {Promise<IModelMessage[]>} - The array of messages for the agent.
          */
         this.toArrayForAgent = function (prompt, system) { return __awaiter(_this, void 0, void 0, function () {
-            var commonMessagesRaw, systemMessagesRaw, _a, _b, _c, content, message, e_2_1, systemMessages, commonMessages, assistantToolCallSet, assistantMessages, promptMessages;
+            var commonMessagesRaw, systemMessagesRaw, _a, _b, _c, content, message, e_2_1, systemMessages, commonMessages, assistantToolOutputCallSet, assistantRawMessages, assistantToolCallSet, assistantMessages, promptMessages;
             var _this = this;
             var _d, e_2, _e, _f;
             var _g;
@@ -1208,22 +1232,41 @@ var ClientHistory = /** @class */ (function () {
                         commonMessages = commonMessagesRaw
                             .filter(this._filterCondition)
                             .slice(-GLOBAL_CONFIG.CC_KEEP_MESSAGES);
-                        assistantToolCallSet = new Set(commonMessages.map(function (_a) {
+                        assistantToolOutputCallSet = new Set(commonMessages.map(function (_a) {
                             var tool_call_id = _a.tool_call_id;
                             return tool_call_id;
                         }));
-                        assistantMessages = commonMessages
+                        assistantRawMessages = commonMessages
                             .map(function (_a) {
                             var tool_calls = _a.tool_calls, message = __rest(_a, ["tool_calls"]);
                             return (__assign(__assign({}, message), { tool_calls: tool_calls === null || tool_calls === void 0 ? void 0 : tool_calls.filter(function (_a) {
                                     var id = _a.id;
-                                    return assistantToolCallSet.has(id);
+                                    return assistantToolOutputCallSet.has(id);
                                 }) }));
                         })
                             .filter(function (_a) {
                             var content = _a.content, tool_calls = _a.tool_calls;
                             return !!content || !!(tool_calls === null || tool_calls === void 0 ? void 0 : tool_calls.length);
                         });
+                        assistantToolCallSet = new Set(assistantRawMessages
+                            .filter(function (_a) {
+                            var role = _a.role;
+                            return role === "tool";
+                        })
+                            .flatMap(function (_a) {
+                            var tool_calls = _a.tool_calls;
+                            return tool_calls === null || tool_calls === void 0 ? void 0 : tool_calls.map(function (_a) {
+                                var id = _a.id;
+                                return id;
+                            });
+                        }));
+                        assistantMessages = assistantRawMessages.filter(function (_a) {
+                            var role = _a.role, tool_call_id = _a.tool_call_id;
+                            if (role === "tool") {
+                                return assistantToolCallSet.has(tool_call_id);
+                            }
+                            return true;
+                        });
                         promptMessages = [];
                         {
                             promptMessages.push({
@@ -1356,12 +1399,17 @@ var HistoryConnectionService = /** @class */ (function () {
          * @returns {Promise<void>} A promise that resolves when the service is disposed.
          */
         this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
+            var key;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("historyConnectionService dispose", {
                             context: this.contextService.context,
                         });
+                        key = "".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.agentName);
+                        if (!this.getHistory.has(key)) {
+                            return [2 /*return*/];
+                        }
                         if (!GLOBAL_CONFIG.CC_AGENT_SEPARATE_HISTORY) return [3 /*break*/, 2];
                         return [4 /*yield*/, this.getItems(this.contextService.context.clientId, this.contextService.context.agentName).clear()];
                     case 1:
@@ -1369,7 +1417,7 @@ var HistoryConnectionService = /** @class */ (function () {
                         this.getItems.clear(this.contextService.context.clientId);
                         _a.label = 2;
                     case 2:
-                        this.getHistory.clear("".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.agentName));
+                        this.getHistory.clear(key);
                         this.sessionValidationService.removeHistoryUsage(this.contextService.context.clientId, this.contextService.context.agentName);
                         return [2 /*return*/];
                 }
@@ -1628,6 +1676,7 @@ var SwarmConnectionService = /** @class */ (function () {
                         });
                     });
                 },
+                callbacks: callbacks,
             });
         });
         /**
@@ -1719,11 +1768,16 @@ var SwarmConnectionService = /** @class */ (function () {
          * @returns {Promise<void>}
          */
         this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
+            var key;
             return __generator(this, function (_a) {
                 this.loggerService.log("swarmConnectionService dispose", {
                     context: this.contextService.context,
                 });
-                this.getSwarm.clear("".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.swarmName));
+                key = "".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.swarmName);
+                if (!this.getSwarm.has(key)) {
+                    return [2 /*return*/];
+                }
+                this.getSwarm.clear(key);
                 return [2 /*return*/];
             });
         }); };
@@ -1771,7 +1825,7 @@ var CompletionSchemaService = /** @class */ (function () {
         this.registry = new functoolsKit.ToolRegistry("completionSchemaService");
         /**
          * Registers a new completion schema.
-         * @param {string} key - The key for the schema.
+         * @param {CompletionName} key - The key for the schema.
          * @param {ICompletionSchema} value - The schema to register.
          */
         this.register = function (key, value) {
@@ -1780,7 +1834,7 @@ var CompletionSchemaService = /** @class */ (function () {
         };
         /**
          * Retrieves a completion schema by key.
-         * @param {string} key - The key of the schema to retrieve.
+         * @param {CompletionName} key - The key of the schema to retrieve.
          * @returns {ICompletionSchema} The retrieved schema.
          */
         this.get = function (key) {
@@ -2155,16 +2209,21 @@ var SessionConnectionService = /** @class */ (function () {
          * @returns {Promise<void>} A promise that resolves when the service is disposed.
          */
         this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
+            var key;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("sessionConnectionService dispose", {
                             context: this.contextService.context,
                         });
+                        key = "".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.swarmName);
+                        if (!this.getSession.has(key)) {
+                            return [2 /*return*/];
+                        }
                         return [4 /*yield*/, this.getSession(this.contextService.context.clientId, this.contextService.context.swarmName).dispose()];
                     case 1:
                         _a.sent();
-                        this.getSession.clear("".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.swarmName));
+                        this.getSession.clear(key);
                         return [2 /*return*/];
                 }
             });
@@ -2207,6 +2266,7 @@ var AgentPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2241,6 +2301,7 @@ var AgentPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2272,6 +2333,7 @@ var AgentPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2307,6 +2369,7 @@ var AgentPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2340,6 +2403,7 @@ var AgentPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2373,6 +2437,7 @@ var AgentPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2404,6 +2469,7 @@ var AgentPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2435,6 +2501,7 @@ var AgentPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2480,6 +2547,7 @@ var HistoryPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2513,6 +2581,7 @@ var HistoryPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2544,6 +2613,7 @@ var HistoryPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2575,6 +2645,7 @@ var HistoryPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2620,6 +2691,7 @@ var SessionPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2654,6 +2726,7 @@ var SessionPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2687,6 +2760,7 @@ var SessionPublicService = /** @class */ (function () {
                                     clientId: clientId,
                                     swarmName: swarmName,
                                     agentName: "",
+                                    storageName: "",
                                 })];
                             case 1: return [2 /*return*/, _a.sent()];
                         }
@@ -2699,12 +2773,14 @@ var SessionPublicService = /** @class */ (function () {
                         clientId: clientId,
                         swarmName: swarmName,
                         agentName: "",
+                        storageName: "",
                     });
                 };
             }, {
                 clientId: clientId,
                 swarmName: swarmName,
                 agentName: "",
+                storageName: "",
             });
         };
         /**
@@ -2737,6 +2813,7 @@ var SessionPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2770,6 +2847,7 @@ var SessionPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2803,6 +2881,7 @@ var SessionPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2834,6 +2913,7 @@ var SessionPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2865,6 +2945,7 @@ var SessionPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2908,6 +2989,7 @@ var SwarmPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2939,6 +3021,7 @@ var SwarmPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2970,6 +3053,7 @@ var SwarmPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -3005,6 +3089,7 @@ var SwarmPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -3038,6 +3123,7 @@ var SwarmPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -3069,6 +3155,7 @@ var SwarmPublicService = /** @class */ (function () {
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
+                                storageName: "",
                             })];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -3087,7 +3174,20 @@ var AgentValidationService = /** @class */ (function () {
         this.loggerService = inject(TYPES.loggerService);
         this.toolValidationService = inject(TYPES.toolValidationService);
         this.completionValidationService = inject(TYPES.completionValidationService);
+        this.storageValidationService = inject(TYPES.storageValidationService);
         this._agentMap = new Map();
+        /**
+         * Retrieves the storages used by the agent
+         * @param {agentName} agentName - The name of the swarm.
+         * @returns {string[]} The list of storage names.
+         * @throws Will throw an error if the swarm is not found.
+         */
+        this.getStorageList = function (agentName) {
+            if (!_this._agentMap.has(agentName)) {
+                throw new Error("agent-swarm agent ".concat(agentName, " not exist (getStorageList)"));
+            }
+            return _this._agentMap.get(agentName).storages;
+        };
         /**
          * Adds a new agent to the validation service.
          * @param {AgentName} agentName - The name of the agent.
@@ -3104,6 +3204,23 @@ var AgentValidationService = /** @class */ (function () {
             }
             _this._agentMap.set(agentName, agentSchema);
         };
+        /**
+         * Check if agent got registered storage
+         */
+        this.hasStorage = functoolsKit.memoize(function (_a) {
+            var _b = __read(_a, 2), agentName = _b[0], storageName = _b[1];
+            return "".concat(agentName, "-").concat(storageName);
+        }, function (agentName, storageName) {
+            _this.loggerService.log("agentValidationService hasStorage", {
+                agentName: agentName,
+                storageName: storageName,
+            });
+            if (!_this._agentMap.has(agentName)) {
+                throw new Error("agent-swarm agent ".concat(agentName, " not exist (hasStorage)"));
+            }
+            var _a = _this._agentMap.get(agentName).storages, storages = _a === void 0 ? [] : _a;
+            return storages.includes(storageName);
+        });
         /**
          * Validates an agent by its name and source.
          * @param {AgentName} agentName - The name of the agent.
@@ -3114,7 +3231,7 @@ var AgentValidationService = /** @class */ (function () {
             var _b = __read(_a, 1), agentName = _b[0];
             return agentName;
         }, function (agentName, source) {
-            var _a;
+            var _a, _b;
             _this.loggerService.log("agentValidationService validate", {
                 agentName: agentName,
                 source: source,
@@ -3130,6 +3247,12 @@ var AgentValidationService = /** @class */ (function () {
                 }
                 _this.toolValidationService.validate(toolName, source);
             });
+            (_b = agent.storages) === null || _b === void 0 ? void 0 : _b.forEach(function (storageName) {
+                if (typeof storageName !== "string") {
+                    throw new Error("agent-swarm agent ".concat(agentName, " storage list is invalid (storageName=").concat(String(storageName), ") source=").concat(source));
+                }
+                _this.storageValidationService.validate(storageName, source);
+            });
         });
     }
     return AgentValidationService;
@@ -3511,128 +3634,914 @@ var ToolValidationService = /** @class */ (function () {
     return ToolValidationService;
 }());
 
-{
-    provide(TYPES.loggerService, function () { return new LoggerService(); });
-    provide(TYPES.contextService, function () { return new ContextService(); });
-}
-{
-    provide(TYPES.agentConnectionService, function () { return new AgentConnectionService(); });
-    provide(TYPES.historyConnectionService, function () { return new HistoryConnectionService(); });
-    provide(TYPES.swarmConnectionService, function () { return new SwarmConnectionService(); });
-    provide(TYPES.sessionConnectionService, function () { return new SessionConnectionService(); });
-}
-{
-    provide(TYPES.agentSchemaService, function () { return new AgentSchemaService(); });
-    provide(TYPES.toolSchemaService, function () { return new ToolSchemaService(); });
-    provide(TYPES.swarmSchemaService, function () { return new SwarmSchemaService(); });
-    provide(TYPES.completionSchemaService, function () { return new CompletionSchemaService(); });
-}
-{
-    provide(TYPES.agentPublicService, function () { return new AgentPublicService(); });
-    provide(TYPES.historyPublicService, function () { return new HistoryPublicService(); });
-    provide(TYPES.sessionPublicService, function () { return new SessionPublicService(); });
-    provide(TYPES.swarmPublicService, function () { return new SwarmPublicService(); });
-}
-{
-    provide(TYPES.agentPublicService, function () { return new AgentPublicService(); });
-    provide(TYPES.historyPublicService, function () { return new HistoryPublicService(); });
-    provide(TYPES.sessionPublicService, function () { return new SessionPublicService(); });
-    provide(TYPES.swarmPublicService, function () { return new SwarmPublicService(); });
-}
-{
-    provide(TYPES.agentValidationService, function () { return new AgentValidationService(); });
-    provide(TYPES.completionValidationService, function () { return new CompletionValidationService(); });
-    provide(TYPES.sessionValidationService, function () { return new SessionValidationService(); });
-    provide(TYPES.swarmValidationService, function () { return new SwarmValidationService(); });
-    provide(TYPES.toolValidationService, function () { return new ToolValidationService(); });
-}
-
-var baseServices = {
-    loggerService: inject(TYPES.loggerService),
-    contextService: inject(TYPES.contextService),
-};
-var connectionServices = {
-    agentConnectionService: inject(TYPES.agentConnectionService),
-    historyConnectionService: inject(TYPES.historyConnectionService),
-    swarmConnectionService: inject(TYPES.swarmConnectionService),
-    sessionConnectionService: inject(TYPES.sessionConnectionService),
-};
-var specServices = {
-    agentSchemaService: inject(TYPES.agentSchemaService),
-    toolSchemaService: inject(TYPES.toolSchemaService),
-    swarmSchemaService: inject(TYPES.swarmSchemaService),
-    completionSchemaService: inject(TYPES.completionSchemaService),
-};
-var publicServices = {
-    agentPublicService: inject(TYPES.agentPublicService),
-    historyPublicService: inject(TYPES.historyPublicService),
-    sessionPublicService: inject(TYPES.sessionPublicService),
-    swarmPublicService: inject(TYPES.swarmPublicService),
-};
-var validationServices = {
-    agentValidationService: inject(TYPES.agentValidationService),
-    toolValidationService: inject(TYPES.toolValidationService),
-    sessionValidationService: inject(TYPES.sessionValidationService),
-    swarmValidationService: inject(TYPES.swarmValidationService),
-    completionValidationService: inject(TYPES.completionValidationService),
-};
-var swarm = __assign(__assign(__assign(__assign(__assign({}, baseServices), connectionServices), specServices), publicServices), validationServices);
-init();
-
 /**
- * Adds a new agent to the agent registry. The swarm takes only those agents which was registered
- *
- * @param {IAgentSchema} agentSchema - The schema of the agent to be added.
- * @returns {string} The name of the added agent.
+ * Service for managing embedding schemas.
  */
-var addAgent = function (agentSchema) {
-    swarm.loggerService.log('function addAgent', {
-        agentSchema: agentSchema
-    });
-    swarm.agentValidationService.addAgent(agentSchema.agentName, agentSchema);
-    swarm.agentSchemaService.register(agentSchema.agentName, agentSchema);
-    return agentSchema.agentName;
-};
+var EmbeddingSchemaService = /** @class */ (function () {
+    function EmbeddingSchemaService() {
+        var _this = this;
+        this.loggerService = inject(TYPES.loggerService);
+        this.registry = new functoolsKit.ToolRegistry("embeddingSchemaService");
+        /**
+         * Registers a embedding with the given key and value.
+         * @param {EmbeddingName} key - The name of the embedding.
+         * @param {IAgentTool} value - The embedding to register.
+         */
+        this.register = function (key, value) {
+            _this.loggerService.log('embeddingSchemaService register');
+            _this.registry = _this.registry.register(key, value);
+        };
+        /**
+         * Retrieves a embedding by its key.
+         * @param {EmbeddingName} key - The name of the embedding.
+         * @returns {IAgentTool} The embedding associated with the given key.
+         */
+        this.get = function (key) {
+            _this.loggerService.log('embeddingSchemaService get', { key: key });
+            return _this.registry.get(key);
+        };
+    }
+    return EmbeddingSchemaService;
+}());
 
 /**
- * Adds a completion engine for agents. Agents could use different models and
- * framewords for completion like: mock, gpt4all, ollama, openai
- *
- * @param {ICompletionSchema} completionSchema - The completion schema to be added.
- * @returns {string} The name of the completion that was added.
+ * Service for managing storage schemas.
  */
-var addCompletion = function (completionSchema) {
-    swarm.loggerService.log('function addCompletion', {
-        completionSchema: completionSchema,
-    });
-    swarm.completionValidationService.addCompletion(completionSchema.completionName);
-    swarm.completionSchemaService.register(completionSchema.completionName, completionSchema);
-    return completionSchema.completionName;
-};
+var StorageSchemaService = /** @class */ (function () {
+    function StorageSchemaService() {
+        var _this = this;
+        this.loggerService = inject(TYPES.loggerService);
+        this.registry = new functoolsKit.ToolRegistry("storageSchemaService");
+        /**
+         * Registers a new storage schema.
+         * @param {StorageName} key - The key for the schema.
+         * @param {IStorageSchema} value - The schema to register.
+         */
+        this.register = function (key, value) {
+            _this.loggerService.log("storageSchemaService register", { key: key });
+            _this.registry = _this.registry.register(key, value);
+        };
+        /**
+         * Retrieves a storage schema by key.
+         * @param {StorageName} key - The key of the schema to retrieve.
+         * @returns {IStorageSchema} The retrieved schema.
+         */
+        this.get = function (key) {
+            _this.loggerService.log("storageSchemaService get", { key: key });
+            return _this.registry.get(key);
+        };
+    }
+    return StorageSchemaService;
+}());
 
-/**
- * Adds a new swarm to the system. The swarm is a root for starting client session
- *
- * @param {ISwarmSchema} swarmSchema - The schema of the swarm to be added.
- * @returns {string} The name of the added swarm.
- */
-var addSwarm = function (swarmSchema) {
-    swarm.loggerService.log('function addSwarm', {
-        swarmSchema: swarmSchema,
-    });
-    swarm.swarmValidationService.addSwarm(swarmSchema.swarmName, swarmSchema);
-    swarm.swarmSchemaService.register(swarmSchema.swarmName, swarmSchema);
-    return swarmSchema.swarmName;
-};
+var ClientStorage = /** @class */ (function () {
+    function ClientStorage(params) {
+        var _this = this;
+        this.params = params;
+        this._itemMap = new Map();
+        this._createEmbedding = functoolsKit.memoize(function (_a) {
+            var _b = __read(_a, 1), id = _b[0].id;
+            return id;
+        }, function (item) { return __awaiter(_this, void 0, void 0, function () {
+            var index, embeddings;
+            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"), {
+                            id: item.id,
+                        });
+                        return [4 /*yield*/, this.params.createIndex(item)];
+                    case 1:
+                        index = _a.sent();
+                        return [4 /*yield*/, this.params.createEmbedding(index)];
+                    case 2:
+                        embeddings = _a.sent();
+                        if (this.params.onCreate) {
+                            this.params.onCreate(index, embeddings, this.params.clientId, this.params.embedding);
+                        }
+                        return [2 /*return*/, [embeddings, index]];
+                }
+            });
+        }); });
+        this.waitForInit = functoolsKit.singleshot(function () { return __awaiter(_this, void 0, void 0, function () {
+            var data;
+            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"));
+                        if (!this.params.getData) {
+                            return [2 /*return*/];
+                        }
+                        return [4 /*yield*/, this.params.getData(this.params.clientId, this.params.storageName)];
+                    case 1:
+                        data = _a.sent();
+                        return [4 /*yield*/, Promise.all(data.map(functoolsKit.execpool(this._createEmbedding, {
+                                delay: 10,
+                                maxExec: GLOBAL_CONFIG.CC_STORAGE_SEARCH_POOL,
+                            })))];
+                    case 2:
+                        _a.sent();
+                        this._itemMap = new Map(data.map(function (item) { return [item.id, item]; }));
+                        return [2 /*return*/];
+                }
+            });
+        }); });
+        this.take = function (search, total) { return __awaiter(_this, void 0, void 0, function () {
+            var indexed, searchEmbeddings, filtered;
+            var _this = this;
+            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, " take"), {
+                            search: search,
+                            total: total,
+                        });
+                        indexed = [];
+                        return [4 /*yield*/, this.params.createEmbedding(search)];
+                    case 1:
+                        searchEmbeddings = _a.sent();
+                        if (this.params.onCreate) {
+                            this.params.onCreate(search, searchEmbeddings, this.params.clientId, this.params.embedding);
+                        }
+                        return [4 /*yield*/, Promise.all(Array.from(this._itemMap.values()).map(functoolsKit.execpool(function (item) { return __awaiter(_this, void 0, void 0, function () {
+                                var _a, targetEmbeddings, index, score;
+                                return __generator(this, function (_b) {
+                                    switch (_b.label) {
+                                        case 0: return [4 /*yield*/, this._createEmbedding(item)];
+                                        case 1:
+                                            _a = __read.apply(void 0, [_b.sent(), 2]), targetEmbeddings = _a[0], index = _a[1];
+                                            return [4 /*yield*/, this.params.calculateSimilarity(searchEmbeddings, targetEmbeddings)];
+                                        case 2:
+                                            score = _b.sent();
+                                            if (this.params.onCompare) {
+                                                this.params.onCompare(search, index, score, this.params.clientId, this.params.embedding);
+                                            }
+                                            indexed.push({ data: item, score: score });
+                                            return [2 /*return*/];
+                                    }
+                                });
+                            }); }, {
+                                delay: 10,
+                                maxExec: GLOBAL_CONFIG.CC_STORAGE_SEARCH_POOL,
+                            })))];
+                    case 2:
+                        _a.sent();
+                        filtered = indexed
+                            .filter(function (_a) {
+                            var score = _a.score;
+                            return score > GLOBAL_CONFIG.CC_STORAGE_SEARCH_SIMILARITY;
+                        })
+                            .sort(function (_a, _b) {
+                            var a = _a.score;
+                            var b = _b.score;
+                            return b - a;
+                        });
+                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " take filtered"), {
+                            filtered: filtered.map(function (_a) {
+                                var data = _a.data, score = _a.score;
+                                return ({ id: data.id, score: score });
+                            }),
+                        });
+                        return [2 /*return*/, filtered.map(function (_a) {
+                                var data = _a.data;
+                                return data;
+                            })];
+                }
+            });
+        }); };
+        this.upsert = function (item) { return __awaiter(_this, void 0, void 0, function () {
+            var _a, _b;
+            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"), {
+                            item: item,
+                        });
+                        this._itemMap.set(item.id, item);
+                        this._createEmbedding.clear(item.id);
+                        return [4 /*yield*/, this._createEmbedding(item)];
+                    case 1:
+                        _c.sent();
+                        if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onUpdate) {
+                            (_b = this.params.callbacks) === null || _b === void 0 ? void 0 : _b.onUpdate(__spreadArray([], __read(this._itemMap.values()), false), this.params.clientId, this.params.storageName);
+                        }
+                        return [2 /*return*/];
+                }
+            });
+        }); };
+        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"), {
+                    id: itemId,
+                });
+                this._itemMap.delete(itemId);
+                this._createEmbedding.clear(itemId);
+                if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onUpdate) {
+                    (_b = this.params.callbacks) === null || _b === void 0 ? void 0 : _b.onUpdate(__spreadArray([], __read(this._itemMap.values()), false), this.params.clientId, this.params.storageName);
+                }
+                return [2 /*return*/];
+            });
+        }); };
+        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._itemMap.clear();
+                this._createEmbedding.clear();
+                return [2 /*return*/];
+            });
+        }); };
+        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"), {
+                    id: itemId,
+                });
+                return [2 /*return*/, (_a = this._itemMap.get(itemId)) !== null && _a !== void 0 ? _a : null];
+            });
+        }); };
+        this.list = function (filter) { return __awaiter(_this, void 0, void 0, 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"));
+                if (!filter) {
+                    return [2 /*return*/, __spreadArray([], __read(this._itemMap.values()), false)];
+                }
+                result = [];
+                try {
+                    for (_a = __values(this._itemMap.values()), _b = _a.next(); !_b.done; _b = _a.next()) {
+                        item = _b.value;
+                        if (filter(item)) {
+                            result.push(item);
+                        }
+                    }
+                }
+                catch (e_1_1) { e_1 = { error: e_1_1 }; }
+                finally {
+                    try {
+                        if (_b && !_b.done && (_c = _a.return)) _c.call(_a);
+                    }
+                    finally { if (e_1) throw e_1.error; }
+                }
+                return [2 /*return*/, result];
+            });
+        }); };
+        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " CTOR"), {
+            params: params,
+        });
+    }
+    return ClientStorage;
+}());
 
 /**
- * Adds a new tool for agents in a swarm. Tool should be registered in `addAgent`
- * declaration
- *
- * @param {IAgentTool} toolSchema - The schema of the tool to be added.
- * @returns {string} The name of the tool that was added.
+ * Service for managing storage connections.
+ * @implements {IStorage}
  */
-var addTool = function (toolSchema) {
+var StorageConnectionService = /** @class */ (function () {
+    function StorageConnectionService() {
+        var _this = this;
+        this.loggerService = inject(TYPES.loggerService);
+        this.contextService = inject(TYPES.contextService);
+        this.storageSchemaService = inject(TYPES.storageSchemaService);
+        this.embeddingSchemaService = inject(TYPES.embeddingSchemaService);
+        /**
+         * Retrieves a storage instance based on client ID and storage name.
+         * @param {string} clientId - The client ID.
+         * @param {string} storageName - The storage name.
+         * @returns {ClientStorage} The client storage instance.
+         */
+        this.getStorage = functoolsKit.memoize(function (_a) {
+            var _b = __read(_a, 2), clientId = _b[0], storageName = _b[1];
+            return "".concat(clientId, "-").concat(storageName);
+        }, function (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 }));
+        });
+        /**
+         * Retrieves a list of storage data based on a search query and total number of items.
+         * @param {string} search - The search query.
+         * @param {number} total - The total number of items to retrieve.
+         * @returns {Promise<IStorageData[]>} The list of storage data.
+         */
+        this.take = function (search, total) { return __awaiter(_this, void 0, void 0, function () {
+            var storage;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("storageConnectionService take", {
+                            context: this.contextService.context,
+                            search: search,
+                            total: total,
+                        });
+                        storage = this.getStorage(this.contextService.context.clientId, this.contextService.context.storageName);
+                        return [4 /*yield*/, storage.waitForInit()];
+                    case 1:
+                        _a.sent();
+                        return [4 /*yield*/, storage.take(search, total)];
+                    case 2: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Upserts an item in the storage.
+         * @param {IStorageData} item - The item to upsert.
+         * @returns {Promise<void>}
+         */
+        this.upsert = function (item) { return __awaiter(_this, void 0, void 0, function () {
+            var storage;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("storageConnectionService upsert", {
+                            context: this.contextService.context,
+                            item: item,
+                        });
+                        storage = this.getStorage(this.contextService.context.clientId, this.contextService.context.storageName);
+                        return [4 /*yield*/, storage.waitForInit()];
+                    case 1:
+                        _a.sent();
+                        return [4 /*yield*/, storage.upsert(item)];
+                    case 2: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Removes an item from the storage.
+         * @param {IStorageData["id"]} itemId - The ID of the item to remove.
+         * @returns {Promise<void>}
+         */
+        this.remove = function (itemId) { return __awaiter(_this, void 0, void 0, function () {
+            var storage;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("storageConnectionService remove", {
+                            context: this.contextService.context,
+                            itemId: itemId,
+                        });
+                        storage = this.getStorage(this.contextService.context.clientId, this.contextService.context.storageName);
+                        return [4 /*yield*/, storage.waitForInit()];
+                    case 1:
+                        _a.sent();
+                        return [4 /*yield*/, storage.remove(itemId)];
+                    case 2: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Retrieves an item from the storage by its ID.
+         * @param {IStorageData["id"]} itemId - The ID of the item to retrieve.
+         * @returns {Promise<IStorageData>} The retrieved item.
+         */
+        this.get = function (itemId) { return __awaiter(_this, void 0, void 0, function () {
+            var storage;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("storageConnectionService get", {
+                            context: this.contextService.context,
+                            itemId: itemId,
+                        });
+                        storage = this.getStorage(this.contextService.context.clientId, this.contextService.context.storageName);
+                        return [4 /*yield*/, storage.waitForInit()];
+                    case 1:
+                        _a.sent();
+                        return [4 /*yield*/, storage.get(itemId)];
+                    case 2: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Retrieves a list of items from the storage, optionally filtered by a predicate function.
+         * @param {function(IStorageData): boolean} [filter] - The optional filter function.
+         * @returns {Promise<IStorageData[]>} The list of items.
+         */
+        this.list = function (filter) { return __awaiter(_this, void 0, void 0, function () {
+            var storage;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("storageConnectionService list", {
+                            context: this.contextService.context,
+                        });
+                        storage = this.getStorage(this.contextService.context.clientId, this.contextService.context.storageName);
+                        return [4 /*yield*/, storage.waitForInit()];
+                    case 1:
+                        _a.sent();
+                        return [4 /*yield*/, storage.list(filter)];
+                    case 2: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Clears all items from the storage.
+         * @returns {Promise<void>}
+         */
+        this.clear = function () { return __awaiter(_this, void 0, void 0, function () {
+            var storage;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("storageConnectionService clear", {
+                            context: this.contextService.context,
+                        });
+                        storage = this.getStorage(this.contextService.context.clientId, this.contextService.context.storageName);
+                        return [4 /*yield*/, storage.waitForInit()];
+                    case 1:
+                        _a.sent();
+                        return [4 /*yield*/, storage.clear()];
+                    case 2: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Disposes of the storage connection.
+         * @returns {Promise<void>}
+         */
+        this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
+            var key;
+            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*/];
+                }
+                this.getStorage.clear(key);
+                return [2 /*return*/];
+            });
+        }); };
+    }
+    return StorageConnectionService;
+}());
+
+/**
+ * Service for managing public storage interactions.
+ */
+var StoragePublicService = /** @class */ (function () {
+    function StoragePublicService() {
+        var _this = this;
+        this.loggerService = inject(TYPES.loggerService);
+        this.storageConnectionService = inject(TYPES.storageConnectionService);
+        /**
+         * Retrieves a list of storage data based on a search query and total number of items.
+         * @param {string} search - The search query.
+         * @param {number} total - The total number of items to retrieve.
+         * @returns {Promise<IStorageData[]>} The list of storage data.
+         */
+        this.take = function (search, total, clientId, storageName) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("storagePublicService take", {
+                            search: search,
+                            total: total,
+                            clientId: clientId,
+                            storageName: storageName,
+                        });
+                        return [4 /*yield*/, ContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.storageConnectionService.take(search, total)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                clientId: clientId,
+                                storageName: storageName,
+                                agentName: "",
+                                swarmName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Upserts an item in the storage.
+         * @param {IStorageData} item - The item to upsert.
+         * @returns {Promise<void>}
+         */
+        this.upsert = function (item, clientId, storageName) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("storagePublicService upsert", {
+                            item: item,
+                            clientId: clientId,
+                            storageName: storageName,
+                        });
+                        return [4 /*yield*/, ContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.storageConnectionService.upsert(item)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                clientId: clientId,
+                                storageName: storageName,
+                                agentName: "",
+                                swarmName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Removes an item from the storage.
+         * @param {IStorageData["id"]} itemId - The ID of the item to remove.
+         * @returns {Promise<void>}
+         */
+        this.remove = function (itemId, clientId, storageName) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("storagePublicService remove", {
+                            itemId: itemId,
+                            clientId: clientId,
+                            storageName: storageName,
+                        });
+                        return [4 /*yield*/, ContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.storageConnectionService.remove(itemId)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                clientId: clientId,
+                                storageName: storageName,
+                                agentName: "",
+                                swarmName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Retrieves an item from the storage by its ID.
+         * @param {IStorageData["id"]} itemId - The ID of the item to retrieve.
+         * @returns {Promise<IStorageData>} The retrieved item.
+         */
+        this.get = function (itemId, clientId, storageName) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("storagePublicService get", {
+                            itemId: itemId,
+                            clientId: clientId,
+                            storageName: storageName,
+                        });
+                        return [4 /*yield*/, ContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.storageConnectionService.get(itemId)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                clientId: clientId,
+                                storageName: storageName,
+                                agentName: "",
+                                swarmName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Retrieves a list of items from the storage, optionally filtered by a predicate function.
+         * @param {function(IStorageData): boolean} [filter] - The optional filter function.
+         * @returns {Promise<IStorageData[]>} The list of items.
+         */
+        this.list = function (clientId, storageName, filter) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("storagePublicService list", {
+                            clientId: clientId,
+                            storageName: storageName,
+                        });
+                        return [4 /*yield*/, ContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.storageConnectionService.list(filter)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                clientId: clientId,
+                                storageName: storageName,
+                                agentName: "",
+                                swarmName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Clears all items from the storage.
+         * @returns {Promise<void>}
+         */
+        this.clear = function (clientId, storageName) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("storagePublicService clear", {
+                            clientId: clientId,
+                            storageName: storageName,
+                        });
+                        return [4 /*yield*/, ContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.storageConnectionService.clear()];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                clientId: clientId,
+                                storageName: storageName,
+                                agentName: "",
+                                swarmName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Disposes of the storage.
+         * @param {string} clientId - The client ID.
+         * @param {StorageName} storageName - The storage name.
+         * @returns {Promise<void>}
+         */
+        this.dispose = function (clientId, storageName) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("storagePublicService dispose", {
+                            clientId: clientId,
+                            storageName: storageName,
+                        });
+                        return [4 /*yield*/, ContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.storageConnectionService.dispose()];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                clientId: clientId,
+                                storageName: storageName,
+                                agentName: "",
+                                swarmName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+    }
+    return StoragePublicService;
+}());
+
+/**
+ * Service for validating storages within the storage swarm.
+ */
+var StorageValidationService = /** @class */ (function () {
+    function StorageValidationService() {
+        var _this = this;
+        this.loggerService = inject(TYPES.loggerService);
+        this.embeddingValidationService = inject(TYPES.embeddingValidationService);
+        this._storageMap = new Map();
+        /**
+         * Adds a new storage to the validation service.
+         * @param {StorageName} storageName - The name of the storage.
+         * @param {IStorageSchema} storageSchema - The schema of the storage.
+         * @throws {Error} If the storage already exists.
+         */
+        this.addStorage = function (storageName, storageSchema) {
+            _this.loggerService.log("storageValidationService addStorage", {
+                storageName: storageName,
+                storageSchema: storageSchema,
+            });
+            if (_this._storageMap.has(storageName)) {
+                throw new Error("storage-swarm storage ".concat(storageName, " already exist"));
+            }
+            _this._storageMap.set(storageName, storageSchema);
+        };
+        /**
+         * Validates an storage by its name and source.
+         * @param {StorageName} storageName - The name of the storage.
+         * @param {string} source - The source of the validation request.
+         * @throws {Error} If the storage is not found.
+         */
+        this.validate = functoolsKit.memoize(function (_a) {
+            var _b = __read(_a, 1), storageName = _b[0];
+            return storageName;
+        }, function (storageName, source) {
+            _this.loggerService.log("storageValidationService validate", {
+                storageName: storageName,
+                source: source,
+            });
+            var storage = _this._storageMap.get(storageName);
+            if (!storage) {
+                throw new Error("storage-swarm storage ".concat(storageName, " not found source=").concat(source));
+            }
+            _this.embeddingValidationService.validate(storage.embedding, source);
+        });
+    }
+    return StorageValidationService;
+}());
+
+/**
+ * Service for validating embeddings within the agent-swarm.
+ */
+var EmbeddingValidationService = /** @class */ (function () {
+    function EmbeddingValidationService() {
+        var _this = this;
+        this.loggerService = inject(TYPES.loggerService);
+        this._embeddingMap = new Map();
+        /**
+         * Adds a new embedding to the validation service.
+         * @param {EmbeddingName} embeddingName - The name of the embedding to add.
+         * @param {IAgentEmbedding} embeddingSchema - The schema of the embedding to add.
+         * @throws Will throw an error if the embedding already exists.
+         */
+        this.addEmbedding = function (embeddingName, embeddingSchema) {
+            _this.loggerService.log("embeddingValidationService addEmbedding", {
+                embeddingName: embeddingName,
+                embeddingSchema: embeddingSchema,
+            });
+            if (_this._embeddingMap.has(embeddingName)) {
+                throw new Error("agent-swarm embedding ".concat(embeddingName, " already exist"));
+            }
+            _this._embeddingMap.set(embeddingName, embeddingSchema);
+        };
+        /**
+         * Validates if a embedding exists in the validation service.
+         * @param {EmbeddingName} embeddingName - The name of the embedding to validate.
+         * @param {string} source - The source of the validation request.
+         * @throws Will throw an error if the embedding is not found.
+         */
+        this.validate = functoolsKit.memoize(function (_a) {
+            var _b = __read(_a, 1), embeddingName = _b[0];
+            return embeddingName;
+        }, function (embeddingName, source) {
+            _this.loggerService.log("embeddingValidationService validate", {
+                embeddingName: embeddingName,
+                source: source,
+            });
+            if (!_this._embeddingMap.has(embeddingName)) {
+                throw new Error("agent-swarm embedding ".concat(embeddingName, " not found source=").concat(source));
+            }
+        });
+    }
+    return EmbeddingValidationService;
+}());
+
+{
+    provide(TYPES.loggerService, function () { return new LoggerService(); });
+    provide(TYPES.contextService, function () { return new ContextService(); });
+}
+{
+    provide(TYPES.agentConnectionService, function () { return new AgentConnectionService(); });
+    provide(TYPES.historyConnectionService, function () { return new HistoryConnectionService(); });
+    provide(TYPES.swarmConnectionService, function () { return new SwarmConnectionService(); });
+    provide(TYPES.sessionConnectionService, function () { return new SessionConnectionService(); });
+    provide(TYPES.storageConnectionService, function () { return new StorageConnectionService(); });
+}
+{
+    provide(TYPES.agentSchemaService, function () { return new AgentSchemaService(); });
+    provide(TYPES.toolSchemaService, function () { return new ToolSchemaService(); });
+    provide(TYPES.swarmSchemaService, function () { return new SwarmSchemaService(); });
+    provide(TYPES.completionSchemaService, function () { return new CompletionSchemaService(); });
+    provide(TYPES.embeddingSchemaService, function () { return new EmbeddingSchemaService(); });
+    provide(TYPES.storageSchemaService, function () { return new StorageSchemaService(); });
+}
+{
+    provide(TYPES.agentPublicService, function () { return new AgentPublicService(); });
+    provide(TYPES.historyPublicService, function () { return new HistoryPublicService(); });
+    provide(TYPES.sessionPublicService, function () { return new SessionPublicService(); });
+    provide(TYPES.swarmPublicService, function () { return new SwarmPublicService(); });
+    provide(TYPES.storagePublicService, function () { return new StoragePublicService(); });
+}
+{
+    provide(TYPES.agentPublicService, function () { return new AgentPublicService(); });
+    provide(TYPES.historyPublicService, function () { return new HistoryPublicService(); });
+    provide(TYPES.sessionPublicService, function () { return new SessionPublicService(); });
+    provide(TYPES.swarmPublicService, function () { return new SwarmPublicService(); });
+}
+{
+    provide(TYPES.agentValidationService, function () { return new AgentValidationService(); });
+    provide(TYPES.completionValidationService, function () { return new CompletionValidationService(); });
+    provide(TYPES.sessionValidationService, function () { return new SessionValidationService(); });
+    provide(TYPES.swarmValidationService, function () { return new SwarmValidationService(); });
+    provide(TYPES.toolValidationService, function () { return new ToolValidationService(); });
+    provide(TYPES.storageValidationService, function () { return new StorageValidationService(); });
+    provide(TYPES.embeddingValidationService, function () { return new EmbeddingValidationService(); });
+}
+
+var baseServices = {
+    loggerService: inject(TYPES.loggerService),
+    contextService: inject(TYPES.contextService),
+};
+var connectionServices = {
+    agentConnectionService: inject(TYPES.agentConnectionService),
+    historyConnectionService: inject(TYPES.historyConnectionService),
+    swarmConnectionService: inject(TYPES.swarmConnectionService),
+    sessionConnectionService: inject(TYPES.sessionConnectionService),
+    storageConnectionService: inject(TYPES.storageConnectionService),
+};
+var schemaServices = {
+    agentSchemaService: inject(TYPES.agentSchemaService),
+    toolSchemaService: inject(TYPES.toolSchemaService),
+    swarmSchemaService: inject(TYPES.swarmSchemaService),
+    completionSchemaService: inject(TYPES.completionSchemaService),
+    embeddingSchemaService: inject(TYPES.embeddingSchemaService),
+    storageSchemaService: inject(TYPES.storageSchemaService),
+};
+var publicServices = {
+    agentPublicService: inject(TYPES.agentPublicService),
+    historyPublicService: inject(TYPES.historyPublicService),
+    sessionPublicService: inject(TYPES.sessionPublicService),
+    swarmPublicService: inject(TYPES.swarmPublicService),
+    storagePublicService: inject(TYPES.storagePublicService),
+};
+var validationServices = {
+    agentValidationService: inject(TYPES.agentValidationService),
+    toolValidationService: inject(TYPES.toolValidationService),
+    sessionValidationService: inject(TYPES.sessionValidationService),
+    swarmValidationService: inject(TYPES.swarmValidationService),
+    completionValidationService: inject(TYPES.completionValidationService),
+    storageValidationService: inject(TYPES.storageValidationService),
+    embeddingValidationService: inject(TYPES.embeddingValidationService),
+};
+var swarm = __assign(__assign(__assign(__assign(__assign({}, baseServices), connectionServices), schemaServices), publicServices), validationServices);
+init();
+
+/**
+ * Adds a new agent to the agent registry. The swarm takes only those agents which was registered
+ *
+ * @param {IAgentSchema} agentSchema - The schema of the agent to be added.
+ * @returns {string} The name of the added agent.
+ */
+var addAgent = function (agentSchema) {
+    swarm.loggerService.log('function addAgent', {
+        agentSchema: agentSchema
+    });
+    swarm.agentValidationService.addAgent(agentSchema.agentName, agentSchema);
+    swarm.agentSchemaService.register(agentSchema.agentName, agentSchema);
+    return agentSchema.agentName;
+};
+
+/**
+ * Adds a completion engine for agents. Agents could use different models and
+ * framewords for completion like: mock, gpt4all, ollama, openai
+ *
+ * @param {ICompletionSchema} completionSchema - The completion schema to be added.
+ * @returns {string} The name of the completion that was added.
+ */
+var addCompletion = function (completionSchema) {
+    swarm.loggerService.log('function addCompletion', {
+        completionSchema: completionSchema,
+    });
+    swarm.completionValidationService.addCompletion(completionSchema.completionName);
+    swarm.completionSchemaService.register(completionSchema.completionName, completionSchema);
+    return completionSchema.completionName;
+};
+
+/**
+ * Adds a new swarm to the system. The swarm is a root for starting client session
+ *
+ * @param {ISwarmSchema} swarmSchema - The schema of the swarm to be added.
+ * @returns {string} The name of the added swarm.
+ */
+var addSwarm = function (swarmSchema) {
+    swarm.loggerService.log('function addSwarm', {
+        swarmSchema: swarmSchema,
+    });
+    swarm.swarmValidationService.addSwarm(swarmSchema.swarmName, swarmSchema);
+    swarm.swarmSchemaService.register(swarmSchema.swarmName, swarmSchema);
+    return swarmSchema.swarmName;
+};
+
+/**
+ * Adds a new tool for agents in a swarm. Tool should be registered in `addAgent`
+ * declaration
+ *
+ * @param {IAgentTool} toolSchema - The schema of the tool to be added.
+ * @returns {string} The name of the tool that was added.
+ */
+var addTool = function (toolSchema) {
     swarm.loggerService.log('function addTool', {
         toolSchema: toolSchema,
     });
@@ -3641,6 +4550,36 @@ var addTool = function (toolSchema) {
     return toolSchema.toolName;
 };
 
+/**
+ * Adds a new embedding to the embedding registry. The swarm takes only those embeddings which was registered
+ *
+ * @param {IEmbeddingSchema} embeddingSchema - The schema of the embedding to be added.
+ * @returns {string} The name of the added embedding.
+ */
+var addEmbedding = function (embeddingSchema) {
+    swarm.loggerService.log('function addEmbedding', {
+        embeddingSchema: embeddingSchema
+    });
+    swarm.embeddingValidationService.addEmbedding(embeddingSchema.embeddingName, embeddingSchema);
+    swarm.embeddingSchemaService.register(embeddingSchema.embeddingName, embeddingSchema);
+    return embeddingSchema.embeddingName;
+};
+
+/**
+ * Adds a new storage to the storage registry. The swarm takes only those storages which was registered
+ *
+ * @param {IStorageSchema} storageSchema - The schema of the storage to be added.
+ * @returns {string} The name of the added storage.
+ */
+var addStorage = function (storageSchema) {
+    swarm.loggerService.log('function addStorage', {
+        storageSchema: storageSchema
+    });
+    swarm.storageValidationService.addStorage(storageSchema.storageName, storageSchema);
+    swarm.storageSchemaService.register(storageSchema.storageName, storageSchema);
+    return storageSchema.storageName;
+};
+
 /**
  * Commits a user message to the active agent history in as swarm without answer.
  *
@@ -3952,6 +4891,25 @@ var disposeConnection = function (clientId, swarmName) { return __awaiter(void 0
                         }); }),
                     ])];
             case 3:
+                _a.sent();
+                return [4 /*yield*/, Promise.all([
+                        swarm.swarmValidationService
+                            .getAgentList(swarmName)
+                            .flatMap(function (agentName) {
+                            return swarm.agentValidationService.getStorageList(agentName);
+                        })
+                            .map(function (storageName) { return __awaiter(void 0, void 0, void 0, function () {
+                            return __generator(this, function (_a) {
+                                switch (_a.label) {
+                                    case 0: return [4 /*yield*/, swarm.storagePublicService.dispose(clientId, storageName)];
+                                    case 1:
+                                        _a.sent();
+                                        return [2 /*return*/];
+                                }
+                            });
+                        }); }),
+                    ])];
+            case 4:
                 _a.sent();
                 return [2 /*return*/];
         }
@@ -4797,9 +5755,179 @@ var makeAutoDispose = function (clientId, swarmName, _a) {
     };
 };
 
+var StorageUtils = /** @class */ (function () {
+    function StorageUtils() {
+        var _this = this;
+        /**
+         * Takes items from the storage.
+         * @param {string} search - The search query.
+         * @param {number} total - The total number of items to take.
+         * @param {string} clientId - The client ID.
+         * @param {AgentName} agentName - The agent name.
+         * @param {StorageName} storageName - The storage name.
+         * @returns {Promise<T[]>} - A promise that resolves to an array of items.
+         * @template T
+         */
+        this.take = function (search, total, clientId, agentName, storageName) { 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,
+                        });
+                        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)"));
+                        }
+                        return [4 /*yield*/, swarm.storagePublicService.take(search, total, clientId, storageName)];
+                    case 1: return [2 /*return*/, (_a.sent())];
+                }
+            });
+        }); };
+        /**
+         * Upserts an item in the storage.
+         * @param {T} item - The item to upsert.
+         * @param {string} clientId - The client ID.
+         * @param {AgentName} agentName - The agent name.
+         * @param {StorageName} storageName - The storage name.
+         * @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 () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        swarm.loggerService.log("StorageStatic upsert", {
+                            item: item,
+                            clientId: clientId,
+                            storageName: 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)"));
+                        }
+                        return [4 /*yield*/, swarm.storagePublicService.upsert(item, clientId, storageName)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Removes an item from the storage.
+         * @param {IStorageData["id"]} itemId - The ID of the item to remove.
+         * @param {string} clientId - The client ID.
+         * @param {AgentName} agentName - The agent name.
+         * @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 () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        swarm.loggerService.log("StorageStatic remove", {
+                            itemId: itemId,
+                            clientId: clientId,
+                            storageName: 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)"));
+                        }
+                        return [4 /*yield*/, swarm.storagePublicService.remove(itemId, clientId, storageName)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Gets an item from the storage.
+         * @param {IStorageData["id"]} itemId - The ID of the item to get.
+         * @param {string} clientId - The client ID.
+         * @param {AgentName} agentName - The agent name.
+         * @param {StorageName} storageName - The storage name.
+         * @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 () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        swarm.loggerService.log("StorageStatic get", {
+                            itemId: itemId,
+                            clientId: clientId,
+                            storageName: 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)"));
+                        }
+                        return [4 /*yield*/, swarm.storagePublicService.get(itemId, clientId, storageName)];
+                    case 1: return [2 /*return*/, (_a.sent())];
+                }
+            });
+        }); };
+        /**
+         * Lists items from the storage.
+         * @param {string} clientId - The client ID.
+         * @param {AgentName} agentName - The agent name.
+         * @param {StorageName} storageName - The storage name.
+         * @param {(item: T) => boolean} [filter] - Optional filter 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 () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        swarm.loggerService.log("StorageStatic list", {
+                            clientId: clientId,
+                            storageName: 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)"));
+                        }
+                        return [4 /*yield*/, swarm.storagePublicService.list(clientId, storageName, filter)];
+                    case 1: return [2 /*return*/, (_a.sent())];
+                }
+            });
+        }); };
+        /**
+         * Clears the storage.
+         * @param {string} clientId - The client ID.
+         * @param {AgentName} agentName - The agent name.
+         * @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 () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        swarm.loggerService.log("StorageStatic clear", {
+                            clientId: clientId,
+                            storageName: 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)"));
+                        }
+                        return [4 /*yield*/, swarm.storagePublicService.clear(clientId, storageName)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+    }
+    return StorageUtils;
+}());
+var Storage = new StorageUtils();
+
 exports.ContextService = ContextService;
+exports.Storage = Storage;
 exports.addAgent = addAgent;
 exports.addCompletion = addCompletion;
+exports.addEmbedding = addEmbedding;
+exports.addStorage = addStorage;
 exports.addSwarm = addSwarm;
 exports.addTool = addTool;
 exports.changeAgent = changeAgent;