agent-swarm-kit 1.0.32 → 1.0.34

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,25 +547,30 @@ var ClientAgent = /** @class */ (function () {
          * @returns {Promise<IModelMessage>}
          */
         this.getCompletion = function (mode) { return __awaiter(_this, void 0, void 0, function () {
-            var messages;
-            var _a;
-            return __generator(this, function (_b) {
-                switch (_b.label) {
+            var messages, args, output;
+            var _a, _b, _c;
+            return __generator(this, function (_d) {
+                switch (_d.label) {
                     case 0:
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " getCompletion"));
                         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()];
+                        messages = _d.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", "callbacks");
+                            }),
+                        };
+                        return [4 /*yield*/, this.params.completion.getCompletion(args)];
+                    case 2:
+                        output = _d.sent();
+                        ((_b = this.params.completion.callbacks) === null || _b === void 0 ? void 0 : _b.onComplete) &&
+                            ((_c = this.params.completion.callbacks) === null || _c === void 0 ? void 0 : _c.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,
@@ -660,11 +680,13 @@ var ClientAgent = /** @class */ (function () {
         this.execute = functoolsKit.queued(function (incoming, mode) { return __awaiter(_this, void 0, void 0, function () {
             var message, _loop_1, this_1, _a, _b, tool, state_1, e_1_1, result, validation, result1;
             var e_1, _c;
-            var _d;
-            return __generator(this, function (_e) {
-                switch (_e.label) {
+            var _d, _e, _f, _g, _h;
+            return __generator(this, function (_j) {
+                switch (_j.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,
@@ -672,44 +694,49 @@ var ClientAgent = /** @class */ (function () {
                                 content: incoming.trim(),
                             })];
                     case 1:
-                        _e.sent();
+                        _j.sent();
                         return [4 /*yield*/, this.getCompletion(mode)];
                     case 2:
-                        message = _e.sent();
+                        message = _j.sent();
                         if (!message.tool_calls) return [3 /*break*/, 11];
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " tool call begin"));
                         _loop_1 = function (tool) {
                             var targetFn, result_2, result_3;
-                            return __generator(this, function (_f) {
-                                switch (_f.label) {
+                            return __generator(this, function (_k) {
+                                switch (_k.label) {
                                     case 0:
                                         targetFn = (_d = this_1.params.tools) === null || _d === void 0 ? void 0 : _d.find(function (t) { return t.function.name === tool.function.name; });
                                         return [4 /*yield*/, this_1.params.history.push(__assign(__assign({}, message), { agentName: this_1.params.agentName }))];
                                     case 1:
-                                        _f.sent();
+                                        _k.sent();
                                         if (!!targetFn) return [3 /*break*/, 4];
                                         this_1.params.logger.debug("ClientAgent agentName=".concat(this_1.params.agentName, " clientId=").concat(this_1.params.clientId, " functionName=").concat(tool.function.name, " tool function not found"), this_1.params.tools);
                                         return [4 /*yield*/, this_1._resurrectModel(mode, "No target function for ".concat(tool.function.name))];
                                     case 2:
-                                        result_2 = _f.sent();
+                                        result_2 = _k.sent();
                                         this_1.params.logger.debug("ClientAgent agentName=".concat(this_1.params.agentName, " clientId=").concat(this_1.params.clientId, " execute end result=").concat(result_2));
                                         return [4 /*yield*/, this_1._emitOuput(mode, result_2)];
                                     case 3:
-                                        _f.sent();
+                                        _k.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:
+                                        ((_e = targetFn.callbacks) === null || _e === void 0 ? void 0 : _e.onValidate) &&
+                                            ((_f = targetFn.callbacks) === null || _f === void 0 ? void 0 : _f.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];
+                                        if (!_k.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"));
                                         return [4 /*yield*/, this_1._resurrectModel(mode, "Function validation failed: name=".concat(tool.function.name, " arguments=").concat(JSON.stringify(tool.function.arguments)))];
                                     case 6:
-                                        result_3 = _f.sent();
+                                        result_3 = _k.sent();
                                         this_1.params.logger.debug("ClientAgent agentName=".concat(this_1.params.agentName, " clientId=").concat(this_1.params.clientId, " execute end result=").concat(result_3));
                                         return [4 /*yield*/, this_1._emitOuput(mode, result_3)];
                                     case 7:
-                                        _f.sent();
+                                        _k.sent();
                                         return [2 /*return*/, { value: void 0 }];
                                     case 8:
+                                        ((_g = targetFn.callbacks) === null || _g === void 0 ? void 0 : _g.onCall) &&
+                                            ((_h = targetFn.callbacks) === null || _h === void 0 ? void 0 : _h.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
                                          */
@@ -720,33 +747,33 @@ var ClientAgent = /** @class */ (function () {
                                                 this_1._outputSubject.toPromise(),
                                             ])];
                                     case 9:
-                                        _f.sent();
+                                        _k.sent();
                                         this_1.params.logger.debug("ClientAgent agentName=".concat(this_1.params.agentName, " clientId=").concat(this_1.params.clientId, " functionName=").concat(tool.function.name, " tool call end"));
                                         return [2 /*return*/];
                                 }
                             });
                         };
                         this_1 = this;
-                        _e.label = 3;
+                        _j.label = 3;
                     case 3:
-                        _e.trys.push([3, 8, 9, 10]);
+                        _j.trys.push([3, 8, 9, 10]);
                         _a = __values(message.tool_calls), _b = _a.next();
-                        _e.label = 4;
+                        _j.label = 4;
                     case 4:
                         if (!!_b.done) return [3 /*break*/, 7];
                         tool = _b.value;
                         return [5 /*yield**/, _loop_1(tool)];
                     case 5:
-                        state_1 = _e.sent();
+                        state_1 = _j.sent();
                         if (typeof state_1 === "object")
                             return [2 /*return*/, state_1.value];
-                        _e.label = 6;
+                        _j.label = 6;
                     case 6:
                         _b = _a.next();
                         return [3 /*break*/, 4];
                     case 7: return [3 /*break*/, 10];
                     case 8:
-                        e_1_1 = _e.sent();
+                        e_1_1 = _j.sent();
                         e_1 = { error: e_1_1 };
                         return [3 /*break*/, 10];
                     case 9:
@@ -763,31 +790,44 @@ var ClientAgent = /** @class */ (function () {
                         result = message.content;
                         return [4 /*yield*/, this.params.history.push(__assign(__assign({}, message), { agentName: this.params.agentName }))];
                     case 12:
-                        _e.sent();
+                        _j.sent();
                         validation = null;
                         return [4 /*yield*/, this.params.validate(result)];
                     case 13:
-                        if (!(validation = _e.sent())) return [3 /*break*/, 16];
+                        if (!(validation = _j.sent())) return [3 /*break*/, 16];
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " execute invalid tool call detected: ").concat(validation), { result: result });
                         return [4 /*yield*/, this._resurrectModel(mode, "Invalid model output: ".concat(result))];
                     case 14:
-                        result1 = _e.sent();
+                        result1 = _j.sent();
                         return [4 /*yield*/, this._emitOuput(mode, result1)];
                     case 15:
-                        _e.sent();
+                        _j.sent();
                         return [2 /*return*/];
                     case 16:
                         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " execute end result=").concat(result));
                         return [4 /*yield*/, this._emitOuput(mode, result)];
                     case 17:
-                        _e.sent();
+                        _j.sent();
                         return [2 /*return*/];
                 }
             });
         }); });
+        /**
+         * Should call on agent dispose
+         * @returns {Promise<void>}
+         */
+        this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " dispose"));
+                this.params.onDispose &&
+                    this.params.onDispose(this.params.clientId, this.params.agentName);
+                return [2 /*return*/];
+            });
+        }); };
         this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " CTOR"), {
             params: params,
         });
+        this.params.onInit && this.params.onInit(params.clientId, params.agentName);
     }
     return ClientAgent;
 }());
