agent-swarm-kit 1.0.11 → 1.0.13

package/build/index.cjs

@@ -425,7 +425,7 @@ var ClientAgent = /** @class */ (function () {
                                     case 1:
                                         _f.sent();
                                         if (!!targetFn) return [3 /*break*/, 4];
-                                        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.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("No target function for ".concat(tool.function.name))];
                                     case 2:
                                         result_2 = _f.sent();
@@ -522,7 +522,9 @@ var ClientAgent = /** @class */ (function () {
                 }
             });
         }); });
-        this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " CTOR"));
+        this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " CTOR"), {
+            params: params,
+        });
     }
     return ClientAgent;
 }());
@@ -848,7 +850,9 @@ var ClientHistory = /** @class */ (function () {
                 }
             });
         }); };
-        this.params.logger.debug("ClientHistory agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " CTOR"));
+        this.params.logger.debug("ClientHistory agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " CTOR"), {
+            params: params,
+        });
     }
     return ClientHistory;
 }());
@@ -951,10 +955,12 @@ var ToolSchemaService = /** @class */ (function () {
     return ToolSchemaService;
 }());
 
+var AGENT_REF_CHANGED = Symbol('agent-ref-changed');
 var ClientSwarm = /** @class */ (function () {
     function ClientSwarm(params) {
         var _this = this;
         this.params = params;
+        this._agentChangedSubject = new functoolsKit.Subject();
         this.waitForOutput = functoolsKit.queued(function () { return __awaiter(_this, void 0, void 0, function () {
             var START_TIME, _a, agentName, output, _b;
             var _this = this;
@@ -968,7 +974,7 @@ var ClientSwarm = /** @class */ (function () {
                         if (Date.now() - START_TIME >= GLOBAL_CONFIG.CC_ANSWER_TIMEOUT_SECONDS) {
                             throw new Error("agent-swarm ClientSwarm waitForOutput timeout reached for ".concat(this.params.swarmName));
                         }
-                        return [4 /*yield*/, Promise.race(Object.entries(this.params.agentMap).map(function (_a) { return __awaiter(_this, [_a], void 0, function (_b) {
+                        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) {
@@ -981,9 +987,14 @@ var ClientSwarm = /** @class */ (function () {
                                             ])];
                                     }
                                 });
-                            }); }))];
+                            }); })), 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;
                         return [4 /*yield*/, this.getAgentName()];
                     case 3:
@@ -1014,6 +1025,22 @@ var ClientSwarm = /** @class */ (function () {
                 }
             });
         }); };
+        this.setAgentRef = function (agentName, agent) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.params.logger.debug("ClientSwarm swarmName=".concat(this.params.swarmName, " clientId=").concat(this.params.clientId, " setAgentRef agentName=").concat(agentName));
+                        if (!this.params.agentMap[agentName]) {
+                            throw new Error("agent-swarm agent ".concat(agentName, " not in the swarm"));
+                        }
+                        this.params.agentMap[agentName] = agent;
+                        return [4 /*yield*/, this._agentChangedSubject.next()];
+                    case 1:
+                        _a.sent();
+                        return [2 /*return*/];
+                }
+            });
+        }); };
         this.setAgentName = function (agentName) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 this.params.logger.debug("ClientSwarm swarmName=".concat(this.params.swarmName, " clientId=").concat(this.params.clientId, " setAgentName agentName=").concat(agentName));
@@ -1021,7 +1048,9 @@ var ClientSwarm = /** @class */ (function () {
                 return [2 /*return*/];
             });
         }); };
-        this.params.logger.debug("ClientSwarm swarmName=".concat(this.params.swarmName, " clientId=").concat(this.params.clientId, " CTOR"));
+        this.params.logger.debug("ClientSwarm swarmName=".concat(this.params.swarmName, " clientId=").concat(this.params.clientId, " CTOR"), {
+            params: params,
+        });
         this._activeAgent = params.defaultAgent;
     }
     return ClientSwarm;
@@ -1098,6 +1127,18 @@ var SwarmConnectionService = /** @class */ (function () {
                 }
             });
         }); };
