agent-swarm-kit 1.0.110 → 1.0.112

package/build/index.cjs

@@ -1671,6 +1671,7 @@ var AgentSchemaService = /** @class */ (function () {
 
 var AGENT_CHANGE_SYMBOL = Symbol("agent-change");
 var TOOL_ERROR_SYMBOL = Symbol("tool-error");
+var TOOL_STOP_SYMBOL = Symbol("tool-stop");
 var getPlaceholder = function () {
     return GLOBAL_CONFIG.CC_EMPTY_OUTPUT_PLACEHOLDERS[Math.floor(Math.random() * GLOBAL_CONFIG.CC_EMPTY_OUTPUT_PLACEHOLDERS.length)];
 };
@@ -1687,6 +1688,7 @@ var ClientAgent = /** @class */ (function () {
         var _this = this;
         this.params = params;
         this._agentChangeSubject = new functoolsKit.Subject();
+        this._toolStopSubject = new functoolsKit.Subject();
         this._toolErrorSubject = new functoolsKit.Subject();
         this._toolCommitSubject = new functoolsKit.Subject();
         this._outputSubject = new functoolsKit.Subject();
@@ -1987,6 +1989,35 @@ var ClientAgent = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Commits change of agent to prevent the next tool execution from being called.
+         * @returns {Promise<void>}
+         */
+        this.commitStopTools = function () { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " commitStopTools"));
+                        return [4 /*yield*/, this._toolStopSubject.next(TOOL_STOP_SYMBOL)];
+                    case 1:
+                        _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "commit-stop-tools",
+                                source: "agent-bus",
+                                input: {},
+                                output: {},
+                                context: {
+                                    agentName: this.params.agentName,
+                                },
+                                clientId: this.params.clientId,
+                            })];
+                    case 2:
+                        _a.sent();
+                        return [2 /*return*/];
+                }
+            });
+        }); };
         /**
          * Commits a system message to the history.
          * @param {string} message - The system message to commit.
@@ -2117,9 +2148,9 @@ var ClientAgent = /** @class */ (function () {
         this.execute = functoolsKit.queued(function (incoming, mode) { return __awaiter(_this, void 0, void 0, function () {
             var rawMessage, message, toolCalls, _loop_1, this_1, idx, state_1, result, validation, result1;
             var _this = this;
-            var _a, _b, _c, _d, _e, _f, _g;
-            return __generator(this, function (_h) {
-                switch (_h.label) {
+            var _a, _b, _c, _d, _e, _f, _g, _h;
+            return __generator(this, function (_j) {
+                switch (_j.label) {
                     case 0:
                         GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                             this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " execute begin"), { incoming: incoming, mode: mode });
@@ -2132,13 +2163,13 @@ var ClientAgent = /** @class */ (function () {
                                 content: incoming.trim(),
                             })];
                     case 1:
-                        _h.sent();
+                        _j.sent();
                         return [4 /*yield*/, this.getCompletion(mode)];
                     case 2:
-                        rawMessage = _h.sent();
+                        rawMessage = _j.sent();
                         return [4 /*yield*/, this.params.map(rawMessage, this.params.clientId, this.params.agentName)];
                     case 3:
-                        message = _h.sent();
+                        message = _j.sent();
                         if (!message.tool_calls) return [3 /*break*/, 9];
                         GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                             this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " tool call begin"));
@@ -2152,11 +2183,11 @@ var ClientAgent = /** @class */ (function () {
                         });
                         return [4 /*yield*/, this.params.history.push(__assign(__assign({}, message), { agentName: this.params.agentName }))];
                     case 4:
-                        _h.sent();
+                        _j.sent();
                         _loop_1 = function (idx) {
                             var tool, targetFn, result_2, result_3, status, result_4;
-                            return __generator(this, function (_j) {
-                                switch (_j.label) {
+                            return __generator(this, function (_k) {
+                                switch (_k.label) {
                                     case 0:
                                         tool = toolCalls[idx];
                                         targetFn = (_a = this_1.params.tools) === null || _a === void 0 ? void 0 : _a.find(function (t) { return t.function.name === tool.function.name; });
@@ -2165,12 +2196,12 @@ var ClientAgent = /** @class */ (function () {
                                             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 1:
-                                        result_2 = _j.sent();
+                                        result_2 = _k.sent();
                                         GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                                             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 2:
-                                        _j.sent();
+                                        _k.sent();
                                         return [2 /*return*/, { value: void 0 }];
                                     case 3:
                                         ((_b = targetFn.callbacks) === null || _b === void 0 ? void 0 : _b.onValidate) &&
@@ -2182,17 +2213,17 @@ var ClientAgent = /** @class */ (function () {
                                                 toolCalls: toolCalls,
                                             }))];
                                     case 4:
-                                        if (!_j.sent()) return [3 /*break*/, 7];
+                                        if (!_k.sent()) return [3 /*break*/, 7];
                                         GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                                             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 5:
-                                        result_3 = _j.sent();
+                                        result_3 = _k.sent();
                                         GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                                             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 6:
-                                        _j.sent();
+                                        _k.sent();
                                         return [2 /*return*/, { value: void 0 }];
                                     case 7:
                                         ((_d = targetFn.callbacks) === null || _d === void 0 ? void 0 : _d.onBeforeCall) &&
@@ -2232,10 +2263,11 @@ var ClientAgent = /** @class */ (function () {
                                                 this_1._agentChangeSubject.toPromise(),
                                                 this_1._toolCommitSubject.toPromise(),
                                                 this_1._toolErrorSubject.toPromise(),
+                                                this_1._toolStopSubject.toPromise(),
                                                 this_1._outputSubject.toPromise(),
                                             ])];
                                     case 8:
-                                        status = _j.sent();
+                                        status = _k.sent();
                                         GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                                             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"));
                                         if (status === AGENT_CHANGE_SYMBOL) {
@@ -2245,17 +2277,24 @@ var ClientAgent = /** @class */ (function () {
                                                 this_1.params.callbacks.onAfterToolCalls(this_1.params.clientId, this_1.params.agentName, toolCalls);
                                             return [2 /*return*/, { value: void 0 }];
                                         }
+                                        if (status === TOOL_STOP_SYMBOL) {
+                                            GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                                                this_1.params.logger.debug("ClientAgent agentName=".concat(this_1.params.agentName, " clientId=").concat(this_1.params.clientId, " functionName=").concat(tool.function.name, " the next tool execution stopped due to the commitStopTools call"));
+                                            ((_g = this_1.params.callbacks) === null || _g === void 0 ? void 0 : _g.onAfterToolCalls) &&
+                                                this_1.params.callbacks.onAfterToolCalls(this_1.params.clientId, this_1.params.agentName, toolCalls);
+                                            return [2 /*return*/, { value: void 0 }];
+                                        }
                                         if (!(status === TOOL_ERROR_SYMBOL)) return [3 /*break*/, 11];
                                         GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                                             this_1.params.logger.debug("ClientAgent agentName=".concat(this_1.params.agentName, " clientId=").concat(this_1.params.clientId, " functionName=").concat(tool.function.name, " the next tool execution stopped due to the call error"));
                                         return [4 /*yield*/, this_1._resurrectModel(mode, "Function call failed with error: name=".concat(tool.function.name, " arguments=").concat(JSON.stringify(tool.function.arguments)))];
                                     case 9:
-                                        result_4 = _j.sent();
+                                        result_4 = _k.sent();
                                         GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                                             this_1.params.logger.debug("ClientAgent agentName=".concat(this_1.params.agentName, " clientId=").concat(this_1.params.clientId, " execute end result=").concat(result_4));
                                         return [4 /*yield*/, this_1._emitOuput(mode, result_4)];
                                     case 10:
-                                        _j.sent();
+                                        _k.sent();
                                         return [2 /*return*/, { value: void 0 }];
                                     case 11: return [2 /*return*/];
                                 }
@@ -2263,20 +2302,20 @@ var ClientAgent = /** @class */ (function () {
                         };
                         this_1 = this;
                         idx = 0;
-                        _h.label = 5;
+                        _j.label = 5;
                     case 5:
                         if (!(idx !== toolCalls.length)) return [3 /*break*/, 8];
                         return [5 /*yield**/, _loop_1(idx)];
                     case 6:
-                        state_1 = _h.sent();
+                        state_1 = _j.sent();
                         if (typeof state_1 === "object")
                             return [2 /*return*/, state_1.value];
-                        _h.label = 7;
+                        _j.label = 7;
                     case 7:
                         idx++;
                         return [3 /*break*/, 5];
                     case 8:
-                        ((_g = this.params.callbacks) === null || _g === void 0 ? void 0 : _g.onAfterToolCalls) &&
+                        ((_h = this.params.callbacks) === null || _h === void 0 ? void 0 : _h.onAfterToolCalls) &&
                             this.params.callbacks.onAfterToolCalls(this.params.clientId, this.params.agentName, toolCalls);
                         return [2 /*return*/];
                     case 9:
@@ -2286,29 +2325,29 @@ var ClientAgent = /** @class */ (function () {
                         }
                         return [4 /*yield*/, this.params.transform(message.content, this.params.clientId, this.params.agentName)];
                     case 10:
-                        result = _h.sent();
+                        result = _j.sent();
                         return [4 /*yield*/, this.params.history.push(__assign(__assign({}, message), { agentName: this.params.agentName }))];
                     case 11:
-                        _h.sent();
+                        _j.sent();
                         validation = null;
                         return [4 /*yield*/, this.params.validate(result)];
                     case 12:
-                        if (!(validation = _h.sent())) return [3 /*break*/, 15];
+                        if (!(validation = _j.sent())) return [3 /*break*/, 15];
                         GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                             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 13:
-                        result1 = _h.sent();
+                        result1 = _j.sent();
                         return [4 /*yield*/, this._emitOuput(mode, result1)];
                     case 14:
-                        _h.sent();
+                        _j.sent();
                         return [2 /*return*/];
                     case 15:
                         GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                             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 16:
-                        _h.sent();
+                        _j.sent();
                         return [2 /*return*/];
                 }
             });
@@ -2499,6 +2538,21 @@ var AgentConnectionService = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Prevent the next tool from being executed
+         * @returns {Promise<any>} The commit result.
+         */
+        this.commitStopTools = function () { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                            this.loggerService.info("agentConnectionService commitStopTools");
+                        return [4 /*yield*/, this.getAgent(this.methodContextService.context.clientId, this.methodContextService.context.agentName).commitStopTools()];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Commits flush of agent history
          * @returns {Promise<any>} The commit result.
@@ -3792,6 +3846,39 @@ var ClientSession = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Commits stop of the nexttool execution
+         * @returns {Promise<void>}
+         */
+        this.commitStopTools = function () { return __awaiter(_this, void 0, void 0, function () {
+            var agent, result;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            this.params.logger.debug("ClientSession clientId=".concat(this.params.clientId, " commitStopTools"));
+                        return [4 /*yield*/, this.params.swarm.getAgent()];
+                    case 1:
+                        agent = _a.sent();
+                        return [4 /*yield*/, agent.commitStopTools()];
+                    case 2:
+                        result = _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "commit-stop-tools",
+                                source: "session-bus",
+                                input: {},
+                                output: {},
+                                context: {
+                                    swarmName: this.params.swarmName,
+                                },
+                                clientId: this.params.clientId,
+                            })];
+                    case 3:
+                        _a.sent();
+                        return [2 /*return*/, result];
+                }
+            });
+        }); };
         /**
          * Commits a system message.
          * @param {string} message - The system message to commit.
@@ -4119,6 +4206,22 @@ var SessionConnectionService = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * 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.commitStopTools = function () { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                            this.loggerService.info("sessionConnectionService commitStopTools");
+                        return [4 /*yield*/, this.getSession(this.methodContextService.context.clientId, this.methodContextService.context.swarmName).commitStopTools()];
+                    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.
@@ -4491,6 +4594,42 @@ var AgentPublicService = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Prevent the next tool from being executed
+         * @param {string} clientId - The client ID.
+         * @param {AgentName} agentName - The name of the agent.
+         * @returns {Promise<unknown>} The commit result.
+         */
+        this.commitStopTools = function (methodName, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                            this.loggerService.info("agentPublicService commitStopTools", {
+                                methodName: methodName,
+                                clientId: clientId,
+                                agentName: agentName,
+                            });
+                        return [4 /*yield*/, MethodContextService.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.commitStopTools()];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                methodName: methodName,
+                                clientId: clientId,
+                                agentName: agentName,
+                                swarmName: "",
+                                storageName: "",
+                                stateName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Disposes of the agent.
          * @param {string} clientId - The client ID.
@@ -5004,6 +5143,41 @@ var SessionPublicService = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Prevent the next tool from being executed
+         * @param {string} clientId - The client ID.
+         * @param {SwarmName} swarmName - The swarm name.
+         * @returns {Promise<void>}
+         */
+        this.commitStopTools = function (methodName, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                            this.loggerService.info("sessionPublicService commitStopTools", {
+                                clientId: clientId,
+                                swarmName: swarmName,
+                            });
+                        return [4 /*yield*/, MethodContextService.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.commitStopTools()];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                methodName: methodName,
+                                clientId: clientId,
+                                swarmName: swarmName,
+                                agentName: "",
+                                storageName: "",
+                                stateName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Disposes of the session.
          * @param {string} clientId - The client ID.
@@ -5859,7 +6033,10 @@ var SessionValidationService = /** @class */ (function () {
          * @param {string} source - The source of the validation request.
          * @throws Will throw an error if the session does not exist.
          */
-        this.validate = function (clientId, source) {
+        this.validate = functoolsKit.memoize(function (_a) {
+            var _b = __read(_a, 1), clientId = _b[0];
+            return "".concat(clientId);
+        }, function (clientId, source) {
             GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
                 _this.loggerService.info("sessionValidationService validate", {
                     clientId: clientId,
@@ -5871,7 +6048,7 @@ var SessionValidationService = /** @class */ (function () {
                 throw new Error("agent-swarm session clientId=".concat(clientId, " not exist source=").concat(source));
             }
             return {};
-        };
+        });
         /**
          * Removes a session.
          * @param {SessionId} clientId - The ID of the client.
@@ -5884,6 +6061,13 @@ var SessionValidationService = /** @class */ (function () {
             _this._sessionSwarmMap.delete(clientId);
             _this._sessionModeMap.delete(clientId);
         };
+        /**
+         * Dispose a session validation cache.
+         * @param {SessionId} clientId - The ID of the client.
+         */
+        this.dispose = function (clientId) {
+            _this.validate.clear(clientId);
+        };
     }
     return SessionValidationService;
 }());
@@ -7246,6 +7430,53 @@ var ClientState = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Sets the to initial value
+         * @returns {Promise<State>}
+         */
+        this.clearState = function () { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            var _a;
+            return __generator(this, function (_b) {
+                switch (_b.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " clearState"));
+                        return [4 /*yield*/, this.dispatch("write", function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.params.getState(this.params.clientId, this.params.stateName)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); })];
+                    case 1:
+                        _b.sent();
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            this.params.logger.debug("ClientState stateName=".concat(this.params.stateName, " clientId=").concat(this.params.clientId, " shared=").concat(this.params.shared, " clearState output"), { pendingState: this._state });
+                        this.params.setState &&
+                            this.params.setState(this._state, this.params.clientId, this.params.stateName);
+                        if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onWrite) {
+                            this.params.callbacks.onWrite(this._state, this.params.clientId, this.params.stateName);
+                        }
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "clear-state",
+                                source: "state-bus",
+                                input: {},
+                                output: {
+                                    state: this._state,
+                                },
+                                context: {
+                                    stateName: this.params.stateName,
+                                },
+                                clientId: this.params.clientId,
+                            })];
+                    case 2:
+                        _b.sent();
+                        return [2 /*return*/, this._state];
+                }
+            });
+        }); };
         /**
          * Gets the current state.
          * @returns {Promise<State>}
@@ -7383,6 +7614,26 @@ var StateConnectionService = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Set the state to initial value
+         * @returns {Promise<T>} The initial state.
+         */
+        this.clearState = function () { return __awaiter(_this, void 0, void 0, function () {
+            var state;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                            this.loggerService.info("stateConnectionService clearState");
+                        state = this.getStateRef(this.methodContextService.context.clientId, this.methodContextService.context.stateName);
+                        return [4 /*yield*/, state.waitForInit()];
+                    case 1:
+                        _a.sent();
+                        return [4 /*yield*/, state.clearState()];
+                    case 2: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Gets the state.
          * @returns {Promise<T>} The current state.
@@ -7480,6 +7731,42 @@ var StatePublicService = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Set the state to initial value
+         * @param {string} clientId - The client ID.
+         * @param {StateName} stateName - The name of the state.
+         * @returns {Promise<T>} - The initial state.
+         */
+        this.clearState = function (methodName, clientId, stateName) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                            this.loggerService.info("statePublicService clearState", {
+                                methodName: methodName,
+                                clientId: clientId,
+                                stateName: stateName,
+                            });
+                        return [4 /*yield*/, MethodContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.stateConnectionService.clearState()];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                methodName: methodName,
+                                clientId: clientId,
+                                stateName: stateName,
+                                agentName: "",
+                                swarmName: "",
+                                storageName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Gets the current state.
          * @param {string} clientId - The client ID.
@@ -8349,6 +8636,26 @@ var SharedStateConnectionService = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Set the state to initial value
+         * @returns {Promise<T>} The new state.
+         */
+        this.clearState = function () { return __awaiter(_this, void 0, void 0, function () {
+            var state;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                            this.loggerService.info("sharedStateConnectionService clearState");
+                        state = this.getStateRef(this.methodContextService.context.stateName);
+                        return [4 /*yield*/, state.waitForInit()];
+                    case 1:
+                        _a.sent();
+                        return [4 /*yield*/, state.clearState()];
+                    case 2: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Gets the state.
          * @returns {Promise<T>} The current state.
@@ -8582,6 +8889,40 @@ var SharedStatePublicService = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Set the state to initial value
+         * @param {StateName} stateName - The name of the state.
+         * @returns {Promise<T>} - The initial state.
+         */
+        this.clearState = function (methodName, stateName) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                            this.loggerService.info("sharedStatePublicService clearState", {
+                                methodName: methodName,
+                                stateName: stateName,
+                            });
+                        return [4 /*yield*/, MethodContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.sharedStateConnectionService.clearState()];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                methodName: methodName,
+                                clientId: "",
+                                stateName: stateName,
+                                agentName: "",
+                                swarmName: "",
+                                storageName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Gets the current state.
          * @param {StateName} stateName - The name of the state.
@@ -8952,7 +9293,7 @@ var swarm = __assign(__assign(__assign(__assign(__assign(__assign(__assign({}, b
 init();
 var swarm$1 = swarm;
 
-var METHOD_NAME$J = "cli.dumpDocs";
+var METHOD_NAME$L = "cli.dumpDocs";
 /**
  * Dumps the documentation for the agents and swarms.
  *
@@ -8963,7 +9304,7 @@ var METHOD_NAME$J = "cli.dumpDocs";
 var dumpDocs = function (dirName, PlantUML) {
     if (dirName === void 0) { dirName = "./docs/chat"; }
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$J, {
+        swarm$1.loggerService.log(METHOD_NAME$L, {
             dirName: dirName,
         });
     if (PlantUML) {
@@ -8974,12 +9315,12 @@ var dumpDocs = function (dirName, PlantUML) {
     swarm$1.agentValidationService
         .getAgentList()
         .forEach(function (agentName) {
-        return swarm$1.agentValidationService.validate(agentName, METHOD_NAME$J);
+        return swarm$1.agentValidationService.validate(agentName, METHOD_NAME$L);
     });
     swarm$1.swarmValidationService
         .getSwarmList()
         .forEach(function (swarmName) {
-        return swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$J);
+        return swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$L);
     });
     swarm$1.agentValidationService.getAgentList().forEach(function (agentName) {
         var dependsOn = swarm$1.agentSchemaService.get(agentName).dependsOn;
@@ -8990,7 +9331,7 @@ var dumpDocs = function (dirName, PlantUML) {
     return swarm$1.docService.dumpDocs(dirName);
 };
 
-var METHOD_NAME$I = "cli.dumpAgent";
+var METHOD_NAME$K = "cli.dumpAgent";
 /**
  * Dumps the agent information into PlantUML format.
  *
@@ -9000,14 +9341,14 @@ var METHOD_NAME$I = "cli.dumpAgent";
 var dumpAgent = function (agentName, _a) {
     var _b = _a === void 0 ? {} : _a, _c = _b.withSubtree, withSubtree = _c === void 0 ? false : _c;
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$I, {
+        swarm$1.loggerService.log(METHOD_NAME$K, {
             agentName: agentName,
         });
-    swarm$1.agentValidationService.validate(agentName, METHOD_NAME$I);
+    swarm$1.agentValidationService.validate(agentName, METHOD_NAME$K);
     return swarm$1.agentMetaService.toUML(agentName, withSubtree);
 };
 
-var METHOD_NAME$H = "cli.dumpSwarm";
+var METHOD_NAME$J = "cli.dumpSwarm";
 /**
  * Dumps the swarm information into PlantUML format.
  *
@@ -9016,14 +9357,14 @@ var METHOD_NAME$H = "cli.dumpSwarm";
  */
 var dumpSwarm = function (swarmName) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$H, {
+        swarm$1.loggerService.log(METHOD_NAME$J, {
             swarmName: swarmName,
         });
-    swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$H);
+    swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$J);
     return swarm$1.swarmMetaService.toUML(swarmName);
 };
 
-var METHOD_NAME$G = "function.addAgent";
+var METHOD_NAME$I = "function.addAgent";
 /**
  * Adds a new agent to the agent registry. The swarm takes only those agents which was registered
  *
@@ -9032,7 +9373,7 @@ var METHOD_NAME$G = "function.addAgent";
  */
 var addAgent = function (agentSchema) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$G, {
+        swarm$1.loggerService.log(METHOD_NAME$I, {
             agentSchema: agentSchema,
         });
     swarm$1.agentValidationService.addAgent(agentSchema.agentName, agentSchema);
@@ -9040,7 +9381,7 @@ var addAgent = function (agentSchema) {
     return agentSchema.agentName;
 };
 
-var METHOD_NAME$F = "function.addCompletion";
+var METHOD_NAME$H = "function.addCompletion";
 /**
  * Adds a completion engine for agents. Agents could use different models and
  * framewords for completion like: mock, gpt4all, ollama, openai
@@ -9050,7 +9391,7 @@ var METHOD_NAME$F = "function.addCompletion";
  */
 var addCompletion = function (completionSchema) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$F, {
+        swarm$1.loggerService.log(METHOD_NAME$H, {
             completionSchema: completionSchema,
         });
     swarm$1.completionValidationService.addCompletion(completionSchema.completionName);
@@ -9058,7 +9399,7 @@ var addCompletion = function (completionSchema) {
     return completionSchema.completionName;
 };
 
-var METHOD_NAME$E = "function.addSwarm";
+var METHOD_NAME$G = "function.addSwarm";
 /**
  * Adds a new swarm to the system. The swarm is a root for starting client session
  *
@@ -9067,7 +9408,7 @@ var METHOD_NAME$E = "function.addSwarm";
  */
 var addSwarm = function (swarmSchema) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$E, {
+        swarm$1.loggerService.log(METHOD_NAME$G, {
             swarmSchema: swarmSchema,
         });
     swarm$1.swarmValidationService.addSwarm(swarmSchema.swarmName, swarmSchema);
@@ -9075,7 +9416,7 @@ var addSwarm = function (swarmSchema) {
     return swarmSchema.swarmName;
 };
 
-var METHOD_NAME$D = "function.addTool";
+var METHOD_NAME$F = "function.addTool";
 /**
  * Adds a new tool for agents in a swarm. Tool should be registered in `addAgent`
  * declaration
@@ -9085,7 +9426,7 @@ var METHOD_NAME$D = "function.addTool";
  */
 var addTool = function (toolSchema) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$D, {
+        swarm$1.loggerService.log(METHOD_NAME$F, {
             toolSchema: toolSchema,
         });
     swarm$1.toolValidationService.addTool(toolSchema.toolName, toolSchema);
@@ -9093,7 +9434,7 @@ var addTool = function (toolSchema) {
     return toolSchema.toolName;
 };
 
-var METHOD_NAME$C = "function.addState";
+var METHOD_NAME$E = "function.addState";
 /**
  * Adds a new state to the state registry. The swarm takes only those states which was registered
  *
@@ -9102,7 +9443,7 @@ var METHOD_NAME$C = "function.addState";
  */
 var addState = function (stateSchema) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$C, {
+        swarm$1.loggerService.log(METHOD_NAME$E, {
             stateSchema: stateSchema,
         });
     swarm$1.stateSchemaService.register(stateSchema.stateName, stateSchema);
@@ -9114,7 +9455,7 @@ var addState = function (stateSchema) {
     return stateSchema.stateName;
 };
 
-var METHOD_NAME$B = "function.addEmbedding";
+var METHOD_NAME$D = "function.addEmbedding";
 /**
  * Adds a new embedding to the embedding registry. The swarm takes only those embeddings which was registered
  *
@@ -9123,7 +9464,7 @@ var METHOD_NAME$B = "function.addEmbedding";
  */
 var addEmbedding = function (embeddingSchema) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$B, {
+        swarm$1.loggerService.log(METHOD_NAME$D, {
             embeddingSchema: embeddingSchema,
         });
     swarm$1.embeddingValidationService.addEmbedding(embeddingSchema.embeddingName, embeddingSchema);
@@ -9131,7 +9472,7 @@ var addEmbedding = function (embeddingSchema) {
     return embeddingSchema.embeddingName;
 };
 
-var METHOD_NAME$A = "function.addStorage";
+var METHOD_NAME$C = "function.addStorage";
 /**
  * Adds a new storage to the storage registry. The swarm takes only those storages which was registered
  *
@@ -9140,7 +9481,7 @@ var METHOD_NAME$A = "function.addStorage";
  */
 var addStorage = function (storageSchema) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$A, {
+        swarm$1.loggerService.log(METHOD_NAME$C, {
             storageSchema: storageSchema,
         });
     swarm$1.storageValidationService.addStorage(storageSchema.storageName, storageSchema);
@@ -9153,7 +9494,7 @@ var addStorage = function (storageSchema) {
     return storageSchema.storageName;
 };
 
-var METHOD_NAME$z = "function.commitToolOutput";
+var METHOD_NAME$B = "function.commitToolOutput";
 /**
  * Commits the tool output to the active agent in a swarm session
  *
@@ -9168,17 +9509,17 @@ var commitToolOutput = function (toolId, content, clientId, agentName) { return
         switch (_a.label) {
             case 0:
                 GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-                    swarm$1.loggerService.log(METHOD_NAME$z, {
+                    swarm$1.loggerService.log(METHOD_NAME$B, {
                         toolId: toolId,
                         content: content,
                         clientId: clientId,
                         agentName: agentName,
                     });
-                swarm$1.agentValidationService.validate(agentName, METHOD_NAME$z);
-                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$z);
+                swarm$1.agentValidationService.validate(agentName, METHOD_NAME$B);
+                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$B);
                 swarmName = swarm$1.sessionValidationService.getSwarm(clientId);
-                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$z);
-                return [4 /*yield*/, swarm$1.swarmPublicService.getAgentName(METHOD_NAME$z, clientId, swarmName)];
+                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$B);
+                return [4 /*yield*/, swarm$1.swarmPublicService.getAgentName(METHOD_NAME$B, clientId, swarmName)];
             case 1:
                 currentAgentName = _a.sent();
                 if (currentAgentName !== agentName) {
@@ -9191,7 +9532,7 @@ var commitToolOutput = function (toolId, content, clientId, agentName) { return
                         });
                     return [2 /*return*/];
                 }
-                return [4 /*yield*/, swarm$1.sessionPublicService.commitToolOutput(toolId, content, METHOD_NAME$z, clientId, swarmName)];
+                return [4 /*yield*/, swarm$1.sessionPublicService.commitToolOutput(toolId, content, METHOD_NAME$B, clientId, swarmName)];
             case 2:
                 _a.sent();
                 return [2 /*return*/];
@@ -9199,7 +9540,7 @@ var commitToolOutput = function (toolId, content, clientId, agentName) { return
     });
 }); };
 
-var METHOD_NAME$y = "function.commitSystemMessage";
+var METHOD_NAME$A = "function.commitSystemMessage";
 /**
  * Commits a system message to the active agent in the swarm.
  *
@@ -9214,16 +9555,16 @@ var commitSystemMessage = function (content, clientId, agentName) { return __awa
         switch (_a.label) {
             case 0:
                 GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-                    swarm$1.loggerService.log(METHOD_NAME$y, {
+                    swarm$1.loggerService.log(METHOD_NAME$A, {
                         content: content,
                         clientId: clientId,
                         agentName: agentName,
                     });
-                swarm$1.agentValidationService.validate(agentName, METHOD_NAME$y);
-                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$y);
+                swarm$1.agentValidationService.validate(agentName, METHOD_NAME$A);
+                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$A);
                 swarmName = swarm$1.sessionValidationService.getSwarm(clientId);
-                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$y);
-                return [4 /*yield*/, swarm$1.swarmPublicService.getAgentName(METHOD_NAME$y, clientId, swarmName)];
+                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$A);
+                return [4 /*yield*/, swarm$1.swarmPublicService.getAgentName(METHOD_NAME$A, clientId, swarmName)];
             case 1:
                 currentAgentName = _a.sent();
                 if (currentAgentName !== agentName) {
@@ -9235,7 +9576,7 @@ var commitSystemMessage = function (content, clientId, agentName) { return __awa
                         });
                     return [2 /*return*/];
                 }
-                return [4 /*yield*/, swarm$1.sessionPublicService.commitSystemMessage(content, METHOD_NAME$y, clientId, swarmName)];
+                return [4 /*yield*/, swarm$1.sessionPublicService.commitSystemMessage(content, METHOD_NAME$A, clientId, swarmName)];
             case 2:
                 _a.sent();
                 return [2 /*return*/];
@@ -9243,7 +9584,7 @@ var commitSystemMessage = function (content, clientId, agentName) { return __awa
     });
 }); };
 
-var METHOD_NAME$x = "function.commitFlush";
+var METHOD_NAME$z = "function.commitFlush";
 /**
  * Commits flush of agent history
  *
@@ -9257,15 +9598,15 @@ var commitFlush = function (clientId, agentName) { return __awaiter(void 0, void
         switch (_a.label) {
             case 0:
                 GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-                    swarm$1.loggerService.log(METHOD_NAME$x, {
+                    swarm$1.loggerService.log(METHOD_NAME$z, {
                         clientId: clientId,
                         agentName: agentName,
                     });
-                swarm$1.agentValidationService.validate(agentName, METHOD_NAME$x);
-                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$x);
+                swarm$1.agentValidationService.validate(agentName, METHOD_NAME$z);
+                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$z);
                 swarmName = swarm$1.sessionValidationService.getSwarm(clientId);
-                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$x);
-                return [4 /*yield*/, swarm$1.swarmPublicService.getAgentName(METHOD_NAME$x, clientId, swarmName)];
+                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$z);
+                return [4 /*yield*/, swarm$1.swarmPublicService.getAgentName(METHOD_NAME$z, clientId, swarmName)];
             case 1:
                 currentAgentName = _a.sent();
                 if (currentAgentName !== agentName) {
@@ -9277,7 +9618,7 @@ var commitFlush = function (clientId, agentName) { return __awaiter(void 0, void
                         });
                     return [2 /*return*/];
                 }
-                return [4 /*yield*/, swarm$1.sessionPublicService.commitFlush(METHOD_NAME$x, clientId, swarmName)];
+                return [4 /*yield*/, swarm$1.sessionPublicService.commitFlush(METHOD_NAME$z, clientId, swarmName)];
             case 2:
                 _a.sent();
                 return [2 /*return*/];
@@ -9285,7 +9626,7 @@ var commitFlush = function (clientId, agentName) { return __awaiter(void 0, void
     });
 }); };
 
-var METHOD_NAME$w = "function.commitSystemMessage";
+var METHOD_NAME$y = "function.commitSystemMessage";
 /**
  * Commits a user message to the active agent history in as swarm without answer.
  *
@@ -9300,16 +9641,16 @@ var commitUserMessage = function (content, clientId, agentName) { return __await
         switch (_a.label) {
             case 0:
                 GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-                    swarm$1.loggerService.log(METHOD_NAME$w, {
+                    swarm$1.loggerService.log(METHOD_NAME$y, {
                         content: content,
                         clientId: clientId,
                         agentName: agentName,
                     });
-                swarm$1.agentValidationService.validate(agentName, METHOD_NAME$w);
-                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$w);
+                swarm$1.agentValidationService.validate(agentName, METHOD_NAME$y);
+                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$y);
                 swarmName = swarm$1.sessionValidationService.getSwarm(clientId);
-                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$w);
-                return [4 /*yield*/, swarm$1.swarmPublicService.getAgentName(METHOD_NAME$w, clientId, swarmName)];
+                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$y);
+                return [4 /*yield*/, swarm$1.swarmPublicService.getAgentName(METHOD_NAME$y, clientId, swarmName)];
             case 1:
                 currentAgentName = _a.sent();
                 if (currentAgentName !== agentName) {
@@ -9321,7 +9662,7 @@ var commitUserMessage = function (content, clientId, agentName) { return __await
                         });
                     return [2 /*return*/];
                 }
