agent-swarm-kit 1.0.69 → 1.0.71

package/build/index.cjs

@@ -227,6 +227,7 @@ var _a;
 var init = (_a = diKit.createActivator("agent-swarm"), _a.init), inject = _a.inject, provide = _a.provide;
 
 var baseServices$1 = {
+    busService: Symbol('busService'),
     loggerService: Symbol('loggerService'),
     contextService: Symbol('contextService'),
 };
@@ -953,7 +954,7 @@ var ClientAgent = /** @class */ (function () {
                         validation = null;
                         return [4 /*yield*/, this.params.validate(result)];
                     case 2:
-                        if (!(validation = _a.sent())) return [3 /*break*/, 7];
+                        if (!(validation = _a.sent())) return [3 /*break*/, 8];
                         return [4 /*yield*/, this._resurrectModel(mode, validation)];
                     case 3:
                         rawResult_1 = _a.sent();
@@ -970,12 +971,44 @@ var ClientAgent = /** @class */ (function () {
                         return [4 /*yield*/, this._outputSubject.next(result_1)];
                     case 6:
                         _a.sent();
-                        return [2 /*return*/];
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "emit-output",
+                                source: "agent",
+                                input: {
+                                    mode: mode,
+                                    rawResult: rawResult_1,
+                                },
+                                output: {
+                                    result: result_1,
+                                },
+                                context: {
+                                    agentName: this.params.agentName,
+                                }
+                            })];
                     case 7:
+                        _a.sent();
+                        return [2 /*return*/];
+                    case 8:
                         this.params.onOutput &&
                             this.params.onOutput(this.params.clientId, this.params.agentName, result);
                         return [4 /*yield*/, this._outputSubject.next(result)];
-                    case 8:
+                    case 9:
+                        _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "emit-output",
+                                source: "agent",
+                                input: {
+                                    mode: mode,
+                                    rawResult: rawResult,
+                                },
+                                output: {
+                                    result: result,
+                                },
+                                context: {
+                                    agentName: this.params.agentName,
+                                }
+                            })];
+                    case 10:
                         _a.sent();
                         return [2 /*return*/];
                 }
@@ -1107,6 +1140,19 @@ var ClientAgent = /** @class */ (function () {
                                 content: message.trim(),
                             })];
                     case 1:
+                        _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "commit-user-message",
+                                source: "agent",
+                                input: {
+                                    message: message,
+                                },
+                                output: {},
+                                context: {
+                                    agentName: this.params.agentName,
+                                }
+                            })];
+                    case 2:
                         _a.sent();
                         return [2 /*return*/];
                 }
@@ -1130,6 +1176,17 @@ var ClientAgent = /** @class */ (function () {
                                 content: "",
                             })];
                     case 1:
+                        _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "commit-flush",
+                                source: "agent",
+                                input: {},
+                                output: {},
+                                context: {
+                                    agentName: this.params.agentName,
+                                }
+                            })];
+                    case 2:
                         _a.sent();
                         return [2 /*return*/];
                 }
@@ -1146,6 +1203,17 @@ var ClientAgent = /** @class */ (function () {
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " commitAgentChange"));
                         return [4 /*yield*/, this._agentChangeSubject.next(AGENT_CHANGE_SYMBOL)];
                     case 1:
+                        _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "commit-agent-change",
+                                source: "agent",
+                                input: {},
+                                output: {},
+                                context: {
+                                    agentName: this.params.agentName,
+                                }
+                            })];
+                    case 2:
                         _a.sent();
                         return [2 /*return*/];
                 }
@@ -1170,6 +1238,19 @@ var ClientAgent = /** @class */ (function () {
                                 content: message.trim(),
                             })];
                     case 1:
+                        _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "commit-system-message",
+                                source: "agent",
+                                input: {
+                                    message: message,
+                                },
+                                output: {},
+                                context: {
+                                    agentName: this.params.agentName,
+                                }
+                            })];
+                    case 2:
                         _a.sent();
                         return [2 /*return*/];
                 }
@@ -1198,6 +1279,20 @@ var ClientAgent = /** @class */ (function () {
                         _a.sent();
                         return [4 /*yield*/, this._toolCommitSubject.next()];
                     case 2:
+                        _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "commit-tool-output",
+                                source: "agent",
+                                input: {
+                                    toolId: toolId,
+                                    content: content,
+                                },
+                                output: {},
+                                context: {
+                                    agentName: this.params.agentName,
+                                }
+                            })];
+                    case 3:
                         _a.sent();
                         return [2 /*return*/];
                 }
@@ -1411,6 +1506,7 @@ var AgentConnectionService = /** @class */ (function () {
     function AgentConnectionService() {
         var _this = this;
         this.loggerService = inject(TYPES.loggerService);
+        this.busService = inject(TYPES.busService);
         this.contextService = inject(TYPES.contextService);
         this.sessionValidationService = inject(TYPES.sessionValidationService);
         this.historyConnectionService = inject(TYPES.historyConnectionService);
@@ -1441,7 +1537,7 @@ var AgentConnectionService = /** @class */ (function () {
                     .getStorage(clientId, storageName)
                     .waitForInit();
             });
-            return new ClientAgent(__assign({ clientId: clientId, agentName: agentName, validate: validate, logger: _this.loggerService, history: _this.historyConnectionService.getHistory(clientId, agentName), prompt: prompt, system: system, transform: transform, map: map, tools: tools === null || tools === void 0 ? void 0 : tools.map(_this.toolSchemaService.get), completion: completion }, callbacks));
+            return new ClientAgent(__assign({ clientId: clientId, agentName: agentName, validate: validate, logger: _this.loggerService, bus: _this.busService, history: _this.historyConnectionService.getHistory(clientId, agentName), prompt: prompt, system: system, transform: transform, map: map, tools: tools === null || tools === void 0 ? void 0 : tools.map(_this.toolSchemaService.get), completion: completion }, callbacks));
         });
         /**
          * Executes an input command.
@@ -1619,6 +1715,19 @@ var ClientHistory = /** @class */ (function () {
                         this.params.logger.debug("ClientHistory agentName=".concat(this.params.agentName, " push"), { message: message });
                         return [4 /*yield*/, this.params.items.push(message, this.params.clientId, this.params.agentName)];
                     case 1:
+                        _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "push",
+                                source: "history",
+                                input: {
+                                    message: message
+                                },
+                                output: {},
+                                context: {
+                                    agentName: this.params.agentName,
+                                }
+                            })];
+                    case 2:
                         _a.sent();
                         return [2 /*return*/];
                 }
