agent-swarm-kit 1.0.111 → 1.0.113

package/build/index.mjs

@@ -1669,6 +1669,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)];
 };
@@ -1686,6 +1687,7 @@ var ClientAgent = /** @class */ (function () {
         this.params = params;
         this._agentChangeSubject = new Subject();
         this._toolErrorSubject = new Subject();
+        this._toolStopSubject = new Subject();
         this._toolCommitSubject = new Subject();
         this._outputSubject = new Subject();
         /**
@@ -1985,6 +1987,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.
@@ -2115,9 +2146,9 @@ var ClientAgent = /** @class */ (function () {
         this.execute = 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 });
@@ -2130,13 +2161,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"));
@@ -2150,11 +2181,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; });
@@ -2163,12 +2194,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) &&
@@ -2180,17 +2211,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) &&
@@ -2230,10 +2261,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) {
@@ -2243,17 +2275,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*/];
                                 }
@@ -2261,20 +2300,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:
@@ -2284,29 +2323,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*/];
                 }
             });
@@ -2497,6 +2536,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.
@@ -3790,6 +3844,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.
@@ -4117,6 +4204,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.
@@ -4489,6 +4592,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.
@@ -5002,6 +5141,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.
@@ -7254,6 +7428,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>}
@@ -7391,6 +7612,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.
@@ -7488,6 +7729,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.
@@ -8357,6 +8634,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.
@@ -8590,6 +8887,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.
@@ -8889,12 +9220,6 @@ var SharedStoragePublicService = /** @class */ (function () {
     provide(TYPES.swarmMetaService, function () { return new SwarmMetaService(); });
     provide(TYPES.agentMetaService, function () { return new AgentMetaService(); });
 }
-{
-    provide(TYPES.agentPublicService, function () { return new AgentPublicService(); });
-    provide(TYPES.historyPublicService, function () { return new HistoryPublicService(); });
-    provide(TYPES.sessionPublicService, function () { return new SessionPublicService(); });
-    provide(TYPES.swarmPublicService, function () { return new SwarmPublicService(); });
-}
 {
     provide(TYPES.agentValidationService, function () { return new AgentValidationService(); });
     provide(TYPES.completionValidationService, function () { return new CompletionValidationService(); });
@@ -8960,7 +9285,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.
  *
@@ -8971,7 +9296,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) {
@@ -8982,12 +9307,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;
@@ -8998,7 +9323,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.
  *
@@ -9008,14 +9333,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.
  *
@@ -9024,14 +9349,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
  *
@@ -9040,7 +9365,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);
@@ -9048,7 +9373,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
@@ -9058,7 +9383,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);
@@ -9066,7 +9391,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
  *
@@ -9075,7 +9400,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);
@@ -9083,7 +9408,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
@@ -9093,7 +9418,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);
@@ -9101,7 +9426,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
  *
@@ -9110,7 +9435,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);
@@ -9122,7 +9447,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
  *
@@ -9131,7 +9456,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);
@@ -9139,7 +9464,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
  *
@@ -9148,7 +9473,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);
@@ -9161,7 +9486,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
  *
@@ -9176,17 +9501,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) {
@@ -9199,7 +9524,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*/];
@@ -9207,7 +9532,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.
  *
@@ -9222,16 +9547,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) {
@@ -9243,7 +9568,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*/];
@@ -9251,7 +9576,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
  *
@@ -9265,15 +9590,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) {
@@ -9285,7 +9610,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*/];
@@ -9293,7 +9618,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.
  *
@@ -9308,16 +9633,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) {
@@ -9329,7 +9654,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*/];
@@ -9337,7 +9662,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
  *
@@ -9351,15 +9676,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*/];
@@ -9367,7 +9692,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.
  *
@@ -9381,14 +9706,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*/];
@@ -9396,7 +9721,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
  *
@@ -9409,14 +9734,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*/];
@@ -9424,7 +9749,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
  *
@@ -9438,14 +9763,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*/];
@@ -9453,7 +9778,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.
  *
@@ -9468,16 +9793,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) {
@@ -9489,7 +9814,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*/];
@@ -9497,7 +9822,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.
  *
@@ -9511,14 +9836,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*/];
@@ -9526,7 +9851,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
  *
@@ -9540,15 +9865,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) {
@@ -9560,7 +9885,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*/];
@@ -9568,7 +9893,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
  *
@@ -9577,6 +9902,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) {
@@ -9584,11 +9978,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*/];
@@ -11246,6 +11641,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}
@@ -11316,6 +11712,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;
 }());
@@ -11327,6 +11751,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}
@@ -11391,6 +11816,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;
 }());
@@ -11838,4 +12288,4 @@ var SchemaUtils = /** @class */ (function () {
  */
 var Schema = new SchemaUtils();
 
-export { ExecutionContextService, History, HistoryAdapter, HistoryInstance, Logger, LoggerAdapter, LoggerInstance, MethodContextService, Schema, SharedState, SharedStorage, State, Storage, addAgent, addCompletion, addEmbedding, addState, addStorage, addSwarm, addTool, cancelOutput, cancelOutputForce, changeToAgent, changeToDefaultAgent, changeToPrevAgent, commitAssistantMessage, commitAssistantMessageForce, commitFlush, commitFlushForce, commitSystemMessage, commitSystemMessageForce, commitToolOutput, commitToolOutputForce, commitUserMessage, commitUserMessageForce, complete, disposeConnection, dumpAgent, dumpDocs, dumpSwarm, emit, emitForce, event, execute, executeForce, getAgentHistory, getAgentName, getAssistantHistory, getLastAssistantMessage, getLastSystemMessage, getLastUserMessage, getRawHistory, getSessionMode, getUserHistory, listenAgentEvent, listenAgentEventOnce, listenEvent, listenEventOnce, listenHistoryEvent, listenHistoryEventOnce, listenSessionEvent, listenSessionEventOnce, listenStateEvent, listenStateEventOnce, listenStorageEvent, listenStorageEventOnce, listenSwarmEvent, listenSwarmEventOnce, makeAutoDispose, makeConnection, session, setConfig, swarm };
+export { ExecutionContextService, History, HistoryAdapter, HistoryInstance, Logger, LoggerAdapter, LoggerInstance, MethodContextService, Schema, SharedState, SharedStorage, State, Storage, addAgent, addCompletion, addEmbedding, addState, addStorage, addSwarm, addTool, cancelOutput, cancelOutputForce, changeToAgent, changeToDefaultAgent, changeToPrevAgent, commitAssistantMessage, commitAssistantMessageForce, commitFlush, commitFlushForce, commitStopTools, commitStopToolsForce, commitSystemMessage, commitSystemMessageForce, commitToolOutput, commitToolOutputForce, commitUserMessage, commitUserMessageForce, complete, disposeConnection, dumpAgent, dumpDocs, dumpSwarm, emit, emitForce, event, execute, executeForce, getAgentHistory, getAgentName, getAssistantHistory, getLastAssistantMessage, getLastSystemMessage, getLastUserMessage, getRawHistory, getSessionMode, getUserHistory, listenAgentEvent, listenAgentEventOnce, listenEvent, listenEventOnce, listenHistoryEvent, listenHistoryEventOnce, listenSessionEvent, listenSessionEventOnce, listenStateEvent, listenStateEventOnce, listenStorageEvent, listenStorageEventOnce, listenSwarmEvent, listenSwarmEventOnce, makeAutoDispose, makeConnection, session, setConfig, swarm };