patch

Author: tripolskypetr

package-lock.json

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

package.json

@@ -2,6 +2,7 @@ import { randomString } from "functools-kit";
 import swarm, { ExecutionContextService } from "../../lib";
 import { GLOBAL_CONFIG } from "../../config/params";
 import beginContext from "../../utils/beginContext";
+import { errorSubject } from "../../config/emitters";
 
 const METHOD_NAME = "function.target.executeForce";
 
@@ -32,24 +33,42 @@ const executeForceInternal = beginContext(
         swarm.perfService.startExecution(executionId, clientId, content.length);
         try {
           swarm.busService.commitExecutionBegin(clientId, { swarmName });
-          const result = await swarm.sessionPublicService.execute(
+   
+          let result = "";
+          let errorValue = null;
+
+          const unError = errorSubject.once(([errorClientId, error]) => {
+            if (clientId === errorClientId) {
+              errorValue = error;
+            }
+          });
+
+          result = await swarm.sessionPublicService.execute(
             content,
             "tool",
             METHOD_NAME,
             clientId,
             swarmName
           );
+
+          unError();
+
+          if (errorValue) {
+            throw errorValue;
+          }
+
           isFinished = swarm.perfService.endExecution(
             executionId,
             clientId,
             result.length
           );
-          swarm.busService.commitExecutionEnd(clientId, { swarmName });
+
           return result;
         } finally {
           if (!isFinished) {
             swarm.perfService.endExecution(executionId, clientId, 0);
           }
+          swarm.busService.commitExecutionEnd(clientId, { swarmName });
           swarm.executionValidationService.decrementCount(executionId, clientId, swarmName);
         }
       },

src/functions/target/executeForce.ts

@@ -3,6 +3,7 @@ import { AgentName } from "../../interfaces/Agent.interface";
 import { GLOBAL_CONFIG } from "../../config/params";
 import swarm, { ExecutionContextService } from "../../lib";
 import beginContext from "../../utils/beginContext";
+import { errorSubject } from "../../config/emitters";
 
 const METHOD_NAME = "function.target.runStateless";
 
@@ -57,26 +58,43 @@ const runStatelessInternal = beginContext(
             agentName,
             swarmName,
           });
-          const result = await swarm.sessionPublicService.run(
+
+          let result = "";
+          let errorValue = null;
+
+          const unError = errorSubject.once(([errorClientId, error]) => {
+            if (clientId === errorClientId) {
+              errorValue = error;
+            }
+          });
+
+          result = await swarm.sessionPublicService.run(
             content,
             METHOD_NAME,
             clientId,
             swarmName
           );
+
+          unError();
+
+          if (errorValue) {
+            throw errorValue;
+          }
+
           isFinished = swarm.perfService.endExecution(
             executionId,
             clientId,
             result.length
           );
-          swarm.busService.commitExecutionEnd(clientId, {
-            agentName,
-            swarmName,
-          });
           return result;
         } finally {
           if (!isFinished) {
             swarm.perfService.endExecution(executionId, clientId, 0);
           }
+          swarm.busService.commitExecutionEnd(clientId, {
+            agentName,
+            swarmName,
+          });
           swarm.executionValidationService.decrementCount(executionId, clientId, swarmName);
         }
       },

src/functions/target/runStateless.ts

@@ -2,6 +2,7 @@ import { randomString } from "functools-kit";
 import swarm, { ExecutionContextService } from "../../lib";
 import { GLOBAL_CONFIG } from "../../config/params";
 import beginContext from "../../utils/beginContext";
+import { errorSubject } from "../../config/emitters";
 
 const METHOD_NAME = "function.target.runStatelessForce";
 
@@ -32,23 +33,40 @@ const runStatelessForceInternal = beginContext(
         swarm.perfService.startExecution(executionId, clientId, content.length);
         try {
           swarm.busService.commitExecutionBegin(clientId, { swarmName });
-          const result = await swarm.sessionPublicService.run(
+
+          let result = "";
+          let errorValue = null;
+
+          const unError = errorSubject.once(([errorClientId, error]) => {
+            if (clientId === errorClientId) {
+              errorValue = error;
+            }
+          });
+          
+          result = await swarm.sessionPublicService.run(
             content,
             METHOD_NAME,
             clientId,
             swarmName
           );
+
+          unError();
+
+          if (errorValue) {
+            throw errorValue;
+          }
+
           isFinished = swarm.perfService.endExecution(
             executionId,
             clientId,
             result.length
           );
-          swarm.busService.commitExecutionEnd(clientId, { swarmName });
           return result;
         } finally {
           if (!isFinished) {
             swarm.perfService.endExecution(executionId, clientId, 0);
           }
+          swarm.busService.commitExecutionEnd(clientId, { swarmName });
           swarm.executionValidationService.decrementCount(executionId, clientId, swarmName);
         }
       },