@@ -1852,6 +1961,7 @@ var HistoryConnectionService = /** @class */ (function () {
     function HistoryConnectionService() {
         var _this = this;
         this.loggerService = inject(TYPES.loggerService);
+        this.busService = inject(TYPES.busService);
         this.contextService = inject(TYPES.contextService);
         this.sessionValidationService = inject(TYPES.sessionValidationService);
         /**
@@ -1868,6 +1978,7 @@ var HistoryConnectionService = /** @class */ (function () {
             return new ClientHistory({
                 clientId: clientId,
                 agentName: agentName,
+                bus: _this.busService,
                 items: GLOBAL_CONFIG.CC_GET_AGENT_HISTORY_ADAPTER(clientId, agentName),
                 logger: _this.loggerService,
             });
@@ -1995,10 +2106,44 @@ var ClientSwarm = /** @class */ (function () {
         this.params = params;
         this._agentChangedSubject = new functoolsKit.Subject();
         this._activeAgent = AGENT_NEED_FETCH;
+        this._cancelOutputSubject = new functoolsKit.Subject();
+        /**
+         * Cancel the await of output by emit of empty string
+         * @returns {Promise<string>} - The output from the active agent.
+         */
+        this.cancelOutput = function () { return __awaiter(_this, void 0, void 0, function () {
+            var _a, _b;
+            var _c;
+            return __generator(this, function (_d) {
+                switch (_d.label) {
+                    case 0:
+                        this.params.logger.debug("ClientSwarm swarmName=".concat(this.params.swarmName, " clientId=").concat(this.params.clientId, " cancelOutput"));
+                        _b = (_a = this._cancelOutputSubject).next;
+                        _c = {};
+                        return [4 /*yield*/, this.getAgentName()];
+                    case 1: return [4 /*yield*/, _b.apply(_a, [(_c.agentName = _d.sent(),
+                                _c.output = "",
+                                _c)])];
+                    case 2:
+                        _d.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "cancel-output",
+                                source: "swarm",
+                                input: {},
+                                output: {},
+                                context: {
+                                    swarmName: this.params.swarmName,
+                                },
+                            })];
+                    case 3:
+                        _d.sent();
+                        return [2 /*return*/];
+                }
+            });
+        }); };
         /**
          * Waits for output from the active agent.
          * @returns {Promise<string>} - The output from the active agent.
-         * @throws {Error} - If the timeout is reached.
          */
         this.waitForOutput = functoolsKit.queued(function () { return __awaiter(_this, void 0, void 0, function () {
             var _a, awaiter, resolve, getOutput, handleOutput, un, _b, agentName, output, expectAgent;
@@ -2012,18 +2157,23 @@ var ClientSwarm = /** @class */ (function () {
                             var _this = this;
                             return __generator(this, function (_a) {
                                 switch (_a.label) {
-                                    case 0: return [4 /*yield*/, Promise.race(this._agentList.map(function (_a) { return __awaiter(_this, [_a], void 0, function (_b) {
+                                    case 0: return [4 /*yield*/, Promise.race(this._agentList
+                                            .map(function (_a) { return __awaiter(_this, [_a], void 0, function (_b) {
                                             var _c;
                                             var _d = __read(_b, 2), agentName = _d[0], agent = _d[1];
                                             return __generator(this, function (_e) {
                                                 switch (_e.label) {
                                                     case 0:
-                                                        _c = { agentName: agentName };
+                                                        _c = {
+                                                            agentName: agentName
+                                                        };
                                                         return [4 /*yield*/, agent.waitForOutput()];
-                                                    case 1: return [2 /*return*/, (_c.output = _e.sent(), _c)];
+                                                    case 1: return [2 /*return*/, (_c.output = _e.sent(),
+                                                            _c)];
                                                 }
                                             });
-                                        }); }))];
+                                        }); })
+                                            .concat(this._cancelOutputSubject.toPromise()))];
                                     case 1: return [2 /*return*/, _a.sent()];
                                 }
                             });
@@ -2046,7 +2196,21 @@ var ClientSwarm = /** @class */ (function () {
                         return [4 /*yield*/, this.getAgentName()];
                     case 2:
                         expectAgent = _c.sent();
-                        agentName !== expectAgent && this.params.logger.debug("ClientSwarm swarmName=".concat(this.params.swarmName, " clientId=").concat(this.params.clientId, " waitForAgent agent miss"), { agentName: agentName, expectAgent: expectAgent });
+                        agentName !== expectAgent &&
+                            this.params.logger.debug("ClientSwarm swarmName=".concat(this.params.swarmName, " clientId=").concat(this.params.clientId, " waitForAgent agent miss"), { agentName: agentName, expectAgent: expectAgent });
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "wait-for-output",
+                                source: "swarm",
+                                input: {},
+                                output: {
+                                    result: output,
+                                },
+                                context: {
+                                    swarmName: this.params.swarmName,
+                                },
+                            })];
+                    case 3:
+                        _c.sent();
                         return [2 /*return*/, output];
                 }
             });
@@ -2067,7 +2231,20 @@ var ClientSwarm = /** @class */ (function () {
                     case 1:
                         _a._activeAgent = _b.sent();
                         _b.label = 2;
-                    case 2: return [2 /*return*/, this._activeAgent];
+                    case 2: return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                            type: "get-agent-name",
+                            source: "swarm",
+                            input: {},
+                            output: {
+                                activeAgent: this._activeAgent,
+                            },
+                            context: {
+                                swarmName: this.params.swarmName,
+                            },
+                        })];
+                    case 3:
+                        _b.sent();
+                        return [2 /*return*/, this._activeAgent];
                 }
             });
         }); };
@@ -2076,7 +2253,7 @@ var ClientSwarm = /** @class */ (function () {
          * @returns {Promise<IAgent>} - The active agent.
          */
         this.getAgent = function () { return __awaiter(_this, void 0, void 0, function () {
-            var agent;
+            var agent, result;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -2084,7 +2261,21 @@ var ClientSwarm = /** @class */ (function () {
                         return [4 /*yield*/, this.getAgentName()];
                     case 1:
                         agent = _a.sent();
-                        return [2 /*return*/, this.params.agentMap[agent]];
+                        result = this.params.agentMap[agent];
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "get-agent",
+                                source: "swarm",
+                                input: {
+                                    result: result,
+                                },
+                                output: {},
+                                context: {
+                                    swarmName: this.params.swarmName,
+                                },
+                            })];
+                    case 2:
+                        _a.sent();
+                        return [2 /*return*/, result];
                 }
             });
         }); };
@@ -2103,8 +2294,22 @@ var ClientSwarm = /** @class */ (function () {
                             throw new Error("agent-swarm agent ".concat(agentName, " not in the swarm"));
                         }
                         this.params.agentMap[agentName] = agent;
-                        return [4 /*yield*/, this._agentChangedSubject.next([agentName, agent])];
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "set-agent-ref",
+                                source: "swarm",
+                                input: {
+                                    agentName: agentName,
+                                    agent: agent,
+                                },
+                                output: {},
+                                context: {
+                                    swarmName: this.params.swarmName,
+                                },
+                            })];
                     case 1:
