@codemation/core-nodes 0.2.0 → 0.4.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@codemation/core-nodes",
-  "version": "0.2.0",
+  "version": "0.4.0",
   "publishConfig": {
     "access": "public"
   },
@@ -31,7 +31,7 @@
     "@langchain/core": "^1.1.31",
     "@langchain/openai": "^1.2.12",
     "lucide-react": "^0.577.0",
-    "@codemation/core": "0.6.0"
+    "@codemation/core": "0.7.0"
   },
   "devDependencies": {
     "@types/node": "^25.3.5",

package/src/nodes/AIAgentNode.ts

@@ -68,7 +68,7 @@ export class AIAgentNode implements RunnableNode<AIAgent<any, any>> {
   outputPorts = ["main"] as const;
   /**
    * Engine validates {@link RunnableNodeConfig.inputSchema} (Zod) on {@code item.json} before enqueue, then resolves
-   * per-item **`itemValue`** leaves on config before {@link #execute}. Prefer modeling prompts as
+   * per-item **`itemExpr`** leaves on config before {@link #execute}. Prefer modeling prompts as
    * {@code { messages: [{ role, content }, ...] }} (on input or config) so persisted inputs are visible in the UI.
    */
   readonly inputSchema = z.unknown();

package/src/nodes/NodeBackedToolRuntime.ts

@@ -14,7 +14,7 @@ import {
   CoreTokens,
   inject,
   injectable,
-  ItemValueResolver,
+  ItemExprResolver,
   NodeOutputNormalizer,
   RunnableOutputBehaviorResolver,
 } from "@codemation/core";
@@ -25,8 +25,8 @@ export class NodeBackedToolRuntime {
   constructor(
     @inject(CoreTokens.NodeResolver)
     private readonly nodeResolver: NodeResolver,
-    @inject(ItemValueResolver)
-    private readonly itemValueResolver: ItemValueResolver,
+    @inject(ItemExprResolver)
+    private readonly itemExprResolver: ItemExprResolver,
     @inject(NodeOutputNormalizer)
     private readonly outputNormalizer: NodeOutputNormalizer,
     @inject(RunnableOutputBehaviorResolver)
@@ -77,7 +77,7 @@ export class NodeBackedToolRuntime {
       const inputSchema = runnable.inputSchema ?? runnableConfig.inputSchema ?? z.unknown();
       const parsed = inputSchema.parse(nodeInput.json);
       const items = [nodeInput];
-      const resolvedCtx = await this.itemValueResolver.resolveConfigForItem(ctx, nodeInput, 0, items);
+      const resolvedCtx = await this.itemExprResolver.resolveConfigForItem(ctx, nodeInput, 0, items);
       const execArgs: RunnableNodeExecuteArgs = {
         input: parsed,
         item: nodeInput,

package/src/workflowAuthoring/WorkflowBranchBuilder.types.ts

@@ -1,6 +1,8 @@
 import type {
   AnyRunnableNodeConfig,
+  CredentialJsonRecord,
   DefinedNode,
+  DefinedNodeConfigInput,
   Item,
   Items,
   NodeExecutionContext,
@@ -19,6 +21,11 @@ import { WorkflowAgentNodeFactory } from "./WorkflowAgentNodeFactory.types";
 import { WorkflowDefinedNodeResolver } from "./WorkflowDefinedNodeResolver.types";
 import { WorkflowDurationParser } from "./WorkflowDurationParser.types";
 
+type WorkflowMapCallback<TCurrentJson, TNextJson> = (
+  item: Item<TCurrentJson>,
+  ctx: NodeExecutionContext<MapData<TCurrentJson, TNextJson>>,
+) => TNextJson;
+
 export class WorkflowBranchBuilder<TCurrentJson> {
   constructor(private readonly steps: ReadonlyArray<AnyRunnableNodeConfig> = []) {}
 
@@ -28,22 +35,20 @@ export class WorkflowBranchBuilder<TCurrentJson> {
     return new WorkflowBranchBuilder<RunnableNodeOutputJson<TConfig>>([...this.steps, config]);
   }
 
-  map<TNextJson>(mapper: (item: TCurrentJson) => TNextJson): WorkflowBranchBuilder<TNextJson>;
+  map<TNextJson>(mapper: WorkflowMapCallback<TCurrentJson, TNextJson>): WorkflowBranchBuilder<TNextJson>;
   map<TNextJson>(
     name: string,
-    mapper: (item: TCurrentJson) => TNextJson,
+    mapper: WorkflowMapCallback<TCurrentJson, TNextJson>,
     options?: MapDataOptions,
   ): WorkflowBranchBuilder<TNextJson>;
   map<TNextJson>(
-    nameOrMapper: string | ((item: TCurrentJson) => TNextJson),
-    mapperOrUndefined?: (item: TCurrentJson) => TNextJson,
+    nameOrMapper: string | WorkflowMapCallback<TCurrentJson, TNextJson>,
+    mapperOrUndefined?: WorkflowMapCallback<TCurrentJson, TNextJson>,
     options?: MapDataOptions,
   ): WorkflowBranchBuilder<TNextJson> {
     const name = typeof nameOrMapper === "string" ? nameOrMapper : "Map data";
     const mapper = typeof nameOrMapper === "string" ? mapperOrUndefined! : nameOrMapper;
-    return this.then(
-      new MapData<TCurrentJson, TNextJson>(name, (item) => mapper(item.json as TCurrentJson), options),
-    ) as WorkflowBranchBuilder<TNextJson>;
+    return this.then(new MapData<TCurrentJson, TNextJson>(name, mapper, options)) as WorkflowBranchBuilder<TNextJson>;
   }
 
   wait(duration: number | string): WorkflowBranchBuilder<TCurrentJson>;
@@ -172,18 +177,18 @@ export class WorkflowBranchBuilder<TCurrentJson> {
     >;
   }
 
-  node<TConfig extends Record<string, unknown>, TOutputJson>(
-    definitionOrKey: DefinedNode<string, TConfig, TCurrentJson, TOutputJson> | string,
-    config: TConfig,
+  node<TConfig extends CredentialJsonRecord, TInputJson, TOutputJson>(
+    definitionOrKey: DefinedNode<string, TConfig, TInputJson, TOutputJson> | string,
+    config: DefinedNodeConfigInput<TConfig, TCurrentJson>,
     name?: string,
     id?: string,
-  ): WorkflowBranchBuilder<TOutputJson> {
+  ): TCurrentJson extends TInputJson ? WorkflowBranchBuilder<TOutputJson> : never {
     const definition = WorkflowDefinedNodeResolver.resolve(
       definitionOrKey as DefinedNode<string, Record<string, unknown>, unknown, unknown> | string,
-    ) as DefinedNode<string, TConfig, TCurrentJson, TOutputJson>;
+    ) as DefinedNode<string, TConfig, TInputJson, TOutputJson>;
     return this.then(
-      definition.create(config, name, id) as RunnableNodeConfig<TCurrentJson, TOutputJson>,
-    ) as WorkflowBranchBuilder<TOutputJson>;
+      definition.create(config, name, id) as unknown as RunnableNodeConfig<TCurrentJson, TOutputJson>,
+    ) as TCurrentJson extends TInputJson ? WorkflowBranchBuilder<TOutputJson> : never;
   }
 
   getSteps(): ReadonlyArray<AnyRunnableNodeConfig> {

package/src/workflowAuthoring/WorkflowChain.types.ts

@@ -1,5 +1,7 @@
 import type {
+  CredentialJsonRecord,
   DefinedNode,
+  DefinedNodeConfigInput,
   Item,
   Items,
   NodeExecutionContext,
@@ -29,6 +31,18 @@ type BranchCallback<TCurrentJson, TNextJson> = (
 ) => WorkflowBranchBuilder<TNextJson>;
 type RouteBranchCallback<TCurrentJson, TNextJson> = (branch: WorkflowChain<TCurrentJson>) => WorkflowChain<TNextJson>;
 type BranchOutputMatch<TLeft, TRight> = [TLeft] extends [TRight] ? ([TRight] extends [TLeft] ? true : false) : false;
+type WorkflowMapCallback<TCurrentJson, TNextJson> = (
+  item: Item<TCurrentJson>,
+  ctx: NodeExecutionContext<MapData<TCurrentJson, TNextJson>>,
+) => TNextJson;
+type WorkflowIfPredicate<TCurrentJson> = (
+  item: Item<TCurrentJson>,
+  ctx: NodeExecutionContext<If<TCurrentJson>>,
+) => boolean;
+type WorkflowSwitchCaseKeyResolver<TCurrentJson> = (
+  item: Item<TCurrentJson>,
+  ctx: NodeExecutionContext<Switch<TCurrentJson>>,
+) => string | Promise<string>;
 
 export class WorkflowChain<TCurrentJson> {
   constructor(private readonly chain: ChainCursor<TCurrentJson>) {}
@@ -39,22 +53,20 @@ export class WorkflowChain<TCurrentJson> {
     return new WorkflowChain(this.chain.then(config));
   }
 
-  map<TNextJson>(mapper: (item: TCurrentJson) => TNextJson): WorkflowChain<TNextJson>;
+  map<TNextJson>(mapper: WorkflowMapCallback<TCurrentJson, TNextJson>): WorkflowChain<TNextJson>;
   map<TNextJson>(
     name: string,
-    mapper: (item: TCurrentJson) => TNextJson,
+    mapper: WorkflowMapCallback<TCurrentJson, TNextJson>,
     options?: MapDataOptions,
   ): WorkflowChain<TNextJson>;
   map<TNextJson>(
-    nameOrMapper: string | ((item: TCurrentJson) => TNextJson),
-    mapperOrUndefined?: (item: TCurrentJson) => TNextJson,
+    nameOrMapper: string | WorkflowMapCallback<TCurrentJson, TNextJson>,
+    mapperOrUndefined?: WorkflowMapCallback<TCurrentJson, TNextJson>,
     options?: MapDataOptions,
   ): WorkflowChain<TNextJson> {
     const name = typeof nameOrMapper === "string" ? nameOrMapper : "Map data";
     const mapper = typeof nameOrMapper === "string" ? mapperOrUndefined! : nameOrMapper;
-    return this.then(
-      new MapData<TCurrentJson, TNextJson>(name, (item) => mapper(item.json as TCurrentJson), options),
-    ) as WorkflowChain<TNextJson>;
+    return this.then(new MapData<TCurrentJson, TNextJson>(name, mapper, options)) as WorkflowChain<TNextJson>;
   }
 
   wait(duration: number | string): WorkflowChain<TCurrentJson>;
@@ -190,7 +202,7 @@ export class WorkflowChain<TCurrentJson> {
   }
 
   if<TBranchJson>(
-    predicate: (item: TCurrentJson) => boolean,
+    predicate: WorkflowIfPredicate<TCurrentJson>,
     branches: Readonly<{
       true?: BranchCallback<TCurrentJson, TBranchJson>;
       false?: BranchCallback<TCurrentJson, TBranchJson>;
@@ -198,16 +210,16 @@ export class WorkflowChain<TCurrentJson> {
   ): WorkflowChain<TBranchJson>;
   if<TBranchJson>(
     name: string,
-    predicate: (item: TCurrentJson) => boolean,
+    predicate: WorkflowIfPredicate<TCurrentJson>,
     branches: Readonly<{
       true?: BranchCallback<TCurrentJson, TBranchJson>;
       false?: BranchCallback<TCurrentJson, TBranchJson>;
     }>,
   ): WorkflowChain<TBranchJson>;
   if<TTrueJson, TFalseJson>(
-    nameOrPredicate: string | ((item: TCurrentJson) => boolean),
+    nameOrPredicate: string | WorkflowIfPredicate<TCurrentJson>,
     predicateOrBranches:
-      | ((item: TCurrentJson) => boolean)
+      | WorkflowIfPredicate<TCurrentJson>
       | Readonly<{ true?: BranchCallback<TCurrentJson, TTrueJson>; false?: BranchCallback<TCurrentJson, TFalseJson> }>,
     branchesOrUndefined?: Readonly<{
       true?: BranchCallback<TCurrentJson, TTrueJson>;
@@ -216,12 +228,14 @@ export class WorkflowChain<TCurrentJson> {
   ): WorkflowChain<BranchOutputMatch<TTrueJson, TFalseJson> extends true ? TTrueJson : never> {
     const name = typeof nameOrPredicate === "string" ? nameOrPredicate : "If";
     const predicate =
-      typeof nameOrPredicate === "string" ? (predicateOrBranches as (item: TCurrentJson) => boolean) : nameOrPredicate;
+      typeof nameOrPredicate === "string"
+        ? (predicateOrBranches as WorkflowIfPredicate<TCurrentJson>)
+        : nameOrPredicate;
     const branches = (typeof nameOrPredicate === "string" ? branchesOrUndefined : predicateOrBranches) as Readonly<{
       true?: BranchCallback<TCurrentJson, TTrueJson>;
       false?: BranchCallback<TCurrentJson, TFalseJson>;
     }>;
-    const cursor = this.chain.then(new If<TCurrentJson>(name, (item) => predicate(item.json as TCurrentJson)));
+    const cursor = this.chain.then(new If<TCurrentJson>(name, (item, _index, _items, ctx) => predicate(item, ctx)));
     const trueSteps = branches.true?.(new WorkflowBranchBuilder<TCurrentJson>()).getSteps();
     const falseSteps = branches.false?.(new WorkflowBranchBuilder<TCurrentJson>()).getSteps();
     return new WorkflowChain(
@@ -258,7 +272,7 @@ export class WorkflowChain<TCurrentJson> {
     cfg: Readonly<{
       cases: readonly string[];
       defaultCase: string;
-      resolveCaseKey: (item: TCurrentJson) => string | Promise<string>;
+      resolveCaseKey: WorkflowSwitchCaseKeyResolver<TCurrentJson>;
       branches: Readonly<Record<string, RouteBranchCallback<TCurrentJson, TBranchJson> | undefined>>;
     }>,
   ): WorkflowChain<TBranchJson>;
@@ -267,7 +281,7 @@ export class WorkflowChain<TCurrentJson> {
     cfg: Readonly<{
       cases: readonly string[];
       defaultCase: string;
-      resolveCaseKey: (item: TCurrentJson) => string | Promise<string>;
+      resolveCaseKey: WorkflowSwitchCaseKeyResolver<TCurrentJson>;
       branches: Readonly<Record<string, RouteBranchCallback<TCurrentJson, TBranchJson> | undefined>>;
     }>,
     id?: string,
@@ -278,13 +292,13 @@ export class WorkflowChain<TCurrentJson> {
       | Readonly<{
           cases: readonly string[];
           defaultCase: string;
-          resolveCaseKey: (item: TCurrentJson) => string | Promise<string>;
+          resolveCaseKey: WorkflowSwitchCaseKeyResolver<TCurrentJson>;
           branches: Readonly<Record<string, RouteBranchCallback<TCurrentJson, TBranchJson> | undefined>>;
         }>,
     cfgOrUndefined?: Readonly<{
       cases: readonly string[];
       defaultCase: string;
-      resolveCaseKey: (item: TCurrentJson) => string | Promise<string>;
+      resolveCaseKey: WorkflowSwitchCaseKeyResolver<TCurrentJson>;
       branches: Readonly<Record<string, RouteBranchCallback<TCurrentJson, TBranchJson> | undefined>>;
     }>,
     id?: string,
@@ -297,7 +311,7 @@ export class WorkflowChain<TCurrentJson> {
         {
           cases: cfg.cases,
           defaultCase: cfg.defaultCase,
-          resolveCaseKey: (item) => cfg.resolveCaseKey(item.json as TCurrentJson),
+          resolveCaseKey: (item, _index, _items, ctx) => cfg.resolveCaseKey(item, ctx),
         },
         id,
       ),
@@ -322,18 +336,18 @@ export class WorkflowChain<TCurrentJson> {
     >;
   }
 
-  node<TConfig extends Record<string, unknown>, TOutputJson>(
-    definitionOrKey: DefinedNode<string, TConfig, TCurrentJson, TOutputJson> | string,
-    config: TConfig,
+  node<TConfig extends CredentialJsonRecord, TInputJson, TOutputJson>(
+    definitionOrKey: DefinedNode<string, TConfig, TInputJson, TOutputJson> | string,
+    config: DefinedNodeConfigInput<TConfig, TCurrentJson>,
     name?: string,
     id?: string,
-  ): WorkflowChain<TOutputJson> {
+  ): TCurrentJson extends TInputJson ? WorkflowChain<TOutputJson> : never {
     const definition = WorkflowDefinedNodeResolver.resolve(
       definitionOrKey as DefinedNode<string, Record<string, unknown>, unknown, unknown> | string,
-    ) as DefinedNode<string, TConfig, TCurrentJson, TOutputJson>;
+    ) as DefinedNode<string, TConfig, TInputJson, TOutputJson>;
     return this.then(
-      definition.create(config, name, id) as RunnableNodeConfig<TCurrentJson, TOutputJson>,
-    ) as WorkflowChain<TOutputJson>;
+      definition.create(config, name, id) as unknown as RunnableNodeConfig<TCurrentJson, TOutputJson>,
+    ) as TCurrentJson extends TInputJson ? WorkflowChain<TOutputJson> : never;
   }
 
   build(): WorkflowDefinition {