agent-swarm-kit 1.0.131 → 1.0.133

package/build/index.cjs

@@ -358,11 +358,13 @@ var INSTANCE_METHOD_NAME_CTOR = "HistoryInstance.CTOR";
 var INSTANCE_METHOD_NAME_ITERATE_CONDITION = "HistoryInstance.iterate_condition";
 var INSTANCE_METHOD_NAME_ITERATE = "HistoryInstance.iterate";
 var INSTANCE_METHOD_NAME_PUSH = "HistoryInstance.push";
+var INSTANCE_METHOD_NAME_POP = "HistoryInstance.pop";
 var INSTANCE_METHOD_NAME_DISPOSE = "HistoryInstance.dispose";
 var METHOD_NAME_USE_HISTORY_ADAPTER = "HistoryUtils.useHistoryAdapter";
 var METHOD_NAME_USE_HISTORY_CALLBACKS = "HistoryUtils.useHistoryCallbacks";
 var METHOD_NAME_ITERATE = "HistoryUtils.iterate";
 var METHOD_NAME_PUSH = "HistoryUtils.push";
+var METHOD_NAME_POP = "HistoryUtils.pop";
 var METHOD_NAME_DISPOSE = "HistoryUtils.dispose";
 var HISTORY_INSTANCE_WAIT_FOR_INIT_FN = function (agentName, self) { return __awaiter(void 0, void 0, void 0, function () {
     var _b;
@@ -481,7 +483,7 @@ var HistoryInstance = /** @class */ (function () {
                                         role: "system",
                                         content: content,
                                         agentName: agentName,
-                                        mode: "tool"
+                                        mode: "tool",
                                     })];
                             case 14: return [4 /*yield*/, _p.sent()];
                             case 15:
@@ -555,7 +557,7 @@ var HistoryInstance = /** @class */ (function () {
                                         role: "system",
                                         content: content,
                                         agentName: agentName,
-                                        mode: "tool"
+                                        mode: "tool",
                                     })];
                             case 36: return [4 /*yield*/, _p.sent()];
                             case 37:
@@ -698,7 +700,7 @@ var HistoryInstance = /** @class */ (function () {
                                 role: "system",
                                 content: content,
                                 agentName: agentName,
-                                mode: "tool"
+                                mode: "tool",
                             })];
                     case 24: return [4 /*yield*/, _l.sent()];
                     case 25:
@@ -745,6 +747,25 @@ var HistoryInstance = /** @class */ (function () {
             this.callbacks.onChange(this._array, this.clientId, agentName);
         return Promise.resolve();
     };
+    /**
+     * Pop the last message from a history
+     * @param agentName - The agent name.
+     * @returns A promise that resolves when the message is pushed.
+     */
+    HistoryInstance.prototype.pop = function (agentName) {
+        var _b;
+        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+            swarm$1.loggerService.debug(INSTANCE_METHOD_NAME_POP, {
+                clientId: this.clientId,
+                agentName: agentName,
+            });
+        var value = (_b = this._array.pop()) !== null && _b !== void 0 ? _b : null;
+        this.callbacks.onPop &&
+            this.callbacks.onPop(value, this.clientId, agentName);
+        this.callbacks.onChange &&
+            this.callbacks.onChange(this._array, this.clientId, agentName);
+        return Promise.resolve(value);
+    };
     /**
      * Dispose of the history for a given agent.
      * @param agentName - The agent name or null.
@@ -827,6 +848,35 @@ var HistoryUtils = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Pop the last message from the history.
+         * @param value - The model message to push.
+         * @param clientId - The client ID.
+         * @param agentName - The agent name.
+         * @returns A promise that resolves when the message is pushed.
+         */
+        this.pop = function (clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+            var isInitial, history;
+            return __generator(this, function (_b) {
+                switch (_b.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
+                            swarm$1.loggerService.log(METHOD_NAME_POP, {
+                                clientId: clientId,
+                                agentName: agentName,
+                            });
+                        isInitial = this.getHistory.has(clientId);
+                        return [4 /*yield*/, this.getHistory(clientId)];
+                    case 1:
+                        history = _b.sent();
+                        return [4 /*yield*/, history.waitForInit(agentName, isInitial)];
+                    case 2:
+                        _b.sent();
+                        return [4 /*yield*/, history.pop(agentName)];
+                    case 3: return [2 /*return*/, _b.sent()];
+                }
+            });
+        }); };
         /**
          * Dispose of the history for a given client and agent.
          * @param clientId - The client ID.
@@ -1440,7 +1490,15 @@ var Logger = LoggerAdapter;
 /**
  * @description `ask for agent function` in `llama3.1:8b` to troubleshoot (need CC_OLLAMA_EMIT_TOOL_PROTOCOL to be turned off)
  */