+                        _a.sent();
+                        return [4 /*yield*/, this._agentChangedSubject.next([agentName, agent])];
+                    case 2:
                         _a.sent();
                         return [2 /*return*/];
                 }
@@ -2122,6 +2327,19 @@ var ClientSwarm = /** @class */ (function () {
                         this._activeAgent = agentName;
                         return [4 /*yield*/, this.params.onAgentChanged(this.params.clientId, agentName, this.params.swarmName)];
                     case 1:
+                        _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "set-agent-name",
+                                source: "swarm",
+                                input: {
+                                    agentName: agentName,
+                                },
+                                output: {},
+                                context: {
+                                    swarmName: this.params.swarmName,
+                                },
+                            })];
+                    case 2:
                         _a.sent();
                         return [2 /*return*/];
                 }
@@ -2149,6 +2367,7 @@ var SwarmConnectionService = /** @class */ (function () {
     function SwarmConnectionService() {
         var _this = this;
         this.loggerService = inject(TYPES.loggerService);
+        this.busService = inject(TYPES.busService);
         this.contextService = inject(TYPES.contextService);
         this.agentConnectionService = inject(TYPES.agentConnectionService);
         this.swarmSchemaService = inject(TYPES.swarmSchemaService);
@@ -2184,6 +2403,7 @@ var SwarmConnectionService = /** @class */ (function () {
                 defaultAgent: defaultAgent,
                 swarmName: swarmName,
                 logger: _this.loggerService,
+                bus: _this.busService,
                 onAgentChanged: function (clientId, agentName, swarmName) {
                     return __awaiter(this, void 0, void 0, function () {
                         return __generator(this, function (_a) {
@@ -2203,6 +2423,22 @@ var SwarmConnectionService = /** @class */ (function () {
                 callbacks: callbacks,
             });
         });
+        /**
+         * Cancel the await of output by emit of empty string
+         * @returns {Promise<void>}
+         */
+        this.cancelOutput = function () { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("swarmConnectionService cancelOutput", {
+                            context: this.contextService.context,
+                        });
+                        return [4 /*yield*/, this.getSwarm(this.contextService.context.clientId, this.contextService.context.swarmName).cancelOutput()];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Waits for the output from the swarm.
          * @returns {Promise<any>} The output from the swarm.
@@ -2397,6 +2633,19 @@ var ClientSession = /** @class */ (function () {
                             this.params.onEmit(this.params.clientId, this.params.swarmName, message);
                         return [4 /*yield*/, this._emitSubject.next(message)];
                     case 1:
+                        _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "emit",
+                                source: "session",
+                                input: {
+                                    message: message
+                                },
+                                output: {},
+                                context: {
+                                    swarmName: this.params.swarmName,
+                                }
+                            })];
+                    case 2:
                         _a.sent();
                         return [2 /*return*/];
                 }
@@ -2427,6 +2676,22 @@ var ClientSession = /** @class */ (function () {
                         return [4 /*yield*/, outputAwaiter];
                     case 2:
                         output = _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "execute",
+                                source: "session",
+                                input: {
+                                    message: message,
+                                    mode: mode,
+                                },
+                                output: {
+                                    result: output,
+                                },
+                                context: {
+                                    swarmName: this.params.swarmName,
+                                }
+                            })];
+                    case 3:
+                        _a.sent();
                         return [2 /*return*/, output];
                 }
             });
@@ -2438,7 +2703,7 @@ var ClientSession = /** @class */ (function () {
          * @returns {Promise<void>}
          */
         this.commitToolOutput = function (toolId, content) { return __awaiter(_this, void 0, void 0, function () {
-            var agent;
+            var agent, result;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -2450,7 +2715,23 @@ var ClientSession = /** @class */ (function () {
                     case 1:
                         agent = _a.sent();
                         return [4 /*yield*/, agent.commitToolOutput(toolId, content)];
-                    case 2: return [2 /*return*/, _a.sent()];
+                    case 2:
+                        result = _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "commit-tool-output",
+                                source: "session",
+                                input: {
+                                    toolId: toolId,
+                                    content: content,
+                                },
+                                output: {},
+                                context: {
+                                    swarmName: this.params.swarmName,
+                                }
+                            })];
+                    case 3:
+                        _a.sent();
+                        return [2 /*return*/, result];
                 }
             });
         }); };
@@ -2460,7 +2741,7 @@ var ClientSession = /** @class */ (function () {
          * @returns {Promise<void>}
          */
         this.commitUserMessage = function (message) { return __awaiter(_this, void 0, void 0, function () {
-            var agent;
+            var agent, result;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -2471,7 +2752,22 @@ var ClientSession = /** @class */ (function () {
                     case 1:
                         agent = _a.sent();
                         return [4 /*yield*/, agent.commitUserMessage(message)];
-                    case 2: return [2 /*return*/, _a.sent()];
+                    case 2:
+                        result = _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "commit-user-message",
+                                source: "session",
+                                input: {
+                                    message: message,
+                                },
+                                output: {},
+                                context: {
+                                    swarmName: this.params.swarmName,
+                                }
+                            })];
+                    case 3:
+                        _a.sent();
+                        return [2 /*return*/, result];
                 }
             });
         }); };
@@ -2480,7 +2776,7 @@ var ClientSession = /** @class */ (function () {
          * @returns {Promise<void>}
          */
         this.commitFlush = function () { return __awaiter(_this, void 0, void 0, function () {
-            var agent;
+            var agent, result;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -2489,7 +2785,20 @@ var ClientSession = /** @class */ (function () {
                     case 1:
                         agent = _a.sent();
                         return [4 /*yield*/, agent.commitFlush()];
-                    case 2: return [2 /*return*/, _a.sent()];
+                    case 2:
+                        result = _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "commit-flush",
+                                source: "session",
+                                input: {},
+                                output: {},
+                                context: {
+                                    swarmName: this.params.swarmName,
+                                }
+                            })];
+                    case 3:
+                        _a.sent();
+                        return [2 /*return*/, result];
                 }
             });
         }); };
@@ -2499,7 +2808,7 @@ var ClientSession = /** @class */ (function () {
          * @returns {Promise<void>}
          */
         this.commitSystemMessage = function (message) { return __awaiter(_this, void 0, void 0, function () {
-            var agent;
+            var agent, result;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -2510,7 +2819,22 @@ var ClientSession = /** @class */ (function () {
                     case 1:
                         agent = _a.sent();
                         return [4 /*yield*/, agent.commitSystemMessage(message)];
-                    case 2: return [2 /*return*/, _a.sent()];
+                    case 2:
+                        result = _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "commit-system-message",
+                                source: "session",
+                                input: {
+                                    message: message
+                                },
+                                output: {},
+                                context: {
+                                    swarmName: this.params.swarmName,
+                                }
+                            })];
+                    case 3:
+                        _a.sent();
+                        return [2 /*return*/, result];
                 }
             });
         }); };
