agent-swarm-kit 1.0.31 → 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,9 +484,11 @@ 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',
+                                mode: "tool",
                                 agentName: this.params.agentName,
                                 content: reason || "Unknown error",
                             })];
@@ -489,7 +496,7 @@ var ClientAgent = /** @class */ (function () {
                         _a.sent();
                         return [4 /*yield*/, this.params.history.push({
                                 role: "user",
-                                mode: 'tool',
+                                mode: "tool",
                                 agentName: this.params.agentName,
                                 content: GLOBAL_CONFIG.CC_TOOL_CALL_EXCEPTION_PROMPT,
                             })];
@@ -508,7 +515,7 @@ var ClientAgent = /** @class */ (function () {
                         return [4 /*yield*/, this.params.history.push({
                                 agentName: this.params.agentName,
                                 role: "assistant",
-                                mode: 'tool',
+                                mode: "tool",
                                 content: content,
                             })];
                     case 5:
@@ -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];
                 }
             });
         }); };
@@ -571,11 +583,13 @@ var ClientAgent = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " commitUserMessage"));
+                        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,
-                                mode: 'user',
+                                mode: "user",
                                 content: message.trim(),
                             })];
                     case 1:
@@ -584,6 +598,29 @@ var ClientAgent = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Commits flush of agent history
+         * @returns {Promise<void>}
+         */
+        this.commitFlush = function () { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                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,
+                                mode: "tool",
+                                content: "",
+                            })];
+                    case 1:
+                        _a.sent();
+                        return [2 /*return*/];
+                }
+            });
+        }); };
         /**
          * Commits a system message to the history.
          * @param {string} message - The system message to commit.
@@ -593,11 +630,13 @@ var ClientAgent = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " commitSystemMessage"));
+                        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,
-                                mode: 'tool',
+                                mode: "tool",
                                 content: message.trim(),
                             })];
                     case 1:
@@ -615,11 +654,13 @@ var ClientAgent = /** @class */ (function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " commitToolOutput content=").concat(content));
+                        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,
-                                mode: 'tool',
+                                mode: "tool",
                                 content: content,
                             })];
                     case 1:
@@ -644,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,
@@ -676,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"));
@@ -689,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
                                          */
