agent-swarm-kit 1.0.122 → 1.0.124

package/build/index.cjs

@@ -4,6 +4,7 @@ var diScoped = require('di-scoped');
 var diKit = require('di-kit');
 var functoolsKit = require('functools-kit');
 var xml2js = require('xml2js');
+var async_hooks = require('async_hooks');
 var lodashEs = require('lodash-es');
 var path = require('path');
 var promises = require('fs/promises');
@@ -192,6 +193,7 @@ var connectionServices$1 = {
     sharedStorageConnectionService: Symbol('sharedStorageConnectionService'),
     stateConnectionService: Symbol('stateConnectionService'),
     sharedStateConnectionService: Symbol('sharedStateConnectionService'),
+    policyConnectionService: Symbol('policyConnectionService'),
 };
 var schemaServices$1 = {
     completionSchemaService: Symbol('completionSchemaService'),
@@ -202,6 +204,7 @@ var schemaServices$1 = {
     storageSchemaService: Symbol('storageSchemaService'),
     stateSchemaService: Symbol('stateSchemaService'),
     memorySchemaService: Symbol('memorySchemaService'),
+    policySchemaService: Symbol('policySchemaService'),
 };
 var metaServices$1 = {
     agentMetaService: Symbol('agentMetaService'),
@@ -216,6 +219,7 @@ var publicServices$1 = {
     sharedStoragePublicService: Symbol('sharedStoragePublicService'),
     statePublicService: Symbol('statePublicService'),
     sharedStatePublicService: Symbol('sharedStatePublicService'),
+    policyPublicService: Symbol('policyPublicService'),
 };
 var validationServices$1 = {
     agentValidationService: Symbol('agentValidationService'),
@@ -225,6 +229,7 @@ var validationServices$1 = {
     completionValidationService: Symbol('completionValidationService'),
     embeddingValidationService: Symbol('embeddingValidationService'),
     storageValidationService: Symbol('storageValidationService'),
+    policyValidationService: Symbol('policyValidationService'),
 };
 var TYPES = __assign(__assign(__assign(__assign(__assign(__assign(__assign({}, baseServices$1), contextServices$1), schemaServices$1), connectionServices$1), publicServices$1), validationServices$1), metaServices$1);
 
@@ -954,6 +959,40 @@ var nameToTitle = function (name) {
     return __spreadArray(["".concat(word.charAt(0).toUpperCase()).concat(word.slice(1))], __read(rest), false).join(' ');
 };
 
+/**
+ * A higher-order function that ensures execution outside of existing method and execution contexts.
+ *
+ * @template T - Generic type extending any function
+ * @param {T} run - The function to be executed outside of existing contexts
+ * @returns {(...args: Parameters<T>) => ReturnType<T>} A wrapped function that executes outside of any existing contexts
+ *
+ * @example
+ * const myFunction = (arg: string) => console.log(arg);
+ * const contextSafeFunction = beginContext(myFunction);
+ * contextSafeFunction('test'); // Executes myFunction outside of any existing contexts
+ *
+ * @remarks
+ * This utility function checks for both MethodContext and ExecutionContext.
+ * If either context exists, the provided function will be executed outside of those contexts.
+ * This is useful for ensuring clean execution environments for certain operations.
+ */
+var beginContext = function (run) {
+    return function () {
+        var args = [];
+        for (var _i = 0; _i < arguments.length; _i++) {
+            args[_i] = arguments[_i];
+        }
+        if (MethodContextService.hasContext() ||
+            ExecutionContextService.hasContext()) {
+            var resource = new async_hooks.AsyncResource("UNTRACKED");
+            var result = resource.runInAsyncScope(function () { return run.apply(void 0, __spreadArray([], __read(args), false)); });
+            resource.emitDestroy();
+            return result;
+        }
+        return run.apply(void 0, __spreadArray([], __read(args), false));
+    };
+};
+
 var _a;
 var LOGGER_INSTANCE_WAIT_FOR_INIT = Symbol("wait-for-init");
 var LOGGER_INSTANCE_WAIT_FOR_FN = function (self) { return __awaiter(void 0, void 0, void 0, function () {
@@ -1105,7 +1144,7 @@ var LoggerUtils = /** @class */ (function () {
          * @param {...any[]} args - The log arguments.
          * @returns {Promise<void>}
          */
-        this.logClient = function (clientId, topic) {
+        this.logClient = beginContext(function (clientId, topic) {
             var args = [];
             for (var _i = 2; _i < arguments.length; _i++) {
                 args[_i - 2] = arguments[_i];
@@ -1134,6 +1173,7 @@ var LoggerUtils = /** @class */ (function () {
                                 }); }, {
                                     clientId: clientId,
                                     agentName: "",
+                                    policyName: "",
                                     methodName: "LoggerUtils.logClient",
                                     stateName: "",
                                     storageName: "",
@@ -1145,7 +1185,7 @@ var LoggerUtils = /** @class */ (function () {
                     }
                 });
             });
-        };
+        });
         /**
          * @method infoClient
          * @description Logs an info message for a specific client.
@@ -1154,7 +1194,7 @@ var LoggerUtils = /** @class */ (function () {
          * @param {...any[]} args - The info log arguments.
          * @returns {Promise<void>}
          */
-        this.infoClient = function (clientId, topic) {
+        this.infoClient = beginContext(function (clientId, topic) {
             var args = [];
             for (var _i = 2; _i < arguments.length; _i++) {
                 args[_i - 2] = arguments[_i];
@@ -1183,6 +1223,7 @@ var LoggerUtils = /** @class */ (function () {
                                 }); }, {
                                     clientId: clientId,
                                     agentName: "",
+                                    policyName: "",
                                     methodName: "LoggerUtils.infoClient",
                                     stateName: "",
                                     storageName: "",
@@ -1194,7 +1235,7 @@ var LoggerUtils = /** @class */ (function () {
                     }
                 });
             });
-        };
+        });
         /**
          * @method debugClient
          * @description Logs a debug message for a specific client.
@@ -1203,7 +1244,7 @@ var LoggerUtils = /** @class */ (function () {
          * @param {...any[]} args - The debug log arguments.
          * @returns {Promise<void>}
          */
-        this.debugClient = function (clientId, topic) {
+        this.debugClient = beginContext(function (clientId, topic) {
             var args = [];
             for (var _i = 2; _i < arguments.length; _i++) {
                 args[_i - 2] = arguments[_i];
@@ -1232,6 +1273,7 @@ var LoggerUtils = /** @class */ (function () {
                                 }); }, {
                                     clientId: clientId,
                                     agentName: "",
+                                    policyName: "",
                                     methodName: "LoggerUtils.debugClient",
                                     stateName: "",
                                     storageName: "",
@@ -1243,7 +1285,7 @@ var LoggerUtils = /** @class */ (function () {
                     }
                 });
             });
-        };
+        });
         /**
          * @method log
          * @description Logs a message.
@@ -1456,6 +1498,7 @@ var CC_LOGGER_ENABLE_CONSOLE = false;
 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_EMPTY_OUTPUT_PLACEHOLDERS: CC_EMPTY_OUTPUT_PLACEHOLDERS,
@@ -1482,6 +1525,7 @@ var GLOBAL_CONFIG = {
     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,
 };
 var setConfig = function (config) {
     Object.assign(GLOBAL_CONFIG, config);
@@ -3735,6 +3779,7 @@ var SwarmSchemaService = /** @class */ (function () {
          * Validation for swarm schema
          */
         this.validateShallow = function (swarmSchema) {
+            var _a;
             GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
                 _this.loggerService.info("swarmSchemaService validateShallow", {
                     swarmSchema: swarmSchema,
@@ -3754,6 +3799,15 @@ var SwarmSchemaService = /** @class */ (function () {
             if (swarmSchema.agentList.some(function (value) { return typeof value !== "string"; })) {
                 throw new Error("agent-swarm swarm schema validation failed: missing agentList for swarmName=".concat(swarmSchema.swarmName, " value=[").concat(swarmSchema.agentList, "]"));
             }
+            if (swarmSchema.policies && !Array.isArray(swarmSchema.policies)) {
+                throw new Error("agent-swarm swarm schema validation failed: invalid policies for swarmName=".concat(swarmSchema.swarmName, " value=").concat(swarmSchema.policies));
+            }
+            if (swarmSchema.policies && swarmSchema.policies.length !== new Set(swarmSchema.policies).size) {
+                throw new Error("agent-swarm agent schema validation failed: found duplicate policies for swarmName=".concat(swarmSchema.swarmName, " policies=[").concat(swarmSchema.policies, "]"));
+            }
+            if ((_a = swarmSchema.policies) === null || _a === void 0 ? void 0 : _a.some(function (value) { return typeof value !== "string"; })) {
+                throw new Error("agent-swarm swarm schema validation failed: missing policies for swarmName=".concat(swarmSchema.swarmName, " value=[").concat(swarmSchema.policies, "]"));
+            }
         };
         /**
          * Registers a new swarm schema.
@@ -3852,18 +3906,33 @@ var ClientSession = /** @class */ (function () {
      */
     ClientSession.prototype.emit = function (message) {
         return __awaiter(this, void 0, void 0, function () {
-            return __generator(this, function (_a) {
-                switch (_a.label) {
+            var _a, _b;
+            return __generator(this, function (_c) {
+                switch (_c.label) {
                     case 0:
                         GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                             this.params.logger.debug("ClientSession clientId=".concat(this.params.clientId, " emit"), {
                                 message: message,
                             });
+                        return [4 /*yield*/, functoolsKit.not(this.params.policy.validateOutput(message, this.params.clientId, this.params.swarmName))];
+                    case 1:
+                        if (!_c.sent()) return [3 /*break*/, 4];
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            this.params.logger.debug("ClientSession clientId=".concat(this.params.clientId, " emit method canceled due to the banhammer of a client"), {
+                                message: message,
+                            });
+                        _b = (_a = this._emitSubject).next;
+                        return [4 /*yield*/, this.params.policy.getBanMessage(this.params.clientId, this.params.swarmName)];
+                    case 2: return [4 /*yield*/, _b.apply(_a, [_c.sent()])];
+                    case 3:
+                        _c.sent();
+                        return [2 /*return*/];
+                    case 4:
                         this.params.onEmit &&
                             this.params.onEmit(this.params.clientId, this.params.swarmName, message);
                         return [4 /*yield*/, this._emitSubject.next(message)];
-                    case 1:
-                        _a.sent();
+                    case 5:
+                        _c.sent();
                         return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
                                 type: "emit",
                                 source: "session-bus",
@@ -3876,8 +3945,8 @@ var ClientSession = /** @class */ (function () {
                                 },
                                 clientId: this.params.clientId,
                             })];
-                    case 2:
-                        _a.sent();
+                    case 6:
+                        _c.sent();
                         return [2 /*return*/];
                 }
             });
@@ -3900,32 +3969,53 @@ var ClientSession = /** @class */ (function () {
                                 message: message,
                                 mode: mode,
                             });
+                        return [4 /*yield*/, functoolsKit.not(this.params.policy.validateInput(message, this.params.clientId, this.params.swarmName))];
+                    case 1:
+                        if (!_a.sent()) return [3 /*break*/, 3];
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            this.params.logger.debug("ClientSession clientId=".concat(this.params.clientId, " execution canceled due to the banhammer of a client"), {
+                                message: message,
+                                mode: mode,
+                            });
+                        return [4 /*yield*/, this.params.policy.getBanMessage(this.params.clientId, this.params.swarmName)];
+                    case 2: return [2 /*return*/, _a.sent()];
+                    case 3:
                         this.params.onExecute &&
                             this.params.onExecute(this.params.clientId, this.params.swarmName, message, mode);
                         return [4 /*yield*/, this.params.swarm.getAgent()];
-                    case 1:
+                    case 4:
                         agent = _a.sent();
                         outputAwaiter = this.params.swarm.waitForOutput();
                         agent.execute(message, mode);
                         return [4 /*yield*/, outputAwaiter];
-                    case 2:
+                    case 5:
                         output = _a.sent();
-                        return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
-                                type: "execute",
-                                source: "session-bus",
-                                input: {
-                                    message: message,
-                                    mode: mode,
-                                },
-                                output: {
-                                    result: output,
-                                },
-                                context: {
-                                    swarmName: this.params.swarmName,
-                                },
-                                clientId: this.params.clientId,
-                            })];
-                    case 3:
+                        return [4 /*yield*/, functoolsKit.not(this.params.policy.validateOutput(output, this.params.clientId, this.params.swarmName))];
+                    case 6:
+                        if (!_a.sent()) return [3 /*break*/, 8];
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            this.params.logger.debug("ClientSession clientId=".concat(this.params.clientId, " execution output emit canceled due to the banhammer of a client"), {
+                                message: message,
+                                mode: mode,
+                            });
+                        return [4 /*yield*/, this.params.policy.getBanMessage(this.params.clientId, this.params.swarmName)];
+                    case 7: return [2 /*return*/, _a.sent()];
+                    case 8: return [4 /*yield*/, this.params.bus.emit(this.params.clientId, {
+                            type: "execute",
+                            source: "session-bus",
+                            input: {
+                                message: message,
+                                mode: mode,
+                            },
+                            output: {
+                                result: output,
+                            },
+                            context: {
+                                swarmName: this.params.swarmName,
+                            },
+                            clientId: this.params.clientId,
+                        })];
+                    case 9:
                         _a.sent();
                         return [2 /*return*/, output];
                 }
@@ -4295,111 +4385,455 @@ var ClientSession = /** @class */ (function () {
     return ClientSession;
 }());
 
+var METHOD_NAME_BAN_CLIENT = "PolicyUtils.banClient";
+var METHOD_NAME_UNBAN_CLIENT = "PolicyUtils.unbanClient";
 /**
- * Service for managing session connections.
- * @implements {ISession}
+ * NoopPolicy class implements the IPolicy interface with no-op methods.
  */
-var SessionConnectionService = /** @class */ (function () {
-    function SessionConnectionService() {
-        var _this = this;
-        this.loggerService = inject(TYPES.loggerService);
-        this.busService = inject(TYPES.busService);
-        this.methodContextService = inject(TYPES.methodContextService);
-        this.swarmConnectionService = inject(TYPES.swarmConnectionService);
-        this.swarmSchemaService = inject(TYPES.swarmSchemaService);
-        /**
-         * Retrieves a memoized session based on clientId and swarmName.
-         * @param {string} clientId - The client ID.
-         * @param {string} swarmName - The swarm name.
-         * @returns {ClientSession} The client session.
-         */
-        this.getSession = functoolsKit.memoize(function (_a) {
-            var _b = __read(_a, 2), clientId = _b[0], swarmName = _b[1];
-            return "".concat(clientId, "-").concat(swarmName);
-        }, function (clientId, swarmName) {
-            var callbacks = _this.swarmSchemaService.get(swarmName).callbacks;
-            return new ClientSession(__assign({ clientId: clientId, logger: _this.loggerService, bus: _this.busService, swarm: _this.swarmConnectionService.getSwarm(clientId, swarmName), swarmName: swarmName }, callbacks));
-        });
-        /**
-         * Emits a message to the session.
-         * @param {string} content - The content to emit.
-         * @returns {Promise<void>} A promise that resolves when the message is emitted.
-         */
-        this.emit = function (content) { return __awaiter(_this, void 0, void 0, function () {
+var NoopPolicy = /** @class */ (function () {
+    /**
+     * Constructs a NoopPolicy instance.
+     * @param {string} swarmName - The name of the swarm.
+     */
+    function NoopPolicy(swarmName) {
+        this.swarmName = swarmName;
+        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+            swarm$1.loggerService.debug("NoopPolicy CTOR swarmName=".concat(swarmName));
+    }
+    /**
+     * Gets the ban message.
+     * @returns {Promise<string>} The ban message.
+     */
+    NoopPolicy.prototype.getBanMessage = function () {
+        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+            swarm$1.loggerService.debug("NoopPolicy getBanMessage swarmName=".concat(this.swarmName));
+        return Promise.resolve(GLOBAL_CONFIG.CC_BANHAMMER_PLACEHOLDER);
+    };
+    /**
+     * Validates the input.
+     * @returns {Promise<boolean>} True if the input is valid, otherwise false.
+     */
+    NoopPolicy.prototype.validateInput = function () {
+        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+            swarm$1.loggerService.debug("NoopPolicy validateInput swarmName=".concat(this.swarmName));
+        return Promise.resolve(true);
+    };
+    /**
+     * Validates the output.
+     * @returns {Promise<boolean>} True if the output is valid, otherwise false.
+     */
+    NoopPolicy.prototype.validateOutput = function () {
+        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+            swarm$1.loggerService.debug("NoopPolicy validateOutput swarmName=".concat(this.swarmName));
+        return Promise.resolve(true);
+    };
+    /**
+     * Bans a client.
+     * @returns {Promise<void>}
+     */
+    NoopPolicy.prototype.banClient = function () {
+        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+            swarm$1.loggerService.debug("NoopPolicy banClient swarmName=".concat(this.swarmName));
+        return Promise.resolve();
+    };
+    /**
+     * Unbans a client.
+     * @returns {Promise<void>}
+     */
+    NoopPolicy.prototype.unbanClient = function () {
+        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+            swarm$1.loggerService.debug("NoopPolicy unbanClient swarmName=".concat(this.swarmName));
+        return Promise.resolve();
+    };
+    return NoopPolicy;
+}());
+/**
+ * MergePolicy class implements the IPolicy interface and merges multiple policies.
+ */
+var MergePolicy = /** @class */ (function () {
+    /**
+     * Constructs a MergePolicy instance.
+     * @param {IPolicy[]} policies - The policies to merge.
+     * @param {SwarmName} swarmName - The name of the swarm.
+     */
+    function MergePolicy(policies, swarmName) {
+        this.policies = policies;
+        this.swarmName = swarmName;
+        this._targetPolicy = null;
+        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+            swarm$1.loggerService.debug("MergePolicy CTOR swarmName=".concat(swarmName), {
+                policies: policies,
+            });
+    }
+    /**
+     * Gets the ban message.
+     * @param {SessionId} clientId - The client ID.
+     * @param {SwarmName} swarmName - The name of the swarm.
+     * @returns {Promise<string>} The ban message.
+     */
+    MergePolicy.prototype.getBanMessage = function (clientId, swarmName) {
+        return __awaiter(this, void 0, void 0, function () {
+            var policy;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
-                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
-                            this.loggerService.info("sessionConnectionService emit", {
-                                content: content,
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            swarm$1.loggerService.debug("MergePolicy getBanMessage swarmName=".concat(swarmName), {
+                                clientId: clientId,
                             });
-                        return [4 /*yield*/, this.getSession(this.methodContextService.context.clientId, this.methodContextService.context.swarmName).emit(content)];
+                        if (!this._targetPolicy) return [3 /*break*/, 2];
+                        policy = this._targetPolicy;
+                        this._targetPolicy = null;
+                        return [4 /*yield*/, policy.getBanMessage(clientId, swarmName)];
                     case 1: return [2 /*return*/, _a.sent()];
+                    case 2: return [2 /*return*/, GLOBAL_CONFIG.CC_BANHAMMER_PLACEHOLDER];
                 }
             });
-        }); };
-        /**
-         * Executes a command in the session.
-         * @param {string} content - The content to execute.
-         * @returns {Promise<string>} A promise that resolves with the execution result.
-         */
-        this.execute = function (content, mode) { return __awaiter(_this, void 0, void 0, function () {
-            return __generator(this, function (_a) {
-                switch (_a.label) {
+        });
+    };
+    /**
+     * Validates the input.
+     * @param {string} incoming - The incoming data.
+     * @param {SessionId} clientId - The client ID.
+     * @param {SwarmName} swarmName - The name of the swarm.
+     * @returns {Promise<boolean>} True if the input is valid, otherwise false.
+     */
+    MergePolicy.prototype.validateInput = function (incoming, clientId, swarmName) {
+        return __awaiter(this, void 0, void 0, function () {
+            var _a, _b, policy, e_1_1;
+            var e_1, _c;
+            return __generator(this, function (_d) {
+                switch (_d.label) {
                     case 0:
-                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
-                            this.loggerService.info("sessionConnectionService execute", {
-                                content: content,
-                                mode: mode,
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            swarm$1.loggerService.debug("MergePolicy validateInput swarmName=".concat(swarmName), {
+                                clientId: clientId,
                             });
-                        return [4 /*yield*/, this.getSession(this.methodContextService.context.clientId, this.methodContextService.context.swarmName).execute(content, mode)];
-                    case 1: return [2 /*return*/, _a.sent()];
+                        _d.label = 1;
+                    case 1:
+                        _d.trys.push([1, 6, 7, 8]);
+                        _a = __values(this.policies), _b = _a.next();
+                        _d.label = 2;
+                    case 2:
+                        if (!!_b.done) return [3 /*break*/, 5];
+                        policy = _b.value;
+                        return [4 /*yield*/, functoolsKit.not(policy.validateInput(incoming, clientId, swarmName))];
+                    case 3:
+                        if (_d.sent()) {
+                            this._targetPolicy = policy;
+                            return [2 /*return*/, false];
+                        }
+                        _d.label = 4;
+                    case 4:
+                        _b = _a.next();
+                        return [3 /*break*/, 2];
+                    case 5: return [3 /*break*/, 8];
+                    case 6:
+                        e_1_1 = _d.sent();
+                        e_1 = { error: e_1_1 };
+                        return [3 /*break*/, 8];
+                    case 7:
+                        try {
+                            if (_b && !_b.done && (_c = _a.return)) _c.call(_a);
+                        }
+                        finally { if (e_1) throw e_1.error; }
+                        return [7 /*endfinally*/];
+                    case 8:
+                        this._targetPolicy = null;
+                        return [2 /*return*/, true];
                 }
             });
-        }); };
-        /**
-         * Run the completion stateless
-         * @param {string} content - The content to execute.
-         * @returns {Promise<string>} A promise that resolves with the execution result.
-         */
-        this.run = function (content) { return __awaiter(_this, void 0, void 0, function () {
-            return __generator(this, function (_a) {
-                switch (_a.label) {
+        });
+    };
+    /**
+     * Validates the output.
+     * @param {string} outgoing - The outgoing data.
+     * @param {SessionId} clientId - The client ID.
+     * @param {SwarmName} swarmName - The name of the swarm.
+     * @returns {Promise<boolean>} True if the output is valid, otherwise false.
+     */
+    MergePolicy.prototype.validateOutput = function (outgoing, clientId, swarmName) {
+        return __awaiter(this, void 0, void 0, function () {
+            var _a, _b, policy, e_2_1;
+            var e_2, _c;
+            return __generator(this, function (_d) {
+                switch (_d.label) {
                     case 0:
-                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
-                            this.loggerService.info("sessionConnectionService run", {
-                                content: content,
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            swarm$1.loggerService.debug("MergePolicy validateOutput swarmName=".concat(swarmName), {
+                                clientId: clientId,
                             });
-                        return [4 /*yield*/, this.getSession(this.methodContextService.context.clientId, this.methodContextService.context.swarmName).run(content)];
-                    case 1: return [2 /*return*/, _a.sent()];
+                        _d.label = 1;
+                    case 1:
+                        _d.trys.push([1, 6, 7, 8]);
+                        _a = __values(this.policies), _b = _a.next();
+                        _d.label = 2;
+                    case 2:
+                        if (!!_b.done) return [3 /*break*/, 5];
+                        policy = _b.value;
+                        return [4 /*yield*/, functoolsKit.not(policy.validateOutput(outgoing, clientId, swarmName))];
+                    case 3:
+                        if (_d.sent()) {
+                            this._targetPolicy = policy;
+                            return [2 /*return*/, false];
+                        }
+                        _d.label = 4;
+                    case 4:
+                        _b = _a.next();
+                        return [3 /*break*/, 2];
+                    case 5: return [3 /*break*/, 8];
+                    case 6:
+                        e_2_1 = _d.sent();
+                        e_2 = { error: e_2_1 };
+                        return [3 /*break*/, 8];
+                    case 7:
+                        try {
+                            if (_b && !_b.done && (_c = _a.return)) _c.call(_a);
+                        }
+                        finally { if (e_2) throw e_2.error; }
+                        return [7 /*endfinally*/];
+                    case 8:
+                        this._targetPolicy = null;
+                        return [2 /*return*/, true];
                 }
             });
-        }); };
-        /**
-         * Connects to the session using the provided connector.
-         * @param {SendMessageFn} connector - The function to send messages.
-         * @returns {ReceiveMessageFn} The function to receive messages.
-         */
-        this.connect = function (connector, clientId, swarmName) {
-            GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
-                _this.loggerService.info("sessionConnectionService connect");
-            return _this.getSession(clientId, swarmName).connect(connector);
-        };
-        /**
-         * Commits tool output to the session.
-         * @param {string} toolId - The `tool_call_id` for openai history
-         * @param {string} content - The content to commit.
-         * @returns {Promise<void>} A promise that resolves when the content is committed.
-         */
-        this.commitToolOutput = function (toolId, content) { return __awaiter(_this, void 0, void 0, function () {
-            return __generator(this, function (_a) {
-                switch (_a.label) {
-                    case 0:
-                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
-                            this.loggerService.info("sessionConnectionService commitToolOutput", {
-                                content: content,
-                                toolId: toolId,
-                            });
+        });
+    };
+    /**
+     * Bans a client.
+     * @param {SessionId} clientId - The client ID.
+     * @param {SwarmName} swarmName - The name of the swarm.
+     * @returns {Promise<void>}
+     */
+    MergePolicy.prototype.banClient = function (clientId, swarmName) {
+        return __awaiter(this, void 0, void 0, function () {
+            var _a, _b, policy;
+            return __generator(this, function (_c) {
+                switch (_c.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            swarm$1.loggerService.debug("MergePolicy banClient swarmName=".concat(swarmName), {
+                                clientId: clientId,
+                            });
+                        if (!this._targetPolicy) return [3 /*break*/, 2];
+                        return [4 /*yield*/, this._targetPolicy.banClient(clientId, swarmName)];
+                    case 1:
+                        _c.sent();
+                        return [2 /*return*/];
+                    case 2:
+                        _a = __read(this.policies, 1), _b = _a[0], policy = _b === void 0 ? null : _b;
+                        if (!policy) return [3 /*break*/, 4];
+                        return [4 /*yield*/, policy.banClient(clientId, swarmName)];
+                    case 3:
+                        _c.sent();
+                        _c.label = 4;
+                    case 4: return [2 /*return*/];
+                }
+            });
+        });
+    };
+    /**
+     * Unbans a client.
+     * @param {SessionId} clientId - The client ID.
+     * @param {SwarmName} swarmName - The name of the swarm.
+     * @returns {Promise<void>}
+     */
+    MergePolicy.prototype.unbanClient = function (clientId, swarmName) {
+        return __awaiter(this, void 0, void 0, function () {
+            var _a, _b, policy;
+            return __generator(this, function (_c) {
+                switch (_c.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            swarm$1.loggerService.debug("MergePolicy unbanClient swarmName=".concat(swarmName), {
+                                clientId: clientId,
+                            });
+                        if (!this._targetPolicy) return [3 /*break*/, 2];
+                        return [4 /*yield*/, this._targetPolicy.unbanClient(clientId, swarmName)];
+                    case 1:
+                        _c.sent();
+                        return [2 /*return*/];
+                    case 2:
+                        _a = __read(this.policies, 1), _b = _a[0], policy = _b === void 0 ? null : _b;
+                        if (!policy) return [3 /*break*/, 4];
+                        return [4 /*yield*/, policy.unbanClient(clientId, swarmName)];
+                    case 3:
+                        _c.sent();
+                        _c.label = 4;
+                    case 4: return [2 /*return*/];
+                }
+            });
+        });
+    };
+    return MergePolicy;
+}());
+/**
+ * PolicyUtils class provides utility methods for banning and unbanning clients.
+ */
+var PolicyUtils = /** @class */ (function () {
+    function PolicyUtils() {
+        var _this = this;
+        /**
+         * Bans a client.
+         * @param {Object} payload - The payload containing clientId, swarmName, and policyName.
+         * @param {string} payload.clientId - The client ID.
+         * @param {SwarmName} payload.swarmName - The name of the swarm.
+         * @param {PolicyName} payload.policyName - The name of the policy.
+         * @returns {Promise<void>}
+         */
+        this.banClient = beginContext(function (payload) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
+                            swarm$1.loggerService.log(METHOD_NAME_BAN_CLIENT, payload);
+                        swarm$1.sessionValidationService.validate(payload.clientId, METHOD_NAME_BAN_CLIENT);
+                        swarm$1.swarmValidationService.validate(payload.swarmName, METHOD_NAME_BAN_CLIENT);
+                        swarm$1.policyValidationService.validate(payload.policyName, METHOD_NAME_BAN_CLIENT);
+                        return [4 /*yield*/, swarm$1.policyPublicService.banClient(payload.swarmName, METHOD_NAME_BAN_CLIENT, payload.clientId, payload.policyName)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); });
+        /**
+         * Unbans a client.
+         * @param {Object} payload - The payload containing clientId, swarmName, and policyName.
+         * @param {string} payload.clientId - The client ID.
+         * @param {SwarmName} payload.swarmName - The name of the swarm.
+         * @param {PolicyName} payload.policyName - The name of the policy.
+         * @returns {Promise<void>}
+         */
+        this.unbanClient = beginContext(function (payload) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
+                            swarm$1.loggerService.log(METHOD_NAME_UNBAN_CLIENT, payload);
+                        swarm$1.sessionValidationService.validate(payload.clientId, METHOD_NAME_UNBAN_CLIENT);
+                        swarm$1.swarmValidationService.validate(payload.swarmName, METHOD_NAME_UNBAN_CLIENT);
+                        swarm$1.policyValidationService.validate(payload.policyName, METHOD_NAME_UNBAN_CLIENT);
+                        return [4 /*yield*/, swarm$1.policyPublicService.unbanClient(payload.swarmName, METHOD_NAME_UNBAN_CLIENT, payload.clientId, payload.policyName)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); });
+    }
+    return PolicyUtils;
+}());
+/**
+ * An instance of PolicyUtils.
+ * @type {PolicyUtils}
+ */
+var Policy = new PolicyUtils();
+
+/**
+ * Service for managing session connections.
+ * @implements {ISession}
+ */
+var SessionConnectionService = /** @class */ (function () {
+    function SessionConnectionService() {
+        var _this = this;
+        this.loggerService = inject(TYPES.loggerService);
+        this.busService = inject(TYPES.busService);
+        this.methodContextService = inject(TYPES.methodContextService);
+        this.swarmConnectionService = inject(TYPES.swarmConnectionService);
+        this.policyConnectionService = inject(TYPES.policyConnectionService);
+        this.swarmSchemaService = inject(TYPES.swarmSchemaService);
+        /**
+         * Retrieves a memoized session based on clientId and swarmName.
+         * @param {string} clientId - The client ID.
+         * @param {string} swarmName - The swarm name.
+         * @returns {ClientSession} The client session.
+         */
+        this.getSession = functoolsKit.memoize(function (_a) {
+            var _b = __read(_a, 2), clientId = _b[0], swarmName = _b[1];
+            return "".concat(clientId, "-").concat(swarmName);
+        }, function (clientId, swarmName) {
+            var _a = _this.swarmSchemaService.get(swarmName), callbacks = _a.callbacks, policies = _a.policies;
+            return new ClientSession(__assign({ clientId: clientId, policy: policies
+                    ? new MergePolicy(policies.map(_this.policyConnectionService.getPolicy), swarmName)
+                    : new NoopPolicy(swarmName), logger: _this.loggerService, bus: _this.busService, swarm: _this.swarmConnectionService.getSwarm(clientId, swarmName), swarmName: swarmName }, callbacks));
+        });
+        /**
+         * Emits a message to the session.
+         * @param {string} content - The content to emit.
+         * @returns {Promise<void>} A promise that resolves when the message is emitted.
+         */
+        this.emit = function (content) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                            this.loggerService.info("sessionConnectionService emit", {
+                                content: content,
+                            });
+                        return [4 /*yield*/, this.getSession(this.methodContextService.context.clientId, this.methodContextService.context.swarmName).emit(content)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Executes a command in the session.
+         * @param {string} content - The content to execute.
+         * @returns {Promise<string>} A promise that resolves with the execution result.
+         */
+        this.execute = function (content, mode) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                            this.loggerService.info("sessionConnectionService execute", {
+                                content: content,
+                                mode: mode,
+                            });
+                        return [4 /*yield*/, this.getSession(this.methodContextService.context.clientId, this.methodContextService.context.swarmName).execute(content, mode)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Run the completion stateless
+         * @param {string} content - The content to execute.
+         * @returns {Promise<string>} A promise that resolves with the execution result.
+         */
+        this.run = function (content) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                            this.loggerService.info("sessionConnectionService run", {
+                                content: content,
+                            });
+                        return [4 /*yield*/, this.getSession(this.methodContextService.context.clientId, this.methodContextService.context.swarmName).run(content)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Connects to the session using the provided connector.
+         * @param {SendMessageFn} connector - The function to send messages.
+         * @returns {ReceiveMessageFn} The function to receive messages.
+         */
+        this.connect = function (connector, clientId, swarmName) {
+            GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                _this.loggerService.info("sessionConnectionService connect");
+            return _this.getSession(clientId, swarmName).connect(connector);
+        };
+        /**
+         * Commits tool output to the session.
+         * @param {string} toolId - The `tool_call_id` for openai history
+         * @param {string} content - The content to commit.
+         * @returns {Promise<void>} A promise that resolves when the content is committed.
+         */
+        this.commitToolOutput = function (toolId, content) { return __awaiter(_this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                            this.loggerService.info("sessionConnectionService commitToolOutput", {
+                                content: content,
+                                toolId: toolId,
+                            });
                         return [4 /*yield*/, this.getSession(this.methodContextService.context.clientId, this.methodContextService.context.swarmName).commitToolOutput(toolId, content)];
                     case 1: return [2 /*return*/, _a.sent()];
                 }
@@ -4554,6 +4988,7 @@ var AgentPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 agentName: agentName,
+                                policyName: "",
                                 swarmName: "",
                                 storageName: "",
                                 stateName: "",
@@ -4593,6 +5028,7 @@ var AgentPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 agentName: agentName,
+                                policyName: "",
                                 swarmName: "",
                                 storageName: "",
                                 stateName: "",
@@ -4631,6 +5067,7 @@ var AgentPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 agentName: agentName,
+                                policyName: "",
                                 swarmName: "",
                                 storageName: "",
                                 stateName: "",
@@ -4667,6 +5104,7 @@ var AgentPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 agentName: agentName,
+                                policyName: "",
                                 swarmName: "",
                                 storageName: "",
                                 stateName: "",
@@ -4707,6 +5145,7 @@ var AgentPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 agentName: agentName,
+                                policyName: "",
                                 swarmName: "",
                                 storageName: "",
                                 stateName: "",
@@ -4745,6 +5184,7 @@ var AgentPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 agentName: agentName,
+                                policyName: "",
                                 swarmName: "",
                                 storageName: "",
                                 stateName: "",
@@ -4783,6 +5223,7 @@ var AgentPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 agentName: agentName,
+                                policyName: "",
                                 swarmName: "",
                                 storageName: "",
                                 stateName: "",
@@ -4821,6 +5262,7 @@ var AgentPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 agentName: agentName,
+                                policyName: "",
                                 swarmName: "",
                                 storageName: "",
                                 stateName: "",
@@ -4857,6 +5299,7 @@ var AgentPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 agentName: agentName,
+                                policyName: "",
                                 swarmName: "",
                                 storageName: "",
                                 stateName: "",
@@ -4893,6 +5336,7 @@ var AgentPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 agentName: agentName,
+                                policyName: "",
                                 swarmName: "",
                                 storageName: "",
                                 stateName: "",
@@ -4929,6 +5373,7 @@ var AgentPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 agentName: agentName,
+                                policyName: "",
                                 swarmName: "",
                                 storageName: "",
                                 stateName: "",
@@ -4965,6 +5410,7 @@ var AgentPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 agentName: agentName,
+                                policyName: "",
                                 swarmName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5015,6 +5461,7 @@ var HistoryPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 agentName: agentName,
+                                policyName: "",
                                 swarmName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5052,6 +5499,7 @@ var HistoryPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 agentName: agentName,
+                                policyName: "",
                                 swarmName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5088,6 +5536,7 @@ var HistoryPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 agentName: agentName,
+                                policyName: "",
                                 swarmName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5123,6 +5572,7 @@ var HistoryPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 agentName: agentName,
+                                policyName: "",
                                 swarmName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5175,6 +5625,7 @@ var SessionPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5213,6 +5664,7 @@ var SessionPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5250,6 +5702,7 @@ var SessionPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5349,6 +5802,7 @@ var SessionPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5387,6 +5841,7 @@ var SessionPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5425,6 +5880,7 @@ var SessionPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5463,6 +5919,7 @@ var SessionPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5498,6 +5955,7 @@ var SessionPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5533,6 +5991,7 @@ var SessionPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5569,6 +6028,7 @@ var SessionPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5614,6 +6074,7 @@ var SwarmPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5649,6 +6110,7 @@ var SwarmPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5685,6 +6147,7 @@ var SwarmPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5721,6 +6184,7 @@ var SwarmPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5756,6 +6220,7 @@ var SwarmPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5796,6 +6261,7 @@ var SwarmPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5834,6 +6300,7 @@ var SwarmPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -5870,6 +6337,7 @@ var SwarmPublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 swarmName: swarmName,
+                                policyName: "",
                                 agentName: "",
                                 storageName: "",
                                 stateName: "",
@@ -6448,6 +6916,7 @@ var SwarmValidationService = /** @class */ (function () {
         var _this = this;
         this.loggerService = inject(TYPES.loggerService);
         this.agentValidationService = inject(TYPES.agentValidationService);
+        this.policyValidationService = inject(TYPES.policyValidationService);
         this._swarmMap = new Map();
         /**
          * Adds a new swarm to the swarm map.
@@ -6502,6 +6971,7 @@ var SwarmValidationService = /** @class */ (function () {
             var _b = __read(_a, 1), swarmName = _b[0];
             return swarmName;
         }, function (swarmName, source) {
+            var _a;
             GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
                 _this.loggerService.info("swarmValidationService validate", {
                     swarmName: swarmName,
@@ -6517,6 +6987,9 @@ var SwarmValidationService = /** @class */ (function () {
             swarm.agentList.forEach(function (agentName) {
                 return _this.agentValidationService.validate(agentName, source);
             });
+            (_a = swarm.policies) === null || _a === void 0 ? void 0 : _a.forEach(function (policyName) {
+                return _this.policyValidationService.validate(policyName, source);
+            });
             return {};
         });
     }
@@ -7306,6 +7779,7 @@ var StoragePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 storageName: storageName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 stateName: "",
@@ -7341,6 +7815,7 @@ var StoragePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 storageName: storageName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 stateName: "",
@@ -7376,6 +7851,7 @@ var StoragePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 storageName: storageName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 stateName: "",
@@ -7412,6 +7888,7 @@ var StoragePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 storageName: storageName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 stateName: "",
@@ -7447,6 +7924,7 @@ var StoragePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 storageName: storageName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 stateName: "",
@@ -7481,6 +7959,7 @@ var StoragePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 storageName: storageName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 stateName: "",
@@ -7516,6 +7995,7 @@ var StoragePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 storageName: storageName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 stateName: "",
@@ -8127,6 +8607,7 @@ var StatePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 stateName: stateName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 storageName: "",
@@ -8163,6 +8644,7 @@ var StatePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 stateName: stateName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 storageName: "",
@@ -8198,6 +8680,7 @@ var StatePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 stateName: stateName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 storageName: "",
@@ -8234,6 +8717,7 @@ var StatePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: clientId,
                                 stateName: stateName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 storageName: "",
@@ -8638,6 +9122,7 @@ var DocService = /** @class */ (function () {
         this.agentValidationService = inject(TYPES.agentValidationService);
         this.swarmSchemaService = inject(TYPES.swarmSchemaService);
         this.agentSchemaService = inject(TYPES.agentSchemaService);
+        this.policySchemaService = inject(TYPES.policySchemaService);
         this.toolSchemaService = inject(TYPES.toolSchemaService);
         this.storageSchemaService = inject(TYPES.storageSchemaService);
         this.stateSchemaService = inject(TYPES.stateSchemaService);
@@ -8650,7 +9135,7 @@ var DocService = /** @class */ (function () {
          * @returns {Promise<void>}
          */
         this.writeSwarmDoc = functoolsKit.execpool(function (swarmSchema, dirName) { return __awaiter(_this, void 0, void 0, function () {
-            var result, umlSchema, umlName, umlSvg, docDescription, i, docDescription, callbackList, i;
+            var result, umlSchema, umlName, umlSvg, docDescription, i, docDescription, i, docDescription, callbackList, i;
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -8707,6 +9192,26 @@ var DocService = /** @class */ (function () {
                                 result.push("");
                             }
                         }
+                        if (swarmSchema.policies) {
+                            result.push("## Banhammer policies");
+                            result.push("");
+                            for (i = 0; i !== swarmSchema.policies.length; i++) {
+                                if (!swarmSchema.policies[i]) {
+                                    continue;
+                                }
+                                result.push("".concat(i + 1, ". ").concat(swarmSchema.policies[i]));
+                                docDescription = this.policySchemaService.get(swarmSchema.policies[i]).docDescription;
+                                if (docDescription) {
+                                    result.push("");
+                                    result.push("\t".concat(docDescription));
+                                }
+                                result.push("");
+                            }
+                            if (!swarmSchema.policies.length) {
+                                result.push("");
+                                result.push("*Empty policies*");
+                            }
+                        }
                         if (swarmSchema.callbacks) {
                             result.push("## Used callbacks");
                             result.push("");
@@ -9452,6 +9957,7 @@ var SharedStatePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: "",
                                 stateName: stateName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 storageName: "",
@@ -9486,6 +9992,7 @@ var SharedStatePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: "",
                                 stateName: stateName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 storageName: "",
@@ -9519,6 +10026,7 @@ var SharedStatePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: "",
                                 stateName: stateName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 storageName: "",
@@ -9569,6 +10077,7 @@ var SharedStoragePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: "",
                                 storageName: storageName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 stateName: "",
@@ -9603,6 +10112,7 @@ var SharedStoragePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: "",
                                 storageName: storageName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 stateName: "",
@@ -9637,6 +10147,7 @@ var SharedStoragePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: "",
                                 storageName: storageName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 stateName: "",
@@ -9672,6 +10183,7 @@ var SharedStoragePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: "",
                                 storageName: storageName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 stateName: "",
@@ -9706,6 +10218,7 @@ var SharedStoragePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: "",
                                 storageName: storageName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 stateName: "",
@@ -9739,6 +10252,7 @@ var SharedStoragePublicService = /** @class */ (function () {
                                 methodName: methodName,
                                 clientId: "",
                                 storageName: storageName,
+                                policyName: "",
                                 agentName: "",
                                 swarmName: "",
                                 stateName: "",
@@ -10236,68 +10750,818 @@ var PerfService = /** @class */ (function () {
     return PerfService;
 }());
 
-{
-    provide(TYPES.docService, function () { return new DocService(); });
-    provide(TYPES.busService, function () { return new BusService(); });
-    provide(TYPES.perfService, function () { return new PerfService(); });
-    provide(TYPES.loggerService, function () { return new LoggerService(); });
-}
-{
-    provide(TYPES.methodContextService, function () { return new MethodContextService(); });
-    provide(TYPES.executionContextService, function () { return new ExecutionContextService(); });
-}
-{
-    provide(TYPES.agentConnectionService, function () { return new AgentConnectionService(); });
-    provide(TYPES.historyConnectionService, function () { return new HistoryConnectionService(); });
-    provide(TYPES.swarmConnectionService, function () { return new SwarmConnectionService(); });
-    provide(TYPES.sessionConnectionService, function () { return new SessionConnectionService(); });
-    provide(TYPES.storageConnectionService, function () { return new StorageConnectionService(); });
-    provide(TYPES.sharedStorageConnectionService, function () { return new SharedStorageConnectionService(); });
-    provide(TYPES.stateConnectionService, function () { return new StateConnectionService(); });
-    provide(TYPES.sharedStateConnectionService, function () { return new SharedStateConnectionService(); });
-}
-{
-    provide(TYPES.agentSchemaService, function () { return new AgentSchemaService(); });
-    provide(TYPES.toolSchemaService, function () { return new ToolSchemaService(); });
-    provide(TYPES.swarmSchemaService, function () { return new SwarmSchemaService(); });
-    provide(TYPES.completionSchemaService, function () { return new CompletionSchemaService(); });
-    provide(TYPES.embeddingSchemaService, function () { return new EmbeddingSchemaService(); });
-    provide(TYPES.storageSchemaService, function () { return new StorageSchemaService(); });
-    provide(TYPES.stateSchemaService, function () { return new StateSchemaService(); });
-    provide(TYPES.memorySchemaService, function () { return new MemorySchemaService(); });
-}
-{
-    provide(TYPES.agentPublicService, function () { return new AgentPublicService(); });
-    provide(TYPES.historyPublicService, function () { return new HistoryPublicService(); });
-    provide(TYPES.sessionPublicService, function () { return new SessionPublicService(); });
-    provide(TYPES.swarmPublicService, function () { return new SwarmPublicService(); });
-    provide(TYPES.storagePublicService, function () { return new StoragePublicService(); });
-    provide(TYPES.sharedStoragePublicService, function () { return new SharedStoragePublicService(); });
-    provide(TYPES.statePublicService, function () { return new StatePublicService(); });
-    provide(TYPES.sharedStatePublicService, function () { return new SharedStatePublicService(); });
-}
-{
-    provide(TYPES.swarmMetaService, function () { return new SwarmMetaService(); });
-    provide(TYPES.agentMetaService, function () { return new AgentMetaService(); });
-}
-{
-    provide(TYPES.agentValidationService, function () { return new AgentValidationService(); });
-    provide(TYPES.completionValidationService, function () { return new CompletionValidationService(); });
-    provide(TYPES.sessionValidationService, function () { return new SessionValidationService(); });
-    provide(TYPES.swarmValidationService, function () { return new SwarmValidationService(); });
-    provide(TYPES.toolValidationService, function () { return new ToolValidationService(); });
-    provide(TYPES.storageValidationService, function () { return new StorageValidationService(); });
-    provide(TYPES.embeddingValidationService, function () { return new EmbeddingValidationService(); });
-}
-
-var baseServices = {
-    docService: inject(TYPES.docService),
-    busService: inject(TYPES.busService),
-    perfService: inject(TYPES.perfService),
-    loggerService: inject(TYPES.loggerService),
-};
-var contextServices = {
-    methodContextService: inject(TYPES.methodContextService),
+/**
+ * Service for managing policy schemas.
+ */
+var PolicySchemaService = /** @class */ (function () {
+    function PolicySchemaService() {
+        var _this = this;
+        this.loggerService = inject(TYPES.loggerService);
+        this.registry = new functoolsKit.ToolRegistry("policySchemaService");
+        /**
+         * Validation for policy schema
+         */
+        this.validateShallow = function (policySchema) {
+            GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                _this.loggerService.info("policySchemaService validateShallow", {
+                    policySchema: policySchema,
+                });
+            if (typeof policySchema.policyName !== "string") {
+                throw new Error("agent-swarm policy schema validation failed: missing policyName");
+            }
+            if (typeof policySchema.getBannedClients !== "function") {
+                throw new Error("agent-swarm policy schema validation failed: missing getBannedClients policyName=".concat(policySchema.policyName));
+            }
+        };
+        /**
+         * Registers a new policy schema.
+         * @param {PolicyName} key - The name of the policy.
+         * @param {IPolicySchema} value - The schema of the policy.
+         */
+        this.register = function (key, value) {
+            GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                _this.loggerService.info("policySchemaService register", { key: key });
+            _this.validateShallow(value);
+            _this.registry = _this.registry.register(key, value);
+        };
+        /**
+         * Retrieves an policy schema by name.
+         * @param {PolicyName} key - The name of the policy.
+         * @returns {IPolicySchema} The schema of the policy.
+         */
+        this.get = function (key) {
+            GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                _this.loggerService.info("policySchemaService get", { key: key });
+            return _this.registry.get(key);
+        };
+    }
+    return PolicySchemaService;
+}());
+
+/**
+ * Service for validating policys within the agent-swarm.
+ */
+var PolicyValidationService = /** @class */ (function () {
+    function PolicyValidationService() {
+        var _this = this;
+        this.loggerService = inject(TYPES.loggerService);
+        this._policyMap = new Map();
+        /**
+         * Adds a new policy to the validation service.
+         * @param {PolicyName} policyName - The name of the policy to add.
+         * @param {IPolicySchema} policySchema - The schema of the policy to add.
+         * @throws Will throw an error if the policy already exists.
+         */
+        this.addPolicy = function (policyName, policySchema) {
+            GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                _this.loggerService.info("policyValidationService addPolicy", {
+                    policyName: policyName,
+                    policySchema: policySchema,
+                });
+            if (_this._policyMap.has(policyName)) {
+                throw new Error("agent-swarm policy ".concat(policyName, " already exist"));
+            }
+            _this._policyMap.set(policyName, policySchema);
+        };
+        /**
+         * Validates if a policy exists in the validation service.
+         * @param {PolicyName} policyName - The name of the policy to validate.
+         * @param {string} source - The source of the validation request.
+         * @throws Will throw an error if the policy is not found.
+         */
+        this.validate = functoolsKit.memoize(function (_a) {
+            var _b = __read(_a, 1), policyName = _b[0];
+            return policyName;
+        }, function (policyName, source) {
+            GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                _this.loggerService.info("policyValidationService validate", {
+                    policyName: policyName,
+                    source: source,
+                });
+            if (!_this._policyMap.has(policyName)) {
+                throw new Error("agent-swarm policy ".concat(policyName, " not found source=").concat(source));
+            }
+            return {};
+        });
+    }
+    return PolicyValidationService;
+}());
+
+/**
+ * Service for handling public policy operations.
+ */
+var PolicyPublicService = /** @class */ (function () {
+    function PolicyPublicService() {
+        var _this = this;
+        this.loggerService = inject(TYPES.loggerService);
+        this.policyConnectionService = inject(TYPES.policyConnectionService);
+        /**
+         * Retrieves the ban message for a client in a specific swarm.
+         * @param {SwarmName} swarmName - The name of the swarm.
+         * @param {string} methodName - The name of the method.
+         * @param {string} clientId - The ID of the client.
+         * @param {PolicyName} policyName - The name of the policy.
+         * @returns {Promise<string>} The ban message.
+         */
+        this.getBanMessage = function (swarmName, methodName, clientId, policyName) { 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("policyPublicService getBanMessage", {
+                                methodName: methodName,
+                                clientId: clientId,
+                                swarmName: swarmName,
+                                policyName: policyName,
+                            });
+                        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.policyConnectionService.getBanMessage(clientId, swarmName)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                methodName: methodName,
+                                clientId: clientId,
+                                agentName: "",
+                                swarmName: swarmName,
+                                policyName: policyName,
+                                storageName: "",
+                                stateName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Validates the input for a specific policy.
+         * @param {string} incoming - The incoming data to validate.
+         * @param {SwarmName} swarmName - The name of the swarm.
+         * @param {string} methodName - The name of the method.
+         * @param {string} clientId - The ID of the client.
+         * @param {PolicyName} policyName - The name of the policy.
+         * @returns {Promise<boolean>} The result of the validation.
+         */
+        this.validateInput = function (incoming, swarmName, methodName, clientId, policyName) { 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("policyPublicService validateInput", {
+                                incoming: incoming,
+                                methodName: methodName,
+                                clientId: clientId,
+                                swarmName: swarmName,
+                                policyName: policyName,
+                            });
+                        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.policyConnectionService.validateInput(incoming, clientId, swarmName)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                methodName: methodName,
+                                clientId: clientId,
+                                agentName: "",
+                                swarmName: swarmName,
+                                policyName: policyName,
+                                storageName: "",
+                                stateName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Validates the output for a specific policy.
+         * @param {string} outgoing - The outgoing data to validate.
+         * @param {SwarmName} swarmName - The name of the swarm.
+         * @param {string} methodName - The name of the method.
+         * @param {string} clientId - The ID of the client.
+         * @param {PolicyName} policyName - The name of the policy.
+         * @returns {Promise<boolean>} The result of the validation.
+         */
+        this.validateOutput = function (outgoing, swarmName, methodName, clientId, policyName) { 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("policyPublicService validateOutput", {
+                                outgoing: outgoing,
+                                methodName: methodName,
+                                clientId: clientId,
+                                swarmName: swarmName,
+                                policyName: policyName,
+                            });
+                        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.policyConnectionService.validateOutput(outgoing, clientId, swarmName)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                methodName: methodName,
+                                clientId: clientId,
+                                agentName: "",
+                                swarmName: swarmName,
+                                policyName: policyName,
+                                storageName: "",
+                                stateName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Bans a client from a specific swarm.
+         * @param {SwarmName} swarmName - The name of the swarm.
+         * @param {string} methodName - The name of the method.
+         * @param {string} clientId - The ID of the client.
+         * @param {PolicyName} policyName - The name of the policy.
+         * @returns {Promise<void>}
+         */
+        this.banClient = function (swarmName, methodName, clientId, policyName) { 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("policyPublicService banClient", {
+                                methodName: methodName,
+                                clientId: clientId,
+                                swarmName: swarmName,
+                                policyName: policyName,
+                            });
+                        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.policyConnectionService.banClient(clientId, swarmName)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                methodName: methodName,
+                                clientId: clientId,
+                                agentName: "",
+                                swarmName: swarmName,
+                                policyName: policyName,
+                                storageName: "",
+                                stateName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Unbans a client from a specific swarm.
+         * @param {SwarmName} swarmName - The name of the swarm.
+         * @param {string} methodName - The name of the method.
+         * @param {string} clientId - The ID of the client.
+         * @param {PolicyName} policyName - The name of the policy.
+         * @returns {Promise<void>}
+         */
+        this.unbanClient = function (swarmName, methodName, clientId, policyName) { 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("policyPublicService unbanClient", {
+                                methodName: methodName,
+                                clientId: clientId,
+                                swarmName: swarmName,
+                                policyName: policyName,
+                            });
+                        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.policyConnectionService.unbanClient(clientId, swarmName)];
+                                        case 1: return [2 /*return*/, _a.sent()];
+                                    }
+                                });
+                            }); }, {
+                                methodName: methodName,
+                                clientId: clientId,
+                                agentName: "",
+                                swarmName: swarmName,
+                                policyName: policyName,
+                                storageName: "",
+                                stateName: "",
+                            })];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+    }
+    return PolicyPublicService;
+}());
+
+var BAN_NEED_FETCH = Symbol("ban-need-fetch");
+/**
+ * Class representing a client policy.
+ * @implements {IPolicy}
+ */
+var ClientPolicy = /** @class */ (function () {
+    /**
+     * Creates an instance of ClientPolicy.
+     * @param {IPolicyParams} params - The policy parameters.
+     */
+    function ClientPolicy(params) {
+        var _a;
+        this.params = params;
+        this._banSet = BAN_NEED_FETCH;
+        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+            this.params.logger.debug("ClientPolicy policyName=".concat(this.params.policyName, " CTOR"), {
+                params: params,
+            });
+        if ((_a = this.params.callbacks) === null || _a === void 0 ? void 0 : _a.onInit) {
+            this.params.callbacks.onInit(params.policyName);
+        }
+    }
+    /**
+     * Gets the ban message for a client.
+     * @param {SessionId} clientId - The client ID.
+     * @param {SwarmName} swarmName - The swarm name.
+     * @returns {Promise<string>} The ban message.
+     */
+    ClientPolicy.prototype.getBanMessage = function (clientId, swarmName) {
+        return __awaiter(this, void 0, void 0, function () {
+            var banMessage;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            this.params.logger.debug("ClientPolicy policyName=".concat(this.params.policyName, " getBanMessage"), {
+                                clientId: clientId,
+                                swarmName: swarmName,
+                            });
+                        if (!this.params.getBanMessage) {
+                            return [2 /*return*/, this.params.banMessage];
+                        }
+                        banMessage = null;
+                        return [4 /*yield*/, this.params.getBanMessage(clientId, this.params.policyName, swarmName)];
+                    case 1:
+                        if ((banMessage = _a.sent())) {
+                            return [2 /*return*/, banMessage];
+                        }
+                        return [2 /*return*/, this.params.banMessage];
+                }
+            });
+        });
+    };
+    /**
+     * Validates the input from a client.
+     * @param {string} incoming - The incoming message.
+     * @param {SessionId} clientId - The client ID.
+     * @param {SwarmName} swarmName - The swarm name.
+     * @returns {Promise<boolean>} Whether the input is valid.
+     */
+    ClientPolicy.prototype.validateInput = function (incoming, clientId, swarmName) {
+        return __awaiter(this, void 0, void 0, function () {
+            var _a, _b;
+            var _c;
+            return __generator(this, function (_d) {
+                switch (_d.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            this.params.logger.debug("ClientPolicy policyName=".concat(this.params.policyName, " validateInput"), {
+                                incoming: incoming,
+                                clientId: clientId,
+                                swarmName: swarmName,
+                            });
+                        if ((_c = this.params.callbacks) === null || _c === void 0 ? void 0 : _c.onValidateInput) {
+                            this.params.callbacks.onValidateInput(incoming, clientId, swarmName, this.params.policyName);
+                        }
+                        return [4 /*yield*/, this.params.bus.emit(clientId, {
+                                type: "validate-input",
+                                source: "policy-bus",
+                                input: {
+                                    incoming: incoming,
+                                },
+                                output: {},
+                                context: {
+                                    policyName: this.params.policyName,
+                                    swarmName: swarmName,
+                                },
+                                clientId: clientId,
+                            })];
+                    case 1:
+                        _d.sent();
+                        if (!(this._banSet === BAN_NEED_FETCH)) return [3 /*break*/, 3];
+                        _a = this;
+                        _b = Set.bind;
+                        return [4 /*yield*/, this.params.getBannedClients(this.params.policyName, swarmName)];
+                    case 2:
+                        _a._banSet = new (_b.apply(Set, [void 0, _d.sent()]))();
+                        _d.label = 3;
+                    case 3:
+                        if (this._banSet.has(clientId)) {
+                            return [2 /*return*/, false];
+                        }
+                        if (!this.params.validateInput) {
+                            return [2 /*return*/, true];
+                        }
+                        return [4 /*yield*/, this.params.validateInput(incoming, clientId, this.params.policyName, swarmName)];
+                    case 4:
+                        if (_d.sent()) {
+                            return [2 /*return*/, true];
+                        }
+                        return [4 /*yield*/, this.banClient(clientId, swarmName)];
+                    case 5:
+                        _d.sent();
+                        return [2 /*return*/, false];
+                }
+            });
+        });
+    };
+    /**
+     * Validates the output to a client.
+     * @param {string} outgoing - The outgoing message.
+     * @param {SessionId} clientId - The client ID.
+     * @param {SwarmName} swarmName - The swarm name.
+     * @returns {Promise<boolean>} Whether the output is valid.
+     */
+    ClientPolicy.prototype.validateOutput = function (outgoing, clientId, swarmName) {
+        return __awaiter(this, void 0, void 0, function () {
+            var _a, _b;
+            var _c;
+            return __generator(this, function (_d) {
+                switch (_d.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            this.params.logger.debug("ClientPolicy policyName=".concat(this.params.policyName, " validateOutput"), {
+                                clientId: clientId,
+                                swarmName: swarmName,
+                                outgoing: outgoing,
+                            });
+                        if ((_c = this.params.callbacks) === null || _c === void 0 ? void 0 : _c.onValidateOutput) {
+                            this.params.callbacks.onValidateOutput(outgoing, clientId, swarmName, this.params.policyName);
+                        }
+                        return [4 /*yield*/, this.params.bus.emit(clientId, {
+                                type: "validate-output",
+                                source: "policy-bus",
+                                input: {
+                                    outgoing: outgoing,
+                                },
+                                output: {},
+                                context: {
+                                    policyName: this.params.policyName,
+                                    swarmName: swarmName,
+                                },
+                                clientId: clientId,
+                            })];
+                    case 1:
+                        _d.sent();
+                        if (!(this._banSet === BAN_NEED_FETCH)) return [3 /*break*/, 3];
+                        _a = this;
+                        _b = Set.bind;
+                        return [4 /*yield*/, this.params.getBannedClients(this.params.policyName, swarmName)];
+                    case 2:
+                        _a._banSet = new (_b.apply(Set, [void 0, _d.sent()]))();
+                        _d.label = 3;
+                    case 3:
+                        if (this._banSet.has(clientId)) {
+                            return [2 /*return*/, false];
+                        }
+                        if (!this.params.validateOutput) {
+                            return [2 /*return*/, true];
+                        }
+                        return [4 /*yield*/, this.params.validateOutput(outgoing, clientId, this.params.policyName, swarmName)];
+                    case 4:
+                        if (_d.sent()) {
+                            return [2 /*return*/, true];
+                        }
+                        return [4 /*yield*/, this.banClient(clientId, swarmName)];
+                    case 5:
+                        _d.sent();
+                        return [2 /*return*/, false];
+                }
+            });
+        });
+    };
+    /**
+     * Bans a client.
+     * @param {SessionId} clientId - The client ID.
+     * @param {SwarmName} swarmName - The swarm name.
+     * @returns {Promise<void>}
+     */
+    ClientPolicy.prototype.banClient = function (clientId, swarmName) {
+        return __awaiter(this, void 0, void 0, function () {
+            var _a, _b;
+            var _c;
+            return __generator(this, function (_d) {
+                switch (_d.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            this.params.logger.debug("ClientPolicy policyName=".concat(this.params.policyName, " banClient"), {
+                                clientId: clientId,
+                                swarmName: swarmName,
+                            });
+                        if ((_c = this.params.callbacks) === null || _c === void 0 ? void 0 : _c.onBanClient) {
+                            this.params.callbacks.onBanClient(clientId, swarmName, this.params.policyName);
+                        }
+                        return [4 /*yield*/, this.params.bus.emit(clientId, {
+                                type: "ban-client",
+                                source: "policy-bus",
+                                input: {},
+                                output: {},
+                                context: {
+                                    policyName: this.params.policyName,
+                                    swarmName: swarmName,
+                                },
+                                clientId: clientId,
+                            })];
+                    case 1:
+                        _d.sent();
+                        if (!(this._banSet === BAN_NEED_FETCH)) return [3 /*break*/, 3];
+                        _a = this;
+                        _b = Set.bind;
+                        return [4 /*yield*/, this.params.getBannedClients(this.params.policyName, swarmName)];
+                    case 2:
+                        _a._banSet = new (_b.apply(Set, [void 0, _d.sent()]))();
+                        _d.label = 3;
+                    case 3:
+                        if (this._banSet.has(clientId)) {
+                            return [2 /*return*/];
+                        }
+                        this._banSet = new Set(this._banSet).add(clientId);
+                        if (!this.params.setBannedClients) return [3 /*break*/, 5];
+                        return [4 /*yield*/, this.params.setBannedClients(__spreadArray([], __read(this._banSet), false), this.params.policyName, swarmName)];
+                    case 4:
+                        _d.sent();
+                        _d.label = 5;
+                    case 5: return [2 /*return*/];
+                }
+            });
+        });
+    };
+    /**
+     * Unbans a client.
+     * @param {SessionId} clientId - The client ID.
+     * @param {SwarmName} swarmName - The swarm name.
+     * @returns {Promise<void>}
+     */
+    ClientPolicy.prototype.unbanClient = function (clientId, swarmName) {
+        return __awaiter(this, void 0, void 0, function () {
+            var _a, _b, banSet;
+            var _c;
+            return __generator(this, function (_d) {
+                switch (_d.label) {
+                    case 0:
+                        GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
+                            this.params.logger.debug("ClientPolicy policyName=".concat(this.params.policyName, " unbanClient"), {
+                                clientId: clientId,
+                                swarmName: swarmName,
+                            });
+                        if ((_c = this.params.callbacks) === null || _c === void 0 ? void 0 : _c.onUnbanClient) {
+                            this.params.callbacks.onUnbanClient(clientId, swarmName, this.params.policyName);
+                        }
+                        return [4 /*yield*/, this.params.bus.emit(clientId, {
+                                type: "unban-client",
+                                source: "policy-bus",
+                                input: {},
+                                output: {},
+                                context: {
+                                    policyName: this.params.policyName,
+                                    swarmName: swarmName,
+                                },
+                                clientId: clientId,
+                            })];
+                    case 1:
+                        _d.sent();
+                        if (!(this._banSet === BAN_NEED_FETCH)) return [3 /*break*/, 3];
+                        _a = this;
+                        _b = Set.bind;
+                        return [4 /*yield*/, this.params.getBannedClients(this.params.policyName, swarmName)];
+                    case 2:
+                        _a._banSet = new (_b.apply(Set, [void 0, _d.sent()]))();
+                        _d.label = 3;
+                    case 3:
+                        if (!this._banSet.has(clientId)) {
+                            return [2 /*return*/];
+                        }
+                        {
+                            banSet = new Set(this._banSet);
+                            banSet.delete(clientId);
+                            this._banSet = banSet;
+                        }
+                        if (!this.params.setBannedClients) return [3 /*break*/, 5];
+                        return [4 /*yield*/, this.params.setBannedClients(__spreadArray([], __read(this._banSet), false), this.params.policyName, swarmName)];
+                    case 4:
+                        _d.sent();
+                        _d.label = 5;
+                    case 5: return [2 /*return*/];
+                }
+            });
+        });
+    };
+    return ClientPolicy;
+}());
+
+/**
+ * Service for managing policy connections.
+ * @implements {IPolicy}
+ */
+var PolicyConnectionService = /** @class */ (function () {
+    function PolicyConnectionService() {
+        var _this = this;
+        this.loggerService = inject(TYPES.loggerService);
+        this.busService = inject(TYPES.busService);
+        this.methodContextService = inject(TYPES.methodContextService);
+        this.policySchemaService = inject(TYPES.policySchemaService);
+        /**
+         * Retrieves a policy based on the policy name.
+         * @param {PolicyName} policyName - The name of the policy.
+         * @returns {ClientPolicy} The client policy.
+         */
+        this.getPolicy = functoolsKit.memoize(function (_a) {
+            var _b = __read(_a, 1), policyName = _b[0];
+            return "".concat(policyName);
+        }, function (policyName) {
+            var schema = _this.policySchemaService.get(policyName);
+            return new ClientPolicy(__assign({ policyName: policyName, bus: _this.busService, logger: _this.loggerService }, schema));
+        });
+        /**
+         * Retrieves the ban message for a client in a swarm.
+         * @param {SessionId} clientId - The ID of the client.
+         * @param {SwarmName} swarmName - The name of the swarm.
+         * @returns {Promise<string>} The ban message.
+         */
+        this.getBanMessage = function (clientId, swarmName) { 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("policyConnectionService getBanMessage", {
+                                clientId: clientId,
+                                swarmName: swarmName,
+                            });
+                        return [4 /*yield*/, this.getPolicy(this.methodContextService.context.policyName).getBanMessage(clientId, swarmName)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Validates the input for a client in a swarm.
+         * @param {string} incoming - The incoming input.
+         * @param {SessionId} clientId - The ID of the client.
+         * @param {SwarmName} swarmName - The name of the swarm.
+         * @returns {Promise<boolean>} Whether the input is valid.
+         */
+        this.validateInput = function (incoming, clientId, swarmName) { 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("policyConnectionService validateInput", {
+                                incoming: incoming,
+                                clientId: clientId,
+                                swarmName: swarmName,
+                            });
+                        return [4 /*yield*/, this.getPolicy(this.methodContextService.context.policyName).validateInput(incoming, clientId, swarmName)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Validates the output for a client in a swarm.
+         * @param {string} outgoing - The outgoing output.
+         * @param {SessionId} clientId - The ID of the client.
+         * @param {SwarmName} swarmName - The name of the swarm.
+         * @returns {Promise<boolean>} Whether the output is valid.
+         */
+        this.validateOutput = function (outgoing, clientId, swarmName) { 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("policyConnectionService validateOutput", {
+                                outgoing: outgoing,
+                                clientId: clientId,
+                                swarmName: swarmName,
+                            });
+                        return [4 /*yield*/, this.getPolicy(this.methodContextService.context.policyName).validateOutput(outgoing, clientId, swarmName)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Bans a client from a swarm.
+         * @param {SessionId} clientId - The ID of the client.
+         * @param {SwarmName} swarmName - The name of the swarm.
+         * @returns {Promise<void>}
+         */
+        this.banClient = function (clientId, swarmName) { 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("policyConnectionService banClient", {
+                                clientId: clientId,
+                                swarmName: swarmName,
+                            });
+                        return [4 /*yield*/, this.getPolicy(this.methodContextService.context.policyName).banClient(clientId, swarmName)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+        /**
+         * Unbans a client from a swarm.
+         * @param {SessionId} clientId - The ID of the client.
+         * @param {SwarmName} swarmName - The name of the swarm.
+         * @returns {Promise<void>}
+         */
+        this.unbanClient = function (clientId, swarmName) { 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("policyConnectionService unbanClient", {
+                                clientId: clientId,
+                                swarmName: swarmName,
+                            });
+                        return [4 /*yield*/, this.getPolicy(this.methodContextService.context.policyName).unbanClient(clientId, swarmName)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); };
+    }
+    return PolicyConnectionService;
+}());
+
+{
+    provide(TYPES.docService, function () { return new DocService(); });
+    provide(TYPES.busService, function () { return new BusService(); });
+    provide(TYPES.perfService, function () { return new PerfService(); });
+    provide(TYPES.loggerService, function () { return new LoggerService(); });
+}
+{
+    provide(TYPES.methodContextService, function () { return new MethodContextService(); });
+    provide(TYPES.executionContextService, function () { return new ExecutionContextService(); });
+}
+{
+    provide(TYPES.agentConnectionService, function () { return new AgentConnectionService(); });
+    provide(TYPES.historyConnectionService, function () { return new HistoryConnectionService(); });
+    provide(TYPES.swarmConnectionService, function () { return new SwarmConnectionService(); });
+    provide(TYPES.sessionConnectionService, function () { return new SessionConnectionService(); });
+    provide(TYPES.storageConnectionService, function () { return new StorageConnectionService(); });
+    provide(TYPES.sharedStorageConnectionService, function () { return new SharedStorageConnectionService(); });
+    provide(TYPES.stateConnectionService, function () { return new StateConnectionService(); });
+    provide(TYPES.sharedStateConnectionService, function () { return new SharedStateConnectionService(); });
+    provide(TYPES.policyConnectionService, function () { return new PolicyConnectionService(); });
+}
+{
+    provide(TYPES.agentSchemaService, function () { return new AgentSchemaService(); });
+    provide(TYPES.toolSchemaService, function () { return new ToolSchemaService(); });
+    provide(TYPES.swarmSchemaService, function () { return new SwarmSchemaService(); });
+    provide(TYPES.completionSchemaService, function () { return new CompletionSchemaService(); });
+    provide(TYPES.embeddingSchemaService, function () { return new EmbeddingSchemaService(); });
+    provide(TYPES.storageSchemaService, function () { return new StorageSchemaService(); });
+    provide(TYPES.stateSchemaService, function () { return new StateSchemaService(); });
+    provide(TYPES.memorySchemaService, function () { return new MemorySchemaService(); });
+    provide(TYPES.policySchemaService, function () { return new PolicySchemaService(); });
+}
+{
+    provide(TYPES.agentPublicService, function () { return new AgentPublicService(); });
+    provide(TYPES.historyPublicService, function () { return new HistoryPublicService(); });
+    provide(TYPES.sessionPublicService, function () { return new SessionPublicService(); });
+    provide(TYPES.swarmPublicService, function () { return new SwarmPublicService(); });
+    provide(TYPES.storagePublicService, function () { return new StoragePublicService(); });
+    provide(TYPES.sharedStoragePublicService, function () { return new SharedStoragePublicService(); });
+    provide(TYPES.statePublicService, function () { return new StatePublicService(); });
+    provide(TYPES.sharedStatePublicService, function () { return new SharedStatePublicService(); });
+    provide(TYPES.policyPublicService, function () { return new PolicyPublicService(); });
+}
+{
+    provide(TYPES.swarmMetaService, function () { return new SwarmMetaService(); });
+    provide(TYPES.agentMetaService, function () { return new AgentMetaService(); });
+}
+{
+    provide(TYPES.agentValidationService, function () { return new AgentValidationService(); });
+    provide(TYPES.completionValidationService, function () { return new CompletionValidationService(); });
+    provide(TYPES.sessionValidationService, function () { return new SessionValidationService(); });
+    provide(TYPES.swarmValidationService, function () { return new SwarmValidationService(); });
+    provide(TYPES.toolValidationService, function () { return new ToolValidationService(); });
+    provide(TYPES.storageValidationService, function () { return new StorageValidationService(); });
+    provide(TYPES.embeddingValidationService, function () { return new EmbeddingValidationService(); });
+    provide(TYPES.policyValidationService, function () { return new PolicyValidationService(); });
+}
+
+var baseServices = {
+    docService: inject(TYPES.docService),
+    busService: inject(TYPES.busService),
+    perfService: inject(TYPES.perfService),
+    loggerService: inject(TYPES.loggerService),
+};
+var contextServices = {
+    methodContextService: inject(TYPES.methodContextService),
     executionContextService: inject(TYPES.executionContextService),
 };
 var connectionServices = {
@@ -10309,6 +11573,7 @@ var connectionServices = {
     sharedStorageConnectionService: inject(TYPES.sharedStorageConnectionService),
     stateConnectionService: inject(TYPES.stateConnectionService),
     sharedStateConnectionService: inject(TYPES.sharedStateConnectionService),
+    policyConnectionService: inject(TYPES.policyConnectionService),
 };
 var schemaServices = {
     agentSchemaService: inject(TYPES.agentSchemaService),
@@ -10319,6 +11584,7 @@ var schemaServices = {
     storageSchemaService: inject(TYPES.storageSchemaService),
     stateSchemaService: inject(TYPES.stateSchemaService),
     memorySchemaService: inject(TYPES.memorySchemaService),
+    policySchemaService: inject(TYPES.policySchemaService),
 };
 var publicServices = {
     agentPublicService: inject(TYPES.agentPublicService),
@@ -10329,6 +11595,7 @@ var publicServices = {
     sharedStoragePublicService: inject(TYPES.sharedStoragePublicService),
     statePublicService: inject(TYPES.statePublicService),
     sharedStatePublicService: inject(TYPES.sharedStatePublicService),
+    policyPublicService: inject(TYPES.policyPublicService),
 };
 var metaServices = {
     agentMetaService: inject(TYPES.agentMetaService),
@@ -10342,12 +11609,13 @@ var validationServices = {
     completionValidationService: inject(TYPES.completionValidationService),
     storageValidationService: inject(TYPES.storageValidationService),
     embeddingValidationService: inject(TYPES.embeddingValidationService),
+    policyValidationService: inject(TYPES.policyValidationService),
 };
 var swarm = __assign(__assign(__assign(__assign(__assign(__assign(__assign({}, baseServices), contextServices), connectionServices), schemaServices), publicServices), metaServices), validationServices);
 init();
 var swarm$1 = swarm;
 
-var METHOD_NAME$Q = "cli.dumpDocs";
+var METHOD_NAME$R = "cli.dumpDocs";
 /**
  * Dumps the documentation for the agents and swarms.
  *
@@ -10355,10 +11623,10 @@ var METHOD_NAME$Q = "cli.dumpDocs";
  * @param {function} [PlantUML] - An optional function to process PlantUML diagrams.
  * @returns {Promise<void>} - A promise that resolves when the documentation has been dumped.
  */
-var dumpDocs = function (dirName, PlantUML) {
+var dumpDocs = beginContext(function (dirName, PlantUML) {
     if (dirName === void 0) { dirName = "./docs/chat"; }
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$Q, {
+        swarm$1.loggerService.log(METHOD_NAME$R, {
             dirName: dirName,
         });
     if (PlantUML) {
@@ -10369,12 +11637,12 @@ var dumpDocs = function (dirName, PlantUML) {
     swarm$1.agentValidationService
         .getAgentList()
         .forEach(function (agentName) {
-        return swarm$1.agentValidationService.validate(agentName, METHOD_NAME$Q);
+        return swarm$1.agentValidationService.validate(agentName, METHOD_NAME$R);
     });
     swarm$1.swarmValidationService
         .getSwarmList()
         .forEach(function (swarmName) {
-        return swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$Q);
+        return swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$R);
     });
     swarm$1.agentValidationService.getAgentList().forEach(function (agentName) {
         var dependsOn = swarm$1.agentSchemaService.get(agentName).dependsOn;
@@ -10383,48 +11651,48 @@ var dumpDocs = function (dirName, PlantUML) {
         }
     });
     return swarm$1.docService.dumpDocs(dirName);
-};
+});
 
-var METHOD_NAME$P = "cli.dumpAgent";
+var METHOD_NAME$Q = "cli.dumpAgent";
 /**
  * Dumps the agent information into PlantUML format.
  *
  * @param {SwarmName} swarmName - The name of the swarm to be dumped.
  * @returns {string} The UML representation of the swarm.
  */
-var dumpAgent = function (agentName, _a) {
+var dumpAgent = beginContext(function (agentName, _a) {
     var _b = _a === void 0 ? {} : _a, _c = _b.withSubtree, withSubtree = _c === void 0 ? false : _c;
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$P, {
+        swarm$1.loggerService.log(METHOD_NAME$Q, {
             agentName: agentName,
         });
-    swarm$1.agentValidationService.validate(agentName, METHOD_NAME$P);
+    swarm$1.agentValidationService.validate(agentName, METHOD_NAME$Q);
     return swarm$1.agentMetaService.toUML(agentName, withSubtree);
-};
+});
 
-var METHOD_NAME$O = "cli.dumpSwarm";
+var METHOD_NAME$P = "cli.dumpSwarm";
 /**
  * Dumps the swarm information into PlantUML format.
  *
  * @param {SwarmName} swarmName - The name of the swarm to be dumped.
  * @returns {string} The UML representation of the swarm.
  */
-var dumpSwarm = function (swarmName) {
+var dumpSwarm = beginContext(function (swarmName) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$O, {
+        swarm$1.loggerService.log(METHOD_NAME$P, {
             swarmName: swarmName,
         });
-    swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$O);
+    swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$P);
     return swarm$1.swarmMetaService.toUML(swarmName);
-};
+});
 
-var METHOD_NAME$N = "cli.dumpPerfomance";
+var METHOD_NAME$O = "cli.dumpPerfomance";
 var METHOD_NAME_INTERNAL$1 = "cli.dumpPerfomance.internal";
 var METHOD_NAME_INTERVAL = "cli.dumpPerfomance.interval";
 /**
  * The internal HOF for handling the performance dump
  */
-var dumpPerfomanceInternal = functoolsKit.trycatch(function () {
+var dumpPerfomanceInternal = beginContext(function () {
     var args_1 = [];
     for (var _i = 0; _i < arguments.length; _i++) {
         args_1[_i] = arguments[_i];
@@ -10441,12 +11709,6 @@ var dumpPerfomanceInternal = functoolsKit.trycatch(function () {
             }
         });
     });
-}, {
-    fallback: function (error) {
-        swarm$1.loggerService.log("agent-swarm dumpPerfomanceInternal exception error=".concat(functoolsKit.getErrorMessage(error)), {
-            errorData: functoolsKit.errorData(error),
-        });
-    },
 });
 /**
  * Dumps the performance data using the swarm's document service.
@@ -10465,7 +11727,7 @@ var dumpPerfomance = function () {
         return __generator(this, function (_a) {
             switch (_a.label) {
                 case 0:
-                    GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG && swarm$1.loggerService.log(METHOD_NAME$N);
+                    GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG && swarm$1.loggerService.log(METHOD_NAME$O);
                     return [4 /*yield*/, dumpPerfomanceInternal(dirName)];
                 case 1:
                     _a.sent();
@@ -10481,15 +11743,15 @@ var dumpPerfomance = function () {
  * @param {string} [dirName="./logs/meta"] - The directory name where the performance data will be dumped.
  * @param {number} [interval=30000] - The interval in milliseconds at which to run the dumpPerfomance function.
  */
-dumpPerfomance.runInterval = function (dirName, interval) {
+dumpPerfomance.runInterval = beginContext(function (dirName, interval) {
     if (dirName === void 0) { dirName = "./logs/meta"; }
     if (interval === void 0) { interval = 30000; }
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
         swarm$1.loggerService.log(METHOD_NAME_INTERVAL);
     return functoolsKit.Source.fromInterval(interval).connect(function () { return dumpPerfomance(dirName); });
-};
+});
 
-var validateClientId$f = function (clientId) {
+var validateClientId$h = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -10503,20 +11765,20 @@ var validateClientId$f = function (clientId) {
  * @param {string} clientId - The ID of the client to subscribe to events for.
  * @param {function} fn - The callback function to handle the event.
  */
-var listenExecutionEvent = function (clientId, fn) {
+var listenExecutionEvent = beginContext(function (clientId, fn) {
     swarm$1.loggerService.log("middleware listenExecutionEvent", {
         clientId: clientId,
     });
-    validateClientId$f(clientId);
+    validateClientId$h(clientId);
     return swarm$1.busService.subscribe(clientId, "execution-bus", functoolsKit.queued(function (e) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
         switch (_a.label) {
             case 0: return [4 /*yield*/, fn(e)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     }); }); }));
-};
+});
 
-var METHOD_NAME$M = "cli.dumpClientPerformance";
+var METHOD_NAME$N = "cli.dumpClientPerformance";
 var METHOD_NAME_INTERNAL = "cli.dumpClientPerformance.internal";
 var METHOD_NAME_EXECUTE = "cli.dumpClientPerformance.execute";
 /**
@@ -10526,7 +11788,7 @@ var METHOD_NAME_EXECUTE = "cli.dumpClientPerformance.execute";
  * @param {string} [dirName="./logs/client"] - The directory name where the performance data will be dumped.
  * @returns {Promise<void>} A promise that resolves when the performance data has been dumped.
  */
-var dumpClientPerformanceInternal = functoolsKit.trycatch(function (clientId_1) {
+var dumpClientPerformanceInternal = beginContext(function (clientId_1) {
     var args_1 = [];
     for (var _i = 1; _i < arguments.length; _i++) {
         args_1[_i - 1] = arguments[_i];
@@ -10543,12 +11805,6 @@ var dumpClientPerformanceInternal = functoolsKit.trycatch(function (clientId_1)
             }
         });
     });
-}, {
-    fallback: function (error) {
-        swarm$1.loggerService.log("agent-swarm dumpClientPerformanceInternal exception error=".concat(functoolsKit.getErrorMessage(error)), {
-            errorData: functoolsKit.errorData(error),
-        });
-    },
 });
 /**
  * Dumps the performance data using the swarm's document service.
@@ -10568,7 +11824,7 @@ var dumpClientPerformance = function (clientId_1) {
         return __generator(this, function (_a) {
             switch (_a.label) {
                 case 0:
-                    GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG && swarm$1.loggerService.log(METHOD_NAME$M);
+                    GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG && swarm$1.loggerService.log(METHOD_NAME$N);
                     return [4 /*yield*/, dumpClientPerformanceInternal(clientId, dirName)];
                 case 1:
                     _a.sent();
@@ -10584,7 +11840,7 @@ var dumpClientPerformance = function (clientId_1) {
  * @param {string} [dirName="./logs/client"] - The directory name where the performance data will be dumped.
  * @returns {Promise<void>} A promise that resolves when the listener has been set up.
  */
-dumpClientPerformance.runAfterExecute = function () {
+dumpClientPerformance.runAfterExecute = beginContext(function () {
     var args_1 = [];
     for (var _i = 0; _i < arguments.length; _i++) {
         args_1[_i] = arguments[_i];
@@ -10610,26 +11866,26 @@ dumpClientPerformance.runAfterExecute = function () {
                 }); })];
         });
     });
-};
+});
 
-var METHOD_NAME$L = "function.setup.addAgent";
+var METHOD_NAME$M = "function.setup.addAgent";
 /**
  * Adds a new agent to the agent registry. The swarm takes only those agents which was registered
  *
  * @param {IAgentSchema} agentSchema - The schema of the agent to be added.
  * @returns {string} The name of the added agent.
  */
-var addAgent = function (agentSchema) {
+var addAgent = beginContext(function (agentSchema) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$L, {
+        swarm$1.loggerService.log(METHOD_NAME$M, {
             agentSchema: agentSchema,
         });
     swarm$1.agentValidationService.addAgent(agentSchema.agentName, agentSchema);
     swarm$1.agentSchemaService.register(agentSchema.agentName, agentSchema);
     return agentSchema.agentName;
-};
+});
 
-var METHOD_NAME$K = "function.setup.addCompletion";
+var METHOD_NAME$L = "function.setup.addCompletion";
 /**
  * Adds a completion engine for agents. Agents could use different models and
  * framewords for completion like: mock, gpt4all, ollama, openai
@@ -10637,34 +11893,34 @@ var METHOD_NAME$K = "function.setup.addCompletion";
  * @param {ICompletionSchema} completionSchema - The completion schema to be added.
  * @returns {string} The name of the completion that was added.
  */
-var addCompletion = function (completionSchema) {
+var addCompletion = beginContext(function (completionSchema) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$K, {
+        swarm$1.loggerService.log(METHOD_NAME$L, {
             completionSchema: completionSchema,
         });
     swarm$1.completionValidationService.addCompletion(completionSchema.completionName);
     swarm$1.completionSchemaService.register(completionSchema.completionName, completionSchema);
     return completionSchema.completionName;
-};
+});
 
-var METHOD_NAME$J = "function.setup.addSwarm";
+var METHOD_NAME$K = "function.setup.addSwarm";
 /**
  * Adds a new swarm to the system. The swarm is a root for starting client session
  *
  * @param {ISwarmSchema} swarmSchema - The schema of the swarm to be added.
  * @returns {string} The name of the added swarm.
  */
-var addSwarm = function (swarmSchema) {
+var addSwarm = beginContext(function (swarmSchema) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$J, {
+        swarm$1.loggerService.log(METHOD_NAME$K, {
             swarmSchema: swarmSchema,
         });
     swarm$1.swarmValidationService.addSwarm(swarmSchema.swarmName, swarmSchema);
     swarm$1.swarmSchemaService.register(swarmSchema.swarmName, swarmSchema);
     return swarmSchema.swarmName;
-};
+});
 
-var METHOD_NAME$I = "function.setup.addTool";
+var METHOD_NAME$J = "function.setup.addTool";
 /**
  * Adds a new tool for agents in a swarm. Tool should be registered in `addAgent`
  * declaration
@@ -10672,26 +11928,26 @@ var METHOD_NAME$I = "function.setup.addTool";
  * @param {IAgentTool} toolSchema - The schema of the tool to be added.
  * @returns {string} The name of the tool that was added.
  */
-var addTool = function (toolSchema) {
+var addTool = beginContext(function (toolSchema) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$I, {
+        swarm$1.loggerService.log(METHOD_NAME$J, {
             toolSchema: toolSchema,
         });
     swarm$1.toolValidationService.addTool(toolSchema.toolName, toolSchema);
     swarm$1.toolSchemaService.register(toolSchema.toolName, toolSchema);
     return toolSchema.toolName;
-};
+});
 
-var METHOD_NAME$H = "function.setup.addState";
+var METHOD_NAME$I = "function.setup.addState";
 /**
  * Adds a new state to the state registry. The swarm takes only those states which was registered
  *
  * @param {IStateSchema} stateSchema - The schema of the state to be added.
  * @returns {string} The name of the added state.
  */
-var addState = function (stateSchema) {
+var addState = beginContext(function (stateSchema) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$H, {
+        swarm$1.loggerService.log(METHOD_NAME$I, {
             stateSchema: stateSchema,
         });
     swarm$1.stateSchemaService.register(stateSchema.stateName, stateSchema);
@@ -10701,35 +11957,35 @@ var addState = function (stateSchema) {
             .waitForInit();
     }
     return stateSchema.stateName;
-};
+});
 
-var METHOD_NAME$G = "function.setup.addEmbedding";
+var METHOD_NAME$H = "function.setup.addEmbedding";
 /**
  * Adds a new embedding to the embedding registry. The swarm takes only those embeddings which was registered
  *
  * @param {IEmbeddingSchema} embeddingSchema - The schema of the embedding to be added.
  * @returns {string} The name of the added embedding.
  */
-var addEmbedding = function (embeddingSchema) {
+var addEmbedding = beginContext(function (embeddingSchema) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$G, {
+        swarm$1.loggerService.log(METHOD_NAME$H, {
             embeddingSchema: embeddingSchema,
         });
     swarm$1.embeddingValidationService.addEmbedding(embeddingSchema.embeddingName, embeddingSchema);
     swarm$1.embeddingSchemaService.register(embeddingSchema.embeddingName, embeddingSchema);
     return embeddingSchema.embeddingName;
-};
+});
 
-var METHOD_NAME$F = "function.setup.addStorage";
+var METHOD_NAME$G = "function.setup.addStorage";
 /**
  * Adds a new storage to the storage registry. The swarm takes only those storages which was registered
  *
  * @param {IStorageSchema} storageSchema - The schema of the storage to be added.
  * @returns {string} The name of the added storage.
  */
-var addStorage = function (storageSchema) {
+var addStorage = beginContext(function (storageSchema) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
-        swarm$1.loggerService.log(METHOD_NAME$F, {
+        swarm$1.loggerService.log(METHOD_NAME$G, {
             storageSchema: storageSchema,
         });
     swarm$1.storageValidationService.addStorage(storageSchema.storageName, storageSchema);
@@ -10740,7 +11996,25 @@ var addStorage = function (storageSchema) {
             .waitForInit();
     }
     return storageSchema.storageName;
-};
+});
+
+var METHOD_NAME$F = "function.setup.addPolicy";
+/**
+ * Adds a new policy for agents in a swarm. Policy should be registered in `addPolicy`
+ * declaration
+ *
+ * @param {IPolicySchema} policySchema - The schema of the policy to be added.
+ * @returns {string} The name of the policy that was added.
+ */
+var addPolicy = beginContext(function (policySchema) {
+    GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
+        swarm$1.loggerService.log(METHOD_NAME$F, {
+            policySchema: policySchema,
+        });
+    swarm$1.policyValidationService.addPolicy(policySchema.policyName, policySchema);
+    swarm$1.policySchemaService.register(policySchema.policyName, policySchema);
+    return policySchema.policyName;
+});
 
 var METHOD_NAME$E = "function.commit.commitToolOutput";
 /**
@@ -10751,7 +12025,7 @@ var METHOD_NAME$E = "function.commit.commitToolOutput";
  * @param {AgentName} agentName - The name of the agent committing the output.
  * @returns {Promise<void>} - A promise that resolves when the operation is complete.
  */
-var commitToolOutput = function (toolId, content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
+var commitToolOutput = beginContext(function (toolId, content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -10786,7 +12060,7 @@ var commitToolOutput = function (toolId, content, clientId, agentName) { return
                 return [2 /*return*/];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$D = "function.commit.commitSystemMessage";
 /**
@@ -10797,7 +12071,7 @@ var METHOD_NAME$D = "function.commit.commitSystemMessage";
  * @param {string} agentName - The name of the agent.
  * @returns {Promise<void>} - A promise that resolves when the message is committed.
  */
-var commitSystemMessage = function (content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
+var commitSystemMessage = beginContext(function (content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -10830,7 +12104,7 @@ var commitSystemMessage = function (content, clientId, agentName) { return __awa
                 return [2 /*return*/];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$C = "function.commit.commitFlush";
 /**
@@ -10840,7 +12114,7 @@ var METHOD_NAME$C = "function.commit.commitFlush";
  * @param {string} agentName - The name of the agent.
  * @returns {Promise<void>} - A promise that resolves when the message is committed.
  */
-var commitFlush = function (clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
+var commitFlush = beginContext(function (clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -10872,7 +12146,7 @@ var commitFlush = function (clientId, agentName) { return __awaiter(void 0, void
                 return [2 /*return*/];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$B = "function.commit.commitSystemMessage";
 /**
@@ -10883,7 +12157,7 @@ var METHOD_NAME$B = "function.commit.commitSystemMessage";
  * @param {string} agentName - The name of the agent.
  * @returns {Promise<void>} - A promise that resolves when the message is committed.
  */
-var commitUserMessage = function (content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
+var commitUserMessage = beginContext(function (content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -10916,7 +12190,7 @@ var commitUserMessage = function (content, clientId, agentName) { return __await
                 return [2 /*return*/];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$A = "function.commit.commitToolOutputForce";
 /**
@@ -10926,7 +12200,7 @@ var METHOD_NAME$A = "function.commit.commitToolOutputForce";
  * @param {string} clientId - The client ID associated with the session.
  * @returns {Promise<void>} - A promise that resolves when the operation is complete.
  */
-var commitToolOutputForce = function (toolId, content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var commitToolOutputForce = beginContext(function (toolId, content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -10946,7 +12220,7 @@ var commitToolOutputForce = function (toolId, content, clientId) { return __awai
                 return [2 /*return*/];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$z = "function.commit.commitSystemMessageForce";
 /**
@@ -10956,7 +12230,7 @@ var METHOD_NAME$z = "function.commit.commitSystemMessageForce";
  * @param {string} clientId - The ID of the client.
  * @returns {Promise<void>} - A promise that resolves when the message is committed.
  */
-var commitSystemMessageForce = function (content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var commitSystemMessageForce = beginContext(function (content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -10975,7 +12249,7 @@ var commitSystemMessageForce = function (content, clientId) { return __awaiter(v
                 return [2 /*return*/];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$y = "function.commit.commitFlushForce";
 /**
@@ -10984,7 +12258,7 @@ var METHOD_NAME$y = "function.commit.commitFlushForce";
  * @param {string} clientId - The ID of the client.
  * @returns {Promise<void>} - A promise that resolves when the message is committed.
  */
-var commitFlushForce = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var commitFlushForce = beginContext(function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -11003,7 +12277,7 @@ var commitFlushForce = function (clientId) { return __awaiter(void 0, void 0, vo
                 return [2 /*return*/];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$x = "function.commit.commitSystemMessage";
 /**
@@ -11013,7 +12287,7 @@ var METHOD_NAME$x = "function.commit.commitSystemMessage";
  * @param {string} clientId - The ID of the client.
  * @returns {Promise<void>} - A promise that resolves when the message is committed.
  */
-var commitUserMessageForce = function (content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var commitUserMessageForce = beginContext(function (content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -11032,7 +12306,7 @@ var commitUserMessageForce = function (content, clientId) { return __awaiter(voi
                 return [2 /*return*/];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$w = "function.commit.commitAssistantMessage";
 /**
@@ -11043,7 +12317,7 @@ var METHOD_NAME$w = "function.commit.commitAssistantMessage";
  * @param {string} agentName - The name of the agent.
  * @returns {Promise<void>} - A promise that resolves when the message is committed.
  */
-var commitAssistantMessage = function (content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
+var commitAssistantMessage = beginContext(function (content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -11076,7 +12350,7 @@ var commitAssistantMessage = function (content, clientId, agentName) { return __
                 return [2 /*return*/];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$v = "function.commit.commitAssistantMessageForce";
 /**
@@ -11086,7 +12360,7 @@ var METHOD_NAME$v = "function.commit.commitAssistantMessageForce";
  * @param {string} clientId - The ID of the client.
  * @returns {Promise<void>} - A promise that resolves when the message is committed.
  */
-var commitAssistantMessageForce = function (content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var commitAssistantMessageForce = beginContext(function (content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -11105,7 +12379,7 @@ var commitAssistantMessageForce = function (content, clientId) { return __awaite
                 return [2 /*return*/];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$u = "function.commit.cancelOutput";
 /**
@@ -11115,7 +12389,7 @@ var METHOD_NAME$u = "function.commit.cancelOutput";
  * @param {string} agentName - The name of the agent.
  * @returns {Promise<void>} - A promise that resolves when the output is canceled
  */
-var cancelOutput = function (clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
+var cancelOutput = beginContext(function (clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -11147,7 +12421,7 @@ var cancelOutput = function (clientId, agentName) { return __awaiter(void 0, voi
                 return [2 /*return*/];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$t = "function.commit.cancelOutputForce";
 /**
@@ -11157,7 +12431,7 @@ var METHOD_NAME$t = "function.commit.cancelOutputForce";
  * @param {string} agentName - The name of the agent.
  * @returns {Promise<void>} - A promise that resolves when the output is canceled
  */
-var cancelOutputForce = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var cancelOutputForce = beginContext(function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -11175,7 +12449,7 @@ var cancelOutputForce = function (clientId) { return __awaiter(void 0, void 0, v
                 return [2 /*return*/];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$s = "function.commit.commitStopTools";
 /**
@@ -11185,7 +12459,7 @@ var METHOD_NAME$s = "function.commit.commitStopTools";
  * @param {string} agentName - The name of the agent.
  * @returns {Promise<void>} - A promise that resolves when the message is committed.
  */
-var commitStopTools = function (clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
+var commitStopTools = beginContext(function (clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -11217,7 +12491,7 @@ var commitStopTools = function (clientId, agentName) { return __awaiter(void 0,
                 return [2 /*return*/];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$r = "function.commit.commitStopToolsForce";
 /**
@@ -11226,7 +12500,7 @@ var METHOD_NAME$r = "function.commit.commitStopToolsForce";
  * @param {string} clientId - The ID of the client.
  * @returns {Promise<void>} - A promise that resolves when the message is committed.
  */
-var commitStopToolsForce = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var commitStopToolsForce = beginContext(function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -11245,7 +12519,7 @@ var commitStopToolsForce = function (clientId) { return __awaiter(void 0, void 0
                 return [2 /*return*/];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$q = "function.target.emitForce";
 /**
@@ -11258,7 +12532,7 @@ var METHOD_NAME$q = "function.target.emitForce";
  * @throws Will throw an error if the session mode is not "makeConnection".
  * @returns {Promise<void>} A promise that resolves when the content is emitted.
  */
-var emitForce = function (content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var emitForce = beginContext(function (content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -11268,7 +12542,8 @@ var emitForce = function (content, clientId) { return __awaiter(void 0, void 0,
                         content: content,
                         clientId: clientId,
                     });
-                if (swarm$1.sessionValidationService.getSessionMode(clientId) !== "makeConnection") {
+                if (swarm$1.sessionValidationService.getSessionMode(clientId) !==
+                    "makeConnection") {
                     throw new Error("agent-swarm-kit emitForce session is not makeConnection clientId=".concat(clientId));
                 }
                 swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$q);
@@ -11278,7 +12553,7 @@ var emitForce = function (content, clientId) { return __awaiter(void 0, void 0,
             case 1: return [2 /*return*/, _a.sent()];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$p = "function.target.executeForce";
 /**
@@ -11291,7 +12566,7 @@ var METHOD_NAME$p = "function.target.executeForce";
  * @param {string} clientId - The ID of the client requesting execution.
  * @returns {Promise<void>} - A promise that resolves when the execution is complete.
  */
-var executeForce = function (content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var executeForce = beginContext(function (content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var executionId, swarmName;
     return __generator(this, function (_a) {
         executionId = functoolsKit.randomString();
@@ -11334,7 +12609,7 @@ var executeForce = function (content, clientId) { return __awaiter(void 0, void
                 executionId: executionId,
             })];
     });
-}); };
+}); });
 
 var METHOD_NAME$o = "function.target.disposeConnection";
 /**
@@ -11344,7 +12619,7 @@ var METHOD_NAME$o = "function.target.disposeConnection";
  * @param {SwarmName} swarmName - The name of the swarm.
  * @returns {Promise<void>} A promise that resolves when the connection is disposed.
  */
-var disposeConnection = function (clientId_1, swarmName_1) {
+var disposeConnection = beginContext(function (clientId_1, swarmName_1) {
     var args_1 = [];
     for (var _i = 2; _i < arguments.length; _i++) {
         args_1[_i - 2] = arguments[_i];
@@ -11456,7 +12731,7 @@ var disposeConnection = function (clientId_1, swarmName_1) {
             }
         });
     });
-};
+});
 
 var METHOD_NAME$n = "function.target.makeAutoDispose";
 var DEFAULT_TIMEOUT = 15 * 60;
@@ -11468,7 +12743,7 @@ var DEFAULT_TIMEOUT = 15 * 60;
  * @param {Partial<IMakeDisposeParams>} [params={}] - Optional parameters for auto-dispose.
  * @returns {Object} An object with tick and stop methods to control the auto-dispose.
  */
-var makeAutoDispose = function (clientId, swarmName, _a) {
+var makeAutoDispose = beginContext(function (clientId, swarmName, _a) {
     var _b = _a === void 0 ? {} : _a, _c = _b.timeoutSeconds, timeoutSeconds = _c === void 0 ? DEFAULT_TIMEOUT : _c, onDestroy = _b.onDestroy;
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
         swarm$1.loggerService.log(METHOD_NAME$n, {
@@ -11516,7 +12791,7 @@ var makeAutoDispose = function (clientId, swarmName, _a) {
             onDestroy && onDestroy(clientId, swarmName);
         },
     };
-};
+});
 
 var METHOD_NAME$m = "function.target.execute";
 /**
@@ -11528,7 +12803,7 @@ var METHOD_NAME$m = "function.target.execute";
  * @param {AgentName} agentName - The name of the agent executing the command.
  * @returns {Promise<void>} - A promise that resolves when the execution is complete.
  */
-var execute = function (content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
+var execute = beginContext(function (content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
     var executionId, swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -11575,7 +12850,10 @@ var execute = function (content, clientId, agentName) { return __awaiter(void 0,
                                 case 2:
                                     result = _a.sent();
                                     isFinished = swarm$1.perfService.endExecution(executionId, clientId, result.length);
-                                    swarm$1.busService.commitExecutionEnd(clientId, { agentName: agentName, swarmName: swarmName });
+                                    swarm$1.busService.commitExecutionEnd(clientId, {
+                                        agentName: agentName,
+                                        swarmName: swarmName,
+                                    });
                                     return [2 /*return*/, result];
                                 case 3:
                                     if (!isFinished) {
@@ -11591,7 +12869,7 @@ var execute = function (content, clientId, agentName) { return __awaiter(void 0,
                     })];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$l = "function.target.emit";
 /**
@@ -11604,7 +12882,7 @@ var METHOD_NAME$l = "function.target.emit";
  * @throws Will throw an error if the session mode is not "makeConnection".
  * @returns {Promise<void>} A promise that resolves when the content is emitted.
  */
-var emit = function (content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
+var emit = beginContext(function (content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -11615,7 +12893,8 @@ var emit = function (content, clientId, agentName) { return __awaiter(void 0, vo
                         clientId: clientId,
                         agentName: agentName,
                     });
-                if (swarm$1.sessionValidationService.getSessionMode(clientId) !== "makeConnection") {
+                if (swarm$1.sessionValidationService.getSessionMode(clientId) !==
+                    "makeConnection") {
                     throw new Error("agent-swarm-kit emit session is not makeConnection clientId=".concat(clientId));
                 }
                 swarm$1.agentValidationService.validate(agentName, METHOD_NAME$l);
@@ -11638,7 +12917,7 @@ var emit = function (content, clientId, agentName) { return __awaiter(void 0, vo
             case 2: return [2 /*return*/, _a.sent()];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$k = "function.target.runStateless";
 /**
@@ -11650,7 +12929,7 @@ var METHOD_NAME$k = "function.target.runStateless";
  * @param {AgentName} agentName - The name of the agent running the command.
  * @returns {Promise<string>} - A promise that resolves the run result
  */
-var runStateless = function (content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
+var runStateless = beginContext(function (content, clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
     var executionId, swarmName, currentAgentName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -11716,7 +12995,7 @@ var runStateless = function (content, clientId, agentName) { return __awaiter(vo
                     })];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$j = "function.target.runStatelessForce";
 /**
@@ -11729,7 +13008,7 @@ var METHOD_NAME$j = "function.target.runStatelessForce";
  * @param {string} clientId - The ID of the client requesting run.
  * @returns {Promise<string>} - A promise that resolves the run result
  */
-var runStatelessForce = function (content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var runStatelessForce = beginContext(function (content, clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var executionId, swarmName;
     return __generator(this, function (_a) {
         executionId = functoolsKit.randomString();
@@ -11772,7 +13051,7 @@ var runStatelessForce = function (content, clientId) { return __awaiter(void 0,
                 executionId: executionId,
             })];
     });
-}); };
+}); });
 
 var METHOD_NAME$i = "function.common.getAgentName";
 /**
@@ -11782,7 +13061,7 @@ var METHOD_NAME$i = "function.common.getAgentName";
  * @returns {Promise<string>} The name of the agent.
  * @throws Will throw an error if the client ID is invalid or if the swarm validation fails.
  */
-var getAgentName = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var getAgentName = beginContext(function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -11798,20 +13077,12 @@ var getAgentName = function (clientId) { return __awaiter(void 0, void 0, void 0
             case 1: return [2 /*return*/, _a.sent()];
         }
     });
-}); };
+}); });
 
 var SCHEDULED_DELAY$1 = 1000;
 var RATE_DELAY = 10000;
 var METHOD_NAME$h = "function.target.makeConnection";
-/**
- * A connection factory for a client to a swarm and returns a function to send messages.
- *
- * @param {ReceiveMessageFn} connector - The function to receive messages.
- * @param {string} clientId - The unique identifier of the client.
- * @param {SwarmName} swarmName - The name of the swarm.
- * @returns {SendMessageFn} - A function to send messages to the swarm.
- */
-var makeConnection = function (connector, clientId, swarmName) {
+var makeConnectionInternal = beginContext(function (connector, clientId, swarmName) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
         swarm$1.loggerService.log(METHOD_NAME$h, {
             clientId: clientId,
@@ -11820,7 +13091,7 @@ var makeConnection = function (connector, clientId, swarmName) {
     swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$h);
     swarm$1.sessionValidationService.addSession(clientId, swarmName, "makeConnection");
     var send = swarm$1.sessionPublicService.connect(connector, METHOD_NAME$h, clientId, swarmName);
-    return functoolsKit.queued(function (outgoing) { return __awaiter(void 0, void 0, void 0, function () {
+    return functoolsKit.queued(beginContext(function (outgoing) { return __awaiter(void 0, void 0, void 0, function () {
         var _a;
         var _b;
         return __generator(this, function (_c) {
@@ -11838,8 +13109,17 @@ var makeConnection = function (connector, clientId, swarmName) {
                 case 2: return [2 /*return*/, _c.sent()];
             }
         });
-    }); });
-};
+    }); }));
+});
+/**
+ * A connection factory for a client to a swarm and returns a function to send messages.
+ *
+ * @param {ReceiveMessageFn} connector - The function to receive messages.
+ * @param {string} clientId - The unique identifier of the client.
+ * @param {SwarmName} swarmName - The name of the swarm.
+ * @returns {SendMessageFn} - A function to send messages to the swarm.
+ */
+var makeConnection = function (connector, clientId, swarmName) { return makeConnectionInternal(connector, clientId, swarmName); };
 /**
  * A scheduled connection factory for a client to a swarm and returns a function to send messages.
  *
@@ -11849,7 +13129,7 @@ var makeConnection = function (connector, clientId, swarmName) {
  * @param {Partial<IMakeConnectionConfig>} [config] - The configuration for scheduling.
  * @returns {SendMessageFn} - A function to send scheduled messages to the swarm.
  */
-makeConnection.scheduled = function (connector, clientId, swarmName, _a) {
+makeConnection.scheduled = beginContext(function (connector, clientId, swarmName, _a) {
     var _b = _a === void 0 ? {} : _a, _c = _b.delay, delay = _c === void 0 ? SCHEDULED_DELAY$1 : _c;
     var send = makeConnection(connector, clientId, swarmName);
     /**
@@ -11858,7 +13138,7 @@ makeConnection.scheduled = function (connector, clientId, swarmName, _a) {
      * @param {string} content - The message content to be sent.
      * @returns {Promise<void>} - A promise that resolves when the message is sent.
      */
-    var wrappedSend = functoolsKit.schedule(function (content) { return __awaiter(void 0, void 0, void 0, function () {
+    var wrappedSend = functoolsKit.schedule(beginContext(function (content) { return __awaiter(void 0, void 0, void 0, function () {
         return __generator(this, function (_a) {
             switch (_a.label) {
                 case 0:
@@ -11869,14 +13149,14 @@ makeConnection.scheduled = function (connector, clientId, swarmName, _a) {
                 case 1: return [2 /*return*/, _a.sent()];
             }
         });
-    }); }, {
+    }); }), {
         /**
          * A function that is called when a message is scheduled.
          *
          * @param {[string]} content - The message content to be scheduled.
          * @returns {Promise<void>} - A promise that resolves when the message is committed.
          */
-        onSchedule: function (_a) { return __awaiter(void 0, [_a], void 0, function (_b) {
+        onSchedule: beginContext(function (_a) { return __awaiter(void 0, [_a], void 0, function (_b) {
             var _c, _d;
             var _e = __read(_b, 1), content = _e[0];
             return __generator(this, function (_f) {
@@ -11895,7 +13175,7 @@ makeConnection.scheduled = function (connector, clientId, swarmName, _a) {
                         return [2 /*return*/];
                 }
             });
-        }); },
+        }); }),
         /**
          * The delay for message scheduler
          */
@@ -11915,7 +13195,7 @@ makeConnection.scheduled = function (connector, clientId, swarmName, _a) {
             }
         });
     }); };
-};
+});
 /**
  * A rate-limited connection factory for a client to a swarm and returns a function to send messages.
  *
@@ -11926,7 +13206,7 @@ makeConnection.scheduled = function (connector, clientId, swarmName, _a) {
  * @param {number} [config.delay=RATE_DELAY] - The delay in milliseconds for rate limiting messages.
  * @returns {SendMessageFn} - A function to send rate-limited messages to the swarm.
  */
-makeConnection.rate = function (connector, clientId, swarmName, _a) {
+makeConnection.rate = beginContext(function (connector, clientId, swarmName, _a) {
     var _b = _a === void 0 ? {} : _a, _c = _b.delay, delay = _c === void 0 ? RATE_DELAY : _c;
     var send = makeConnection(connector, clientId, swarmName);
     /**
@@ -11935,7 +13215,7 @@ makeConnection.rate = function (connector, clientId, swarmName, _a) {
      * @param {string} content - The message content to be sent.
      * @returns {Promise<void>} - A promise that resolves when the message is sent.
      */
-    var wrappedSend = functoolsKit.rate(function (content) { return __awaiter(void 0, void 0, void 0, function () {
+    var wrappedSend = functoolsKit.rate(beginContext(function (content) { return __awaiter(void 0, void 0, void 0, function () {
         return __generator(this, function (_a) {
             switch (_a.label) {
                 case 0:
@@ -11946,7 +13226,7 @@ makeConnection.rate = function (connector, clientId, swarmName, _a) {
                 case 1: return [2 /*return*/, _a.sent()];
             }
         });
-    }); }, {
+    }); }), {
         key: function () { return clientId; },
         rateName: "makeConnection.rate clientId=".concat(clientId),
         delay: delay,
@@ -11968,15 +13248,15 @@ makeConnection.rate = function (connector, clientId, swarmName, _a) {
                 case 2:
                     error_1 = _a.sent();
                     if ((error_1 === null || error_1 === void 0 ? void 0 : error_1.type) === "rate-error") {
-                        console.warn("agent-swarm rate limit reached for clientId=".concat(clientId));
-                        return [2 /*return*/];
+                        console.warn("agent-swarm makeConnection.rate rate limit reached for clientId=".concat(clientId));
+                        return [2 /*return*/, ""];
                     }
                     throw error_1;
                 case 3: return [2 /*return*/];
             }
         });
     }); };
-};
+});
 
 var METHOD_NAME$g = "function.target.complete";
 var COMPLETE_TTL = 15 * 60 * 1000;
@@ -12031,7 +13311,7 @@ var createGc$3 = functoolsKit.singleshot(function () { return __awaiter(void 0,
  * @param {SwarmName} swarmName - The swarm name.
  * @returns {Promise<string>} The result of the complete function.
  */
-var complete = function (content, clientId, swarmName) { return __awaiter(void 0, void 0, void 0, function () {
+var complete = beginContext(function (content, clientId, swarmName) { return __awaiter(void 0, void 0, void 0, function () {
     var executionId, run;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -12079,20 +13359,11 @@ var complete = function (content, clientId, swarmName) { return __awaiter(void 0
                     })];
         }
     });
-}); };
+}); });
 
 var SCHEDULED_DELAY = 1000;
 var METHOD_NAME$f = "function.target.session";
-/**
- * Creates a session for the given client and swarm.
- *
- * @param {string} clientId - The ID of the client.
- * @param {SwarmName} swarmName - The name of the swarm.
- * @returns {Object} An object containing the session methods.
- * @returns {TComplete} complete - A function to complete the session with content.
- * @returns {Function} dispose - A function to dispose of the session.
- */
-var session = function (clientId, swarmName) {
+var sessionInternal = beginContext(function (clientId, swarmName) {
     var executionId = functoolsKit.randomString();
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
         swarm$1.loggerService.log(METHOD_NAME$f, {
@@ -12109,7 +13380,7 @@ var session = function (clientId, swarmName) {
          * @param {string} content - The content to complete the session with.
          * @returns {Promise<string>} A promise that resolves with the result of the session execution.
          */
-        complete: functoolsKit.queued(function (content) { return __awaiter(void 0, void 0, void 0, function () {
+        complete: functoolsKit.queued(beginContext(function (content) { return __awaiter(void 0, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 swarm$1.sessionValidationService.validate(clientId, METHOD_NAME$f);
                 return [2 /*return*/, ExecutionContextService.runInContext(function () { return __awaiter(void 0, void 0, void 0, function () {
@@ -12144,34 +13415,134 @@ var session = function (clientId, swarmName) {
                         executionId: executionId,
                     })];
             });
-        }); }),
+        }); })),
+        /**
+         * Disposes of the session.
+         *
+         * @returns {Promise<void>} A promise that resolves when the session is disposed.
+         */
+        dispose: beginContext(function () { return __awaiter(void 0, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0: return [4 /*yield*/, disposeConnection(clientId, swarmName, METHOD_NAME$f)];
+                    case 1: return [2 /*return*/, _a.sent()];
+                }
+            });
+        }); }),
+    };
+});
+/**
+ * Creates a session for the given client and swarm.
+ *
+ * @param {string} clientId - The ID of the client.
+ * @param {SwarmName} swarmName - The name of the swarm.
+ * @returns {Object} An object containing the session methods.
+ * @returns {TComplete} complete - A function to complete the session with content.
+ * @returns {Function} dispose - A function to dispose of the session.
+ */
+var session = function (clientId, swarmName) {
+    return sessionInternal(clientId, swarmName);
+};
+/**
+ * Creates a scheduled session for the given client and swarm.
+ *
+ * @param {string} clientId - The ID of the client.
+ * @param {SwarmName} swarmName - The name of the swarm.
+ * @param {Partial<ISessionConfig>} [config] - The configuration for the scheduled session.
+ * @param {number} [config.delay] - The delay for the scheduled session.
+ * @returns {Object} An object containing the scheduled session methods.
+ * @returns {TComplete} complete - A function to complete the session with content.
+ * @returns {Function} dispose - A function to dispose of the session.
+ */
+session.scheduled = function (clientId, swarmName, _a) {
+    var _b = _a === void 0 ? {} : _a, _c = _b.delay, delay = _c === void 0 ? SCHEDULED_DELAY : _c;
+    var _d = session(clientId, swarmName), complete = _d.complete, dispose = _d.dispose;
+    var isMounted = true;
+    /**
+     * Completes the scheduled session with the given content.
+     *
+     * @param {string} content - The content to complete the session with.
+     * @returns {Promise<string>} A promise that resolves with the result of the session execution.
+     */
+    var wrappedComplete = functoolsKit.schedule(beginContext(function (content) { return __awaiter(void 0, void 0, void 0, function () {
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    if (!isMounted) {
+                        return [2 /*return*/];
+                    }
+                    return [4 /*yield*/, complete(content)];
+                case 1: return [2 /*return*/, _a.sent()];
+            }
+        });
+    }); }), {
+        onSchedule: beginContext(function (_a) { return __awaiter(void 0, [_a], void 0, function (_b) {
+            var _c, _d;
+            var _e = __read(_b, 1), content = _e[0];
+            return __generator(this, function (_f) {
+                switch (_f.label) {
+                    case 0:
+                        if (!isMounted) {
+                            return [2 /*return*/];
+                        }
+                        _c = commitUserMessage;
+                        _d = [content,
+                            clientId];
+                        return [4 /*yield*/, getAgentName(clientId)];
+                    case 1: return [4 /*yield*/, _c.apply(void 0, _d.concat([_f.sent()]))];
+                    case 2:
+                        _f.sent();
+                        return [2 /*return*/];
+                }
+            });
+        }); }),
+        delay: delay,
+    });
+    return {
+        /**
+         * Completes the scheduled session with the given content.
+         *
+         * @param {string} content - The content to complete the session with.
+         * @returns {Promise<string>} A promise that resolves with the result of the session execution.
+         */
+        complete: function (content) {
+            return __awaiter(this, void 0, void 0, function () {
+                return __generator(this, function (_a) {
+                    switch (_a.label) {
+                        case 0: return [4 /*yield*/, wrappedComplete(content)];
+                        case 1: return [2 /*return*/, _a.sent()];
+                    }
+                });
+            });
+        },
         /**
-         * Disposes of the session.
+         * Disposes of the scheduled session.
          *
          * @returns {Promise<void>} A promise that resolves when the session is disposed.
          */
-        dispose: function () { return __awaiter(void 0, void 0, void 0, function () {
-            return __generator(this, function (_a) {
-                switch (_a.label) {
-                    case 0: return [4 /*yield*/, disposeConnection(clientId, swarmName, METHOD_NAME$f)];
-                    case 1: return [2 /*return*/, _a.sent()];
-                }
+        dispose: function () {
+            return __awaiter(this, void 0, void 0, function () {
+                return __generator(this, function (_a) {
+                    switch (_a.label) {
+                        case 0:
+                            isMounted = false;
+                            return [4 /*yield*/, dispose()];
+                        case 1: return [2 /*return*/, _a.sent()];
+                    }
+                });
             });
-        }); },
+        },
     };
 };
 /**
- * Creates a scheduled session for the given client and swarm.
+ * A rate-limited connection factory for a client to a swarm and returns a function to send messages.
  *
- * @param {string} clientId - The ID of the client.
+ * @param {string} clientId - The unique identifier of the client.
  * @param {SwarmName} swarmName - The name of the swarm.
- * @param {Partial<ISessionConfig>} [config] - The configuration for the scheduled session.
- * @param {number} [config.delay] - The delay for the scheduled session.
- * @returns {Object} An object containing the scheduled session methods.
- * @returns {TComplete} complete - A function to complete the session with content.
- * @returns {Function} dispose - A function to dispose of the session.
+ * @param {Partial<ISessionConfig>} [config] - The configuration for rate limiting.
+ * @param {number} [config.delay=SCHEDULED_DELAY] - The delay in milliseconds for rate limiting messages.
  */
-session.scheduled = function (clientId, swarmName, _a) {
+session.rate = function (clientId, swarmName, _a) {
     var _b = _a === void 0 ? {} : _a, _c = _b.delay, delay = _c === void 0 ? SCHEDULED_DELAY : _c;
     var _d = session(clientId, swarmName), complete = _d.complete, dispose = _d.dispose;
     var isMounted = true;
@@ -12181,7 +13552,7 @@ session.scheduled = function (clientId, swarmName, _a) {
      * @param {string} content - The content to complete the session with.
      * @returns {Promise<string>} A promise that resolves with the result of the session execution.
      */
-    var wrappedComplete = functoolsKit.schedule(function (content) { return __awaiter(void 0, void 0, void 0, function () {
+    var wrappedComplete = functoolsKit.rate(beginContext(function (content) { return __awaiter(void 0, void 0, void 0, function () {
         return __generator(this, function (_a) {
             switch (_a.label) {
                 case 0:
@@ -12192,27 +13563,9 @@ session.scheduled = function (clientId, swarmName, _a) {
                 case 1: return [2 /*return*/, _a.sent()];
             }
         });
-    }); }, {
-        onSchedule: function (_a) { return __awaiter(void 0, [_a], void 0, function (_b) {
-            var _c, _d;
-            var _e = __read(_b, 1), content = _e[0];
-            return __generator(this, function (_f) {
-                switch (_f.label) {
-                    case 0:
-                        if (!isMounted) {
-                            return [2 /*return*/];
-                        }
-                        _c = commitUserMessage;
-                        _d = [content,
-                            clientId];
-                        return [4 /*yield*/, getAgentName(clientId)];
-                    case 1: return [4 /*yield*/, _c.apply(void 0, _d.concat([_f.sent()]))];
-                    case 2:
-                        _f.sent();
-                        return [2 /*return*/];
-                }
-            });
-        }); },
+    }); }), {
+        key: function () { return clientId; },
+        rateName: "makeConnection.rate clientId=".concat(clientId),
         delay: delay,
     });
     return {
@@ -12224,10 +13577,21 @@ session.scheduled = function (clientId, swarmName, _a) {
          */
         complete: function (content) {
             return __awaiter(this, void 0, void 0, function () {
+                var error_1;
                 return __generator(this, function (_a) {
                     switch (_a.label) {
-                        case 0: return [4 /*yield*/, wrappedComplete(content)];
+                        case 0:
+                            _a.trys.push([0, 2, , 3]);
+                            return [4 /*yield*/, wrappedComplete(content)];
                         case 1: return [2 /*return*/, _a.sent()];
+                        case 2:
+                            error_1 = _a.sent();
+                            if ((error_1 === null || error_1 === void 0 ? void 0 : error_1.type) === "rate-error") {
+                                console.warn("agent-swarm session.rate rate limit reached for clientId=".concat(clientId));
+                                return [2 /*return*/, ""];
+                            }
+                            throw error_1;
+                        case 3: return [2 /*return*/];
                     }
                 });
             });
@@ -12260,7 +13624,7 @@ var METHOD_NAME$e = "function.common.getAgentHistory";
  * @param {AgentName} agentName - The name of the agent.
  * @returns {Promise<Array>} - A promise that resolves to an array containing the agent's history.
  */
-var getAgentHistory = function (clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
+var getAgentHistory = beginContext(function (clientId, agentName) { return __awaiter(void 0, void 0, void 0, function () {
     var prompt, history;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -12279,7 +13643,7 @@ var getAgentHistory = function (clientId, agentName) { return __awaiter(void 0,
                 return [2 /*return*/, __spreadArray([], __read(history), false)];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$d = "function.common.getSessionMode";
 /**
@@ -12287,7 +13651,7 @@ var METHOD_NAME$d = "function.common.getSessionMode";
  *
  * @param {string} clientId - The client ID of the session.
  */
-var getSessionMode = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var getSessionMode = beginContext(function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName;
     return __generator(this, function (_a) {
         GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
@@ -12299,7 +13663,7 @@ var getSessionMode = function (clientId) { return __awaiter(void 0, void 0, void
         swarm$1.swarmValidationService.validate(swarmName, METHOD_NAME$d);
         return [2 /*return*/, swarm$1.sessionValidationService.getSessionMode(clientId)];
     });
-}); };
+}); });
 
 var METHOD_NAME$c = "function.common.getSessionContext";
 /**
@@ -12336,7 +13700,7 @@ var METHOD_NAME$b = "function.history.getRawHistory";
  * @param {string} clientId - The ID of the client whose history is to be retrieved.
  * @returns {Promise<Array>} A promise that resolves to an array containing the raw history.
  */
-var getRawHistory = function (clientId_1) {
+var getRawHistory = beginContext(function (clientId_1) {
     var args_1 = [];
     for (var _i = 1; _i < arguments.length; _i++) {
         args_1[_i - 1] = arguments[_i];
@@ -12364,7 +13728,7 @@ var getRawHistory = function (clientId_1) {
             }
         });
     });
-};
+});
 
 var METHOD_NAME$a = "function.history.getLastUserMessage";
 /**
@@ -12373,7 +13737,7 @@ var METHOD_NAME$a = "function.history.getLastUserMessage";
  * @param {string} clientId - The ID of the client whose message history is being retrieved.
  * @returns {Promise<string | null>} - The content of the last user message, or null if no user message is found.
  */
-var getLastUserMessage = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var getLastUserMessage = beginContext(function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var history, last;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -12392,7 +13756,7 @@ var getLastUserMessage = function (clientId) { return __awaiter(void 0, void 0,
                 return [2 /*return*/, last ? last.content : null];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$9 = "function.history.getUserHistory";
 /**
@@ -12401,7 +13765,7 @@ var METHOD_NAME$9 = "function.history.getUserHistory";
  * @param {string} clientId - The ID of the client whose history is to be retrieved.
  * @returns {Promise<Array>} A promise that resolves to an array of history objects filtered by user role.
  */
-var getUserHistory = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var getUserHistory = beginContext(function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var history;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -12419,7 +13783,7 @@ var getUserHistory = function (clientId) { return __awaiter(void 0, void 0, void
                     })];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$8 = "function.history.getAssistantHistory";
 /**
@@ -12428,7 +13792,7 @@ var METHOD_NAME$8 = "function.history.getAssistantHistory";
  * @param {string} clientId - The ID of the client.
  * @returns {Promise<Array>} - A promise that resolves to an array of history objects where the role is "assistant".
  */
-var getAssistantHistory = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var getAssistantHistory = beginContext(function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var history;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -12446,7 +13810,7 @@ var getAssistantHistory = function (clientId) { return __awaiter(void 0, void 0,
                     })];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$7 = "function.history.getLastAssistantMessage";
 /**
@@ -12455,7 +13819,7 @@ var METHOD_NAME$7 = "function.history.getLastAssistantMessage";
  * @param {string} clientId - The ID of the client whose message history is being retrieved.
  * @returns {Promise<string | null>} - The content of the last assistant message, or null if no user message is found.
  */
-var getLastAssistantMessage = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var getLastAssistantMessage = beginContext(function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var history, last;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -12474,7 +13838,7 @@ var getLastAssistantMessage = function (clientId) { return __awaiter(void 0, voi
                 return [2 /*return*/, last ? last.content : null];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$6 = "function.history.getLastSystemMessage";
 /**
@@ -12483,7 +13847,7 @@ var METHOD_NAME$6 = "function.history.getLastSystemMessage";
  * @param {string} clientId - The ID of the client whose message history is being retrieved.
  * @returns {Promise<string | null>} - The content of the last system message, or null if no user message is found.
  */
-var getLastSystemMessage = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var getLastSystemMessage = beginContext(function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var history, last;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -12502,7 +13866,7 @@ var getLastSystemMessage = function (clientId) { return __awaiter(void 0, void 0
                 return [2 /*return*/, last ? last.content : null];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$5 = "function.event.listenEvent";
 var DISALLOWED_EVENT_SOURCE_LIST$2 = new Set([
@@ -12512,7 +13876,8 @@ var DISALLOWED_EVENT_SOURCE_LIST$2 = new Set([
     "state-bus",
     "storage-bus",
     "swarm-bus",
-    "execution-bus"
+    "execution-bus",
+    "policy-bus",
 ]);
 /**
  * Emits an event to the swarm bus service.
@@ -12522,7 +13887,7 @@ var DISALLOWED_EVENT_SOURCE_LIST$2 = new Set([
  * @param {T} payload - The payload of the event.
  * @returns {boolean} - Returns true if the event was successfully emitted.
  */
-var event = function (clientId, topicName, payload) {
+var event = beginContext(function (clientId, topicName, payload) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
         swarm$1.loggerService.log(METHOD_NAME$5, {
             clientId: clientId,
@@ -12535,7 +13900,7 @@ var event = function (clientId, topicName, payload) {
         payload: payload,
         clientId: clientId,
     });
-};
+});
 
 var METHOD_NAME$4 = "function.event.listenEvent";
 var DISALLOWED_EVENT_SOURCE_LIST$1 = new Set([
@@ -12545,9 +13910,10 @@ var DISALLOWED_EVENT_SOURCE_LIST$1 = new Set([
     "state-bus",
     "storage-bus",
     "swarm-bus",
-    "execution-bus"
+    "execution-bus",
+    "policy-bus",
 ]);
-var validateClientId$e = function (clientId) {
+var validateClientId$g = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -12562,7 +13928,7 @@ var validateClientId$e = function (clientId) {
  * @param {string} clientId - The ID of the client to listen for events from.
  * @param {(data: T) => void} fn - The callback function to execute when the event is received. The data payload is passed as an argument to this function.
  */
-var listenEvent = function (clientId, topicName, fn) {
+var listenEvent = beginContext(function (clientId, topicName, fn) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
         swarm$1.loggerService.log(METHOD_NAME$4, {
             clientId: clientId,
@@ -12570,7 +13936,7 @@ var listenEvent = function (clientId, topicName, fn) {
     if (DISALLOWED_EVENT_SOURCE_LIST$1.has(topicName)) {
         throw new Error("agent-swarm listenEvent topic is reserved topicName=".concat(topicName));
     }
-    validateClientId$e(clientId);
+    validateClientId$g(clientId);
     return swarm$1.busService.subscribe(clientId, topicName, functoolsKit.queued(function (_a) { return __awaiter(void 0, [_a], void 0, function (_b) {
         var payload = _b.payload;
         return __generator(this, function (_c) {
@@ -12580,7 +13946,7 @@ var listenEvent = function (clientId, topicName, fn) {
             }
         });
     }); }));
-};
+});
 
 var METHOD_NAME$3 = "function.event.listenEventOnce";
 var DISALLOWED_EVENT_SOURCE_LIST = new Set([
@@ -12590,9 +13956,10 @@ var DISALLOWED_EVENT_SOURCE_LIST = new Set([
     "state-bus",
     "storage-bus",
     "swarm-bus",
-    "execution-bus"
+    "execution-bus",
+    "policy-bus",
 ]);
-var validateClientId$d = function (clientId) {
+var validateClientId$f = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -12607,7 +13974,7 @@ var validateClientId$d = function (clientId) {
  * @param {string} clientId - The ID of the client to listen for events from.
  * @param {(data: T) => void} fn - The callback function to execute when the event is received. The data payload is passed as an argument to this function.
  */
-var listenEventOnce = function (clientId, topicName, filterFn, fn) {
+var listenEventOnce = beginContext(function (clientId, topicName, filterFn, fn) {
     GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
         swarm$1.loggerService.log(METHOD_NAME$3, {
             clientId: clientId,
@@ -12615,7 +13982,7 @@ var listenEventOnce = function (clientId, topicName, filterFn, fn) {
     if (DISALLOWED_EVENT_SOURCE_LIST.has(topicName)) {
         throw new Error("agent-swarm listenEventOnce topic is reserved topicName=".concat(topicName));
     }
-    validateClientId$d(clientId);
+    validateClientId$f(clientId);
     return swarm$1.busService.once(clientId, topicName, function (_a) {
         var payload = _a.payload;
         return filterFn(payload);
@@ -12628,7 +13995,7 @@ var listenEventOnce = function (clientId, topicName, filterFn, fn) {
             }
         });
     }); }));
-};
+});
 
 var METHOD_NAME$2 = "function.navigate.changeToAgent";
 /**
@@ -12717,7 +14084,7 @@ var createGc$2 = functoolsKit.singleshot(function () { return __awaiter(void 0,
  * @param {string} clientId - The client ID.
  * @returns {Promise<void>} - A promise that resolves when the agent is changed.
  */
-var changeToAgent = function (agentName, clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var changeToAgent = beginContext(function (agentName, clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName, activeAgent, run;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -12744,7 +14111,7 @@ var changeToAgent = function (agentName, clientId) { return __awaiter(void 0, vo
             case 3: return [2 /*return*/, _a.sent()];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME$1 = "function.navigate.changeToPrevAgent";
 /**
@@ -12832,7 +14199,7 @@ var createGc$1 = functoolsKit.singleshot(function () { return __awaiter(void 0,
  * @param {string} clientId - The client ID.
  * @returns {Promise<void>} - A promise that resolves when the agent is changed.
  */
-var changeToPrevAgent = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var changeToPrevAgent = beginContext(function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName, agentName, run;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -12857,7 +14224,7 @@ var changeToPrevAgent = function (clientId) { return __awaiter(void 0, void 0, v
             case 3: return [2 /*return*/, _a.sent()];
         }
     });
-}); };
+}); });
 
 var METHOD_NAME = "function.navigate.changeToDefaultAgent";
 /**
@@ -12945,7 +14312,7 @@ var createGc = functoolsKit.singleshot(function () { return __awaiter(void 0, vo
  * @param {string} clientId - The client ID.
  * @returns {Promise<void>} - A promise that resolves when the agent is changed.
  */
-var changeToDefaultAgent = function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
+var changeToDefaultAgent = beginContext(function (clientId) { return __awaiter(void 0, void 0, void 0, function () {
     var swarmName, agentName, run;
     return __generator(this, function (_a) {
         switch (_a.label) {
@@ -12968,9 +14335,9 @@ var changeToDefaultAgent = function (clientId) { return __awaiter(void 0, void 0
             case 2: return [2 /*return*/, _a.sent()];
         }
     });
-}); };
+}); });
 
-var validateClientId$c = function (clientId) {
+var validateClientId$e = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -12984,20 +14351,20 @@ var validateClientId$c = function (clientId) {
  * @param {string} clientId - The ID of the client to subscribe to events for.
  * @param {function} fn - The callback function to handle the event.
  */
-var listenAgentEvent = function (clientId, fn) {
+var listenAgentEvent = beginContext(function (clientId, fn) {
     swarm$1.loggerService.log("middleware listenAgentEvent", {
         clientId: clientId,
     });
-    validateClientId$c(clientId);
+    validateClientId$e(clientId);
     return swarm$1.busService.subscribe(clientId, "agent-bus", functoolsKit.queued(function (e) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
         switch (_a.label) {
             case 0: return [4 /*yield*/, fn(e)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     }); }); }));
-};
+});
 
-var validateClientId$b = function (clientId) {
+var validateClientId$d = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -13011,20 +14378,20 @@ var validateClientId$b = function (clientId) {
  * @param {string} clientId - The ID of the client to subscribe to.
  * @param {(event: IBusEvent) => void} fn - The callback function to handle the event.
  */
-var listenHistoryEvent = function (clientId, fn) {
+var listenHistoryEvent = beginContext(function (clientId, fn) {
     swarm$1.loggerService.log("middleware listenHistoryEvent", {
         clientId: clientId,
     });
-    validateClientId$b(clientId);
+    validateClientId$d(clientId);
     return swarm$1.busService.subscribe(clientId, "history-bus", functoolsKit.queued(function (e) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
         switch (_a.label) {
             case 0: return [4 /*yield*/, fn(e)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     }); }); }));
-};
+});
 
-var validateClientId$a = function (clientId) {
+var validateClientId$c = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -13038,20 +14405,20 @@ var validateClientId$a = function (clientId) {
  * @param {string} clientId - The ID of the client to subscribe to session events for.
  * @param {function} fn - The callback function to handle the session events.
  */
-var listenSessionEvent = function (clientId, fn) {
+var listenSessionEvent = beginContext(function (clientId, fn) {
     swarm$1.loggerService.log("middleware listenSessionEvent", {
         clientId: clientId,
     });
-    validateClientId$a(clientId);
+    validateClientId$c(clientId);
     return swarm$1.busService.subscribe(clientId, "session-bus", functoolsKit.queued(function (e) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
         switch (_a.label) {
             case 0: return [4 /*yield*/, fn(e)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     }); }); }));
-};
+});
 
-var validateClientId$9 = function (clientId) {
+var validateClientId$b = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -13065,20 +14432,20 @@ var validateClientId$9 = function (clientId) {
  * @param {string} clientId - The ID of the client to subscribe to.
  * @param {function} fn - The callback function to handle the event.
  */
-var listenStateEvent = function (clientId, fn) {
+var listenStateEvent = beginContext(function (clientId, fn) {
     swarm$1.loggerService.log("middleware listenStateEvent", {
         clientId: clientId,
     });
-    validateClientId$9(clientId);
+    validateClientId$b(clientId);
     return swarm$1.busService.subscribe(clientId, "state-bus", functoolsKit.queued(function (e) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
         switch (_a.label) {
             case 0: return [4 /*yield*/, fn(e)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     }); }); }));
-};
+});
 
-var validateClientId$8 = function (clientId) {
+var validateClientId$a = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -13092,20 +14459,20 @@ var validateClientId$8 = function (clientId) {
  * @param {string} clientId - The ID of the client to subscribe to storage events for.
  * @param {function} fn - The callback function to handle the storage event.
  */
-var listenStorageEvent = function (clientId, fn) {
+var listenStorageEvent = beginContext(function (clientId, fn) {
     swarm$1.loggerService.log("middleware listenStorageEvent", {
         clientId: clientId,
     });
-    validateClientId$8(clientId);
+    validateClientId$a(clientId);
     return swarm$1.busService.subscribe(clientId, "storage-bus", functoolsKit.queued(function (e) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
         switch (_a.label) {
             case 0: return [4 /*yield*/, fn(e)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     }); }); }));
-};
+});
 
-var validateClientId$7 = function (clientId) {
+var validateClientId$9 = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -13119,20 +14486,47 @@ var validateClientId$7 = function (clientId) {
  * @param {string} clientId - The ID of the client to subscribe to events for.
  * @param {(event: IBusEvent) => void} fn - The callback function to handle the event.
  */
-var listenSwarmEvent = function (clientId, fn) {
+var listenSwarmEvent = beginContext(function (clientId, fn) {
     swarm$1.loggerService.log("middleware listenSwarmEvent", {
         clientId: clientId,
     });
-    validateClientId$7(clientId);
+    validateClientId$9(clientId);
     return swarm$1.busService.subscribe(clientId, "swarm-bus", functoolsKit.queued(function (e) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
         switch (_a.label) {
             case 0: return [4 /*yield*/, fn(e)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     }); }); }));
+});
+
+var validateClientId$8 = function (clientId) {
+    if (clientId === "*") {
+        return;
+    }
+    if (!swarm$1.sessionValidationService.hasSession(clientId)) {
+        throw new Error("agent-swarm listenPolicyEvent session not found for clientId=".concat(clientId));
+    }
 };
+/**
+ * Hook to subscribe to swarm events for a specific client.
+ *
+ * @param {string} clientId - The ID of the client to subscribe to events for.
+ * @param {(event: IBusEvent) => void} fn - The callback function to handle the event.
+ */
+var listenPolicyEvent = beginContext(function (clientId, fn) {
+    swarm$1.loggerService.log("middleware listenPolicyEvent", {
+        clientId: clientId,
+    });
+    validateClientId$8(clientId);
+    return swarm$1.busService.subscribe(clientId, "policy-bus", functoolsKit.queued(function (e) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
+        switch (_a.label) {
+            case 0: return [4 /*yield*/, fn(e)];
+            case 1: return [2 /*return*/, _a.sent()];
+        }
+    }); }); }));
+});
 
-var validateClientId$6 = function (clientId) {
+var validateClientId$7 = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -13146,20 +14540,20 @@ var validateClientId$6 = function (clientId) {
  * @param {string} clientId - The ID of the client to subscribe to events for.
  * @param {function} fn - The callback function to handle the event.
  */
-var listenAgentEventOnce = function (clientId, filterFn, fn) {
+var listenAgentEventOnce = beginContext(function (clientId, filterFn, fn) {
     swarm$1.loggerService.log("middleware listenAgentEventOnce", {
         clientId: clientId,
     });
-    validateClientId$6(clientId);
+    validateClientId$7(clientId);
     return swarm$1.busService.once(clientId, "agent-bus", filterFn, functoolsKit.queued(function (e) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
         switch (_a.label) {
             case 0: return [4 /*yield*/, fn(e)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     }); }); }));
-};
+});
 
-var validateClientId$5 = function (clientId) {
+var validateClientId$6 = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -13173,20 +14567,20 @@ var validateClientId$5 = function (clientId) {
  * @param {string} clientId - The ID of the client to subscribe to.
  * @param {(event: IBusEvent) => void} fn - The callback function to handle the event.
  */
-var listenHistoryEventOnce = function (clientId, filterFn, fn) {
+var listenHistoryEventOnce = beginContext(function (clientId, filterFn, fn) {
     swarm$1.loggerService.log("middleware listenHistoryEventOnce", {
         clientId: clientId,
     });
-    validateClientId$5(clientId);
+    validateClientId$6(clientId);
     return swarm$1.busService.once(clientId, "history-bus", filterFn, functoolsKit.queued(function (e) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
         switch (_a.label) {
             case 0: return [4 /*yield*/, fn(e)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     }); }); }));
-};
+});
 
-var validateClientId$4 = function (clientId) {
+var validateClientId$5 = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -13200,20 +14594,20 @@ var validateClientId$4 = function (clientId) {
  * @param {string} clientId - The ID of the client to subscribe to session events for.
  * @param {function} fn - The callback function to handle the session events.
  */
-var listenSessionEventOnce = function (clientId, filterFn, fn) {
+var listenSessionEventOnce = beginContext(function (clientId, filterFn, fn) {
     swarm$1.loggerService.log("middleware listenSessionEventOnce", {
         clientId: clientId,
     });
-    validateClientId$4(clientId);
+    validateClientId$5(clientId);
     return swarm$1.busService.once(clientId, "session-bus", filterFn, functoolsKit.queued(function (e) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
         switch (_a.label) {
             case 0: return [4 /*yield*/, fn(e)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     }); }); }));
-};
+});
 
-var validateClientId$3 = function (clientId) {
+var validateClientId$4 = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -13227,20 +14621,20 @@ var validateClientId$3 = function (clientId) {
  * @param {string} clientId - The ID of the client to subscribe to.
  * @param {function} fn - The callback function to handle the event.
  */
-var listenStateEventOnce = function (clientId, filterFn, fn) {
+var listenStateEventOnce = beginContext(function (clientId, filterFn, fn) {
     swarm$1.loggerService.log("middleware listenStateEventOnce", {
         clientId: clientId,
     });
-    validateClientId$3(clientId);
+    validateClientId$4(clientId);
     return swarm$1.busService.once(clientId, "state-bus", filterFn, functoolsKit.queued(function (e) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
         switch (_a.label) {
             case 0: return [4 /*yield*/, fn(e)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     }); }); }));
-};
+});
 
-var validateClientId$2 = function (clientId) {
+var validateClientId$3 = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -13254,20 +14648,20 @@ var validateClientId$2 = function (clientId) {
  * @param {string} clientId - The ID of the client to subscribe to storage events for.
  * @param {function} fn - The callback function to handle the storage event.
  */
-var listenStorageEventOnce = function (clientId, filterFn, fn) {
+var listenStorageEventOnce = beginContext(function (clientId, filterFn, fn) {
     swarm$1.loggerService.log("middleware listenStorageEventOnce", {
         clientId: clientId,
     });
-    validateClientId$2(clientId);
+    validateClientId$3(clientId);
     return swarm$1.busService.once(clientId, "storage-bus", filterFn, functoolsKit.queued(function (e) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
         switch (_a.label) {
             case 0: return [4 /*yield*/, fn(e)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     }); }); }));
-};
+});
 
-var validateClientId$1 = function (clientId) {
+var validateClientId$2 = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -13281,20 +14675,20 @@ var validateClientId$1 = function (clientId) {
  * @param {string} clientId - The ID of the client to subscribe to events for.
  * @param {(event: IBusEvent) => void} fn - The callback function to handle the event.
  */
-var listenSwarmEventOnce = function (clientId, filterFn, fn) {
+var listenSwarmEventOnce = beginContext(function (clientId, filterFn, fn) {
     swarm$1.loggerService.log("middleware listenSwarmEventOnce", {
         clientId: clientId,
     });
-    validateClientId$1(clientId);
+    validateClientId$2(clientId);
     return swarm$1.busService.once(clientId, "swarm-bus", filterFn, functoolsKit.queued(function (e) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
         switch (_a.label) {
             case 0: return [4 /*yield*/, fn(e)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     }); }); }));
-};
+});
 
-var validateClientId = function (clientId) {
+var validateClientId$1 = function (clientId) {
     if (clientId === "*") {
         return;
     }
@@ -13308,18 +14702,45 @@ var validateClientId = function (clientId) {
  * @param {string} clientId - The ID of the client to subscribe to events for.
  * @param {function} fn - The callback function to handle the event.
  */
-var listenExecutionEventOnce = function (clientId, filterFn, fn) {
+var listenExecutionEventOnce = beginContext(function (clientId, filterFn, fn) {
     swarm$1.loggerService.log("middleware listenExecutionEventOnce", {
         clientId: clientId,
     });
-    validateClientId(clientId);
+    validateClientId$1(clientId);
     return swarm$1.busService.once(clientId, "execution-bus", filterFn, functoolsKit.queued(function (e) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
         switch (_a.label) {
             case 0: return [4 /*yield*/, fn(e)];
             case 1: return [2 /*return*/, _a.sent()];
         }
     }); }); }));
+});
+
+var validateClientId = function (clientId) {
+    if (clientId === "*") {
+        return;
+    }
+    if (!swarm$1.sessionValidationService.hasSession(clientId)) {
+        throw new Error("agent-swarm listenPolicyEventOnce session not found for clientId=".concat(clientId));
+    }
 };
+/**
+ * Hook to subscribe to swarm events for a specific client.
+ *
+ * @param {string} clientId - The ID of the client to subscribe to events for.
+ * @param {(event: IBusEvent) => void} fn - The callback function to handle the event.
+ */
+var listenPolicyEventOnce = beginContext(function (clientId, filterFn, fn) {
+    swarm$1.loggerService.log("middleware listenPolicyEventOnce", {
+        clientId: clientId,
+    });
+    validateClientId(clientId);
+    return swarm$1.busService.once(clientId, "policy-bus", filterFn, functoolsKit.queued(function (e) { return __awaiter(void 0, void 0, void 0, function () { return __generator(this, function (_a) {
+        switch (_a.label) {
+            case 0: return [4 /*yield*/, fn(e)];
+            case 1: return [2 /*return*/, _a.sent()];
+        }
+    }); }); }));
+});
 
 var METHOD_NAME_GET$3 = "StateUtils.getState";
 var METHOD_NAME_SET$1 = "StateUtils.setState";
@@ -13341,7 +14762,7 @@ var StateUtils = /** @class */ (function () {
          * @returns {Promise<T>} The state data.
          * @throws Will throw an error if the state is not registered in the agent.
          */
-        this.getState = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+        this.getState = beginContext(function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -13358,7 +14779,7 @@ var StateUtils = /** @class */ (function () {
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
         /**
          * Sets the state for a given client and state name.
          * @template T
@@ -13370,7 +14791,7 @@ var StateUtils = /** @class */ (function () {
          * @returns {Promise<void>}
          * @throws Will throw an error if the state is not registered in the agent.
          */
-        this.setState = function (dispatchFn, payload) { return __awaiter(_this, void 0, void 0, function () {
+        this.setState = beginContext(function (dispatchFn, payload) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -13393,7 +14814,7 @@ var StateUtils = /** @class */ (function () {
                     case 3: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
         /**
          * Set the state to initial value
          * @template T
@@ -13404,7 +14825,7 @@ var StateUtils = /** @class */ (function () {
          * @returns {Promise<void>}
          * @throws Will throw an error if the state is not registered in the agent.
          */
-        this.clearState = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+        this.clearState = beginContext(function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -13421,7 +14842,7 @@ var StateUtils = /** @class */ (function () {
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
     }
     return StateUtils;
 }());
@@ -13449,7 +14870,7 @@ var SharedStateUtils = /** @class */ (function () {
          * @returns {Promise<T>} The state data.
          * @throws Will throw an error if the state is not registered in the agent.
          */
-        this.getState = function (stateName) { return __awaiter(_this, void 0, void 0, function () {
+        this.getState = beginContext(function (stateName) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -13461,7 +14882,7 @@ var SharedStateUtils = /** @class */ (function () {
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
         /**
          * Sets the state for a given client and state name.
          * @template T
@@ -13470,7 +14891,7 @@ var SharedStateUtils = /** @class */ (function () {
          * @returns {Promise<void>}
          * @throws Will throw an error if the state is not registered in the agent.
          */
-        this.setState = function (dispatchFn, stateName) { return __awaiter(_this, void 0, void 0, function () {
+        this.setState = beginContext(function (dispatchFn, stateName) { return __awaiter(_this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
                 switch (_a.label) {
@@ -13488,7 +14909,7 @@ var SharedStateUtils = /** @class */ (function () {
                     case 3: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
         /**
          * Set the state to initial value
          * @template T
@@ -13496,7 +14917,7 @@ var SharedStateUtils = /** @class */ (function () {
          * @returns {Promise<void>}
          * @throws Will throw an error if the state is not registered in the agent.
          */
-        this.clearState = function (stateName) { return __awaiter(_this, void 0, void 0, function () {
+        this.clearState = beginContext(function (stateName) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -13508,7 +14929,7 @@ var SharedStateUtils = /** @class */ (function () {
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
     }
     return SharedStateUtils;
 }());
@@ -13537,7 +14958,7 @@ var StorageUtils = /** @class */ (function () {
          * @returns {Promise<T[]>} - A promise that resolves to an array of items.
          * @template T
          */
-        this.take = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+        this.take = beginContext(function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -13555,10 +14976,10 @@ var StorageUtils = /** @class */ (function () {
                             throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (take)"));
                         }
                         return [4 /*yield*/, swarm$1.storagePublicService.take(payload.search, payload.total, METHOD_NAME_TAKE$1, payload.clientId, payload.storageName, payload.score)];
-                    case 1: return [2 /*return*/, (_a.sent())];
+                    case 1: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
         /**
          * Upserts an item in the storage.
          * @param {T} item - The item to upsert.
@@ -13568,7 +14989,7 @@ var StorageUtils = /** @class */ (function () {
          * @returns {Promise<void>} - A promise that resolves when the operation is complete.
          * @template T
          */
-        this.upsert = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+        this.upsert = beginContext(function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -13587,7 +15008,7 @@ var StorageUtils = /** @class */ (function () {
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
         /**
          * Removes an item from the storage.
          * @param {IStorageData["id"]} itemId - The ID of the item to remove.
@@ -13596,7 +15017,7 @@ var StorageUtils = /** @class */ (function () {
          * @param {StorageName} storageName - The storage name.
          * @returns {Promise<void>} - A promise that resolves when the operation is complete.
          */
-        this.remove = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+        this.remove = beginContext(function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -13615,7 +15036,7 @@ var StorageUtils = /** @class */ (function () {
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
         /**
          * Gets an item from the storage.
          * @param {IStorageData["id"]} itemId - The ID of the item to get.
@@ -13625,7 +15046,7 @@ var StorageUtils = /** @class */ (function () {
          * @returns {Promise<T | null>} - A promise that resolves to the item or null if not found.
          * @template T
          */
-        this.get = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+        this.get = beginContext(function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -13640,10 +15061,10 @@ var StorageUtils = /** @class */ (function () {
                             throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (get)"));
                         }
                         return [4 /*yield*/, swarm$1.storagePublicService.get(payload.itemId, METHOD_NAME_GET$1, payload.clientId, payload.storageName)];
-                    case 1: return [2 /*return*/, (_a.sent())];
+                    case 1: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
         /**
          * Lists items from the storage.
          * @param {string} clientId - The client ID.
@@ -13653,7 +15074,7 @@ var StorageUtils = /** @class */ (function () {
          * @returns {Promise<T[]>} - A promise that resolves to an array of items.
          * @template T
          */
-        this.list = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+        this.list = beginContext(function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -13667,10 +15088,10 @@ var StorageUtils = /** @class */ (function () {
                             throw new Error("agent-swarm StorageUtils ".concat(payload.storageName, " not registered in ").concat(payload.agentName, " (list)"));
                         }
                         return [4 /*yield*/, swarm$1.storagePublicService.list(METHOD_NAME_LIST$1, payload.clientId, payload.storageName, payload.filter)];
-                    case 1: return [2 /*return*/, (_a.sent())];
+                    case 1: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
         /**
          * Clears the storage.
          * @param {string} clientId - The client ID.
@@ -13678,7 +15099,7 @@ var StorageUtils = /** @class */ (function () {
          * @param {StorageName} storageName - The storage name.
          * @returns {Promise<void>} - A promise that resolves when the operation is complete.
          */
-        this.clear = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+        this.clear = beginContext(function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -13695,7 +15116,7 @@ var StorageUtils = /** @class */ (function () {
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
     }
     return StorageUtils;
 }());
@@ -13718,7 +15139,7 @@ var SharedStorageUtils = /** @class */ (function () {
          * @returns {Promise<T[]>} - A promise that resolves to an array of items.
          * @template T
          */
-        this.take = function (payload) { return __awaiter(_this, void 0, void 0, function () {
+        this.take = beginContext(function (payload) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -13731,10 +15152,10 @@ var SharedStorageUtils = /** @class */ (function () {
                             });
                         swarm$1.storageValidationService.validate(payload.storageName, METHOD_NAME_TAKE);
                         return [4 /*yield*/, swarm$1.sharedStoragePublicService.take(payload.search, payload.total, METHOD_NAME_TAKE, payload.storageName, payload.score)];
-                    case 1: return [2 /*return*/, (_a.sent())];
+                    case 1: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
         /**
          * Upserts an item in the storage.
          * @param {T} item - The item to upsert.
@@ -13742,7 +15163,7 @@ var SharedStorageUtils = /** @class */ (function () {
          * @returns {Promise<void>} - A promise that resolves when the operation is complete.
          * @template T
          */
-        this.upsert = function (item, storageName) { return __awaiter(_this, void 0, void 0, function () {
+        this.upsert = beginContext(function (item, storageName) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -13756,14 +15177,14 @@ var SharedStorageUtils = /** @class */ (function () {
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
         /**
          * Removes an item from the storage.
          * @param {IStorageData["id"]} itemId - The ID of the item to remove.
          * @param {StorageName} storageName - The storage name.
          * @returns {Promise<void>} - A promise that resolves when the operation is complete.
          */
-        this.remove = function (itemId, storageName) { return __awaiter(_this, void 0, void 0, function () {
+        this.remove = beginContext(function (itemId, storageName) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -13777,7 +15198,7 @@ var SharedStorageUtils = /** @class */ (function () {
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
         /**
          * Gets an item from the storage.
          * @param {IStorageData["id"]} itemId - The ID of the item to get.
@@ -13785,7 +15206,7 @@ var SharedStorageUtils = /** @class */ (function () {
          * @returns {Promise<T | null>} - A promise that resolves to the item or null if not found.
          * @template T
          */
-        this.get = function (itemId, storageName) { return __awaiter(_this, void 0, void 0, function () {
+        this.get = beginContext(function (itemId, storageName) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -13796,10 +15217,10 @@ var SharedStorageUtils = /** @class */ (function () {
                             });
                         swarm$1.storageValidationService.validate(storageName, METHOD_NAME_GET);
                         return [4 /*yield*/, swarm$1.sharedStoragePublicService.get(itemId, METHOD_NAME_GET, storageName)];
-                    case 1: return [2 /*return*/, (_a.sent())];
+                    case 1: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
         /**
          * Lists items from the storage.
          * @param {StorageName} storageName - The storage name.
@@ -13807,7 +15228,7 @@ var SharedStorageUtils = /** @class */ (function () {
          * @returns {Promise<T[]>} - A promise that resolves to an array of items.
          * @template T
          */
-        this.list = function (storageName, filter) { return __awaiter(_this, void 0, void 0, function () {
+        this.list = beginContext(function (storageName, filter) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -13817,16 +15238,16 @@ var SharedStorageUtils = /** @class */ (function () {
                             });
                         swarm$1.storageValidationService.validate(storageName, METHOD_NAME_LIST);
                         return [4 /*yield*/, swarm$1.sharedStoragePublicService.list(METHOD_NAME_LIST, storageName, filter)];
-                    case 1: return [2 /*return*/, (_a.sent())];
+                    case 1: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
         /**
          * Clears the storage.
          * @param {StorageName} storageName - The storage name.
          * @returns {Promise<void>} - A promise that resolves when the operation is complete.
          */
-        this.clear = function (storageName) { return __awaiter(_this, void 0, void 0, function () {
+        this.clear = beginContext(function (storageName) { return __awaiter(_this, void 0, void 0, function () {
             return __generator(this, function (_a) {
                 switch (_a.label) {
                     case 0:
@@ -13839,7 +15260,7 @@ var SharedStorageUtils = /** @class */ (function () {
                     case 1: return [2 /*return*/, _a.sent()];
                 }
             });
-        }); };
+        }); });
     }
     return SharedStorageUtils;
 }());
@@ -13895,7 +15316,7 @@ var SchemaUtils = /** @class */ (function () {
          * @param {T} value - The value to write to the session memory.
          * @returns {T} The actual value from the session memory.
          */
-        this.writeSessionMemory = function (clientId, value) {
+        this.writeSessionMemory = beginContext(function (clientId, value) {
             GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
                 swarm$1.loggerService.log(METHOD_NAME_WRITE, {
                     clientId: clientId,
@@ -13903,7 +15324,7 @@ var SchemaUtils = /** @class */ (function () {
                 });
             swarm$1.sessionValidationService.validate(clientId, METHOD_NAME_WRITE);
             return swarm$1.memorySchemaService.writeValue(clientId, value);
-        };
+        });
         /**
          * Reads a value from the session memory for a given client.
          *
@@ -13911,14 +15332,14 @@ var SchemaUtils = /** @class */ (function () {
          * @param {string} clientId - The ID of the client.
          * @returns {T} The value read from the session memory.
          */
-        this.readSessionMemory = function (clientId) {
+        this.readSessionMemory = beginContext(function (clientId) {
             GLOBAL_CONFIG.CC_LOGGER_ENABLE_LOG &&
                 swarm$1.loggerService.log(METHOD_NAME_READ, {
                     clientId: clientId,
                 });
             swarm$1.sessionValidationService.validate(clientId, METHOD_NAME_READ);
             return swarm$1.memorySchemaService.readValue(clientId);
-        };
+        });
         /**
          * Serializes an object or an array of objects into a formatted string.
          *
@@ -13974,6 +15395,7 @@ exports.Logger = Logger;
 exports.LoggerAdapter = LoggerAdapter;
 exports.LoggerInstance = LoggerInstance;
 exports.MethodContextService = MethodContextService;
+exports.Policy = Policy;
 exports.Schema = Schema;
 exports.SharedState = SharedState;
 exports.SharedStorage = SharedStorage;
@@ -13982,10 +15404,12 @@ exports.Storage = Storage;
 exports.addAgent = addAgent;
 exports.addCompletion = addCompletion;
 exports.addEmbedding = addEmbedding;
+exports.addPolicy = addPolicy;
 exports.addState = addState;
 exports.addStorage = addStorage;
 exports.addSwarm = addSwarm;
 exports.addTool = addTool;
+exports.beginContext = beginContext;
 exports.cancelOutput = cancelOutput;
 exports.cancelOutputForce = cancelOutputForce;
 exports.changeToAgent = changeToAgent;
@@ -14033,6 +15457,8 @@ exports.listenExecutionEvent = listenExecutionEvent;
 exports.listenExecutionEventOnce = listenExecutionEventOnce;
 exports.listenHistoryEvent = listenHistoryEvent;
 exports.listenHistoryEventOnce = listenHistoryEventOnce;
+exports.listenPolicyEvent = listenPolicyEvent;
+exports.listenPolicyEventOnce = listenPolicyEventOnce;
 exports.listenSessionEvent = listenSessionEvent;
 exports.listenSessionEventOnce = listenSessionEventOnce;
 exports.listenStateEvent = listenStateEvent;