+        this.setAgentRef = function (agentName, agent) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("swarmConnectionService setAgentRef", {
+                            context: this.contextService.context,
+                        });
+                        return [4 /*yield*/, this.getSwarm(this.contextService.context.clientId, this.contextService.context.swarmName).setAgentRef(agentName, agent)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         this.setAgentName = function (agentName) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -1273,7 +1314,9 @@ var ClientSession = /** @class */ (function () {
                 });
             }); };
         };
-        this.params.logger.debug("ClientSession clientId=".concat(this.params.clientId, " CTOR"));
+        this.params.logger.debug("ClientSession clientId=".concat(this.params.clientId, " CTOR"), {
+            params: params,
+        });
     }
     return ClientSession;
 }());
@@ -1354,6 +1397,31 @@ var AgentPublicService = /** @class */ (function () {
         var _this = this;
         this.loggerService = inject(TYPES.loggerService);
         this.agentConnectionService = inject(TYPES.agentConnectionService);
+        this.createAgentRef = function (clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("agentPublicService createAgentRef", {
+                            clientId: clientId,
+                            agentName: agentName,
+                        });
+                        return [4 /*yield*/, ContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.agentConnectionService.getAgent(clientId, agentName)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                clientId: clientId,
+                                agentName: agentName,
+                                swarmName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         this.execute = function (input, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
@@ -1830,6 +1898,33 @@ var SwarmPublicService = /** @class */ (function () {
                 }
             });
         }); };
+        this.setAgentRef = function (clientId, swarmName, agentName, agent) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("swarmPublicService setAgentRef", {
+                            agentName: agentName,
+                            agent: agent,
+                            clientId: clientId,
+                            swarmName: swarmName,
+                        });
+                        return [4 /*yield*/, ContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.swarmConnectionService.setAgentRef(agentName, agent)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                clientId: clientId,
+                                swarmName: swarmName,
+                                agentName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         this.setAgentName = function (agentName, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
@@ -2231,30 +2326,46 @@ var swarm = __assign(__assign(__assign(__assign(__assign({}, baseServices), conn
 init();
 
 var addAgent = function (agentSchema) {
+    swarm.loggerService.log('function addAgent', {
+        agentSchema: agentSchema
+    });
     swarm.agentValidationService.addAgent(agentSchema.agentName, agentSchema);
     swarm.agentSchemaService.register(agentSchema.agentName, agentSchema);
     return agentSchema.agentName;
 };
 
 var addCompletion = function (completionSchema) {
+    swarm.loggerService.log('function addCompletion', {
+        completionSchema: completionSchema,
+    });
     swarm.completionValidationService.addCompletion(completionSchema.completionName);
     swarm.completionSchemaService.register(completionSchema.completionName, completionSchema);
     return completionSchema.completionName;
 };
 
 var addSwarm = function (swarmSchema) {
+    swarm.loggerService.log('function addSwarm', {
+        swarmSchema: swarmSchema,
+    });
     swarm.swarmValidationService.addSwarm(swarmSchema.swarmName, swarmSchema);
     swarm.swarmSchemaService.register(swarmSchema.swarmName, swarmSchema);
     return swarmSchema.swarmName;
 };
 
 var addTool = function (toolSchema) {
+    swarm.loggerService.log('function addTool', {
+        toolSchema: toolSchema,
+    });
     swarm.toolValidationService.addTool(toolSchema.toolName, toolSchema);
     swarm.toolSchemaService.register(toolSchema.toolName, toolSchema);
     return toolSchema.toolName;
 };
 
 var makeConnection = function (connector, clientId, swarmName) {
+    swarm.loggerService.log("function makeConnection", {
+        clientId: clientId,
+        swarmName: swarmName,
+    });
     swarm.swarmValidationService.validate(swarmName, "makeConnection");
     swarm.sessionValidationService.addSession(clientId, swarmName);
     var send = swarm.sessionPublicService.connect(connector, clientId, swarmName);
@@ -2279,18 +2390,66 @@ var makeConnection = function (connector, clientId, swarmName) {
     }); });
 };
 
+var CHANGE_AGENT_TTL = 15 * 60 * 1000;
+var CHANGE_AGENT_GC = 60 * 1000;
+var createChangeAgent = functoolsKit.ttl(function (clientId) {
+    return functoolsKit.queued(function (agentName) { return __awaiter(void 0, void 0, void 0, function () {
+        var swarmName, _a, _b, _c;
+        return __generator(this, function (_d) {
+            switch (_d.label) {
+                case 0:
+                    swarm.sessionValidationService.validate(clientId, "changeAgent");
+                    swarm.agentValidationService.validate(agentName, "changeAgent");
+                    swarmName = swarm.sessionValidationService.getSwarm(clientId);
+                    return [4 /*yield*/, swarm.agentPublicService.dispose(clientId, agentName)];
+                case 1:
+                    _d.sent();
+                    return [4 /*yield*/, swarm.historyPublicService.dispose(clientId, agentName)];
+                case 2:
+                    _d.sent();
+                    _b = (_a = swarm.swarmPublicService).setAgentRef;
+                    _c = [clientId,
+                        swarmName,
+                        agentName];
+                    return [4 /*yield*/, swarm.agentPublicService.createAgentRef(clientId, agentName)];
+                case 3: return [4 /*yield*/, _b.apply(_a, _c.concat([_d.sent()]))];
+                case 4:
+                    _d.sent();
+                    return [4 /*yield*/, swarm.swarmPublicService.setAgentName(agentName, clientId, swarmName)];
+                case 5:
+                    _d.sent();
+                    return [2 /*return*/];
+            }
+        });
+    }); });
+}, {
+    key: function (_a) {
+        var _b = __read(_a, 1), clientId = _b[0];
+        return "".concat(clientId);
+    },
+    timeout: CHANGE_AGENT_TTL,
+});
+var createGc$1 = functoolsKit.singleshot(function () { return __awaiter(void 0, void 0, void 0, function () {
+    return __generator(this, function (_a) {
+        setInterval(createChangeAgent.gc, CHANGE_AGENT_GC);
+        return [2 /*return*/];
+    });
+}); });
 var changeAgent = function (agentName, clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName;
+    var run;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
-                swarm.sessionValidationService.validate(clientId, "changeAgent");
-                swarm.agentValidationService.validate(agentName, "changeAgent");
-                swarmName = swarm.sessionValidationService.getSwarm(clientId);
-                return [4 /*yield*/, swarm.swarmPublicService.setAgentName(agentName, clientId, swarmName)];
+                swarm.loggerService.log('function changeAgent', {
+                    agentName: agentName,
+                    clientId: clientId,
+                });
+                return [4 /*yield*/, createChangeAgent(clientId)];
             case 1:
-                _a.sent();
-                return [2 /*return*/];
+                run = _a.sent();
+                createGc$1();
+                return [4 /*yield*/, run(agentName)];
+            case 2: return [2 /*return*/, _a.sent()];
         }
     });
 }); };