-                return [4 /*yield*/, swarm$1.sessionPublicService.commitUserMessage(content, METHOD_NAME$w, clientId, swarmName)];
+                return [4 /*yield*/, swarm$1.sessionPublicService.commitUserMessage(content, METHOD_NAME$y, clientId, swarmName)];
             case 2:
                 _a.sent();
                 return [2 /*return*/];
@@ -9329,7 +9670,7 @@ var commitUserMessage = function (content, clientId, agentName) { return __await
     });
 }); };
 
-var METHOD_NAME$v = "function.commitToolOutputForce";
+var METHOD_NAME$x = "function.commitToolOutputForce";
 /**
  * Commits the tool output to the active agent in a swarm session without checking active agent
  *
@@ -9343,15 +9684,15 @@ var commitToolOutputForce = function (toolId, content, clientId) { return __awai
         switch (_a.label) {
             case 0:
                 GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-                    swarm$1.loggerService.log(METHOD_NAME$v, {
+                    swarm$1.loggerService.log(METHOD_NAME$x, {
                         toolId: toolId,
                         content: content,
                         clientId: clientId,
                     });
-                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$v);
+                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$x);
                 swarmName = swarm$1.sessionValidationService.getSwarm(clientId);
-                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$v);
-                return [4 /*yield*/, swarm$1.sessionPublicService.commitToolOutput(toolId, content, METHOD_NAME$v, clientId, swarmName)];
+                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$x);
+                return [4 /*yield*/, swarm$1.sessionPublicService.commitToolOutput(toolId, content, METHOD_NAME$x, clientId, swarmName)];
             case 1:
                 _a.sent();
                 return [2 /*return*/];
