@codemation/core-nodes 0.1.1 → 0.3.0

package/src/nodes/NodeBackedToolRuntime.ts

@@ -10,7 +10,14 @@ import type {
   ToolExecuteArgs,
   ZodSchemaAny,
 } from "@codemation/core";
-import { CoreTokens, inject, injectable, ItemValueResolver, NodeOutputNormalizer } from "@codemation/core";
+import {
+  CoreTokens,
+  inject,
+  injectable,
+  ItemValueResolver,
+  NodeOutputNormalizer,
+  RunnableOutputBehaviorResolver,
+} from "@codemation/core";
 import { z } from "zod";
 
 @injectable()
@@ -22,6 +29,8 @@ export class NodeBackedToolRuntime {
     private readonly itemValueResolver: ItemValueResolver,
     @inject(NodeOutputNormalizer)
     private readonly outputNormalizer: NodeOutputNormalizer,
+    @inject(RunnableOutputBehaviorResolver)
+    private readonly outputBehaviorResolver: RunnableOutputBehaviorResolver,
   ) {}
 
   async execute(
@@ -64,7 +73,7 @@ export class NodeBackedToolRuntime {
     if (this.isRunnableNode(resolvedNode)) {
       const runnable = resolvedNode;
       const runnableConfig = ctx.config;
-      const carry = runnableConfig.lineageCarry ?? "emitOnly";
+      const behavior = this.outputBehaviorResolver.resolve(runnableConfig);
       const inputSchema = runnable.inputSchema ?? runnableConfig.inputSchema ?? z.unknown();
       const parsed = inputSchema.parse(nodeInput.json);
       const items = [nodeInput];
@@ -80,7 +89,7 @@ export class NodeBackedToolRuntime {
       return this.outputNormalizer.normalizeExecuteResult({
         baseItem: nodeInput,
         raw,
-        carry,
+        behavior,
       });
     }
     throw new Error(`Node-backed tool expected a runnable node instance for "${ctx.config.name ?? ctx.nodeId}".`);

package/src/nodes/aiAgent.ts

@@ -1,5 +1,7 @@
 export { AgentMessageFactory } from "./AgentMessageFactory";
 export { AgentOutputFactory } from "./AgentOutputFactory";
+export { AgentStructuredOutputRepairPromptFactory } from "./AgentStructuredOutputRepairPromptFactory";
+export { AgentStructuredOutputRunner } from "./AgentStructuredOutputRunner";
 export { AgentToolCallPortMap } from "./AgentToolCallPortMapFactory";
 export { AIAgent } from "./AIAgentConfig";
 export { AIAgentExecutionHelpersFactory } from "./AIAgentExecutionHelpersFactory";

package/src/nodes/mapData.ts

@@ -2,6 +2,11 @@ import type { Item, NodeExecutionContext, RunnableNodeConfig, TypeToken } from "
 
 import { MapDataNode } from "./MapDataNode";
 
+export interface MapDataOptions {
+  readonly id?: string;
+  readonly keepBinaries?: boolean;
+}
+
 export class MapData<TInputJson = unknown, TOutputJson = unknown> implements RunnableNodeConfig<
   TInputJson,
   TOutputJson
@@ -11,14 +16,22 @@ export class MapData<TInputJson = unknown, TOutputJson = unknown> implements Run
   readonly execution = { hint: "local" } as const;
   /** Zero mapped items should still allow downstream nodes to run. */
   readonly continueWhenEmptyOutput = true as const;
+  readonly keepBinaries: boolean;
+
   constructor(
     public readonly name: string,
     public readonly map: (
       item: Item<TInputJson>,
       ctx: NodeExecutionContext<MapData<TInputJson, TOutputJson>>,
     ) => TOutputJson,
-    public readonly id?: string,
-  ) {}
+    private readonly options: MapDataOptions = {},
+  ) {
+    this.keepBinaries = options.keepBinaries ?? true;
+  }
+
+  get id(): string | undefined {
+    return this.options.id;
+  }
 }
 
 export { MapDataNode } from "./MapDataNode";

package/src/nodes/switch.ts

@@ -14,7 +14,6 @@ export class Switch<TInputJson = unknown> implements RunnableNodeConfig<TInputJs
   readonly type: TypeToken<unknown> = SwitchNode;
   readonly execution = { hint: "local" } as const;
   readonly icon = "lucide:git-branch-plus" as const;
-  readonly lineageCarry = "carryThrough" as const;
   readonly declaredOutputPorts: ReadonlyArray<string>;
 
   constructor(

package/src/workflowAuthoring/WorkflowAgentNodeFactory.types.ts

@@ -10,22 +10,18 @@ export class WorkflowAgentNodeFactory {
   ): AIAgent<TCurrentJson, TOutputSchema extends z.ZodTypeAny ? z.output<TOutputSchema> : Record<string, unknown>> {
     const options = typeof nameOrOptions === "string" ? optionsOrUndefined! : nameOrOptions;
     const name = typeof nameOrOptions === "string" ? nameOrOptions : "AI agent";
-    const prompt = options.prompt;
-    const messages = [
-      {
-        role: "user" as const,
-        content:
-          typeof prompt === "function" ? ({ item }: { item: { json: TCurrentJson } }) => prompt(item.json) : prompt,
-      },
-    ] as const;
+    const outputSchema = options.outputSchema as
+      | z.ZodType<TOutputSchema extends z.ZodTypeAny ? z.output<TOutputSchema> : Record<string, unknown>>
+      | undefined;
     return new AIAgent<
       TCurrentJson,
       TOutputSchema extends z.ZodTypeAny ? z.output<TOutputSchema> : Record<string, unknown>
     >({
       name,
-      messages,
+      messages: options.messages,
       chatModel: WorkflowChatModelFactory.create(options.model),
       tools: options.tools,
+      outputSchema,
       id: options.id,
       retryPolicy: options.retryPolicy,
       guardrails: options.guardrails,

package/src/workflowAuthoring/WorkflowAuthoringOptions.types.ts

@@ -1,10 +1,16 @@
-import type { AgentGuardrailConfig, ChatModelConfig, RunnableNodeConfig, ToolConfig } from "@codemation/core";
+import type {
+  AgentGuardrailConfig,
+  AgentMessageConfig,
+  ChatModelConfig,
+  RunnableNodeConfig,
+  ToolConfig,
+} from "@codemation/core";
 import { z } from "zod";
 
-export type WorkflowAgentPrompt<TCurrentJson> = string | ((item: TCurrentJson) => string);
+export type WorkflowAgentMessages<TCurrentJson> = AgentMessageConfig<TCurrentJson>;
 
 export interface WorkflowAgentOptions<TCurrentJson, TOutputSchema extends z.ZodTypeAny | undefined = undefined> {
-  readonly prompt: WorkflowAgentPrompt<TCurrentJson>;
+  readonly messages: WorkflowAgentMessages<TCurrentJson>;
   readonly model: string | ChatModelConfig;
   readonly tools?: ReadonlyArray<ToolConfig>;
   readonly outputSchema?: TOutputSchema;

package/src/workflowAuthoring/WorkflowBranchBuilder.types.ts

@@ -10,6 +10,7 @@ import type {
 import { z } from "zod";
 import { Aggregate } from "../nodes/aggregate";
 import { Filter } from "../nodes/filter";
+import type { MapDataOptions } from "../nodes/mapData";
 import { MapData } from "../nodes/mapData";
 import { Split } from "../nodes/split";
 import { Wait } from "../nodes/wait";
@@ -18,6 +19,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> = []) {}
 
@@ -27,22 +33,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,
-    id?: string,
+    mapper: WorkflowMapCallback<TCurrentJson, TNextJson>,
+    options?: MapDataOptions,
   ): WorkflowBranchBuilder<TNextJson>;
   map<TNextJson>(
-    nameOrMapper: string | ((item: TCurrentJson) => TNextJson),
-    mapperOrUndefined?: (item: TCurrentJson) => TNextJson,
-    id?: string,
+    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), id),
-    ) as WorkflowBranchBuilder<TNextJson>;
+    return this.then(new MapData<TCurrentJson, TNextJson>(name, mapper, options)) as WorkflowBranchBuilder<TNextJson>;
   }
 
   wait(duration: number | string): WorkflowBranchBuilder<TCurrentJson>;

package/src/workflowAuthoring/WorkflowChain.types.ts

@@ -12,6 +12,7 @@ import { z } from "zod";
 import { Aggregate } from "../nodes/aggregate";
 import { Filter } from "../nodes/filter";
 import { If } from "../nodes/if";
+import type { MapDataOptions } from "../nodes/mapData";
 import { MapData } from "../nodes/mapData";
 import { Merge, type MergeMode } from "../nodes/merge";
 import { Split } from "../nodes/split";
@@ -28,6 +29,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>) {}
@@ -38,18 +51,20 @@ export class WorkflowChain<TCurrentJson> {
     return new WorkflowChain(this.chain.then(config));
   }
 
-  map<TNextJson>(mapper: (item: TCurrentJson) => TNextJson): WorkflowChain<TNextJson>;
-  map<TNextJson>(name: string, mapper: (item: TCurrentJson) => TNextJson, id?: string): WorkflowChain<TNextJson>;
+  map<TNextJson>(mapper: WorkflowMapCallback<TCurrentJson, TNextJson>): WorkflowChain<TNextJson>;
   map<TNextJson>(
-    nameOrMapper: string | ((item: TCurrentJson) => TNextJson),
-    mapperOrUndefined?: (item: TCurrentJson) => TNextJson,
-    id?: string,
+    name: string,
+    mapper: WorkflowMapCallback<TCurrentJson, TNextJson>,
+    options?: MapDataOptions,
+  ): WorkflowChain<TNextJson>;
+  map<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), id),
-    ) as WorkflowChain<TNextJson>;
+    return this.then(new MapData<TCurrentJson, TNextJson>(name, mapper, options)) as WorkflowChain<TNextJson>;
   }
 
   wait(duration: number | string): WorkflowChain<TCurrentJson>;