@@ -2541,6 +2865,15 @@ var ClientSession = /** @class */ (function () {
                     }
                 });
             }); });
+            _this.params.bus.emit(_this.params.clientId, {
+                type: "connect",
+                source: "session",
+                input: {},
+                output: {},
+                context: {
+                    swarmName: _this.params.swarmName,
+                }
+            });
             return function (incoming) { return __awaiter(_this, void 0, void 0, function () {
                 var _a;
                 var _b;
@@ -2592,6 +2925,7 @@ var SessionConnectionService = /** @class */ (function () {
     function SessionConnectionService() {
         var _this = this;
         this.loggerService = inject(TYPES.loggerService);
+        this.busService = inject(TYPES.busService);
         this.contextService = inject(TYPES.contextService);
         this.swarmConnectionService = inject(TYPES.swarmConnectionService);
         this.swarmSchemaService = inject(TYPES.swarmSchemaService);
@@ -2606,7 +2940,7 @@ var SessionConnectionService = /** @class */ (function () {
             return "".concat(clientId, "-").concat(swarmName);
         }, function (clientId, swarmName) {
             var callbacks = _this.swarmSchemaService.get(swarmName).callbacks;
-            return new ClientSession(__assign({ clientId: clientId, logger: _this.loggerService, swarm: _this.swarmConnectionService.getSwarm(clientId, swarmName), swarmName: swarmName }, callbacks));
+            return new ClientSession(__assign({ clientId: clientId, logger: _this.loggerService, bus: _this.busService, swarm: _this.swarmConnectionService.getSwarm(clientId, swarmName), swarmName: swarmName }, callbacks));
         });
         /**
          * Emits a message to the session.
@@ -3542,6 +3876,39 @@ var SwarmPublicService = /** @class */ (function () {
         var _this = this;
         this.loggerService = inject(TYPES.loggerService);
         this.swarmConnectionService = inject(TYPES.swarmConnectionService);
+        /**
+         * Cancel the await of output by emit of empty string
+         * @param {string} clientId - The client ID.
+         * @param {SwarmName} swarmName - The swarm name.
+         * @returns {Promise<void>}
+         */
+        this.cancelOutput = function (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 cancelOutput", {
+                            clientId: clientId,
+                            swarmName: swarmName,
+                        });
+                        return [4 /*yield*/, ContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.swarmConnectionService.cancelOutput()];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                clientId: clientId,
+                                swarmName: swarmName,
+                                agentName: "",
+                                storageName: "",
+                                stateName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Waits for output from the swarm.
          * @param {string} clientId - The client ID.
@@ -4416,6 +4783,7 @@ var ClientStorage = /** @class */ (function () {
      */
     function ClientStorage(params) {
         var _this = this;
+        var _a;
         this.params = params;
         this._itemMap = new Map();
         /**
@@ -4534,6 +4902,23 @@ var ClientStorage = /** @class */ (function () {
                             if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onSearch) {
                                 (_b = this.params.callbacks) === null || _b === void 0 ? void 0 : _b.onSearch(search, indexed, this.params.clientId, this.params.storageName);
                             }
+                            return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                    type: "take",
+                                    source: "storage",
+                                    input: {
+                                        search: search,
+                                        total: total,
+                                        score: score,
+                                    },
+                                    output: {
+                                        indexed: indexed,
+                                    },
+                                    context: {
+                                        storageName: this.params.storageName,
+                                    }
+                                })];
+                        case 3:
+                            _c.sent();
                             return [2 /*return*/, indexed.take(total, score)];
                     }
                 });
@@ -4560,6 +4945,19 @@ var ClientStorage = /** @class */ (function () {
                         if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onUpdate) {
                             (_b = this.params.callbacks) === null || _b === void 0 ? void 0 : _b.onUpdate(__spreadArray([], __read(this._itemMap.values()), false), this.params.clientId, this.params.storageName);
                         }
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "upsert",
+                                source: "storage",
+                                input: {
+                                    item: item,
+                                },
+                                output: {},
+                                context: {
+                                    storageName: this.params.storageName,
+                                }
+                            })];
+                    case 2:
+                        _c.sent();
                         return [2 /*return*/];
                 }
             });
@@ -4572,15 +4970,31 @@ var ClientStorage = /** @class */ (function () {
         this.remove = function (itemId) { return __awaiter(_this, void 0, void 0, function () {
             var _a, _b;
             return __generator(this, function (_c) {
-                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " remove"), {
-                    id: itemId,
-                });
-                this._itemMap.delete(itemId);
-                this._createEmbedding.clear(itemId);
-                if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onUpdate) {
-                    (_b = this.params.callbacks) === null || _b === void 0 ? void 0 : _b.onUpdate(__spreadArray([], __read(this._itemMap.values()), false), this.params.clientId, this.params.storageName);
+                switch (_c.label) {
+                    case 0:
+                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " remove"), {
+                            id: itemId,
+                        });
+                        this._itemMap.delete(itemId);
+                        this._createEmbedding.clear(itemId);
+                        if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onUpdate) {
+                            (_b = this.params.callbacks) === null || _b === void 0 ? void 0 : _b.onUpdate(__spreadArray([], __read(this._itemMap.values()), false), this.params.clientId, this.params.storageName);
+                        }
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "remove",
+                                source: "storage",
+                                input: {
+                                    itemId: itemId,
+                                },
+                                output: {},
+                                context: {
+                                    storageName: this.params.storageName,
+                                }
+                            })];
+                    case 1:
+                        _c.sent();
+                        return [2 /*return*/];
                 }