@@ -9359,7 +9700,7 @@ var commitToolOutputForce = function (toolId, content, clientId) { return __awai
     });
 }); };
 
-var METHOD_NAME$u = "function.commitSystemMessageForce";
+var METHOD_NAME$w = "function.commitSystemMessageForce";
 /**
  * Commits a system message to the active agent in as swarm without checking active agent.
  *
@@ -9373,14 +9714,14 @@ var commitSystemMessageForce = function (content, clientId) { return __awaiter(v
         switch (_a.label) {
             case 0:
                 GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-                    swarm$1.loggerService.log(METHOD_NAME$u, {
+                    swarm$1.loggerService.log(METHOD_NAME$w, {
                         content: content,
                         clientId: clientId,
                     });
-                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$u);
+                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$w);
                 swarmName = swarm$1.sessionValidationService.getSwarm(clientId);
-                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$u);
-                return [4 /*yield*/, swarm$1.sessionPublicService.commitSystemMessage(content, METHOD_NAME$u, clientId, swarmName)];
+                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$w);
+                return [4 /*yield*/, swarm$1.sessionPublicService.commitSystemMessage(content, METHOD_NAME$w, clientId, swarmName)];
             case 1:
                 _a.sent();
                 return [2 /*return*/];
@@ -9388,7 +9729,7 @@ var commitSystemMessageForce = function (content, clientId) { return __awaiter(v
     });
 }); };
 