-var CC_TOOL_CALL_EXCEPTION_PROMPT = "Start the conversation";
+var CC_TOOL_CALL_EXCEPTION_FLUSH_PROMPT = "Start the conversation";
+/**
+ * @description fix for invalid tool calls on IlyaGusev/saiga_yandexgpt_8b_gguf (LMStudio, appear time to time)
+ */
+var CC_TOOL_CALL_EXCEPTION_RECOMPLETE_PROMPT = "I see your previous message is malformed. Think again and resend it correct";
+/**
+ * @description custom function to fix the model
+ */
+var CC_TOOL_CALL_EXCEPTION_CUSTON_FUNCTION = function () { return Promise.resolve(); };
 /**
  * @description When the model output is empty just say hello to the customer
  */
@@ -1495,12 +1553,14 @@ var CC_LOGGER_ENABLE_INFO = false;
 var CC_LOGGER_ENABLE_DEBUG = false;
 var CC_LOGGER_ENABLE_LOG = true;
 var CC_LOGGER_ENABLE_CONSOLE = false;
+var CC_RESQUE_STRATEGY;
 var CC_NAME_TO_TITLE = nameToTitle;
 var CC_FN_PLANTUML = function () { return Promise.resolve(""); };
 var CC_PROCESS_UUID = functoolsKit.randomString();
 var CC_BANHAMMER_PLACEHOLDER = "You have been banned! To continue conversation, please contact the administrator.";
 var GLOBAL_CONFIG = {
-    CC_TOOL_CALL_EXCEPTION_PROMPT: CC_TOOL_CALL_EXCEPTION_PROMPT,
+    CC_TOOL_CALL_EXCEPTION_FLUSH_PROMPT: CC_TOOL_CALL_EXCEPTION_FLUSH_PROMPT,
+    CC_TOOL_CALL_EXCEPTION_RECOMPLETE_PROMPT: CC_TOOL_CALL_EXCEPTION_RECOMPLETE_PROMPT,
     CC_EMPTY_OUTPUT_PLACEHOLDERS: CC_EMPTY_OUTPUT_PLACEHOLDERS,
     CC_KEEP_MESSAGES: CC_KEEP_MESSAGES,
     CC_GET_AGENT_HISTORY_ADAPTER: CC_GET_AGENT_HISTORY_ADAPTER,
@@ -1522,11 +1582,14 @@ var GLOBAL_CONFIG = {
     CC_LOGGER_ENABLE_DEBUG: CC_LOGGER_ENABLE_DEBUG,
     CC_LOGGER_ENABLE_LOG: CC_LOGGER_ENABLE_LOG,
     CC_LOGGER_ENABLE_CONSOLE: CC_LOGGER_ENABLE_CONSOLE,
+    CC_RESQUE_STRATEGY: CC_RESQUE_STRATEGY,
     CC_NAME_TO_TITLE: CC_NAME_TO_TITLE,
     CC_FN_PLANTUML: CC_FN_PLANTUML,
     CC_PROCESS_UUID: CC_PROCESS_UUID,
     CC_BANHAMMER_PLACEHOLDER: CC_BANHAMMER_PLACEHOLDER,
+    CC_TOOL_CALL_EXCEPTION_CUSTON_FUNCTION: CC_TOOL_CALL_EXCEPTION_CUSTON_FUNCTION,
 };
+GLOBAL_CONFIG.CC_RESQUE_STRATEGY = "recomplete";
 var setConfig = function (config) {
     Object.assign(GLOBAL_CONFIG, config);
 };
@@ -2165,45 +2228,67 @@ var ClientAgent = /** @class */ (function () {
      * @returns {Promise<string>}
      * @private
      */
-    ClientAgent.prototype._resurrectModel = function (mode, reason) {
-        return __awaiter(this, void 0, void 0, function () {
+    ClientAgent.prototype._resurrectModel = function (mode_1) {
+        return __awaiter(this, arguments, void 0, function (mode, reason) {
             var rawMessage, message, result, validation, content;
+            if (reason === void 0) { reason = "unknown"; }
             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, " _resurrectModel"));
+                        console.warn("agent-swarm model ressurect for agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " strategy=").concat(GLOBAL_CONFIG.CC_RESQUE_STRATEGY, " reason=").concat(reason));
                         this.params.onResurrect &&
                             this.params.onResurrect(this.params.clientId, this.params.agentName, mode, reason);
+                        if (!(GLOBAL_CONFIG.CC_RESQUE_STRATEGY === "recomplete")) return [3 /*break*/, 2];
+                        return [4 /*yield*/, this.params.history.push({
+                                role: "user",
+                                mode: "tool",
+                                agentName: this.params.agentName,
+                                content: GLOBAL_CONFIG.CC_TOOL_CALL_EXCEPTION_RECOMPLETE_PROMPT,
+                            })];
+                    case 1:
+                        _a.sent();
+                        return [3 /*break*/, 8];
+                    case 2:
+                        if (!(GLOBAL_CONFIG.CC_RESQUE_STRATEGY === "flush")) return [3 /*break*/, 5];
                         return [4 /*yield*/, this.params.history.push({
                                 role: "resque",
                                 mode: "tool",
                                 agentName: this.params.agentName,
                                 content: reason || "Unknown error",
                             })];
-                    case 1:
+                    case 3:
                         _a.sent();
                         return [4 /*yield*/, this.params.history.push({
                                 role: "user",
                                 mode: "tool",
                                 agentName: this.params.agentName,
-                                content: GLOBAL_CONFIG.CC_TOOL_CALL_EXCEPTION_PROMPT,
+                                content: GLOBAL_CONFIG.CC_TOOL_CALL_EXCEPTION_FLUSH_PROMPT,
                             })];
-                    case 2:
+                    case 4:
                         _a.sent();
-                        return [4 /*yield*/, this.getCompletion(mode)];
-                    case 3:
+                        return [3 /*break*/, 8];
+                    case 5:
+                        if (!(GLOBAL_CONFIG.CC_RESQUE_STRATEGY === "custom")) return [3 /*break*/, 7];
+                        return [4 /*yield*/, GLOBAL_CONFIG.CC_TOOL_CALL_EXCEPTION_CUSTON_FUNCTION(this.params.clientId, this.params.agentName)];
+                    case 6:
+                        _a.sent();
+                        return [3 /*break*/, 8];
+                    case 7: throw new Error("agent-swarm _resurrectModel invalid strategy value=".concat(GLOBAL_CONFIG.CC_RESQUE_STRATEGY, " agentName=").concat(this.params.agentName, " clientId=").concat(this.params.clientId));
+                    case 8: return [4 /*yield*/, this.getCompletion(mode)];
+                    case 9:
                         rawMessage = _a.sent();
                         return [4 /*yield*/, this.params.map(rawMessage, this.params.clientId, this.params.agentName)];
-                    case 4:
+                    case 10:
                         message = _a.sent();
                         return [4 /*yield*/, this.params.transform(message.content, this.params.clientId, this.params.agentName)];
-                    case 5:
+                    case 11:
                         result = _a.sent();
                         validation = null;
                         return [4 /*yield*/, this.params.validate(result)];
-                    case 6:
-                        if (!(validation = _a.sent())) return [3 /*break*/, 8];
+                    case 12:
+                        if (!(validation = _a.sent())) return [3 /*break*/, 14];
                         GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                             this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " _resurrectModel validation error: ").concat(validation));
                         content = getPlaceholder();
@@ -2213,11 +2298,11 @@ var ClientAgent = /** @class */ (function () {
                                 mode: "tool",
                                 content: content,
                             })];
-                    case 7:
+                    case 13:
                         _a.sent();
                         return [2 /*return*/, content];
-                    case 8: return [4 /*yield*/, this.params.history.push(__assign(__assign({}, message), { agentName: this.params.agentName }))];
-                    case 9:
+                    case 14: return [4 /*yield*/, this.params.history.push(__assign(__assign({}, message), { agentName: this.params.agentName }))];
+                    case 15:
                         _a.sent();
                         return [2 /*return*/, result];
                 }
@@ -2855,6 +2940,40 @@ var ClientHistory = /** @class */ (function () {
             });
         });
     };
+    /**
+     * Pushes a message to the history.
+     * @returns {Promise<IModelMessage | null>}
+     */
+    ClientHistory.prototype.pop = function () {
+        return __awaiter(this, void 0, void 0, function () {
+            var value;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            this.params.logger.debug("ClientHistory agentName=".concat(this.params.agentName, " pop"));
+                        return [4 /*yield*/, this.params.items.pop(this.params.clientId, this.params.agentName)];
+                    case 1:
+                        value = _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "pop",
+                                source: "history-bus",
+                                input: {},
+                                output: {
+                                    value: value
+                                },
+                                context: {
+                                    agentName: this.params.agentName,
+                                },
+                                clientId: this.params.clientId,
+                            })];
+                    case 2:
+                        _a.sent();
+                        return [2 /*return*/, value];
+                }
+            });
+        });
+    };
     /**
      * Converts the history to an array of raw messages.
      * @returns {Promise<IModelMessage[]>} - The array of raw messages.
@@ -3127,6 +3246,21 @@ var HistoryConnectionService = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Pop a message from the history.
+         * @returns {Promise<IModelMessage | null>} A promise that resolves when the message is popped.
+         */
+        this.pop = 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("historyConnectionService pop");
+                        return [4 /*yield*/, this.getHistory(this.methodContextService.context.clientId, this.methodContextService.context.agentName).pop()];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Converts the history to an array for the agent.
          * @param {string} prompt - The prompt.