-                return [2 /*return*/];
             });
         }); };
         /**
@@ -4589,10 +5003,24 @@ var ClientStorage = /** @class */ (function () {
          */
         this.clear = function () { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
-                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " clear"));
-                this._itemMap.clear();
-                this._createEmbedding.clear();
-                return [2 /*return*/];
+                switch (_a.label) {
+                    case 0:
+                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " clear"));
+                        this._itemMap.clear();
+                        this._createEmbedding.clear();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "clear",
+                                source: "storage",
+                                input: {},
+                                output: {},
+                                context: {
+                                    storageName: this.params.storageName,
+                                }
+                            })];
+                    case 1:
+                        _a.sent();
+                        return [2 /*return*/];
+                }
             });
         }); };
         /**
@@ -4601,12 +5029,32 @@ var ClientStorage = /** @class */ (function () {
          * @returns {Promise<T | null>} - The item or null if not found.
          */
         this.get = function (itemId) { return __awaiter(_this, void 0, void 0, function () {
+            var result;
             var _a;
             return __generator(this, function (_b) {
-                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " get"), {
-                    id: itemId,
-                });
-                return [2 /*return*/, (_a = this._itemMap.get(itemId)) !== null && _a !== void 0 ? _a : null];
+                switch (_b.label) {
+                    case 0:
+                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " get"), {
+                            id: itemId,
+                        });
+                        result = (_a = this._itemMap.get(itemId)) !== null && _a !== void 0 ? _a : null;
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "get",
+                                source: "storage",
+                                input: {
+                                    itemId: itemId,
+                                },
+                                output: {
+                                    result: result,
+                                },
+                                context: {
+                                    storageName: this.params.storageName,
+                                }
+                            })];
+                    case 1:
+                        _b.sent();
+                        return [2 /*return*/, result];
+                }
             });
         }); };
         /**
@@ -4618,32 +5066,65 @@ var ClientStorage = /** @class */ (function () {
             var result, _a, _b, item;
             var e_1, _c;
             return __generator(this, function (_d) {
-                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " list"));
-                if (!filter) {
-                    return [2 /*return*/, __spreadArray([], __read(this._itemMap.values()), false)];
-                }
-                result = [];
-                try {
-                    for (_a = __values(this._itemMap.values()), _b = _a.next(); !_b.done; _b = _a.next()) {
-                        item = _b.value;
-                        if (filter(item)) {
-                            result.push(item);
+                switch (_d.label) {
+                    case 0:
+                        this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " list"));
+                        if (!filter) {
+                            return [2 /*return*/, __spreadArray([], __read(this._itemMap.values()), false)];
                         }
-                    }
+                        result = [];
+                        try {
+                            for (_a = __values(this._itemMap.values()), _b = _a.next(); !_b.done; _b = _a.next()) {
+                                item = _b.value;
+                                if (filter(item)) {
+                                    result.push(item);
+                                }
+                            }
+                        }
+                        catch (e_1_1) { e_1 = { error: e_1_1 }; }
+                        finally {
+                            try {
+                                if (_b && !_b.done && (_c = _a.return)) _c.call(_a);
+                            }
+                            finally { if (e_1) throw e_1.error; }
+                        }
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "list",
+                                source: "storage",
+                                input: {},
+                                output: {
+                                    result: result,
+                                },
+                                context: {
+                                    storageName: this.params.storageName,
+                                }
+                            })];
+                    case 1:
+                        _d.sent();
+                        return [2 /*return*/, result];
                 }
-                catch (e_1_1) { e_1 = { error: e_1_1 }; }
-                finally {
-                    try {
-                        if (_b && !_b.done && (_c = _a.return)) _c.call(_a);
-                    }
-                    finally { if (e_1) throw e_1.error; }
+            });
+        }); };
+        /**
+         * Disposes of the state.
+         * @returns {Promise<void>}
+         */
+        this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
+            var _a;
+            return __generator(this, function (_b) {
+                this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " dispose"));
+                if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onDispose) {
+                    this.params.callbacks.onDispose(this.params.clientId, this.params.storageName);
                 }
-                return [2 /*return*/, result];
+                return [2 /*return*/];
             });
         }); };
         this.params.logger.debug("ClientStorage storageName=".concat(this.params.storageName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " CTOR"), {
             params: params,
         });
+        if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onInit) {
+            this.params.callbacks.onInit(this.params.clientId, this.params.storageName);
+        }
     }
     return ClientStorage;
 }());
@@ -4656,6 +5137,7 @@ var StorageConnectionService = /** @class */ (function () {
     function StorageConnectionService() {
         var _this = this;
         this.loggerService = inject(TYPES.loggerService);
+        this.busService = inject(TYPES.busService);
         this.contextService = inject(TYPES.contextService);
         this.storageSchemaService = inject(TYPES.storageSchemaService);
         this.sessionValidationService = inject(TYPES.sessionValidationService);
@@ -4675,7 +5157,7 @@ var StorageConnectionService = /** @class */ (function () {
             if (!shared) {
                 throw new Error("agent-swarm storage not shared storageName=".concat(storageName));
             }
-            return new ClientStorage(__assign(__assign({ clientId: clientId, storageName: storageName, embedding: embeddingName, calculateSimilarity: calculateSimilarity, createEmbedding: createEmbedding, createIndex: createIndex, getData: getData, shared: shared, logger: _this.loggerService }, embedding), { callbacks: callbacks }));
+            return new ClientStorage(__assign(__assign({ clientId: clientId, storageName: storageName, embedding: embeddingName, calculateSimilarity: calculateSimilarity, createEmbedding: createEmbedding, createIndex: createIndex, getData: getData, shared: shared, logger: _this.loggerService, bus: _this.busService }, embedding), { callbacks: callbacks }));
         });
         /**
          * Retrieves a storage instance based on client ID and storage name.
@@ -4693,7 +5175,7 @@ var StorageConnectionService = /** @class */ (function () {
                 return _this.getSharedStorage(clientId, storageName);
             }
             var _c = _this.embeddingSchemaService.get(embeddingName), calculateSimilarity = _c.calculateSimilarity, createEmbedding = _c.createEmbedding, embedding = _c.callbacks;
-            return new ClientStorage(__assign(__assign({ clientId: clientId, storageName: storageName, embedding: embeddingName, calculateSimilarity: calculateSimilarity, createEmbedding: createEmbedding, createIndex: createIndex, getData: getData, shared: shared, logger: _this.loggerService }, embedding), { callbacks: callbacks }));
+            return new ClientStorage(__assign(__assign({ clientId: clientId, storageName: storageName, embedding: embeddingName, calculateSimilarity: calculateSimilarity, createEmbedding: createEmbedding, createIndex: createIndex, getData: getData, shared: shared, logger: _this.loggerService, bus: _this.busService }, embedding), { callbacks: callbacks }));
         });
         /**
          * Retrieves a list of storage data based on a search query and total number of items.
@@ -4838,18 +5320,31 @@ var StorageConnectionService = /** @class */ (function () {
          * @returns {Promise<void>}
          */
         this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
-            var key;
+            var key, storage;
             return __generator(this, function (_a) {
-                this.loggerService.log("storageConnectionService dispose", {
-                    context: this.contextService.context,
-                });
-                key = "".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.storageName);
-                if (!this.getStorage.has(key)) {
-                    return [2 /*return*/];
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("storageConnectionService dispose", {
+                            context: this.contextService.context,
+                        });
+                        key = "".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.storageName);
+                        if (!this.getStorage.has(key)) {
+                            return [2 /*return*/];
+                        }
+                        if (!!this.getSharedStorage.has(this.contextService.context.storageName)) return [3 /*break*/, 3];
+                        storage = this.getSharedStorage(this.contextService.context.clientId, this.contextService.context.storageName);
+                        return [4 /*yield*/, storage.waitForInit()];
+                    case 1:
+                        _a.sent();
+                        return [4 /*yield*/, storage.dispose()];
+                    case 2:
+                        _a.sent();
+                        _a.label = 3;
+                    case 3:
+                        this.getStorage.clear(key);
+                        this.sessionValidationService.removeStorageUsage(this.contextService.context.clientId, this.contextService.context.storageName);
+                        return [2 /*return*/];
                 }
-                this.getStorage.clear(key);
-                this.sessionValidationService.removeStorageUsage(this.contextService.context.clientId, this.contextService.context.storageName);
-                return [2 /*return*/];
             });
         }); };
     }
