pipeai 0.1.0

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts","../src/agent.ts","../src/tool-provider.ts","../src/utils.ts","../src/workflow.ts"],"sourcesContent":["export { Agent } from \"./agent\";\nexport type {\n  AgentConfig,\n  GenerateTextResult,\n  StreamTextResult,\n} from \"./agent\";\n\nexport { Workflow, WorkflowBranchError, WorkflowLoopError } from \"./workflow\";\nexport type { SealedWorkflow } from \"./workflow\";\nexport type {\n  AgentStepHooks,\n  StepOptions,\n  BranchCase,\n  BranchSelect,\n  RepeatOptions,\n  WorkflowResult,\n  WorkflowStreamResult,\n  WorkflowStreamOptions,\n} from \"./workflow\";\n\nexport { defineTool } from \"./tool-provider\";\nexport type { ToolProviderConfig, IToolProvider } from \"./tool-provider\";\n\nexport type { MaybePromise, Resolvable } from \"./utils\";\n","import {\n  generateText,\n  streamText,\n  tool,\n  Output,\n  type GenerateTextResult as AIGenerateTextResult,\n  type StreamTextResult as AIStreamTextResult,\n  type ModelMessage,\n  type LanguageModel,\n  type Tool,\n  type ToolSet,\n  type StopCondition,\n  type ToolChoice,\n  type OnStepFinishEvent,\n  type OnFinishEvent,\n} from \"ai\";\n\n// Extract the Output interface type from the Output.object return type\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ntype OutputType<T = any> = ReturnType<typeof Output.object<T>>;\nimport type { ZodType } from \"zod\";\nimport { isToolProvider, TOOL_PROVIDER_BRAND, type IToolProvider } from \"./tool-provider\";\nimport { extractOutput, resolveValue, type MaybePromise, type Resolvable } from \"./utils\";\n\n// Tools config accepts both AI SDK tools and context-aware ToolProviders\ntype AgentToolSet<TContext> = Record<string, Tool | IToolProvider<TContext>>;\n\n// ── Result type aliases ─────────────────────────────────────────────\n\nexport type GenerateTextResult<TOOLS extends ToolSet = ToolSet, OUTPUT extends OutputType = OutputType> = AIGenerateTextResult<TOOLS, OUTPUT>;\nexport type StreamTextResult<TOOLS extends ToolSet = ToolSet, OUTPUT extends OutputType = OutputType> = AIStreamTextResult<TOOLS, OUTPUT>;\n\n// ── AI SDK passthrough types ────────────────────────────────────────\n\n// Extract options types from both AI SDK entry points\ntype StreamTextOptions = Parameters<typeof streamText>[0];\ntype GenerateTextOptions = Parameters<typeof generateText>[0];\n\n// Keys we replace with resolvable or context-enriched versions\ntype ManagedKeys =\n  | 'model' | 'system' | 'prompt' | 'messages'\n  | 'tools' | 'activeTools' | 'toolChoice' | 'stopWhen'\n  | 'output' | 'onFinish' | 'onStepFinish' | 'onError';\n\n// Combine options from both streamText and generateText.\n// Each side contributes its unique props; shared props merge naturally.\n// Stream-only props (onChunk, onAbort) are ignored by generateText.\n// Generate-only props (experimental_include.responseBody) are ignored by streamText.\ntype AIPassthroughOptions =\n  Omit<StreamTextOptions, ManagedKeys> &\n  Omit<GenerateTextOptions, ManagedKeys>;\n\n// ── Resolved config (output of resolveConfig / resolveConfigAsync) ──\n\ninterface ResolvedAgentConfig {\n  model: LanguageModel;\n  prompt: string | undefined;\n  system: string | undefined;\n  messages: ModelMessage[] | undefined;\n  tools: Record<string, Tool>;\n  activeTools: string[] | undefined;\n  toolChoice: ToolChoice<ToolSet> | undefined;\n  stopWhen: StopCondition<ToolSet> | Array<StopCondition<ToolSet>> | undefined;\n}\n\n// ── Agent Configuration ─────────────────────────────────────────────\n\nexport interface AgentConfig<\n  TContext,\n  TInput = void,\n  TOutput = void,\n> extends AIPassthroughOptions {\n  // ── Custom (not in AI SDK) ──\n  id: string;\n  description?: string;\n  input?: ZodType<TInput>;\n  output?: OutputType<TOutput>;\n\n  // ── Resolvable (our versions of AI SDK properties) ──\n  model: Resolvable<TContext, TInput, LanguageModel>;\n  system?: Resolvable<TContext, TInput, string>;\n  prompt?: Resolvable<TContext, TInput, string>;\n  messages?: Resolvable<TContext, TInput, ModelMessage[]>;\n  tools?: Resolvable<TContext, TInput, AgentToolSet<TContext>>;\n  activeTools?: Resolvable<TContext, TInput, string[]>;\n  toolChoice?: Resolvable<TContext, TInput, ToolChoice<ToolSet>>;\n  stopWhen?: Resolvable<TContext, TInput, StopCondition<ToolSet> | Array<StopCondition<ToolSet>>>;\n\n  // ── Context-enriched callbacks (replace AI SDK versions) ──\n  onStepFinish?: (params: { result: OnStepFinishEvent; ctx: Readonly<TContext>; input: TInput }) => MaybePromise<void>;\n  onFinish?: (params: { result: OnFinishEvent; ctx: Readonly<TContext>; input: TInput }) => MaybePromise<void>;\n  onError?: (params: { error: unknown; ctx: Readonly<TContext>; input: TInput }) => MaybePromise<void>;\n}\n\n// ── Agent ───────────────────────────────────────────────────────────\n\nexport class Agent<\n  TContext,\n  TInput = void,\n  TOutput = void,\n> {\n  readonly id: string;\n  readonly description: string;\n  readonly hasOutput: boolean;\n  private readonly config: AgentConfig<TContext, TInput, TOutput>;\n  private readonly _hasDynamicConfig: boolean;\n  private readonly _resolvedStaticTools: Record<string, Tool> | null = null;\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  private readonly _passthrough: Record<string, any>;\n  private readonly _onStepFinish: AgentConfig<TContext, TInput, TOutput>['onStepFinish'];\n  private readonly _onFinish: AgentConfig<TContext, TInput, TOutput>['onFinish'];\n\n  constructor(config: AgentConfig<TContext, TInput, TOutput>) {\n    this.id = config.id;\n    this.description = config.description ?? \"\";\n    this.hasOutput = config.output !== undefined;\n    this.config = config;\n    this._hasDynamicConfig = [\n      config.model, config.system, config.prompt,\n      config.messages, config.tools, config.activeTools,\n      config.toolChoice, config.stopWhen,\n    ].some(v => typeof v === \"function\");\n\n    // Cache tools when config is static and contains no ToolProviders.\n    // Avoids re-iterating the tools map on every generate()/stream() call.\n    if (!this._hasDynamicConfig) {\n      const rawTools = (config.tools as AgentToolSet<TContext> | undefined) ?? {};\n      const hasProvider = Object.values(rawTools).some(v => isToolProvider(v));\n      if (!hasProvider) {\n        this._resolvedStaticTools = rawTools as Record<string, Tool>;\n      }\n    }\n\n    // Pre-compute the passthrough (AI SDK options we don't manage) once,\n    // rather than destructuring on every generate()/stream() call.\n    const {\n      id: _id, description: _desc, input: _inputSchema, output: _output,\n      model: _m, system: _s, prompt: _p, messages: _msg,\n      tools: _t, activeTools: _at, toolChoice: _tc, stopWhen: _sw,\n      onStepFinish, onFinish, onError: _onError,\n      ...passthrough\n    } = config;\n    this._passthrough = passthrough;\n    this._onStepFinish = onStepFinish;\n    this._onFinish = onFinish;\n  }\n\n  async generate(ctx: TContext, ...args: TInput extends void ? [input?: TInput] : [input: TInput]): Promise<GenerateTextResult> {\n    const input = args[0] as TInput;\n    const resolved = await this.resolveConfig(ctx, input);\n    const options = this.buildCallOptions(resolved, ctx, input);\n\n    try {\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      return await generateText(options as any);\n    } catch (error: unknown) {\n      if (this.config.onError) {\n        await this.config.onError({ error, ctx, input });\n      }\n      throw error;\n    }\n  }\n\n  async stream(ctx: TContext, ...args: TInput extends void ? [input?: TInput] : [input: TInput]): Promise<StreamTextResult> {\n    const input = args[0] as TInput;\n    const resolved = await this.resolveConfig(ctx, input);\n    const options = this.buildCallOptions(resolved, ctx, input);\n\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    return streamText({\n      ...options,\n      onError: this.config.onError\n        ? ({ error }: { error: unknown }) => this.config.onError!({ error, ctx, input })\n        : undefined,\n    } as any);\n  }\n\n  asTool(ctx: TContext, options?: {\n    mapOutput?: (result: GenerateTextResult) => MaybePromise<TOutput>;\n  }): Tool {\n    return this.createToolInstance(ctx, options);\n  }\n\n  asToolProvider(options?: {\n    mapOutput?: (result: GenerateTextResult) => MaybePromise<TOutput>;\n  }): IToolProvider<TContext> {\n    if (!this.config.input) {\n      throw new Error(`Agent \"${this.id}\": asToolProvider() requires an input schema`);\n    }\n\n    return {\n      [TOOL_PROVIDER_BRAND]: true as const,\n      createTool: (ctx: Readonly<TContext>) => this.createToolInstance(ctx as TContext, options),\n    };\n  }\n\n  private createToolInstance(ctx: TContext, options?: {\n    mapOutput?: (result: GenerateTextResult) => MaybePromise<TOutput>;\n  }): Tool {\n    if (!this.config.input) {\n      throw new Error(`Agent \"${this.id}\": asTool() requires an input schema`);\n    }\n\n    return tool({\n      description: this.description,\n      parameters: this.config.input,\n      execute: async (toolInput: TInput) => {\n        const result = await (this.generate as (ctx: TContext, input: TInput) => Promise<GenerateTextResult>)(ctx, toolInput);\n        if (options?.mapOutput) return options.mapOutput(result);\n        return extractOutput(result, this.hasOutput);\n      },\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    } as any);\n  }\n\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  private buildCallOptions(resolved: ResolvedAgentConfig, ctx: TContext, input: TInput): Record<string, any> {\n    return {\n      ...this._passthrough,\n      model: resolved.model,\n      tools: resolved.tools,\n      activeTools: resolved.activeTools,\n      toolChoice: resolved.toolChoice,\n      stopWhen: resolved.stopWhen,\n      ...(resolved.messages\n        ? { messages: resolved.messages }\n        : { prompt: resolved.prompt ?? \"\" }),\n      ...(resolved.system ? { system: resolved.system } : {}),\n      ...(this.config.output ? { output: this.config.output } : {}),\n      onStepFinish: this._onStepFinish\n        ? (event: OnStepFinishEvent) => this._onStepFinish!({ result: event, ctx, input })\n        : undefined,\n      onFinish: this._onFinish\n        ? (event: OnFinishEvent) => this._onFinish!({ result: event, ctx, input })\n        : undefined,\n    };\n  }\n\n  private resolveConfig(ctx: TContext, input: TInput): ResolvedAgentConfig | Promise<ResolvedAgentConfig> {\n    if (!this._hasDynamicConfig) {\n      return {\n        model: this.config.model as LanguageModel,\n        prompt: this.config.prompt as string | undefined,\n        system: this.config.system as string | undefined,\n        messages: this.config.messages as ModelMessage[] | undefined,\n        tools: this._resolvedStaticTools ?? this.resolveTools(\n          (this.config.tools as AgentToolSet<TContext> | undefined) ?? {}, ctx\n        ),\n        activeTools: this.config.activeTools as string[] | undefined,\n        toolChoice: this.config.toolChoice as ToolChoice<ToolSet> | undefined,\n        stopWhen: this.config.stopWhen as StopCondition<ToolSet> | Array<StopCondition<ToolSet>> | undefined,\n      };\n    }\n    return this.resolveConfigAsync(ctx, input);\n  }\n\n  private async resolveConfigAsync(ctx: TContext, input: TInput): Promise<ResolvedAgentConfig> {\n    const [model, prompt, system, messages, rawTools, activeTools, toolChoice, stopWhen] = await Promise.all([\n      resolveValue(this.config.model, ctx, input),\n      resolveValue(this.config.prompt, ctx, input),\n      resolveValue(this.config.system, ctx, input),\n      resolveValue(this.config.messages, ctx, input),\n      resolveValue(this.config.tools, ctx, input),\n      resolveValue(this.config.activeTools, ctx, input),\n      resolveValue(this.config.toolChoice, ctx, input),\n      resolveValue(this.config.stopWhen, ctx, input),\n    ]);\n    const tools = this.resolveTools(rawTools ?? {}, ctx);\n    return { model, prompt, system, messages, tools, activeTools, toolChoice, stopWhen };\n  }\n\n  private resolveTools(\n    tools: AgentToolSet<TContext>,\n    ctx: TContext\n  ): Record<string, Tool> {\n    const entries = Object.entries(tools);\n    if (entries.length === 0) return tools as Record<string, Tool>;\n    let hasProvider = false;\n    const resolved: Record<string, Tool> = {};\n    for (const [key, toolOrProvider] of entries) {\n      if (isToolProvider<TContext>(toolOrProvider)) {\n        hasProvider = true;\n        resolved[key] = toolOrProvider.createTool(ctx as Readonly<TContext>);\n      } else {\n        resolved[key] = toolOrProvider as Tool;\n      }\n    }\n    return hasProvider ? resolved : (tools as Record<string, Tool>);\n  }\n}\n","import { tool, type Tool, type ToolExecutionOptions, type FlexibleSchema } from \"ai\";\n\nexport const TOOL_PROVIDER_BRAND = Symbol.for(\"agent-workflow.ToolProvider\");\n\nexport type ToolProviderConfig<TContext, TInput, TOutput> = {\n  description?: string;\n  input: FlexibleSchema<TInput>;\n  output?: FlexibleSchema<unknown>;\n  providerOptions?: unknown;\n  execute: (input: TInput, ctx: Readonly<TContext>, options?: ToolExecutionOptions) => Promise<TOutput>;\n};\n\nexport interface IToolProvider<TContext> {\n  readonly [TOOL_PROVIDER_BRAND]: true;\n  createTool(context: Readonly<TContext>): Tool;\n}\n\nexport class ToolProvider<\n  TContext,\n  TInput = unknown,\n  TOutput = unknown,\n> implements IToolProvider<TContext> {\n  readonly [TOOL_PROVIDER_BRAND] = true as const;\n  private readonly config: ToolProviderConfig<TContext, TInput, TOutput>;\n\n  constructor(config: ToolProviderConfig<TContext, TInput, TOutput>) {\n    this.config = config;\n  }\n\n  createTool(context: Readonly<TContext>): Tool {\n    const { execute, input: inputSchema, ...toolDef } = this.config;\n    return tool({\n      ...toolDef,\n      parameters: inputSchema,\n      execute: (input: TInput, options?: ToolExecutionOptions) => execute(input, context, options),\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    } as any);\n  }\n}\n\nexport function defineTool<TContext>() {\n  return <TInput, TOutput>(\n    config: ToolProviderConfig<TContext, TInput, TOutput>\n  ): ToolProvider<TContext, TInput, TOutput> => new ToolProvider(config);\n}\n\nexport function isToolProvider<TContext>(obj: unknown): obj is IToolProvider<TContext> {\n  return (\n    typeof obj === \"object\" &&\n    obj !== null &&\n    TOOL_PROVIDER_BRAND in obj\n  );\n}\n","export type MaybePromise<T> = T | Promise<T>;\n\n/**\n * A value that can be static or derived from context and input.\n * Used for agent config fields that may need runtime resolution.\n *\n * Functions may return a Promise for async resolution; static values are always sync.\n */\nexport type Resolvable<TCtx, TInput, TValue> =\n  | TValue\n  | ((ctx: Readonly<TCtx>, input: TInput) => TValue | Promise<TValue>);\n\nexport function resolveValue<TCtx, TInput, TValue>(\n  value: Resolvable<TCtx, TInput, TValue>,\n  ctx: TCtx,\n  input: TInput\n): TValue | Promise<TValue>;\nexport function resolveValue<TCtx, TInput, TValue>(\n  value: Resolvable<TCtx, TInput, TValue> | undefined,\n  ctx: TCtx,\n  input: TInput\n): TValue | Promise<TValue> | undefined {\n  if (typeof value === \"function\") {\n    return (value as (ctx: TCtx, input: TInput) => TValue | Promise<TValue>)(ctx, input);\n  }\n  return value;\n}\n\n/**\n * Extract structured output from an AI SDK result, falling back to text.\n * Works for both generate (sync .output/.text) and stream (async .output/.text) results.\n */\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport async function extractOutput(result: any, hasStructuredOutput: boolean): Promise<unknown> {\n  if (hasStructuredOutput) {\n    const output = await result.output;\n    if (output !== undefined) return output;\n  }\n  return await result.text;\n}\n","import {\n  createUIMessageStream,\n  type UIMessageStreamWriter,\n} from \"ai\";\nimport { type Agent, type GenerateTextResult, type StreamTextResult } from \"./agent\";\nimport { extractOutput, type MaybePromise } from \"./utils\";\n\n// ── Error Types ─────────────────────────────────────────────────────\n\nexport class WorkflowBranchError extends Error {\n  constructor(\n    public readonly branchType: \"predicate\" | \"select\",\n    message: string,\n  ) {\n    super(message);\n    this.name = \"WorkflowBranchError\";\n  }\n}\n\nexport class WorkflowLoopError extends Error {\n  constructor(\n    public readonly iterations: number,\n    public readonly maxIterations: number,\n  ) {\n    super(`Loop exceeded maximum iterations (${maxIterations})`);\n    this.name = \"WorkflowLoopError\";\n  }\n}\n\n// ── Shared Agent Step Hooks ─────────────────────────────────────────\n\nexport interface AgentStepHooks<TContext, TOutput, TNextOutput> {\n  mapGenerateResult?: (params: { result: GenerateTextResult; ctx: Readonly<TContext>; input: TOutput }) => MaybePromise<TNextOutput>;\n  mapStreamResult?: (params: { result: StreamTextResult; ctx: Readonly<TContext>; input: TOutput }) => MaybePromise<TNextOutput>;\n  onGenerateResult?: (params: { result: GenerateTextResult; ctx: Readonly<TContext>; input: TOutput }) => MaybePromise<void>;\n  onStreamResult?: (params: { result: StreamTextResult; ctx: Readonly<TContext>; input: TOutput }) => MaybePromise<void>;\n  handleStream?: (params: {\n    result: StreamTextResult;\n    writer: UIMessageStreamWriter;\n    ctx: Readonly<TContext>;\n  }) => MaybePromise<void>;\n}\n\n// ── Step Options ────────────────────────────────────────────────────\n\nexport type StepOptions<TContext, TOutput, TNextOutput> = AgentStepHooks<TContext, TOutput, TNextOutput>;\n\n// ── Branch Types ────────────────────────────────────────────────────\n\nexport interface BranchCase<TContext, TOutput, TNextOutput> extends AgentStepHooks<TContext, TOutput, TNextOutput> {\n  when?: (params: { ctx: Readonly<TContext>; input: TOutput }) => MaybePromise<boolean>;\n  agent: Agent<TContext, TOutput, TNextOutput>;\n}\n\nexport interface BranchSelect<TContext, TOutput, TKeys extends string, TNextOutput> extends AgentStepHooks<TContext, TOutput, TNextOutput> {\n  select: (params: { ctx: Readonly<TContext>; input: TOutput }) => MaybePromise<TKeys>;\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  agents: Record<TKeys, Agent<TContext, any, TNextOutput>>;\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  fallback?: Agent<TContext, any, TNextOutput>;\n}\n\n// ── Result Types ────────────────────────────────────────────────────\n\nexport interface WorkflowResult<TOutput> {\n  output: TOutput;\n}\n\nexport interface WorkflowStreamResult<TOutput> {\n  stream: ReadableStream;\n  output: Promise<TOutput>;\n}\n\nexport interface WorkflowStreamOptions {\n  onError?: (error: unknown) => string;\n  onFinish?: () => MaybePromise<void>;\n}\n\n// ── Loop Types ──────────────────────────────────────────────────────\n\ntype LoopPredicate<TContext, TOutput> = (params: {\n  output: TOutput;\n  ctx: Readonly<TContext>;\n  iterations: number;\n}) => MaybePromise<boolean>;\n\n// Exactly one of `until` or `while` — never both.\nexport type RepeatOptions<TContext, TOutput> =\n  | { until: LoopPredicate<TContext, TOutput>; while?: never; maxIterations?: number }\n  | { while: LoopPredicate<TContext, TOutput>; until?: never; maxIterations?: number };\n\n// Extracts the element type from an array type. Resolves to `never` for non-arrays,\n// making foreach uncallable at compile time when the previous step doesn't produce an array.\ntype ElementOf<T> = T extends readonly (infer E)[] ? E : never;\n\n// ── Step Node ───────────────────────────────────────────────────────\n\ntype StepNode =\n  | { readonly type: \"step\"; readonly id: string; readonly execute: (state: RuntimeState) => MaybePromise<void> }\n  | { readonly type: \"catch\"; readonly id: string; readonly catchFn: (params: { error: unknown; ctx: unknown; lastOutput: unknown; stepId: string }) => MaybePromise<unknown> }\n  | { readonly type: \"finally\"; readonly id: string; readonly execute: (state: RuntimeState) => MaybePromise<void> };\n\ninterface RuntimeState {\n  ctx: unknown;\n  output: unknown;\n  mode: \"generate\" | \"stream\";\n  writer?: UIMessageStreamWriter;\n}\n\n// ── Sealed Workflow (returned by finally — execution only) ───────────\n\nexport class SealedWorkflow<\n  TContext,\n  TInput = void,\n  TOutput = void,\n> {\n  readonly id?: string;\n  protected readonly steps: ReadonlyArray<StepNode>;\n\n  protected constructor(steps: ReadonlyArray<StepNode>, id?: string) {\n    this.steps = steps;\n    this.id = id;\n  }\n\n  // ── Execution ─────────────────────────────────────────────────\n\n  async generate(ctx: TContext, ...args: TInput extends void ? [input?: TInput] : [input: TInput]): Promise<WorkflowResult<TOutput>> {\n    const input = args[0];\n    const state: RuntimeState = {\n      ctx,\n      output: input,\n      mode: \"generate\",\n    };\n\n    await this.execute(state);\n\n    return {\n      output: state.output as TOutput,\n    };\n  }\n\n  stream(\n    ctx: TContext,\n    ...args: TInput extends void\n      ? [input?: TInput, options?: WorkflowStreamOptions]\n      : [input: TInput, options?: WorkflowStreamOptions]\n  ): WorkflowStreamResult<TOutput> {\n    const input = args[0];\n    const options = args[1] as WorkflowStreamOptions | undefined;\n\n    let resolveOutput: (value: TOutput) => void;\n    let rejectOutput: (error: unknown) => void;\n    const outputPromise = new Promise<TOutput>((res, rej) => {\n      resolveOutput = res;\n      rejectOutput = rej;\n    });\n\n    // Prevent unhandled rejection warning if the consumer never awaits `output`.\n    // The original promise still rejects normally when awaited.\n    outputPromise.catch(() => {});\n\n    const stream = createUIMessageStream({\n      execute: async ({ writer }) => {\n        const state: RuntimeState = {\n          ctx,\n          output: input,\n          mode: \"stream\",\n          writer,\n        };\n\n        try {\n          await this.execute(state);\n          resolveOutput(state.output as TOutput);\n        } catch (error) {\n          rejectOutput!(error);\n          throw error;\n        }\n      },\n      ...(options?.onError ? { onError: options.onError } : {}),\n      ...(options?.onFinish ? { onFinish: options.onFinish } : {}),\n    });\n\n    return {\n      stream,\n      output: outputPromise,\n    };\n  }\n\n  // ── Internal: execute pipeline ────────────────────────────────\n\n  protected async execute(state: RuntimeState): Promise<void> {\n    if (this.steps.length === 0) {\n      throw new Error(\"Workflow has no steps. Add at least one step before calling generate() or stream().\");\n    }\n\n    let pendingError: { error: unknown; stepId: string } | null = null;\n\n    for (const node of this.steps) {\n      if (node.type === \"finally\") {\n        await node.execute(state);\n        continue;\n      }\n\n      if (node.type === \"catch\") {\n        if (!pendingError) continue;\n        try {\n          state.output = await node.catchFn({\n            error: pendingError.error,\n            ctx: state.ctx,\n            lastOutput: state.output,\n            stepId: pendingError.stepId,\n          });\n          pendingError = null;\n        } catch (catchError) {\n          pendingError = { error: catchError, stepId: node.id };\n        }\n        continue;\n      }\n\n      // type === \"step\" — skip while in error state\n      if (pendingError) continue;\n\n      try {\n        await node.execute(state);\n      } catch (error) {\n        pendingError = { error, stepId: node.id };\n      }\n    }\n\n    if (pendingError) throw pendingError.error;\n  }\n\n  // ── Internal: execute a nested workflow within a step/loop ─────\n  // Defined on SealedWorkflow (not Workflow) because TypeScript's protected\n  // access rules only allow calling workflow.execute() from the same class.\n\n  protected async executeNestedWorkflow(\n    state: RuntimeState,\n    workflow: SealedWorkflow<TContext, unknown, unknown>,\n  ): Promise<void> {\n    await workflow.execute(state);\n  }\n\n  // ── Internal: execute an agent within a step/branch ───────────\n  // In stream mode, output extraction awaits the full stream before returning.\n  // Streaming benefits the client (incremental output), not pipeline throughput —\n  // each step still runs sequentially.\n\n  protected async executeAgent<TAgentInput, TNextOutput>(\n    state: RuntimeState,\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    agent: Agent<TContext, any, TNextOutput>,\n    ctx: TContext,\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    options?: AgentStepHooks<TContext, any, TNextOutput>,\n  ): Promise<void> {\n    const input = state.output as TAgentInput;\n    const hasStructuredOutput = agent.hasOutput;\n\n    if (state.mode === \"stream\" && state.writer) {\n      const result = await (agent.stream as (ctx: TContext, input: unknown) => Promise<StreamTextResult>)(ctx, state.output);\n\n      if (options?.handleStream) {\n        await options.handleStream({ result, writer: state.writer, ctx });\n      } else {\n        state.writer.merge(result.toUIMessageStream());\n      }\n\n      if (options?.onStreamResult) {\n        await options.onStreamResult({ result, ctx, input });\n      }\n\n      if (options?.mapStreamResult) {\n        state.output = await options.mapStreamResult({ result, ctx, input });\n      } else {\n        state.output = await extractOutput(result, hasStructuredOutput);\n      }\n    } else {\n      const result = await (agent.generate as (ctx: TContext, input: unknown) => Promise<GenerateTextResult>)(ctx, state.output);\n\n      if (options?.onGenerateResult) {\n        await options.onGenerateResult({ result, ctx, input });\n      }\n\n      if (options?.mapGenerateResult) {\n        state.output = await options.mapGenerateResult({ result, ctx, input });\n      } else {\n        state.output = await extractOutput(result, hasStructuredOutput);\n      }\n    }\n  }\n}\n\n// ── Workflow ────────────────────────────────────────────────────────\n\nexport class Workflow<\n  TContext,\n  TInput = void,\n  TOutput = void,\n> extends SealedWorkflow<TContext, TInput, TOutput> {\n\n  private constructor(steps: ReadonlyArray<StepNode> = [], id?: string) {\n    super(steps, id);\n  }\n\n  static create<TContext, TInput = void>(options?: { id?: string }): Workflow<TContext, TInput, TInput> {\n    return new Workflow<TContext, TInput, TInput>([], options?.id);\n  }\n\n  static from<TContext, TInput, TOutput>(\n    agent: Agent<TContext, TInput, TOutput>,\n    options?: StepOptions<TContext, TInput, TOutput>\n  ): Workflow<TContext, TInput, TOutput> {\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    return new Workflow<TContext, TInput, any>([]).step(agent, options);\n  }\n\n  // ── step: agent overload ──────────────────────────────────────\n\n  step<TNextOutput>(\n    agent: Agent<TContext, TOutput, TNextOutput>,\n    options?: StepOptions<TContext, TOutput, TNextOutput>\n  ): Workflow<TContext, TInput, TNextOutput>;\n\n  // ── step: nested workflow overload ─────────────────────────────\n\n  step<TNextOutput>(\n    workflow: SealedWorkflow<TContext, TOutput, TNextOutput>,\n  ): Workflow<TContext, TInput, TNextOutput>;\n\n  // ── step: transform overload (replaces map + tap) ─────────────\n\n  step<TNextOutput>(\n    id: string,\n    fn: (params: { ctx: Readonly<TContext>; input: TOutput }) => MaybePromise<TNextOutput>\n  ): Workflow<TContext, TInput, TNextOutput>;\n\n  // ── step: implementation ──────────────────────────────────────\n\n  step<TNextOutput>(\n    target: Agent<TContext, TOutput, TNextOutput> | SealedWorkflow<TContext, TOutput, TNextOutput> | string,\n    optionsOrFn?: StepOptions<TContext, TOutput, TNextOutput> | ((params: { ctx: Readonly<TContext>; input: TOutput }) => MaybePromise<TNextOutput>)\n  ): Workflow<TContext, TInput, TNextOutput> {\n    // Nested workflow overload: step(workflow)\n    if (target instanceof SealedWorkflow) {\n      const workflow = target;\n      const node: StepNode = {\n        type: \"step\",\n        id: workflow.id ?? \"nested-workflow\",\n        execute: async (state) => {\n          await this.executeNestedWorkflow(state, workflow as SealedWorkflow<TContext, unknown, unknown>);\n        },\n      };\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      return new Workflow<TContext, TInput, TNextOutput>([...this.steps, node] as any, this.id);\n    }\n\n    // Transform overload: step(id, fn)\n    if (typeof target === \"string\") {\n      if (typeof optionsOrFn !== \"function\") {\n        throw new Error(`Workflow step(\"${target}\"): second argument must be a function`);\n      }\n      const fn = optionsOrFn as (params: { ctx: Readonly<TContext>; input: TOutput }) => MaybePromise<TNextOutput>;\n      const node: StepNode = {\n        type: \"step\",\n        id: target,\n        execute: async (state) => {\n          state.output = await fn({\n            ctx: state.ctx as Readonly<TContext>,\n            input: state.output as TOutput,\n          });\n        },\n      };\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      return new Workflow<TContext, TInput, TNextOutput>([...this.steps, node] as any, this.id);\n    }\n\n    // Agent overload: step(agent, options?)\n    const agent = target;\n    const options = optionsOrFn as StepOptions<TContext, TOutput, TNextOutput> | undefined;\n    const node: StepNode = {\n      type: \"step\",\n      id: agent.id,\n      execute: async (state) => {\n        const ctx = state.ctx as TContext;\n        await this.executeAgent(state, agent, ctx, options);\n      },\n    };\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    return new Workflow<TContext, TInput, TNextOutput>([...this.steps, node] as any, this.id);\n  }\n\n  // ── branch: predicate routing (array) ─────────────────────────\n\n  branch<TNextOutput>(\n    cases: BranchCase<TContext, TOutput, TNextOutput>[]\n  ): Workflow<TContext, TInput, TNextOutput>;\n\n  // ── branch: key routing (select) ──────────────────────────────\n\n  branch<TKeys extends string, TNextOutput>(\n    config: BranchSelect<TContext, TOutput, TKeys, TNextOutput>\n  ): Workflow<TContext, TInput, TNextOutput>;\n\n  // ── branch: implementation ────────────────────────────────────\n\n  branch<TKeys extends string, TNextOutput>(\n    casesOrConfig: BranchCase<TContext, TOutput, TNextOutput>[] | BranchSelect<TContext, TOutput, TKeys, TNextOutput>\n  ): Workflow<TContext, TInput, TNextOutput> {\n    if (Array.isArray(casesOrConfig)) {\n      return this.branchPredicate(casesOrConfig);\n    }\n    return this.branchSelect(casesOrConfig);\n  }\n\n  private branchPredicate<TNextOutput>(\n    cases: BranchCase<TContext, TOutput, TNextOutput>[]\n  ): Workflow<TContext, TInput, TNextOutput> {\n    const node: StepNode = {\n      type: \"step\",\n      id: \"branch:predicate\",\n      execute: async (state) => {\n        const ctx = state.ctx as TContext;\n        const input = state.output as TOutput;\n\n        for (const branchCase of cases) {\n          if (branchCase.when) {\n            const match = await branchCase.when({ ctx, input });\n            if (!match) continue;\n          }\n\n          // Matched (or no `when` = default)\n          await this.executeAgent(state, branchCase.agent, ctx, branchCase);\n          return;\n        }\n\n        throw new WorkflowBranchError(\"predicate\", `No branch matched and no default branch (a case without \\`when\\`) was provided. Input: ${JSON.stringify(input)}`);\n      },\n    };\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    return new Workflow<TContext, TInput, TNextOutput>([...this.steps, node] as any, this.id);\n  }\n\n  private branchSelect<TKeys extends string, TNextOutput>(\n    config: BranchSelect<TContext, TOutput, TKeys, TNextOutput>\n  ): Workflow<TContext, TInput, TNextOutput> {\n    const node: StepNode = {\n      type: \"step\",\n      id: \"branch:select\",\n      execute: async (state) => {\n        const ctx = state.ctx as TContext;\n        const input = state.output as TOutput;\n        const key = await config.select({ ctx, input });\n\n        let agent = config.agents[key];\n        if (!agent) {\n          if (config.fallback) {\n            agent = config.fallback;\n          } else {\n            throw new WorkflowBranchError(\"select\", `No agent found for key \"${key}\" and no fallback provided. Available keys: ${Object.keys(config.agents).join(\", \")}`);\n          }\n        }\n\n        await this.executeAgent(state, agent, ctx, config);\n      },\n    };\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    return new Workflow<TContext, TInput, TNextOutput>([...this.steps, node] as any, this.id);\n  }\n\n  // ── foreach: array iteration ─────────────────────────────────\n\n  foreach<TNextOutput>(\n    target: Agent<TContext, ElementOf<TOutput>, TNextOutput> | SealedWorkflow<TContext, ElementOf<TOutput>, TNextOutput>,\n    options?: { concurrency?: number },\n  ): Workflow<TContext, TInput, TNextOutput[]> {\n    const concurrency = options?.concurrency ?? 1;\n    const isWorkflow = target instanceof SealedWorkflow;\n    const id = isWorkflow ? (target.id ?? \"foreach\") : `foreach:${(target as Agent<TContext, ElementOf<TOutput>, TNextOutput>).id}`;\n\n    const node: StepNode = {\n      type: \"step\",\n      id,\n      execute: async (state) => {\n        const items = state.output;\n        if (!Array.isArray(items)) {\n          throw new Error(`foreach \"${id}\": expected array input, got ${typeof items}`);\n        }\n\n        const ctx = state.ctx as TContext;\n        const results: unknown[] = new Array(items.length);\n\n        // Streaming is intentionally not propagated to foreach items —\n        // each item runs in generate mode because merging interleaved\n        // streams from parallel items into a single writer is not supported.\n        const executeItem = async (item: unknown, index: number) => {\n          const itemState: RuntimeState = { ctx: state.ctx, output: item, mode: \"generate\" };\n          if (isWorkflow) {\n            await this.executeNestedWorkflow(itemState, target as SealedWorkflow<TContext, unknown, unknown>);\n          } else {\n            await this.executeAgent(itemState, target as Agent<TContext, unknown, TNextOutput>, ctx);\n          }\n          results[index] = itemState.output;\n        };\n\n        if (concurrency <= 1) {\n          for (let i = 0; i < items.length; i++) {\n            await executeItem(items[i], i);\n          }\n        } else {\n          for (let i = 0; i < items.length; i += concurrency) {\n            const batch = items.slice(i, i + concurrency);\n            await Promise.all(batch.map((item, j) => executeItem(item, i + j)));\n          }\n        }\n\n        state.output = results;\n      },\n    };\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    return new Workflow<TContext, TInput, TNextOutput[]>([...this.steps, node] as any, this.id);\n  }\n\n  // ── repeat: conditional loop ─────────────────────────────────\n\n  repeat(\n    target: Agent<TContext, TOutput, TOutput> | SealedWorkflow<TContext, TOutput, TOutput>,\n    options: RepeatOptions<TContext, TOutput>,\n  ): Workflow<TContext, TInput, TOutput> {\n    const maxIterations = options.maxIterations ?? 10;\n    const isWorkflow = target instanceof SealedWorkflow;\n    const id = isWorkflow ? (target.id ?? \"repeat\") : `repeat:${(target as Agent<TContext, TOutput, TOutput>).id}`;\n    const predicate: LoopPredicate<TContext, TOutput> = options.until\n      ?? (async (p) => !(await options.while!(p)));\n\n    const node: StepNode = {\n      type: \"step\",\n      id,\n      execute: async (state) => {\n        const ctx = state.ctx as TContext;\n\n        for (let i = 1; i <= maxIterations; i++) {\n          if (isWorkflow) {\n            await this.executeNestedWorkflow(state, target as SealedWorkflow<TContext, unknown, unknown>);\n          } else {\n            await this.executeAgent(state, target as Agent<TContext, TOutput, TOutput>, ctx);\n          }\n\n          const done = await predicate({\n            output: state.output as TOutput,\n            ctx: ctx as Readonly<TContext>,\n            iterations: i,\n          });\n          if (done) return;\n        }\n\n        throw new WorkflowLoopError(maxIterations, maxIterations);\n      },\n    };\n    return new Workflow<TContext, TInput, TOutput>([...this.steps, node], this.id);\n  }\n\n  // ── catch ─────────────────────────────────────────────────────\n\n  catch(\n    id: string,\n    fn: (params: { error: unknown; ctx: Readonly<TContext>; lastOutput: TOutput; stepId: string }) => MaybePromise<TOutput>\n  ): Workflow<TContext, TInput, TOutput> {\n    if (!this.steps.some(s => s.type === \"step\")) {\n      throw new Error(`Workflow: catch(\"${id}\") requires at least one preceding step.`);\n    }\n    const node: StepNode = {\n      type: \"catch\",\n      id,\n      catchFn: fn as (params: { error: unknown; ctx: unknown; lastOutput: unknown; stepId: string }) => MaybePromise<unknown>,\n    };\n    return new Workflow<TContext, TInput, TOutput>([...this.steps, node], this.id);\n  }\n\n  // ── finally (terminal — returns sealed workflow) ──────────────\n\n  finally(\n    id: string,\n    fn: (params: { ctx: Readonly<TContext> }) => MaybePromise<void>\n  ): SealedWorkflow<TContext, TInput, TOutput> {\n    const node: StepNode = {\n      type: \"finally\",\n      id,\n      execute: async (state) => {\n        await fn({ ctx: state.ctx as Readonly<TContext> });\n      },\n    };\n    return new SealedWorkflow<TContext, TInput, TOutput>([...this.steps, node], this.id);\n  }\n}\n\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,IAAAA,aAeO;;;ACfP,gBAAgF;AAEzE,IAAM,sBAAsB,uBAAO,IAAI,6BAA6B;AAepE,IAAM,eAAN,MAI8B;AAAA,EACnC,CAAU,mBAAmB,IAAI;AAAA,EAChB;AAAA,EAEjB,YAAY,QAAuD;AACjE,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,WAAW,SAAmC;AAC5C,UAAM,EAAE,SAAS,OAAO,aAAa,GAAG,QAAQ,IAAI,KAAK;AACzD,eAAO,gBAAK;AAAA,MACV,GAAG;AAAA,MACH,YAAY;AAAA,MACZ,SAAS,CAAC,OAAe,YAAmC,QAAQ,OAAO,SAAS,OAAO;AAAA;AAAA,IAE7F,CAAQ;AAAA,EACV;AACF;AAEO,SAAS,aAAuB;AACrC,SAAO,CACL,WAC4C,IAAI,aAAa,MAAM;AACvE;AAEO,SAAS,eAAyB,KAA8C;AACrF,SACE,OAAO,QAAQ,YACf,QAAQ,QACR,uBAAuB;AAE3B;;;ACnCO,SAAS,aACd,OACA,KACA,OACsC;AACtC,MAAI,OAAO,UAAU,YAAY;AAC/B,WAAQ,MAAiE,KAAK,KAAK;AAAA,EACrF;AACA,SAAO;AACT;AAOA,eAAsB,cAAc,QAAa,qBAAgD;AAC/F,MAAI,qBAAqB;AACvB,UAAM,SAAS,MAAM,OAAO;AAC5B,QAAI,WAAW,OAAW,QAAO;AAAA,EACnC;AACA,SAAO,MAAM,OAAO;AACtB;;;AFyDO,IAAM,QAAN,MAIL;AAAA,EACS;AAAA,EACA;AAAA,EACA;AAAA,EACQ;AAAA,EACA;AAAA,EACA,uBAAoD;AAAA;AAAA,EAEpD;AAAA,EACA;AAAA,EACA;AAAA,EAEjB,YAAY,QAAgD;AAC1D,SAAK,KAAK,OAAO;AACjB,SAAK,cAAc,OAAO,eAAe;AACzC,SAAK,YAAY,OAAO,WAAW;AACnC,SAAK,SAAS;AACd,SAAK,oBAAoB;AAAA,MACvB,OAAO;AAAA,MAAO,OAAO;AAAA,MAAQ,OAAO;AAAA,MACpC,OAAO;AAAA,MAAU,OAAO;AAAA,MAAO,OAAO;AAAA,MACtC,OAAO;AAAA,MAAY,OAAO;AAAA,IAC5B,EAAE,KAAK,OAAK,OAAO,MAAM,UAAU;AAInC,QAAI,CAAC,KAAK,mBAAmB;AAC3B,YAAM,WAAY,OAAO,SAAgD,CAAC;AAC1E,YAAM,cAAc,OAAO,OAAO,QAAQ,EAAE,KAAK,OAAK,eAAe,CAAC,CAAC;AACvE,UAAI,CAAC,aAAa;AAChB,aAAK,uBAAuB;AAAA,MAC9B;AAAA,IACF;AAIA,UAAM;AAAA,MACJ,IAAI;AAAA,MAAK,aAAa;AAAA,MAAO,OAAO;AAAA,MAAc,QAAQ;AAAA,MAC1D,OAAO;AAAA,MAAI,QAAQ;AAAA,MAAI,QAAQ;AAAA,MAAI,UAAU;AAAA,MAC7C,OAAO;AAAA,MAAI,aAAa;AAAA,MAAK,YAAY;AAAA,MAAK,UAAU;AAAA,MACxD;AAAA,MAAc;AAAA,MAAU,SAAS;AAAA,MACjC,GAAG;AAAA,IACL,IAAI;AACJ,SAAK,eAAe;AACpB,SAAK,gBAAgB;AACrB,SAAK,YAAY;AAAA,EACnB;AAAA,EAEA,MAAM,SAAS,QAAkB,MAA6F;AAC5H,UAAM,QAAQ,KAAK,CAAC;AACpB,UAAM,WAAW,MAAM,KAAK,cAAc,KAAK,KAAK;AACpD,UAAM,UAAU,KAAK,iBAAiB,UAAU,KAAK,KAAK;AAE1D,QAAI;AAEF,aAAO,UAAM,yBAAa,OAAc;AAAA,IAC1C,SAAS,OAAgB;AACvB,UAAI,KAAK,OAAO,SAAS;AACvB,cAAM,KAAK,OAAO,QAAQ,EAAE,OAAO,KAAK,MAAM,CAAC;AAAA,MACjD;AACA,YAAM;AAAA,IACR;AAAA,EACF;AAAA,EAEA,MAAM,OAAO,QAAkB,MAA2F;AACxH,UAAM,QAAQ,KAAK,CAAC;AACpB,UAAM,WAAW,MAAM,KAAK,cAAc,KAAK,KAAK;AACpD,UAAM,UAAU,KAAK,iBAAiB,UAAU,KAAK,KAAK;AAG1D,eAAO,uBAAW;AAAA,MAChB,GAAG;AAAA,MACH,SAAS,KAAK,OAAO,UACjB,CAAC,EAAE,MAAM,MAA0B,KAAK,OAAO,QAAS,EAAE,OAAO,KAAK,MAAM,CAAC,IAC7E;AAAA,IACN,CAAQ;AAAA,EACV;AAAA,EAEA,OAAO,KAAe,SAEb;AACP,WAAO,KAAK,mBAAmB,KAAK,OAAO;AAAA,EAC7C;AAAA,EAEA,eAAe,SAEa;AAC1B,QAAI,CAAC,KAAK,OAAO,OAAO;AACtB,YAAM,IAAI,MAAM,UAAU,KAAK,EAAE,8CAA8C;AAAA,IACjF;AAEA,WAAO;AAAA,MACL,CAAC,mBAAmB,GAAG;AAAA,MACvB,YAAY,CAAC,QAA4B,KAAK,mBAAmB,KAAiB,OAAO;AAAA,IAC3F;AAAA,EACF;AAAA,EAEQ,mBAAmB,KAAe,SAEjC;AACP,QAAI,CAAC,KAAK,OAAO,OAAO;AACtB,YAAM,IAAI,MAAM,UAAU,KAAK,EAAE,sCAAsC;AAAA,IACzE;AAEA,eAAO,iBAAK;AAAA,MACV,aAAa,KAAK;AAAA,MAClB,YAAY,KAAK,OAAO;AAAA,MACxB,SAAS,OAAO,cAAsB;AACpC,cAAM,SAAS,MAAO,KAAK,SAA2E,KAAK,SAAS;AACpH,YAAI,SAAS,UAAW,QAAO,QAAQ,UAAU,MAAM;AACvD,eAAO,cAAc,QAAQ,KAAK,SAAS;AAAA,MAC7C;AAAA;AAAA,IAEF,CAAQ;AAAA,EACV;AAAA;AAAA,EAGQ,iBAAiB,UAA+B,KAAe,OAAoC;AACzG,WAAO;AAAA,MACL,GAAG,KAAK;AAAA,MACR,OAAO,SAAS;AAAA,MAChB,OAAO,SAAS;AAAA,MAChB,aAAa,SAAS;AAAA,MACtB,YAAY,SAAS;AAAA,MACrB,UAAU,SAAS;AAAA,MACnB,GAAI,SAAS,WACT,EAAE,UAAU,SAAS,SAAS,IAC9B,EAAE,QAAQ,SAAS,UAAU,GAAG;AAAA,MACpC,GAAI,SAAS,SAAS,EAAE,QAAQ,SAAS,OAAO,IAAI,CAAC;AAAA,MACrD,GAAI,KAAK,OAAO,SAAS,EAAE,QAAQ,KAAK,OAAO,OAAO,IAAI,CAAC;AAAA,MAC3D,cAAc,KAAK,gBACf,CAAC,UAA6B,KAAK,cAAe,EAAE,QAAQ,OAAO,KAAK,MAAM,CAAC,IAC/E;AAAA,MACJ,UAAU,KAAK,YACX,CAAC,UAAyB,KAAK,UAAW,EAAE,QAAQ,OAAO,KAAK,MAAM,CAAC,IACvE;AAAA,IACN;AAAA,EACF;AAAA,EAEQ,cAAc,KAAe,OAAmE;AACtG,QAAI,CAAC,KAAK,mBAAmB;AAC3B,aAAO;AAAA,QACL,OAAO,KAAK,OAAO;AAAA,QACnB,QAAQ,KAAK,OAAO;AAAA,QACpB,QAAQ,KAAK,OAAO;AAAA,QACpB,UAAU,KAAK,OAAO;AAAA,QACtB,OAAO,KAAK,wBAAwB,KAAK;AAAA,UACtC,KAAK,OAAO,SAAgD,CAAC;AAAA,UAAG;AAAA,QACnE;AAAA,QACA,aAAa,KAAK,OAAO;AAAA,QACzB,YAAY,KAAK,OAAO;AAAA,QACxB,UAAU,KAAK,OAAO;AAAA,MACxB;AAAA,IACF;AACA,WAAO,KAAK,mBAAmB,KAAK,KAAK;AAAA,EAC3C;AAAA,EAEA,MAAc,mBAAmB,KAAe,OAA6C;AAC3F,UAAM,CAAC,OAAO,QAAQ,QAAQ,UAAU,UAAU,aAAa,YAAY,QAAQ,IAAI,MAAM,QAAQ,IAAI;AAAA,MACvG,aAAa,KAAK,OAAO,OAAO,KAAK,KAAK;AAAA,MAC1C,aAAa,KAAK,OAAO,QAAQ,KAAK,KAAK;AAAA,MAC3C,aAAa,KAAK,OAAO,QAAQ,KAAK,KAAK;AAAA,MAC3C,aAAa,KAAK,OAAO,UAAU,KAAK,KAAK;AAAA,MAC7C,aAAa,KAAK,OAAO,OAAO,KAAK,KAAK;AAAA,MAC1C,aAAa,KAAK,OAAO,aAAa,KAAK,KAAK;AAAA,MAChD,aAAa,KAAK,OAAO,YAAY,KAAK,KAAK;AAAA,MAC/C,aAAa,KAAK,OAAO,UAAU,KAAK,KAAK;AAAA,IAC/C,CAAC;AACD,UAAM,QAAQ,KAAK,aAAa,YAAY,CAAC,GAAG,GAAG;AACnD,WAAO,EAAE,OAAO,QAAQ,QAAQ,UAAU,OAAO,aAAa,YAAY,SAAS;AAAA,EACrF;AAAA,EAEQ,aACN,OACA,KACsB;AACtB,UAAM,UAAU,OAAO,QAAQ,KAAK;AACpC,QAAI,QAAQ,WAAW,EAAG,QAAO;AACjC,QAAI,cAAc;AAClB,UAAM,WAAiC,CAAC;AACxC,eAAW,CAAC,KAAK,cAAc,KAAK,SAAS;AAC3C,UAAI,eAAyB,cAAc,GAAG;AAC5C,sBAAc;AACd,iBAAS,GAAG,IAAI,eAAe,WAAW,GAAyB;AAAA,MACrE,OAAO;AACL,iBAAS,GAAG,IAAI;AAAA,MAClB;AAAA,IACF;AACA,WAAO,cAAc,WAAY;AAAA,EACnC;AACF;;;AGjSA,IAAAC,aAGO;AAMA,IAAM,sBAAN,cAAkC,MAAM;AAAA,EAC7C,YACkB,YAChB,SACA;AACA,UAAM,OAAO;AAHG;AAIhB,SAAK,OAAO;AAAA,EACd;AACF;AAEO,IAAM,oBAAN,cAAgC,MAAM;AAAA,EAC3C,YACkB,YACA,eAChB;AACA,UAAM,qCAAqC,aAAa,GAAG;AAH3C;AACA;AAGhB,SAAK,OAAO;AAAA,EACd;AACF;AAoFO,IAAM,iBAAN,MAIL;AAAA,EACS;AAAA,EACU;AAAA,EAET,YAAY,OAAgC,IAAa;AACjE,SAAK,QAAQ;AACb,SAAK,KAAK;AAAA,EACZ;AAAA;AAAA,EAIA,MAAM,SAAS,QAAkB,MAAkG;AACjI,UAAM,QAAQ,KAAK,CAAC;AACpB,UAAM,QAAsB;AAAA,MAC1B;AAAA,MACA,QAAQ;AAAA,MACR,MAAM;AAAA,IACR;AAEA,UAAM,KAAK,QAAQ,KAAK;AAExB,WAAO;AAAA,MACL,QAAQ,MAAM;AAAA,IAChB;AAAA,EACF;AAAA,EAEA,OACE,QACG,MAG4B;AAC/B,UAAM,QAAQ,KAAK,CAAC;AACpB,UAAM,UAAU,KAAK,CAAC;AAEtB,QAAI;AACJ,QAAI;AACJ,UAAM,gBAAgB,IAAI,QAAiB,CAAC,KAAK,QAAQ;AACvD,sBAAgB;AAChB,qBAAe;AAAA,IACjB,CAAC;AAID,kBAAc,MAAM,MAAM;AAAA,IAAC,CAAC;AAE5B,UAAM,aAAS,kCAAsB;AAAA,MACnC,SAAS,OAAO,EAAE,OAAO,MAAM;AAC7B,cAAM,QAAsB;AAAA,UAC1B;AAAA,UACA,QAAQ;AAAA,UACR,MAAM;AAAA,UACN;AAAA,QACF;AAEA,YAAI;AACF,gBAAM,KAAK,QAAQ,KAAK;AACxB,wBAAc,MAAM,MAAiB;AAAA,QACvC,SAAS,OAAO;AACd,uBAAc,KAAK;AACnB,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,MACA,GAAI,SAAS,UAAU,EAAE,SAAS,QAAQ,QAAQ,IAAI,CAAC;AAAA,MACvD,GAAI,SAAS,WAAW,EAAE,UAAU,QAAQ,SAAS,IAAI,CAAC;AAAA,IAC5D,CAAC;AAED,WAAO;AAAA,MACL;AAAA,MACA,QAAQ;AAAA,IACV;AAAA,EACF;AAAA;AAAA,EAIA,MAAgB,QAAQ,OAAoC;AAC1D,QAAI,KAAK,MAAM,WAAW,GAAG;AAC3B,YAAM,IAAI,MAAM,qFAAqF;AAAA,IACvG;AAEA,QAAI,eAA0D;AAE9D,eAAW,QAAQ,KAAK,OAAO;AAC7B,UAAI,KAAK,SAAS,WAAW;AAC3B,cAAM,KAAK,QAAQ,KAAK;AACxB;AAAA,MACF;AAEA,UAAI,KAAK,SAAS,SAAS;AACzB,YAAI,CAAC,aAAc;AACnB,YAAI;AACF,gBAAM,SAAS,MAAM,KAAK,QAAQ;AAAA,YAChC,OAAO,aAAa;AAAA,YACpB,KAAK,MAAM;AAAA,YACX,YAAY,MAAM;AAAA,YAClB,QAAQ,aAAa;AAAA,UACvB,CAAC;AACD,yBAAe;AAAA,QACjB,SAAS,YAAY;AACnB,yBAAe,EAAE,OAAO,YAAY,QAAQ,KAAK,GAAG;AAAA,QACtD;AACA;AAAA,MACF;AAGA,UAAI,aAAc;AAElB,UAAI;AACF,cAAM,KAAK,QAAQ,KAAK;AAAA,MAC1B,SAAS,OAAO;AACd,uBAAe,EAAE,OAAO,QAAQ,KAAK,GAAG;AAAA,MAC1C;AAAA,IACF;AAEA,QAAI,aAAc,OAAM,aAAa;AAAA,EACvC;AAAA;AAAA;AAAA;AAAA,EAMA,MAAgB,sBACd,OACA,UACe;AACf,UAAM,SAAS,QAAQ,KAAK;AAAA,EAC9B;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAgB,aACd,OAEA,OACA,KAEA,SACe;AACf,UAAM,QAAQ,MAAM;AACpB,UAAM,sBAAsB,MAAM;AAElC,QAAI,MAAM,SAAS,YAAY,MAAM,QAAQ;AAC3C,YAAM,SAAS,MAAO,MAAM,OAAwE,KAAK,MAAM,MAAM;AAErH,UAAI,SAAS,cAAc;AACzB,cAAM,QAAQ,aAAa,EAAE,QAAQ,QAAQ,MAAM,QAAQ,IAAI,CAAC;AAAA,MAClE,OAAO;AACL,cAAM,OAAO,MAAM,OAAO,kBAAkB,CAAC;AAAA,MAC/C;AAEA,UAAI,SAAS,gBAAgB;AAC3B,cAAM,QAAQ,eAAe,EAAE,QAAQ,KAAK,MAAM,CAAC;AAAA,MACrD;AAEA,UAAI,SAAS,iBAAiB;AAC5B,cAAM,SAAS,MAAM,QAAQ,gBAAgB,EAAE,QAAQ,KAAK,MAAM,CAAC;AAAA,MACrE,OAAO;AACL,cAAM,SAAS,MAAM,cAAc,QAAQ,mBAAmB;AAAA,MAChE;AAAA,IACF,OAAO;AACL,YAAM,SAAS,MAAO,MAAM,SAA4E,KAAK,MAAM,MAAM;AAEzH,UAAI,SAAS,kBAAkB;AAC7B,cAAM,QAAQ,iBAAiB,EAAE,QAAQ,KAAK,MAAM,CAAC;AAAA,MACvD;AAEA,UAAI,SAAS,mBAAmB;AAC9B,cAAM,SAAS,MAAM,QAAQ,kBAAkB,EAAE,QAAQ,KAAK,MAAM,CAAC;AAAA,MACvE,OAAO;AACL,cAAM,SAAS,MAAM,cAAc,QAAQ,mBAAmB;AAAA,MAChE;AAAA,IACF;AAAA,EACF;AACF;AAIO,IAAM,WAAN,MAAM,kBAIH,eAA0C;AAAA,EAE1C,YAAY,QAAiC,CAAC,GAAG,IAAa;AACpE,UAAM,OAAO,EAAE;AAAA,EACjB;AAAA,EAEA,OAAO,OAAgC,SAA+D;AACpG,WAAO,IAAI,UAAmC,CAAC,GAAG,SAAS,EAAE;AAAA,EAC/D;AAAA,EAEA,OAAO,KACL,OACA,SACqC;AAErC,WAAO,IAAI,UAAgC,CAAC,CAAC,EAAE,KAAK,OAAO,OAAO;AAAA,EACpE;AAAA;AAAA,EAwBA,KACE,QACA,aACyC;AAEzC,QAAI,kBAAkB,gBAAgB;AACpC,YAAM,WAAW;AACjB,YAAMC,QAAiB;AAAA,QACrB,MAAM;AAAA,QACN,IAAI,SAAS,MAAM;AAAA,QACnB,SAAS,OAAO,UAAU;AACxB,gBAAM,KAAK,sBAAsB,OAAO,QAAsD;AAAA,QAChG;AAAA,MACF;AAEA,aAAO,IAAI,UAAwC,CAAC,GAAG,KAAK,OAAOA,KAAI,GAAU,KAAK,EAAE;AAAA,IAC1F;AAGA,QAAI,OAAO,WAAW,UAAU;AAC9B,UAAI,OAAO,gBAAgB,YAAY;AACrC,cAAM,IAAI,MAAM,kBAAkB,MAAM,wCAAwC;AAAA,MAClF;AACA,YAAM,KAAK;AACX,YAAMA,QAAiB;AAAA,QACrB,MAAM;AAAA,QACN,IAAI;AAAA,QACJ,SAAS,OAAO,UAAU;AACxB,gBAAM,SAAS,MAAM,GAAG;AAAA,YACtB,KAAK,MAAM;AAAA,YACX,OAAO,MAAM;AAAA,UACf,CAAC;AAAA,QACH;AAAA,MACF;AAEA,aAAO,IAAI,UAAwC,CAAC,GAAG,KAAK,OAAOA,KAAI,GAAU,KAAK,EAAE;AAAA,IAC1F;AAGA,UAAM,QAAQ;AACd,UAAM,UAAU;AAChB,UAAM,OAAiB;AAAA,MACrB,MAAM;AAAA,MACN,IAAI,MAAM;AAAA,MACV,SAAS,OAAO,UAAU;AACxB,cAAM,MAAM,MAAM;AAClB,cAAM,KAAK,aAAa,OAAO,OAAO,KAAK,OAAO;AAAA,MACpD;AAAA,IACF;AAEA,WAAO,IAAI,UAAwC,CAAC,GAAG,KAAK,OAAO,IAAI,GAAU,KAAK,EAAE;AAAA,EAC1F;AAAA;AAAA,EAgBA,OACE,eACyC;AACzC,QAAI,MAAM,QAAQ,aAAa,GAAG;AAChC,aAAO,KAAK,gBAAgB,aAAa;AAAA,IAC3C;AACA,WAAO,KAAK,aAAa,aAAa;AAAA,EACxC;AAAA,EAEQ,gBACN,OACyC;AACzC,UAAM,OAAiB;AAAA,MACrB,MAAM;AAAA,MACN,IAAI;AAAA,MACJ,SAAS,OAAO,UAAU;AACxB,cAAM,MAAM,MAAM;AAClB,cAAM,QAAQ,MAAM;AAEpB,mBAAW,cAAc,OAAO;AAC9B,cAAI,WAAW,MAAM;AACnB,kBAAM,QAAQ,MAAM,WAAW,KAAK,EAAE,KAAK,MAAM,CAAC;AAClD,gBAAI,CAAC,MAAO;AAAA,UACd;AAGA,gBAAM,KAAK,aAAa,OAAO,WAAW,OAAO,KAAK,UAAU;AAChE;AAAA,QACF;AAEA,cAAM,IAAI,oBAAoB,aAAa,0FAA0F,KAAK,UAAU,KAAK,CAAC,EAAE;AAAA,MAC9J;AAAA,IACF;AAEA,WAAO,IAAI,UAAwC,CAAC,GAAG,KAAK,OAAO,IAAI,GAAU,KAAK,EAAE;AAAA,EAC1F;AAAA,EAEQ,aACN,QACyC;AACzC,UAAM,OAAiB;AAAA,MACrB,MAAM;AAAA,MACN,IAAI;AAAA,MACJ,SAAS,OAAO,UAAU;AACxB,cAAM,MAAM,MAAM;AAClB,cAAM,QAAQ,MAAM;AACpB,cAAM,MAAM,MAAM,OAAO,OAAO,EAAE,KAAK,MAAM,CAAC;AAE9C,YAAI,QAAQ,OAAO,OAAO,GAAG;AAC7B,YAAI,CAAC,OAAO;AACV,cAAI,OAAO,UAAU;AACnB,oBAAQ,OAAO;AAAA,UACjB,OAAO;AACL,kBAAM,IAAI,oBAAoB,UAAU,2BAA2B,GAAG,+CAA+C,OAAO,KAAK,OAAO,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE;AAAA,UAC9J;AAAA,QACF;AAEA,cAAM,KAAK,aAAa,OAAO,OAAO,KAAK,MAAM;AAAA,MACnD;AAAA,IACF;AAEA,WAAO,IAAI,UAAwC,CAAC,GAAG,KAAK,OAAO,IAAI,GAAU,KAAK,EAAE;AAAA,EAC1F;AAAA;AAAA,EAIA,QACE,QACA,SAC2C;AAC3C,UAAM,cAAc,SAAS,eAAe;AAC5C,UAAM,aAAa,kBAAkB;AACrC,UAAM,KAAK,aAAc,OAAO,MAAM,YAAa,WAAY,OAA4D,EAAE;AAE7H,UAAM,OAAiB;AAAA,MACrB,MAAM;AAAA,MACN;AAAA,MACA,SAAS,OAAO,UAAU;AACxB,cAAM,QAAQ,MAAM;AACpB,YAAI,CAAC,MAAM,QAAQ,KAAK,GAAG;AACzB,gBAAM,IAAI,MAAM,YAAY,EAAE,gCAAgC,OAAO,KAAK,EAAE;AAAA,QAC9E;AAEA,cAAM,MAAM,MAAM;AAClB,cAAM,UAAqB,IAAI,MAAM,MAAM,MAAM;AAKjD,cAAM,cAAc,OAAO,MAAe,UAAkB;AAC1D,gBAAM,YAA0B,EAAE,KAAK,MAAM,KAAK,QAAQ,MAAM,MAAM,WAAW;AACjF,cAAI,YAAY;AACd,kBAAM,KAAK,sBAAsB,WAAW,MAAoD;AAAA,UAClG,OAAO;AACL,kBAAM,KAAK,aAAa,WAAW,QAAiD,GAAG;AAAA,UACzF;AACA,kBAAQ,KAAK,IAAI,UAAU;AAAA,QAC7B;AAEA,YAAI,eAAe,GAAG;AACpB,mBAAS,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK;AACrC,kBAAM,YAAY,MAAM,CAAC,GAAG,CAAC;AAAA,UAC/B;AAAA,QACF,OAAO;AACL,mBAAS,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK,aAAa;AAClD,kBAAM,QAAQ,MAAM,MAAM,GAAG,IAAI,WAAW;AAC5C,kBAAM,QAAQ,IAAI,MAAM,IAAI,CAAC,MAAM,MAAM,YAAY,MAAM,IAAI,CAAC,CAAC,CAAC;AAAA,UACpE;AAAA,QACF;AAEA,cAAM,SAAS;AAAA,MACjB;AAAA,IACF;AAEA,WAAO,IAAI,UAA0C,CAAC,GAAG,KAAK,OAAO,IAAI,GAAU,KAAK,EAAE;AAAA,EAC5F;AAAA;AAAA,EAIA,OACE,QACA,SACqC;AACrC,UAAM,gBAAgB,QAAQ,iBAAiB;AAC/C,UAAM,aAAa,kBAAkB;AACrC,UAAM,KAAK,aAAc,OAAO,MAAM,WAAY,UAAW,OAA6C,EAAE;AAC5G,UAAM,YAA8C,QAAQ,UACtD,OAAO,MAAM,CAAE,MAAM,QAAQ,MAAO,CAAC;AAE3C,UAAM,OAAiB;AAAA,MACrB,MAAM;AAAA,MACN;AAAA,MACA,SAAS,OAAO,UAAU;AACxB,cAAM,MAAM,MAAM;AAElB,iBAAS,IAAI,GAAG,KAAK,eAAe,KAAK;AACvC,cAAI,YAAY;AACd,kBAAM,KAAK,sBAAsB,OAAO,MAAoD;AAAA,UAC9F,OAAO;AACL,kBAAM,KAAK,aAAa,OAAO,QAA6C,GAAG;AAAA,UACjF;AAEA,gBAAM,OAAO,MAAM,UAAU;AAAA,YAC3B,QAAQ,MAAM;AAAA,YACd;AAAA,YACA,YAAY;AAAA,UACd,CAAC;AACD,cAAI,KAAM;AAAA,QACZ;AAEA,cAAM,IAAI,kBAAkB,eAAe,aAAa;AAAA,MAC1D;AAAA,IACF;AACA,WAAO,IAAI,UAAoC,CAAC,GAAG,KAAK,OAAO,IAAI,GAAG,KAAK,EAAE;AAAA,EAC/E;AAAA;AAAA,EAIA,MACE,IACA,IACqC;AACrC,QAAI,CAAC,KAAK,MAAM,KAAK,OAAK,EAAE,SAAS,MAAM,GAAG;AAC5C,YAAM,IAAI,MAAM,oBAAoB,EAAE,0CAA0C;AAAA,IAClF;AACA,UAAM,OAAiB;AAAA,MACrB,MAAM;AAAA,MACN;AAAA,MACA,SAAS;AAAA,IACX;AACA,WAAO,IAAI,UAAoC,CAAC,GAAG,KAAK,OAAO,IAAI,GAAG,KAAK,EAAE;AAAA,EAC/E;AAAA;AAAA,EAIA,QACE,IACA,IAC2C;AAC3C,UAAM,OAAiB;AAAA,MACrB,MAAM;AAAA,MACN;AAAA,MACA,SAAS,OAAO,UAAU;AACxB,cAAM,GAAG,EAAE,KAAK,MAAM,IAA0B,CAAC;AAAA,MACnD;AAAA,IACF;AACA,WAAO,IAAI,eAA0C,CAAC,GAAG,KAAK,OAAO,IAAI,GAAG,KAAK,EAAE;AAAA,EACrF;AACF;","names":["import_ai","import_ai","node"]}