-var METHOD_NAME$t = "function.commitFlushForce";
+var METHOD_NAME$v = "function.commitFlushForce";
 /**
  * Commits flush of agent history without active agent check
  *
@@ -9401,14 +9742,14 @@ var commitFlushForce = function (clientId) { return __awaiter(void 0, void 0, vo
         switch (_a.label) {
             case 0:
                 GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-                    swarm$1.loggerService.log(METHOD_NAME$t, {
+                    swarm$1.loggerService.log(METHOD_NAME$v, {
                         clientId: clientId,
-                        METHOD_NAME: METHOD_NAME$t,
+                        METHOD_NAME: METHOD_NAME$v,
                     });
-                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$t);
+                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$v);
                 swarmName = swarm$1.sessionValidationService.getSwarm(clientId);
-                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$t);
-                return [4 /*yield*/, swarm$1.sessionPublicService.commitFlush(METHOD_NAME$t, clientId, swarmName)];
+                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$v);
+                return [4 /*yield*/, swarm$1.sessionPublicService.commitFlush(METHOD_NAME$v, clientId, swarmName)];
             case 1:
                 _a.sent();
                 return [2 /*return*/];
@@ -9416,7 +9757,7 @@ var commitFlushForce = function (clientId) { return __awaiter(void 0, void 0, vo
     });
 }); };
 
