agent-swarm-kit 1.0.34 → 1.0.36

package/build/index.cjs

@@ -355,6 +355,10 @@ var validateDefault = function (output) { return __awaiter(void 0, void 0, void
     });
 }); };
 
+var removeXmlTags = function (input) {
+    return input.replace(/<[^>]*>.*?<\/[^>]*>/g, '');
+};
+
 /**
  * @description `ask for agent function` in `llama3.1:8b` to troubleshoot (need CC_OLLAMA_EMIT_TOOL_PROTOCOL to be turned off)
  */
@@ -396,18 +400,29 @@ var CC_AGENT_DISALLOWED_SYMBOLS = [
     "{",
     "}",
 ];
+var CC_AGENT_HISTORY_FILTER = function (agentName) { return function (message) {
+    var isOk = true;
+    if (message.role === "tool") {
+        isOk = isOk && message.agentName === agentName;
+    }
+    if (message.tool_calls) {
+        isOk = isOk && message.agentName === agentName;
+    }
+    return isOk;
+}; };
+var CC_AGENT_OUTPUT_TRANSFORM = removeXmlTags;
 var CC_KEEP_MESSAGES = 5;
-var CC_ANSWER_TIMEOUT = 120000;
 var CC_GET_AGENT_HISTORY = function () { return new functoolsKit.PubsubArrayAdapter(); };
 var GLOBAL_CONFIG = {
     CC_TOOL_CALL_EXCEPTION_PROMPT: CC_TOOL_CALL_EXCEPTION_PROMPT,
     CC_EMPTY_OUTPUT_PLACEHOLDERS: CC_EMPTY_OUTPUT_PLACEHOLDERS,
     CC_KEEP_MESSAGES: CC_KEEP_MESSAGES,
-    CC_ANSWER_TIMEOUT: CC_ANSWER_TIMEOUT,
     CC_GET_AGENT_HISTORY: CC_GET_AGENT_HISTORY,
     CC_SWARM_AGENT_CHANGED: CC_SWARM_AGENT_CHANGED,
     CC_SWARM_DEFAULT_AGENT: CC_SWARM_DEFAULT_AGENT,
     CC_AGENT_DEFAULT_VALIDATION: CC_AGENT_DEFAULT_VALIDATION,
+    CC_AGENT_HISTORY_FILTER: CC_AGENT_HISTORY_FILTER,
+    CC_AGENT_OUTPUT_TRANSFORM: CC_AGENT_OUTPUT_TRANSFORM,
     CC_AGENT_DISALLOWED_TAGS: CC_AGENT_DISALLOWED_TAGS,
     CC_AGENT_DISALLOWED_SYMBOLS: CC_AGENT_DISALLOWED_SYMBOLS,
 };