package/dist/index.d.cts

@@ -0,0 +1,240 @@
+import { Tool, FlexibleSchema, ToolExecutionOptions, streamText, generateText, Output, LanguageModel, ModelMessage, ToolChoice, ToolSet, StopCondition, OnStepFinishEvent, OnFinishEvent, GenerateTextResult as GenerateTextResult$1, StreamTextResult as StreamTextResult$1, UIMessageStreamWriter } from 'ai';
+import { ZodType } from 'zod';
+
+declare const TOOL_PROVIDER_BRAND: unique symbol;
+type ToolProviderConfig<TContext, TInput, TOutput> = {
+    description?: string;
+    input: FlexibleSchema<TInput>;
+    output?: FlexibleSchema<unknown>;
+    providerOptions?: unknown;
+    execute: (input: TInput, ctx: Readonly<TContext>, options?: ToolExecutionOptions) => Promise<TOutput>;
+};
+interface IToolProvider<TContext> {
+    readonly [TOOL_PROVIDER_BRAND]: true;
+    createTool(context: Readonly<TContext>): Tool;
+}
+declare class ToolProvider<TContext, TInput = unknown, TOutput = unknown> implements IToolProvider<TContext> {
+    readonly [TOOL_PROVIDER_BRAND]: true;
+    private readonly config;
+    constructor(config: ToolProviderConfig<TContext, TInput, TOutput>);
+    createTool(context: Readonly<TContext>): Tool;
+}
+declare function defineTool<TContext>(): <TInput, TOutput>(config: ToolProviderConfig<TContext, TInput, TOutput>) => ToolProvider<TContext, TInput, TOutput>;
+
+type MaybePromise<T> = T | Promise<T>;
+/**
+ * A value that can be static or derived from context and input.
+ * Used for agent config fields that may need runtime resolution.
+ *
+ * Functions may return a Promise for async resolution; static values are always sync.
+ */
+type Resolvable<TCtx, TInput, TValue> = TValue | ((ctx: Readonly<TCtx>, input: TInput) => TValue | Promise<TValue>);
+
+type OutputType<T = any> = ReturnType<typeof Output.object<T>>;
+
+type AgentToolSet<TContext> = Record<string, Tool | IToolProvider<TContext>>;
+type GenerateTextResult<TOOLS extends ToolSet = ToolSet, OUTPUT extends OutputType = OutputType> = GenerateTextResult$1<TOOLS, OUTPUT>;
+type StreamTextResult<TOOLS extends ToolSet = ToolSet, OUTPUT extends OutputType = OutputType> = StreamTextResult$1<TOOLS, OUTPUT>;
+type StreamTextOptions = Parameters<typeof streamText>[0];
+type GenerateTextOptions = Parameters<typeof generateText>[0];
+type ManagedKeys = 'model' | 'system' | 'prompt' | 'messages' | 'tools' | 'activeTools' | 'toolChoice' | 'stopWhen' | 'output' | 'onFinish' | 'onStepFinish' | 'onError';
+type AIPassthroughOptions = Omit<StreamTextOptions, ManagedKeys> & Omit<GenerateTextOptions, ManagedKeys>;
+interface AgentConfig<TContext, TInput = void, TOutput = void> extends AIPassthroughOptions {
+    id: string;
+    description?: string;
+    input?: ZodType<TInput>;
+    output?: OutputType<TOutput>;
+    model: Resolvable<TContext, TInput, LanguageModel>;
+    system?: Resolvable<TContext, TInput, string>;
+    prompt?: Resolvable<TContext, TInput, string>;
+    messages?: Resolvable<TContext, TInput, ModelMessage[]>;
+    tools?: Resolvable<TContext, TInput, AgentToolSet<TContext>>;
+    activeTools?: Resolvable<TContext, TInput, string[]>;
+    toolChoice?: Resolvable<TContext, TInput, ToolChoice<ToolSet>>;
+    stopWhen?: Resolvable<TContext, TInput, StopCondition<ToolSet> | Array<StopCondition<ToolSet>>>;
+    onStepFinish?: (params: {
+        result: OnStepFinishEvent;
+        ctx: Readonly<TContext>;
+        input: TInput;
+    }) => MaybePromise<void>;
+    onFinish?: (params: {
+        result: OnFinishEvent;
+        ctx: Readonly<TContext>;
+        input: TInput;
+    }) => MaybePromise<void>;
+    onError?: (params: {
+        error: unknown;
+        ctx: Readonly<TContext>;
+        input: TInput;
+    }) => MaybePromise<void>;
+}
+declare class Agent<TContext, TInput = void, TOutput = void> {
+    readonly id: string;
+    readonly description: string;
+    readonly hasOutput: boolean;
+    private readonly config;
+    private readonly _hasDynamicConfig;
+    private readonly _resolvedStaticTools;
+    private readonly _passthrough;
+    private readonly _onStepFinish;
+    private readonly _onFinish;
+    constructor(config: AgentConfig<TContext, TInput, TOutput>);
+    generate(ctx: TContext, ...args: TInput extends void ? [input?: TInput] : [input: TInput]): Promise<GenerateTextResult>;
+    stream(ctx: TContext, ...args: TInput extends void ? [input?: TInput] : [input: TInput]): Promise<StreamTextResult>;
+    asTool(ctx: TContext, options?: {
+        mapOutput?: (result: GenerateTextResult) => MaybePromise<TOutput>;
+    }): Tool;
+    asToolProvider(options?: {
+        mapOutput?: (result: GenerateTextResult) => MaybePromise<TOutput>;
+    }): IToolProvider<TContext>;
+    private createToolInstance;
+    private buildCallOptions;
+    private resolveConfig;
+    private resolveConfigAsync;
+    private resolveTools;
+}
+
+declare class WorkflowBranchError extends Error {
+    readonly branchType: "predicate" | "select";
+    constructor(branchType: "predicate" | "select", message: string);
+}
+declare class WorkflowLoopError extends Error {
+    readonly iterations: number;
+    readonly maxIterations: number;
+    constructor(iterations: number, maxIterations: number);
+}
+interface AgentStepHooks<TContext, TOutput, TNextOutput> {
+    mapGenerateResult?: (params: {
+        result: GenerateTextResult;
+        ctx: Readonly<TContext>;
+        input: TOutput;
+    }) => MaybePromise<TNextOutput>;
+    mapStreamResult?: (params: {
+        result: StreamTextResult;
+        ctx: Readonly<TContext>;
+        input: TOutput;
+    }) => MaybePromise<TNextOutput>;
+    onGenerateResult?: (params: {
+        result: GenerateTextResult;
+        ctx: Readonly<TContext>;
+        input: TOutput;
+    }) => MaybePromise<void>;
+    onStreamResult?: (params: {
+        result: StreamTextResult;
+        ctx: Readonly<TContext>;
+        input: TOutput;
+    }) => MaybePromise<void>;
+    handleStream?: (params: {
+        result: StreamTextResult;
+        writer: UIMessageStreamWriter;
+        ctx: Readonly<TContext>;
+    }) => MaybePromise<void>;
+}
+type StepOptions<TContext, TOutput, TNextOutput> = AgentStepHooks<TContext, TOutput, TNextOutput>;
+interface BranchCase<TContext, TOutput, TNextOutput> extends AgentStepHooks<TContext, TOutput, TNextOutput> {
+    when?: (params: {
+        ctx: Readonly<TContext>;
+        input: TOutput;
+    }) => MaybePromise<boolean>;
+    agent: Agent<TContext, TOutput, TNextOutput>;
+}
+interface BranchSelect<TContext, TOutput, TKeys extends string, TNextOutput> extends AgentStepHooks<TContext, TOutput, TNextOutput> {
+    select: (params: {
+        ctx: Readonly<TContext>;
+        input: TOutput;
+    }) => MaybePromise<TKeys>;
+    agents: Record<TKeys, Agent<TContext, any, TNextOutput>>;
+    fallback?: Agent<TContext, any, TNextOutput>;
+}
+interface WorkflowResult<TOutput> {
+    output: TOutput;
+}
+interface WorkflowStreamResult<TOutput> {
+    stream: ReadableStream;
+    output: Promise<TOutput>;
+}
+interface WorkflowStreamOptions {
+    onError?: (error: unknown) => string;
+    onFinish?: () => MaybePromise<void>;
+}
+type LoopPredicate<TContext, TOutput> = (params: {
+    output: TOutput;
+    ctx: Readonly<TContext>;
+    iterations: number;
+}) => MaybePromise<boolean>;
+type RepeatOptions<TContext, TOutput> = {
+    until: LoopPredicate<TContext, TOutput>;
+    while?: never;
+    maxIterations?: number;
+} | {
+    while: LoopPredicate<TContext, TOutput>;
+    until?: never;
+    maxIterations?: number;
+};
+type ElementOf<T> = T extends readonly (infer E)[] ? E : never;
+type StepNode = {
+    readonly type: "step";
+    readonly id: string;
+    readonly execute: (state: RuntimeState) => MaybePromise<void>;
+} | {
+    readonly type: "catch";
+    readonly id: string;
+    readonly catchFn: (params: {
+        error: unknown;
+        ctx: unknown;
+        lastOutput: unknown;
+        stepId: string;
+    }) => MaybePromise<unknown>;
+} | {
+    readonly type: "finally";
+    readonly id: string;
+    readonly execute: (state: RuntimeState) => MaybePromise<void>;
+};
+interface RuntimeState {
+    ctx: unknown;
+    output: unknown;
+    mode: "generate" | "stream";
+    writer?: UIMessageStreamWriter;
+}
+declare class SealedWorkflow<TContext, TInput = void, TOutput = void> {
+    readonly id?: string;
+    protected readonly steps: ReadonlyArray<StepNode>;
+    protected constructor(steps: ReadonlyArray<StepNode>, id?: string);
+    generate(ctx: TContext, ...args: TInput extends void ? [input?: TInput] : [input: TInput]): Promise<WorkflowResult<TOutput>>;
+    stream(ctx: TContext, ...args: TInput extends void ? [input?: TInput, options?: WorkflowStreamOptions] : [input: TInput, options?: WorkflowStreamOptions]): WorkflowStreamResult<TOutput>;
+    protected execute(state: RuntimeState): Promise<void>;
+    protected executeNestedWorkflow(state: RuntimeState, workflow: SealedWorkflow<TContext, unknown, unknown>): Promise<void>;
+    protected executeAgent<TAgentInput, TNextOutput>(state: RuntimeState, agent: Agent<TContext, any, TNextOutput>, ctx: TContext, options?: AgentStepHooks<TContext, any, TNextOutput>): Promise<void>;
+}
+declare class Workflow<TContext, TInput = void, TOutput = void> extends SealedWorkflow<TContext, TInput, TOutput> {
+    private constructor();
+    static create<TContext, TInput = void>(options?: {
+        id?: string;
+    }): Workflow<TContext, TInput, TInput>;
+    static from<TContext, TInput, TOutput>(agent: Agent<TContext, TInput, TOutput>, options?: StepOptions<TContext, TInput, TOutput>): Workflow<TContext, TInput, TOutput>;
+    step<TNextOutput>(agent: Agent<TContext, TOutput, TNextOutput>, options?: StepOptions<TContext, TOutput, TNextOutput>): Workflow<TContext, TInput, TNextOutput>;
+    step<TNextOutput>(workflow: SealedWorkflow<TContext, TOutput, TNextOutput>): Workflow<TContext, TInput, TNextOutput>;
+    step<TNextOutput>(id: string, fn: (params: {
+        ctx: Readonly<TContext>;
+        input: TOutput;
+    }) => MaybePromise<TNextOutput>): Workflow<TContext, TInput, TNextOutput>;
+    branch<TNextOutput>(cases: BranchCase<TContext, TOutput, TNextOutput>[]): Workflow<TContext, TInput, TNextOutput>;
+    branch<TKeys extends string, TNextOutput>(config: BranchSelect<TContext, TOutput, TKeys, TNextOutput>): Workflow<TContext, TInput, TNextOutput>;
+    private branchPredicate;
+    private branchSelect;
+    foreach<TNextOutput>(target: Agent<TContext, ElementOf<TOutput>, TNextOutput> | SealedWorkflow<TContext, ElementOf<TOutput>, TNextOutput>, options?: {
+        concurrency?: number;
+    }): Workflow<TContext, TInput, TNextOutput[]>;
+    repeat(target: Agent<TContext, TOutput, TOutput> | SealedWorkflow<TContext, TOutput, TOutput>, options: RepeatOptions<TContext, TOutput>): Workflow<TContext, TInput, TOutput>;
+    catch(id: string, fn: (params: {
+        error: unknown;
+        ctx: Readonly<TContext>;
+        lastOutput: TOutput;
+        stepId: string;
+    }) => MaybePromise<TOutput>): Workflow<TContext, TInput, TOutput>;
+    finally(id: string, fn: (params: {
+        ctx: Readonly<TContext>;
+    }) => MaybePromise<void>): SealedWorkflow<TContext, TInput, TOutput>;
+}
+
+export { Agent, type AgentConfig, type AgentStepHooks, type BranchCase, type BranchSelect, type GenerateTextResult, type IToolProvider, type MaybePromise, type RepeatOptions, type Resolvable, SealedWorkflow, type StepOptions, type StreamTextResult, type ToolProviderConfig, Workflow, WorkflowBranchError, WorkflowLoopError, type WorkflowResult, type WorkflowStreamOptions, type WorkflowStreamResult, defineTool };

