agent-swarm-kit 1.0.69 → 1.0.71

package/build/index.mjs

@@ -225,6 +225,7 @@ 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'),
 };
@@ -951,7 +952,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();
@@ -968,12 +969,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*/];
                 }
@@ -1105,6 +1138,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*/];
                 }
@@ -1128,6 +1174,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*/];
                 }
@@ -1144,6 +1201,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*/];
                 }
@@ -1168,6 +1236,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*/];
                 }
@@ -1196,6 +1277,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*/];
                 }
@@ -1409,6 +1504,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);
@@ -1439,7 +1535,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.
@@ -1617,6 +1713,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*/];
                 }
@@ -1850,6 +1959,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);
         /**
@@ -1866,6 +1976,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,
             });
@@ -1993,10 +2104,44 @@ var ClientSwarm = /** @class */ (function () {
         this.params = params;
         this._agentChangedSubject = new Subject();
         this._activeAgent = AGENT_NEED_FETCH;
+        this._cancelOutputSubject = new 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 = queued(function () { return __awaiter(_this, void 0, void 0, function () {
             var _a, awaiter, resolve, getOutput, handleOutput, un, _b, agentName, output, expectAgent;
@@ -2010,18 +2155,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()];
                                 }
                             });
@@ -2044,7 +2194,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];
                 }
             });
@@ -2065,7 +2229,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];
                 }
             });
         }); };
@@ -2074,7 +2251,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:
@@ -2082,7 +2259,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];
                 }
             });
         }); };
@@ -2101,8 +2292,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*/];
                 }
@@ -2120,6 +2325,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*/];
                 }
@@ -2147,6 +2365,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);
@@ -2182,6 +2401,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) {
@@ -2201,6 +2421,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.
@@ -2395,6 +2631,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*/];
                 }
@@ -2425,6 +2674,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];
                 }
             });
@@ -2436,7 +2701,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:
@@ -2448,7 +2713,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];
                 }
             });
         }); };
@@ -2458,7 +2739,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:
@@ -2469,7 +2750,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];
                 }
             });
         }); };
@@ -2478,7 +2774,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:
@@ -2487,7 +2783,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];
                 }
             });
         }); };
@@ -2497,7 +2806,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:
@@ -2508,7 +2817,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];
                 }
             });
         }); };
@@ -2539,6 +2863,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;
@@ -2590,6 +2923,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);
@@ -2604,7 +2938,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.
@@ -3540,6 +3874,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.
@@ -4414,6 +4781,7 @@ var ClientStorage = /** @class */ (function () {
      */
     function ClientStorage(params) {
         var _this = this;
+        var _a;
         this.params = params;
         this._itemMap = new Map();
         /**
@@ -4532,6 +4900,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)];
                     }
                 });
@@ -4558,6 +4943,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*/];
                 }
             });
@@ -4570,15 +4968,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*/];
             });
         }); };
         /**
@@ -4587,10 +5001,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*/];
+                }
             });
         }); };
         /**
@@ -4599,12 +5027,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];
+                }
             });
         }); };
         /**
@@ -4616,32 +5064,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;
 }());
@@ -4654,6 +5135,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);
@@ -4673,7 +5155,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.
@@ -4691,7 +5173,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.
@@ -4836,18 +5318,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*/];
             });
         }); };
     }
@@ -5266,12 +5761,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);
                         }
@@ -5290,7 +5785,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;
@@ -5328,12 +5823,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];
                 }
             });
@@ -5347,13 +5855,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];
                 }
             });
@@ -5365,13 +5886,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);
         }
@@ -5388,9 +5912,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 = 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
+                    ? 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.
@@ -5402,11 +5966,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
                     ? queued(function () {
                         var args = [];
@@ -5485,6 +6053,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:
@@ -5492,6 +6061,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*/];
@@ -5611,7 +6182,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 = memoize(function (_a) {
+            var _b = __read(_a, 2), clientId = _b[0], source = _b[1];
+            return "".concat(clientId, "-").concat(source);
+        }, function () { return new 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(); });
 }
@@ -5657,6 +6324,7 @@ var StatePublicService = /** @class */ (function () {
 }
 
 var baseServices = {
+    busService: inject(TYPES.busService),
     loggerService: inject(TYPES.loggerService),
     contextService: inject(TYPES.contextService),
 };
@@ -5766,10 +6434,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;
 };
 
@@ -6109,69 +6782,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)];
@@ -6839,6 +7507,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.
  *
@@ -7021,6 +7718,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;
@@ -7264,4 +8139,4 @@ var StateUtils = /** @class */ (function () {
  */
 var State = new StateUtils();
 
-export { ContextService, History, HistoryAdapter, HistoryInstance, State, Storage, addAgent, addCompletion, addEmbedding, addState, addStorage, addSwarm, addTool, changeAgent, commitFlush, commitFlushForce, commitSystemMessage, commitSystemMessageForce, commitToolOutput, commitToolOutputForce, commitUserMessage, commitUserMessageForce, complete, disposeConnection, emit, emitForce, execute, executeForce, getAgentHistory, getAgentName, getAssistantHistory, getLastAssistantMessage, getLastSystemMessage, getLastUserMessage, getRawHistory, getSessionMode, getUserHistory, makeAutoDispose, makeConnection, session, setConfig, swarm };
+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 };