@@ -2299,6 +2458,10 @@ var disposeConnection = function (clientId, swarmName) { return __awaiter(void 0
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                swarm.loggerService.log("function disposeConnection", {
+                    clientId: clientId,
+                    swarmName: swarmName,
+                });
                 swarm.swarmValidationService.validate(swarmName, "disposeConnection");
                 swarm.sessionValidationService.removeSession(clientId);
                 return [4 /*yield*/, swarm.sessionPublicService.dispose(clientId, swarmName)];
@@ -2368,7 +2531,13 @@ var complete = function (content, clientId, swarmName) { return __awaiter(void 0
     var run;
     return __generator(this, function (_a) {
         switch (_a.label) {
-            case 0: return [4 /*yield*/, createComplete(clientId, swarmName)];
+            case 0:
+                swarm.loggerService.log("function complete", {
+                    content: content,
+                    clientId: clientId,
+                    swarmName: swarmName,
+                });
+                return [4 /*yield*/, createComplete(clientId, swarmName)];
             case 1:
                 run = _a.sent();
                 createGc();
@@ -2379,6 +2548,10 @@ var complete = function (content, clientId, swarmName) { return __awaiter(void 0
 }); };
 
 var session = function (clientId, swarmName) {
+    swarm.loggerService.log("function session", {
+        clientId: clientId,
+        swarmName: swarmName,
+    });
     swarm.swarmValidationService.validate(swarmName, "session");
     swarm.sessionValidationService.addSession(clientId, swarmName);
     return {
@@ -2408,6 +2581,9 @@ var getRawHistory = function (clientId) { return __awaiter(void 0, void 0, void
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                swarm.loggerService.log("function getRawHistory", {
+                    clientId: clientId,
+                });
                 swarm.sessionValidationService.validate(clientId, "getRawHistory");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "getRawHistory");
@@ -2427,6 +2603,10 @@ var getAgentHistory = function (clientId, agentName) { return __awaiter(void 0,
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                swarm.loggerService.log("function getAgentHistory", {
+                    clientId: clientId,
+                    agentName: agentName,
+                });
                 swarm.agentValidationService.validate(agentName, "getAgentHistory");
                 prompt = swarm.agentSchemaService.get(agentName).prompt;
                 return [4 /*yield*/, swarm.historyPublicService.toArrayForAgent(prompt, clientId, agentName)];
@@ -2442,6 +2622,11 @@ var commitToolOutput = function (content, clientId, agentName) { return __awaite
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                swarm.loggerService.log('function commitToolOutput', {
+                    content: content,
+                    clientId: clientId,
+                    agentName: agentName,
+                });
                 swarm.agentValidationService.validate(agentName, "commitSystemMessage");
                 swarm.sessionValidationService.validate(clientId, "commitToolOutput");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
@@ -2470,6 +2655,11 @@ var commitSystemMessage = function (content, clientId, agentName) { return __awa
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                swarm.loggerService.log('function commitSystemMessage', {
+                    content: content,
+                    clientId: clientId,
+                    agentName: agentName,
+                });
                 swarm.agentValidationService.validate(agentName, "commitSystemMessage");
                 swarm.sessionValidationService.validate(clientId, "commitSystemMessage");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
@@ -2498,6 +2688,11 @@ var execute = function (content, clientId, agentName) { return __awaiter(void 0,
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                swarm.loggerService.log("function execute", {
+                    content: content,
+                    clientId: clientId,
+                    agentName: agentName,
+                });
                 swarm.agentValidationService.validate(agentName, "commitSystemMessage");
                 swarm.sessionValidationService.validate(clientId, "execute");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);

package/build/index.mjs

@@ -423,7 +423,7 @@ var ClientAgent = /** @class */ (function () {
                                     case 1:
                                         _f.sent();
                                         if (!!targetFn) return [3 /*break*/, 4];
-                                        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.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("No target function for ".concat(tool.function.name))];
                                     case 2:
                                         result_2 = _f.sent();
@@ -520,7 +520,9 @@ var ClientAgent = /** @class */ (function () {
                 }
             });
         }); });
-        this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " CTOR"));
+        this.params.logger.debug("ClientAgent agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " CTOR"), {
+            params: params,
+        });
     }
     return ClientAgent;
 }());
@@ -846,7 +848,9 @@ var ClientHistory = /** @class */ (function () {
                 }
             });
         }); };