package/dist/index.d.ts

@@ -0,0 +1,240 @@
+import { Tool, FlexibleSchema, ToolExecutionOptions, streamText, generateText, Output, LanguageModel, ModelMessage, ToolChoice, ToolSet, StopCondition, OnStepFinishEvent, OnFinishEvent, GenerateTextResult as GenerateTextResult$1, StreamTextResult as StreamTextResult$1, UIMessageStreamWriter } from 'ai';
+import { ZodType } from 'zod';
+
+declare const TOOL_PROVIDER_BRAND: unique symbol;
+type ToolProviderConfig<TContext, TInput, TOutput> = {
+    description?: string;
+    input: FlexibleSchema<TInput>;
+    output?: FlexibleSchema<unknown>;
+    providerOptions?: unknown;
+    execute: (input: TInput, ctx: Readonly<TContext>, options?: ToolExecutionOptions) => Promise<TOutput>;
+};
+interface IToolProvider<TContext> {
+    readonly [TOOL_PROVIDER_BRAND]: true;
+    createTool(context: Readonly<TContext>): Tool;
+}
+declare class ToolProvider<TContext, TInput = unknown, TOutput = unknown> implements IToolProvider<TContext> {
+    readonly [TOOL_PROVIDER_BRAND]: true;
+    private readonly config;
+    constructor(config: ToolProviderConfig<TContext, TInput, TOutput>);
+    createTool(context: Readonly<TContext>): Tool;
+}
+declare function defineTool<TContext>(): <TInput, TOutput>(config: ToolProviderConfig<TContext, TInput, TOutput>) => ToolProvider<TContext, TInput, TOutput>;
+
+type MaybePromise<T> = T | Promise<T>;
+/**
+ * A value that can be static or derived from context and input.
+ * Used for agent config fields that may need runtime resolution.
+ *
+ * Functions may return a Promise for async resolution; static values are always sync.
+ */
+type Resolvable<TCtx, TInput, TValue> = TValue | ((ctx: Readonly<TCtx>, input: TInput) => TValue | Promise<TValue>);
+
+type OutputType<T = any> = ReturnType<typeof Output.object<T>>;
+
+type AgentToolSet<TContext> = Record<string, Tool | IToolProvider<TContext>>;
+type GenerateTextResult<TOOLS extends ToolSet = ToolSet, OUTPUT extends OutputType = OutputType> = GenerateTextResult$1<TOOLS, OUTPUT>;
+type StreamTextResult<TOOLS extends ToolSet = ToolSet, OUTPUT extends OutputType = OutputType> = StreamTextResult$1<TOOLS, OUTPUT>;
+type StreamTextOptions = Parameters<typeof streamText>[0];
+type GenerateTextOptions = Parameters<typeof generateText>[0];
+type ManagedKeys = 'model' | 'system' | 'prompt' | 'messages' | 'tools' | 'activeTools' | 'toolChoice' | 'stopWhen' | 'output' | 'onFinish' | 'onStepFinish' | 'onError';
+type AIPassthroughOptions = Omit<StreamTextOptions, ManagedKeys> & Omit<GenerateTextOptions, ManagedKeys>;
+interface AgentConfig<TContext, TInput = void, TOutput = void> extends AIPassthroughOptions {
+    id: string;
+    description?: string;
+    input?: ZodType<TInput>;
+    output?: OutputType<TOutput>;
+    model: Resolvable<TContext, TInput, LanguageModel>;
+    system?: Resolvable<TContext, TInput, string>;
+    prompt?: Resolvable<TContext, TInput, string>;
+    messages?: Resolvable<TContext, TInput, ModelMessage[]>;
+    tools?: Resolvable<TContext, TInput, AgentToolSet<TContext>>;
+    activeTools?: Resolvable<TContext, TInput, string[]>;
+    toolChoice?: Resolvable<TContext, TInput, ToolChoice<ToolSet>>;
+    stopWhen?: Resolvable<TContext, TInput, StopCondition<ToolSet> | Array<StopCondition<ToolSet>>>;
+    onStepFinish?: (params: {
+        result: OnStepFinishEvent;
+        ctx: Readonly<TContext>;
+        input: TInput;
+    }) => MaybePromise<void>;
+    onFinish?: (params: {
+        result: OnFinishEvent;
+        ctx: Readonly<TContext>;
+        input: TInput;
+    }) => MaybePromise<void>;
+    onError?: (params: {
+        error: unknown;
+        ctx: Readonly<TContext>;
+        input: TInput;
+    }) => MaybePromise<void>;
+}
+declare class Agent<TContext, TInput = void, TOutput = void> {
+    readonly id: string;
+    readonly description: string;
+    readonly hasOutput: boolean;
+    private readonly config;
+    private readonly _hasDynamicConfig;
+    private readonly _resolvedStaticTools;
+    private readonly _passthrough;
+    private readonly _onStepFinish;
+    private readonly _onFinish;
+    constructor(config: AgentConfig<TContext, TInput, TOutput>);
+    generate(ctx: TContext, ...args: TInput extends void ? [input?: TInput] : [input: TInput]): Promise<GenerateTextResult>;
+    stream(ctx: TContext, ...args: TInput extends void ? [input?: TInput] : [input: TInput]): Promise<StreamTextResult>;
+    asTool(ctx: TContext, options?: {
+        mapOutput?: (result: GenerateTextResult) => MaybePromise<TOutput>;
+    }): Tool;
+    asToolProvider(options?: {
+        mapOutput?: (result: GenerateTextResult) => MaybePromise<TOutput>;
+    }): IToolProvider<TContext>;
+    private createToolInstance;
+    private buildCallOptions;
+    private resolveConfig;
+    private resolveConfigAsync;
+    private resolveTools;
+}
+
+declare class WorkflowBranchError extends Error {
+    readonly branchType: "predicate" | "select";
+    constructor(branchType: "predicate" | "select", message: string);
+}
+declare class WorkflowLoopError extends Error {
+    readonly iterations: number;
+    readonly maxIterations: number;
+    constructor(iterations: number, maxIterations: number);
+}
+interface AgentStepHooks<TContext, TOutput, TNextOutput> {
+    mapGenerateResult?: (params: {
+        result: GenerateTextResult;
+        ctx: Readonly<TContext>;
+        input: TOutput;
+    }) => MaybePromise<TNextOutput>;
+    mapStreamResult?: (params: {
+        result: StreamTextResult;
+        ctx: Readonly<TContext>;
+        input: TOutput;
+    }) => MaybePromise<TNextOutput>;
+    onGenerateResult?: (params: {
+        result: GenerateTextResult;
+        ctx: Readonly<TContext>;
+        input: TOutput;
+    }) => MaybePromise<void>;
+    onStreamResult?: (params: {
+        result: StreamTextResult;
+        ctx: Readonly<TContext>;
+        input: TOutput;
+    }) => MaybePromise<void>;
+    handleStream?: (params: {
+        result: StreamTextResult;
+        writer: UIMessageStreamWriter;
+        ctx: Readonly<TContext>;
+    }) => MaybePromise<void>;
+}
+type StepOptions<TContext, TOutput, TNextOutput> = AgentStepHooks<TContext, TOutput, TNextOutput>;
+interface BranchCase<TContext, TOutput, TNextOutput> extends AgentStepHooks<TContext, TOutput, TNextOutput> {
+    when?: (params: {
+        ctx: Readonly<TContext>;
+        input: TOutput;
+    }) => MaybePromise<boolean>;
+    agent: Agent<TContext, TOutput, TNextOutput>;
+}
+interface BranchSelect<TContext, TOutput, TKeys extends string, TNextOutput> extends AgentStepHooks<TContext, TOutput, TNextOutput> {
+    select: (params: {
+        ctx: Readonly<TContext>;
+        input: TOutput;
+    }) => MaybePromise<TKeys>;
+    agents: Record<TKeys, Agent<TContext, any, TNextOutput>>;
+    fallback?: Agent<TContext, any, TNextOutput>;
+}
+interface WorkflowResult<TOutput> {
+    output: TOutput;
+}
+interface WorkflowStreamResult<TOutput> {
+    stream: ReadableStream;
+    output: Promise<TOutput>;
+}
+interface WorkflowStreamOptions {
+    onError?: (error: unknown) => string;
+    onFinish?: () => MaybePromise<void>;
+}
+type LoopPredicate<TContext, TOutput> = (params: {
+    output: TOutput;
+    ctx: Readonly<TContext>;
+    iterations: number;
+}) => MaybePromise<boolean>;
+type RepeatOptions<TContext, TOutput> = {
+    until: LoopPredicate<TContext, TOutput>;
+    while?: never;
+    maxIterations?: number;
+} | {
+    while: LoopPredicate<TContext, TOutput>;
+    until?: never;
+    maxIterations?: number;
+};
+type ElementOf<T> = T extends readonly (infer E)[] ? E : never;
+type StepNode = {
+    readonly type: "step";
+    readonly id: string;
+    readonly execute: (state: RuntimeState) => MaybePromise<void>;
+} | {
+    readonly type: "catch";
+    readonly id: string;
+    readonly catchFn: (params: {
+        error: unknown;
+        ctx: unknown;
+        lastOutput: unknown;
+        stepId: string;
+    }) => MaybePromise<unknown>;
+} | {
+    readonly type: "finally";
+    readonly id: string;
+    readonly execute: (state: RuntimeState) => MaybePromise<void>;
+};
+interface RuntimeState {
+    ctx: unknown;
+    output: unknown;
+    mode: "generate" | "stream";
+    writer?: UIMessageStreamWriter;
+}
+declare class SealedWorkflow<TContext, TInput = void, TOutput = void> {
+    readonly id?: string;
+    protected readonly steps: ReadonlyArray<StepNode>;
+    protected constructor(steps: ReadonlyArray<StepNode>, id?: string);
+    generate(ctx: TContext, ...args: TInput extends void ? [input?: TInput] : [input: TInput]): Promise<WorkflowResult<TOutput>>;
+    stream(ctx: TContext, ...args: TInput extends void ? [input?: TInput, options?: WorkflowStreamOptions] : [input: TInput, options?: WorkflowStreamOptions]): WorkflowStreamResult<TOutput>;
+    protected execute(state: RuntimeState): Promise<void>;
+    protected executeNestedWorkflow(state: RuntimeState, workflow: SealedWorkflow<TContext, unknown, unknown>): Promise<void>;
+    protected executeAgent<TAgentInput, TNextOutput>(state: RuntimeState, agent: Agent<TContext, any, TNextOutput>, ctx: TContext, options?: AgentStepHooks<TContext, any, TNextOutput>): Promise<void>;
+}
+declare class Workflow<TContext, TInput = void, TOutput = void> extends SealedWorkflow<TContext, TInput, TOutput> {
+    private constructor();
+    static create<TContext, TInput = void>(options?: {
+        id?: string;
+    }): Workflow<TContext, TInput, TInput>;
+    static from<TContext, TInput, TOutput>(agent: Agent<TContext, TInput, TOutput>, options?: StepOptions<TContext, TInput, TOutput>): Workflow<TContext, TInput, TOutput>;
+    step<TNextOutput>(agent: Agent<TContext, TOutput, TNextOutput>, options?: StepOptions<TContext, TOutput, TNextOutput>): Workflow<TContext, TInput, TNextOutput>;
+    step<TNextOutput>(workflow: SealedWorkflow<TContext, TOutput, TNextOutput>): Workflow<TContext, TInput, TNextOutput>;
+    step<TNextOutput>(id: string, fn: (params: {
+        ctx: Readonly<TContext>;
+        input: TOutput;
+    }) => MaybePromise<TNextOutput>): Workflow<TContext, TInput, TNextOutput>;
+    branch<TNextOutput>(cases: BranchCase<TContext, TOutput, TNextOutput>[]): Workflow<TContext, TInput, TNextOutput>;
+    branch<TKeys extends string, TNextOutput>(config: BranchSelect<TContext, TOutput, TKeys, TNextOutput>): Workflow<TContext, TInput, TNextOutput>;
+    private branchPredicate;
+    private branchSelect;
+    foreach<TNextOutput>(target: Agent<TContext, ElementOf<TOutput>, TNextOutput> | SealedWorkflow<TContext, ElementOf<TOutput>, TNextOutput>, options?: {
+        concurrency?: number;
+    }): Workflow<TContext, TInput, TNextOutput[]>;
+    repeat(target: Agent<TContext, TOutput, TOutput> | SealedWorkflow<TContext, TOutput, TOutput>, options: RepeatOptions<TContext, TOutput>): Workflow<TContext, TInput, TOutput>;
+    catch(id: string, fn: (params: {
+        error: unknown;
+        ctx: Readonly<TContext>;
+        lastOutput: TOutput;
+        stepId: string;
+    }) => MaybePromise<TOutput>): Workflow<TContext, TInput, TOutput>;
+    finally(id: string, fn: (params: {
+        ctx: Readonly<TContext>;
+    }) => MaybePromise<void>): SealedWorkflow<TContext, TInput, TOutput>;
+}
+
+export { Agent, type AgentConfig, type AgentStepHooks, type BranchCase, type BranchSelect, type GenerateTextResult, type IToolProvider, type MaybePromise, type RepeatOptions, type Resolvable, SealedWorkflow, type StepOptions, type StreamTextResult, type ToolProviderConfig, Workflow, WorkflowBranchError, WorkflowLoopError, type WorkflowResult, type WorkflowStreamOptions, type WorkflowStreamResult, defineTool };