agent-swarm-kit 1.0.74 → 1.0.76

package/build/index.mjs

@@ -1,6 +1,6 @@
 import { scoped } from 'di-scoped';
-import { ToolRegistry, trycatch, singleshot, memoize, sleep, Subject, queued, getErrorMessage, errorData, not, randomString, createAwaiter, cancelable, CANCELED_PROMISE_SYMBOL, execpool, SortedArray, schedule, ttl, Source } from 'functools-kit';
 import { createActivator } from 'di-kit';
+import { ToolRegistry, trycatch, singleshot, memoize, sleep, Subject, queued, getErrorMessage, errorData, not, randomString, createAwaiter, cancelable, CANCELED_PROMISE_SYMBOL, execpool, SortedArray, schedule, ttl, Source } from 'functools-kit';
 import { omit } from 'lodash-es';
 import xml2js from 'xml2js';
 
@@ -165,6 +165,50 @@ var ContextService = scoped(/** @class */ (function () {
     return class_1;
 }()));
 
+var _a;
+var init = (_a = createActivator("agent-swarm"), _a.init), inject = _a.inject, provide = _a.provide;
+
+var baseServices$1 = {
+    busService: Symbol('busService'),
+    loggerService: Symbol('loggerService'),
+    contextService: Symbol('contextService'),
+};
+var connectionServices$1 = {
+    agentConnectionService: Symbol('agentConnectionService'),
+    historyConnectionService: Symbol('historyConnectionService'),
+    swarmConnectionService: Symbol('swarmConnectionService'),
+    sessionConnectionService: Symbol('sessionConnectionService'),
+    storageConnectionService: Symbol('storageConnectionService'),
+    stateConnectionService: Symbol('stateConnectionService'),
+};
+var schemaServices$1 = {
+    completionSchemaService: Symbol('completionSchemaService'),
+    agentSchemaService: Symbol('agentSchemaService'),
+    swarmSchemaService: Symbol('swarmSchemaService'),
+    toolSchemaService: Symbol('toolSchemaService'),
+    embeddingSchemaService: Symbol('embeddingSchemaService'),
+    storageSchemaService: Symbol('storageSchemaService'),
+    stateSchemaService: Symbol('stateSchemaService'),
+};
+var publicServices$1 = {
+    agentPublicService: Symbol('agentPublicService'),
+    historyPublicService: Symbol('historyPublicService'),
+    sessionPublicService: Symbol('sessionPublicService'),
+    swarmPublicService: Symbol('swarmPublicService'),
+    storagePublicService: Symbol('storagePublicService'),
+    statePublicService: Symbol('statePublicService'),
+};
+var validationServices$1 = {
+    agentValidationService: Symbol('agentValidationService'),
+    toolValidationService: Symbol('toolValidationService'),
+    sessionValidationService: Symbol('sessionValidationService'),
+    swarmValidationService: Symbol('swarmValidationService'),
+    completionValidationService: Symbol('completionValidationService'),
+    embeddingValidationService: Symbol('embeddingValidationService'),
+    storageValidationService: Symbol('storageValidationService'),
+};
+var TYPES = __assign(__assign(__assign(__assign(__assign({}, baseServices$1), schemaServices$1), connectionServices$1), publicServices$1), validationServices$1);
+
 /**
  * LoggerService class that implements the ILogger interface.
  * Provides methods to log and debug messages.
@@ -172,6 +216,7 @@ var ContextService = scoped(/** @class */ (function () {
 var LoggerService = /** @class */ (function () {
     function LoggerService() {
         var _this = this;
+        this.contextService = inject(TYPES.contextService);
         this._logger = {
             /**
              * Logs messages.
@@ -191,24 +236,32 @@ var LoggerService = /** @class */ (function () {
          * @param {...any} args - The messages to log.
          */
         this.log = function () {
-            var _a;
+            var _a, _b;
             var args = [];
             for (var _i = 0; _i < arguments.length; _i++) {
                 args[_i] = arguments[_i];
             }
-            (_a = _this._logger).log.apply(_a, __spreadArray([], __read(args), false));
+            if (ContextService.hasContext()) {
+                (_a = _this._logger).log.apply(_a, __spreadArray(__spreadArray([], __read(args), false), [_this.contextService.context], false));
+                return;
+            }
+            (_b = _this._logger).log.apply(_b, __spreadArray([], __read(args), false));
         };
         /**
          * Logs debug messages using the current logger.
          * @param {...any} args - The debug messages to log.
          */
         this.debug = function () {
-            var _a;
+            var _a, _b;
             var args = [];
             for (var _i = 0; _i < arguments.length; _i++) {
                 args[_i] = arguments[_i];
             }
-            (_a = _this._logger).debug.apply(_a, __spreadArray([], __read(args), false));
+            if (ContextService.hasContext()) {
+                (_a = _this._logger).debug.apply(_a, __spreadArray(__spreadArray([], __read(args), false), [_this.contextService.context], false));
+                return;
+            }
+            (_b = _this._logger).debug.apply(_b, __spreadArray([], __read(args), false));
         };
         /**
          * Sets a new logger.
@@ -221,50 +274,6 @@ var LoggerService = /** @class */ (function () {
     return LoggerService;
 }());
 
-var _a;
-var init = (_a = createActivator("agent-swarm"), _a.init), inject = _a.inject, provide = _a.provide;
-
-var baseServices$1 = {
-    busService: Symbol('busService'),
-    loggerService: Symbol('loggerService'),
-    contextService: Symbol('contextService'),
-};
-var connectionServices$1 = {
-    agentConnectionService: Symbol('agentConnectionService'),
-    historyConnectionService: Symbol('historyConnectionService'),
-    swarmConnectionService: Symbol('swarmConnectionService'),
-    sessionConnectionService: Symbol('sessionConnectionService'),
-    storageConnectionService: Symbol('storageConnectionService'),
-    stateConnectionService: Symbol('stateConnectionService'),
-};
-var schemaServices$1 = {
-    completionSchemaService: Symbol('completionSchemaService'),
-    agentSchemaService: Symbol('agentSchemaService'),
-    swarmSchemaService: Symbol('swarmSchemaService'),
-    toolSchemaService: Symbol('toolSchemaService'),
-    embeddingSchemaService: Symbol('embeddingSchemaService'),
-    storageSchemaService: Symbol('storageSchemaService'),
-    stateSchemaService: Symbol('stateSchemaService'),
-};
-var publicServices$1 = {
-    agentPublicService: Symbol('agentPublicService'),
-    historyPublicService: Symbol('historyPublicService'),
-    sessionPublicService: Symbol('sessionPublicService'),
-    swarmPublicService: Symbol('swarmPublicService'),
-    storagePublicService: Symbol('storagePublicService'),
-    statePublicService: Symbol('statePublicService'),
-};
-var validationServices$1 = {
-    agentValidationService: Symbol('agentValidationService'),
-    toolValidationService: Symbol('toolValidationService'),
-    sessionValidationService: Symbol('sessionValidationService'),
-    swarmValidationService: Symbol('swarmValidationService'),
-    completionValidationService: Symbol('completionValidationService'),
-    embeddingValidationService: Symbol('embeddingValidationService'),
-    storageValidationService: Symbol('storageValidationService'),
-};
-var TYPES = __assign(__assign(__assign(__assign(__assign({}, baseServices$1), schemaServices$1), connectionServices$1), publicServices$1), validationServices$1);
-
 /**
  * Service for managing agent schemas.
  */
@@ -1556,7 +1565,6 @@ var AgentConnectionService = /** @class */ (function () {
                         this.loggerService.log("agentConnectionService execute", {
                             input: input,
                             mode: mode,
-                            context: this.contextService.context,
                         });
                         return [4 /*yield*/, this.getAgent(this.contextService.context.clientId, this.contextService.context.agentName).execute(input, mode)];
                     case 1: return [2 /*return*/, _a.sent()];
@@ -1571,9 +1579,7 @@ var AgentConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("agentConnectionService waitForOutput", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("agentConnectionService waitForOutput");
                         return [4 /*yield*/, this.getAgent(this.contextService.context.clientId, this.contextService.context.agentName).waitForOutput()];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -1592,7 +1598,6 @@ var AgentConnectionService = /** @class */ (function () {
                         this.loggerService.log("agentConnectionService commitToolOutput", {
                             content: content,
                             toolId: toolId,
-                            context: this.contextService.context,
                         });
                         return [4 /*yield*/, this.getAgent(this.contextService.context.clientId, this.contextService.context.agentName).commitToolOutput(toolId, content)];
                     case 1: return [2 /*return*/, _a.sent()];
@@ -1610,7 +1615,6 @@ var AgentConnectionService = /** @class */ (function () {
                     case 0:
                         this.loggerService.log("agentConnectionService commitSystemMessage", {
                             message: message,
-                            context: this.contextService.context,
                         });
                         return [4 /*yield*/, this.getAgent(this.contextService.context.clientId, this.contextService.context.agentName).commitSystemMessage(message)];
                     case 1: return [2 /*return*/, _a.sent()];
@@ -1628,7 +1632,6 @@ var AgentConnectionService = /** @class */ (function () {
                     case 0:
                         this.loggerService.log("agentConnectionService commitUserMessage", {
                             message: message,
-                            context: this.contextService.context,
                         });
                         return [4 /*yield*/, this.getAgent(this.contextService.context.clientId, this.contextService.context.agentName).commitUserMessage(message)];
                     case 1: return [2 /*return*/, _a.sent()];
@@ -1643,9 +1646,7 @@ var AgentConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("agentConnectionService commitAgentChange", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("agentConnectionService commitAgentChange");
                         return [4 /*yield*/, this.getAgent(this.contextService.context.clientId, this.contextService.context.agentName).commitAgentChange()];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -1659,9 +1660,7 @@ var AgentConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("agentConnectionService commitFlush", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("agentConnectionService commitFlush");
                         return [4 /*yield*/, this.getAgent(this.contextService.context.clientId, this.contextService.context.agentName).commitFlush()];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -1676,9 +1675,7 @@ var AgentConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("agentConnectionService dispose", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("agentConnectionService dispose");
                         key = "".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.agentName);
                         if (!this.getAgent.has(key)) {
                             return [2 /*return*/];
@@ -1999,7 +1996,7 @@ var HistoryConnectionService = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("historyConnectionService push", {
-                            context: this.contextService.context,
+                            message: message,
                         });
                         return [4 /*yield*/, this.getHistory(this.contextService.context.clientId, this.contextService.context.agentName).push(message)];
                     case 1: return [2 /*return*/, _a.sent()];
@@ -2016,7 +2013,7 @@ var HistoryConnectionService = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("historyConnectionService toArrayForAgent", {
-                            context: this.contextService.context,
+                            prompt: prompt,
                         });
                         return [4 /*yield*/, this.getHistory(this.contextService.context.clientId, this.contextService.context.agentName).toArrayForAgent(prompt)];
                     case 1: return [2 /*return*/, _a.sent()];
@@ -2031,9 +2028,7 @@ var HistoryConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("historyConnectionService toArrayForRaw", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("historyConnectionService toArrayForRaw");
                         return [4 /*yield*/, this.getHistory(this.contextService.context.clientId, this.contextService.context.agentName).toArrayForRaw()];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2048,9 +2043,7 @@ var HistoryConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("historyConnectionService dispose", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("historyConnectionService dispose");
                         key = "".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.agentName);
                         if (!this.getHistory.has(key)) {
                             return [2 /*return*/];
@@ -2443,9 +2436,7 @@ var SwarmConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("swarmConnectionService cancelOutput", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("swarmConnectionService cancelOutput");
                         return [4 /*yield*/, this.getSwarm(this.contextService.context.clientId, this.contextService.context.swarmName).cancelOutput()];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2459,9 +2450,7 @@ var SwarmConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("swarmConnectionService waitForOutput", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("swarmConnectionService waitForOutput");
                         return [4 /*yield*/, this.getSwarm(this.contextService.context.clientId, this.contextService.context.swarmName).waitForOutput()];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2475,9 +2464,7 @@ var SwarmConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("swarmConnectionService getAgentName", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("swarmConnectionService getAgentName");
                         return [4 /*yield*/, this.getSwarm(this.contextService.context.clientId, this.contextService.context.swarmName).getAgentName()];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2491,9 +2478,7 @@ var SwarmConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("swarmConnectionService getAgent", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("swarmConnectionService getAgent");
                         return [4 /*yield*/, this.getSwarm(this.contextService.context.clientId, this.contextService.context.swarmName).getAgent()];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -2510,7 +2495,7 @@ var SwarmConnectionService = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("swarmConnectionService setAgentRef", {
-                            context: this.contextService.context,
+                            agentName: agentName,
                         });
                         return [4 /*yield*/, this.getSwarm(this.contextService.context.clientId, this.contextService.context.swarmName).setAgentRef(agentName, agent)];
                     case 1: return [2 /*return*/, _a.sent()];
@@ -2528,7 +2513,6 @@ var SwarmConnectionService = /** @class */ (function () {
                     case 0:
                         this.loggerService.log("swarmConnectionService setAgentName", {
                             agentName: agentName,
-                            context: this.contextService.context,
                         });
                         return [4 /*yield*/, this.getSwarm(this.contextService.context.clientId, this.contextService.context.swarmName).setAgentName(agentName)];
                     case 1: return [2 /*return*/, _a.sent()];
@@ -2542,9 +2526,7 @@ var SwarmConnectionService = /** @class */ (function () {
         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.loggerService.log("swarmConnectionService dispose");
                 key = "".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.swarmName);
                 if (!this.getSwarm.has(key)) {
                     return [2 /*return*/];
@@ -2894,17 +2876,22 @@ var ClientSession = /** @class */ (function () {
                 clientId: _this.params.clientId,
             });
             return function (incoming) { return __awaiter(_this, void 0, void 0, function () {
-                var _a;
+                var data, _a;
                 var _b;
                 return __generator(this, function (_c) {
                     switch (_c.label) {
                         case 0:
                             this.params.logger.debug("ClientSession clientId=".concat(this.params.clientId, " connect call"));
-                            _a = connector;
-                            _b = {};
                             return [4 /*yield*/, this.execute(incoming.data, "user")];
                         case 1:
-                            _b.data = _c.sent();
+                            data = _c.sent();
+                            if (!data) {
+                                return [2 /*return*/];
+                            }
+                            _a = connector;
+                            _b = {
+                                data: data
+                            };
                             return [4 /*yield*/, this.params.swarm.getAgentName()];
                         case 2: return [4 /*yield*/, _a.apply(void 0, [(_b.agentName = _c.sent(),
                                     _b.clientId = incoming.clientId,
@@ -2971,7 +2958,7 @@ var SessionConnectionService = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("sessionConnectionService emit", {
-                            context: this.contextService.context,
+                            content: content,
                         });
                         return [4 /*yield*/, this.getSession(this.contextService.context.clientId, this.contextService.context.swarmName).emit(content)];
                     case 1: return [2 /*return*/, _a.sent()];
@@ -2988,7 +2975,6 @@ var SessionConnectionService = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("sessionConnectionService execute", {
-                            context: this.contextService.context,
                             content: content,
                             mode: mode,
                         });
@@ -3002,11 +2988,9 @@ var SessionConnectionService = /** @class */ (function () {
          * @param {SendMessageFn} connector - The function to send messages.
          * @returns {ReceiveMessageFn} The function to receive messages.
          */
-        this.connect = function (connector) {
-            _this.loggerService.log("sessionConnectionService connect", {
-                context: _this.contextService.context,
-            });
-            return _this.getSession(_this.contextService.context.clientId, _this.contextService.context.swarmName).connect(connector);
+        this.connect = function (connector, clientId, swarmName) {
+            _this.loggerService.log("sessionConnectionService connect");
+            return _this.getSession(clientId, swarmName).connect(connector);
         };
         /**
          * Commits tool output to the session.
@@ -3019,7 +3003,6 @@ var SessionConnectionService = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("sessionConnectionService commitToolOutput", {
-                            context: this.contextService.context,
                             content: content,
                             toolId: toolId,
                         });
@@ -3038,7 +3021,6 @@ var SessionConnectionService = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("sessionConnectionService commitSystemMessage", {
-                            context: this.contextService.context,
                             message: message,
                         });
                         return [4 /*yield*/, this.getSession(this.contextService.context.clientId, this.contextService.context.swarmName).commitSystemMessage(message)];
@@ -3056,7 +3038,6 @@ var SessionConnectionService = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("sessionConnectionService commitUserMessage", {
-                            context: this.contextService.context,
                             message: message,
                         });
                         return [4 /*yield*/, this.getSession(this.contextService.context.clientId, this.contextService.context.swarmName).commitUserMessage(message)];
@@ -3073,9 +3054,7 @@ var SessionConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("sessionConnectionService commitFlush", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("sessionConnectionService commitFlush");
                         return [4 /*yield*/, this.getSession(this.contextService.context.clientId, this.contextService.context.swarmName).commitFlush()];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -3090,9 +3069,7 @@ var SessionConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("sessionConnectionService dispose", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("sessionConnectionService dispose");
                         key = "".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.swarmName);
                         if (!this.getSession.has(key)) {
                             return [2 /*return*/];
@@ -3123,12 +3100,13 @@ var AgentPublicService = /** @class */ (function () {
          * @param {AgentName} agentName - The name of the agent.
          * @returns {Promise<unknown>} The agent reference.
          */
-        this.createAgentRef = function (clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+        this.createAgentRef = function (requestId, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("agentPublicService createAgentRef", {
+                            requestId: requestId,
                             clientId: clientId,
                             agentName: agentName,
                         });
@@ -3140,6 +3118,7 @@ var AgentPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
@@ -3157,12 +3136,13 @@ var AgentPublicService = /** @class */ (function () {
          * @param {AgentName} agentName - The name of the agent.
          * @returns {Promise<unknown>} The execution result.
          */
-        this.execute = function (input, mode, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+        this.execute = function (input, mode, requestId, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("agentPublicService execute", {
+                            requestId: requestId,
                             input: input,
                             clientId: clientId,
                             agentName: agentName,
@@ -3176,6 +3156,7 @@ var AgentPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
@@ -3192,12 +3173,13 @@ var AgentPublicService = /** @class */ (function () {
          * @param {AgentName} agentName - The name of the agent.
          * @returns {Promise<unknown>} The output result.
          */
-        this.waitForOutput = function (clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+        this.waitForOutput = function (requestId, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("agentPublicService waitForOutput", {
+                            requestId: requestId,
                             clientId: clientId,
                             agentName: agentName,
                         });
@@ -3209,6 +3191,7 @@ var AgentPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
@@ -3227,12 +3210,13 @@ var AgentPublicService = /** @class */ (function () {
          * @param {AgentName} agentName - The name of the agent.
          * @returns {Promise<unknown>} The commit result.
          */
-        this.commitToolOutput = function (toolId, content, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+        this.commitToolOutput = function (toolId, content, requestId, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("agentPublicService commitToolOutput", {
+                            requestId: requestId,
                             content: content,
                             clientId: clientId,
                             toolId: toolId,
@@ -3246,6 +3230,7 @@ var AgentPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
@@ -3263,12 +3248,13 @@ var AgentPublicService = /** @class */ (function () {
          * @param {AgentName} agentName - The name of the agent.
          * @returns {Promise<unknown>} The commit result.
          */
-        this.commitSystemMessage = function (message, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+        this.commitSystemMessage = function (message, requestId, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("agentPublicService commitSystemMessage", {
+                            requestId: requestId,
                             message: message,
                             clientId: clientId,
                             agentName: agentName,
@@ -3281,6 +3267,7 @@ var AgentPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
@@ -3298,12 +3285,13 @@ var AgentPublicService = /** @class */ (function () {
          * @param {AgentName} agentName - The name of the agent.
          * @returns {Promise<unknown>} The commit result.
          */
-        this.commitUserMessage = function (message, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+        this.commitUserMessage = function (message, requestId, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("agentPublicService commitUserMessage", {
+                            requestId: requestId,
                             message: message,
                             clientId: clientId,
                             agentName: agentName,
@@ -3316,6 +3304,7 @@ var AgentPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
@@ -3332,12 +3321,13 @@ var AgentPublicService = /** @class */ (function () {
          * @param {AgentName} agentName - The name of the agent.
          * @returns {Promise<unknown>} The commit result.
          */
-        this.commitFlush = function (clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+        this.commitFlush = function (requestId, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("agentPublicService commitFlush", {
+                            requestId: requestId,
                             clientId: clientId,
                             agentName: agentName,
                         });
@@ -3349,6 +3339,7 @@ var AgentPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
@@ -3365,12 +3356,13 @@ var AgentPublicService = /** @class */ (function () {
          * @param {AgentName} agentName - The name of the agent.
          * @returns {Promise<unknown>} The commit result.
          */
-        this.commitAgentChange = function (clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+        this.commitAgentChange = function (requestId, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("agentPublicService commitAgentChange", {
+                            requestId: requestId,
                             clientId: clientId,
                             agentName: agentName,
                         });
@@ -3382,6 +3374,7 @@ var AgentPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
@@ -3398,12 +3391,13 @@ var AgentPublicService = /** @class */ (function () {
          * @param {AgentName} agentName - The name of the agent.
          * @returns {Promise<unknown>} The dispose result.
          */
-        this.dispose = function (clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+        this.dispose = function (requestId, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("agentPublicService dispose", {
+                            requestId: requestId,
                             clientId: clientId,
                             agentName: agentName,
                         });
@@ -3415,6 +3409,7 @@ var AgentPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
@@ -3444,12 +3439,13 @@ var HistoryPublicService = /** @class */ (function () {
          * @param {AgentName} agentName - The agent name.
          * @returns {Promise<void>} A promise that resolves when the operation is complete.
          */
-        this.push = function (message, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+        this.push = function (message, requestId, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("historyPublicService push", {
+                            requestId: requestId,
                             message: message,
                             clientId: clientId,
                             agentName: agentName,
@@ -3462,6 +3458,7 @@ var HistoryPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
@@ -3479,7 +3476,7 @@ var HistoryPublicService = /** @class */ (function () {
          * @param {AgentName} agentName - The agent name.
          * @returns {Promise<any[]>} A promise that resolves to an array of history items.
          */
-        this.toArrayForAgent = function (prompt, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+        this.toArrayForAgent = function (prompt, requestId, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -3497,6 +3494,7 @@ var HistoryPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
@@ -3513,12 +3511,13 @@ var HistoryPublicService = /** @class */ (function () {
          * @param {AgentName} agentName - The agent name.
          * @returns {Promise<any[]>} A promise that resolves to a raw array of history items.
          */
-        this.toArrayForRaw = function (clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+        this.toArrayForRaw = function (requestId, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("historyPublicService toArrayForRaw", {
+                            requestId: requestId,
                             clientId: clientId,
                             agentName: agentName,
                         });
@@ -3530,6 +3529,7 @@ var HistoryPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
@@ -3546,7 +3546,7 @@ var HistoryPublicService = /** @class */ (function () {
          * @param {AgentName} agentName - The agent name.
          * @returns {Promise<void>} A promise that resolves when the operation is complete.
          */
-        this.dispose = function (clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+        this.dispose = function (requestId, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -3563,6 +3563,7 @@ var HistoryPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 agentName: agentName,
                                 swarmName: "",
@@ -3592,7 +3593,7 @@ var SessionPublicService = /** @class */ (function () {
          * @param {SwarmName} swarmName - The swarm name.
          * @returns {Promise<void>}
          */
-        this.emit = function (content, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
+        this.emit = function (content, requestId, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -3600,6 +3601,7 @@ var SessionPublicService = /** @class */ (function () {
                         this.loggerService.log("sessionPublicService emit", {
                             content: content,
                             clientId: clientId,
+                            requestId: requestId,
                             swarmName: swarmName,
                         });
                         return [4 /*yield*/, ContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
@@ -3610,6 +3612,7 @@ var SessionPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
@@ -3627,7 +3630,7 @@ var SessionPublicService = /** @class */ (function () {
          * @param {SwarmName} swarmName - The swarm name.
          * @returns {Promise<void>}
          */
-        this.execute = function (content, mode, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
+        this.execute = function (content, mode, requestId, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -3646,6 +3649,7 @@ var SessionPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
@@ -3663,52 +3667,13 @@ var SessionPublicService = /** @class */ (function () {
          * @param {SwarmName} swarmName - The swarm name.
          * @returns {ReceiveMessageFn}
          */
-        this.connect = function (connector, clientId, swarmName) {
+        this.connect = function (connector, requestId, clientId, swarmName) {
             _this.loggerService.log("sessionPublicService connect", {
+                requestId: requestId,
                 clientId: clientId,
                 swarmName: swarmName,
             });
-            return ContextService.runInContext(function () {
-                var receive = _this.sessionConnectionService.connect(function (outgoing) { return __awaiter(_this, void 0, void 0, function () {
-                    var _this = this;
-                    return __generator(this, function (_a) {
-                        switch (_a.label) {
-                            case 0: 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*/, connector(outgoing)];
-                                            case 1: return [2 /*return*/, _a.sent()];
-                                        }
-                                    });
-                                }); }, {
-                                    clientId: clientId,
-                                    swarmName: swarmName,
-                                    agentName: "",
-                                    storageName: "",
-                                    stateName: "",
-                                })];
-                            case 1: return [2 /*return*/, _a.sent()];
-                        }
-                    });
-                }); });
-                return function (incoming) {
-                    return ContextService.runInContext(function () {
-                        return receive(incoming);
-                    }, {
-                        clientId: clientId,
-                        swarmName: swarmName,
-                        agentName: "",
-                        storageName: "",
-                        stateName: "",
-                    });
-                };
-            }, {
-                clientId: clientId,
-                swarmName: swarmName,
-                agentName: "",
-                storageName: "",
-                stateName: "",
-            });
+            return _this.sessionConnectionService.connect(connector, clientId, swarmName);
         };
         /**
          * Commits tool output to the session.
@@ -3718,12 +3683,13 @@ var SessionPublicService = /** @class */ (function () {
          * @param {SwarmName} swarmName - The swarm name.
          * @returns {Promise<void>}
          */
-        this.commitToolOutput = function (toolId, content, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
+        this.commitToolOutput = function (toolId, content, requestId, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("sessionPublicService commitToolOutput", {
+                            requestId: requestId,
                             toolId: toolId,
                             content: content,
                             clientId: clientId,
@@ -3737,6 +3703,7 @@ var SessionPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
@@ -3754,12 +3721,13 @@ var SessionPublicService = /** @class */ (function () {
          * @param {SwarmName} swarmName - The swarm name.
          * @returns {Promise<void>}
          */
-        this.commitSystemMessage = function (message, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
+        this.commitSystemMessage = function (message, requestId, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("sessionPublicService commitSystemMessage", {
+                            requestId: requestId,
                             message: message,
                             clientId: clientId,
                             swarmName: swarmName,
@@ -3772,6 +3740,7 @@ var SessionPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
@@ -3789,12 +3758,13 @@ var SessionPublicService = /** @class */ (function () {
          * @param {SwarmName} swarmName - The swarm name.
          * @returns {Promise<void>}
          */
-        this.commitUserMessage = function (message, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
+        this.commitUserMessage = function (message, requestId, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("sessionPublicService commitUserMessage", {
+                            requestId: requestId,
                             message: message,
                             clientId: clientId,
                             swarmName: swarmName,
@@ -3807,6 +3777,7 @@ var SessionPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
@@ -3823,7 +3794,7 @@ var SessionPublicService = /** @class */ (function () {
          * @param {SwarmName} swarmName - The swarm name.
          * @returns {Promise<void>}
          */
-        this.commitFlush = function (clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
+        this.commitFlush = function (requestId, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -3840,6 +3811,7 @@ var SessionPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
@@ -3856,12 +3828,13 @@ var SessionPublicService = /** @class */ (function () {
          * @param {SwarmName} swarmName - The swarm name.
          * @returns {Promise<void>}
          */
-        this.dispose = function (clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
+        this.dispose = function (requestId, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("sessionPublicService dispose", {
+                            requestId: requestId,
                             clientId: clientId,
                             swarmName: swarmName,
                         });
@@ -3873,6 +3846,7 @@ var SessionPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
@@ -3901,7 +3875,7 @@ var SwarmPublicService = /** @class */ (function () {
          * @param {SwarmName} swarmName - The swarm name.
          * @returns {Promise<void>}
          */
-        this.cancelOutput = function (clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
+        this.cancelOutput = function (requestId, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -3918,6 +3892,7 @@ var SwarmPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
@@ -3934,13 +3909,14 @@ var SwarmPublicService = /** @class */ (function () {
          * @param {SwarmName} swarmName - The swarm name.
          * @returns {Promise<void>}
          */
-        this.waitForOutput = function (clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
+        this.waitForOutput = function (requestId, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("swarmPublicService waitForOutput", {
                             clientId: clientId,
+                            requestId: requestId,
                             swarmName: swarmName,
                         });
                         return [4 /*yield*/, ContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
@@ -3951,6 +3927,7 @@ var SwarmPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
@@ -3967,7 +3944,7 @@ var SwarmPublicService = /** @class */ (function () {
          * @param {SwarmName} swarmName - The swarm name.
          * @returns {Promise<string>}
          */
-        this.getAgentName = function (clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
+        this.getAgentName = function (requestId, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -3975,6 +3952,7 @@ var SwarmPublicService = /** @class */ (function () {
                         this.loggerService.log("swarmPublicService getAgentName", {
                             clientId: clientId,
                             swarmName: swarmName,
+                            requestId: requestId,
                         });
                         return [4 /*yield*/, ContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
                                 return __generator(this, function (_a) {
@@ -3984,6 +3962,7 @@ var SwarmPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
@@ -4000,7 +3979,7 @@ var SwarmPublicService = /** @class */ (function () {
          * @param {SwarmName} swarmName - The swarm name.
          * @returns {Promise<IAgent>}
          */
-        this.getAgent = function (clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
+        this.getAgent = function (requestId, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -4017,6 +3996,7 @@ var SwarmPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
@@ -4035,12 +4015,13 @@ var SwarmPublicService = /** @class */ (function () {
          * @param {IAgent} agent - The agent instance.
          * @returns {Promise<void>}
          */
-        this.setAgentRef = function (clientId, swarmName, agentName, agent) { return __awaiter(_this, void 0, void 0, function () {
+        this.setAgentRef = function (requestId, clientId, swarmName, agentName, agent) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("swarmPublicService setAgentRef", {
+                            requestId: requestId,
                             agentName: agentName,
                             agent: agent,
                             clientId: clientId,
@@ -4054,6 +4035,7 @@ var SwarmPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
@@ -4071,12 +4053,13 @@ var SwarmPublicService = /** @class */ (function () {
          * @param {SwarmName} swarmName - The swarm name.
          * @returns {Promise<void>}
          */
-        this.setAgentName = function (agentName, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
+        this.setAgentName = function (agentName, requestId, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("swarmPublicService setAgentName", {
+                            requestId: requestId,
                             agentName: agentName,
                             clientId: clientId,
                             swarmName: swarmName,
@@ -4089,6 +4072,7 @@ var SwarmPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
@@ -4105,12 +4089,13 @@ var SwarmPublicService = /** @class */ (function () {
          * @param {SwarmName} swarmName - The swarm name.
          * @returns {Promise<void>}
          */
-        this.dispose = function (clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
+        this.dispose = function (requestId, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("swarmPublicService dispose", {
+                            requestId: requestId,
                             clientId: clientId,
                             swarmName: swarmName,
                         });
@@ -4122,6 +4107,7 @@ var SwarmPublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 swarmName: swarmName,
                                 agentName: "",
@@ -5214,7 +5200,6 @@ var StorageConnectionService = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("storageConnectionService take", {
-                            context: this.contextService.context,
                             search: search,
                             total: total,
                             score: score,
@@ -5239,7 +5224,6 @@ var StorageConnectionService = /** @class */ (function () {
                 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);
@@ -5262,7 +5246,6 @@ var StorageConnectionService = /** @class */ (function () {
                 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);
@@ -5285,7 +5268,6 @@ var StorageConnectionService = /** @class */ (function () {
                 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);
@@ -5307,9 +5289,7 @@ var StorageConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("storageConnectionService list", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("storageConnectionService list");
                         storage = this.getStorage(this.contextService.context.clientId, this.contextService.context.storageName);
                         return [4 /*yield*/, storage.waitForInit()];
                     case 1:
@@ -5328,9 +5308,7 @@ var StorageConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("storageConnectionService clear", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("storageConnectionService clear");
                         storage = this.getStorage(this.contextService.context.clientId, this.contextService.context.storageName);
                         return [4 /*yield*/, storage.waitForInit()];
                     case 1:
@@ -5349,9 +5327,7 @@ var StorageConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("storageConnectionService dispose", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("storageConnectionService dispose");
                         key = "".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.storageName);
                         if (!this.getStorage.has(key)) {
                             return [2 /*return*/];
@@ -5390,12 +5366,13 @@ var StoragePublicService = /** @class */ (function () {
          * @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, score) { return __awaiter(_this, void 0, void 0, function () {
+        this.take = function (search, total, requestId, clientId, storageName, score) { 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", {
+                            requestId: requestId,
                             search: search,
                             total: total,
                             clientId: clientId,
@@ -5410,6 +5387,7 @@ var StoragePublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 storageName: storageName,
                                 agentName: "",
@@ -5425,7 +5403,7 @@ var StoragePublicService = /** @class */ (function () {
          * @param {IStorageData} item - The item to upsert.
          * @returns {Promise<void>}
          */
-        this.upsert = function (item, clientId, storageName) { return __awaiter(_this, void 0, void 0, function () {
+        this.upsert = function (item, requestId, clientId, storageName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -5443,6 +5421,7 @@ var StoragePublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 storageName: storageName,
                                 agentName: "",
@@ -5458,7 +5437,7 @@ var StoragePublicService = /** @class */ (function () {
          * @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 () {
+        this.remove = function (itemId, requestId, clientId, storageName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -5476,6 +5455,7 @@ var StoragePublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 storageName: storageName,
                                 agentName: "",
@@ -5491,12 +5471,13 @@ var StoragePublicService = /** @class */ (function () {
          * @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 () {
+        this.get = function (itemId, requestId, 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", {
+                            requestId: requestId,
                             itemId: itemId,
                             clientId: clientId,
                             storageName: storageName,
@@ -5509,6 +5490,7 @@ var StoragePublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 storageName: storageName,
                                 agentName: "",
@@ -5524,12 +5506,13 @@ var StoragePublicService = /** @class */ (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 () {
+        this.list = function (requestId, 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", {
+                            requestId: requestId,
                             clientId: clientId,
                             storageName: storageName,
                         });
@@ -5541,6 +5524,7 @@ var StoragePublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 storageName: storageName,
                                 agentName: "",
@@ -5555,12 +5539,13 @@ var StoragePublicService = /** @class */ (function () {
          * Clears all items from the storage.
          * @returns {Promise<void>}
          */
-        this.clear = function (clientId, storageName) { return __awaiter(_this, void 0, void 0, function () {
+        this.clear = function (requestId, 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", {
+                            requestId: requestId,
                             clientId: clientId,
                             storageName: storageName,
                         });
@@ -5572,6 +5557,7 @@ var StoragePublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 storageName: storageName,
                                 agentName: "",
@@ -5588,7 +5574,7 @@ var StoragePublicService = /** @class */ (function () {
          * @param {StorageName} storageName - The storage name.
          * @returns {Promise<void>}
          */
-        this.dispose = function (clientId, storageName) { return __awaiter(_this, void 0, void 0, function () {
+        this.dispose = function (requestId, clientId, storageName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -5605,6 +5591,7 @@ var StoragePublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 storageName: storageName,
                                 agentName: "",
@@ -6033,9 +6020,7 @@ var StateConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("stateConnectionService setState", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("stateConnectionService setState");
                         state = this.getStateRef(this.contextService.context.clientId, this.contextService.context.stateName);
                         return [4 /*yield*/, state.waitForInit()];
                     case 1:
@@ -6054,9 +6039,7 @@ var StateConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("stateConnectionService getState", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("stateConnectionService getState");
                         state = this.getStateRef(this.contextService.context.clientId, this.contextService.context.stateName);
                         return [4 /*yield*/, state.waitForInit()];
                     case 1:
@@ -6075,9 +6058,7 @@ var StateConnectionService = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.loggerService.log("stateConnectionService dispose", {
-                            context: this.contextService.context,
-                        });
+                        this.loggerService.log("stateConnectionService dispose");
                         key = "".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.stateName);
                         if (!this.getStateRef.has(key)) {
                             return [2 /*return*/];
@@ -6114,12 +6095,13 @@ var StatePublicService = /** @class */ (function () {
          * @param {StateName} stateName - The name of the state.
          * @returns {Promise<T>} - The updated state.
          */
-        this.setState = function (dispatchFn, clientId, stateName) { return __awaiter(_this, void 0, void 0, function () {
+        this.setState = function (dispatchFn, requestId, clientId, stateName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("statePublicService setState", {
+                            requestId: requestId,
                             clientId: clientId,
                             stateName: stateName,
                         });
@@ -6131,6 +6113,7 @@ var StatePublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 stateName: stateName,
                                 agentName: "",
@@ -6147,7 +6130,7 @@ var StatePublicService = /** @class */ (function () {
          * @param {StateName} stateName - The name of the state.
          * @returns {Promise<T>} - The current state.
          */
-        this.getState = function (clientId, stateName) { return __awaiter(_this, void 0, void 0, function () {
+        this.getState = function (requestId, clientId, stateName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -6164,6 +6147,7 @@ var StatePublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 stateName: stateName,
                                 agentName: "",
@@ -6180,12 +6164,13 @@ var StatePublicService = /** @class */ (function () {
          * @param {StateName} stateName - The name of the state.
          * @returns {Promise<void>} - A promise that resolves when the state is disposed.
          */
-        this.dispose = function (clientId, stateName) { return __awaiter(_this, void 0, void 0, function () {
+        this.dispose = function (requestId, clientId, stateName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
                         this.loggerService.log("statePublicService dispose", {
+                            requestId: requestId,
                             clientId: clientId,
                             stateName: stateName,
                         });
@@ -6197,6 +6182,7 @@ var StatePublicService = /** @class */ (function () {
                                     }
                                 });
                             }); }, {
+                                requestId: requestId,
                                 clientId: clientId,
                                 stateName: stateName,
                                 agentName: "",
@@ -6237,7 +6223,7 @@ var BusService = /** @class */ (function () {
                 _this._eventWildcardMap.set(source, true);
             }
             _this._eventSourceSet.add(source);
-            _this.getEventSubject(clientId, source).subscribe(fn);
+            return _this.getEventSubject(clientId, source).subscribe(fn);
         };
         /**
          * Subscribes to a single event for a specific client and source.
@@ -6534,20 +6520,22 @@ var addStorage = function (storageSchema) {
  * @returns {Promise<void>} - A promise that resolves when the message is committed.
  */
 var commitUserMessage = function (content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName, currentAgentName;
+    var requestId, swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log('function commitSystemMessage', {
                     content: content,
                     clientId: clientId,
                     agentName: agentName,
+                    requestId: requestId,
                 });
                 swarm.agentValidationService.validate(agentName, "commitUserMessage");
                 swarm.sessionValidationService.validate(clientId, "commitUserMessage");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "commitUserMessage");
-                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(clientId, swarmName)];
+                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(requestId, clientId, swarmName)];
             case 1:
                 currentAgentName = _a.sent();
                 if (currentAgentName !== agentName) {
@@ -6558,7 +6546,7 @@ var commitUserMessage = function (content, clientId, agentName) { return __await
                     });
                     return [2 /*return*/];
                 }
-                return [4 /*yield*/, swarm.sessionPublicService.commitUserMessage(content, clientId, swarmName)];
+                return [4 /*yield*/, swarm.sessionPublicService.commitUserMessage(content, requestId, clientId, swarmName)];
             case 2:
                 _a.sent();
                 return [2 /*return*/];
@@ -6574,17 +6562,19 @@ var commitUserMessage = function (content, clientId, agentName) { return __await
  * @throws Will throw an error if the client ID is invalid or if the swarm validation fails.
  */
 var getAgentName = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName;
+    var requestId, swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log("function getAgentName", {
                     clientId: clientId,
+                    requestId: requestId,
                 });
                 swarm.sessionValidationService.validate(clientId, "getAgentName");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "getAgentName");
-                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(clientId, swarmName)];
+                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(requestId, clientId, swarmName)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     });
@@ -6600,13 +6590,15 @@ var SCHEDULED_DELAY$1 = 1000;
  * @returns {SendMessageFn} - A function to send messages to the swarm.
  */
 var makeConnection = function (connector, clientId, swarmName) {
+    var requestId = randomString();
     swarm.loggerService.log("function makeConnection", {
         clientId: clientId,
         swarmName: swarmName,
+        requestId: requestId,
     });
     swarm.swarmValidationService.validate(swarmName, "makeConnection");
     swarm.sessionValidationService.addSession(clientId, swarmName, "makeConnection");
-    var send = swarm.sessionPublicService.connect(connector, clientId, swarmName);
+    var send = swarm.sessionPublicService.connect(connector, requestId, clientId, swarmName);
     return queued(function (outgoing) { return __awaiter(void 0, void 0, void 0, function () {
         var _a;
         var _b;
@@ -6618,7 +6610,7 @@ var makeConnection = function (connector, clientId, swarmName) {
                     _b = {
                         data: outgoing
                     };
-                    return [4 /*yield*/, swarm.swarmPublicService.getAgentName(clientId, swarmName)];
+                    return [4 /*yield*/, swarm.swarmPublicService.getAgentName(requestId, clientId, swarmName)];
                 case 1: return [4 /*yield*/, _a.apply(void 0, [(_b.agentName = _c.sent(),
                             _b.clientId = clientId,
                             _b)])];
@@ -6721,7 +6713,7 @@ var CHANGE_AGENT_GC = 60 * 1000;
  * @returns {TChangeAgentRun} - The change agent function.
  */
 var createChangeAgent = ttl(function (clientId) {
-    return queued(function (agentName) { return __awaiter(void 0, void 0, void 0, function () {
+    return queued(function (requestId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
         var swarmName, _a, _b, _c;
         return __generator(this, function (_d) {
             switch (_d.label) {
@@ -6734,7 +6726,7 @@ var createChangeAgent = ttl(function (clientId) {
                             .map(function (agentName) { return __awaiter(void 0, void 0, void 0, function () {
                             return __generator(this, function (_a) {
                                 switch (_a.label) {
-                                    case 0: return [4 /*yield*/, swarm.agentPublicService.commitAgentChange(clientId, agentName)];
+                                    case 0: return [4 /*yield*/, swarm.agentPublicService.commitAgentChange(requestId, clientId, agentName)];
                                     case 1:
                                         _a.sent();
                                         return [2 /*return*/];
@@ -6743,21 +6735,22 @@ var createChangeAgent = ttl(function (clientId) {
                         }); }))];
                 case 1:
                     _d.sent();
-                    return [4 /*yield*/, swarm.agentPublicService.dispose(clientId, agentName)];
+                    return [4 /*yield*/, swarm.agentPublicService.dispose(requestId, clientId, agentName)];
                 case 2:
                     _d.sent();
-                    return [4 /*yield*/, swarm.historyPublicService.dispose(clientId, agentName)];
+                    return [4 /*yield*/, swarm.historyPublicService.dispose(requestId, clientId, agentName)];
                 case 3:
                     _d.sent();
                     _b = (_a = swarm.swarmPublicService).setAgentRef;
-                    _c = [clientId,
+                    _c = [requestId,
+                        clientId,
                         swarmName,
                         agentName];
-                    return [4 /*yield*/, swarm.agentPublicService.createAgentRef(clientId, agentName)];
+                    return [4 /*yield*/, swarm.agentPublicService.createAgentRef(requestId, clientId, agentName)];
                 case 4: return [4 /*yield*/, _b.apply(_a, _c.concat([_d.sent()]))];
                 case 5:
                     _d.sent();
-                    return [4 /*yield*/, swarm.swarmPublicService.setAgentName(agentName, clientId, swarmName)];
+                    return [4 /*yield*/, swarm.swarmPublicService.setAgentName(agentName, requestId, clientId, swarmName)];
                 case 6:
                     _d.sent();
                     return [2 /*return*/];
@@ -6791,10 +6784,11 @@ var createGc$1 = singleshot(function () { return __awaiter(void 0, void 0, void
  * @returns {Promise<void>} - A promise that resolves when the agent is changed.
  */
 var changeAgent = function (agentName, clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var run;
+    var requestId, run;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log("function changeAgent", {
                     agentName: agentName,
                     clientId: clientId,
@@ -6803,7 +6797,7 @@ var changeAgent = function (agentName, clientId) { return __awaiter(void 0, void
             case 1:
                 run = _a.sent();
                 createGc$1();
-                return [4 /*yield*/, run(agentName)];
+                return [4 /*yield*/, run(requestId, agentName)];
             case 2: return [2 /*return*/, _a.sent()];
         }
     });
@@ -6816,83 +6810,91 @@ var changeAgent = function (agentName, clientId) { return __awaiter(void 0, void
  * @param {SwarmName} swarmName - The name of the swarm.
  * @returns {Promise<void>} A promise that resolves when the connection is disposed.
  */
-var disposeConnection = function (clientId, swarmName) { return __awaiter(void 0, void 0, void 0, function () {
-    return __generator(this, function (_a) {
-        switch (_a.label) {
-            case 0:
-                swarm.loggerService.log("function disposeConnection", {
-                    clientId: clientId,
-                    swarmName: swarmName,
-                });
-                swarm.swarmValidationService.validate(swarmName, "disposeConnection");
-                swarm.sessionValidationService.removeSession(clientId);
-                swarm.busService.dispose(clientId);
-                return [4 /*yield*/, swarm.sessionPublicService.dispose(clientId, swarmName)];
-            case 1:
-                _a.sent();
-                return [4 /*yield*/, swarm.swarmPublicService.dispose(clientId, swarmName)];
-            case 2:
-                _a.sent();
-                return [4 /*yield*/, Promise.all(swarm.swarmValidationService
-                        .getAgentList(swarmName)
-                        .map(function (agentName) { return __awaiter(void 0, void 0, void 0, function () {
-                        return __generator(this, function (_a) {
-                            switch (_a.label) {
-                                case 0: return [4 /*yield*/, swarm.agentPublicService.dispose(clientId, agentName)];
-                                case 1:
-                                    _a.sent();
-                                    return [4 /*yield*/, swarm.historyPublicService.dispose(clientId, agentName)];
-                                case 2:
-                                    _a.sent();
-                                    return [2 /*return*/];
-                            }
-                        });
-                    }); }))];
-            case 3:
-                _a.sent();
-                return [4 /*yield*/, Promise.all(swarm.swarmValidationService
-                        .getAgentList(swarmName)
-                        .flatMap(function (agentName) {
-                        return swarm.agentValidationService.getStorageList(agentName);
-                    })
-                        .filter(function (storageName) { return !!storageName; })
-                        .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 [4 /*yield*/, Promise.all(swarm.swarmValidationService
-                        .getAgentList(swarmName)
-                        .flatMap(function (agentName) {
-                        return swarm.agentValidationService.getStateList(agentName);
-                    })
-                        .filter(function (stateName) { return !!stateName; })
-                        .map(function (stateName) { return __awaiter(void 0, void 0, void 0, function () {
-                        return __generator(this, function (_a) {
-                            switch (_a.label) {
-                                case 0: return [4 /*yield*/, swarm.statePublicService.dispose(clientId, stateName)];
-                                case 1:
-                                    _a.sent();
-                                    return [2 /*return*/];
-                            }
-                        });
-                    }); }))];
-            case 5:
-                _a.sent();
-                return [4 /*yield*/, HistoryAdapter.dispose(clientId, null)];
-            case 6:
-                _a.sent();
-                return [2 /*return*/];
-        }
+var disposeConnection = function (clientId_1, swarmName_1) {
+    var args_1 = [];
+    for (var _i = 2; _i < arguments.length; _i++) {
+        args_1[_i - 2] = arguments[_i];
+    }
+    return __awaiter(void 0, __spreadArray([clientId_1, swarmName_1], __read(args_1), false), void 0, function (clientId, swarmName, requestId) {
+        if (requestId === void 0) { requestId = randomString(); }
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    swarm.loggerService.log("function disposeConnection", {
+                        clientId: clientId,
+                        swarmName: swarmName,
+                        requestId: requestId,
+                    });
+                    swarm.swarmValidationService.validate(swarmName, "disposeConnection");
+                    swarm.sessionValidationService.removeSession(clientId);
+                    swarm.busService.dispose(clientId);
+                    return [4 /*yield*/, swarm.sessionPublicService.dispose(requestId, clientId, swarmName)];
+                case 1:
+                    _a.sent();
+                    return [4 /*yield*/, swarm.swarmPublicService.dispose(requestId, clientId, swarmName)];
+                case 2:
+                    _a.sent();
+                    return [4 /*yield*/, Promise.all(swarm.swarmValidationService
+                            .getAgentList(swarmName)
+                            .map(function (agentName) { return __awaiter(void 0, void 0, void 0, function () {
+                            return __generator(this, function (_a) {
+                                switch (_a.label) {
+                                    case 0: return [4 /*yield*/, swarm.agentPublicService.dispose(requestId, clientId, agentName)];
+                                    case 1:
+                                        _a.sent();
+                                        return [4 /*yield*/, swarm.historyPublicService.dispose(requestId, clientId, agentName)];
+                                    case 2:
+                                        _a.sent();
+                                        return [2 /*return*/];
+                                }
+                            });
+                        }); }))];
+                case 3:
+                    _a.sent();
+                    return [4 /*yield*/, Promise.all(swarm.swarmValidationService
+                            .getAgentList(swarmName)
+                            .flatMap(function (agentName) {
+                            return swarm.agentValidationService.getStorageList(agentName);
+                        })
+                            .filter(function (storageName) { return !!storageName; })
+                            .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(requestId, clientId, storageName)];
+                                    case 1:
+                                        _a.sent();
+                                        return [2 /*return*/];
+                                }
+                            });
+                        }); }))];
+                case 4:
+                    _a.sent();
+                    return [4 /*yield*/, Promise.all(swarm.swarmValidationService
+                            .getAgentList(swarmName)
+                            .flatMap(function (agentName) {
+                            return swarm.agentValidationService.getStateList(agentName);
+                        })
+                            .filter(function (stateName) { return !!stateName; })
+                            .map(function (stateName) { return __awaiter(void 0, void 0, void 0, function () {
+                            return __generator(this, function (_a) {
+                                switch (_a.label) {
+                                    case 0: return [4 /*yield*/, swarm.statePublicService.dispose(requestId, clientId, stateName)];
+                                    case 1:
+                                        _a.sent();
+                                        return [2 /*return*/];
+                                }
+                            });
+                        }); }))];
+                case 5:
+                    _a.sent();
+                    return [4 /*yield*/, HistoryAdapter.dispose(clientId, null)];
+                case 6:
+                    _a.sent();
+                    return [2 /*return*/];
+            }
+        });
     });
-}); };
+};
 
 var COMPLETE_TTL = 15 * 60 * 1000;
 var COMPLETE_GC = 60 * 1000;
@@ -6903,17 +6905,17 @@ var COMPLETE_GC = 60 * 1000;
  * @returns {TCompleteRun} The complete run function.
  */
 var createComplete = ttl(function (clientId, swarmName) {
-    return queued(function (content) { return __awaiter(void 0, void 0, void 0, function () {
+    return queued(function (requestId, content) { return __awaiter(void 0, void 0, void 0, function () {
         var result;
         return __generator(this, function (_a) {
             switch (_a.label) {
                 case 0:
                     swarm.swarmValidationService.validate(swarmName, "complete");
                     swarm.sessionValidationService.addSession(clientId, swarmName, "complete");
-                    return [4 /*yield*/, swarm.sessionPublicService.execute(content, "user", clientId, swarmName)];
+                    return [4 /*yield*/, swarm.sessionPublicService.execute(content, "user", requestId, clientId, swarmName)];
                 case 1:
                     result = _a.sent();
-                    return [4 /*yield*/, disposeConnection(clientId, swarmName)];
+                    return [4 /*yield*/, disposeConnection(clientId, swarmName, requestId)];
                 case 2:
                     _a.sent();
                     return [2 /*return*/, result];
@@ -6947,20 +6949,22 @@ var createGc = singleshot(function () { return __awaiter(void 0, void 0, void 0,
  * @returns {Promise<string>} The result of the complete function.
  */
 var complete = function (content, clientId, swarmName) { return __awaiter(void 0, void 0, void 0, function () {
-    var run;
+    var requestId, run;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log("function complete", {
                     content: content,
                     clientId: clientId,
                     swarmName: swarmName,
+                    requestId: requestId,
                 });
                 return [4 /*yield*/, createComplete(clientId, swarmName)];
             case 1:
                 run = _a.sent();
                 createGc();
-                return [4 /*yield*/, run(content)];
+                return [4 /*yield*/, run(requestId, content)];
             case 2: return [2 /*return*/, _a.sent()];
         }
     });
@@ -6977,9 +6981,11 @@ var SCHEDULED_DELAY = 1000;
  * @returns {Function} dispose - A function to dispose of the session.
  */
 var session = function (clientId, swarmName) {
+    var requestId = randomString();
     swarm.loggerService.log("function session", {
         clientId: clientId,
         swarmName: swarmName,
+        requestId: requestId,
     });
     swarm.swarmValidationService.validate(swarmName, "session");
     swarm.sessionValidationService.addSession(clientId, swarmName, "session");
@@ -6995,7 +7001,7 @@ var session = function (clientId, swarmName) {
                 switch (_a.label) {
                     case 0:
                         swarm.sessionValidationService.validate(clientId, "session");
-                        return [4 /*yield*/, swarm.sessionPublicService.execute(content, "user", clientId, swarmName)];
+                        return [4 /*yield*/, swarm.sessionPublicService.execute(content, "user", requestId, clientId, swarmName)];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
@@ -7008,7 +7014,7 @@ var session = function (clientId, swarmName) {
         dispose: function () { return __awaiter(void 0, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
-                    case 0: return [4 /*yield*/, disposeConnection(clientId, swarmName)];
+                    case 0: return [4 /*yield*/, disposeConnection(clientId, swarmName, requestId)];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
@@ -7113,27 +7119,35 @@ session.scheduled = function (clientId, swarmName, _a) {
  * @param {string} clientId - The ID of the client whose history is to be retrieved.
  * @returns {Promise<Array>} A promise that resolves to an array containing the raw history.
  */
-var getRawHistory = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName, agentName, history;
-    return __generator(this, function (_a) {
-        switch (_a.label) {
-            case 0:
-                swarm.loggerService.log("function getRawHistory", {
-                    clientId: clientId,
-                });
-                swarm.sessionValidationService.validate(clientId, "getRawHistory");
-                swarmName = swarm.sessionValidationService.getSwarm(clientId);
-                swarm.swarmValidationService.validate(swarmName, "getRawHistory");
-                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(clientId, swarmName)];
-            case 1:
-                agentName = _a.sent();
-                return [4 /*yield*/, swarm.historyPublicService.toArrayForRaw(clientId, agentName)];
-            case 2:
-                history = _a.sent();
-                return [2 /*return*/, __spreadArray([], __read(history), false)];
-        }
+var getRawHistory = function (clientId_1) {
+    var args_1 = [];
+    for (var _i = 1; _i < arguments.length; _i++) {
+        args_1[_i - 1] = arguments[_i];
+    }
+    return __awaiter(void 0, __spreadArray([clientId_1], __read(args_1), false), void 0, function (clientId, requestId) {
+        var swarmName, agentName, history;
+        if (requestId === void 0) { requestId = randomString(); }
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    swarm.loggerService.log("function getRawHistory", {
+                        clientId: clientId,
+                        requestId: requestId,
+                    });
+                    swarm.sessionValidationService.validate(clientId, "getRawHistory");
+                    swarmName = swarm.sessionValidationService.getSwarm(clientId);
+                    swarm.swarmValidationService.validate(swarmName, "getRawHistory");
+                    return [4 /*yield*/, swarm.swarmPublicService.getAgentName(requestId, clientId, swarmName)];
+                case 1:
+                    agentName = _a.sent();
+                    return [4 /*yield*/, swarm.historyPublicService.toArrayForRaw(requestId, clientId, agentName)];
+                case 2:
+                    history = _a.sent();
+                    return [2 /*return*/, __spreadArray([], __read(history), false)];
+            }
+        });
     });
-}); };
+};
 
 /**
  * Retrieves the history prepared for a specific agent with resque algorithm tweaks
@@ -7143,17 +7157,19 @@ var getRawHistory = function (clientId) { return __awaiter(void 0, void 0, void
  * @returns {Promise<Array>} - A promise that resolves to an array containing the agent's history.
  */
 var getAgentHistory = function (clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
-    var prompt, history;
+    var requestId, prompt, history;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log("function getAgentHistory", {
                     clientId: clientId,
                     agentName: agentName,
+                    requestId: requestId,
                 });
                 swarm.agentValidationService.validate(agentName, "getAgentHistory");
                 prompt = swarm.agentSchemaService.get(agentName).prompt;
-                return [4 /*yield*/, swarm.historyPublicService.toArrayForAgent(prompt, clientId, agentName)];
+                return [4 /*yield*/, swarm.historyPublicService.toArrayForAgent(prompt, requestId, clientId, agentName)];
             case 1:
                 history = _a.sent();
                 return [2 /*return*/, __spreadArray([], __read(history), false)];
@@ -7187,21 +7203,23 @@ var getSessionMode = function (clientId) { return __awaiter(void 0, void 0, void
  * @returns {Promise<void>} - A promise that resolves when the operation is complete.
  */
 var commitToolOutput = function (toolId, content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName, currentAgentName;
+    var requestId, swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log('function commitToolOutput', {
                     toolId: toolId,
                     content: content,
                     clientId: clientId,
                     agentName: agentName,
+                    requestId: requestId,
                 });
                 swarm.agentValidationService.validate(agentName, "commitSystemMessage");
                 swarm.sessionValidationService.validate(clientId, "commitToolOutput");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "commitToolOutput");
-                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(clientId, swarmName)];
+                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(requestId, clientId, swarmName)];
             case 1:
                 currentAgentName = _a.sent();
                 if (currentAgentName !== agentName) {
@@ -7213,7 +7231,7 @@ var commitToolOutput = function (toolId, content, clientId, agentName) { return
                     });
                     return [2 /*return*/];
                 }
-                return [4 /*yield*/, swarm.sessionPublicService.commitToolOutput(toolId, content, clientId, swarmName)];
+                return [4 /*yield*/, swarm.sessionPublicService.commitToolOutput(toolId, content, requestId, clientId, swarmName)];
             case 2:
                 _a.sent();
                 return [2 /*return*/];
@@ -7230,20 +7248,22 @@ var commitToolOutput = function (toolId, content, clientId, agentName) { return
  * @returns {Promise<void>} - A promise that resolves when the message is committed.
  */
 var commitSystemMessage = function (content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName, currentAgentName;
+    var requestId, swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log('function commitSystemMessage', {
                     content: content,
                     clientId: clientId,
                     agentName: agentName,
+                    requestId: requestId,
                 });
                 swarm.agentValidationService.validate(agentName, "commitSystemMessage");
                 swarm.sessionValidationService.validate(clientId, "commitSystemMessage");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "commitSystemMessage");
-                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(clientId, swarmName)];
+                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(requestId, clientId, swarmName)];
             case 1:
                 currentAgentName = _a.sent();
                 if (currentAgentName !== agentName) {
@@ -7254,7 +7274,7 @@ var commitSystemMessage = function (content, clientId, agentName) { return __awa
                     });
                     return [2 /*return*/];
                 }
-                return [4 /*yield*/, swarm.sessionPublicService.commitSystemMessage(content, clientId, swarmName)];
+                return [4 /*yield*/, swarm.sessionPublicService.commitSystemMessage(content, requestId, clientId, swarmName)];
             case 2:
                 _a.sent();
                 return [2 /*return*/];
@@ -7270,19 +7290,21 @@ var commitSystemMessage = function (content, clientId, agentName) { return __awa
  * @returns {Promise<void>} - A promise that resolves when the message is committed.
  */
 var commitFlush = function (clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName, currentAgentName;
+    var requestId, swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log('function commitFlush', {
                     clientId: clientId,
                     agentName: agentName,
+                    requestId: requestId,
                 });
                 swarm.agentValidationService.validate(agentName, "commitFlush");
                 swarm.sessionValidationService.validate(clientId, "commitFlush");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "commitFlush");
-                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(clientId, swarmName)];
+                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(requestId, clientId, swarmName)];
             case 1:
                 currentAgentName = _a.sent();
                 if (currentAgentName !== agentName) {
@@ -7293,7 +7315,7 @@ var commitFlush = function (clientId, agentName) { return __awaiter(void 0, void
                     });
                     return [2 /*return*/];
                 }
-                return [4 /*yield*/, swarm.sessionPublicService.commitFlush(clientId, swarmName)];
+                return [4 /*yield*/, swarm.sessionPublicService.commitFlush(requestId, clientId, swarmName)];
             case 2:
                 _a.sent();
                 return [2 /*return*/];
@@ -7311,20 +7333,22 @@ var commitFlush = function (clientId, agentName) { return __awaiter(void 0, void
  * @returns {Promise<void>} - A promise that resolves when the execution is complete.
  */
 var execute = function (content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName, currentAgentName;
+    var requestId, swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log("function execute", {
                     content: content,
                     clientId: clientId,
                     agentName: agentName,
+                    requestId: requestId,
                 });
                 swarm.agentValidationService.validate(agentName, "execute");
                 swarm.sessionValidationService.validate(clientId, "execute");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "execute");
-                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(clientId, swarmName)];
+                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(requestId, clientId, swarmName)];
             case 1:
                 currentAgentName = _a.sent();
                 if (currentAgentName !== agentName) {
@@ -7335,7 +7359,7 @@ var execute = function (content, clientId, agentName) { return __awaiter(void 0,
                     });
                     return [2 /*return*/];
                 }
-                return [4 /*yield*/, swarm.sessionPublicService.execute(content, "tool", clientId, swarmName)];
+                return [4 /*yield*/, swarm.sessionPublicService.execute(content, "tool", requestId, clientId, swarmName)];
             case 2: return [2 /*return*/, _a.sent()];
         }
     });
@@ -7352,14 +7376,16 @@ var execute = function (content, clientId, agentName) { return __awaiter(void 0,
  * @returns {Promise<void>} A promise that resolves when the content is emitted.
  */
 var emit = function (content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName, currentAgentName;
+    var requestId, swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log("function emit", {
                     content: content,
                     clientId: clientId,
                     agentName: agentName,
+                    requestId: requestId,
                 });
                 if (swarm.sessionValidationService.getSessionMode(clientId) !== "makeConnection") {
                     throw new Error("agent-swarm-kit emit session is not makeConnection clientId=".concat(clientId));
@@ -7368,7 +7394,7 @@ var emit = function (content, clientId, agentName) { return __awaiter(void 0, vo
                 swarm.sessionValidationService.validate(clientId, "emit");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "emit");
-                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(clientId, swarmName)];
+                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(requestId, clientId, swarmName)];
             case 1:
                 currentAgentName = _a.sent();
                 if (currentAgentName !== agentName) {
@@ -7379,7 +7405,7 @@ var emit = function (content, clientId, agentName) { return __awaiter(void 0, vo
                     });
                     return [2 /*return*/];
                 }
-                return [4 /*yield*/, swarm.sessionPublicService.emit(content, clientId, swarmName)];
+                return [4 /*yield*/, swarm.sessionPublicService.emit(content, requestId, clientId, swarmName)];
             case 2: return [2 /*return*/, _a.sent()];
         }
     });
@@ -7393,19 +7419,21 @@ var emit = function (content, clientId, agentName) { return __awaiter(void 0, vo
  * @returns {Promise<void>} - A promise that resolves when the operation is complete.
  */
 var commitToolOutputForce = function (toolId, content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName;
+    var requestId, swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log('function commitToolOutputForce', {
                     toolId: toolId,
                     content: content,
                     clientId: clientId,
+                    requestId: requestId,
                 });
                 swarm.sessionValidationService.validate(clientId, "commitToolOutputForce");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "commitToolOutputForce");
-                return [4 /*yield*/, swarm.sessionPublicService.commitToolOutput(toolId, content, clientId, swarmName)];
+                return [4 /*yield*/, swarm.sessionPublicService.commitToolOutput(toolId, content, requestId, clientId, swarmName)];
             case 1:
                 _a.sent();
                 return [2 /*return*/];
@@ -7421,18 +7449,20 @@ var commitToolOutputForce = function (toolId, content, clientId) { return __awai
  * @returns {Promise<void>} - A promise that resolves when the message is committed.
  */
 var commitSystemMessageForce = function (content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName;
+    var requestId, swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log('function commitSystemMessageForce', {
                     content: content,
                     clientId: clientId,
+                    requestId: requestId,
                 });
                 swarm.sessionValidationService.validate(clientId, "commitSystemMessageForce");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "commitSystemMessageForce");
-                return [4 /*yield*/, swarm.sessionPublicService.commitSystemMessage(content, clientId, swarmName)];
+                return [4 /*yield*/, swarm.sessionPublicService.commitSystemMessage(content, requestId, clientId, swarmName)];
             case 1:
                 _a.sent();
                 return [2 /*return*/];
@@ -7447,17 +7477,19 @@ var commitSystemMessageForce = function (content, clientId) { return __awaiter(v
  * @returns {Promise<void>} - A promise that resolves when the message is committed.
  */
 var commitFlushForce = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName;
+    var requestId, swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log('function commitFlushForce', {
                     clientId: clientId,
+                    requestId: requestId,
                 });
                 swarm.sessionValidationService.validate(clientId, "commitFlushForce");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "commitFlushForce");
-                return [4 /*yield*/, swarm.sessionPublicService.commitFlush(clientId, swarmName)];
+                return [4 /*yield*/, swarm.sessionPublicService.commitFlush(requestId, clientId, swarmName)];
             case 1:
                 _a.sent();
                 return [2 /*return*/];
@@ -7473,18 +7505,20 @@ var commitFlushForce = function (clientId) { return __awaiter(void 0, void 0, vo
  * @returns {Promise<void>} - A promise that resolves when the message is committed.
  */
 var commitUserMessageForce = function (content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName;
+    var requestId, swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log('function commitSystemMessage', {
                     content: content,
                     clientId: clientId,
+                    requestId: requestId,
                 });
                 swarm.sessionValidationService.validate(clientId, "commitUserMessageForce");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "commitUserMessageForce");
-                return [4 /*yield*/, swarm.sessionPublicService.commitUserMessage(content, clientId, swarmName)];
+                return [4 /*yield*/, swarm.sessionPublicService.commitUserMessage(content, requestId, clientId, swarmName)];
             case 1:
                 _a.sent();
                 return [2 /*return*/];
@@ -7503,13 +7537,15 @@ var commitUserMessageForce = function (content, clientId) { return __awaiter(voi
  * @returns {Promise<void>} A promise that resolves when the content is emitted.
  */
 var emitForce = function (content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName;
+    var requestId, swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log("function emitForce", {
                     content: content,
                     clientId: clientId,
+                    requestId: requestId,
                 });
                 if (swarm.sessionValidationService.getSessionMode(clientId) !== "makeConnection") {
                     throw new Error("agent-swarm-kit emitForce session is not makeConnection clientId=".concat(clientId));
@@ -7517,7 +7553,7 @@ var emitForce = function (content, clientId) { return __awaiter(void 0, void 0,
                 swarm.sessionValidationService.validate(clientId, "emitForce");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "emitForce");
-                return [4 /*yield*/, swarm.sessionPublicService.emit(content, clientId, swarmName)];
+                return [4 /*yield*/, swarm.sessionPublicService.emit(content, requestId, clientId, swarmName)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     });
@@ -7534,24 +7570,26 @@ var emitForce = function (content, clientId) { return __awaiter(void 0, void 0,
  * @returns {Promise<void>} - A promise that resolves when the execution is complete.
  */
 var executeForce = function (content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName;
+    var requestId, swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log("function executeForce", {
                     content: content,
                     clientId: clientId,
+                    requestId: requestId,
                 });
                 swarm.sessionValidationService.validate(clientId, "executeForce");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "executeForce");
-                return [4 /*yield*/, swarm.sessionPublicService.execute(content, "tool", clientId, swarmName)];
+                return [4 /*yield*/, swarm.sessionPublicService.execute(content, "tool", requestId, clientId, swarmName)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     });
 }); };
 
-var DISALLOWED_EVENT_SOURCE_LIST$1 = new Set([
+var DISALLOWED_EVENT_SOURCE_LIST$2 = new Set([
     "agent-bus",
     "history-bus",
     "session-bus",
@@ -7559,7 +7597,7 @@ var DISALLOWED_EVENT_SOURCE_LIST$1 = new Set([
     "storage-bus",
     "swarm-bus",
 ]);
-var validateClientId$6 = function (clientId) {
+var validateClientId$d = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -7578,16 +7616,56 @@ var listenEvent = function (clientId, topicName, fn) {
     swarm.loggerService.log("function listenEvent", {
         clientId: clientId,
     });
-    if (DISALLOWED_EVENT_SOURCE_LIST$1.has(topicName)) {
+    if (DISALLOWED_EVENT_SOURCE_LIST$2.has(topicName)) {
         throw new Error("agent-swarm listenEvent topic is reserved topicName=".concat(topicName));
     }
-    validateClientId$6(clientId);
+    validateClientId$d(clientId);
     return swarm.busService.subscribe(clientId, topicName, function (_a) {
         var payload = _a.payload;
         return fn(payload);
     });
 };
 
+var DISALLOWED_EVENT_SOURCE_LIST$1 = new Set([
+    "agent-bus",
+    "history-bus",
+    "session-bus",
+    "state-bus",
+    "storage-bus",
+    "swarm-bus",
+]);
+var validateClientId$c = function (clientId) {
+    if (clientId === "*") {
+        return;
+    }
+    if (!swarm.sessionValidationService.hasSession(clientId)) {
+        throw new Error("agent-swarm listenEventOnce session not found for clientId=".concat(clientId));
+    }
+};
+/**
+ * Listens for an event on the swarm bus service and executes a callback function when the event is received.
+ *
+ * @template T - The type of the data payload.
+ * @param {string} clientId - The ID of the client to listen for events from.
+ * @param {(data: T) => void} fn - The callback function to execute when the event is received. The data payload is passed as an argument to this function.
+ */
+var listenEventOnce = function (clientId, topicName, filterFn, fn) {
+    swarm.loggerService.log("function listenEventOnce", {
+        clientId: clientId,
+    });
+    if (DISALLOWED_EVENT_SOURCE_LIST$1.has(topicName)) {
+        throw new Error("agent-swarm listenEventOnce topic is reserved topicName=".concat(topicName));
+    }
+    validateClientId$c(clientId);
+    return swarm.busService.once(clientId, topicName, function (_a) {
+        var payload = _a.payload;
+        return filterFn(payload);
+    }, function (_a) {
+        var payload = _a.payload;
+        return fn(payload);
+    });
+};
+
 /**
  * Retrieves the last message sent by the user from the client's message history.
  *
@@ -7595,14 +7673,16 @@ var listenEvent = function (clientId, topicName, fn) {
  * @returns {Promise<string | null>} - The content of the last user message, or null if no user message is found.
  */
 var getLastUserMessage = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var history, last;
+    var requestId, history, last;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log("function getLastUserMessage", {
                     clientId: clientId,
+                    requestId: requestId,
                 });
-                return [4 /*yield*/, getRawHistory(clientId)];
+                return [4 /*yield*/, getRawHistory(clientId, requestId)];
             case 1:
                 history = _a.sent();
                 last = history.findLast(function (_a) {
@@ -7621,14 +7701,16 @@ var getLastUserMessage = function (clientId) { return __awaiter(void 0, void 0,
  * @returns {Promise<Array>} A promise that resolves to an array of history objects filtered by user role.
  */
 var getUserHistory = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var history;
+    var requestId, history;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log("function getUserHistory", {
                     clientId: clientId,
+                    requestId: requestId,
                 });
-                return [4 /*yield*/, getRawHistory(clientId)];
+                return [4 /*yield*/, getRawHistory(clientId, requestId)];
             case 1:
                 history = _a.sent();
                 return [2 /*return*/, history.filter(function (_a) {
@@ -7646,14 +7728,16 @@ var getUserHistory = function (clientId) { return __awaiter(void 0, void 0, void
  * @returns {Promise<Array>} - A promise that resolves to an array of history objects where the role is "assistant".
  */
 var getAssistantHistory = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var history;
+    var requestId, history;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log("function getAssistantHistory", {
                     clientId: clientId,
+                    requestId: requestId,
                 });
-                return [4 /*yield*/, getRawHistory(clientId)];
+                return [4 /*yield*/, getRawHistory(clientId, requestId)];
             case 1:
                 history = _a.sent();
                 return [2 /*return*/, history.filter(function (_a) {
@@ -7671,14 +7755,16 @@ var getAssistantHistory = function (clientId) { return __awaiter(void 0, void 0,
  * @returns {Promise<string | null>} - The content of the last assistant message, or null if no user message is found.
  */
 var getLastAssistantMessage = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var history, last;
+    var requestId, history, last;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log("function getLastAssistantMessage", {
                     clientId: clientId,
+                    requestId: requestId,
                 });
-                return [4 /*yield*/, getRawHistory(clientId)];
+                return [4 /*yield*/, getRawHistory(clientId, requestId)];
             case 1:
                 history = _a.sent();
                 last = history.findLast(function (_a) {
@@ -7697,14 +7783,16 @@ var getLastAssistantMessage = function (clientId) { return __awaiter(void 0, voi
  * @returns {Promise<string | null>} - The content of the last system message, or null if no user message is found.
  */
 var getLastSystemMessage = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var history, last;
+    var requestId, history, last;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log("function getLastSystemMessage", {
                     clientId: clientId,
+                    requestId: requestId,
                 });
-                return [4 /*yield*/, getRawHistory(clientId)];
+                return [4 /*yield*/, getRawHistory(clientId, requestId)];
             case 1:
                 history = _a.sent();
                 last = history.findLast(function (_a) {
@@ -7808,19 +7896,21 @@ var event = function (clientId, topicName, payload) {
  * @returns {Promise<void>} - A promise that resolves when the output is canceled
  */
 var cancelOutput = function (clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName, currentAgentName;
+    var requestId, swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log('function cancelOutput', {
                     clientId: clientId,
                     agentName: agentName,
+                    requestId: requestId,
                 });
                 swarm.agentValidationService.validate(agentName, "cancelOutput");
                 swarm.sessionValidationService.validate(clientId, "cancelOutput");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "cancelOutput");
-                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(clientId, swarmName)];
+                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(requestId, clientId, swarmName)];
             case 1:
                 currentAgentName = _a.sent();
                 if (currentAgentName !== agentName) {
@@ -7831,7 +7921,7 @@ var cancelOutput = function (clientId, agentName) { return __awaiter(void 0, voi
                     });
                     return [2 /*return*/];
                 }
-                return [4 /*yield*/, swarm.swarmPublicService.cancelOutput(clientId, swarmName)];
+                return [4 /*yield*/, swarm.swarmPublicService.cancelOutput(requestId, clientId, swarmName)];
             case 2:
                 _a.sent();
                 return [2 /*return*/];
@@ -7847,17 +7937,19 @@ var cancelOutput = function (clientId, agentName) { return __awaiter(void 0, voi
  * @returns {Promise<void>} - A promise that resolves when the output is canceled
  */
 var cancelOutputForce = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName;
+    var requestId, swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                requestId = randomString();
                 swarm.loggerService.log('function cancelOutputForce', {
                     clientId: clientId,
+                    requestId: requestId
                 });
                 swarm.sessionValidationService.validate(clientId, "cancelOutputForce");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "cancelOutputForce");
-                return [4 /*yield*/, swarm.swarmPublicService.cancelOutput(clientId, swarmName)];
+                return [4 /*yield*/, swarm.swarmPublicService.cancelOutput(requestId, clientId, swarmName)];
             case 1:
                 _a.sent();
                 return [2 /*return*/];
@@ -7865,7 +7957,7 @@ var cancelOutputForce = function (clientId) { return __awaiter(void 0, void 0, v
     });
 }); };
 
-var validateClientId$5 = function (clientId) {
+var validateClientId$b = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -7883,11 +7975,11 @@ var listenAgentEvent = function (clientId, fn) {
     swarm.loggerService.log("middleware listenAgentEvent", {
         clientId: clientId,
     });
-    validateClientId$5(clientId);
-    swarm.busService.subscribe(clientId, "agent-bus", fn);
+    validateClientId$b(clientId);
+    return swarm.busService.subscribe(clientId, "agent-bus", fn);
 };
 
-var validateClientId$4 = function (clientId) {
+var validateClientId$a = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -7905,11 +7997,11 @@ var listenHistoryEvent = function (clientId, fn) {
     swarm.loggerService.log("middleware listenHistoryEvent", {
         clientId: clientId,
     });
-    validateClientId$4(clientId);
-    swarm.busService.subscribe(clientId, "history-bus", fn);
+    validateClientId$a(clientId);
+    return swarm.busService.subscribe(clientId, "history-bus", fn);
 };
 
-var validateClientId$3 = function (clientId) {
+var validateClientId$9 = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -7927,11 +8019,11 @@ var listenSessionEvent = function (clientId, fn) {
     swarm.loggerService.log("middleware listenSessionEvent", {
         clientId: clientId,
     });
-    validateClientId$3(clientId);
-    swarm.busService.subscribe(clientId, "session-bus", fn);
+    validateClientId$9(clientId);
+    return swarm.busService.subscribe(clientId, "session-bus", fn);
 };
 
-var validateClientId$2 = function (clientId) {
+var validateClientId$8 = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -7949,11 +8041,11 @@ var listenStateEvent = function (clientId, fn) {
     swarm.loggerService.log("middleware listenStateEvent", {
         clientId: clientId,
     });
-    validateClientId$2(clientId);
-    swarm.busService.subscribe(clientId, "state-bus", fn);
+    validateClientId$8(clientId);
+    return swarm.busService.subscribe(clientId, "state-bus", fn);
 };
 
-var validateClientId$1 = function (clientId) {
+var validateClientId$7 = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -7971,11 +8063,11 @@ var listenStorageEvent = function (clientId, fn) {
     swarm.loggerService.log("middleware listenStorageEvent", {
         clientId: clientId,
     });
-    validateClientId$1(clientId);
-    swarm.busService.subscribe(clientId, "storage-bus", fn);
+    validateClientId$7(clientId);
+    return swarm.busService.subscribe(clientId, "storage-bus", fn);
 };
 
-var validateClientId = function (clientId) {
+var validateClientId$6 = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -7993,8 +8085,140 @@ var listenSwarmEvent = function (clientId, fn) {
     swarm.loggerService.log("middleware listenSwarmEvent", {
         clientId: clientId,
     });
+    validateClientId$6(clientId);
+    return swarm.busService.subscribe(clientId, "swarm-bus", fn);
+};
+
+var validateClientId$5 = function (clientId) {
+    if (clientId === "*") {
+        return;
+    }
+    if (!swarm.sessionValidationService.hasSession(clientId)) {
+        throw new Error("agent-swarm listenAgentEventOnce session not found for clientId=".concat(clientId));
+    }
+};
+/**
+ * Hook to subscribe to agent events for a specific client.
+ *
+ * @param {string} clientId - The ID of the client to subscribe to events for.
+ * @param {function} fn - The callback function to handle the event.
+ */
+var listenAgentEventOnce = function (clientId, filterFn, fn) {
+    swarm.loggerService.log("middleware listenAgentEventOnce", {
+        clientId: clientId,
+    });
+    validateClientId$5(clientId);
+    return swarm.busService.once(clientId, "agent-bus", filterFn, fn);
+};
+
+var validateClientId$4 = function (clientId) {
+    if (clientId === "*") {
+        return;
+    }
+    if (!swarm.sessionValidationService.hasSession(clientId)) {
+        throw new Error("agent-swarm listenHistoryEventOnce session not found for clientId=".concat(clientId));
+    }
+};
+/**
+ * Hook to subscribe to history events for a specific client.
+ *
+ * @param {string} clientId - The ID of the client to subscribe to.
+ * @param {(event: IBusEvent) => void} fn - The callback function to handle the event.
+ */
+var listenHistoryEventOnce = function (clientId, filterFn, fn) {
+    swarm.loggerService.log("middleware listenHistoryEventOnce", {
+        clientId: clientId,
+    });
+    validateClientId$4(clientId);
+    return swarm.busService.once(clientId, "history-bus", filterFn, fn);
+};
+
+var validateClientId$3 = function (clientId) {
+    if (clientId === "*") {
+        return;
+    }
+    if (!swarm.sessionValidationService.hasSession(clientId)) {
+        throw new Error("agent-swarm listenSessionEventOnce session not found for clientId=".concat(clientId));
+    }
+};
+/**
+ * Hook to subscribe to session events for a specific client.
+ *
+ * @param {string} clientId - The ID of the client to subscribe to session events for.
+ * @param {function} fn - The callback function to handle the session events.
+ */
+var listenSessionEventOnce = function (clientId, filterFn, fn) {
+    swarm.loggerService.log("middleware listenSessionEventOnce", {
+        clientId: clientId,
+    });
+    validateClientId$3(clientId);
+    return swarm.busService.once(clientId, "session-bus", filterFn, fn);
+};
+
+var validateClientId$2 = function (clientId) {
+    if (clientId === "*") {
+        return;
+    }
+    if (!swarm.sessionValidationService.hasSession(clientId)) {
+        throw new Error("agent-swarm listenStateEventOnce session not found for clientId=".concat(clientId));
+    }
+};
+/**
+ * Hook to subscribe to state events for a specific client.
+ *
+ * @param {string} clientId - The ID of the client to subscribe to.
+ * @param {function} fn - The callback function to handle the event.
+ */
+var listenStateEventOnce = function (clientId, filterFn, fn) {
+    swarm.loggerService.log("middleware listenStateEventOnce", {
+        clientId: clientId,
+    });
+    validateClientId$2(clientId);
+    return swarm.busService.once(clientId, "state-bus", filterFn, fn);
+};
+
+var validateClientId$1 = function (clientId) {
+    if (clientId === "*") {
+        return;
+    }
+    if (!swarm.sessionValidationService.hasSession(clientId)) {
+        throw new Error("agent-swarm listenStorageEventOnce session not found for clientId=".concat(clientId));
+    }
+};
+/**
+ * Hook to subscribe to storage events for a specific client.
+ *
+ * @param {string} clientId - The ID of the client to subscribe to storage events for.
+ * @param {function} fn - The callback function to handle the storage event.
+ */
+var listenStorageEventOnce = function (clientId, filterFn, fn) {
+    swarm.loggerService.log("middleware listenStorageEventOnce", {
+        clientId: clientId,
+    });
+    validateClientId$1(clientId);
+    return swarm.busService.once(clientId, "storage-bus", filterFn, fn);
+};
+
+var validateClientId = function (clientId) {
+    if (clientId === "*") {
+        return;
+    }
+    if (!swarm.sessionValidationService.hasSession(clientId)) {
+        throw new Error("agent-swarm listenSwarmEventOnce session not found for clientId=".concat(clientId));
+    }
+};
+/**
+ * Hook to subscribe to swarm events for a specific client.
+ *
+ * @param {string} clientId - The ID of the client to subscribe to events for.
+ * @param {(event: IBusEvent) => void} fn - The callback function to handle the event.
+ */
+var listenSwarmEventOnce = function (clientId, filterFn, fn) {
+    swarm.loggerService.log("middleware listenSwarmEventOnce", {
+        clientId: clientId,
+    });
     validateClientId(clientId);
-    swarm.busService.subscribe(clientId, "swarm-bus", fn);
+    return swarm.busService.once(clientId, "swarm-bus", filterFn, fn);
 };
 
 var StorageUtils = /** @class */ (function () {
@@ -8011,21 +8235,24 @@ var StorageUtils = /** @class */ (function () {
          * @template T
          */
         this.take = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+            var requestId;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
+                        requestId = randomString();
                         swarm.loggerService.log("StorageUtils take", {
                             search: payload.search,
                             total: payload.total,
                             clientId: payload.clientId,
                             storageName: payload.storageName,
                             score: payload.score,
+                            requestId: requestId,
                         });
                         swarm.storageValidationService.validate(payload.storageName, "StorageUtils");
                         if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
                             throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (take)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.take(payload.search, payload.total, payload.clientId, payload.storageName, payload.score)];
+                        return [4 /*yield*/, swarm.storagePublicService.take(payload.search, payload.total, requestId, payload.clientId, payload.storageName, payload.score)];
                     case 1: return [2 /*return*/, (_a.sent())];
                 }
             });
@@ -8040,19 +8267,22 @@ var StorageUtils = /** @class */ (function () {
          * @template T
          */
         this.upsert = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+            var requestId;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
+                        requestId = randomString();
                         swarm.loggerService.log("StorageUtils upsert", {
                             item: payload.item,
                             clientId: payload.clientId,
                             storageName: payload.storageName,
+                            requestId: requestId,
                         });
                         swarm.storageValidationService.validate(payload.storageName, "StorageUtils");
                         if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
                             throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (upsert)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.upsert(payload.item, payload.clientId, payload.storageName)];
+                        return [4 /*yield*/, swarm.storagePublicService.upsert(payload.item, requestId, payload.clientId, payload.storageName)];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
@@ -8066,19 +8296,22 @@ var StorageUtils = /** @class */ (function () {
          * @returns {Promise<void>} - A promise that resolves when the operation is complete.
          */
         this.remove = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+            var requestId;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
+                        requestId = randomString();
                         swarm.loggerService.log("StorageUtils remove", {
                             itemId: payload.itemId,
                             clientId: payload.clientId,
                             storageName: payload.storageName,
+                            requestId: requestId,
                         });
                         swarm.storageValidationService.validate(payload.storageName, "StorageUtils");
                         if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
                             throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (remove)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.remove(payload.itemId, payload.clientId, payload.storageName)];
+                        return [4 /*yield*/, swarm.storagePublicService.remove(payload.itemId, requestId, payload.clientId, payload.storageName)];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
@@ -8093,19 +8326,22 @@ var StorageUtils = /** @class */ (function () {
          * @template T
          */
         this.get = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+            var requestId;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
+                        requestId = randomString();
                         swarm.loggerService.log("StorageUtils get", {
                             itemId: payload.itemId,
                             clientId: payload.clientId,
                             storageName: payload.storageName,
+                            requestId: requestId,
                         });
                         swarm.storageValidationService.validate(payload.storageName, "StorageUtils");
                         if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
                             throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (get)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.get(payload.itemId, payload.clientId, payload.storageName)];
+                        return [4 /*yield*/, swarm.storagePublicService.get(payload.itemId, requestId, payload.clientId, payload.storageName)];
                     case 1: return [2 /*return*/, (_a.sent())];
                 }
             });
@@ -8120,18 +8356,21 @@ var StorageUtils = /** @class */ (function () {
          * @template T
          */
         this.list = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+            var requestId;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
+                        requestId = randomString();
                         swarm.loggerService.log("StorageUtils list", {
                             clientId: payload.clientId,
                             storageName: payload.storageName,
+                            requestId: requestId,
                         });
                         swarm.storageValidationService.validate(payload.storageName, "StorageUtils");
                         if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
                             throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (list)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.list(payload.clientId, payload.storageName, payload.filter)];
+                        return [4 /*yield*/, swarm.storagePublicService.list(requestId, payload.clientId, payload.storageName, payload.filter)];
                     case 1: return [2 /*return*/, (_a.sent())];
                 }
             });
@@ -8144,18 +8383,21 @@ var StorageUtils = /** @class */ (function () {
          * @returns {Promise<void>} - A promise that resolves when the operation is complete.
          */
         this.clear = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+            var requestId;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
+                        requestId = randomString();
                         swarm.loggerService.log("StorageUtils clear", {
                             clientId: payload.clientId,
                             storageName: payload.storageName,
+                            requestId: requestId,
                         });
                         swarm.storageValidationService.validate(payload.storageName, "StorageUtils");
                         if (!swarm.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
                             throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (clear)"));
                         }
-                        return [4 /*yield*/, swarm.storagePublicService.clear(payload.clientId, payload.storageName)];
+                        return [4 /*yield*/, swarm.storagePublicService.clear(requestId, payload.clientId, payload.storageName)];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
@@ -8183,17 +8425,20 @@ var StateUtils = /** @class */ (function () {
          * @throws Will throw an error if the state is not registered in the agent.
          */
         this.getState = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+            var requestId;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
+                        requestId = randomString();
                         swarm.loggerService.log("StateUtils getState", {
                             clientId: payload.clientId,
                             stateName: payload.stateName,
+                            requestId: requestId,
                         });
                         if (!swarm.agentValidationService.hasState(payload.agentName, payload.stateName)) {
                             throw new Error("agent-swarm StateUtils ".concat(payload.stateName, " not registered in ").concat(payload.agentName, " (getState)"));
                         }
-                        return [4 /*yield*/, swarm.statePublicService.getState(payload.clientId, payload.stateName)];
+                        return [4 /*yield*/, swarm.statePublicService.getState(requestId, payload.clientId, payload.stateName)];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
@@ -8210,23 +8455,26 @@ var StateUtils = /** @class */ (function () {
          * @throws Will throw an error if the state is not registered in the agent.
          */
         this.setState = function (dispatchFn, payload) { return __awaiter(_this, void 0, void 0, function () {
+            var requestId;
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
+                        requestId = randomString();
                         swarm.loggerService.log("StateUtils setState", {
                             clientId: payload.clientId,
                             stateName: payload.stateName,
+                            requestId: requestId,
                         });
                         if (!swarm.agentValidationService.hasState(payload.agentName, payload.stateName)) {
                             throw new Error("agent-swarm StateUtils ".concat(payload.stateName, " not registered in ").concat(payload.agentName, " (setState)"));
                         }
                         if (!(typeof dispatchFn === "function")) return [3 /*break*/, 2];
-                        return [4 /*yield*/, swarm.statePublicService.setState(dispatchFn, payload.clientId, payload.stateName)];
+                        return [4 /*yield*/, swarm.statePublicService.setState(dispatchFn, requestId, payload.clientId, payload.stateName)];
                     case 1: return [2 /*return*/, _a.sent()];
                     case 2: return [4 /*yield*/, swarm.statePublicService.setState(function () { return __awaiter(_this, void 0, void 0, function () { return __generator(this, function (_a) {
                             return [2 /*return*/, dispatchFn];
-                        }); }); }, payload.clientId, payload.stateName)];
+                        }); }); }, requestId, payload.clientId, payload.stateName)];
                     case 3: return [2 /*return*/, _a.sent()];
                 }
             });
@@ -8240,4 +8488,23 @@ var StateUtils = /** @class */ (function () {
  */
 var State = new StateUtils();
 
-export { ContextService, History, HistoryAdapter, HistoryInstance, State, Storage, addAgent, addCompletion, addEmbedding, addState, addStorage, addSwarm, addTool, cancelOutput, cancelOutputForce, changeAgent, commitFlush, commitFlushForce, commitSystemMessage, commitSystemMessageForce, commitToolOutput, commitToolOutputForce, commitUserMessage, commitUserMessageForce, complete, disposeConnection, emit, emitForce, event, execute, executeForce, getAgentHistory, getAgentName, getAssistantHistory, getLastAssistantMessage, getLastSystemMessage, getLastUserMessage, getRawHistory, getSessionMode, getUserHistory, listenAgentEvent, listenEvent, listenHistoryEvent, listenSessionEvent, listenStateEvent, listenStorageEvent, listenSwarmEvent, makeAutoDispose, makeConnection, session, setConfig, swarm };
+var LoggerUtils = /** @class */ (function () {
+    function LoggerUtils() {
+        /**
+         * Sets the provided logger to the logger service.
+         * @param {ILogger} logger - The logger instance to be used.
+         */
+        this.useLogger = function (logger) {
+            swarm.loggerService.log("HistoryInstance useLogger");
+            swarm.loggerService.setLogger(logger);
+        };
+    }
+    return LoggerUtils;
+}());
+/**
+ * Instance of LoggerUtils to be used for logging.
+ * @type {LoggerUtils}
+ */
+var Logger = new LoggerUtils();
+
+export { ContextService, History, HistoryAdapter, HistoryInstance, Logger, State, Storage, addAgent, addCompletion, addEmbedding, addState, addStorage, addSwarm, addTool, cancelOutput, cancelOutputForce, changeAgent, commitFlush, commitFlushForce, commitSystemMessage, commitSystemMessageForce, commitToolOutput, commitToolOutputForce, commitUserMessage, commitUserMessageForce, complete, disposeConnection, emit, emitForce, event, execute, executeForce, getAgentHistory, getAgentName, getAssistantHistory, getLastAssistantMessage, getLastSystemMessage, getLastUserMessage, getRawHistory, getSessionMode, getUserHistory, listenAgentEvent, listenAgentEventOnce, listenEvent, listenEventOnce, listenHistoryEvent, listenHistoryEventOnce, listenSessionEvent, listenSessionEventOnce, listenStateEvent, listenStateEventOnce, listenStorageEvent, listenStorageEventOnce, listenSwarmEvent, listenSwarmEventOnce, makeAutoDispose, makeConnection, session, setConfig, swarm };