agent-swarm-kit 1.1.96 → 1.1.97

package/build/index.cjs

@@ -7346,6 +7346,7 @@ class ClientSession {
         const outputAwaiter = this.params.swarm.waitForOutput();
         agent.execute(message, mode);
         const output = await outputAwaiter;
+        await swarm$1.executionValidationService.flushCount(this.params.clientId, this.params.swarmName);
         if (await functoolsKit.not(this.params.policy.validateOutput(output, this.params.clientId, this.params.swarmName))) {
             GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                 this.params.logger.debug(`ClientSession clientId=${this.params.clientId} execution output emit canceled due to the banhammer of a client`, {
@@ -7610,6 +7611,7 @@ class ClientSession {
             clientId: this.params.clientId,
         });
         this._notifySubject.subscribe(async (data) => {
+            await swarm$1.executionValidationService.flushCount(this.params.clientId, this.params.swarmName);
             await connector({
                 data,
                 agentName: await this.params.swarm.getAgentName(),
@@ -7620,6 +7622,7 @@ class ClientSession {
             GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                 this.params.logger.debug(`ClientSession clientId=${this.params.clientId} connect call`);
             const data = await this.execute(incoming.data, "user");
+            await swarm$1.executionValidationService.flushCount(this.params.clientId, this.params.swarmName);
             if (!data) {
                 return "";
             }
@@ -17847,8 +17850,30 @@ class ExecutionValidationService {
             if (!this.sessionValidationService.hasSession(clientId)) {
                 return;
             }
-            const { executionSet } = this.getExecutionCount(clientId, swarmName);
+            const { executionSet, executionIgnore } = this.getExecutionCount(clientId, swarmName);
             executionSet.delete(executionId);
+            executionIgnore.add(executionId);
+        };
+        /**
+         * Clears all tracked execution IDs for a specific client and swarm.
+         * This effectively resets the execution count for the given client and swarm context,
+         * but does not remove the memoized entry itself.
+         *
+         * @param {string} clientId - The unique identifier for the client.
+         * @param {SwarmName} swarmName - The name of the swarm associated with the client.
+         * @returns {void}
+         */
+        this.flushCount = (clientId, swarmName) => {
+            GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                this.loggerService.info("executionValidationService dispose", {
+                    clientId,
+                    swarmName,
+                });
+            if (!this.sessionValidationService.hasSession(clientId)) {
+                return;
+            }
+            const { executionSet } = this.getExecutionCount(clientId, swarmName);
+            executionSet.clear();
         };
         /**
          * Clears the memoized execution count for a specific client and swarm.

package/build/index.mjs

@@ -7344,6 +7344,7 @@ class ClientSession {
         const outputAwaiter = this.params.swarm.waitForOutput();
         agent.execute(message, mode);
         const output = await outputAwaiter;
+        await swarm$1.executionValidationService.flushCount(this.params.clientId, this.params.swarmName);
         if (await not(this.params.policy.validateOutput(output, this.params.clientId, this.params.swarmName))) {
             GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                 this.params.logger.debug(`ClientSession clientId=${this.params.clientId} execution output emit canceled due to the banhammer of a client`, {
@@ -7608,6 +7609,7 @@ class ClientSession {
             clientId: this.params.clientId,
         });
         this._notifySubject.subscribe(async (data) => {
+            await swarm$1.executionValidationService.flushCount(this.params.clientId, this.params.swarmName);
             await connector({
                 data,
                 agentName: await this.params.swarm.getAgentName(),
@@ -7618,6 +7620,7 @@ class ClientSession {
             GLOBAL_CONFIG.CC_LOGGER_ENABLE_DEBUG &&
                 this.params.logger.debug(`ClientSession clientId=${this.params.clientId} connect call`);
             const data = await this.execute(incoming.data, "user");
+            await swarm$1.executionValidationService.flushCount(this.params.clientId, this.params.swarmName);
             if (!data) {
                 return "";
             }
@@ -17845,8 +17848,30 @@ class ExecutionValidationService {
             if (!this.sessionValidationService.hasSession(clientId)) {
                 return;
             }
-            const { executionSet } = this.getExecutionCount(clientId, swarmName);
+            const { executionSet, executionIgnore } = this.getExecutionCount(clientId, swarmName);
             executionSet.delete(executionId);
+            executionIgnore.add(executionId);
+        };
+        /**
+         * Clears all tracked execution IDs for a specific client and swarm.
+         * This effectively resets the execution count for the given client and swarm context,
+         * but does not remove the memoized entry itself.
+         *
+         * @param {string} clientId - The unique identifier for the client.
+         * @param {SwarmName} swarmName - The name of the swarm associated with the client.
+         * @returns {void}
+         */
+        this.flushCount = (clientId, swarmName) => {
+            GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
+                this.loggerService.info("executionValidationService dispose", {
+                    clientId,
+                    swarmName,
+                });
+            if (!this.sessionValidationService.hasSession(clientId)) {
+                return;
+            }
+            const { executionSet } = this.getExecutionCount(clientId, swarmName);
+            executionSet.clear();
         };
         /**
          * Clears the memoized execution count for a specific client and swarm.

package/package.json

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

package/types.d.ts

@@ -10665,6 +10665,16 @@ declare class ExecutionValidationService {
      * @returns {void}
      */
     decrementCount: (executionId: string, clientId: string, swarmName: SwarmName) => void;
+    /**
+     * Clears all tracked execution IDs for a specific client and swarm.
+     * This effectively resets the execution count for the given client and swarm context,
+     * but does not remove the memoized entry itself.
+     *
+     * @param {string} clientId - The unique identifier for the client.
+     * @param {SwarmName} swarmName - The name of the swarm associated with the client.
+     * @returns {void}
+     */
+    flushCount: (clientId: string, swarmName: SwarmName) => void;
     /**
      * Clears the memoized execution count for a specific client and swarm.
      * @param {string} clientId - The unique identifier for the client.