-        this.params.logger.debug("ClientHistory agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " CTOR"));
+        this.params.logger.debug("ClientHistory agentName=".concat(this.params.agentName, " clientId=").concat(this.params.clientId, " CTOR"), {
+            params: params,
+        });
     }
     return ClientHistory;
 }());
@@ -949,10 +953,12 @@ var ToolSchemaService = /** @class */ (function () {
     return ToolSchemaService;
 }());
 
+var AGENT_REF_CHANGED = Symbol('agent-ref-changed');
 var ClientSwarm = /** @class */ (function () {
     function ClientSwarm(params) {
         var _this = this;
         this.params = params;
+        this._agentChangedSubject = new Subject();
         this.waitForOutput = queued(function () { return __awaiter(_this, void 0, void 0, function () {
             var START_TIME, _a, agentName, output, _b;
             var _this = this;
@@ -966,7 +972,7 @@ var ClientSwarm = /** @class */ (function () {
                         if (Date.now() - START_TIME >= GLOBAL_CONFIG.CC_ANSWER_TIMEOUT_SECONDS) {
                             throw new Error("agent-swarm ClientSwarm waitForOutput timeout reached for ".concat(this.params.swarmName));
                         }
-                        return [4 /*yield*/, Promise.race(Object.entries(this.params.agentMap).map(function (_a) { return __awaiter(_this, [_a], void 0, function (_b) {
+                        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) {
@@ -979,9 +985,14 @@ var ClientSwarm = /** @class */ (function () {
                                             ])];
                                     }
                                 });
-                            }); }))];
+                            }); })), 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;
                         return [4 /*yield*/, this.getAgentName()];
                     case 3:
