@codemation/core-nodes 0.1.1 → 0.2.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";
@@ -31,17 +32,17 @@ export class WorkflowBranchBuilder<TCurrentJson> {
   map<TNextJson>(
     name: string,
     mapper: (item: TCurrentJson) => TNextJson,
-    id?: string,
+    options?: MapDataOptions,
   ): WorkflowBranchBuilder<TNextJson>;
   map<TNextJson>(
     nameOrMapper: string | ((item: TCurrentJson) => TNextJson),
     mapperOrUndefined?: (item: TCurrentJson) => TNextJson,
-    id?: string,
+    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),
+      new MapData<TCurrentJson, TNextJson>(name, (item) => mapper(item.json as TCurrentJson), options),
     ) as WorkflowBranchBuilder<TNextJson>;
   }
 

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";
@@ -39,16 +40,20 @@ export class WorkflowChain<TCurrentJson> {
   }
 
   map<TNextJson>(mapper: (item: TCurrentJson) => TNextJson): WorkflowChain<TNextJson>;
-  map<TNextJson>(name: string, mapper: (item: TCurrentJson) => TNextJson, id?: string): WorkflowChain<TNextJson>;
+  map<TNextJson>(
+    name: string,
+    mapper: (item: TCurrentJson) => TNextJson,
+    options?: MapDataOptions,
+  ): WorkflowChain<TNextJson>;
   map<TNextJson>(
     nameOrMapper: string | ((item: TCurrentJson) => TNextJson),
     mapperOrUndefined?: (item: TCurrentJson) => TNextJson,
-    id?: string,
+    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),
+      new MapData<TCurrentJson, TNextJson>(name, (item) => mapper(item.json as TCurrentJson), options),
     ) as WorkflowChain<TNextJson>;
   }
 

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";