@@ -5268,12 +5763,12 @@ var ClientState = /** @class */ (function () {
             return __generator(this, function (_c) {
                 switch (_c.label) {
                     case 0:
-                        this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " waitForInit"));
+                        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " waitForInit"));
                         _a = this;
                         return [4 /*yield*/, this.params.getState(this.params.clientId, this.params.stateName)];
                     case 1:
                         _a._state = _c.sent();
-                        this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " waitForInit output"), { initialState: this._state });
+                        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " waitForInit output"), { initialState: this._state });
                         if ((_b = this.params.callbacks) === null || _b === void 0 ? void 0 : _b.onLoad) {
                             this.params.callbacks.onLoad(this._state, this.params.clientId, this.params.stateName);
                         }
@@ -5292,7 +5787,7 @@ var ClientState = /** @class */ (function () {
             return __generator(this, function (_b) {
                 switch (_b.label) {
                     case 0:
-                        this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " setState"));
+                        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " setState"));
                         return [4 /*yield*/, this.dispatch("write", function (currentState) { return __awaiter(_this, void 0, void 0, function () {
                                 var _a, _b, middleware, e_1_1;
                                 var e_1, _c;
@@ -5330,12 +5825,25 @@ var ClientState = /** @class */ (function () {
                             }); })];
                     case 1:
                         _b.sent();
-                        this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " setState output"), { pendingState: this._state });
+                        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " setState output"), { pendingState: this._state });
                         this.params.setState &&
                             this.params.setState(this._state, this.params.clientId, this.params.stateName);
                         if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onWrite) {
                             this.params.callbacks.onWrite(this._state, this.params.clientId, this.params.stateName);
                         }
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "set-state",
+                                source: "state",
+                                input: {},
+                                output: {
+                                    state: this._state,
+                                },
+                                context: {
+                                    stateName: this.params.stateName,
+                                }
+                            })];
+                    case 2:
+                        _b.sent();
                         return [2 /*return*/, this._state];
                 }
             });
@@ -5349,13 +5857,26 @@ var ClientState = /** @class */ (function () {
             return __generator(this, function (_b) {
                 switch (_b.label) {
                     case 0:
-                        this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " getState"));
+                        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " getState"));
                         return [4 /*yield*/, this.dispatch("read")];
                     case 1:
                         _b.sent();
                         if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onRead) {
                             this.params.callbacks.onRead(this._state, this.params.clientId, this.params.stateName);
                         }
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "get-state",
+                                source: "state",
+                                input: {},
+                                output: {
+                                    state: this._state,
+                                },
+                                context: {
+                                    stateName: this.params.stateName,
+                                }
+                            })];
+                    case 2:
+                        _b.sent();
                         return [2 /*return*/, this._state];
                 }
             });
@@ -5367,13 +5888,16 @@ var ClientState = /** @class */ (function () {
         this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
             var _a;
             return __generator(this, function (_b) {
-                this.params.logger.debug("ClientStorage stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " dispose"));
+                this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " dispose"));
                 if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onDispose) {
                     this.params.callbacks.onDispose(this.params.clientId, this.params.stateName);
                 }
                 return [2 /*return*/];
             });
         }); };
+        this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " CTOR"), {
+            params: params,
+        });
         if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onInit) {
             this.params.callbacks.onInit(this.params.clientId, this.params.stateName);
         }