@@ -5553,6 +5687,43 @@ var HistoryPublicService = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Pushes a message to the history.
+         * @param {string} clientId - The client ID.
+         * @param {AgentName} agentName - The agent name.
+         * @returns {Promise<IModelMessage | null>} A promise that resolves when the operation is complete.
+         */
+        this.pop = 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("historyPublicService pop", {
+                                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.historyConnectionService.pop()];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                methodName: methodName,
+                                clientId: clientId,
+                                agentName: agentName,
+                                policyName: "",
+                                swarmName: "",
+                                storageName: "",
+                                stateName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Converts history to an array for a specific agent.
          * @param {string} prompt - The prompt.
@@ -15556,6 +15727,7 @@ var SchemaUtils = /** @class */ (function () {
          * @returns {string} The serialized string.
          */
         this.serialize = function (data, map) {
+            if (map === void 0) { map = {}; }
             GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
                 swarm$1.loggerService.log(METHOD_NAME_SERIALIZE, {
                     data: data,

package/build/index.mjs

@@ -356,11 +356,13 @@ var INSTANCE_METHOD_NAME_CTOR = "HistoryInstance.CTOR";
 var INSTANCE_METHOD_NAME_ITERATE_CONDITION = "HistoryInstance.iterate_condition";
 var INSTANCE_METHOD_NAME_ITERATE = "HistoryInstance.iterate";
 var INSTANCE_METHOD_NAME_PUSH = "HistoryInstance.push";
+var INSTANCE_METHOD_NAME_POP = "HistoryInstance.pop";
 var INSTANCE_METHOD_NAME_DISPOSE = "HistoryInstance.dispose";
 var METHOD_NAME_USE_HISTORY_ADAPTER = "HistoryUtils.useHistoryAdapter";
 var METHOD_NAME_USE_HISTORY_CALLBACKS = "HistoryUtils.useHistoryCallbacks";
 var METHOD_NAME_ITERATE = "HistoryUtils.iterate";
 var METHOD_NAME_PUSH = "HistoryUtils.push";
+var METHOD_NAME_POP = "HistoryUtils.pop";
 var METHOD_NAME_DISPOSE = "HistoryUtils.dispose";
 var HISTORY_INSTANCE_WAIT_FOR_INIT_FN = function (agentName, self) { return __awaiter(void 0, void 0, void 0, function () {
     var _b;
@@ -479,7 +481,7 @@ var HistoryInstance = /** @class */ (function () {
                                         role: "system",
                                         content: content,
                                         agentName: agentName,
-                                        mode: "tool"
+                                        mode: "tool",
                                     })];
                             case 14: return [4 /*yield*/, _p.sent()];
                             case 15:
@@ -553,7 +555,7 @@ var HistoryInstance = /** @class */ (function () {
                                         role: "system",
                                         content: content,
                                         agentName: agentName,
-                                        mode: "tool"
+                                        mode: "tool",
                                     })];
                             case 36: return [4 /*yield*/, _p.sent()];
                             case 37:
@@ -696,7 +698,7 @@ var HistoryInstance = /** @class */ (function () {
                                 role: "system",
                                 content: content,
                                 agentName: agentName,
-                                mode: "tool"
+                                mode: "tool",
                             })];
                     case 24: return [4 /*yield*/, _l.sent()];
                     case 25:
@@ -743,6 +745,25 @@ var HistoryInstance = /** @class */ (function () {
             this.callbacks.onChange(this._array, this.clientId, agentName);
         return Promise.resolve();
     };
+    /**
+     * Pop the last message from a history
+     * @param agentName - The agent name.
+     * @returns A promise that resolves when the message is pushed.
+     */
+    HistoryInstance.prototype.pop = function (agentName) {
+        var _b;
+        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+            swarm$1.loggerService.debug(INSTANCE_METHOD_NAME_POP, {
+                clientId: this.clientId,
+                agentName: agentName,
+            });
+        var value = (_b = this._array.pop()) !== null && _b !== void 0 ? _b : null;
+        this.callbacks.onPop &&
+            this.callbacks.onPop(value, this.clientId, agentName);
+        this.callbacks.onChange &&
+            this.callbacks.onChange(this._array, this.clientId, agentName);
+        return Promise.resolve(value);
+    };
     /**
      * Dispose of the history for a given agent.
      * @param agentName - The agent name or null.
@@ -825,6 +846,35 @@ var HistoryUtils = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Pop the last message from the history.
+         * @param value - The model message to push.
+         * @param clientId - The client ID.
+         * @param agentName - The agent name.
+         * @returns A promise that resolves when the message is pushed.
+         */
+        this.pop = function (clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+            var isInitial, history;
+            return __generator(this, function (_b) {
+                switch (_b.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
+                            swarm$1.loggerService.log(METHOD_NAME_POP, {
+                                clientId: clientId,
+                                agentName: agentName,
+                            });
+                        isInitial = this.getHistory.has(clientId);
+                        return [4 /*yield*/, this.getHistory(clientId)];
+                    case 1:
+                        history = _b.sent();
+                        return [4 /*yield*/, history.waitForInit(agentName, isInitial)];
+                    case 2:
+                        _b.sent();
+                        return [4 /*yield*/, history.pop(agentName)];
+                    case 3: return [2 /*return*/, _b.sent()];
+                }
+            });
+        }); };
         /**
          * Dispose of the history for a given client and agent.
          * @param clientId - The client ID.
@@ -1438,7 +1488,15 @@ var Logger = LoggerAdapter;
 /**
  * @description `ask for agent function` in `llama3.1:8b` to troubleshoot (need CC_OLLAMA_EMIT_TOOL_PROTOCOL to be turned off)
  */
-var CC_TOOL_CALL_EXCEPTION_PROMPT = "Start the conversation";
+var CC_TOOL_CALL_EXCEPTION_FLUSH_PROMPT = "Start the conversation";
+/**
+ * @description fix for invalid tool calls on IlyaGusev/saiga_yandexgpt_8b_gguf (LMStudio, appear time to time)
+ */
+var CC_TOOL_CALL_EXCEPTION_RECOMPLETE_PROMPT = "I see your previous message is malformed. Think again and resend it correct";
+/**
+ * @description custom function to fix the model
+ */
+var CC_TOOL_CALL_EXCEPTION_CUSTON_FUNCTION = function () { return Promise.resolve(); };
 /**
  * @description When the model output is empty just say hello to the customer
  */
@@ -1493,12 +1551,14 @@ var CC_LOGGER_ENABLE_INFO = false;
 var CC_LOGGER_ENABLE_DEBUG = false;
 var CC_LOGGER_ENABLE_LOG = true;
 var CC_LOGGER_ENABLE_CONSOLE = false;
+var CC_RESQUE_STRATEGY;
 var CC_NAME_TO_TITLE = nameToTitle;
 var CC_FN_PLANTUML = function () { return Promise.resolve(""); };
 var CC_PROCESS_UUID = randomString();
 var CC_BANHAMMER_PLACEHOLDER = "You have been banned! To continue conversation, please contact the administrator.";
 var GLOBAL_CONFIG = {
-    CC_TOOL_CALL_EXCEPTION_PROMPT: CC_TOOL_CALL_EXCEPTION_PROMPT,
+    CC_TOOL_CALL_EXCEPTION_FLUSH_PROMPT: CC_TOOL_CALL_EXCEPTION_FLUSH_PROMPT,
+    CC_TOOL_CALL_EXCEPTION_RECOMPLETE_PROMPT: CC_TOOL_CALL_EXCEPTION_RECOMPLETE_PROMPT,
     CC_EMPTY_OUTPUT_PLACEHOLDERS: CC_EMPTY_OUTPUT_PLACEHOLDERS,
     CC_KEEP_MESSAGES: CC_KEEP_MESSAGES,
     CC_GET_AGENT_HISTORY_ADAPTER: CC_GET_AGENT_HISTORY_ADAPTER,
@@ -1520,11 +1580,14 @@ var GLOBAL_CONFIG = {
     CC_LOGGER_ENABLE_DEBUG: CC_LOGGER_ENABLE_DEBUG,
     CC_LOGGER_ENABLE_LOG: CC_LOGGER_ENABLE_LOG,
     CC_LOGGER_ENABLE_CONSOLE: CC_LOGGER_ENABLE_CONSOLE,
+    CC_RESQUE_STRATEGY: CC_RESQUE_STRATEGY,
     CC_NAME_TO_TITLE: CC_NAME_TO_TITLE,
     CC_FN_PLANTUML: CC_FN_PLANTUML,
     CC_PROCESS_UUID: CC_PROCESS_UUID,
     CC_BANHAMMER_PLACEHOLDER: CC_BANHAMMER_PLACEHOLDER,
+    CC_TOOL_CALL_EXCEPTION_CUSTON_FUNCTION: CC_TOOL_CALL_EXCEPTION_CUSTON_FUNCTION,
 };
+GLOBAL_CONFIG.CC_RESQUE_STRATEGY = "recomplete";
 var setConfig = function (config) {
     Object.assign(GLOBAL_CONFIG, config);
 };
@@ -2163,45 +2226,67 @@ var ClientAgent = /** @class */ (function () {
      * @returns {Promise<string>}
      * @private
      */
-    ClientAgent.prototype._resurrectModel = function (mode, reason) {
-        return __awaiter(this, void 0, void 0, function () {
+    ClientAgent.prototype._resurrectModel = function (mode_1) {
+        return __awaiter(this, arguments, void 0, function (mode, reason) {
             var rawMessage, message, result, validation, content;
+            if (reason === void 0) { reason = "unknown"; }
             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, " _resurrectModel"));
+                        console.warn("agent-swarm model ressurect for agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " strategy=").concat(GLOBAL_CONFIG.CC_RESQUE_STRATEGY, " reason=").concat(reason));
                         this.params.onResurrect &&
                             this.params.onResurrect(this.params.clientId, this.params.agentName, mode, reason);
+                        if (!(GLOBAL_CONFIG.CC_RESQUE_STRATEGY === "recomplete")) return [3 /*break*/, 2];
+                        return [4 /*yield*/, this.params.history.push({
+                                role: "user",
+                                mode: "tool",
+                                agentName: this.params.agentName,
+                                content: GLOBAL_CONFIG.CC_TOOL_CALL_EXCEPTION_RECOMPLETE_PROMPT,
+                            })];
+                    case 1:
+                        _a.sent();
+                        return [3 /*break*/, 8];
+                    case 2:
+                        if (!(GLOBAL_CONFIG.CC_RESQUE_STRATEGY === "flush")) return [3 /*break*/, 5];
                         return [4 /*yield*/, this.params.history.push({
                                 role: "resque",
                                 mode: "tool",
                                 agentName: this.params.agentName,
                                 content: reason || "Unknown error",
                             })];
-                    case 1:
+                    case 3:
                         _a.sent();
                         return [4 /*yield*/, this.params.history.push({
                                 role: "user",
                                 mode: "tool",
                                 agentName: this.params.agentName,
-                                content: GLOBAL_CONFIG.CC_TOOL_CALL_EXCEPTION_PROMPT,
+                                content: GLOBAL_CONFIG.CC_TOOL_CALL_EXCEPTION_FLUSH_PROMPT,
                             })];
-                    case 2:
+                    case 4:
                         _a.sent();
-                        return [4 /*yield*/, this.getCompletion(mode)];
-                    case 3:
+                        return [3 /*break*/, 8];
+                    case 5:
+                        if (!(GLOBAL_CONFIG.CC_RESQUE_STRATEGY === "custom")) return [3 /*break*/, 7];
+                        return [4 /*yield*/, GLOBAL_CONFIG.CC_TOOL_CALL_EXCEPTION_CUSTON_FUNCTION(this.params.clientId, this.params.agentName)];
+                    case 6:
+                        _a.sent();
+                        return [3 /*break*/, 8];
+                    case 7: throw new Error("agent-swarm _resurrectModel invalid strategy value=".concat(GLOBAL_CONFIG.CC_RESQUE_STRATEGY, " agentName=").concat(this.params.agentName, " clientId=").concat(this.params.clientId));
+                    case 8: return [4 /*yield*/, this.getCompletion(mode)];
+                    case 9:
                         rawMessage = _a.sent();
                         return [4 /*yield*/, this.params.map(rawMessage, this.params.clientId, this.params.agentName)];
-                    case 4:
+                    case 10:
                         message = _a.sent();
                         return [4 /*yield*/, this.params.transform(message.content, this.params.clientId, this.params.agentName)];
-                    case 5:
+                    case 11:
                         result = _a.sent();
                         validation = null;
                         return [4 /*yield*/, this.params.validate(result)];
-                    case 6:
-                        if (!(validation = _a.sent())) return [3 /*break*/, 8];
+                    case 12:
+                        if (!(validation = _a.sent())) return [3 /*break*/, 14];
                         GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                             this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " _resurrectModel validation error: ").concat(validation));
                         content = getPlaceholder();
@@ -2211,11 +2296,11 @@ var ClientAgent = /** @class */ (function () {
                                 mode: "tool",
                                 content: content,
                             })];
-                    case 7:
+                    case 13:
                         _a.sent();
                         return [2 /*return*/, content];
-                    case 8: return [4 /*yield*/, this.params.history.push(__assign(__assign({}, message), { agentName: this.params.agentName }))];
-                    case 9:
+                    case 14: return [4 /*yield*/, this.params.history.push(__assign(__assign({}, message), { agentName: this.params.agentName }))];
+                    case 15:
                         _a.sent();
                         return [2 /*return*/, result];
                 }
@@ -2853,6 +2938,40 @@ var ClientHistory = /** @class */ (function () {
             });
         });
     };
+    /**
+     * Pushes a message to the history.
+     * @returns {Promise<IModelMessage | null>}
+     */
+    ClientHistory.prototype.pop = function () {
+        return __awaiter(this, void 0, void 0, function () {
+            var value;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            this.params.logger.debug("ClientHistory agentName=".concat(this.params.agentName, " pop"));
+                        return [4 /*yield*/, this.params.items.pop(this.params.clientId, this.params.agentName)];
+                    case 1:
+                        value = _a.sent();
+                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                                type: "pop",
+                                source: "history-bus",
+                                input: {},
+                                output: {
+                                    value: value
+                                },
+                                context: {
+                                    agentName: this.params.agentName,
+                                },
+                                clientId: this.params.clientId,
+                            })];
+                    case 2:
+                        _a.sent();
+                        return [2 /*return*/, value];
+                }
+            });
+        });
+    };
     /**
      * Converts the history to an array of raw messages.
      * @returns {Promise<IModelMessage[]>} - The array of raw messages.
@@ -3125,6 +3244,21 @@ var HistoryConnectionService = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Pop a message from the history.
+         * @returns {Promise<IModelMessage | null>} A promise that resolves when the message is popped.
+         */
+        this.pop = 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("historyConnectionService pop");
+                        return [4 /*yield*/, this.getHistory(this.methodContextService.context.clientId, this.methodContextService.context.agentName).pop()];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Converts the history to an array for the agent.
          * @param {string} prompt - The prompt.
