@n1k1t/pipelain 0.4.13 → 0.4.15

package/lib/src/models/pipeline/steps/swarm.js

@@ -11,9 +11,13 @@ class PipelineSwarmStepCompiler extends model_1.PipelineStepCompiler {
         this.definition = definition;
         this.type = 'swarm';
     }
-    /** Provides subtask executors */
-    subtasks(predicate) {
-        this.definition.subtasks = predicate;
+    /** Provides a list of subtask executors */
+    list(predicate) {
+        return this.subtasks(predicate);
+    }
+    /** Provides a collection of subtask executors keyed by name */
+    map(predicate) {
+        this.definition.map = predicate;
         return this;
     }
     /** Provides a bottle neck limit for parallel executions */
@@ -22,17 +26,26 @@ class PipelineSwarmStepCompiler extends model_1.PipelineStepCompiler {
         return this;
     }
     compile(provided) {
-        if (!this.definition.subtasks) {
+        if (!this.definition.list && !this.definition.map) {
             throw new errors_1.PipelineStepCompilationError(this.type, '[subtasks] are empty');
         }
+        if (this.definition.list && this.definition.map) {
+            throw new errors_1.PipelineStepCompilationError(this.type, '[map] and [list] are conflicted');
+        }
         return new PipelineSwarmStep(this.type, {
             title: this.definition.title,
-            subtasks: this.definition.subtasks,
             limit: this.definition.limit,
+            list: this.definition.list,
+            map: this.definition.map,
             pipeline: provided.pipeline,
             parent: provided.parent,
         });
     }
+    /** @deprecated Provides subtask executors (use `list` method instead) */
+    subtasks(predicate) {
+        this.definition.list = predicate;
+        return this;
+    }
     static build(title) {
         return new PipelineSwarmStepCompiler({ title });
     }
@@ -41,22 +54,30 @@ exports.PipelineSwarmStepCompiler = PipelineSwarmStepCompiler;
 class PipelineSwarmStep extends model_1.PipelineStep {
     async run(parameters) {
         const meta = (0, utils_1.buildMetaManager)();
+        const exec = async (subtask) => {
+            const compiled = await subtask.compile({ pipeline: this.pipeline, parent: this });
+            return compiled instanceof model_1.PipelineStep
+                ? compiled.run(parameters)
+                : compiled.run(undefined);
+        };
         this.pipeline.session.emit('step:run', { step: this, meta: meta.init() });
         try {
-            const subtasks = typeof this.definition.subtasks === 'function'
-                ? await this.definition.subtasks(parameters)
-                : this.definition.subtasks;
-            const handled = await (0, utils_2.chunkify)(subtasks, this.definition.limit ?? Infinity, async (subtask) => {
-                const compiled = await subtask.compile({
-                    pipeline: this.pipeline,
-                    parent: this,
-                });
-                return compiled instanceof model_1.PipelineStep
-                    ? compiled.run(parameters)
-                    : compiled.run(undefined);
-            });
+            if (this.definition.map) {
+                const map = typeof this.definition.map === 'function'
+                    ? await this.definition.map(parameters)
+                    : this.definition.map;
+                const handled = await (0, utils_2.chunkify)(Object.keys(map), this.definition.limit ?? Infinity, (key) => exec(map[key]));
+                const result = Object.fromEntries(handled.map((subtask) => [subtask.payload, subtask.result]));
+                this.pipeline.session.emit('step:run', { step: this, meta: meta.done() });
+                return result;
+            }
+            const subtasks = typeof this.definition.list === 'function'
+                ? await this.definition.list(parameters)
+                : this.definition.list ?? [];
+            const handled = await (0, utils_2.chunkify)(subtasks, this.definition.limit ?? Infinity, exec);
+            const result = handled.map((subtask) => subtask.result);
             this.pipeline.session.emit('step:run', { step: this, meta: meta.done() });
-            return handled.map((subtask) => subtask.result);
+            return result;
         }
         catch (error) {
             this.pipeline.session.emit('step:run', { step: this, meta: meta.error() });

package/lib/src/models/pipeline/steps/swarm.js.map

@@ -1 +1 @@
-{"version":3,"file":"swarm.js","sourceRoot":"","sources":["../../../../../src/models/pipeline/steps/swarm.ts"],"names":[],"mappings":";;;AAIA,mCAA2G;AAC3G,sCAAyD;AAEzD,0CAAkD;AAClD,0CAA0C;AAU1C,MAAa,yBAGX,SAAQ,4BAA6B;IAGrC,YAAsB,UAAgD;QACpE,KAAK,EAAE,CAAC;QADY,eAAU,GAAV,UAAU,CAAsC;QAF9D,SAAI,GAAwC,OAAO,CAAC;IAI5D,CAAC;IAED,iCAAiC;IAC1B,QAAQ,CAGb,SAAgE;QAChE,IAAI,CAAC,UAAU,CAAC,QAAQ,GAAG,SAAS,CAAC;QACrC,OAAuB,IAAI,CAAC;IAC9B,CAAC;IAED,2DAA2D;IACpD,KAAK,CAAC,KAAa;QACxB,IAAI,CAAC,UAAU,CAAC,KAAK,GAAG,KAAK,CAAC;QAC9B,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,OAAO,CAAC,QAA6B;QAC1C,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC;YAC9B,MAAM,IAAI,qCAA4B,CAAC,IAAI,CAAC,IAAI,EAAE,sBAAsB,CAAC,CAAC;QAC5E,CAAC;QAED,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,IAAI,EAAE;YACtC,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,KAAK;YAE5B,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,QAAQ;YAClC,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,KAAK;YAE5B,QAAQ,EAAE,QAAQ,CAAC,QAAQ;YAC3B,MAAM,EAAE,QAAQ,CAAC,MAAM;SACxB,CAAC,CAAC;IACL,CAAC;IAED,MAAM,CAAC,KAAK,CACV,KAAc;QAEd,OAAO,IAAI,yBAAyB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAClD,CAAC;CACF;AA9CD,8DA8CC;AAED,MAAa,iBAGX,SAAQ,oBAA2E;IAC5E,KAAK,CAAC,GAAG,CAAC,UAA8C;QAC7D,MAAM,IAAI,GAAG,IAAA,wBAAgB,GAAE,CAAC;QAChC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAE1E,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,OAAO,IAAI,CAAC,UAAU,CAAC,QAAQ,KAAK,UAAU;gBAC7D,CAAC,CAAC,MAAM,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC;gBAC5C,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;YAE7B,MAAM,OAAO,GAAG,MAAM,IAAA,gBAAQ,EAAC,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,KAAK,IAAI,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;gBAC5F,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC;oBACrC,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,MAAM,EAAE,IAAI;iBACb,CAAC,CAAC;gBAEH,OAAO,QAAQ,YAAY,oBAAY;oBACrC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC;oBAC1B,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAC9B,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAE1E,OAAgB,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC3D,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YAC3E,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF;AAhCD,8CAgCC"}
+{"version":3,"file":"swarm.js","sourceRoot":"","sources":["../../../../../src/models/pipeline/steps/swarm.ts"],"names":[],"mappings":";;;AAIA,mCAA2G;AAC3G,sCAAyD;AAEzD,0CAAkD;AAClD,0CAA0C;AAa1C,MAAa,yBAIX,SAAQ,4BAA6B;IAGrC,YAAsB,UAAgD;QACpE,KAAK,EAAE,CAAC;QADY,eAAU,GAAV,UAAU,CAAsC;QAF9D,SAAI,GAAwC,OAAO,CAAC;IAI5D,CAAC;IAED,2CAA2C;IACpC,IAAI,CAGT,SAAgE;QAChE,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAClC,CAAC;IAED,+DAA+D;IACxD,GAAG,CAKR,SAA4D;QAC5D,IAAI,CAAC,UAAU,CAAC,GAAG,GAAG,SAAS,CAAC;QAChC,OAAuB,IAAI,CAAC;IAC9B,CAAC;IAED,2DAA2D;IACpD,KAAK,CAAC,KAAa;QACxB,IAAI,CAAC,UAAU,CAAC,KAAK,GAAG,KAAK,CAAC;QAC9B,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,OAAO,CAAC,QAA6B;QAC1C,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC;YAClD,MAAM,IAAI,qCAA4B,CAAC,IAAI,CAAC,IAAI,EAAE,sBAAsB,CAAC,CAAC;QAC5E,CAAC;QACD,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC;YAChD,MAAM,IAAI,qCAA4B,CAAC,IAAI,CAAC,IAAI,EAAE,iCAAiC,CAAC,CAAC;QACvF,CAAC;QAED,OAAO,IAAI,iBAAiB,CAAC,IAAI,CAAC,IAAI,EAAE;YACtC,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,KAAK;YAC5B,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,KAAK;YAE5B,IAAI,EAAE,IAAI,CAAC,UAAU,CAAC,IAAI;YAC1B,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,GAAG;YAExB,QAAQ,EAAE,QAAQ,CAAC,QAAQ;YAC3B,MAAM,EAAE,QAAQ,CAAC,MAAM;SACxB,CAAC,CAAC;IACL,CAAC;IAED,yEAAyE;IAClE,QAAQ,CAGb,SAAgE;QAChE,IAAI,CAAC,UAAU,CAAC,IAAI,GAAG,SAAS,CAAC;QACjC,OAAuB,IAAI,CAAC;IAC9B,CAAC;IAED,MAAM,CAAC,KAAK,CAGV,KAAc;QACd,OAAO,IAAI,yBAAyB,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAClD,CAAC;CACF;AAvED,8DAuEC;AAED,MAAa,iBAGX,SAAQ,oBAA2E;IAC5E,KAAK,CAAC,GAAG,CAAC,UAA8C;QAC7D,MAAM,IAAI,GAAG,IAAA,wBAAgB,GAAE,CAAC;QAChC,MAAM,IAAI,GAAG,KAAK,EAAE,OAAgD,EAAE,EAAE;YACtE,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAElF,OAAO,QAAQ,YAAY,oBAAY;gBACrC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC;gBAC1B,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC9B,CAAC,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAE1E,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC;gBACxB,MAAM,GAAG,GAAG,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,KAAK,UAAU;oBACnD,CAAC,CAAC,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC;oBACvC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBAExB,MAAM,OAAO,GAAG,MAAM,IAAA,gBAAQ,EAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,KAAK,IAAI,QAAQ,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAC7G,MAAM,MAAM,GAAY,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBAExG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAC1E,OAAO,MAAM,CAAC;YAChB,CAAC;YAED,MAAM,QAAQ,GAAG,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,KAAK,UAAU;gBACzD,CAAC,CAAC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC;gBACxC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,IAAI,EAAE,CAAC;YAE/B,MAAM,OAAO,GAAG,MAAM,IAAA,gBAAQ,EAAC,QAAQ,EAAE,IAAI,CAAC,UAAU,CAAC,KAAK,IAAI,QAAQ,EAAE,IAAI,CAAC,CAAC;YAClF,MAAM,MAAM,GAAY,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAEjE,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1E,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YAC3E,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF;AA3CD,8CA2CC"}

package/package.json

@@ -1,16 +1,17 @@
 {
   "name": "@n1k1t/pipelain",
-  "version": "0.4.13",
+  "version": "0.4.15",
   "description": "Powerful utility to build and execute type-safe AI pipelines with structured outputs and tool integration",
   "main": "lib/src/index.js",
   "types": "lib/src/index.d.ts",
   "scripts": {
     "test": "NODE_ENV=test jest --silent",
     "test:coverage": "NODE_ENV=test jest --coverage",
-    "start:dev:main": "npx ts-node -r ./src/opentelemetry.ts test/index.ts",
-    "start:dev:tool": "npx ts-node -r ./src/opentelemetry.ts test/tool.ts",
-    "start:dev:skill": "npx ts-node -r ./src/opentelemetry.ts test/skill.ts",
-    "start:dev:report": "npx ts-node -r ./src/opentelemetry.ts test/report.ts",
+    "run:pipelain:main": "npx ts-node test/index.ts",
+    "run:pipelain:tool": "npx ts-node test/tool.ts",
+    "run:pipelain:skill": "npx ts-node test/skill.ts",
+    "run:pipelain:report": "npx ts-node test/report.ts",
+    "run:pipelain:fallback": "npx ts-node test/fallback.ts",
     "build": "rm -rf lib && npx tsc",
     "build:check": "npx tsc --noEmit",
     "prepare": "husky",
@@ -90,14 +91,14 @@
     "typescript": "5.7.2"
   },
   "dependencies": {
-    "@ai-sdk/anthropic": "4.0.25",
-    "@ai-sdk/google": "4.0.29",
-    "@ai-sdk/mcp": "2.0.20",
-    "@ai-sdk/mistral": "4.0.18",
-    "@ai-sdk/openai": "4.0.25",
-    "@ai-sdk/openai-compatible": "3.0.18",
+    "@ai-sdk/anthropic": "4.0.49",
+    "@ai-sdk/google": "4.0.63",
+    "@ai-sdk/mcp": "2.0.44",
+    "@ai-sdk/mistral": "4.0.39",
+    "@ai-sdk/openai": "4.0.58",
+    "@ai-sdk/openai-compatible": "3.0.43",
     "@ast-grep/napi": "0.42.0",
-    "ai": "7.0.44",
+    "ai": "7.0.92",
     "colors": "1.4.0",
     "commander": "9.4.0",
     "dayjs": "1.11.21",