@@ -5390,9 +5914,49 @@ var StateConnectionService = /** @class */ (function () {
     function StateConnectionService() {
         var _this = this;
         this.loggerService = inject(TYPES.loggerService);
+        this.busService = inject(TYPES.busService);
         this.contextService = inject(TYPES.contextService);
         this.stateSchemaService = inject(TYPES.stateSchemaService);
         this.sessionValidationService = inject(TYPES.sessionValidationService);
+        /**
+         * Memoized function to get a shared state reference.
+         * @param {string} clientId - The client ID.
+         * @param {StateName} stateName - The state name.
+         * @returns {ClientState} The client state.
+         */
+        this.getSharedStateRef = functoolsKit.memoize(function (_a) {
+            var _b = __read(_a, 2), stateName = _b[1];
+            return "".concat(stateName);
+        }, function (clientId, stateName) {
+            _this.sessionValidationService.addStateUsage(clientId, stateName);
+            var _a = _this.stateSchemaService.get(stateName), getState = _a.getState, setState = _a.setState, _b = _a.middlewares, middlewares = _b === void 0 ? [] : _b, shared = _a.shared, callbacks = _a.callbacks;
+            if (!shared) {
+                throw new Error("agent-swarm state not shared stateName=".concat(stateName));
+            }
+            return new ClientState({
+                clientId: clientId,
+                stateName: stateName,
+                logger: _this.loggerService,
+                bus: _this.busService,
+                setState: setState
+                    ? functoolsKit.queued(function () {
+                        var args = [];
+                        for (var _i = 0; _i < arguments.length; _i++) {
+                            args[_i] = arguments[_i];
+                        }
+                        return __awaiter(_this, void 0, void 0, function () { return __generator(this, function (_a) {
+                            switch (_a.label) {
+                                case 0: return [4 /*yield*/, setState.apply(void 0, __spreadArray([], __read(args), false))];
+                                case 1: return [2 /*return*/, _a.sent()];
+                            }
+                        }); });
+                    })
+                    : setState,
+                getState: getState,
+                middlewares: middlewares,
+                callbacks: callbacks,
+            });
+        });
         /**
          * Memoized function to get a state reference.
          * @param {string} clientId - The client ID.
@@ -5404,11 +5968,15 @@ var StateConnectionService = /** @class */ (function () {
             return "".concat(clientId, "-").concat(stateName);
         }, function (clientId, stateName) {
             _this.sessionValidationService.addStateUsage(clientId, stateName);
-            var _a = _this.stateSchemaService.get(stateName), getState = _a.getState, setState = _a.setState, _b = _a.middlewares, middlewares = _b === void 0 ? [] : _b, callbacks = _a.callbacks;
+            var _a = _this.stateSchemaService.get(stateName), getState = _a.getState, setState = _a.setState, _b = _a.middlewares, middlewares = _b === void 0 ? [] : _b, callbacks = _a.callbacks, _c = _a.shared, shared = _c === void 0 ? false : _c;
+            if (shared) {
+                return _this.getSharedStateRef(clientId, stateName);
+            }
             return new ClientState({
                 clientId: clientId,
                 stateName: stateName,
                 logger: _this.loggerService,
+                bus: _this.busService,
                 setState: setState
                     ? functoolsKit.queued(function () {
                         var args = [];
@@ -5487,6 +6055,7 @@ var StateConnectionService = /** @class */ (function () {
                         if (!this.getStateRef.has(key)) {
                             return [2 /*return*/];
                         }
+                        if (!!this.getSharedStateRef.has(this.contextService.context.stateName)) return [3 /*break*/, 3];
                         state = this.getStateRef(this.contextService.context.clientId, this.contextService.context.stateName);
                         return [4 /*yield*/, state.waitForInit()];
                     case 1:
@@ -5494,6 +6063,8 @@ var StateConnectionService = /** @class */ (function () {
                         return [4 /*yield*/, state.dispose()];
                     case 2:
                         _a.sent();
+                        _a.label = 3;
+                    case 3:
                         this.getStateRef.clear(key);
                         this.sessionValidationService.removeStateUsage(this.contextService.context.clientId, this.contextService.context.stateName);
                         return [2 /*return*/];
@@ -5613,7 +6184,103 @@ var StatePublicService = /** @class */ (function () {
     return StatePublicService;
 }());
 
+var BusService = /** @class */ (function () {
+    function BusService() {
+        var _this = this;
+        this.loggerService = inject(TYPES.loggerService);
+        this._eventSourceSet = new Set();
+        this.getEventSubject = functoolsKit.memoize(function (_a) {
+            var _b = __read(_a, 2), clientId = _b[0], source = _b[1];
+            return "".concat(clientId, "-").concat(source);
+        }, function () { return new functoolsKit.Subject(); });
+        /**
+         * Subscribes to events for a specific client and source.
+         * @param {string} clientId - The client ID.
+         * @param {EventSource} source - The event source.
+         * @param {(event: T) => void} fn - The callback function to handle the event.
+         * @returns {Subscription} The subscription object.
+         */
+        this.subscribe = function (clientId, source, fn) {
+            _this.loggerService.log("busService subscribe", {
+                clientId: clientId,
+                source: source,
+            });
+            _this._eventSourceSet.add(source);
+            return _this.getEventSubject(clientId, source).subscribe(fn);
+        };
+        /**
+         * Subscribes to a single event for a specific client and source.
+         * @param {string} clientId - The client ID.
+         * @param {EventSource} source - The event source.
+         * @param {(event: T) => boolean} filterFn - The filter function to determine if the event should be handled.
+         * @param {(event: T) => void} fn - The callback function to handle the event.
+         * @returns {Subscription} The subscription object.
+         */
+        this.once = function (clientId, source, filterFn, fn) {
+            _this.loggerService.log("busService once", {
+                clientId: clientId,
+                source: source,
+            });
+            _this._eventSourceSet.add(source);
+            return _this.getEventSubject(clientId, source).filter(filterFn).once(fn);
+        };
+        /**
+         * Emits an event for a specific client.
+         * @param {string} clientId - The client ID.
+         * @param {T} event - The event to emit.
+         * @returns {Promise<void>} A promise that resolves when the event has been emitted.
+         */
+        this.emit = function (clientId, event) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.debug("busService emit", {
+                            clientId: clientId,
+                            event: event,
+                        });
+                        if (!this.getEventSubject.has("".concat(clientId, "-").concat(event.source))) {
+                            return [2 /*return*/];
+                        }
+                        return [4 /*yield*/, this.getEventSubject(clientId, event.source).next(event)];
+                    case 1:
+                        _a.sent();
+                        return [2 /*return*/];
+                }
+            });
+        }); };
+        /**
+         * Disposes of all event subscriptions for a specific client.
+         * @param {string} clientId - The client ID.
+         */
+        this.dispose = function (clientId) {
+            var e_1, _a;
+            _this.loggerService.log("busService dispose", {
+                clientId: clientId,
+            });
+            try {
+                for (var _b = __values(_this._eventSourceSet), _c = _b.next(); !_c.done; _c = _b.next()) {
+                    var source = _c.value;
+                    var key = "".concat(clientId, "-").concat(source);
+                    if (_this.getEventSubject.has(key)) {
+                        _this.getEventSubject(clientId, source).unsubscribeAll();
+                    }
+                    _this.getEventSubject.clear("".concat(clientId, "-").concat(source));
+                }
+            }
+            catch (e_1_1) { e_1 = { error: e_1_1 }; }
+            finally {
+                try {
+                    if (_c && !_c.done && (_a = _b.return)) _a.call(_b);
+                }
+                finally { if (e_1) throw e_1.error; }
+            }
+        };
+    }
+    return BusService;
+}());
+
 {
+    provide(TYPES.busService, function () { return new BusService(); });
     provide(TYPES.loggerService, function () { return new LoggerService(); });
     provide(TYPES.contextService, function () { return new ContextService(); });
 }
@@ -5659,6 +6326,7 @@ var StatePublicService = /** @class */ (function () {
 }
 
 var baseServices = {
+    busService: inject(TYPES.busService),
     loggerService: inject(TYPES.loggerService),
     contextService: inject(TYPES.contextService),
 };
@@ -5768,10 +6436,15 @@ var addTool = function (toolSchema) {
  * @returns {string} The name of the added state.
  */
 var addState = function (stateSchema) {
-    swarm.loggerService.log('function addState', {
-        stateSchema: stateSchema
+    swarm.loggerService.log("function addState", {
+        stateSchema: stateSchema,
     });
     swarm.stateSchemaService.register(stateSchema.stateName, stateSchema);
+    if (stateSchema.shared) {
+        swarm.stateConnectionService
+            .getSharedStateRef("shared", stateSchema.stateName)
+            .waitForInit();
+    }
     return stateSchema.stateName;
 };
 
@@ -6111,69 +6784,64 @@ var disposeConnection = function (clientId, swarmName) { return __awaiter(void 0
                 });
                 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*/];
-                                }
-                            });
-                        }); }),
-                    ])];
+                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*/];
-                                }
-                            });
-                        }); }),
-                    ])];
+                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*/];
-                                }
-                            });
-                        }); }),
-                    ])];
+                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)];
@@ -6841,6 +7509,35 @@ var executeForce = function (content, clientId) { return __awaiter(void 0, void
     });
 }); };
 