-var METHOD_NAME$s = "function.commitSystemMessage";
+var METHOD_NAME$u = "function.commitSystemMessage";
 /**
  * Commits a user message to the active agent history in as swarm without answer and checking active agent
  *
@@ -9430,14 +9771,14 @@ var commitUserMessageForce = function (content, clientId) { return __awaiter(voi
         switch (_a.label) {
             case 0:
                 GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-                    swarm$1.loggerService.log(METHOD_NAME$s, {
+                    swarm$1.loggerService.log(METHOD_NAME$u, {
                         content: content,
                         clientId: clientId,
                     });
-                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$s);
+                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$u);
                 swarmName = swarm$1.sessionValidationService.getSwarm(clientId);
-                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$s);
-                return [4 /*yield*/, swarm$1.sessionPublicService.commitUserMessage(content, METHOD_NAME$s, clientId, swarmName)];
+                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$u);
+                return [4 /*yield*/, swarm$1.sessionPublicService.commitUserMessage(content, METHOD_NAME$u, clientId, swarmName)];
             case 1:
                 _a.sent();
                 return [2 /*return*/];
@@ -9445,7 +9786,7 @@ var commitUserMessageForce = function (content, clientId) { return __awaiter(voi
     });
 }); };
 
-var METHOD_NAME$r = "function.commitAssistantMessage";
+var METHOD_NAME$t = "function.commitAssistantMessage";
 /**
  * Commits an assistant message to the active agent in the swarm.
  *
@@ -9460,16 +9801,16 @@ var commitAssistantMessage = function (content, clientId, agentName) { return __
         switch (_a.label) {
             case 0:
                 GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-                    swarm$1.loggerService.log(METHOD_NAME$r, {
+                    swarm$1.loggerService.log(METHOD_NAME$t, {
                         content: content,
                         clientId: clientId,
                         agentName: agentName,
                     });
-                swarm$1.agentValidationService.validate(agentName, METHOD_NAME$r);
-                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$r);
+                swarm$1.agentValidationService.validate(agentName, METHOD_NAME$t);
+                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$t);
                 swarmName = swarm$1.sessionValidationService.getSwarm(clientId);
-                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$r);
-                return [4 /*yield*/, swarm$1.swarmPublicService.getAgentName(METHOD_NAME$r, clientId, swarmName)];
+                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$t);
+                return [4 /*yield*/, swarm$1.swarmPublicService.getAgentName(METHOD_NAME$t, clientId, swarmName)];
             case 1:
                 currentAgentName = _a.sent();
                 if (currentAgentName !== agentName) {
@@ -9481,7 +9822,7 @@ var commitAssistantMessage = function (content, clientId, agentName) { return __
                         });
                     return [2 /*return*/];
                 }