@@ -185,7 +200,7 @@ export class WorkflowChain<TCurrentJson> {
   }
 
   if<TBranchJson>(
-    predicate: (item: TCurrentJson) => boolean,
+    predicate: WorkflowIfPredicate<TCurrentJson>,
     branches: Readonly<{
       true?: BranchCallback<TCurrentJson, TBranchJson>;
       false?: BranchCallback<TCurrentJson, TBranchJson>;
@@ -193,16 +208,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>;
@@ -211,12 +226,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(
@@ -253,7 +270,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>;
@@ -262,7 +279,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,
@@ -273,13 +290,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,
@@ -292,7 +309,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,
       ),

package/src/workflowAuthoring.types.ts

@@ -1,4 +1,4 @@
-export type { WorkflowAgentOptions } from "./workflowAuthoring/WorkflowAuthoringOptions.types";
+export type { WorkflowAgentMessages, WorkflowAgentOptions } from "./workflowAuthoring/WorkflowAuthoringOptions.types";
 export { WorkflowAuthoringBuilder } from "./workflowAuthoring/WorkflowAuthoringBuilder.types";
 export { WorkflowBranchBuilder } from "./workflowAuthoring/WorkflowBranchBuilder.types";
 export { WorkflowChain } from "./workflowAuthoring/WorkflowChain.types";