agent-swarm-kit 1.0.32 → 1.0.33

package/build/index.cjs

@@ -456,11 +456,16 @@ var ClientAgent = /** @class */ (function () {
                         if ((validation = _a.sent())) {
                             throw new Error("agent-swarm-kit ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " model ressurect failed: ").concat(validation));
                         }
+                        this.params.onOutput &&
+                            this.params.onOutput(this.params.clientId, this.params.agentName, result_1);
                         return [4 /*yield*/, this._outputSubject.next(result_1)];
                     case 4:
                         _a.sent();
                         return [2 /*return*/];
-                    case 5: return [4 /*yield*/, this._outputSubject.next(result)];
+                    case 5:
+                        this.params.onOutput &&
+                            this.params.onOutput(this.params.clientId, this.params.agentName, result);
+                        return [4 /*yield*/, this._outputSubject.next(result)];
                     case 6:
                         _a.sent();
                         return [2 /*return*/];
@@ -479,6 +484,8 @@ var ClientAgent = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " _resurrectModel"));
+                        this.params.onResurrect &&
+                            this.params.onResurrect(this.params.clientId, this.params.agentName, mode, reason);
                         return [4 /*yield*/, this.params.history.push({
                                 role: "resque",
                                 mode: "tool",
@@ -540,7 +547,7 @@ var ClientAgent = /** @class */ (function () {
          * @returns {Promise<IModelMessage>}
          */
         this.getCompletion = function (mode) { return __awaiter(_this, void 0, void 0, function () {
-            var messages;
+            var messages, args, output;
             var _a;
             return __generator(this, function (_b) {
                 switch (_b.label) {
@@ -549,16 +556,21 @@ var ClientAgent = /** @class */ (function () {
                         return [4 /*yield*/, this.params.history.toArrayForAgent(this.params.prompt, this.params.system)];
                     case 1:
                         messages = _b.sent();
-                        return [4 /*yield*/, this.params.completion.getCompletion({
-                                clientId: this.params.clientId,
-                                agentName: this.params.agentName,
-                                messages: messages,
-                                mode: mode,
-                                tools: (_a = this.params.tools) === null || _a === void 0 ? void 0 : _a.map(function (t) {
-                                    return lodashEs.omit(t, "toolName", "call", "validate");
-                                }),
-                            })];
-                    case 2: return [2 /*return*/, _b.sent()];
+                        args = {
+                            clientId: this.params.clientId,
+                            agentName: this.params.agentName,
+                            messages: messages,
+                            mode: mode,
+                            tools: (_a = this.params.tools) === null || _a === void 0 ? void 0 : _a.map(function (t) {
+                                return lodashEs.omit(t, "toolName", "call", "validate", "onCall", "onValidate");
+                            }),
+                        };
+                        return [4 /*yield*/, this.params.completion.getCompletion(args)];
+                    case 2:
+                        output = _b.sent();
+                        this.params.completion.onComplete &&
+                            this.params.completion.onComplete(args, output);
+                        return [2 /*return*/, output];
                 }
             });
         }); };
@@ -572,6 +584,8 @@ var ClientAgent = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " commitUserMessage"), { message: message });
+                        this.params.onUserMessage &&
+                            this.params.onUserMessage(this.params.clientId, this.params.agentName, message);
                         return [4 /*yield*/, this.params.history.push({
                                 role: "user",
                                 agentName: this.params.agentName,
@@ -593,6 +607,8 @@ var ClientAgent = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " commitFlush"));
+                        this.params.onFlush &&
+                            this.params.onFlush(this.params.clientId, this.params.agentName);
                         return [4 /*yield*/, this.params.history.push({
                                 role: "flush",
                                 agentName: this.params.agentName,
@@ -615,6 +631,8 @@ var ClientAgent = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " commitSystemMessage"), { message: message });
+                        this.params.onSystemMessage &&
+                            this.params.onSystemMessage(this.params.clientId, this.params.agentName, message);
                         return [4 /*yield*/, this.params.history.push({
                                 role: "system",
                                 agentName: this.params.agentName,
@@ -637,6 +655,8 @@ var ClientAgent = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " commitToolOutput"), { content: content });
+                        this.params.onToolOutput &&
+                            this.params.onToolOutput(this.params.clientId, this.params.agentName, content);
                         return [4 /*yield*/, this.params.history.push({
                                 role: "tool",
                                 agentName: this.params.agentName,
@@ -665,6 +685,8 @@ var ClientAgent = /** @class */ (function () {
                 switch (_e.label) {
                     case 0:
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " execute begin"), { incoming: incoming, mode: mode });
+                        this.params.onExecute &&
+                            this.params.onExecute(this.params.clientId, this.params.agentName, incoming, mode);
                         return [4 /*yield*/, this.params.history.push({
                                 role: "user",
                                 mode: mode,
@@ -697,7 +719,9 @@ var ClientAgent = /** @class */ (function () {
                                     case 3:
                                         _f.sent();
                                         return [2 /*return*/, { value: void 0 }];
-                                    case 4: return [4 /*yield*/, functoolsKit.not(targetFn.validate(this_1.params.clientId, this_1.params.agentName, tool.function.arguments))];
+                                    case 4:
+                                        targetFn.onValidate && targetFn.onValidate(this_1.params.clientId, this_1.params.agentName, tool.function.arguments);
+                                        return [4 /*yield*/, functoolsKit.not(targetFn.validate(this_1.params.clientId, this_1.params.agentName, tool.function.arguments))];
                                     case 5:
                                         if (!_f.sent()) return [3 /*break*/, 8];
                                         this_1.params.logger.debug("ClientAgent agentName=".concat(this_1.params.agentName, " clientId=").concat(this_1.params.clientId, " functionName=").concat(tool.function.name, " tool validation not passed"));
@@ -710,6 +734,8 @@ var ClientAgent = /** @class */ (function () {
                                         _f.sent();
                                         return [2 /*return*/, { value: void 0 }];
                                     case 8:
+                                        targetFn.onCall &&
+                                            targetFn.onCall(this_1.params.clientId, this_1.params.agentName, tool.function.arguments);
                                         /**
                                          * @description Do not await to avoid deadlock! The tool can send the message to other agents by emulating user messages
                                          */
@@ -816,7 +842,7 @@ var AgentConnectionService = /** @class */ (function () {
             var _b = __read(_a, 2), clientId = _b[0], agentName = _b[1];
             return "".concat(clientId, "-").concat(agentName);
         }, function (clientId, agentName) {
-            var _a = _this.agentSchemaService.get(agentName), prompt = _a.prompt, system = _a.system, tools = _a.tools, completionName = _a.completion, _b = _a.validate, validate = _b === void 0 ? validateDefault : _b;
+            var _a = _this.agentSchemaService.get(agentName), prompt = _a.prompt, system = _a.system, tools = _a.tools, onExecute = _a.onExecute, onOutput = _a.onOutput, onResurrect = _a.onResurrect, onSystemMessage = _a.onSystemMessage, onToolOutput = _a.onToolOutput, onUserMessage = _a.onUserMessage, completionName = _a.completion, _b = _a.validate, validate = _b === void 0 ? validateDefault : _b;
             var completion = _this.completionSchemaService.get(completionName);
             _this.sessionValidationService.addAgentUsage(clientId, agentName);
             return new ClientAgent({
@@ -829,6 +855,12 @@ var AgentConnectionService = /** @class */ (function () {
                 system: system,
                 tools: tools === null || tools === void 0 ? void 0 : tools.map(_this.toolSchemaService.get),
                 completion: completion,
+                onExecute: onExecute,
+                onOutput: onOutput,
+                onResurrect: onResurrect,
+                onSystemMessage: onSystemMessage,
+                onToolOutput: onToolOutput,
+                onUserMessage: onUserMessage,
             });
         });
         /**
@@ -1459,7 +1491,7 @@ var SwarmConnectionService = /** @class */ (function () {
             return "".concat(clientId, "-").concat(swarmName);
         }, function (clientId, swarmName) {
             var e_1, _a;
-            var _b = _this.swarmSchemaService.get(swarmName), agentList = _b.agentList, defaultAgent = _b.defaultAgent;
+            var _b = _this.swarmSchemaService.get(swarmName), agentList = _b.agentList, defaultAgent = _b.defaultAgent, onAgentChanged = _b.onAgentChanged;
             var agentMap = {};
             try {
                 for (var agentList_1 = __values(agentList), agentList_1_1 = agentList_1.next(); !agentList_1_1.done; agentList_1_1 = agentList_1.next()) {
@@ -1484,7 +1516,9 @@ var SwarmConnectionService = /** @class */ (function () {
                     return __awaiter(this, void 0, void 0, function () {
                         return __generator(this, function (_a) {
                             switch (_a.label) {
-                                case 0: return [4 /*yield*/, GLOBAL_CONFIG.CC_SWARM_AGENT_CHANGED(clientId, agentName, swarmName)];
+                                case 0:
+                                    onAgentChanged && onAgentChanged(clientId, agentName, swarmName);
+                                    return [4 /*yield*/, GLOBAL_CONFIG.CC_SWARM_AGENT_CHANGED(clientId, agentName, swarmName)];
                                 case 1:
                                     _a.sent();
                                     return [2 /*return*/];
@@ -1679,6 +1713,8 @@ var ClientSession = /** @class */ (function () {
                         this.params.logger.debug("ClientSession clientId=".concat(this.params.clientId, " emit"), {
                             message: message,
                         });
+                        this.params.onEmit &&
+                            this.params.onEmit(this.params.clientId, this.params.swarmName, message);
                         return [4 /*yield*/, this._emitSubject.next(message)];
                     case 1:
                         _a.sent();
@@ -1701,6 +1737,8 @@ var ClientSession = /** @class */ (function () {
                             message: message,
                             mode: mode,
                         });
+                        this.params.onExecute &&
+                            this.params.onExecute(this.params.clientId, this.params.swarmName, message, mode);
                         return [4 /*yield*/, this.params.swarm.getAgent()];
                     case 1:
                         agent = _a.sent();
@@ -1801,6 +1839,8 @@ var ClientSession = /** @class */ (function () {
          */
         this.connect = function (connector) {
             _this.params.logger.debug("ClientSession clientId=".concat(_this.params.clientId, " connect"));
+            _this.params.onConnect &&
+                _this.params.onConnect(_this.params.clientId, _this.params.swarmName);
             _this._emitSubject.subscribe(function (data) { return __awaiter(_this, void 0, void 0, function () {
                 var _a;
                 var _b;
@@ -1859,6 +1899,7 @@ var SessionConnectionService = /** @class */ (function () {
         this.loggerService = inject(TYPES.loggerService);
         this.contextService = inject(TYPES.contextService);
         this.swarmConnectionService = inject(TYPES.swarmConnectionService);
+        this.swarmSchemaService = inject(TYPES.swarmSchemaService);
         /**
          * Retrieves a memoized session based on clientId and swarmName.
          * @param {string} clientId - The client ID.
@@ -1869,10 +1910,15 @@ var SessionConnectionService = /** @class */ (function () {
             var _b = __read(_a, 2), clientId = _b[0], swarmName = _b[1];
             return "".concat(clientId, "-").concat(swarmName);
         }, function (clientId, swarmName) {
+            var _a = _this.swarmSchemaService.get(swarmName), onConnect = _a.onConnect, onEmit = _a.onEmit, onExecute = _a.onExecute;
             return new ClientSession({
                 clientId: clientId,
                 logger: _this.loggerService,
                 swarm: _this.swarmConnectionService.getSwarm(clientId, swarmName),
+                swarmName: swarmName,
+                onConnect: onConnect,
+                onEmit: onEmit,
+                onExecute: onExecute,
             });
         });
         /**

package/build/index.mjs

@@ -454,11 +454,16 @@ var ClientAgent = /** @class */ (function () {
                         if ((validation = _a.sent())) {
                             throw new Error("agent-swarm-kit ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " model ressurect failed: ").concat(validation));
                         }
+                        this.params.onOutput &&
+                            this.params.onOutput(this.params.clientId, this.params.agentName, result_1);
                         return [4 /*yield*/, this._outputSubject.next(result_1)];
                     case 4:
                         _a.sent();
                         return [2 /*return*/];
-                    case 5: return [4 /*yield*/, this._outputSubject.next(result)];
+                    case 5:
+                        this.params.onOutput &&
+                            this.params.onOutput(this.params.clientId, this.params.agentName, result);
+                        return [4 /*yield*/, this._outputSubject.next(result)];
                     case 6:
                         _a.sent();
                         return [2 /*return*/];
@@ -477,6 +482,8 @@ var ClientAgent = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " _resurrectModel"));
+                        this.params.onResurrect &&
+                            this.params.onResurrect(this.params.clientId, this.params.agentName, mode, reason);
                         return [4 /*yield*/, this.params.history.push({
                                 role: "resque",
                                 mode: "tool",
@@ -538,7 +545,7 @@ var ClientAgent = /** @class */ (function () {
          * @returns {Promise<IModelMessage>}
          */
         this.getCompletion = function (mode) { return __awaiter(_this, void 0, void 0, function () {
-            var messages;
+            var messages, args, output;
             var _a;
             return __generator(this, function (_b) {
                 switch (_b.label) {
@@ -547,16 +554,21 @@ var ClientAgent = /** @class */ (function () {
                         return [4 /*yield*/, this.params.history.toArrayForAgent(this.params.prompt, this.params.system)];
                     case 1:
                         messages = _b.sent();
-                        return [4 /*yield*/, this.params.completion.getCompletion({
-                                clientId: this.params.clientId,
-                                agentName: this.params.agentName,
-                                messages: messages,
-                                mode: mode,
-                                tools: (_a = this.params.tools) === null || _a === void 0 ? void 0 : _a.map(function (t) {
-                                    return omit(t, "toolName", "call", "validate");
-                                }),
-                            })];
-                    case 2: return [2 /*return*/, _b.sent()];
+                        args = {
+                            clientId: this.params.clientId,
+                            agentName: this.params.agentName,
+                            messages: messages,
+                            mode: mode,
+                            tools: (_a = this.params.tools) === null || _a === void 0 ? void 0 : _a.map(function (t) {
+                                return omit(t, "toolName", "call", "validate", "onCall", "onValidate");
+                            }),
+                        };
+                        return [4 /*yield*/, this.params.completion.getCompletion(args)];
+                    case 2:
+                        output = _b.sent();
+                        this.params.completion.onComplete &&
+                            this.params.completion.onComplete(args, output);
+                        return [2 /*return*/, output];
                 }
             });
         }); };
@@ -570,6 +582,8 @@ var ClientAgent = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " commitUserMessage"), { message: message });
+                        this.params.onUserMessage &&
+                            this.params.onUserMessage(this.params.clientId, this.params.agentName, message);
                         return [4 /*yield*/, this.params.history.push({
                                 role: "user",
                                 agentName: this.params.agentName,
@@ -591,6 +605,8 @@ var ClientAgent = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " commitFlush"));
+                        this.params.onFlush &&
+                            this.params.onFlush(this.params.clientId, this.params.agentName);
                         return [4 /*yield*/, this.params.history.push({
                                 role: "flush",
                                 agentName: this.params.agentName,
@@ -613,6 +629,8 @@ var ClientAgent = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " commitSystemMessage"), { message: message });
+                        this.params.onSystemMessage &&
+                            this.params.onSystemMessage(this.params.clientId, this.params.agentName, message);
                         return [4 /*yield*/, this.params.history.push({
                                 role: "system",
                                 agentName: this.params.agentName,
@@ -635,6 +653,8 @@ var ClientAgent = /** @class */ (function () {
                 switch (_a.label) {
                     case 0:
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " commitToolOutput"), { content: content });
+                        this.params.onToolOutput &&
+                            this.params.onToolOutput(this.params.clientId, this.params.agentName, content);
                         return [4 /*yield*/, this.params.history.push({
                                 role: "tool",
                                 agentName: this.params.agentName,
@@ -663,6 +683,8 @@ var ClientAgent = /** @class */ (function () {
                 switch (_e.label) {
                     case 0:
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " execute begin"), { incoming: incoming, mode: mode });
+                        this.params.onExecute &&
+                            this.params.onExecute(this.params.clientId, this.params.agentName, incoming, mode);
                         return [4 /*yield*/, this.params.history.push({
                                 role: "user",
                                 mode: mode,
@@ -695,7 +717,9 @@ var ClientAgent = /** @class */ (function () {
                                     case 3:
                                         _f.sent();
                                         return [2 /*return*/, { value: void 0 }];
-                                    case 4: return [4 /*yield*/, not(targetFn.validate(this_1.params.clientId, this_1.params.agentName, tool.function.arguments))];
+                                    case 4:
+                                        targetFn.onValidate && targetFn.onValidate(this_1.params.clientId, this_1.params.agentName, tool.function.arguments);
+                                        return [4 /*yield*/, not(targetFn.validate(this_1.params.clientId, this_1.params.agentName, tool.function.arguments))];
                                     case 5:
                                         if (!_f.sent()) return [3 /*break*/, 8];
                                         this_1.params.logger.debug("ClientAgent agentName=".concat(this_1.params.agentName, " clientId=").concat(this_1.params.clientId, " functionName=").concat(tool.function.name, " tool validation not passed"));
@@ -708,6 +732,8 @@ var ClientAgent = /** @class */ (function () {
                                         _f.sent();
                                         return [2 /*return*/, { value: void 0 }];
                                     case 8:
+                                        targetFn.onCall &&
+                                            targetFn.onCall(this_1.params.clientId, this_1.params.agentName, tool.function.arguments);
                                         /**
                                          * @description Do not await to avoid deadlock! The tool can send the message to other agents by emulating user messages
                                          */
@@ -814,7 +840,7 @@ var AgentConnectionService = /** @class */ (function () {
             var _b = __read(_a, 2), clientId = _b[0], agentName = _b[1];
             return "".concat(clientId, "-").concat(agentName);
         }, function (clientId, agentName) {
-            var _a = _this.agentSchemaService.get(agentName), prompt = _a.prompt, system = _a.system, tools = _a.tools, completionName = _a.completion, _b = _a.validate, validate = _b === void 0 ? validateDefault : _b;
+            var _a = _this.agentSchemaService.get(agentName), prompt = _a.prompt, system = _a.system, tools = _a.tools, onExecute = _a.onExecute, onOutput = _a.onOutput, onResurrect = _a.onResurrect, onSystemMessage = _a.onSystemMessage, onToolOutput = _a.onToolOutput, onUserMessage = _a.onUserMessage, completionName = _a.completion, _b = _a.validate, validate = _b === void 0 ? validateDefault : _b;
             var completion = _this.completionSchemaService.get(completionName);
             _this.sessionValidationService.addAgentUsage(clientId, agentName);
             return new ClientAgent({
@@ -827,6 +853,12 @@ var AgentConnectionService = /** @class */ (function () {
                 system: system,
                 tools: tools === null || tools === void 0 ? void 0 : tools.map(_this.toolSchemaService.get),
                 completion: completion,
+                onExecute: onExecute,
+                onOutput: onOutput,
+                onResurrect: onResurrect,
+                onSystemMessage: onSystemMessage,
+                onToolOutput: onToolOutput,
+                onUserMessage: onUserMessage,
             });
         });
         /**
@@ -1457,7 +1489,7 @@ var SwarmConnectionService = /** @class */ (function () {
             return "".concat(clientId, "-").concat(swarmName);
         }, function (clientId, swarmName) {
             var e_1, _a;
-            var _b = _this.swarmSchemaService.get(swarmName), agentList = _b.agentList, defaultAgent = _b.defaultAgent;
+            var _b = _this.swarmSchemaService.get(swarmName), agentList = _b.agentList, defaultAgent = _b.defaultAgent, onAgentChanged = _b.onAgentChanged;
             var agentMap = {};
             try {
                 for (var agentList_1 = __values(agentList), agentList_1_1 = agentList_1.next(); !agentList_1_1.done; agentList_1_1 = agentList_1.next()) {
@@ -1482,7 +1514,9 @@ var SwarmConnectionService = /** @class */ (function () {
                     return __awaiter(this, void 0, void 0, function () {
                         return __generator(this, function (_a) {
                             switch (_a.label) {
-                                case 0: return [4 /*yield*/, GLOBAL_CONFIG.CC_SWARM_AGENT_CHANGED(clientId, agentName, swarmName)];
+                                case 0:
+                                    onAgentChanged && onAgentChanged(clientId, agentName, swarmName);
+                                    return [4 /*yield*/, GLOBAL_CONFIG.CC_SWARM_AGENT_CHANGED(clientId, agentName, swarmName)];
                                 case 1:
                                     _a.sent();
                                     return [2 /*return*/];
@@ -1677,6 +1711,8 @@ var ClientSession = /** @class */ (function () {
                         this.params.logger.debug("ClientSession clientId=".concat(this.params.clientId, " emit"), {
                             message: message,
                         });
+                        this.params.onEmit &&
+                            this.params.onEmit(this.params.clientId, this.params.swarmName, message);
                         return [4 /*yield*/, this._emitSubject.next(message)];
                     case 1:
                         _a.sent();
@@ -1699,6 +1735,8 @@ var ClientSession = /** @class */ (function () {
                             message: message,
                             mode: mode,
                         });
+                        this.params.onExecute &&
+                            this.params.onExecute(this.params.clientId, this.params.swarmName, message, mode);
                         return [4 /*yield*/, this.params.swarm.getAgent()];
                     case 1:
                         agent = _a.sent();
@@ -1799,6 +1837,8 @@ var ClientSession = /** @class */ (function () {
          */
         this.connect = function (connector) {
             _this.params.logger.debug("ClientSession clientId=".concat(_this.params.clientId, " connect"));
+            _this.params.onConnect &&
+                _this.params.onConnect(_this.params.clientId, _this.params.swarmName);
             _this._emitSubject.subscribe(function (data) { return __awaiter(_this, void 0, void 0, function () {
                 var _a;
                 var _b;
@@ -1857,6 +1897,7 @@ var SessionConnectionService = /** @class */ (function () {
         this.loggerService = inject(TYPES.loggerService);
         this.contextService = inject(TYPES.contextService);
         this.swarmConnectionService = inject(TYPES.swarmConnectionService);
+        this.swarmSchemaService = inject(TYPES.swarmSchemaService);
         /**
          * Retrieves a memoized session based on clientId and swarmName.
          * @param {string} clientId - The client ID.
@@ -1867,10 +1908,15 @@ var SessionConnectionService = /** @class */ (function () {
             var _b = __read(_a, 2), clientId = _b[0], swarmName = _b[1];
             return "".concat(clientId, "-").concat(swarmName);
         }, function (clientId, swarmName) {
+            var _a = _this.swarmSchemaService.get(swarmName), onConnect = _a.onConnect, onEmit = _a.onEmit, onExecute = _a.onExecute;
             return new ClientSession({
                 clientId: clientId,
                 logger: _this.loggerService,
                 swarm: _this.swarmConnectionService.getSwarm(clientId, swarmName),
+                swarmName: swarmName,
+                onConnect: onConnect,
+                onEmit: onEmit,
+                onExecute: onExecute,
             });
         });
         /**

package/package.json

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

package/types.d.ts

@@ -1,44 +1,40 @@
 import * as di_scoped from 'di-scoped';
-import { ExecutionMode as ExecutionMode$1 } from 'src/interfaces/Session.interface';
 import * as functools_kit from 'functools-kit';
 import { IPubsubArray, Subject } from 'functools-kit';
 
 /**
- * Interface representing a model message.
+ * Interface representing an incoming message.
  */
-interface IModelMessage {
+interface IIncomingMessage {
     /**
-     * The role of the message sender.
-     * @type {'assistant' | 'system' | 'tool' | 'user' | 'resque' | 'flush'}
+     * The ID of the client sending the message.
      */
-    role: "assistant" | "system" | "tool" | "user" | "resque" | "flush";
+    clientId: string;
+    /**
+     * The data contained in the message.
+     */
+    data: string;
     /**
      * The name of the agent sending the message.
-     * @type {string}
      */
-    agentName: string;
+    agentName: AgentName;
+}
+/**
+ * Interface representing an outgoing message.
+ */
+interface IOutgoingMessage {
     /**
-     * The content of the message.
-     * @type {string}
+     * The ID of the client receiving the message.
      */
-    content: string;
+    clientId: string;
     /**
-     * The source of message: tool or user
-     * @type {ExecutionMode}
+     * The data contained in the message.
      */
-    mode: ExecutionMode$1;
+    data: string;
     /**
-     * Optional tool calls associated with the message.
-     * @type {Array<{ function: { name: string; arguments: { [key: string]: any; }; }; }>}
+     * The name of the agent sending the message.
      */
-    tool_calls?: {
-        function: {
-            name: string;
-            arguments: {
-                [key: string]: any;
-            };
-        };
-    }[];
+    agentName: AgentName;
 }
 
 /**
@@ -57,163 +53,29 @@ interface ILogger {
     debug(...args: any[]): void;
 }
 
-/**
- * Interface representing the history of model messages.
- */
-interface IHistory {
+interface ISwarmSession {
     /**
-     * Pushes a message to the history.
-     * @param {IModelMessage} message - The message to push.
-     * @returns {Promise<void>}
-     */
-    push(message: IModelMessage): Promise<void>;
-    /**
-     * Converts the history to an array of messages for a specific agent.
-     * @param {string} prompt - The prompt to filter messages for the agent.
-     * @returns {Promise<IModelMessage[]>}
-     */
-    toArrayForAgent(prompt: string, system?: string[]): Promise<IModelMessage[]>;
-    /**
-     * Converts the history to an array of raw messages.
-     * @returns {Promise<IModelMessage[]>}
-     */
-    toArrayForRaw(): Promise<IModelMessage[]>;
-}
-/**
- * Interface representing the parameters required to create a history instance.
- */
-interface IHistoryParams extends IHistorySchema {
-    /**
-     * The name of the agent.
-     * @type {AgentName}
-     */
-    agentName: AgentName;
-    /**
-     * The client ID.
-     * @type {string}
-     */
-    clientId: string;
-    /**
-     * The logger instance.
-     * @type {ILogger}
-     */
-    logger: ILogger;
-}
-/**
- * Interface representing the schema of the history.
- */
-interface IHistorySchema {
-    /**
-     * The array of model messages.
-     * @type {IPubsubArray<IModelMessage>}
-     */
-    items: IPubsubArray<IModelMessage>;
-}
-
-/**
- * Represents a tool call with a function name and arguments.
- */
-interface IToolCall {
-    function: {
-        /**
-         * The name of the function to be called.
-         */
-        name: string;
-        /**
-         * The arguments to be passed to the function.
-         */
-        arguments: {
-            [key: string]: any;
-        };
-    };
-}
-/**
- * Represents a tool with a type and function details.
- */
-interface ITool {
-    /**
-     * The type of the tool.
-     */
-    type: string;
-    function: {
-        /**
-         * The name of the function.
-         */
-        name: string;
-        /**
-         * The description of the function.
-         */
-        description: string;
-        /**
-         * The parameters required by the function.
-         */
-        parameters: {
-            /**
-             * The type of the parameters.
-             */
-            type: string;
-            /**
-             * The list of required parameters.
-             */
-            required: string[];
-            /**
-             * The properties of the parameters.
-             */
-            properties: {
-                [key: string]: {
-                    /**
-                     * The type of the property.
-                     */
-                    type: string;
-                    /**
-                     * The description of the property.
-                     */
-                    description: string;
-                    /**
-                     * The possible values for the property.
-                     */
-                    enum?: string[];
-                };
-            };
-        };
-    };
-}
-
-/**
- * Interface representing an incoming message.
- */
-interface IIncomingMessage {
-    /**
-     * The ID of the client sending the message.
-     */
-    clientId: string;
-    /**
-     * The data contained in the message.
-     */
-    data: string;
-    /**
-     * The name of the agent sending the message.
-     */
-    agentName: AgentName;
-}
-/**
- * Interface representing an outgoing message.
- */
-interface IOutgoingMessage {
-    /**
-     * The ID of the client receiving the message.
+     * Callback triggered when a client connects.
+     * @param clientId - The ID of the client.
+     * @param swarmName - The name of the swarm.
      */
-    clientId: string;
+    onConnect?: (clientId: string, swarmName: SwarmName) => void;
     /**
-     * The data contained in the message.
+     * Callback triggered when a command is executed.
+     * @param clientId - The ID of the client.
+     * @param swarmName - The name of the swarm.
+     * @param content - The content to execute.
+     * @param mode - The source of execution: tool or user.
      */
-    data: string;
+    onExecute?: (clientId: string, swarmName: SwarmName, content: string, mode: ExecutionMode) => void;
     /**
-     * The name of the agent sending the message.
+     * Callback triggered when a message is emitted.
+     * @param clientId - The ID of the client.
+     * @param swarmName - The name of the swarm.
+     * @param message - The message to emit.
      */
-    agentName: AgentName;
+    onEmit?: (clientId: string, swarmName: SwarmName, message: string) => void;
 }
-
 /**
  * Parameters for initializing a swarm.
  * @interface
@@ -221,6 +83,7 @@ interface IOutgoingMessage {
  */
 interface ISwarmParams extends Omit<ISwarmSchema, keyof {
     agentList: never;
+    onAgentChanged: never;
 }> {
     /** Client identifier */
     clientId: string;
@@ -229,19 +92,21 @@ interface ISwarmParams extends Omit<ISwarmSchema, keyof {
     /** Map of agent names to agent instances */
     agentMap: Record<AgentName, IAgent>;
     /** Emit the callback on agent change */
-    onAgentChanged(clientId: string, agentName: AgentName, swarmName: SwarmName): Promise<void>;
+    onAgentChanged: (clientId: string, agentName: AgentName, swarmName: SwarmName) => Promise<void>;
 }
 /**
  * Schema for defining a swarm.
  * @interface
  */
-interface ISwarmSchema {
+interface ISwarmSchema extends ISwarmSession {
     /** Default agent name */
     defaultAgent: AgentName;
     /** Name of the swarm */
     swarmName: string;
     /** List of agent names */
     agentList: string[];
+    /** Emit the callback on agent change */
+    onAgentChanged?: (clientId: string, agentName: AgentName, swarmName: SwarmName) => void;
 }
 /**
  * Interface for a swarm.
@@ -284,10 +149,11 @@ type SwarmName = string;
  * Parameters required to create a session.
  * @interface
  */
-interface ISessionParams extends ISessionSchema {
+interface ISessionParams extends ISessionSchema, ISwarmSession {
     clientId: string;
     logger: ILogger;
     swarm: ISwarm;
+    swarmName: SwarmName;
 }
 /**
  * Schema for session data.
@@ -374,6 +240,166 @@ type SessionMode = "session" | "makeConnection" | "complete";
  */
 type ExecutionMode = "tool" | "user";
 
+/**
+ * Interface representing a model message.
+ */
+interface IModelMessage {
+    /**
+     * The role of the message sender.
+     * @type {'assistant' | 'system' | 'tool' | 'user' | 'resque' | 'flush'}
+     */
+    role: "assistant" | "system" | "tool" | "user" | "resque" | "flush";
+    /**
+     * The name of the agent sending the message.
+     * @type {string}
+     */
+    agentName: string;
+    /**
+     * The content of the message.
+     * @type {string}
+     */
+    content: string;
+    /**
+     * The source of message: tool or user
+     * @type {ExecutionMode}
+     */
+    mode: ExecutionMode;
+    /**
+     * Optional tool calls associated with the message.
+     * @type {Array<{ function: { name: string; arguments: { [key: string]: any; }; }; }>}
+     */
+    tool_calls?: {
+        function: {
+            name: string;
+            arguments: {
+                [key: string]: any;
+            };
+        };
+    }[];
+}
+
+/**
+ * Interface representing the history of model messages.
+ */
+interface IHistory {
+    /**
+     * Pushes a message to the history.
+     * @param {IModelMessage} message - The message to push.
+     * @returns {Promise<void>}
+     */
+    push(message: IModelMessage): Promise<void>;
+    /**
+     * Converts the history to an array of messages for a specific agent.
+     * @param {string} prompt - The prompt to filter messages for the agent.
+     * @returns {Promise<IModelMessage[]>}
+     */
+    toArrayForAgent(prompt: string, system?: string[]): Promise<IModelMessage[]>;
+    /**
+     * Converts the history to an array of raw messages.
+     * @returns {Promise<IModelMessage[]>}
+     */
+    toArrayForRaw(): Promise<IModelMessage[]>;
+}
+/**
+ * Interface representing the parameters required to create a history instance.
+ */
+interface IHistoryParams extends IHistorySchema {
+    /**
+     * The name of the agent.
+     * @type {AgentName}
+     */
+    agentName: AgentName;
+    /**
+     * The client ID.
+     * @type {string}
+     */
+    clientId: string;
+    /**
+     * The logger instance.
+     * @type {ILogger}
+     */
+    logger: ILogger;
+}
+/**
+ * Interface representing the schema of the history.
+ */
+interface IHistorySchema {
+    /**
+     * The array of model messages.
+     * @type {IPubsubArray<IModelMessage>}
+     */
+    items: IPubsubArray<IModelMessage>;
+}
+
+/**
+ * Represents a tool call with a function name and arguments.
+ */
+interface IToolCall {
+    function: {
+        /**
+         * The name of the function to be called.
+         */
+        name: string;
+        /**
+         * The arguments to be passed to the function.
+         */
+        arguments: {
+            [key: string]: any;
+        };
+    };
+}
+/**
+ * Represents a tool with a type and function details.
+ */
+interface ITool {
+    /**
+     * The type of the tool.
+     */
+    type: string;
+    function: {
+        /**
+         * The name of the function.
+         */
+        name: string;
+        /**
+         * The description of the function.
+         */
+        description: string;
+        /**
+         * The parameters required by the function.
+         */
+        parameters: {
+            /**
+             * The type of the parameters.
+             */
+            type: string;
+            /**
+             * The list of required parameters.
+             */
+            required: string[];
+            /**
+             * The properties of the parameters.
+             */
+            properties: {
+                [key: string]: {
+                    /**
+                     * The type of the property.
+                     */
+                    type: string;
+                    /**
+                     * The description of the property.
+                     */
+                    description: string;
+                    /**
+                     * The possible values for the property.
+                     */
+                    enum?: string[];
+                };
+            };
+        };
+    };
+}
+
 /**
  * Interface representing a completion.
  */
@@ -418,6 +444,12 @@ interface ICompletionSchema {
      * @returns A promise that resolves to a model message.
      */
     getCompletion(args: ICompletionArgs): Promise<IModelMessage>;
+    /**
+     * Callback fired after complete.
+     * @param args - Arguments passed to complete
+     * @param output - Output of the model
+     */
+    onComplete?: (args: ICompletionArgs, output: IModelMessage) => void;
 }
 /**
  * Type representing the name of a completion.
@@ -447,6 +479,22 @@ interface IAgentTool<T = Record<string, unknown>> extends ITool {
      * @returns A promise that resolves to a boolean indicating whether the parameters are valid, or a boolean.
      */
     validate(clientId: string, agentName: AgentName, params: T): Promise<boolean> | boolean;
+    /**
+     * Callback triggered when the tool is called.
+     * @param clientId - The ID of the client.
+     * @param agentName - The name of the agent.
+     * @param params - The parameters for the tool.
+     * @returns A promise that resolves when the tool call is complete.
+     */
+    onCall?: (clientId: string, agentName: AgentName, params: T) => Promise<void>;
+    /**
+     * Callback triggered when the tool parameters are validated.
+     * @param clientId - The ID of the client.
+     * @param agentName - The name of the agent.
+     * @param params - The parameters for the tool.
+     * @returns A promise that resolves to a boolean indicating whether the parameters are valid.
+     */
+    onValidate?: (clientId: string, agentName: AgentName, params: T) => Promise<boolean>;
 }
 /**
  * Interface representing the parameters for an agent.
@@ -493,6 +541,56 @@ interface IAgentSchema {
      * @returns A promise that resolves to a string or null.
      */
     validate?: (output: string) => Promise<string | null>;
+    /**
+     * Callback triggered when the agent executes.
+     * @param clientId - The ID of the client.
+     * @param agentName - The name of the agent.
+     * @param input - The input to execute.
+     * @param mode - The source of execution: tool or user.
+     */
+    onExecute?: (clientId: string, agentName: AgentName, input: string, mode: ExecutionMode) => void;
+    /**
+     * Callback triggered when there is tool output.
+     * @param clientId - The ID of the client.
+     * @param agentName - The name of the agent.
+     * @param content - The content of the tool output.
+     */
+    onToolOutput?: (clientId: string, agentName: AgentName, content: string) => void;
+    /**
+     * Callback triggered when there is a system message.
+     * @param clientId - The ID of the client.
+     * @param agentName - The name of the agent.
+     * @param message - The system message.
+     */
+    onSystemMessage?: (clientId: string, agentName: AgentName, message: string) => void;
+    /**
+     * Callback triggered when there is a user message.
+     * @param clientId - The ID of the client.
+     * @param agentName - The name of the agent.
+     * @param message - The user message.
+     */
+    onUserMessage?: (clientId: string, agentName: AgentName, message: string) => void;
+    /**
+     * Callback triggered when the agent history is flushed.
+     * @param clientId - The ID of the client.
+     * @param agentName - The name of the agent.
+     */
+    onFlush?: (clientId: string, agentName: AgentName) => void;
+    /**
+     * Callback triggered when there is output.
+     * @param clientId - The ID of the client.
+     * @param agentName - The name of the agent.
+     * @param output - The output string.
+     */
+    onOutput?: (clientId: string, agentName: AgentName, output: string) => void;
+    /**
+     * Callback triggered when the agent is resurrected.
+     * @param clientId - The ID of the client.
+     * @param agentName - The name of the agent.
+     * @param mode - The source of execution: tool or user.
+     * @param reason - The reason for the resurrection.
+     */
+    onResurrect?: (clientId: string, agentName: AgentName, mode: ExecutionMode, reason?: string) => void;
 }
 /**
  * Interface representing an agent.
@@ -501,7 +599,7 @@ interface IAgent {
     /**
      * Executes the agent with the given input.
      * @param input - The input to execute.
-     * @param mode - The source of execution: tool or user
+     * @param mode - The source of execution: tool or user.
      * @returns A promise that resolves when the execution is complete.
      */
     execute: (input: string, mode: ExecutionMode) => Promise<void>;
@@ -529,7 +627,7 @@ interface IAgent {
      */
     commitUserMessage(message: string): Promise<void>;
     /**
-     * Clear the history for agent
+     * Clears the history for the agent.
      * @returns A promise that resolves when the flush is committed.
      */
     commitFlush(): Promise<void>;
@@ -1022,6 +1120,7 @@ declare class SessionConnectionService implements ISession {
     private readonly loggerService;
     private readonly contextService;
     private readonly swarmConnectionService;
+    private readonly swarmSchemaService;
     /**
      * Retrieves a memoized session based on clientId and swarmName.
      * @param {string} clientId - The client ID.