-                return [4 /*yield*/, swarm$1.sessionPublicService.commitAssistantMessage(content, METHOD_NAME$r, clientId, swarmName)];
+                return [4 /*yield*/, swarm$1.sessionPublicService.commitAssistantMessage(content, METHOD_NAME$t, clientId, swarmName)];
             case 2:
                 _a.sent();
                 return [2 /*return*/];
@@ -9489,7 +9830,7 @@ var commitAssistantMessage = function (content, clientId, agentName) { return __
     });
 }); };
 
-var METHOD_NAME$q = "function.commitAssistantMessageForce";
+var METHOD_NAME$s = "function.commitAssistantMessageForce";
 /**
  * Commits an assistant message to the active agent in as swarm without checking active agent.
  *
@@ -9503,14 +9844,14 @@ var commitAssistantMessageForce = function (content, clientId) { return __awaite
         switch (_a.label) {
             case 0:
                 GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-                    swarm$1.loggerService.log(METHOD_NAME$q, {
+                    swarm$1.loggerService.log(METHOD_NAME$s, {
                         content: content,
                         clientId: clientId,
                     });
-                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$q);
+                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$s);
                 swarmName = swarm$1.sessionValidationService.getSwarm(clientId);
-                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$q);
-                return [4 /*yield*/, swarm$1.sessionPublicService.commitAssistantMessage(content, METHOD_NAME$q, clientId, swarmName)];
+                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$s);
+                return [4 /*yield*/, swarm$1.sessionPublicService.commitAssistantMessage(content, METHOD_NAME$s, clientId, swarmName)];
             case 1:
                 _a.sent();
                 return [2 /*return*/];