@@ -438,19 +453,21 @@ var ClientAgent = /** @class */ (function () {
          * @returns {Promise<void>}
          * @private
          */
-        this._emitOuput = function (mode, result) { return __awaiter(_this, void 0, void 0, function () {
-            var validation, result_1;
+        this._emitOuput = function (mode, rawResult) { return __awaiter(_this, void 0, void 0, function () {
+            var result, validation, rawResult_1, result_1;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " _emitOuput"));
+                        result = this.params.transform(rawResult);
+                        this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " _emitOuput"), { mode: mode, result: result, rawResult: rawResult });
                         validation = null;
                         return [4 /*yield*/, this.params.validate(result)];
                     case 1:
                         if (!(validation = _a.sent())) return [3 /*break*/, 5];
                         return [4 /*yield*/, this._resurrectModel(mode, validation)];
                     case 2:
-                        result_1 = _a.sent();
+                        rawResult_1 = _a.sent();
+                        result_1 = this.params.transform(rawResult_1);
                         return [4 /*yield*/, this.params.validate(result_1)];
                     case 3:
                         if ((validation = _a.sent())) {
@@ -856,10 +873,10 @@ var AgentConnectionService = /** @class */ (function () {
             var _b = __read(_a, 2), clientId = _b[0], agentName = _b[1];
             return "".concat(clientId, "-").concat(agentName);
         }, function (clientId, agentName) {
-            var _a = _this.agentSchemaService.get(agentName), prompt = _a.prompt, system = _a.system, tools = _a.tools, callbacks = _a.callbacks, completionName = _a.completion, _b = _a.validate, validate = _b === void 0 ? validateDefault : _b;
+            var _a = _this.agentSchemaService.get(agentName), prompt = _a.prompt, system = _a.system, tools = _a.tools, _b = _a.transform, transform = _b === void 0 ? GLOBAL_CONFIG.CC_AGENT_OUTPUT_TRANSFORM : _b, callbacks = _a.callbacks, completionName = _a.completion, _c = _a.validate, validate = _c === void 0 ? validateDefault : _c;
             var completion = _this.completionSchemaService.get(completionName);
             _this.sessionValidationService.addAgentUsage(clientId, agentName);
-            return new ClientAgent(__assign({ clientId: clientId, agentName: agentName, validate: validate, logger: _this.loggerService, history: _this.historyConnectionService.getHistory(clientId, agentName), prompt: prompt, system: system, tools: tools === null || tools === void 0 ? void 0 : tools.map(_this.toolSchemaService.get), completion: completion }, callbacks));
+            return new ClientAgent(__assign({ clientId: clientId, agentName: agentName, validate: validate, logger: _this.loggerService, history: _this.historyConnectionService.getHistory(clientId, agentName), prompt: prompt, system: system, transform: transform, tools: tools === null || tools === void 0 ? void 0 : tools.map(_this.toolSchemaService.get), completion: completion }, callbacks));
         });
         /**
          * Executes an input command.
@@ -1139,17 +1156,7 @@ var ClientHistory = /** @class */ (function () {
                             return agentName === _this.params.agentName;
                         });
                         commonMessages = commonMessagesRaw
-                            .filter(function (_a) {
-                            var role = _a.role, agentName = _a.agentName, tool_calls = _a.tool_calls;
-                            var isOk = true;
-                            if (role === "tool") {
-                                isOk = isOk && agentName === _this.params.agentName;
-                            }
-                            if (tool_calls) {
-                                isOk = isOk && agentName === _this.params.agentName;
-                            }
-                            return isOk;
-                        })
+                            .filter(this._filterCondition)
                             .slice(-GLOBAL_CONFIG.CC_KEEP_MESSAGES);
                         promptMessages = [];
                         {
@@ -1176,6 +1183,7 @@ var ClientHistory = /** @class */ (function () {
         this.params.logger.debug("ClientHistory agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " CTOR"), {
             params: params,
         });
+        this._filterCondition = GLOBAL_CONFIG.CC_AGENT_HISTORY_FILTER(this.params.agentName);
     }
     return ClientHistory;
 }());
@@ -1325,7 +1333,6 @@ var ToolSchemaService = /** @class */ (function () {
     return ToolSchemaService;
 }());
 
-var AGENT_REF_CHANGED = Symbol("agent-ref-changed");
 var AGENT_NEED_FETCH = Symbol("agent-need-fetch");
 /**
  * ClientSwarm class implements the ISwarm interface and manages agents within a swarm.
@@ -1346,47 +1353,53 @@ var ClientSwarm = /** @class */ (function () {
          * @throws {Error} - If the timeout is reached.
          */
         this.waitForOutput = functoolsKit.queued(function () { return __awaiter(_this, void 0, void 0, function () {
-            var START_TIME, _a, agentName, output, _b;
+            var _a, awaiter, resolve, getOutput, handleOutput, un, _b, agentName, output, expectAgent;
             var _this = this;
             return __generator(this, function (_c) {
                 switch (_c.label) {
                     case 0:
                         this.params.logger.debug("ClientSwarm swarmName=".concat(this.params.swarmName, " clientId=").concat(this.params.clientId, " waitForOutput"));
-                        START_TIME = Date.now();
-                        _c.label = 1;
+                        _a = __read(functoolsKit.createAwaiter(), 2), awaiter = _a[0], resolve = _a[1].resolve;
+                        getOutput = functoolsKit.cancelable(function () { return __awaiter(_this, void 0, void 0, function () {
+                            var _this = this;
+                            return __generator(this, function (_a) {
+                                switch (_a.label) {
+                                    case 0: return [4 /*yield*/, Promise.race(this._agentList.map(function (_a) { return __awaiter(_this, [_a], void 0, function (_b) {
+                                            var _c;
+                                            var _d = __read(_b, 2), agentName = _d[0], agent = _d[1];
+                                            return __generator(this, function (_e) {
+                                                switch (_e.label) {
+                                                    case 0:
+                                                        _c = { agentName: agentName };
+                                                        return [4 /*yield*/, agent.waitForOutput()];
+                                                    case 1: return [2 /*return*/, (_c.output = _e.sent(), _c)];
+                                                }
+                                            });
+                                        }); }))];
+                                    case 1: return [2 /*return*/, _a.sent()];
+                                }
+                            });
+                        }); });
+                        handleOutput = function () {
+                            getOutput.cancel();
+                            getOutput().then(function (value) {
+                                if (value === functoolsKit.CANCELED_PROMISE_SYMBOL) {
+                                    return;
+                                }
+                                resolve(value);
+                            });
+                        };
+                        un = this._agentChangedSubject.subscribe(handleOutput);
+                        handleOutput();
+                        return [4 /*yield*/, awaiter];
                     case 1:
-                        if (Date.now() - START_TIME >= GLOBAL_CONFIG.CC_ANSWER_TIMEOUT) {
-                            throw new Error("agent-swarm ClientSwarm waitForOutput timeout reached for ".concat(this.params.swarmName));
-                        }
-                        return [4 /*yield*/, Promise.race(__spreadArray(__spreadArray([], __read(Object.entries(this.params.agentMap).map(function (_a) { return __awaiter(_this, [_a], void 0, function (_b) {
-                                var _c;
-                                var _d = __read(_b, 2), agentName = _d[0], agent = _d[1];
-                                return __generator(this, function (_e) {
-                                    switch (_e.label) {
-                                        case 0:
-                                            _c = [agentName];
-                                            return [4 /*yield*/, agent.waitForOutput()];
-                                        case 1: return [2 /*return*/, _c.concat([_e.sent()])];
-                                    }
-                                });
-                            }); })), false), [
-                                this._agentChangedSubject
-                                    .toPromise()
-                                    .then(function () { return [AGENT_REF_CHANGED]; }),
-                            ], false))];
-                    case 2:
-                        _a = __read.apply(void 0, [_c.sent(), 2]), agentName = _a[0], output = _a[1];
-                        if (agentName === AGENT_REF_CHANGED) {
-                            return [3 /*break*/, 1];
-                        }
-                        _b = agentName;
+                        _b = _c.sent(), agentName = _b.agentName, output = _b.output;
+                        un();
                         return [4 /*yield*/, this.getAgentName()];
-                    case 3:
-                        if (_b === (_c.sent())) {
-                            return [2 /*return*/, output];
-                        }
-                        return [3 /*break*/, 1];
-                    case 4: return [2 /*return*/];
+                    case 2:
+                        expectAgent = _c.sent();
+                        agentName !== expectAgent && this.params.logger.debug("ClientSwarm swarmName=".concat(this.params.swarmName, " clientId=").concat(this.params.clientId, " waitForAgent agent miss"), { agentName: agentName, expectAgent: expectAgent });
+                        return [2 /*return*/, output];
                 }
             });
         }); });
@@ -1442,7 +1455,7 @@ var ClientSwarm = /** @class */ (function () {
                             throw new Error("agent-swarm agent ".concat(agentName, " not in the swarm"));
                         }
                         this.params.agentMap[agentName] = agent;
-                        return [4 /*yield*/, this._agentChangedSubject.next()];
+                        return [4 /*yield*/, this._agentChangedSubject.next([agentName, agent])];
                     case 1:
                         _a.sent();
                         return [2 /*return*/];
@@ -1470,6 +1483,13 @@ var ClientSwarm = /** @class */ (function () {
             params: params,
         });
     }
+    Object.defineProperty(ClientSwarm.prototype, "_agentList", {
+        get: function () {
+            return Object.entries(this.params.agentMap);
+        },
+        enumerable: false,
+        configurable: true
+    });
     return ClientSwarm;
 }());
 

package/build/index.mjs

@@ -1,5 +1,5 @@
 import { scoped } from 'di-scoped';
-import { ToolRegistry, trycatch, sleep, PubsubArrayAdapter, Subject, queued, not, memoize, schedule, ttl, singleshot, Source } from 'functools-kit';
+import { ToolRegistry, trycatch, sleep, PubsubArrayAdapter, Subject, queued, not, memoize, createAwaiter, cancelable, CANCELED_PROMISE_SYMBOL, schedule, ttl, singleshot, Source } from 'functools-kit';
 import { createActivator } from 'di-kit';
 import { omit } from 'lodash-es';
 import xml2js from 'xml2js';
@@ -353,6 +353,10 @@ var validateDefault = function (output) { return __awaiter(void 0, void 0, void
     });
 }); };
 
+var removeXmlTags = function (input) {
+    return input.replace(/<[^>]*>.*?<\/[^>]*>/g, '');
+};
+
 /**
  * @description `ask for agent function` in `llama3.1:8b` to troubleshoot (need CC_OLLAMA_EMIT_TOOL_PROTOCOL to be turned off)
  */
@@ -394,18 +398,29 @@ var CC_AGENT_DISALLOWED_SYMBOLS = [
     "{",
     "}",
 ];
+var CC_AGENT_HISTORY_FILTER = function (agentName) { return function (message) {
+    var isOk = true;
+    if (message.role === "tool") {
+        isOk = isOk && message.agentName === agentName;
+    }
+    if (message.tool_calls) {
+        isOk = isOk && message.agentName === agentName;
+    }
+    return isOk;
+}; };
+var CC_AGENT_OUTPUT_TRANSFORM = removeXmlTags;
 var CC_KEEP_MESSAGES = 5;
-var CC_ANSWER_TIMEOUT = 120000;
 var CC_GET_AGENT_HISTORY = function () { return new PubsubArrayAdapter(); };
 var GLOBAL_CONFIG = {
     CC_TOOL_CALL_EXCEPTION_PROMPT: CC_TOOL_CALL_EXCEPTION_PROMPT,
     CC_EMPTY_OUTPUT_PLACEHOLDERS: CC_EMPTY_OUTPUT_PLACEHOLDERS,
     CC_KEEP_MESSAGES: CC_KEEP_MESSAGES,
-    CC_ANSWER_TIMEOUT: CC_ANSWER_TIMEOUT,
     CC_GET_AGENT_HISTORY: CC_GET_AGENT_HISTORY,
     CC_SWARM_AGENT_CHANGED: CC_SWARM_AGENT_CHANGED,
     CC_SWARM_DEFAULT_AGENT: CC_SWARM_DEFAULT_AGENT,
     CC_AGENT_DEFAULT_VALIDATION: CC_AGENT_DEFAULT_VALIDATION,
+    CC_AGENT_HISTORY_FILTER: CC_AGENT_HISTORY_FILTER,
+    CC_AGENT_OUTPUT_TRANSFORM: CC_AGENT_OUTPUT_TRANSFORM,
     CC_AGENT_DISALLOWED_TAGS: CC_AGENT_DISALLOWED_TAGS,
     CC_AGENT_DISALLOWED_SYMBOLS: CC_AGENT_DISALLOWED_SYMBOLS,
 };
@@ -436,19 +451,21 @@ var ClientAgent = /** @class */ (function () {
          * @returns {Promise<void>}
          * @private
          */
-        this._emitOuput = function (mode, result) { return __awaiter(_this, void 0, void 0, function () {
-            var validation, result_1;
+        this._emitOuput = function (mode, rawResult) { return __awaiter(_this, void 0, void 0, function () {
+            var result, validation, rawResult_1, result_1;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " _emitOuput"));
+                        result = this.params.transform(rawResult);
+                        this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " _emitOuput"), { mode: mode, result: result, rawResult: rawResult });
                         validation = null;
                         return [4 /*yield*/, this.params.validate(result)];
                     case 1:
                         if (!(validation = _a.sent())) return [3 /*break*/, 5];
                         return [4 /*yield*/, this._resurrectModel(mode, validation)];
                     case 2:
-                        result_1 = _a.sent();
+                        rawResult_1 = _a.sent();
+                        result_1 = this.params.transform(rawResult_1);
                         return [4 /*yield*/, this.params.validate(result_1)];
                     case 3:
                         if ((validation = _a.sent())) {
@@ -854,10 +871,10 @@ var AgentConnectionService = /** @class */ (function () {
             var _b = __read(_a, 2), clientId = _b[0], agentName = _b[1];
             return "".concat(clientId, "-").concat(agentName);
         }, function (clientId, agentName) {
-            var _a = _this.agentSchemaService.get(agentName), prompt = _a.prompt, system = _a.system, tools = _a.tools, callbacks = _a.callbacks, completionName = _a.completion, _b = _a.validate, validate = _b === void 0 ? validateDefault : _b;
+            var _a = _this.agentSchemaService.get(agentName), prompt = _a.prompt, system = _a.system, tools = _a.tools, _b = _a.transform, transform = _b === void 0 ? GLOBAL_CONFIG.CC_AGENT_OUTPUT_TRANSFORM : _b, callbacks = _a.callbacks, completionName = _a.completion, _c = _a.validate, validate = _c === void 0 ? validateDefault : _c;
             var completion = _this.completionSchemaService.get(completionName);
             _this.sessionValidationService.addAgentUsage(clientId, agentName);
-            return new ClientAgent(__assign({ clientId: clientId, agentName: agentName, validate: validate, logger: _this.loggerService, history: _this.historyConnectionService.getHistory(clientId, agentName), prompt: prompt, system: system, tools: tools === null || tools === void 0 ? void 0 : tools.map(_this.toolSchemaService.get), completion: completion }, callbacks));
+            return new ClientAgent(__assign({ clientId: clientId, agentName: agentName, validate: validate, logger: _this.loggerService, history: _this.historyConnectionService.getHistory(clientId, agentName), prompt: prompt, system: system, transform: transform, tools: tools === null || tools === void 0 ? void 0 : tools.map(_this.toolSchemaService.get), completion: completion }, callbacks));
         });
         /**
          * Executes an input command.
@@ -1137,17 +1154,7 @@ var ClientHistory = /** @class */ (function () {
                             return agentName === _this.params.agentName;
                         });
                         commonMessages = commonMessagesRaw
-                            .filter(function (_a) {
-                            var role = _a.role, agentName = _a.agentName, tool_calls = _a.tool_calls;
-                            var isOk = true;
-                            if (role === "tool") {
-                                isOk = isOk && agentName === _this.params.agentName;
-                            }
-                            if (tool_calls) {
-                                isOk = isOk && agentName === _this.params.agentName;
-                            }
-                            return isOk;
-                        })
+                            .filter(this._filterCondition)
                             .slice(-GLOBAL_CONFIG.CC_KEEP_MESSAGES);
                         promptMessages = [];
                         {
@@ -1174,6 +1181,7 @@ var ClientHistory = /** @class */ (function () {
         this.params.logger.debug("ClientHistory agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " CTOR"), {
             params: params,
         });
+        this._filterCondition = GLOBAL_CONFIG.CC_AGENT_HISTORY_FILTER(this.params.agentName);
     }
     return ClientHistory;
 }());
@@ -1323,7 +1331,6 @@ var ToolSchemaService = /** @class */ (function () {
     return ToolSchemaService;
 }());
 
-var AGENT_REF_CHANGED = Symbol("agent-ref-changed");
 var AGENT_NEED_FETCH = Symbol("agent-need-fetch");
 /**
  * ClientSwarm class implements the ISwarm interface and manages agents within a swarm.
@@ -1344,47 +1351,53 @@ var ClientSwarm = /** @class */ (function () {
          * @throws {Error} - If the timeout is reached.
          */
         this.waitForOutput = queued(function () { return __awaiter(_this, void 0, void 0, function () {
-            var START_TIME, _a, agentName, output, _b;
+            var _a, awaiter, resolve, getOutput, handleOutput, un, _b, agentName, output, expectAgent;
             var _this = this;
             return __generator(this, function (_c) {
                 switch (_c.label) {
                     case 0:
                         this.params.logger.debug("ClientSwarm swarmName=".concat(this.params.swarmName, " clientId=").concat(this.params.clientId, " waitForOutput"));
-                        START_TIME = Date.now();
-                        _c.label = 1;
+                        _a = __read(createAwaiter(), 2), awaiter = _a[0], resolve = _a[1].resolve;
+                        getOutput = cancelable(function () { return __awaiter(_this, void 0, void 0, function () {
+                            var _this = this;
+                            return __generator(this, function (_a) {
+                                switch (_a.label) {
+                                    case 0: return [4 /*yield*/, Promise.race(this._agentList.map(function (_a) { return __awaiter(_this, [_a], void 0, function (_b) {
+                                            var _c;
+                                            var _d = __read(_b, 2), agentName = _d[0], agent = _d[1];
+                                            return __generator(this, function (_e) {
+                                                switch (_e.label) {
+                                                    case 0:
+                                                        _c = { agentName: agentName };
+                                                        return [4 /*yield*/, agent.waitForOutput()];
+                                                    case 1: return [2 /*return*/, (_c.output = _e.sent(), _c)];
+                                                }
+                                            });
+                                        }); }))];
+                                    case 1: return [2 /*return*/, _a.sent()];
+                                }
+                            });
+                        }); });
+                        handleOutput = function () {
+                            getOutput.cancel();
+                            getOutput().then(function (value) {
+                                if (value === CANCELED_PROMISE_SYMBOL) {
+                                    return;
+                                }
+                                resolve(value);
+                            });
+                        };
+                        un = this._agentChangedSubject.subscribe(handleOutput);
+                        handleOutput();
+                        return [4 /*yield*/, awaiter];
                     case 1:
-                        if (Date.now() - START_TIME >= GLOBAL_CONFIG.CC_ANSWER_TIMEOUT) {
-                            throw new Error("agent-swarm ClientSwarm waitForOutput timeout reached for ".concat(this.params.swarmName));
-                        }
-                        return [4 /*yield*/, Promise.race(__spreadArray(__spreadArray([], __read(Object.entries(this.params.agentMap).map(function (_a) { return __awaiter(_this, [_a], void 0, function (_b) {
-                                var _c;
-                                var _d = __read(_b, 2), agentName = _d[0], agent = _d[1];
-                                return __generator(this, function (_e) {
-                                    switch (_e.label) {
-                                        case 0:
-                                            _c = [agentName];
-                                            return [4 /*yield*/, agent.waitForOutput()];
-                                        case 1: return [2 /*return*/, _c.concat([_e.sent()])];
-                                    }
-                                });
-                            }); })), false), [
-                                this._agentChangedSubject
-                                    .toPromise()
-                                    .then(function () { return [AGENT_REF_CHANGED]; }),
-                            ], false))];
-                    case 2:
-                        _a = __read.apply(void 0, [_c.sent(), 2]), agentName = _a[0], output = _a[1];
-                        if (agentName === AGENT_REF_CHANGED) {
-                            return [3 /*break*/, 1];
-                        }
-                        _b = agentName;
+                        _b = _c.sent(), agentName = _b.agentName, output = _b.output;
+                        un();
                         return [4 /*yield*/, this.getAgentName()];
-                    case 3:
-                        if (_b === (_c.sent())) {
-                            return [2 /*return*/, output];
-                        }
-                        return [3 /*break*/, 1];
-                    case 4: return [2 /*return*/];
+                    case 2:
+                        expectAgent = _c.sent();
+                        agentName !== expectAgent && this.params.logger.debug("ClientSwarm swarmName=".concat(this.params.swarmName, " clientId=").concat(this.params.clientId, " waitForAgent agent miss"), { agentName: agentName, expectAgent: expectAgent });
+                        return [2 /*return*/, output];
                 }
             });
         }); });
@@ -1440,7 +1453,7 @@ var ClientSwarm = /** @class */ (function () {
                             throw new Error("agent-swarm agent ".concat(agentName, " not in the swarm"));
                         }
                         this.params.agentMap[agentName] = agent;
-                        return [4 /*yield*/, this._agentChangedSubject.next()];
+                        return [4 /*yield*/, this._agentChangedSubject.next([agentName, agent])];
                     case 1:
                         _a.sent();
                         return [2 /*return*/];
@@ -1468,6 +1481,13 @@ var ClientSwarm = /** @class */ (function () {
             params: params,
         });
     }
+    Object.defineProperty(ClientSwarm.prototype, "_agentList", {
+        get: function () {
+            return Object.entries(this.params.agentMap);
+        },
+        enumerable: false,
+        configurable: true
+    });
     return ClientSwarm;
 }());
 

package/package.json

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

package/types.d.ts

@@ -642,6 +642,8 @@ interface IAgentSchema {
      * @returns A promise that resolves to a string or null.
      */
     validate?: (output: string) => Promise<string | null>;
+    /** The transform function for model output */
+    transform?: (input: string) => string;
     /** The lifecycle calbacks of the agent. */
     callbacks?: Partial<IAgentSchemaCallbacks>;
 }
@@ -751,7 +753,7 @@ declare class ClientAgent implements IAgent {
      * @returns {Promise<void>}
      * @private
      */
-    _emitOuput: (mode: ExecutionMode, result: string) => Promise<void>;
+    _emitOuput: (mode: ExecutionMode, rawResult: string) => Promise<void>;
     /**
      * Resurrects the model based on the given reason.
      * @param {string} [reason] - The reason for resurrecting the model.
@@ -871,6 +873,10 @@ declare class AgentConnectionService implements IAgent {
  */
 declare class ClientHistory implements IHistory {
     readonly params: IHistoryParams;
+    /**
+     * Filter condition for `toArrayForAgent`
+     */
+    _filterCondition: (message: IModelMessage) => boolean;
     /**
      * Creates an instance of ClientHistory.
      * @param {IHistoryParams} params - The parameters for the history.
@@ -989,6 +995,7 @@ declare class ClientSwarm implements ISwarm {
     readonly params: ISwarmParams;
     private _agentChangedSubject;
     private _activeAgent;
+    get _agentList(): [string, IAgent][];
     /**
      * Creates an instance of ClientSwarm.
      * @param {ISwarmParams} params - The parameters for the swarm.
@@ -2039,11 +2046,12 @@ declare const GLOBAL_CONFIG: {
     CC_TOOL_CALL_EXCEPTION_PROMPT: string;
     CC_EMPTY_OUTPUT_PLACEHOLDERS: string[];
     CC_KEEP_MESSAGES: number;
-    CC_ANSWER_TIMEOUT: number;
     CC_GET_AGENT_HISTORY: (clientId: string, agentName: AgentName) => IPubsubArray<IModelMessage>;
     CC_SWARM_AGENT_CHANGED: (clientId: string, agentName: AgentName, swarmName: SwarmName) => Promise<void>;
     CC_SWARM_DEFAULT_AGENT: (clientId: string, swarmName: SwarmName, defaultAgent: AgentName) => Promise<AgentName>;
     CC_AGENT_DEFAULT_VALIDATION: (output: string) => Promise<string | null>;
+    CC_AGENT_HISTORY_FILTER: (agentName: AgentName) => (message: IModelMessage) => boolean;
+    CC_AGENT_OUTPUT_TRANSFORM: (input: string) => string;
     CC_AGENT_DISALLOWED_TAGS: string[];
     CC_AGENT_DISALLOWED_SYMBOLS: string[];
 };