+var DISALLOWED_EVENT_SOURCE_LIST$1 = new Set([
+    "agent",
+    "history",
+    "session",
+    "state",
+    "storage",
+    "swarm",
+]);
+/**
+ * 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.
+ * @returns {void} - Returns nothing.
+ */
+var listenEvent = function (clientId, topicName, fn) {
+    swarm.loggerService.log("function listenEvent", {
+        clientId: clientId,
+    });
+    if (DISALLOWED_EVENT_SOURCE_LIST$1.has(topicName)) {
+        throw new Error("agent-swarm listenEvent topic is reserved topicName=".concat(topicName));
+    }
+    return swarm.busService.subscribe(clientId, topicName, function (_a) {
+        var payload = _a.payload;
+        return fn(payload);
+    });
+};
+
 /**
  * Retrieves the last message sent by the user from the client's message history.
  *
@@ -7023,6 +7720,184 @@ var makeAutoDispose = function (clientId, swarmName, _a) {
     };
 };
 
+var DISALLOWED_EVENT_SOURCE_LIST = new Set([
+    "agent",
+    "history",
+    "session",
+    "state",
+    "storage",
+    "swarm",
+]);
+/**
+ * Emits an event to the swarm bus service.
+ *
+ * @template T - The type of the payload.
+ * @param {string} clientId - The ID of the client emitting the event.
+ * @param {T} payload - The payload of the event.
+ * @returns {boolean} - Returns true if the event was successfully emitted.
+ */
+var event = function (clientId, topicName, payload) {
+    swarm.loggerService.log("function listenEvent", {
+        clientId: clientId,
+    });
+    if (DISALLOWED_EVENT_SOURCE_LIST.has(topicName)) {
+        throw new Error("agent-swarm event topic is reserved topicName=".concat(topicName));
+    }
+    return swarm.busService.emit(clientId, {
+        source: topicName,
+        payload: payload,
+    });
+};
+
+/**
+ * Cancel the await of output by emit of empty string
+ *
+ * @param {string} clientId - The ID of the client.
+ * @param {string} agentName - The name of the agent.
+ * @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;
+    return __generator(this, function (_a) {
+        switch (_a.label) {
+            case 0:
+                swarm.loggerService.log('function cancelOutput', {
+                    clientId: clientId,
+                    agentName: agentName,
+                });
+                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)];
+            case 1:
+                currentAgentName = _a.sent();
+                if (currentAgentName !== agentName) {
+                    swarm.loggerService.log('function "cancelOutput" skipped due to the agent change', {
+                        currentAgentName: currentAgentName,
+                        agentName: agentName,
+                        clientId: clientId,
+                    });
+                    return [2 /*return*/];
+                }
+                return [4 /*yield*/, swarm.swarmPublicService.cancelOutput(clientId, swarmName)];
+            case 2:
+                _a.sent();
+                return [2 /*return*/];
+        }
+    });
+}); };
+
+/**
+ * Cancel the await of output by emit of empty string without checking active agent
+ *
+ * @param {string} clientId - The ID of the client.
+ * @param {string} agentName - The name of the agent.
+ * @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;
+    return __generator(this, function (_a) {
+        switch (_a.label) {
+            case 0:
+                swarm.loggerService.log('function cancelOutputForce', {
+                    clientId: clientId,
+                });
+                swarm.sessionValidationService.validate(clientId, "cancelOutputForce");
+                swarmName = swarm.sessionValidationService.getSwarm(clientId);
+                swarm.swarmValidationService.validate(swarmName, "cancelOutputForce");
+                return [4 /*yield*/, swarm.swarmPublicService.cancelOutput(clientId, swarmName)];
+            case 1:
+                _a.sent();
+                return [2 /*return*/];
+        }
+    });
+}); };
+
+/**
+ * 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.
+ * @returns {function} - A function to unsubscribe from the event.
+ */
+var listenAgentEvent = function (clientId, fn) {
+    swarm.loggerService.log("middleware listenAgentEvent", {
+        clientId: clientId,
+    });
+    return swarm.busService.subscribe(clientId, "agent", fn);
+};
+
+/**
+ * Hook to subscribe to history events for a specific client.
+ *
+ * @param {string} clientId - The ID of the client to subscribe to.
+ * @param {(event: IBaseEvent) => void} fn - The callback function to handle the event.
+ * @returns {Function} - The unsubscribe function.
+ */
+var listenHistoryEvent = function (clientId, fn) {
+    swarm.loggerService.log("middleware listenHistoryEvent", {
+        clientId: clientId,
+    });
+    return swarm.busService.subscribe(clientId, "history", fn);
+};
+
+/**
+ * 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.
+ * @returns {function} - The unsubscribe function to stop listening to session events.
+ */
+var listenSessionEvent = function (clientId, fn) {
+    swarm.loggerService.log("middleware listenSessionEvent", {
+        clientId: clientId,
+    });
+    return swarm.busService.subscribe(clientId, "session", fn);
+};
+
+/**
+ * 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.
+ * @returns {function} - The unsubscribe function to stop listening to the events.
+ */
+var listenStateEvent = function (clientId, fn) {
+    swarm.loggerService.log("middleware listenStateEvent", {
+        clientId: clientId,
+    });
+    return swarm.busService.subscribe(clientId, "state", fn);
+};
+
+/**
+ * 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.
+ * @returns {function} - A function to unsubscribe from the storage events.
+ */
+var listenStorageEvent = function (clientId, fn) {
+    swarm.loggerService.log("middleware listenStorageEvent", {
+        clientId: clientId,
+    });
+    return swarm.busService.subscribe(clientId, "storage", fn);
+};
+
+/**
+ * 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: IBaseEvent) => void} fn - The callback function to handle the event.
+ * @returns {Function} - A function to unsubscribe from the event.
+ */
+var listenSwarmEvent = function (clientId, fn) {
+    swarm.loggerService.log("middleware listenSwarmEvent", {
+        clientId: clientId,
+    });
+    return swarm.busService.subscribe(clientId, "swarm", fn);
+};
+
 var StorageUtils = /** @class */ (function () {
     function StorageUtils() {
         var _this = this;
@@ -7279,6 +8154,8 @@ exports.addState = addState;
 exports.addStorage = addStorage;
 exports.addSwarm = addSwarm;
 exports.addTool = addTool;
+exports.cancelOutput = cancelOutput;
+exports.cancelOutputForce = cancelOutputForce;
 exports.changeAgent = changeAgent;
 exports.commitFlush = commitFlush;
 exports.commitFlushForce = commitFlushForce;
@@ -7292,6 +8169,7 @@ exports.complete = complete;
 exports.disposeConnection = disposeConnection;
 exports.emit = emit;
 exports.emitForce = emitForce;
+exports.event = event;
 exports.execute = execute;
 exports.executeForce = executeForce;
 exports.getAgentHistory = getAgentHistory;
@@ -7303,6 +8181,13 @@ exports.getLastUserMessage = getLastUserMessage;
 exports.getRawHistory = getRawHistory;
 exports.getSessionMode = getSessionMode;
 exports.getUserHistory = getUserHistory;
+exports.listenAgentEvent = listenAgentEvent;
+exports.listenEvent = listenEvent;
+exports.listenHistoryEvent = listenHistoryEvent;
+exports.listenSessionEvent = listenSessionEvent;
+exports.listenStateEvent = listenStateEvent;
+exports.listenStorageEvent = listenStorageEvent;
+exports.listenSwarmEvent = listenSwarmEvent;
 exports.makeAutoDispose = makeAutoDispose;
 exports.makeConnection = makeConnection;
 exports.session = session;