@@ -9518,7 +9859,7 @@ var commitAssistantMessageForce = function (content, clientId) { return __awaite
     });
 }); };
 
-var METHOD_NAME$p = "function.cancelOutput";
+var METHOD_NAME$r = "function.cancelOutput";
 /**
  * Cancel the await of output by emit of empty string
  *
@@ -9532,15 +9873,15 @@ var cancelOutput = function (clientId, agentName) { return __awaiter(void 0, voi
         switch (_a.label) {
             case 0:
                 GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-                    swarm$1.loggerService.log(METHOD_NAME$p, {
+                    swarm$1.loggerService.log(METHOD_NAME$r, {
                         clientId: clientId,
                         agentName: agentName,
                     });
-                swarm$1.agentValidationService.validate(agentName, METHOD_NAME$p);
-                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$p);
+                swarm$1.agentValidationService.validate(agentName, METHOD_NAME$r);
+                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$r);
                 swarmName = swarm$1.sessionValidationService.getSwarm(clientId);
-                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$p);
-                return [4 /*yield*/, swarm$1.swarmPublicService.getAgentName(METHOD_NAME$p, clientId, swarmName)];
+                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$r);
+                return [4 /*yield*/, swarm$1.swarmPublicService.getAgentName(METHOD_NAME$r, clientId, swarmName)];
             case 1:
                 currentAgentName = _a.sent();
                 if (currentAgentName !== agentName) {
@@ -9552,7 +9893,7 @@ var cancelOutput = function (clientId, agentName) { return __awaiter(void 0, voi
                         });
                     return [2 /*return*/];
                 }
-                return [4 /*yield*/, swarm$1.swarmPublicService.cancelOutput(METHOD_NAME$p, clientId, swarmName)];
+                return [4 /*yield*/, swarm$1.swarmPublicService.cancelOutput(METHOD_NAME$r, clientId, swarmName)];
             case 2:
                 _a.sent();
                 return [2 /*return*/];
@@ -9560,7 +9901,7 @@ var cancelOutput = function (clientId, agentName) { return __awaiter(void 0, voi
     });
 }); };
 
-var METHOD_NAME$o = "function.cancelOutputForce";
+var METHOD_NAME$q = "function.cancelOutputForce";
 /**
  * Cancel the await of output by emit of empty string without checking active agent
  *
@@ -9569,6 +9910,75 @@ var METHOD_NAME$o = "function.cancelOutputForce";
  * @returns {Promise<void>} - A promise that resolves when the output is canceled
  */
 var cancelOutputForce = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
+    var swarmName;
+    return __generator(this, function (_a) {
+        switch (_a.label) {
+            case 0:
+                GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
+                    swarm$1.loggerService.log(METHOD_NAME$q, {
+                        clientId: clientId,
+                    });
+                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$q);
+                swarmName = swarm$1.sessionValidationService.getSwarm(clientId);
+                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$q);
+                return [4 /*yield*/, swarm$1.swarmPublicService.cancelOutput(METHOD_NAME$q, clientId, swarmName)];
+            case 1:
+                _a.sent();
+                return [2 /*return*/];
+        }
+    });
+}); };
+
+var METHOD_NAME$p = "function.commitStopTools";
+/**
+ * Prevent the next tool from being executed
+ *
+ * @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 commitStopTools = 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:
+                GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
+                    swarm$1.loggerService.log(METHOD_NAME$p, {
+                        clientId: clientId,
+                        agentName: agentName,
+                    });
+                swarm$1.agentValidationService.validate(agentName, METHOD_NAME$p);
+                swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$p);
+                swarmName = swarm$1.sessionValidationService.getSwarm(clientId);
+                swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$p);
+                return [4 /*yield*/, swarm$1.swarmPublicService.getAgentName(METHOD_NAME$p, clientId, swarmName)];
+            case 1:
+                currentAgentName = _a.sent();
+                if (currentAgentName !== agentName) {
+                    GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
+                        swarm$1.loggerService.log('function "commitStopTools" skipped due to the agent change', {
+                            currentAgentName: currentAgentName,
+                            agentName: agentName,
+                            clientId: clientId,
+                        });
+                    return [2 /*return*/];
+                }
+                return [4 /*yield*/, swarm$1.sessionPublicService.commitStopTools(METHOD_NAME$p, clientId, swarmName)];
+            case 2:
+                _a.sent();
+                return [2 /*return*/];
+        }
+    });
+}); };
+
+var METHOD_NAME$o = "function.commitStopToolsForce";
+/**
+ * Prevent the next tool from being executed without active agent check
+ *
+ * @param {string} clientId - The ID of the client.
+ * @returns {Promise<void>} - A promise that resolves when the message is committed.
+ */
+var commitStopToolsForce = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -9576,11 +9986,12 @@ var cancelOutputForce = function (clientId) { return __awaiter(void 0, void 0, v
                 GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
                     swarm$1.loggerService.log(METHOD_NAME$o, {
                         clientId: clientId,
+                        METHOD_NAME: METHOD_NAME$o,
                     });
                 swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$o);
                 swarmName = swarm$1.sessionValidationService.getSwarm(clientId);
                 swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$o);