@@ -1012,6 +1023,22 @@ var ClientSwarm = /** @class */ (function () {
                 }
             });
         }); };
+        this.setAgentRef = function (agentName, agent) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.params.logger.debug("ClientSwarm swarmName=".concat(this.params.swarmName, " clientId=").concat(this.params.clientId, " setAgentRef agentName=").concat(agentName));
+                        if (!this.params.agentMap[agentName]) {
+                            throw new Error("agent-swarm agent ".concat(agentName, " not in the swarm"));
+                        }
+                        this.params.agentMap[agentName] = agent;
+                        return [4 /*yield*/, this._agentChangedSubject.next()];
+                    case 1:
+                        _a.sent();
+                        return [2 /*return*/];
+                }
+            });
+        }); };
         this.setAgentName = function (agentName) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 this.params.logger.debug("ClientSwarm swarmName=".concat(this.params.swarmName, " clientId=").concat(this.params.clientId, " setAgentName agentName=").concat(agentName));
@@ -1019,7 +1046,9 @@ var ClientSwarm = /** @class */ (function () {
                 return [2 /*return*/];
             });
         }); };
-        this.params.logger.debug("ClientSwarm swarmName=".concat(this.params.swarmName, " clientId=").concat(this.params.clientId, " CTOR"));
+        this.params.logger.debug("ClientSwarm swarmName=".concat(this.params.swarmName, " clientId=").concat(this.params.clientId, " CTOR"), {
+            params: params,
+        });
         this._activeAgent = params.defaultAgent;
     }
     return ClientSwarm;
@@ -1096,6 +1125,18 @@ var SwarmConnectionService = /** @class */ (function () {
                 }
             });
         }); };
+        this.setAgentRef = function (agentName, agent) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("swarmConnectionService setAgentRef", {
+                            context: this.contextService.context,
+                        });
+                        return [4 /*yield*/, this.getSwarm(this.contextService.context.clientId, this.contextService.context.swarmName).setAgentRef(agentName, agent)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         this.setAgentName = function (agentName) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -1271,7 +1312,9 @@ var ClientSession = /** @class */ (function () {
                 });
             }); };
         };
-        this.params.logger.debug("ClientSession clientId=".concat(this.params.clientId, " CTOR"));
+        this.params.logger.debug("ClientSession clientId=".concat(this.params.clientId, " CTOR"), {
+            params: params,
+        });
     }
     return ClientSession;
 }());