@@ -816,20 +856,10 @@ 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, callbacks = _a.callbacks, 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({
-                clientId: clientId,
-                agentName: agentName,
-                validate: validate,
-                logger: _this.loggerService,
-                history: _this.historyConnectionService.getHistory(clientId, agentName),
-                prompt: prompt,
-                system: system,
-                tools: tools === null || tools === void 0 ? void 0 : tools.map(_this.toolSchemaService.get),
-                completion: completion,
-            });
+            return new ClientAgent(__assign({ clientId: clientId, agentName: agentName, validate: validate, logger: _this.loggerService, history: _this.historyConnectionService.getHistory(clientId, agentName), prompt: prompt, system: system, tools: tools === null || tools === void 0 ? void 0 : tools.map(_this.toolSchemaService.get), completion: completion }, callbacks));
         });
         /**
          * Executes an input command.
@@ -942,12 +972,18 @@ var AgentConnectionService = /** @class */ (function () {
          */
         this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
-                this.loggerService.log("agentConnectionService dispose", {
-                    context: this.contextService.context,
-                });
-                this.getAgent.clear("".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.agentName));
-                this.sessionValidationService.removeAgentUsage(this.contextService.context.clientId, this.contextService.context.agentName);
-                return [2 /*return*/];
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("agentConnectionService dispose", {
+                            context: this.contextService.context,
+                        });
+                        return [4 /*yield*/, this.getAgent(this.contextService.context.clientId, this.contextService.context.agentName).dispose()];
+                    case 1:
+                        _a.sent();
+                        this.getAgent.clear("".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.agentName));
+                        this.sessionValidationService.removeAgentUsage(this.contextService.context.clientId, this.contextService.context.agentName);
+                        return [2 /*return*/];
+                }
             });
         }); };
     }