@@ -795,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({
@@ -808,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,
             });
         });
         /**
@@ -899,6 +952,22 @@ var AgentConnectionService = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Commits flush of agent history
+         * @returns {Promise<any>} The commit result.
+         */
+        this.commitFlush = function () { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("agentConnectionService commitFlush", {
+                            context: this.contextService.context,
+                        });
+                        return [4 /*yield*/, this.getAgent(this.contextService.context.clientId, this.contextService.context.agentName).commitFlush()];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Disposes of the agent connection.
          * @returns {Promise<void>} The dispose result.
@@ -1028,6 +1097,11 @@ var ClientHistory = /** @class */ (function () {
                             systemMessagesRaw.splice(0, systemMessagesRaw.length);
                             return [3 /*break*/, 4];
                         }
+                        if (message.role === "flush") {
+                            commonMessagesRaw.splice(0, commonMessagesRaw.length);
+                            systemMessagesRaw.splice(0, systemMessagesRaw.length);
+                            return [3 /*break*/, 4];
+                        }
                         if (message.role === "system") {
                             systemMessagesRaw.push(message);
                         }
@@ -1417,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()) {
@@ -1442,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*/];
@@ -1637,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();
@@ -1659,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();
@@ -1713,6 +1793,24 @@ var ClientSession = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Commits flush of agent history
+         * @returns {Promise<void>}
+         */
+        this.commitFlush = function () { return __awaiter(_this, void 0, void 0, function () {
+            var agent;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.params.logger.debug("ClientSession clientId=".concat(this.params.clientId, " commitFlush"));
+                        return [4 /*yield*/, this.params.swarm.getAgent()];
+                    case 1:
+                        agent = _a.sent();
+                        return [4 /*yield*/, agent.commitFlush()];
+                    case 2: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Commits a system message.
          * @param {string} message - The system message to commit.
@@ -1741,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;
@@ -1799,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.
@@ -1809,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,
             });
         });
         /**
@@ -1873,6 +1979,7 @@ var SessionConnectionService = /** @class */ (function () {
                     case 0:
                         this.loggerService.log("sessionConnectionService commitToolOutput", {
                             context: this.contextService.context,
+                            content: content,
                         });
                         return [4 /*yield*/, this.getSession(this.contextService.context.clientId, this.contextService.context.swarmName).commitToolOutput(content)];
                     case 1: return [2 /*return*/, _a.sent()];
@@ -1890,6 +1997,7 @@ var SessionConnectionService = /** @class */ (function () {
                     case 0:
                         this.loggerService.log("sessionConnectionService commitSystemMessage", {
                             context: this.contextService.context,
+                            message: message,
                         });
                         return [4 /*yield*/, this.getSession(this.contextService.context.clientId, this.contextService.context.swarmName).commitSystemMessage(message)];
                     case 1: return [2 /*return*/, _a.sent()];
@@ -1907,12 +2015,30 @@ var SessionConnectionService = /** @class */ (function () {
                     case 0:
                         this.loggerService.log("sessionConnectionService commitUserMessage", {
                             context: this.contextService.context,
+                            message: message,
                         });
                         return [4 /*yield*/, this.getSession(this.contextService.context.clientId, this.contextService.context.swarmName).commitUserMessage(message)];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
         }); };
+        /**
+         * Commits user message to the agent without answer.
+         * @param {string} message - The message to commit.
+         * @returns {Promise<void>} A promise that resolves when the message is committed.
+         */
+        this.commitFlush = function () { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("sessionConnectionService commitFlush", {
+                            context: this.contextService.context,
+                        });
+                        return [4 /*yield*/, this.getSession(this.contextService.context.clientId, this.contextService.context.swarmName).commitFlush()];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Disposes of the session connection service.
          * @returns {Promise<void>} A promise that resolves when the service is disposed.
@@ -2133,6 +2259,37 @@ var AgentPublicService = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Commits flush of agent history
+         * @param {string} clientId - The client ID.
+         * @param {AgentName} agentName - The name of the agent.
+         * @returns {Promise<unknown>} The commit result.
+         */
+        this.commitFlush = function (clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("agentPublicService commitFlush", {
+                            clientId: clientId,
+                            agentName: agentName,
+                        });
+                        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.agentConnectionService.commitFlush()];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                clientId: clientId,
+                                agentName: agentName,
+                                swarmName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Disposes of the agent.
          * @param {string} clientId - The client ID.
@@ -2530,6 +2687,37 @@ var SessionPublicService = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Commits flush of agent history
+         * @param {string} clientId - The client ID.
+         * @param {SwarmName} swarmName - The swarm name.
+         * @returns {Promise<void>}
+         */
+        this.commitFlush = 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("sessionPublicService commitFlush", {
+                            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.sessionConnectionService.commitFlush()];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                clientId: clientId,
+                                swarmName: swarmName,
+                                agentName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Disposes of the session.
          * @param {string} clientId - The client ID.
@@ -3995,6 +4183,45 @@ var commitSystemMessage = function (content, clientId, agentName) { return __awa
     });
 }); };
 
+/**
+ * Commits flush of agent history
+ *
+ * @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 message is committed.
+ */
+var commitFlush = 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 commitFlush', {
+                    clientId: clientId,
+                    agentName: agentName,
+                });
+                swarm.agentValidationService.validate(agentName, "commitFlush");
+                swarm.sessionValidationService.validate(clientId, "commitFlush");
+                swarmName = swarm.sessionValidationService.getSwarm(clientId);
+                swarm.swarmValidationService.validate(swarmName, "commitFlush");
+                return [4 /*yield*/, swarm.swarmPublicService.getAgentName(clientId, swarmName)];
+            case 1:
+                currentAgentName = _a.sent();
+                if (currentAgentName !== agentName) {
+                    swarm.loggerService.log('function "commitFlush" skipped due to the agent change', {
+                        currentAgentName: currentAgentName,
+                        agentName: agentName,
+                        clientId: clientId,
+                    });
+                    return [2 /*return*/];
+                }
+                return [4 /*yield*/, swarm.sessionPublicService.commitFlush(clientId, swarmName)];
+            case 2:
+                _a.sent();
+                return [2 /*return*/];
+        }
+    });
+}); };
+
 /**
  * Send the message to the active agent in the swarm content like it income from client side
  * Should be used to review tool output and initiate conversation from the model side to client
@@ -4267,6 +4494,7 @@ exports.addCompletion = addCompletion;
 exports.addSwarm = addSwarm;
 exports.addTool = addTool;
 exports.changeAgent = changeAgent;
+exports.commitFlush = commitFlush;
 exports.commitSystemMessage = commitSystemMessage;
 exports.commitToolOutput = commitToolOutput;
 exports.commitUserMessage = commitUserMessage;