-                return [4 /*yield*/, swarm$1.swarmPublicService.cancelOutput(METHOD_NAME$o, clientId, swarmName)];
+                return [4 /*yield*/, swarm$1.sessionPublicService.commitStopTools(METHOD_NAME$o, clientId, swarmName)];
             case 1:
                 _a.sent();
                 return [2 /*return*/];
@@ -9751,6 +10162,7 @@ var disposeConnection = function (clientId_1, swarmName_1) {
                     _a.sent();
                     swarm$1.sessionValidationService.removeSession(clientId);
                     swarm$1.busService.dispose(clientId);
+                    swarm$1.sessionValidationService.dispose(clientId);
                     return [2 /*return*/];
             }
         });
@@ -11237,6 +11649,7 @@ var listenSwarmEventOnce = function (clientId, filterFn, fn) {
 
 var METHOD_NAME_GET$3 = "StateUtils.getState";
 var METHOD_NAME_SET$1 = "StateUtils.setState";
+var METHOD_NAME_CLEAR$3 = "StateUtils.clearState";
 /**
  * Utility class for managing state in the agent swarm.
  * @implements {TState}
@@ -11263,6 +11676,7 @@ var StateUtils = /** @class */ (function () {
                                 clientId: payload.clientId,
                                 stateName: payload.stateName,
                             });
+                        swarm$1.sessionValidationService.validate(payload.clientId, METHOD_NAME_GET$3);
                         if (!swarm$1.agentValidationService.hasState(payload.agentName, payload.stateName)) {
                             throw new Error("agent-swarm StateUtils ".concat(payload.stateName, " not registered in ").concat(payload.agentName, " (getState)"));
                         }
@@ -11292,6 +11706,7 @@ var StateUtils = /** @class */ (function () {
                                 clientId: payload.clientId,
                                 stateName: payload.stateName,
                             });
+                        swarm$1.sessionValidationService.validate(payload.clientId, METHOD_NAME_SET$1);
                         if (!swarm$1.agentValidationService.hasState(payload.agentName, payload.stateName)) {
                             throw new Error("agent-swarm StateUtils ".concat(payload.stateName, " not registered in ").concat(payload.agentName, " (setState)"));
                         }
@@ -11305,6 +11720,34 @@ var StateUtils = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Set the state to initial value
+         * @template T
+         * @param {Object} payload - The payload containing client and state information.
+         * @param {string} payload.clientId - The client ID.
+         * @param {AgentName} payload.agentName - The agent name.
+         * @param {StateName} payload.stateName - The state name.
+         * @returns {Promise<void>}
+         * @throws Will throw an error if the state is not registered in the agent.
+         */
+        this.clearState = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
+                            swarm$1.loggerService.log(METHOD_NAME_SET$1, {
+                                clientId: payload.clientId,
+                                stateName: payload.stateName,
+                            });
+                        swarm$1.sessionValidationService.validate(payload.clientId, METHOD_NAME_SET$1);
+                        if (!swarm$1.agentValidationService.hasState(payload.agentName, payload.stateName)) {
+                            throw new Error("agent-swarm StateUtils ".concat(payload.stateName, " not registered in ").concat(payload.agentName, " (clearState)"));
+                        }
+                        return [4 /*yield*/, swarm$1.statePublicService.clearState(METHOD_NAME_CLEAR$3, payload.clientId, payload.stateName)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
     }
     return StateUtils;
 }());
@@ -11316,6 +11759,7 @@ var State = new StateUtils();
 
 var METHOD_NAME_GET$2 = "SharedStateUtils.getSharedState";
 var METHOD_NAME_SET = "SharedStateUtils.setSharedState";
+var METHOD_NAME_CLEAR$2 = "SharedStateUtils.clearSharedState";
 /**
  * Utility class for managing state in the agent swarm.
  * @implements {TSharedState}
@@ -11380,6 +11824,31 @@ var SharedStateUtils = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Set the state to initial value
+         * @template T
+         * @param {Object} payload - The payload containing client and state information.
+         * @param {AgentName} payload.agentName - The agent name.
+         * @param {StateName} payload.stateName - The state name.
+         * @returns {Promise<void>}
+         * @throws Will throw an error if the state is not registered in the agent.
+         */
+        this.clearState = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
+                            swarm$1.loggerService.log(METHOD_NAME_SET, {
+                                stateName: payload.stateName,
+                            });
+                        if (!swarm$1.agentValidationService.hasState(payload.agentName, payload.stateName)) {
+                            throw new Error("agent-swarm SharedStateUtils ".concat(payload.stateName, " not registered in ").concat(payload.agentName, " (clearState)"));
+                        }
+                        return [4 /*yield*/, swarm$1.sharedStatePublicService.clearState(METHOD_NAME_CLEAR$2, payload.stateName)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
     }
     return SharedStateUtils;
 }());
@@ -11420,6 +11889,7 @@ var StorageUtils = /** @class */ (function () {
                                 storageName: payload.storageName,
                                 score: payload.score,
                             });
+                        swarm$1.sessionValidationService.validate(payload.clientId, METHOD_NAME_TAKE$1);
                         swarm$1.storageValidationService.validate(payload.storageName, METHOD_NAME_TAKE$1);
                         if (!swarm$1.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
                             throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (take)"));
@@ -11448,6 +11918,7 @@ var StorageUtils = /** @class */ (function () {
                                 clientId: payload.clientId,
                                 storageName: payload.storageName,
                             });
+                        swarm$1.sessionValidationService.validate(payload.clientId, METHOD_NAME_UPSERT$1);
                         swarm$1.storageValidationService.validate(payload.storageName, METHOD_NAME_UPSERT$1);
                         if (!swarm$1.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
                             throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (upsert)"));
@@ -11475,6 +11946,7 @@ var StorageUtils = /** @class */ (function () {
                                 clientId: payload.clientId,
                                 storageName: payload.storageName,
                             });
+                        swarm$1.sessionValidationService.validate(payload.clientId, METHOD_NAME_REMOVE$1);
                         swarm$1.storageValidationService.validate(payload.storageName, METHOD_NAME_REMOVE$1);
                         if (!swarm$1.agentValidationService.hasStorage(payload.agentName, payload.storageName)) {
                             throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (remove)"));
@@ -11853,6 +12325,8 @@ exports.commitAssistantMessage = commitAssistantMessage;
 exports.commitAssistantMessageForce = commitAssistantMessageForce;
 exports.commitFlush = commitFlush;
 exports.commitFlushForce = commitFlushForce;
+exports.commitStopTools = commitStopTools;
+exports.commitStopToolsForce = commitStopToolsForce;
 exports.commitSystemMessage = commitSystemMessage;
 exports.commitSystemMessageForce = commitSystemMessageForce;
 exports.commitToolOutput = commitToolOutput;