@@ -1459,7 +1495,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, callbacks = _b.callbacks;
             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 +1520,11 @@ 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:
+                                    if (callbacks && callbacks.onAgentChanged) {
+                                        callbacks.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 +1719,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 +1743,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 +1845,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;
@@ -1842,9 +1888,22 @@ var ClientSession = /** @class */ (function () {
                 });
             }); };
         };
+        /**
+         * Should call on session dispose
+         * @returns {Promise<void>}
+         */
+        this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                this.params.logger.debug("ClientSession clientId=".concat(this.params.clientId, " dispose"));
+                this.params.onDispose &&
+                    this.params.onDispose(this.params.clientId, this.params.swarmName);
+                return [2 /*return*/];
+            });
+        }); };
         this.params.logger.debug("ClientSession clientId=".concat(this.params.clientId, " CTOR"), {
             params: params,
         });
+        this.params.onInit && this.params.onInit(params.clientId, params.swarmName);
     }
     return ClientSession;
 }());
@@ -1859,6 +1918,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,11 +1929,8 @@ var SessionConnectionService = /** @class */ (function () {
             var _b = __read(_a, 2), clientId = _b[0], swarmName = _b[1];
             return "".concat(clientId, "-").concat(swarmName);
         }, function (clientId, swarmName) {
-            return new ClientSession({
-                clientId: clientId,
-                logger: _this.loggerService,
-                swarm: _this.swarmConnectionService.getSwarm(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));
         });
         /**
          * Emits a message to the session.
@@ -1999,11 +2056,17 @@ var SessionConnectionService = /** @class */ (function () {
          */
         this.dispose = function () { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
-                this.loggerService.log("sessionConnectionService dispose", {
-                    context: this.contextService.context,
-                });
-                this.getSession.clear("".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.swarmName));
-                return [2 /*return*/];
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("sessionConnectionService dispose", {
+                            context: this.contextService.context,
+                        });
+                        return [4 /*yield*/, this.getSession(this.contextService.context.clientId, this.contextService.context.swarmName).dispose()];
+                    case 1:
+                        _a.sent();
+                        this.getSession.clear("".concat(this.contextService.context.clientId, "-").concat(this.contextService.context.swarmName));
+                        return [2 /*return*/];
+                }
             });
         }); };
     }