@@ -1352,6 +1395,31 @@ var AgentPublicService = /** @class */ (function () {
         var _this = this;
         this.loggerService = inject(TYPES.loggerService);
         this.agentConnectionService = inject(TYPES.agentConnectionService);
+        this.createAgentRef = function (clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("agentPublicService createAgentRef", {
+                            clientId: clientId,
+                            agentName: agentName,
+                        });
+                        return [4 /*yield*/, ContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.agentConnectionService.getAgent(clientId, agentName)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                clientId: clientId,
+                                agentName: agentName,
+                                swarmName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         this.execute = function (input, clientId, agentName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
@@ -1828,6 +1896,33 @@ var SwarmPublicService = /** @class */ (function () {
                 }
             });
         }); };
+        this.setAgentRef = function (clientId, swarmName, agentName, agent) { return __awaiter(_this, void 0, void 0, function () {
+            var _this = this;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.loggerService.log("swarmPublicService setAgentRef", {
+                            agentName: agentName,
+                            agent: agent,
+                            clientId: clientId,
+                            swarmName: swarmName,
+                        });
+                        return [4 /*yield*/, ContextService.runInContext(function () { return __awaiter(_this, void 0, void 0, function () {
+                                return __generator(this, function (_a) {
+                                    switch (_a.label) {
+                                        case 0: return [4 /*yield*/, this.swarmConnectionService.setAgentRef(agentName, agent)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                clientId: clientId,
+                                swarmName: swarmName,
+                                agentName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
         this.setAgentName = function (agentName, clientId, swarmName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
@@ -2229,30 +2324,46 @@ var swarm = __assign(__assign(__assign(__assign(__assign({}, baseServices), conn
 init();
 
 var addAgent = function (agentSchema) {
+    swarm.loggerService.log('function addAgent', {
+        agentSchema: agentSchema
+    });
     swarm.agentValidationService.addAgent(agentSchema.agentName, agentSchema);
     swarm.agentSchemaService.register(agentSchema.agentName, agentSchema);
     return agentSchema.agentName;
 };
 
 var addCompletion = function (completionSchema) {
+    swarm.loggerService.log('function addCompletion', {
+        completionSchema: completionSchema,
+    });
     swarm.completionValidationService.addCompletion(completionSchema.completionName);
     swarm.completionSchemaService.register(completionSchema.completionName, completionSchema);
     return completionSchema.completionName;
 };
 
 var addSwarm = function (swarmSchema) {
+    swarm.loggerService.log('function addSwarm', {
+        swarmSchema: swarmSchema,
+    });
     swarm.swarmValidationService.addSwarm(swarmSchema.swarmName, swarmSchema);
     swarm.swarmSchemaService.register(swarmSchema.swarmName, swarmSchema);
     return swarmSchema.swarmName;
 };
 
 var addTool = function (toolSchema) {
+    swarm.loggerService.log('function addTool', {
+        toolSchema: toolSchema,
+    });
     swarm.toolValidationService.addTool(toolSchema.toolName, toolSchema);
     swarm.toolSchemaService.register(toolSchema.toolName, toolSchema);
     return toolSchema.toolName;
 };
 
 var makeConnection = function (connector, clientId, swarmName) {
+    swarm.loggerService.log("function makeConnection", {
+        clientId: clientId,
+        swarmName: swarmName,
+    });
     swarm.swarmValidationService.validate(swarmName, "makeConnection");
     swarm.sessionValidationService.addSession(clientId, swarmName);
     var send = swarm.sessionPublicService.connect(connector, clientId, swarmName);
@@ -2277,18 +2388,66 @@ var makeConnection = function (connector, clientId, swarmName) {
     }); });
 };
 
+var CHANGE_AGENT_TTL = 15 * 60 * 1000;
+var CHANGE_AGENT_GC = 60 * 1000;
+var createChangeAgent = ttl(function (clientId) {
+    return queued(function (agentName) { return __awaiter(void 0, void 0, void 0, function () {
+        var swarmName, _a, _b, _c;
+        return __generator(this, function (_d) {
+            switch (_d.label) {
+                case 0:
+                    swarm.sessionValidationService.validate(clientId, "changeAgent");
+                    swarm.agentValidationService.validate(agentName, "changeAgent");
+                    swarmName = swarm.sessionValidationService.getSwarm(clientId);
+                    return [4 /*yield*/, swarm.agentPublicService.dispose(clientId, agentName)];
+                case 1:
+                    _d.sent();
+                    return [4 /*yield*/, swarm.historyPublicService.dispose(clientId, agentName)];
+                case 2:
+                    _d.sent();
+                    _b = (_a = swarm.swarmPublicService).setAgentRef;
+                    _c = [clientId,
+                        swarmName,
+                        agentName];
+                    return [4 /*yield*/, swarm.agentPublicService.createAgentRef(clientId, agentName)];
+                case 3: return [4 /*yield*/, _b.apply(_a, _c.concat([_d.sent()]))];
+                case 4:
+                    _d.sent();
+                    return [4 /*yield*/, swarm.swarmPublicService.setAgentName(agentName, clientId, swarmName)];
+                case 5:
+                    _d.sent();
+                    return [2 /*return*/];
+            }
+        });
+    }); });
+}, {
+    key: function (_a) {
+        var _b = __read(_a, 1), clientId = _b[0];
+        return "".concat(clientId);
+    },
+    timeout: CHANGE_AGENT_TTL,
+});
+var createGc$1 = singleshot(function () { return __awaiter(void 0, void 0, void 0, function () {
+    return __generator(this, function (_a) {
+        setInterval(createChangeAgent.gc, CHANGE_AGENT_GC);
+        return [2 /*return*/];
+    });
+}); });
 var changeAgent = function (agentName, clientId) { return __awaiter(void 0, void 0, void 0, function () {
-    var swarmName;
+    var run;
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
-                swarm.sessionValidationService.validate(clientId, "changeAgent");
-                swarm.agentValidationService.validate(agentName, "changeAgent");
-                swarmName = swarm.sessionValidationService.getSwarm(clientId);
-                return [4 /*yield*/, swarm.swarmPublicService.setAgentName(agentName, clientId, swarmName)];
+                swarm.loggerService.log('function changeAgent', {
+                    agentName: agentName,
+                    clientId: clientId,
+                });
+                return [4 /*yield*/, createChangeAgent(clientId)];
             case 1:
-                _a.sent();
-                return [2 /*return*/];
+                run = _a.sent();
+                createGc$1();
+                return [4 /*yield*/, run(agentName)];
+            case 2: return [2 /*return*/, _a.sent()];
         }
     });
 }); };
@@ -2297,6 +2456,10 @@ var disposeConnection = function (clientId, swarmName) { return __awaiter(void 0
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                swarm.loggerService.log("function disposeConnection", {
+                    clientId: clientId,
+                    swarmName: swarmName,
+                });
                 swarm.swarmValidationService.validate(swarmName, "disposeConnection");
                 swarm.sessionValidationService.removeSession(clientId);
                 return [4 /*yield*/, swarm.sessionPublicService.dispose(clientId, swarmName)];
@@ -2366,7 +2529,13 @@ var complete = function (content, clientId, swarmName) { return __awaiter(void 0
     var run;
     return __generator(this, function (_a) {
         switch (_a.label) {
-            case 0: return [4 /*yield*/, createComplete(clientId, swarmName)];
+            case 0:
+                swarm.loggerService.log("function complete", {
+                    content: content,
+                    clientId: clientId,
+                    swarmName: swarmName,
+                });
+                return [4 /*yield*/, createComplete(clientId, swarmName)];
             case 1:
                 run = _a.sent();
                 createGc();
@@ -2377,6 +2546,10 @@ var complete = function (content, clientId, swarmName) { return __awaiter(void 0
 }); };
 
 var session = function (clientId, swarmName) {
+    swarm.loggerService.log("function session", {
+        clientId: clientId,
+        swarmName: swarmName,
+    });
     swarm.swarmValidationService.validate(swarmName, "session");
     swarm.sessionValidationService.addSession(clientId, swarmName);
     return {
@@ -2406,6 +2579,9 @@ var getRawHistory = function (clientId) { return __awaiter(void 0, void 0, void
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                swarm.loggerService.log("function getRawHistory", {
+                    clientId: clientId,
+                });
                 swarm.sessionValidationService.validate(clientId, "getRawHistory");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
                 swarm.swarmValidationService.validate(swarmName, "getRawHistory");
@@ -2425,6 +2601,10 @@ var getAgentHistory = function (clientId, agentName) { return __awaiter(void 0,
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                swarm.loggerService.log("function getAgentHistory", {
+                    clientId: clientId,
+                    agentName: agentName,
+                });
                 swarm.agentValidationService.validate(agentName, "getAgentHistory");
                 prompt = swarm.agentSchemaService.get(agentName).prompt;
                 return [4 /*yield*/, swarm.historyPublicService.toArrayForAgent(prompt, clientId, agentName)];
@@ -2440,6 +2620,11 @@ var commitToolOutput = function (content, clientId, agentName) { return __awaite
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                swarm.loggerService.log('function commitToolOutput', {
+                    content: content,
+                    clientId: clientId,
+                    agentName: agentName,
+                });
                 swarm.agentValidationService.validate(agentName, "commitSystemMessage");
                 swarm.sessionValidationService.validate(clientId, "commitToolOutput");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
@@ -2468,6 +2653,11 @@ var commitSystemMessage = function (content, clientId, agentName) { return __awa
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                swarm.loggerService.log('function commitSystemMessage', {
+                    content: content,
+                    clientId: clientId,
+                    agentName: agentName,
+                });
                 swarm.agentValidationService.validate(agentName, "commitSystemMessage");
                 swarm.sessionValidationService.validate(clientId, "commitSystemMessage");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);
@@ -2496,6 +2686,11 @@ var execute = function (content, clientId, agentName) { return __awaiter(void 0,
     return __generator(this, function (_a) {
         switch (_a.label) {
             case 0:
+                swarm.loggerService.log("function execute", {
+                    content: content,
+                    clientId: clientId,
+                    agentName: agentName,
+                });
                 swarm.agentValidationService.validate(agentName, "commitSystemMessage");
                 swarm.sessionValidationService.validate(clientId, "execute");
                 swarmName = swarm.sessionValidationService.getSwarm(clientId);

package/package.json

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

package/types.d.ts

@@ -125,6 +125,7 @@ interface ISwarm {
     waitForOutput(): Promise<string>;
     getAgentName(): Promise<AgentName>;
     getAgent(): Promise<IAgent>;
+    setAgentRef(agentName: AgentName, agent: IAgent): Promise<void>;
     setAgentName(agentName: AgentName): Promise<void>;
 }
 type SwarmName = string;
@@ -215,11 +216,13 @@ declare class ToolSchemaService {
 
 declare class ClientSwarm implements ISwarm {
     readonly params: ISwarmParams;
+    private _agentChangedSubject;
     private _activeAgent;
     constructor(params: ISwarmParams);
     waitForOutput: () => Promise<string>;
     getAgentName: () => Promise<AgentName>;
     getAgent: () => Promise<IAgent>;
+    setAgentRef: (agentName: AgentName, agent: IAgent) => Promise<void>;
     setAgentName: (agentName: AgentName) => Promise<void>;
 }
 
@@ -232,6 +235,7 @@ declare class SwarmConnectionService implements ISwarm {
     waitForOutput: () => Promise<string>;
     getAgentName: () => Promise<string>;
     getAgent: () => Promise<IAgent>;
+    setAgentRef: (agentName: AgentName, agent: IAgent) => Promise<void>;
     setAgentName: (agentName: AgentName) => Promise<void>;
     dispose: () => Promise<void>;
 }
@@ -311,6 +315,7 @@ type TAgentConnectionService = {
 declare class AgentPublicService implements TAgentConnectionService {
     private readonly loggerService;
     private readonly agentConnectionService;
+    createAgentRef: (clientId: string, agentName: AgentName) => Promise<ClientAgent>;
     execute: (input: string, clientId: string, agentName: AgentName) => Promise<void>;
     waitForOutput: (clientId: string, agentName: AgentName) => Promise<string>;
     commitToolOutput: (content: string, clientId: string, agentName: AgentName) => Promise<void>;
@@ -368,6 +373,7 @@ declare class SwarmPublicService implements TSwarmConnectionService {
     waitForOutput: (clientId: string, swarmName: SwarmName) => Promise<string>;
     getAgentName: (clientId: string, swarmName: SwarmName) => Promise<string>;
     getAgent: (clientId: string, swarmName: SwarmName) => Promise<IAgent>;
+    setAgentRef: (clientId: string, swarmName: SwarmName, agentName: AgentName, agent: IAgent) => Promise<void>;
     setAgentName: (agentName: AgentName, clientId: string, swarmName: SwarmName) => Promise<void>;
     dispose: (clientId: string, swarmName: SwarmName) => Promise<void>;
 }