agent-swarm-kit 1.0.90 → 1.0.91

package/build/index.cjs

@@ -7341,7 +7341,9 @@ var AgentMetaService = /** @class */ (function () {
                 var childSeen_2 = new Set(seen).add(agentName);
                 return {
                     name: agentName,
-                    child: dependsOn.map(function (name) { return _this.makeAgentNodeCommon(name, childSeen_2); }),
+                    child: dependsOn.map(function (name) {
+                        return _this.makeAgentNodeCommon(name, childSeen_2);
+                    }),
                 };
             }
             else {
@@ -7355,12 +7357,17 @@ var AgentMetaService = /** @class */ (function () {
          * @param {AgentName} agentName - The name of the agent.
          * @returns {string} The UML representation of the agent's meta nodes.
          */
-        this.toUML = function (agentName) {
+        this.toUML = function (agentName, withSubtree) {
+            if (withSubtree === void 0) { withSubtree = false; }
             GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
                 _this.loggerService.info("agentMetaService toUML", {
                     agentName: agentName,
                 });
-            var rootNode = _this.makeAgentNode(agentName);
+            var subtree = new Set();
+            if (!withSubtree) {
+                subtree.add(agentName);
+            }
+            var rootNode = _this.makeAgentNode(agentName, subtree);
             return _this.serialize([rootNode]);
         };
     }
@@ -7520,13 +7527,14 @@ var METHOD_NAME$E = "function.dumpAgent";
  * @param {SwarmName} swarmName - The name of the swarm to be dumped.
  * @returns {string} The UML representation of the swarm.
  */
-var dumpAgent = function (agentName) {
+var dumpAgent = 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$E, {
             agentName: agentName,
         });
     swarm$1.agentValidationService.validate(agentName, METHOD_NAME$E);
-    return swarm$1.agentMetaService.toUML(agentName);
+    return swarm$1.agentMetaService.toUML(agentName, withSubtree);
 };
 
 var METHOD_NAME$D = "function.dumpSwarm";

package/build/index.mjs

@@ -7339,7 +7339,9 @@ var AgentMetaService = /** @class */ (function () {
                 var childSeen_2 = new Set(seen).add(agentName);
                 return {
                     name: agentName,
-                    child: dependsOn.map(function (name) { return _this.makeAgentNodeCommon(name, childSeen_2); }),
+                    child: dependsOn.map(function (name) {
+                        return _this.makeAgentNodeCommon(name, childSeen_2);
+                    }),
                 };
             }
             else {
@@ -7353,12 +7355,17 @@ var AgentMetaService = /** @class */ (function () {
          * @param {AgentName} agentName - The name of the agent.
          * @returns {string} The UML representation of the agent's meta nodes.
          */
-        this.toUML = function (agentName) {
+        this.toUML = function (agentName, withSubtree) {
+            if (withSubtree === void 0) { withSubtree = false; }
             GLOBAL_CONFIG.CC_LOGGER_ENABLE_INFO &&
                 _this.loggerService.info("agentMetaService toUML", {
                     agentName: agentName,
                 });
-            var rootNode = _this.makeAgentNode(agentName);
+            var subtree = new Set();
+            if (!withSubtree) {
+                subtree.add(agentName);
+            }
+            var rootNode = _this.makeAgentNode(agentName, subtree);
             return _this.serialize([rootNode]);
         };
     }
@@ -7518,13 +7525,14 @@ var METHOD_NAME$E = "function.dumpAgent";
  * @param {SwarmName} swarmName - The name of the swarm to be dumped.
  * @returns {string} The UML representation of the swarm.
  */
-var dumpAgent = function (agentName) {
+var dumpAgent = 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$E, {
             agentName: agentName,
         });
     swarm$1.agentValidationService.validate(agentName, METHOD_NAME$E);
-    return swarm$1.agentMetaService.toUML(agentName);
+    return swarm$1.agentMetaService.toUML(agentName, withSubtree);
 };
 
 var METHOD_NAME$D = "function.dumpSwarm";

package/package.json

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

package/types.d.ts

@@ -3066,7 +3066,7 @@ declare class AgentMetaService {
      * @param {AgentName} agentName - The name of the agent.
      * @returns {string} The UML representation of the agent's meta nodes.
      */
-    toUML: (agentName: AgentName) => string;
+    toUML: (agentName: AgentName, withSubtree?: boolean) => string;
 }
 
 /**
@@ -3130,13 +3130,19 @@ declare const swarm: {
     loggerService: LoggerService;
 };
 
+/**
+ * The config for UML generation
+ */
+interface IConfig {
+    withSubtree: boolean;
+}
 /**
  * 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.
  */
-declare const dumpAgent: (agentName: AgentName) => string;
+declare const dumpAgent: (agentName: AgentName, { withSubtree }?: Partial<IConfig>) => string;
 
 /**
  * Dumps the swarm information into PlantUML format.