@@ -5551,6 +5685,43 @@ var HistoryPublicService = /** @class */ (function () {
                 }
             });
         }); };
+        /**
+         * Pushes a message to the history.
+         * @param {string} clientId - The client ID.
+         * @param {AgentName} agentName - The agent name.
+         * @returns {Promise<IModelMessage | null>} A promise that resolves when the operation is complete.
+         */
+        this.pop = 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("historyPublicService pop", {
+                                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.historyConnectionService.pop()];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                methodName: methodName,
+                                clientId: clientId,
+                                agentName: agentName,
+                                policyName: "",
+                                swarmName: "",
+                                storageName: "",
+                                stateName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         /**
          * Converts history to an array for a specific agent.
          * @param {string} prompt - The prompt.
@@ -15554,6 +15725,7 @@ var SchemaUtils = /** @class */ (function () {
          * @returns {string} The serialized string.
          */
         this.serialize = function (data, map) {
+            if (map === void 0) { map = {}; }
             GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
                 swarm$1.loggerService.log(METHOD_NAME_SERIALIZE, {
                     data: data,

package/package.json

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

package/types.d.ts

@@ -1076,6 +1076,13 @@ interface IHistoryInstanceCallbacks {
      * @param agentName - The agent name.
      */
     onPush: (data: IModelMessage, clientId: string, agentName: AgentName) => void;
+    /**
+     * Callback for when the history pop the last message
+     * @param data - The array of model messages.
+     * @param clientId - The client ID.
+     * @param agentName - The agent name.
+     */
+    onPop: (data: IModelMessage | null, clientId: string, agentName: AgentName) => void;
     /**
      * Callback for when the history is read. Will be called for each message
      * @param message - The model message.
@@ -1130,6 +1137,13 @@ interface IHistoryAdapter {
      * @returns A promise that resolves when the message is pushed.
      */
     push: (value: IModelMessage, clientId: string, agentName: AgentName) => Promise<void>;
+    /**
+     * Pop the last message from a history
+     * @param clientId - The client ID.
+     * @param agentName - The agent name.
+     * @returns The last message or null
+     */
+    pop: (clientId: string, agentName: AgentName) => Promise<IModelMessage | null>;
     /**
      * Dispose of the history for a given client and agent.
      * @param clientId - The client ID.
@@ -1177,6 +1191,13 @@ interface IHistoryInstance {
      * @returns A promise that resolves when the message is pushed.
      */
     push(value: IModelMessage, agentName: AgentName): Promise<void>;
+    /**
+     * Pop the last message from a history
+     * @param value - The model message to push.
+     * @param agentName - The agent name.
+     * @returns A promise that resolves the last message or null
+     */
+    pop(agentName: AgentName): Promise<IModelMessage | null>;
     /**
      * Dispose of the history for a given agent.
      * @param agentName - The agent name or null.
@@ -1225,6 +1246,12 @@ declare class HistoryInstance implements IHistoryInstance {
      * @returns A promise that resolves when the message is pushed.
      */
     push(value: IModelMessage, agentName: AgentName): Promise<void>;
+    /**
+     * Pop the last message from a history
+     * @param agentName - The agent name.
+     * @returns A promise that resolves when the message is pushed.
+     */
+    pop(agentName: AgentName): Promise<IModelMessage>;
     /**
      * Dispose of the history for a given agent.
      * @param agentName - The agent name or null.
@@ -1264,6 +1291,14 @@ declare class HistoryUtils implements IHistoryAdapter, IHistoryControl {
      * @returns A promise that resolves when the message is pushed.
      */
     push: (value: IModelMessage, clientId: string, agentName: AgentName) => Promise<void>;
+    /**
+     * Pop the last message from the history.
+     * @param value - The model message to push.
+     * @param clientId - The client ID.
+     * @param agentName - The agent name.
+     * @returns A promise that resolves when the message is pushed.
+     */
+    pop: (clientId: string, agentName: AgentName) => Promise<IModelMessage>;
     /**
      * Dispose of the history for a given client and agent.
      * @param clientId - The client ID.
@@ -1291,6 +1326,11 @@ interface IHistory {
      * @returns {Promise<void>}
      */
     push(message: IModelMessage): Promise<void>;
+    /**
+     * Pop the last message from a history
+     * @returns {Promise<IModelMessage | null>}
+     */
+    pop(): Promise<IModelMessage | null>;
     /**
      * Converts the history to an array of messages for a specific agent.
      * @param {string} prompt - The prompt to filter messages for the agent.
@@ -1401,11 +1441,12 @@ interface ICompletionSchema {
  */
 type CompletionName = string;
 
+type ToolValue = string | number | boolean | null;
 /**
  * Interface representing lifecycle callbacks of a tool
  * @template T - The type of the parameters for the tool.
  */
-interface IAgentToolCallbacks<T = Record<string, unknown>> {
+interface IAgentToolCallbacks<T = Record<string, ToolValue>> {
     /**
      * Callback triggered before the tool is called.
      * @param toolId - The `tool_call_id` for openai history
@@ -1445,7 +1486,7 @@ interface IAgentToolCallbacks<T = Record<string, unknown>> {
  * Interface representing a tool used by an agent.
  * @template T - The type of the parameters for the tool.
  */
-interface IAgentTool<T = Record<string, unknown>> extends ITool {
+interface IAgentTool<T = Record<string, ToolValue>> extends ITool {
     /** The description for documentation */
     docNote?: string;
     /** The name of the tool. */
@@ -1972,6 +2013,11 @@ declare class ClientHistory implements IHistory {
      * @returns {Promise<void>}
      */
     push(message: IModelMessage): Promise<void>;
+    /**
+     * Pushes a message to the history.
+     * @returns {Promise<IModelMessage | null>}
+     */
+    pop(): Promise<IModelMessage | null>;
     /**
      * Converts the history to an array of raw messages.
      * @returns {Promise<IModelMessage[]>} - The array of raw messages.
@@ -2013,6 +2059,11 @@ declare class HistoryConnectionService implements IHistory {
      * @returns {Promise<void>} A promise that resolves when the message is pushed.
      */
     push: (message: IModelMessage) => Promise<void>;
+    /**
+     * Pop a message from the history.
+     * @returns {Promise<IModelMessage | null>} A promise that resolves when the message is popped.
+     */
+    pop: () => Promise<IModelMessage>;
     /**
      * Converts the history to an array for the agent.
      * @param {string} prompt - The prompt.
@@ -2542,6 +2593,13 @@ declare class HistoryPublicService implements THistoryConnectionService {
      * @returns {Promise<void>} A promise that resolves when the operation is complete.
      */
     push: (message: IModelMessage, methodName: string, clientId: string, agentName: AgentName) => Promise<void>;
+    /**
+     * Pushes a message to the history.
+     * @param {string} clientId - The client ID.
+     * @param {AgentName} agentName - The agent name.
+     * @returns {Promise<IModelMessage | null>} A promise that resolves when the operation is complete.
+     */
+    pop: (methodName: string, clientId: string, agentName: AgentName) => Promise<IModelMessage>;
     /**
      * Converts history to an array for a specific agent.
      * @param {string} prompt - The prompt.
@@ -4397,7 +4455,7 @@ declare const addSwarm: (swarmSchema: ISwarmSchema) => string;
  * @param {IAgentTool} toolSchema - The schema of the tool to be added.
  * @returns {string} The name of the tool that was added.
  */
-declare const addTool: (toolSchema: IAgentTool<Record<string, unknown>>) => string;
+declare const addTool: (toolSchema: IAgentTool<Record<string, string | number | boolean>>) => string;
 
 /**
  * Adds a new state to the state registry. The swarm takes only those states which was registered
@@ -5277,7 +5335,8 @@ declare const LoggerAdapter: LoggerUtils;
 declare const Logger: ILoggerControl;
 
 declare const GLOBAL_CONFIG: {
-    CC_TOOL_CALL_EXCEPTION_PROMPT: string;
+    CC_TOOL_CALL_EXCEPTION_FLUSH_PROMPT: string;
+    CC_TOOL_CALL_EXCEPTION_RECOMPLETE_PROMPT: string;
     CC_EMPTY_OUTPUT_PLACEHOLDERS: string[];
     CC_KEEP_MESSAGES: number;
     CC_GET_AGENT_HISTORY_ADAPTER: (clientId: string, agentName: AgentName) => IHistoryAdapter;
@@ -5299,10 +5358,12 @@ declare const GLOBAL_CONFIG: {
     CC_LOGGER_ENABLE_DEBUG: boolean;
     CC_LOGGER_ENABLE_LOG: boolean;
     CC_LOGGER_ENABLE_CONSOLE: boolean;
+    CC_RESQUE_STRATEGY: "flush" | "recomplete" | "custom";
     CC_NAME_TO_TITLE: (name: string) => string;
     CC_FN_PLANTUML: (uml: string) => Promise<string>;
     CC_PROCESS_UUID: string;
     CC_BANHAMMER_PLACEHOLDER: string;
+    CC_TOOL_CALL_EXCEPTION_CUSTON_FUNCTION: (clientId: string, agentName: AgentName) => Promise<void>;
 };
 declare const setConfig: (config: Partial<typeof GLOBAL_CONFIG>) => void;
 
@@ -5640,9 +5701,9 @@ declare class SchemaUtils {
      * @param {T[] | T} data - The data to serialize.
      * @returns {string} The serialized string.
      */
-    serialize: <T extends object = any>(data: T[] | T, map: {
-        mapKey: typeof GLOBAL_CONFIG.CC_NAME_TO_TITLE;
-        mapValue: (key: string, value: string) => string;
+    serialize: <T extends object = any>(data: T[] | T, map?: {
+        mapKey?: typeof GLOBAL_CONFIG.CC_NAME_TO_TITLE;
+        mapValue?: (key: string, value: string) => string;
     }) => string;
 }
 /**