@anvia/core 0.6.3 → 0.7.1

package/dist/index.js

@@ -2,20 +2,21 @@ import {
   SkillValidationError,
   loadSkills,
   skill
-} from "./chunk-ZT2YH2GA.js";
+} from "./chunk-ITYXP4CL.js";
 import {
   AgentBuilder
-} from "./chunk-SYMAVW5T.js";
+} from "./chunk-YQHXHTYN.js";
 import {
   MaxTurnsError,
   PromptCancelledError,
+  ToolApprovalRequiredError,
   cancelPrompt,
   createHook,
   requestToolApproval,
   runControl,
   skipTool,
   toolCallControl
-} from "./chunk-GBBMMU5E.js";
+} from "./chunk-LDB4V2NU.js";
 import "./chunk-XUUY2L2D.js";
 import "./chunk-XXT2UCAR.js";
 import "./chunk-YK4WAAS4.js";
@@ -23,11 +24,13 @@ import {
   createMiddleware,
   createThinkTool,
   createToolMiddleware
-} from "./chunk-5OGAUWM5.js";
+} from "./chunk-USWULPZJ.js";
 import {
   createTool
-} from "./chunk-66IJDP6L.js";
+} from "./chunk-VQE5PAAT.js";
 import "./chunk-MMHG7WAM.js";
+import "./chunk-FTO3W4UP.js";
+import "./chunk-OIMLU4SF.js";
 import {
   createCompletion,
   createCompletionStream,
@@ -41,8 +44,6 @@ import {
   UserContent
 } from "./chunk-WJO6NVE2.js";
 import "./chunk-WQKHFADH.js";
-import "./chunk-FTO3W4UP.js";
-import "./chunk-OIMLU4SF.js";
 export {
   AgentBuilder,
   AssistantContent,
@@ -50,6 +51,7 @@ export {
   Message,
   PromptCancelledError,
   SkillValidationError,
+  ToolApprovalRequiredError,
   ToolContent,
   Usage,
   UserContent,

package/dist/internal/agent.d.ts

@@ -1,8 +1,8 @@
-export { e as Agent, j as AgentEventAppendInput, k as AgentEventRecord, h as AgentEventStore, l as AgentEventStoreInclude, i as AgentEventStoreOptions, I as AgentEventStoreRegistration, J as AgentOptions, K as AgentSession, L as AgentToolOptions, M as DEFAULT_MAX_TURNS, D as DynamicContextOptions, N as DynamicContextRegistration, f as DynamicToolOptions, O as DynamicToolRegistration } from '../agent-CFMdNB6O.js';
+export { e as Agent, j as AgentEventAppendInput, k as AgentEventRecord, h as AgentEventStore, l as AgentEventStoreInclude, i as AgentEventStoreOptions, I as AgentEventStoreRegistration, J as AgentOptions, K as AgentSession, L as AgentToolOptions, M as DEFAULT_MAX_TURNS, D as DynamicContextOptions, N as DynamicContextRegistration, f as DynamicToolOptions, O as DynamicToolRegistration } from '../agent-BW19m79B.js';
 import '../types-C8loR4il.js';
 import '../memory/index.js';
 import '../observability/index.js';
-import '../tool-Bwj_zoTx.js';
-import '../middleware-C0eXGwoh.js';
+import '../tool-BLypI31e.js';
+import '../middleware-CZocYAtZ.js';
 import '../types-IB2e9u5M.js';
 import '../vector-store/index.js';

package/dist/internal/agent.js

@@ -2,17 +2,17 @@ import {
   Agent,
   AgentSession,
   DEFAULT_MAX_TURNS
-} from "../chunk-GBBMMU5E.js";
+} from "../chunk-LDB4V2NU.js";
 import "../chunk-XUUY2L2D.js";
 import "../chunk-YK4WAAS4.js";
-import "../chunk-5OGAUWM5.js";
-import "../chunk-66IJDP6L.js";
+import "../chunk-USWULPZJ.js";
+import "../chunk-VQE5PAAT.js";
 import "../chunk-MMHG7WAM.js";
+import "../chunk-FTO3W4UP.js";
+import "../chunk-OIMLU4SF.js";
 import "../chunk-S5IGJ4RB.js";
 import "../chunk-WJO6NVE2.js";
 import "../chunk-WQKHFADH.js";
-import "../chunk-FTO3W4UP.js";
-import "../chunk-OIMLU4SF.js";
 export {
   Agent,
   AgentSession,

package/dist/mcp/index.d.ts

@@ -1,10 +1,10 @@
-import { M as McpConnection, a as McpServer, b as McpStdioOptions, c as McpHttpOptions, d as McpSseOptions } from '../types-BLyF-Tub.js';
-export { e as McpClient, f as McpToolCallContent, g as McpToolCallResult, h as McpToolDefinition } from '../types-BLyF-Tub.js';
+import { M as McpConnection, a as McpServer, b as McpStdioOptions, c as McpHttpOptions, d as McpSseOptions } from '../types-CIwm9p4w.js';
+export { e as McpClient, f as McpToolCallContent, g as McpToolCallResult, h as McpToolDefinition } from '../types-CIwm9p4w.js';
 import '@modelcontextprotocol/sdk/client/sse.js';
 import '@modelcontextprotocol/sdk/client/stdio.js';
 import '@modelcontextprotocol/sdk/client/streamableHttp.js';
 import '../types-C8loR4il.js';
-import '../tool-Bwj_zoTx.js';
+import '../tool-BLypI31e.js';
 
 declare function connectMcp(connection: McpConnection): Promise<McpServer>;
 

package/dist/{middleware-C0eXGwoh.d.ts → middleware-CZocYAtZ.d.ts}

@@ -1,7 +1,7 @@
 import { h as ToolDefinition, b as CompletionRequest, c as CompletionResponse, e as JsonValue, k as ToolResultContent } from './types-C8loR4il.js';
 import { V as VectorMetadata, E as EmbeddingModel, b as EmbeddedDocument } from './types-IB2e9u5M.js';
 import { VectorSearchIndex } from './vector-store/index.js';
-import { A as AnyTool, c as ToolCallContext, N as NormalizedToolOutput } from './tool-Bwj_zoTx.js';
+import { A as AnyTool, f as ToolCallContext, N as NormalizedToolOutput } from './tool-BLypI31e.js';
 
 declare class ToolSet {
     private readonly tools;

package/dist/observability/index.d.ts

@@ -1,5 +1,5 @@
 import { M as Message, b as CompletionRequest, J as JsonObject, t as CompletionModelCapabilities, c as CompletionResponse, f as ToolCall, h as ToolDefinition, k as ToolResultContent, U as Usage } from '../types-C8loR4il.js';
-import { d as ToolCallStreamEvent } from '../tool-Bwj_zoTx.js';
+import { g as ToolCallStreamEvent } from '../tool-BLypI31e.js';
 
 type AgentTraceInfo = {
     traceId?: string | undefined;

package/dist/pipeline/index.d.ts

@@ -1,14 +1,14 @@
-import { e as Agent } from '../agent-CFMdNB6O.js';
+import { z } from 'zod';
+import { e as Agent } from '../agent-BW19m79B.js';
 import { J as JsonObject, C as CompletionModel } from '../types-C8loR4il.js';
 import { Extractor } from '../extractor/index.js';
 import '../memory/index.js';
 import '../observability/index.js';
-import '../tool-Bwj_zoTx.js';
-import '../middleware-C0eXGwoh.js';
+import '../tool-BLypI31e.js';
+import '../middleware-CZocYAtZ.js';
 import '../types-IB2e9u5M.js';
 import '../vector-store/index.js';
 import '../zod-schema-C7F4clpm.js';
-import 'zod';
 
 /** Minimal interface for anything that can run as a pipeline stage. */
 interface PipelineOp<Input = unknown, Output = unknown> {
@@ -108,8 +108,8 @@ declare class Pipeline<Input, Output> implements PipelineOp<Input, Awaited<Outpu
 declare class PipelineBuilder<Input, Output = Input> {
     private readonly executor;
     private readonly state;
-    constructor();
-    constructor(metadata: PipelineMetadata);
+    constructor(schema: z.ZodType<Output, Input>);
+    constructor(schema: z.ZodType<Output, Input>, metadata: PipelineMetadata);
     constructor(executor: (input: Input) => Output | Promise<Output>);
     constructor(executor: PipelineExecutor<Input, Output>, state: PipelineBuilderState);
     /** Add a synchronous or asynchronous transform stage. */

package/dist/pipeline/index.js

@@ -5,6 +5,9 @@ import {
   mapWithConcurrency
 } from "../chunk-OIMLU4SF.js";
 
+// src/pipeline/builder.ts
+import { z } from "zod";
+
 // src/pipeline/graph.ts
 function initialBuilderState(metadata) {
   return {
@@ -189,10 +192,10 @@ async function runNode(context, node, fn) {
 var PipelineBuilder = class _PipelineBuilder {
   executor;
   state;
-  constructor(metadataOrExecutor, state) {
-    if (state !== void 0) {
+  constructor(metadataOrExecutor, stateOrMetadata) {
+    if (stateOrMetadata !== void 0 && isBuilderState(stateOrMetadata)) {
       this.executor = metadataOrExecutor;
-      this.state = state;
+      this.state = stateOrMetadata;
       return;
     }
     if (typeof metadataOrExecutor === "function") {
@@ -201,6 +204,13 @@ var PipelineBuilder = class _PipelineBuilder {
       this.state = initialBuilderState({});
       return;
     }
+    if (isZodSchema(metadataOrExecutor)) {
+      const schema = metadataOrExecutor;
+      const metadata = stateOrMetadata !== void 0 && isPipelineMetadata(stateOrMetadata) ? stateOrMetadata : {};
+      this.executor = ((input) => schema.parse(input));
+      this.state = initialBuilderState(metadata);
+      return;
+    }
     this.executor = identity;
     this.state = initialBuilderState(metadataOrExecutor ?? {});
   }
@@ -338,6 +348,23 @@ var PipelineBuilder = class _PipelineBuilder {
 function identity(input) {
   return input;
 }
+function isZodSchema(value) {
+  return value instanceof z.ZodType;
+}
+function isBuilderState(value) {
+  if (typeof value !== "object" || value === null || Array.isArray(value)) {
+    return false;
+  }
+  const candidate = value;
+  return "graph" in candidate && "terminalNodeId" in candidate && "terminalNodeIds" in candidate && "nextNodeIndex" in candidate && "nextEdgeIndex" in candidate;
+}
+function isPipelineMetadata(value) {
+  if (typeof value !== "object" || value === null || Array.isArray(value)) {
+    return false;
+  }
+  const candidate = value;
+  return "id" in candidate || "name" in candidate || "description" in candidate || "metadata" in candidate;
+}
 export {
   Pipeline,
   PipelineBuilder

package/dist/pipeline/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/pipeline/graph.ts","../../src/pipeline/pipeline.ts","../../src/pipeline/runtime.ts","../../src/pipeline/builder.ts"],"sourcesContent":["import type { JsonObject } from \"../completion\";\nimport { compact } from \"../internal/compact\";\nimport type {\n  PipelineBuilderState,\n  PipelineGraph,\n  PipelineGraphNode,\n  PipelineMetadata,\n  PipelineStageKind,\n} from \"./types\";\n\nexport function initialBuilderState(metadata: PipelineMetadata): PipelineBuilderState {\n  return {\n    graph: initialGraph(metadata),\n    terminalNodeId: \"input\",\n    terminalNodeIds: [\"input\"],\n    nextNodeIndex: 1,\n    nextEdgeIndex: 1,\n  };\n}\n\nexport function initialGraph(metadata: PipelineMetadata): PipelineGraph {\n  const id = normalizeId(metadata.id ?? \"pipeline\");\n  return {\n    ...compact({\n      id,\n      name: metadata.name,\n      description: metadata.description,\n      metadata: metadata.metadata,\n    }),\n    nodes: [{ id: \"input\", kind: \"input\", label: \"Input\" }],\n    edges: [],\n  };\n}\n\nexport function appendNode(\n  state: PipelineBuilderState,\n  kind: PipelineStageKind,\n  label: string,\n  options: {\n    description?: string | undefined;\n    metadata?: JsonObject | undefined;\n    preferredId?: string | undefined;\n    agentId?: string | undefined;\n    agentName?: string | undefined;\n    pipelineId?: string | undefined;\n  } = {},\n): { state: PipelineBuilderState; node: PipelineGraphNode } {\n  const node = graphNode(kind, label, state.nextNodeIndex, {\n    ...options,\n    existingIds: new Set(state.graph.nodes.map((item) => item.id)),\n  });\n  return {\n    node,\n    state: appendGraphNode(state, node, activeTerminalNodeIds(state), [node.id]),\n  };\n}\n\nexport function appendChildNode(\n  state: PipelineBuilderState,\n  parentId: string,\n  kind: PipelineStageKind,\n  label: string,\n  options: {\n    branchKey?: string | undefined;\n  } = {},\n): { state: PipelineBuilderState; node: PipelineGraphNode } {\n  const node = graphNode(kind, label, state.nextNodeIndex, {\n    ...options,\n    existingIds: new Set(state.graph.nodes.map((item) => item.id)),\n  });\n  return {\n    node,\n    state: appendGraphNode(state, node, [parentId], activeTerminalNodeIds(state)),\n  };\n}\n\nexport function activeTerminalNodeIds(state: PipelineBuilderState): string[] {\n  return state.terminalNodeIds.length > 0 ? state.terminalNodeIds : [state.terminalNodeId];\n}\n\nexport function withTerminalNodes(\n  state: PipelineBuilderState,\n  terminalNodeIds: string[],\n): PipelineBuilderState {\n  return {\n    ...state,\n    terminalNodeId: terminalNodeIds.at(-1) ?? state.terminalNodeId,\n    terminalNodeIds,\n  };\n}\n\nexport function withOutputNode(state: PipelineBuilderState): PipelineGraph {\n  const graph = cloneGraph(state.graph);\n  if (graph.nodes.some((node) => node.id === \"output\")) {\n    return graph;\n  }\n  graph.nodes.push({ id: \"output\", kind: \"output\", label: \"Output\" });\n  graph.edges.push(\n    ...activeTerminalNodeIds(state).map((sourceId, index) => ({\n      id: `edge_${state.nextEdgeIndex + index}`,\n      source: sourceId,\n      target: \"output\",\n    })),\n  );\n  return graph;\n}\n\nexport function nextStageLabel(state: PipelineBuilderState, prefix: string): string {\n  return `${prefix} ${state.nextNodeIndex}`;\n}\n\nexport function cloneGraph(graph: PipelineGraph): PipelineGraph {\n  return {\n    ...graph,\n    nodes: graph.nodes.map((node) => ({ ...node })),\n    edges: graph.edges.map((edge) => ({ ...edge })),\n  };\n}\n\nfunction appendGraphNode(\n  state: PipelineBuilderState,\n  node: PipelineGraphNode,\n  sourceIds: string[],\n  terminalNodeIds: string[],\n): PipelineBuilderState {\n  const edges = sourceIds.map((sourceId, index) => ({\n    id: `edge_${state.nextEdgeIndex + index}`,\n    source: sourceId,\n    target: node.id,\n  }));\n  const terminalNodeId = terminalNodeIds.at(-1) ?? state.terminalNodeId;\n  return {\n    graph: {\n      ...state.graph,\n      nodes: [...state.graph.nodes, node],\n      edges: [...state.graph.edges, ...edges],\n    },\n    terminalNodeId,\n    terminalNodeIds,\n    nextNodeIndex: state.nextNodeIndex + 1,\n    nextEdgeIndex: state.nextEdgeIndex + edges.length,\n  };\n}\n\nfunction graphNode(\n  kind: PipelineStageKind,\n  label: string,\n  index: number,\n  options: {\n    description?: string | undefined;\n    metadata?: JsonObject | undefined;\n    preferredId?: string | undefined;\n    agentId?: string | undefined;\n    agentName?: string | undefined;\n    pipelineId?: string | undefined;\n    branchKey?: string | undefined;\n    existingIds?: Set<string> | undefined;\n  } = {},\n): PipelineGraphNode {\n  const id = uniqueGraphNodeId(\n    normalizeId(options.preferredId ?? `${kind}_${index}`),\n    options.existingIds ?? new Set(),\n  );\n  return compact({\n    id,\n    kind,\n    label,\n    description: options.description,\n    metadata: options.metadata,\n    agentId: options.agentId,\n    agentName: options.agentName,\n    pipelineId: options.pipelineId,\n    branchKey: options.branchKey,\n  }) as PipelineGraphNode;\n}\n\nfunction normalizeId(value: string): string {\n  const normalized = value\n    .trim()\n    .toLowerCase()\n    .replace(/[^a-z0-9_-]+/g, \"_\")\n    .replace(/^_+|_+$/g, \"\");\n  return normalized.length === 0 ? \"pipeline\" : normalized;\n}\n\nfunction uniqueGraphNodeId(baseId: string, existingIds: Set<string>): string {\n  let id = baseId;\n  let suffix = 2;\n  while (existingIds.has(id)) {\n    id = `${baseId}_${suffix}`;\n    suffix += 1;\n  }\n  return id;\n}\n","import type { JsonObject } from \"../completion\";\nimport { mapWithConcurrency } from \"../internal/concurrency\";\nimport { cloneGraph, initialGraph } from \"./graph\";\nimport type {\n  PipelineBatchOptions,\n  PipelineExecutor,\n  PipelineGraph,\n  PipelineOp,\n  PipelineRunOptions,\n} from \"./types\";\n\n/** Runnable pipeline returned by `PipelineBuilder.build()`. */\nexport class Pipeline<Input, Output> implements PipelineOp<Input, Awaited<Output>> {\n  readonly id: string;\n  readonly name: string | undefined;\n  readonly description: string | undefined;\n  readonly metadata: JsonObject | undefined;\n\n  constructor(\n    private readonly executor: PipelineExecutor<Input, Output>,\n    private readonly pipelineGraph: PipelineGraph = initialGraph({}),\n  ) {\n    this.id = pipelineGraph.id;\n    this.name = pipelineGraph.name;\n    this.description = pipelineGraph.description;\n    this.metadata = pipelineGraph.metadata;\n  }\n\n  /** Run one input through the built pipeline and return the final stage output. */\n  async run(input: Input, options: PipelineRunOptions = {}): Promise<Awaited<Output>> {\n    return (await this.executor(input, { observer: options.observer })) as Awaited<Output>;\n  }\n\n  /** Run many inputs through the same pipeline with bounded concurrency. */\n  async batch<I extends Iterable<Input>>(\n    inputs: I,\n    options: PipelineBatchOptions,\n  ): Promise<Array<Awaited<Output>>> {\n    return mapWithConcurrency([...inputs], options.concurrency, (input) => this.run(input));\n  }\n\n  graph(): PipelineGraph {\n    return cloneGraph(this.pipelineGraph);\n  }\n}\n","import type { PipelineGraphNode, PipelineRunContext } from \"./types\";\n\nexport async function runNode<Output>(\n  context: PipelineRunContext,\n  node: PipelineGraphNode,\n  fn: () => Output | Promise<Output>,\n): Promise<Awaited<Output>> {\n  const startedAt = Date.now();\n  await context.observer?.onEvent({ type: \"stage_started\", node });\n  try {\n    const output = (await fn()) as Awaited<Output>;\n    await context.observer?.onEvent({\n      type: \"stage_completed\",\n      node,\n      durationMs: Date.now() - startedAt,\n    });\n    return output;\n  } catch (error) {\n    await context.observer?.onEvent({\n      type: \"stage_failed\",\n      node,\n      durationMs: Date.now() - startedAt,\n      error,\n    });\n    throw error;\n  }\n}\n","import type { Agent } from \"../agent/agent\";\nimport type { CompletionModel } from \"../completion\";\nimport type { Extractor } from \"../extractor\";\nimport {\n  appendChildNode,\n  appendNode,\n  initialBuilderState,\n  nextStageLabel,\n  withOutputNode,\n  withTerminalNodes,\n} from \"./graph\";\nimport { Pipeline } from \"./pipeline\";\nimport { runNode } from \"./runtime\";\nimport type {\n  ParallelOutput,\n  PipelineBuilderState,\n  PipelineExecutor,\n  PipelineGraphNode,\n  PipelineMetadata,\n  PipelineOp,\n  PipelineRunContext,\n  PipelineStageMetadata,\n} from \"./types\";\n\n/** Builds a typed pipeline from an original input type to an inferred output type. */\nexport class PipelineBuilder<Input, Output = Input> {\n  private readonly executor: PipelineExecutor<Input, Output>;\n  private readonly state: PipelineBuilderState;\n\n  constructor();\n  constructor(metadata: PipelineMetadata);\n  constructor(executor: (input: Input) => Output | Promise<Output>);\n  constructor(executor: PipelineExecutor<Input, Output>, state: PipelineBuilderState);\n  constructor(\n    metadataOrExecutor?:\n      | PipelineMetadata\n      | ((input: Input) => Output | Promise<Output>)\n      | PipelineExecutor<Input, Output>,\n    state?: PipelineBuilderState,\n  ) {\n    if (state !== undefined) {\n      this.executor = metadataOrExecutor as PipelineExecutor<Input, Output>;\n      this.state = state;\n      return;\n    }\n\n    if (typeof metadataOrExecutor === \"function\") {\n      const executor = metadataOrExecutor as (input: Input) => Output | Promise<Output>;\n      this.executor = ((input) => executor(input)) as PipelineExecutor<Input, Output>;\n      this.state = initialBuilderState({});\n      return;\n    }\n\n    this.executor = identity as PipelineExecutor<Input, Output>;\n    this.state = initialBuilderState(metadataOrExecutor ?? {});\n  }\n\n  /** Add a synchronous or asynchronous transform stage. */\n  step<Next>(\n    fn: (input: Awaited<Output>) => Next | Promise<Next>,\n    metadata?: PipelineStageMetadata,\n  ): PipelineBuilder<Input, Awaited<Next>> {\n    const next = appendNode(\n      this.state,\n      \"step\",\n      metadata?.name ?? nextStageLabel(this.state, \"Step\"),\n      {\n        description: metadata?.description,\n        metadata: metadata?.metadata,\n        preferredId: metadata?.id,\n      },\n    );\n    return new PipelineBuilder<Input, Awaited<Next>>(\n      async (input, context): Promise<Awaited<Next>> => {\n        const value = await this.runStep(input, context);\n        const result = await runNode(context, next.node, () => fn(value));\n        return result as Awaited<Next>;\n      },\n      next.state,\n    );\n  }\n\n  /** Compose another pipeline operation after the current stage. */\n  use<Next>(\n    op: PipelineOp<Awaited<Output>, Next>,\n    metadata?: PipelineStageMetadata,\n  ): PipelineBuilder<Input, Awaited<Next>> {\n    const nested = op instanceof Pipeline ? op : undefined;\n    const next = appendNode(\n      this.state,\n      nested === undefined ? \"step\" : \"pipeline\",\n      metadata?.name ??\n        nested?.name ??\n        nested?.id ??\n        nextStageLabel(this.state, nested === undefined ? \"Operation\" : \"Pipeline\"),\n      {\n        description: metadata?.description ?? nested?.description,\n        metadata: metadata?.metadata ?? nested?.metadata,\n        preferredId: metadata?.id,\n        pipelineId: nested?.id,\n      },\n    );\n    return new PipelineBuilder<Input, Awaited<Next>>(\n      async (input, context): Promise<Awaited<Next>> => {\n        const value = await this.runStep(input, context);\n        const result = await runNode(context, next.node, () => op.run(value));\n        return result as Awaited<Next>;\n      },\n      next.state,\n    );\n  }\n\n  /** Run named branch operations concurrently from the current value. */\n  parallel<Branches extends Record<string, PipelineOp<Awaited<Output>, unknown>>>(\n    branches: Branches,\n    metadata?: PipelineStageMetadata,\n  ): PipelineBuilder<Input, ParallelOutput<Branches>> {\n    const parallel = appendNode(\n      this.state,\n      \"parallel\",\n      metadata?.name ?? `${Object.keys(branches).length} parallel branches`,\n      {\n        description: metadata?.description,\n        metadata: metadata?.metadata,\n        preferredId: metadata?.id,\n      },\n    );\n    let nextState = parallel.state;\n    const branchNodes: Record<string, PipelineGraphNode> = {};\n    for (const key of Object.keys(branches)) {\n      const branch = appendChildNode(nextState, parallel.node.id, \"branch\", key, {\n        branchKey: key,\n      });\n      nextState = branch.state;\n      branchNodes[key] = branch.node;\n    }\n    nextState = withTerminalNodes(\n      nextState,\n      Object.values(branchNodes).map((node) => node.id),\n    );\n\n    return new PipelineBuilder<Input, ParallelOutput<Branches>>(async (input, context) => {\n      const value = await this.runStep(input, context);\n      const entries = await runNode(context, parallel.node, () =>\n        Promise.all(\n          Object.entries(branches).map(async ([key, op]) => {\n            const node = branchNodes[key] as PipelineGraphNode;\n            const output = await runNode(context, node, () => op.run(value));\n            return [key, output] as const;\n          }),\n        ),\n      );\n      return Object.fromEntries(entries) as ParallelOutput<Branches>;\n    }, nextState);\n  }\n\n  /** Send the current value to an agent as text and continue with the agent output. */\n  prompt(\n    agent: Agent<CompletionModel>,\n    metadata?: PipelineStageMetadata,\n  ): PipelineBuilder<Input, string> {\n    const next = appendNode(this.state, \"agent\", metadata?.name ?? agent.name ?? agent.id, {\n      description: metadata?.description ?? agent.description,\n      metadata: metadata?.metadata,\n      preferredId: metadata?.id,\n      agentId: agent.id,\n      agentName: agent.name,\n    });\n    return new PipelineBuilder<Input, string>(async (input, context) => {\n      const value = await this.runStep(input, context);\n      return runNode(context, next.node, async () => {\n        const response = await agent.prompt(String(value)).send();\n        return response.output;\n      });\n    }, next.state);\n  }\n\n  /** Send the current value to an extractor as text and continue with typed schema data. */\n  extract<T>(\n    extractor: Extractor<T, CompletionModel>,\n    metadata?: PipelineStageMetadata,\n  ): PipelineBuilder<Input, T> {\n    const next = appendNode(\n      this.state,\n      \"extractor\",\n      metadata?.name ?? nextStageLabel(this.state, \"Extractor\"),\n      {\n        description: metadata?.description,\n        metadata: metadata?.metadata,\n        preferredId: metadata?.id,\n      },\n    );\n    return new PipelineBuilder<Input, T>(async (input, context) => {\n      const value = await this.runStep(input, context);\n      return runNode(context, next.node, () => extractor.extract(String(value)));\n    }, next.state);\n  }\n\n  /** Finish the builder and return a runnable pipeline. */\n  build(): Pipeline<Input, Awaited<Output>> {\n    const graph = withOutputNode(this.state);\n    return new Pipeline<Input, Awaited<Output>>(\n      (input, context) => this.runStep(input, context) as Promise<Awaited<Output>>,\n      graph,\n    );\n  }\n\n  private async runStep(input: Input, context: PipelineRunContext): Promise<Awaited<Output>> {\n    return (await this.executor(input, context)) as Awaited<Output>;\n  }\n}\n\nfunction identity<T>(input: T): T {\n  return input;\n}\n"],"mappings":";;;;;;;;AAUO,SAAS,oBAAoB,UAAkD;AACpF,SAAO;AAAA,IACL,OAAO,aAAa,QAAQ;AAAA,IAC5B,gBAAgB;AAAA,IAChB,iBAAiB,CAAC,OAAO;AAAA,IACzB,eAAe;AAAA,IACf,eAAe;AAAA,EACjB;AACF;AAEO,SAAS,aAAa,UAA2C;AACtE,QAAM,KAAK,YAAY,SAAS,MAAM,UAAU;AAChD,SAAO;AAAA,IACL,GAAG,QAAQ;AAAA,MACT;AAAA,MACA,MAAM,SAAS;AAAA,MACf,aAAa,SAAS;AAAA,MACtB,UAAU,SAAS;AAAA,IACrB,CAAC;AAAA,IACD,OAAO,CAAC,EAAE,IAAI,SAAS,MAAM,SAAS,OAAO,QAAQ,CAAC;AAAA,IACtD,OAAO,CAAC;AAAA,EACV;AACF;AAEO,SAAS,WACd,OACA,MACA,OACA,UAOI,CAAC,GACqD;AAC1D,QAAM,OAAO,UAAU,MAAM,OAAO,MAAM,eAAe;AAAA,IACvD,GAAG;AAAA,IACH,aAAa,IAAI,IAAI,MAAM,MAAM,MAAM,IAAI,CAAC,SAAS,KAAK,EAAE,CAAC;AAAA,EAC/D,CAAC;AACD,SAAO;AAAA,IACL;AAAA,IACA,OAAO,gBAAgB,OAAO,MAAM,sBAAsB,KAAK,GAAG,CAAC,KAAK,EAAE,CAAC;AAAA,EAC7E;AACF;AAEO,SAAS,gBACd,OACA,UACA,MACA,OACA,UAEI,CAAC,GACqD;AAC1D,QAAM,OAAO,UAAU,MAAM,OAAO,MAAM,eAAe;AAAA,IACvD,GAAG;AAAA,IACH,aAAa,IAAI,IAAI,MAAM,MAAM,MAAM,IAAI,CAAC,SAAS,KAAK,EAAE,CAAC;AAAA,EAC/D,CAAC;AACD,SAAO;AAAA,IACL;AAAA,IACA,OAAO,gBAAgB,OAAO,MAAM,CAAC,QAAQ,GAAG,sBAAsB,KAAK,CAAC;AAAA,EAC9E;AACF;AAEO,SAAS,sBAAsB,OAAuC;AAC3E,SAAO,MAAM,gBAAgB,SAAS,IAAI,MAAM,kBAAkB,CAAC,MAAM,cAAc;AACzF;AAEO,SAAS,kBACd,OACA,iBACsB;AACtB,SAAO;AAAA,IACL,GAAG;AAAA,IACH,gBAAgB,gBAAgB,GAAG,EAAE,KAAK,MAAM;AAAA,IAChD;AAAA,EACF;AACF;AAEO,SAAS,eAAe,OAA4C;AACzE,QAAM,QAAQ,WAAW,MAAM,KAAK;AACpC,MAAI,MAAM,MAAM,KAAK,CAAC,SAAS,KAAK,OAAO,QAAQ,GAAG;AACpD,WAAO;AAAA,EACT;AACA,QAAM,MAAM,KAAK,EAAE,IAAI,UAAU,MAAM,UAAU,OAAO,SAAS,CAAC;AAClE,QAAM,MAAM;AAAA,IACV,GAAG,sBAAsB,KAAK,EAAE,IAAI,CAAC,UAAU,WAAW;AAAA,MACxD,IAAI,QAAQ,MAAM,gBAAgB,KAAK;AAAA,MACvC,QAAQ;AAAA,MACR,QAAQ;AAAA,IACV,EAAE;AAAA,EACJ;AACA,SAAO;AACT;AAEO,SAAS,eAAe,OAA6B,QAAwB;AAClF,SAAO,GAAG,MAAM,IAAI,MAAM,aAAa;AACzC;AAEO,SAAS,WAAW,OAAqC;AAC9D,SAAO;AAAA,IACL,GAAG;AAAA,IACH,OAAO,MAAM,MAAM,IAAI,CAAC,UAAU,EAAE,GAAG,KAAK,EAAE;AAAA,IAC9C,OAAO,MAAM,MAAM,IAAI,CAAC,UAAU,EAAE,GAAG,KAAK,EAAE;AAAA,EAChD;AACF;AAEA,SAAS,gBACP,OACA,MACA,WACA,iBACsB;AACtB,QAAM,QAAQ,UAAU,IAAI,CAAC,UAAU,WAAW;AAAA,IAChD,IAAI,QAAQ,MAAM,gBAAgB,KAAK;AAAA,IACvC,QAAQ;AAAA,IACR,QAAQ,KAAK;AAAA,EACf,EAAE;AACF,QAAM,iBAAiB,gBAAgB,GAAG,EAAE,KAAK,MAAM;AACvD,SAAO;AAAA,IACL,OAAO;AAAA,MACL,GAAG,MAAM;AAAA,MACT,OAAO,CAAC,GAAG,MAAM,MAAM,OAAO,IAAI;AAAA,MAClC,OAAO,CAAC,GAAG,MAAM,MAAM,OAAO,GAAG,KAAK;AAAA,IACxC;AAAA,IACA;AAAA,IACA;AAAA,IACA,eAAe,MAAM,gBAAgB;AAAA,IACrC,eAAe,MAAM,gBAAgB,MAAM;AAAA,EAC7C;AACF;AAEA,SAAS,UACP,MACA,OACA,OACA,UASI,CAAC,GACc;AACnB,QAAM,KAAK;AAAA,IACT,YAAY,QAAQ,eAAe,GAAG,IAAI,IAAI,KAAK,EAAE;AAAA,IACrD,QAAQ,eAAe,oBAAI,IAAI;AAAA,EACjC;AACA,SAAO,QAAQ;AAAA,IACb;AAAA,IACA;AAAA,IACA;AAAA,IACA,aAAa,QAAQ;AAAA,IACrB,UAAU,QAAQ;AAAA,IAClB,SAAS,QAAQ;AAAA,IACjB,WAAW,QAAQ;AAAA,IACnB,YAAY,QAAQ;AAAA,IACpB,WAAW,QAAQ;AAAA,EACrB,CAAC;AACH;AAEA,SAAS,YAAY,OAAuB;AAC1C,QAAM,aAAa,MAChB,KAAK,EACL,YAAY,EACZ,QAAQ,iBAAiB,GAAG,EAC5B,QAAQ,YAAY,EAAE;AACzB,SAAO,WAAW,WAAW,IAAI,aAAa;AAChD;AAEA,SAAS,kBAAkB,QAAgB,aAAkC;AAC3E,MAAI,KAAK;AACT,MAAI,SAAS;AACb,SAAO,YAAY,IAAI,EAAE,GAAG;AAC1B,SAAK,GAAG,MAAM,IAAI,MAAM;AACxB,cAAU;AAAA,EACZ;AACA,SAAO;AACT;;;ACrLO,IAAM,WAAN,MAA4E;AAAA,EAMjF,YACmB,UACA,gBAA+B,aAAa,CAAC,CAAC,GAC/D;AAFiB;AACA;AAEjB,SAAK,KAAK,cAAc;AACxB,SAAK,OAAO,cAAc;AAC1B,SAAK,cAAc,cAAc;AACjC,SAAK,WAAW,cAAc;AAAA,EAChC;AAAA,EAPmB;AAAA,EACA;AAAA,EAPV;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EAaT,MAAM,IAAI,OAAc,UAA8B,CAAC,GAA6B;AAClF,WAAQ,MAAM,KAAK,SAAS,OAAO,EAAE,UAAU,QAAQ,SAAS,CAAC;AAAA,EACnE;AAAA;AAAA,EAGA,MAAM,MACJ,QACA,SACiC;AACjC,WAAO,mBAAmB,CAAC,GAAG,MAAM,GAAG,QAAQ,aAAa,CAAC,UAAU,KAAK,IAAI,KAAK,CAAC;AAAA,EACxF;AAAA,EAEA,QAAuB;AACrB,WAAO,WAAW,KAAK,aAAa;AAAA,EACtC;AACF;;;AC1CA,eAAsB,QACpB,SACA,MACA,IAC0B;AAC1B,QAAM,YAAY,KAAK,IAAI;AAC3B,QAAM,QAAQ,UAAU,QAAQ,EAAE,MAAM,iBAAiB,KAAK,CAAC;AAC/D,MAAI;AACF,UAAM,SAAU,MAAM,GAAG;AACzB,UAAM,QAAQ,UAAU,QAAQ;AAAA,MAC9B,MAAM;AAAA,MACN;AAAA,MACA,YAAY,KAAK,IAAI,IAAI;AAAA,IAC3B,CAAC;AACD,WAAO;AAAA,EACT,SAAS,OAAO;AACd,UAAM,QAAQ,UAAU,QAAQ;AAAA,MAC9B,MAAM;AAAA,MACN;AAAA,MACA,YAAY,KAAK,IAAI,IAAI;AAAA,MACzB;AAAA,IACF,CAAC;AACD,UAAM;AAAA,EACR;AACF;;;ACDO,IAAM,kBAAN,MAAM,iBAAuC;AAAA,EACjC;AAAA,EACA;AAAA,EAMjB,YACE,oBAIA,OACA;AACA,QAAI,UAAU,QAAW;AACvB,WAAK,WAAW;AAChB,WAAK,QAAQ;AACb;AAAA,IACF;AAEA,QAAI,OAAO,uBAAuB,YAAY;AAC5C,YAAM,WAAW;AACjB,WAAK,YAAY,CAAC,UAAU,SAAS,KAAK;AAC1C,WAAK,QAAQ,oBAAoB,CAAC,CAAC;AACnC;AAAA,IACF;AAEA,SAAK,WAAW;AAChB,SAAK,QAAQ,oBAAoB,sBAAsB,CAAC,CAAC;AAAA,EAC3D;AAAA;AAAA,EAGA,KACE,IACA,UACuC;AACvC,UAAM,OAAO;AAAA,MACX,KAAK;AAAA,MACL;AAAA,MACA,UAAU,QAAQ,eAAe,KAAK,OAAO,MAAM;AAAA,MACnD;AAAA,QACE,aAAa,UAAU;AAAA,QACvB,UAAU,UAAU;AAAA,QACpB,aAAa,UAAU;AAAA,MACzB;AAAA,IACF;AACA,WAAO,IAAI;AAAA,MACT,OAAO,OAAO,YAAoC;AAChD,cAAM,QAAQ,MAAM,KAAK,QAAQ,OAAO,OAAO;AAC/C,cAAM,SAAS,MAAM,QAAQ,SAAS,KAAK,MAAM,MAAM,GAAG,KAAK,CAAC;AAChE,eAAO;AAAA,MACT;AAAA,MACA,KAAK;AAAA,IACP;AAAA,EACF;AAAA;AAAA,EAGA,IACE,IACA,UACuC;AACvC,UAAM,SAAS,cAAc,WAAW,KAAK;AAC7C,UAAM,OAAO;AAAA,MACX,KAAK;AAAA,MACL,WAAW,SAAY,SAAS;AAAA,MAChC,UAAU,QACR,QAAQ,QACR,QAAQ,MACR,eAAe,KAAK,OAAO,WAAW,SAAY,cAAc,UAAU;AAAA,MAC5E;AAAA,QACE,aAAa,UAAU,eAAe,QAAQ;AAAA,QAC9C,UAAU,UAAU,YAAY,QAAQ;AAAA,QACxC,aAAa,UAAU;AAAA,QACvB,YAAY,QAAQ;AAAA,MACtB;AAAA,IACF;AACA,WAAO,IAAI;AAAA,MACT,OAAO,OAAO,YAAoC;AAChD,cAAM,QAAQ,MAAM,KAAK,QAAQ,OAAO,OAAO;AAC/C,cAAM,SAAS,MAAM,QAAQ,SAAS,KAAK,MAAM,MAAM,GAAG,IAAI,KAAK,CAAC;AACpE,eAAO;AAAA,MACT;AAAA,MACA,KAAK;AAAA,IACP;AAAA,EACF;AAAA;AAAA,EAGA,SACE,UACA,UACkD;AAClD,UAAM,WAAW;AAAA,MACf,KAAK;AAAA,MACL;AAAA,MACA,UAAU,QAAQ,GAAG,OAAO,KAAK,QAAQ,EAAE,MAAM;AAAA,MACjD;AAAA,QACE,aAAa,UAAU;AAAA,QACvB,UAAU,UAAU;AAAA,QACpB,aAAa,UAAU;AAAA,MACzB;AAAA,IACF;AACA,QAAI,YAAY,SAAS;AACzB,UAAM,cAAiD,CAAC;AACxD,eAAW,OAAO,OAAO,KAAK,QAAQ,GAAG;AACvC,YAAM,SAAS,gBAAgB,WAAW,SAAS,KAAK,IAAI,UAAU,KAAK;AAAA,QACzE,WAAW;AAAA,MACb,CAAC;AACD,kBAAY,OAAO;AACnB,kBAAY,GAAG,IAAI,OAAO;AAAA,IAC5B;AACA,gBAAY;AAAA,MACV;AAAA,MACA,OAAO,OAAO,WAAW,EAAE,IAAI,CAAC,SAAS,KAAK,EAAE;AAAA,IAClD;AAEA,WAAO,IAAI,iBAAiD,OAAO,OAAO,YAAY;AACpF,YAAM,QAAQ,MAAM,KAAK,QAAQ,OAAO,OAAO;AAC/C,YAAM,UAAU,MAAM;AAAA,QAAQ;AAAA,QAAS,SAAS;AAAA,QAAM,MACpD,QAAQ;AAAA,UACN,OAAO,QAAQ,QAAQ,EAAE,IAAI,OAAO,CAAC,KAAK,EAAE,MAAM;AAChD,kBAAM,OAAO,YAAY,GAAG;AAC5B,kBAAM,SAAS,MAAM,QAAQ,SAAS,MAAM,MAAM,GAAG,IAAI,KAAK,CAAC;AAC/D,mBAAO,CAAC,KAAK,MAAM;AAAA,UACrB,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO,OAAO,YAAY,OAAO;AAAA,IACnC,GAAG,SAAS;AAAA,EACd;AAAA;AAAA,EAGA,OACE,OACA,UACgC;AAChC,UAAM,OAAO,WAAW,KAAK,OAAO,SAAS,UAAU,QAAQ,MAAM,QAAQ,MAAM,IAAI;AAAA,MACrF,aAAa,UAAU,eAAe,MAAM;AAAA,MAC5C,UAAU,UAAU;AAAA,MACpB,aAAa,UAAU;AAAA,MACvB,SAAS,MAAM;AAAA,MACf,WAAW,MAAM;AAAA,IACnB,CAAC;AACD,WAAO,IAAI,iBAA+B,OAAO,OAAO,YAAY;AAClE,YAAM,QAAQ,MAAM,KAAK,QAAQ,OAAO,OAAO;AAC/C,aAAO,QAAQ,SAAS,KAAK,MAAM,YAAY;AAC7C,cAAM,WAAW,MAAM,MAAM,OAAO,OAAO,KAAK,CAAC,EAAE,KAAK;AACxD,eAAO,SAAS;AAAA,MAClB,CAAC;AAAA,IACH,GAAG,KAAK,KAAK;AAAA,EACf;AAAA;AAAA,EAGA,QACE,WACA,UAC2B;AAC3B,UAAM,OAAO;AAAA,MACX,KAAK;AAAA,MACL;AAAA,MACA,UAAU,QAAQ,eAAe,KAAK,OAAO,WAAW;AAAA,MACxD;AAAA,QACE,aAAa,UAAU;AAAA,QACvB,UAAU,UAAU;AAAA,QACpB,aAAa,UAAU;AAAA,MACzB;AAAA,IACF;AACA,WAAO,IAAI,iBAA0B,OAAO,OAAO,YAAY;AAC7D,YAAM,QAAQ,MAAM,KAAK,QAAQ,OAAO,OAAO;AAC/C,aAAO,QAAQ,SAAS,KAAK,MAAM,MAAM,UAAU,QAAQ,OAAO,KAAK,CAAC,CAAC;AAAA,IAC3E,GAAG,KAAK,KAAK;AAAA,EACf;AAAA;AAAA,EAGA,QAA0C;AACxC,UAAM,QAAQ,eAAe,KAAK,KAAK;AACvC,WAAO,IAAI;AAAA,MACT,CAAC,OAAO,YAAY,KAAK,QAAQ,OAAO,OAAO;AAAA,MAC/C;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,QAAQ,OAAc,SAAuD;AACzF,WAAQ,MAAM,KAAK,SAAS,OAAO,OAAO;AAAA,EAC5C;AACF;AAEA,SAAS,SAAY,OAAa;AAChC,SAAO;AACT;","names":[]}
+{"version":3,"sources":["../../src/pipeline/builder.ts","../../src/pipeline/graph.ts","../../src/pipeline/pipeline.ts","../../src/pipeline/runtime.ts"],"sourcesContent":["import { z } from \"zod\";\nimport type { Agent } from \"../agent/agent\";\nimport type { CompletionModel } from \"../completion\";\nimport type { Extractor } from \"../extractor\";\nimport {\n  appendChildNode,\n  appendNode,\n  initialBuilderState,\n  nextStageLabel,\n  withOutputNode,\n  withTerminalNodes,\n} from \"./graph\";\nimport { Pipeline } from \"./pipeline\";\nimport { runNode } from \"./runtime\";\nimport type {\n  ParallelOutput,\n  PipelineBuilderState,\n  PipelineExecutor,\n  PipelineGraphNode,\n  PipelineMetadata,\n  PipelineOp,\n  PipelineRunContext,\n  PipelineStageMetadata,\n} from \"./types\";\n\n/** Builds a typed pipeline from an original input type to an inferred output type. */\nexport class PipelineBuilder<Input, Output = Input> {\n  private readonly executor: PipelineExecutor<Input, Output>;\n  private readonly state: PipelineBuilderState;\n\n  constructor(schema: z.ZodType<Output, Input>);\n  constructor(schema: z.ZodType<Output, Input>, metadata: PipelineMetadata);\n  constructor(executor: (input: Input) => Output | Promise<Output>);\n  constructor(executor: PipelineExecutor<Input, Output>, state: PipelineBuilderState);\n  constructor(\n    metadataOrExecutor?:\n      | PipelineMetadata\n      | z.ZodType\n      | ((input: Input) => Output | Promise<Output>)\n      | PipelineExecutor<Input, Output>,\n    stateOrMetadata?: PipelineBuilderState | PipelineMetadata,\n  ) {\n    if (stateOrMetadata !== undefined && isBuilderState(stateOrMetadata)) {\n      this.executor = metadataOrExecutor as PipelineExecutor<Input, Output>;\n      this.state = stateOrMetadata;\n      return;\n    }\n\n    if (typeof metadataOrExecutor === \"function\") {\n      const executor = metadataOrExecutor as (input: Input) => Output | Promise<Output>;\n      this.executor = ((input) => executor(input)) as PipelineExecutor<Input, Output>;\n      this.state = initialBuilderState({});\n      return;\n    }\n\n    if (isZodSchema(metadataOrExecutor)) {\n      const schema = metadataOrExecutor;\n      const metadata: PipelineMetadata =\n        stateOrMetadata !== undefined && isPipelineMetadata(stateOrMetadata) ? stateOrMetadata : {};\n      this.executor = ((input) => schema.parse(input)) as PipelineExecutor<Input, Output>;\n      this.state = initialBuilderState(metadata);\n      return;\n    }\n\n    this.executor = identity as PipelineExecutor<Input, Output>;\n    this.state = initialBuilderState((metadataOrExecutor as PipelineMetadata | undefined) ?? {});\n  }\n\n  /** Add a synchronous or asynchronous transform stage. */\n  step<Next>(\n    fn: (input: Awaited<Output>) => Next | Promise<Next>,\n    metadata?: PipelineStageMetadata,\n  ): PipelineBuilder<Input, Awaited<Next>> {\n    const next = appendNode(\n      this.state,\n      \"step\",\n      metadata?.name ?? nextStageLabel(this.state, \"Step\"),\n      {\n        description: metadata?.description,\n        metadata: metadata?.metadata,\n        preferredId: metadata?.id,\n      },\n    );\n    return new PipelineBuilder<Input, Awaited<Next>>(\n      async (input, context): Promise<Awaited<Next>> => {\n        const value = await this.runStep(input, context);\n        const result = await runNode(context, next.node, () => fn(value));\n        return result as Awaited<Next>;\n      },\n      next.state,\n    );\n  }\n\n  /** Compose another pipeline operation after the current stage. */\n  use<Next>(\n    op: PipelineOp<Awaited<Output>, Next>,\n    metadata?: PipelineStageMetadata,\n  ): PipelineBuilder<Input, Awaited<Next>> {\n    const nested = op instanceof Pipeline ? op : undefined;\n    const next = appendNode(\n      this.state,\n      nested === undefined ? \"step\" : \"pipeline\",\n      metadata?.name ??\n        nested?.name ??\n        nested?.id ??\n        nextStageLabel(this.state, nested === undefined ? \"Operation\" : \"Pipeline\"),\n      {\n        description: metadata?.description ?? nested?.description,\n        metadata: metadata?.metadata ?? nested?.metadata,\n        preferredId: metadata?.id,\n        pipelineId: nested?.id,\n      },\n    );\n    return new PipelineBuilder<Input, Awaited<Next>>(\n      async (input, context): Promise<Awaited<Next>> => {\n        const value = await this.runStep(input, context);\n        const result = await runNode(context, next.node, () => op.run(value));\n        return result as Awaited<Next>;\n      },\n      next.state,\n    );\n  }\n\n  /** Run named branch operations concurrently from the current value. */\n  parallel<Branches extends Record<string, PipelineOp<Awaited<Output>, unknown>>>(\n    branches: Branches,\n    metadata?: PipelineStageMetadata,\n  ): PipelineBuilder<Input, ParallelOutput<Branches>> {\n    const parallel = appendNode(\n      this.state,\n      \"parallel\",\n      metadata?.name ?? `${Object.keys(branches).length} parallel branches`,\n      {\n        description: metadata?.description,\n        metadata: metadata?.metadata,\n        preferredId: metadata?.id,\n      },\n    );\n    let nextState = parallel.state;\n    const branchNodes: Record<string, PipelineGraphNode> = {};\n    for (const key of Object.keys(branches)) {\n      const branch = appendChildNode(nextState, parallel.node.id, \"branch\", key, {\n        branchKey: key,\n      });\n      nextState = branch.state;\n      branchNodes[key] = branch.node;\n    }\n    nextState = withTerminalNodes(\n      nextState,\n      Object.values(branchNodes).map((node) => node.id),\n    );\n\n    return new PipelineBuilder<Input, ParallelOutput<Branches>>(async (input, context) => {\n      const value = await this.runStep(input, context);\n      const entries = await runNode(context, parallel.node, () =>\n        Promise.all(\n          Object.entries(branches).map(async ([key, op]) => {\n            const node = branchNodes[key] as PipelineGraphNode;\n            const output = await runNode(context, node, () => op.run(value));\n            return [key, output] as const;\n          }),\n        ),\n      );\n      return Object.fromEntries(entries) as ParallelOutput<Branches>;\n    }, nextState);\n  }\n\n  /** Send the current value to an agent as text and continue with the agent output. */\n  prompt(\n    agent: Agent<CompletionModel>,\n    metadata?: PipelineStageMetadata,\n  ): PipelineBuilder<Input, string> {\n    const next = appendNode(this.state, \"agent\", metadata?.name ?? agent.name ?? agent.id, {\n      description: metadata?.description ?? agent.description,\n      metadata: metadata?.metadata,\n      preferredId: metadata?.id,\n      agentId: agent.id,\n      agentName: agent.name,\n    });\n    return new PipelineBuilder<Input, string>(async (input, context) => {\n      const value = await this.runStep(input, context);\n      return runNode(context, next.node, async () => {\n        const response = await agent.prompt(String(value)).send();\n        return response.output;\n      });\n    }, next.state);\n  }\n\n  /** Send the current value to an extractor as text and continue with typed schema data. */\n  extract<T>(\n    extractor: Extractor<T, CompletionModel>,\n    metadata?: PipelineStageMetadata,\n  ): PipelineBuilder<Input, T> {\n    const next = appendNode(\n      this.state,\n      \"extractor\",\n      metadata?.name ?? nextStageLabel(this.state, \"Extractor\"),\n      {\n        description: metadata?.description,\n        metadata: metadata?.metadata,\n        preferredId: metadata?.id,\n      },\n    );\n    return new PipelineBuilder<Input, T>(async (input, context) => {\n      const value = await this.runStep(input, context);\n      return runNode(context, next.node, () => extractor.extract(String(value)));\n    }, next.state);\n  }\n\n  /** Finish the builder and return a runnable pipeline. */\n  build(): Pipeline<Input, Awaited<Output>> {\n    const graph = withOutputNode(this.state);\n    return new Pipeline<Input, Awaited<Output>>(\n      (input, context) => this.runStep(input, context) as Promise<Awaited<Output>>,\n      graph,\n    );\n  }\n\n  private async runStep(input: Input, context: PipelineRunContext): Promise<Awaited<Output>> {\n    return (await this.executor(input, context)) as Awaited<Output>;\n  }\n}\n\nfunction identity<T>(input: T): T {\n  return input;\n}\n\nfunction isZodSchema(value: unknown): value is z.ZodType {\n  return value instanceof z.ZodType;\n}\n\nfunction isBuilderState(value: unknown): value is PipelineBuilderState {\n  if (typeof value !== \"object\" || value === null || Array.isArray(value)) {\n    return false;\n  }\n  const candidate = value as Record<string, unknown>;\n  return (\n    \"graph\" in candidate &&\n    \"terminalNodeId\" in candidate &&\n    \"terminalNodeIds\" in candidate &&\n    \"nextNodeIndex\" in candidate &&\n    \"nextEdgeIndex\" in candidate\n  );\n}\n\nfunction isPipelineMetadata(value: unknown): value is PipelineMetadata {\n  if (typeof value !== \"object\" || value === null || Array.isArray(value)) {\n    return false;\n  }\n  const candidate = value as Record<string, unknown>;\n  return (\n    \"id\" in candidate ||\n    \"name\" in candidate ||\n    \"description\" in candidate ||\n    \"metadata\" in candidate\n  );\n}\n","import type { JsonObject } from \"../completion\";\nimport { compact } from \"../internal/compact\";\nimport type {\n  PipelineBuilderState,\n  PipelineGraph,\n  PipelineGraphNode,\n  PipelineMetadata,\n  PipelineStageKind,\n} from \"./types\";\n\nexport function initialBuilderState(metadata: PipelineMetadata): PipelineBuilderState {\n  return {\n    graph: initialGraph(metadata),\n    terminalNodeId: \"input\",\n    terminalNodeIds: [\"input\"],\n    nextNodeIndex: 1,\n    nextEdgeIndex: 1,\n  };\n}\n\nexport function initialGraph(metadata: PipelineMetadata): PipelineGraph {\n  const id = normalizeId(metadata.id ?? \"pipeline\");\n  return {\n    ...compact({\n      id,\n      name: metadata.name,\n      description: metadata.description,\n      metadata: metadata.metadata,\n    }),\n    nodes: [{ id: \"input\", kind: \"input\", label: \"Input\" }],\n    edges: [],\n  };\n}\n\nexport function appendNode(\n  state: PipelineBuilderState,\n  kind: PipelineStageKind,\n  label: string,\n  options: {\n    description?: string | undefined;\n    metadata?: JsonObject | undefined;\n    preferredId?: string | undefined;\n    agentId?: string | undefined;\n    agentName?: string | undefined;\n    pipelineId?: string | undefined;\n  } = {},\n): { state: PipelineBuilderState; node: PipelineGraphNode } {\n  const node = graphNode(kind, label, state.nextNodeIndex, {\n    ...options,\n    existingIds: new Set(state.graph.nodes.map((item) => item.id)),\n  });\n  return {\n    node,\n    state: appendGraphNode(state, node, activeTerminalNodeIds(state), [node.id]),\n  };\n}\n\nexport function appendChildNode(\n  state: PipelineBuilderState,\n  parentId: string,\n  kind: PipelineStageKind,\n  label: string,\n  options: {\n    branchKey?: string | undefined;\n  } = {},\n): { state: PipelineBuilderState; node: PipelineGraphNode } {\n  const node = graphNode(kind, label, state.nextNodeIndex, {\n    ...options,\n    existingIds: new Set(state.graph.nodes.map((item) => item.id)),\n  });\n  return {\n    node,\n    state: appendGraphNode(state, node, [parentId], activeTerminalNodeIds(state)),\n  };\n}\n\nexport function activeTerminalNodeIds(state: PipelineBuilderState): string[] {\n  return state.terminalNodeIds.length > 0 ? state.terminalNodeIds : [state.terminalNodeId];\n}\n\nexport function withTerminalNodes(\n  state: PipelineBuilderState,\n  terminalNodeIds: string[],\n): PipelineBuilderState {\n  return {\n    ...state,\n    terminalNodeId: terminalNodeIds.at(-1) ?? state.terminalNodeId,\n    terminalNodeIds,\n  };\n}\n\nexport function withOutputNode(state: PipelineBuilderState): PipelineGraph {\n  const graph = cloneGraph(state.graph);\n  if (graph.nodes.some((node) => node.id === \"output\")) {\n    return graph;\n  }\n  graph.nodes.push({ id: \"output\", kind: \"output\", label: \"Output\" });\n  graph.edges.push(\n    ...activeTerminalNodeIds(state).map((sourceId, index) => ({\n      id: `edge_${state.nextEdgeIndex + index}`,\n      source: sourceId,\n      target: \"output\",\n    })),\n  );\n  return graph;\n}\n\nexport function nextStageLabel(state: PipelineBuilderState, prefix: string): string {\n  return `${prefix} ${state.nextNodeIndex}`;\n}\n\nexport function cloneGraph(graph: PipelineGraph): PipelineGraph {\n  return {\n    ...graph,\n    nodes: graph.nodes.map((node) => ({ ...node })),\n    edges: graph.edges.map((edge) => ({ ...edge })),\n  };\n}\n\nfunction appendGraphNode(\n  state: PipelineBuilderState,\n  node: PipelineGraphNode,\n  sourceIds: string[],\n  terminalNodeIds: string[],\n): PipelineBuilderState {\n  const edges = sourceIds.map((sourceId, index) => ({\n    id: `edge_${state.nextEdgeIndex + index}`,\n    source: sourceId,\n    target: node.id,\n  }));\n  const terminalNodeId = terminalNodeIds.at(-1) ?? state.terminalNodeId;\n  return {\n    graph: {\n      ...state.graph,\n      nodes: [...state.graph.nodes, node],\n      edges: [...state.graph.edges, ...edges],\n    },\n    terminalNodeId,\n    terminalNodeIds,\n    nextNodeIndex: state.nextNodeIndex + 1,\n    nextEdgeIndex: state.nextEdgeIndex + edges.length,\n  };\n}\n\nfunction graphNode(\n  kind: PipelineStageKind,\n  label: string,\n  index: number,\n  options: {\n    description?: string | undefined;\n    metadata?: JsonObject | undefined;\n    preferredId?: string | undefined;\n    agentId?: string | undefined;\n    agentName?: string | undefined;\n    pipelineId?: string | undefined;\n    branchKey?: string | undefined;\n    existingIds?: Set<string> | undefined;\n  } = {},\n): PipelineGraphNode {\n  const id = uniqueGraphNodeId(\n    normalizeId(options.preferredId ?? `${kind}_${index}`),\n    options.existingIds ?? new Set(),\n  );\n  return compact({\n    id,\n    kind,\n    label,\n    description: options.description,\n    metadata: options.metadata,\n    agentId: options.agentId,\n    agentName: options.agentName,\n    pipelineId: options.pipelineId,\n    branchKey: options.branchKey,\n  }) as PipelineGraphNode;\n}\n\nfunction normalizeId(value: string): string {\n  const normalized = value\n    .trim()\n    .toLowerCase()\n    .replace(/[^a-z0-9_-]+/g, \"_\")\n    .replace(/^_+|_+$/g, \"\");\n  return normalized.length === 0 ? \"pipeline\" : normalized;\n}\n\nfunction uniqueGraphNodeId(baseId: string, existingIds: Set<string>): string {\n  let id = baseId;\n  let suffix = 2;\n  while (existingIds.has(id)) {\n    id = `${baseId}_${suffix}`;\n    suffix += 1;\n  }\n  return id;\n}\n","import type { JsonObject } from \"../completion\";\nimport { mapWithConcurrency } from \"../internal/concurrency\";\nimport { cloneGraph, initialGraph } from \"./graph\";\nimport type {\n  PipelineBatchOptions,\n  PipelineExecutor,\n  PipelineGraph,\n  PipelineOp,\n  PipelineRunOptions,\n} from \"./types\";\n\n/** Runnable pipeline returned by `PipelineBuilder.build()`. */\nexport class Pipeline<Input, Output> implements PipelineOp<Input, Awaited<Output>> {\n  readonly id: string;\n  readonly name: string | undefined;\n  readonly description: string | undefined;\n  readonly metadata: JsonObject | undefined;\n\n  constructor(\n    private readonly executor: PipelineExecutor<Input, Output>,\n    private readonly pipelineGraph: PipelineGraph = initialGraph({}),\n  ) {\n    this.id = pipelineGraph.id;\n    this.name = pipelineGraph.name;\n    this.description = pipelineGraph.description;\n    this.metadata = pipelineGraph.metadata;\n  }\n\n  /** Run one input through the built pipeline and return the final stage output. */\n  async run(input: Input, options: PipelineRunOptions = {}): Promise<Awaited<Output>> {\n    return (await this.executor(input, { observer: options.observer })) as Awaited<Output>;\n  }\n\n  /** Run many inputs through the same pipeline with bounded concurrency. */\n  async batch<I extends Iterable<Input>>(\n    inputs: I,\n    options: PipelineBatchOptions,\n  ): Promise<Array<Awaited<Output>>> {\n    return mapWithConcurrency([...inputs], options.concurrency, (input) => this.run(input));\n  }\n\n  graph(): PipelineGraph {\n    return cloneGraph(this.pipelineGraph);\n  }\n}\n","import type { PipelineGraphNode, PipelineRunContext } from \"./types\";\n\nexport async function runNode<Output>(\n  context: PipelineRunContext,\n  node: PipelineGraphNode,\n  fn: () => Output | Promise<Output>,\n): Promise<Awaited<Output>> {\n  const startedAt = Date.now();\n  await context.observer?.onEvent({ type: \"stage_started\", node });\n  try {\n    const output = (await fn()) as Awaited<Output>;\n    await context.observer?.onEvent({\n      type: \"stage_completed\",\n      node,\n      durationMs: Date.now() - startedAt,\n    });\n    return output;\n  } catch (error) {\n    await context.observer?.onEvent({\n      type: \"stage_failed\",\n      node,\n      durationMs: Date.now() - startedAt,\n      error,\n    });\n    throw error;\n  }\n}\n"],"mappings":";;;;;;;;AAAA,SAAS,SAAS;;;ACUX,SAAS,oBAAoB,UAAkD;AACpF,SAAO;AAAA,IACL,OAAO,aAAa,QAAQ;AAAA,IAC5B,gBAAgB;AAAA,IAChB,iBAAiB,CAAC,OAAO;AAAA,IACzB,eAAe;AAAA,IACf,eAAe;AAAA,EACjB;AACF;AAEO,SAAS,aAAa,UAA2C;AACtE,QAAM,KAAK,YAAY,SAAS,MAAM,UAAU;AAChD,SAAO;AAAA,IACL,GAAG,QAAQ;AAAA,MACT;AAAA,MACA,MAAM,SAAS;AAAA,MACf,aAAa,SAAS;AAAA,MACtB,UAAU,SAAS;AAAA,IACrB,CAAC;AAAA,IACD,OAAO,CAAC,EAAE,IAAI,SAAS,MAAM,SAAS,OAAO,QAAQ,CAAC;AAAA,IACtD,OAAO,CAAC;AAAA,EACV;AACF;AAEO,SAAS,WACd,OACA,MACA,OACA,UAOI,CAAC,GACqD;AAC1D,QAAM,OAAO,UAAU,MAAM,OAAO,MAAM,eAAe;AAAA,IACvD,GAAG;AAAA,IACH,aAAa,IAAI,IAAI,MAAM,MAAM,MAAM,IAAI,CAAC,SAAS,KAAK,EAAE,CAAC;AAAA,EAC/D,CAAC;AACD,SAAO;AAAA,IACL;AAAA,IACA,OAAO,gBAAgB,OAAO,MAAM,sBAAsB,KAAK,GAAG,CAAC,KAAK,EAAE,CAAC;AAAA,EAC7E;AACF;AAEO,SAAS,gBACd,OACA,UACA,MACA,OACA,UAEI,CAAC,GACqD;AAC1D,QAAM,OAAO,UAAU,MAAM,OAAO,MAAM,eAAe;AAAA,IACvD,GAAG;AAAA,IACH,aAAa,IAAI,IAAI,MAAM,MAAM,MAAM,IAAI,CAAC,SAAS,KAAK,EAAE,CAAC;AAAA,EAC/D,CAAC;AACD,SAAO;AAAA,IACL;AAAA,IACA,OAAO,gBAAgB,OAAO,MAAM,CAAC,QAAQ,GAAG,sBAAsB,KAAK,CAAC;AAAA,EAC9E;AACF;AAEO,SAAS,sBAAsB,OAAuC;AAC3E,SAAO,MAAM,gBAAgB,SAAS,IAAI,MAAM,kBAAkB,CAAC,MAAM,cAAc;AACzF;AAEO,SAAS,kBACd,OACA,iBACsB;AACtB,SAAO;AAAA,IACL,GAAG;AAAA,IACH,gBAAgB,gBAAgB,GAAG,EAAE,KAAK,MAAM;AAAA,IAChD;AAAA,EACF;AACF;AAEO,SAAS,eAAe,OAA4C;AACzE,QAAM,QAAQ,WAAW,MAAM,KAAK;AACpC,MAAI,MAAM,MAAM,KAAK,CAAC,SAAS,KAAK,OAAO,QAAQ,GAAG;AACpD,WAAO;AAAA,EACT;AACA,QAAM,MAAM,KAAK,EAAE,IAAI,UAAU,MAAM,UAAU,OAAO,SAAS,CAAC;AAClE,QAAM,MAAM;AAAA,IACV,GAAG,sBAAsB,KAAK,EAAE,IAAI,CAAC,UAAU,WAAW;AAAA,MACxD,IAAI,QAAQ,MAAM,gBAAgB,KAAK;AAAA,MACvC,QAAQ;AAAA,MACR,QAAQ;AAAA,IACV,EAAE;AAAA,EACJ;AACA,SAAO;AACT;AAEO,SAAS,eAAe,OAA6B,QAAwB;AAClF,SAAO,GAAG,MAAM,IAAI,MAAM,aAAa;AACzC;AAEO,SAAS,WAAW,OAAqC;AAC9D,SAAO;AAAA,IACL,GAAG;AAAA,IACH,OAAO,MAAM,MAAM,IAAI,CAAC,UAAU,EAAE,GAAG,KAAK,EAAE;AAAA,IAC9C,OAAO,MAAM,MAAM,IAAI,CAAC,UAAU,EAAE,GAAG,KAAK,EAAE;AAAA,EAChD;AACF;AAEA,SAAS,gBACP,OACA,MACA,WACA,iBACsB;AACtB,QAAM,QAAQ,UAAU,IAAI,CAAC,UAAU,WAAW;AAAA,IAChD,IAAI,QAAQ,MAAM,gBAAgB,KAAK;AAAA,IACvC,QAAQ;AAAA,IACR,QAAQ,KAAK;AAAA,EACf,EAAE;AACF,QAAM,iBAAiB,gBAAgB,GAAG,EAAE,KAAK,MAAM;AACvD,SAAO;AAAA,IACL,OAAO;AAAA,MACL,GAAG,MAAM;AAAA,MACT,OAAO,CAAC,GAAG,MAAM,MAAM,OAAO,IAAI;AAAA,MAClC,OAAO,CAAC,GAAG,MAAM,MAAM,OAAO,GAAG,KAAK;AAAA,IACxC;AAAA,IACA;AAAA,IACA;AAAA,IACA,eAAe,MAAM,gBAAgB;AAAA,IACrC,eAAe,MAAM,gBAAgB,MAAM;AAAA,EAC7C;AACF;AAEA,SAAS,UACP,MACA,OACA,OACA,UASI,CAAC,GACc;AACnB,QAAM,KAAK;AAAA,IACT,YAAY,QAAQ,eAAe,GAAG,IAAI,IAAI,KAAK,EAAE;AAAA,IACrD,QAAQ,eAAe,oBAAI,IAAI;AAAA,EACjC;AACA,SAAO,QAAQ;AAAA,IACb;AAAA,IACA;AAAA,IACA;AAAA,IACA,aAAa,QAAQ;AAAA,IACrB,UAAU,QAAQ;AAAA,IAClB,SAAS,QAAQ;AAAA,IACjB,WAAW,QAAQ;AAAA,IACnB,YAAY,QAAQ;AAAA,IACpB,WAAW,QAAQ;AAAA,EACrB,CAAC;AACH;AAEA,SAAS,YAAY,OAAuB;AAC1C,QAAM,aAAa,MAChB,KAAK,EACL,YAAY,EACZ,QAAQ,iBAAiB,GAAG,EAC5B,QAAQ,YAAY,EAAE;AACzB,SAAO,WAAW,WAAW,IAAI,aAAa;AAChD;AAEA,SAAS,kBAAkB,QAAgB,aAAkC;AAC3E,MAAI,KAAK;AACT,MAAI,SAAS;AACb,SAAO,YAAY,IAAI,EAAE,GAAG;AAC1B,SAAK,GAAG,MAAM,IAAI,MAAM;AACxB,cAAU;AAAA,EACZ;AACA,SAAO;AACT;;;ACrLO,IAAM,WAAN,MAA4E;AAAA,EAMjF,YACmB,UACA,gBAA+B,aAAa,CAAC,CAAC,GAC/D;AAFiB;AACA;AAEjB,SAAK,KAAK,cAAc;AACxB,SAAK,OAAO,cAAc;AAC1B,SAAK,cAAc,cAAc;AACjC,SAAK,WAAW,cAAc;AAAA,EAChC;AAAA,EAPmB;AAAA,EACA;AAAA,EAPV;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EAaT,MAAM,IAAI,OAAc,UAA8B,CAAC,GAA6B;AAClF,WAAQ,MAAM,KAAK,SAAS,OAAO,EAAE,UAAU,QAAQ,SAAS,CAAC;AAAA,EACnE;AAAA;AAAA,EAGA,MAAM,MACJ,QACA,SACiC;AACjC,WAAO,mBAAmB,CAAC,GAAG,MAAM,GAAG,QAAQ,aAAa,CAAC,UAAU,KAAK,IAAI,KAAK,CAAC;AAAA,EACxF;AAAA,EAEA,QAAuB;AACrB,WAAO,WAAW,KAAK,aAAa;AAAA,EACtC;AACF;;;AC1CA,eAAsB,QACpB,SACA,MACA,IAC0B;AAC1B,QAAM,YAAY,KAAK,IAAI;AAC3B,QAAM,QAAQ,UAAU,QAAQ,EAAE,MAAM,iBAAiB,KAAK,CAAC;AAC/D,MAAI;AACF,UAAM,SAAU,MAAM,GAAG;AACzB,UAAM,QAAQ,UAAU,QAAQ;AAAA,MAC9B,MAAM;AAAA,MACN;AAAA,MACA,YAAY,KAAK,IAAI,IAAI;AAAA,IAC3B,CAAC;AACD,WAAO;AAAA,EACT,SAAS,OAAO;AACd,UAAM,QAAQ,UAAU,QAAQ;AAAA,MAC9B,MAAM;AAAA,MACN;AAAA,MACA,YAAY,KAAK,IAAI,IAAI;AAAA,MACzB;AAAA,IACF,CAAC;AACD,UAAM;AAAA,EACR;AACF;;;AHAO,IAAM,kBAAN,MAAM,iBAAuC;AAAA,EACjC;AAAA,EACA;AAAA,EAMjB,YACE,oBAKA,iBACA;AACA,QAAI,oBAAoB,UAAa,eAAe,eAAe,GAAG;AACpE,WAAK,WAAW;AAChB,WAAK,QAAQ;AACb;AAAA,IACF;AAEA,QAAI,OAAO,uBAAuB,YAAY;AAC5C,YAAM,WAAW;AACjB,WAAK,YAAY,CAAC,UAAU,SAAS,KAAK;AAC1C,WAAK,QAAQ,oBAAoB,CAAC,CAAC;AACnC;AAAA,IACF;AAEA,QAAI,YAAY,kBAAkB,GAAG;AACnC,YAAM,SAAS;AACf,YAAM,WACJ,oBAAoB,UAAa,mBAAmB,eAAe,IAAI,kBAAkB,CAAC;AAC5F,WAAK,YAAY,CAAC,UAAU,OAAO,MAAM,KAAK;AAC9C,WAAK,QAAQ,oBAAoB,QAAQ;AACzC;AAAA,IACF;AAEA,SAAK,WAAW;AAChB,SAAK,QAAQ,oBAAqB,sBAAuD,CAAC,CAAC;AAAA,EAC7F;AAAA;AAAA,EAGA,KACE,IACA,UACuC;AACvC,UAAM,OAAO;AAAA,MACX,KAAK;AAAA,MACL;AAAA,MACA,UAAU,QAAQ,eAAe,KAAK,OAAO,MAAM;AAAA,MACnD;AAAA,QACE,aAAa,UAAU;AAAA,QACvB,UAAU,UAAU;AAAA,QACpB,aAAa,UAAU;AAAA,MACzB;AAAA,IACF;AACA,WAAO,IAAI;AAAA,MACT,OAAO,OAAO,YAAoC;AAChD,cAAM,QAAQ,MAAM,KAAK,QAAQ,OAAO,OAAO;AAC/C,cAAM,SAAS,MAAM,QAAQ,SAAS,KAAK,MAAM,MAAM,GAAG,KAAK,CAAC;AAChE,eAAO;AAAA,MACT;AAAA,MACA,KAAK;AAAA,IACP;AAAA,EACF;AAAA;AAAA,EAGA,IACE,IACA,UACuC;AACvC,UAAM,SAAS,cAAc,WAAW,KAAK;AAC7C,UAAM,OAAO;AAAA,MACX,KAAK;AAAA,MACL,WAAW,SAAY,SAAS;AAAA,MAChC,UAAU,QACR,QAAQ,QACR,QAAQ,MACR,eAAe,KAAK,OAAO,WAAW,SAAY,cAAc,UAAU;AAAA,MAC5E;AAAA,QACE,aAAa,UAAU,eAAe,QAAQ;AAAA,QAC9C,UAAU,UAAU,YAAY,QAAQ;AAAA,QACxC,aAAa,UAAU;AAAA,QACvB,YAAY,QAAQ;AAAA,MACtB;AAAA,IACF;AACA,WAAO,IAAI;AAAA,MACT,OAAO,OAAO,YAAoC;AAChD,cAAM,QAAQ,MAAM,KAAK,QAAQ,OAAO,OAAO;AAC/C,cAAM,SAAS,MAAM,QAAQ,SAAS,KAAK,MAAM,MAAM,GAAG,IAAI,KAAK,CAAC;AACpE,eAAO;AAAA,MACT;AAAA,MACA,KAAK;AAAA,IACP;AAAA,EACF;AAAA;AAAA,EAGA,SACE,UACA,UACkD;AAClD,UAAM,WAAW;AAAA,MACf,KAAK;AAAA,MACL;AAAA,MACA,UAAU,QAAQ,GAAG,OAAO,KAAK,QAAQ,EAAE,MAAM;AAAA,MACjD;AAAA,QACE,aAAa,UAAU;AAAA,QACvB,UAAU,UAAU;AAAA,QACpB,aAAa,UAAU;AAAA,MACzB;AAAA,IACF;AACA,QAAI,YAAY,SAAS;AACzB,UAAM,cAAiD,CAAC;AACxD,eAAW,OAAO,OAAO,KAAK,QAAQ,GAAG;AACvC,YAAM,SAAS,gBAAgB,WAAW,SAAS,KAAK,IAAI,UAAU,KAAK;AAAA,QACzE,WAAW;AAAA,MACb,CAAC;AACD,kBAAY,OAAO;AACnB,kBAAY,GAAG,IAAI,OAAO;AAAA,IAC5B;AACA,gBAAY;AAAA,MACV;AAAA,MACA,OAAO,OAAO,WAAW,EAAE,IAAI,CAAC,SAAS,KAAK,EAAE;AAAA,IAClD;AAEA,WAAO,IAAI,iBAAiD,OAAO,OAAO,YAAY;AACpF,YAAM,QAAQ,MAAM,KAAK,QAAQ,OAAO,OAAO;AAC/C,YAAM,UAAU,MAAM;AAAA,QAAQ;AAAA,QAAS,SAAS;AAAA,QAAM,MACpD,QAAQ;AAAA,UACN,OAAO,QAAQ,QAAQ,EAAE,IAAI,OAAO,CAAC,KAAK,EAAE,MAAM;AAChD,kBAAM,OAAO,YAAY,GAAG;AAC5B,kBAAM,SAAS,MAAM,QAAQ,SAAS,MAAM,MAAM,GAAG,IAAI,KAAK,CAAC;AAC/D,mBAAO,CAAC,KAAK,MAAM;AAAA,UACrB,CAAC;AAAA,QACH;AAAA,MACF;AACA,aAAO,OAAO,YAAY,OAAO;AAAA,IACnC,GAAG,SAAS;AAAA,EACd;AAAA;AAAA,EAGA,OACE,OACA,UACgC;AAChC,UAAM,OAAO,WAAW,KAAK,OAAO,SAAS,UAAU,QAAQ,MAAM,QAAQ,MAAM,IAAI;AAAA,MACrF,aAAa,UAAU,eAAe,MAAM;AAAA,MAC5C,UAAU,UAAU;AAAA,MACpB,aAAa,UAAU;AAAA,MACvB,SAAS,MAAM;AAAA,MACf,WAAW,MAAM;AAAA,IACnB,CAAC;AACD,WAAO,IAAI,iBAA+B,OAAO,OAAO,YAAY;AAClE,YAAM,QAAQ,MAAM,KAAK,QAAQ,OAAO,OAAO;AAC/C,aAAO,QAAQ,SAAS,KAAK,MAAM,YAAY;AAC7C,cAAM,WAAW,MAAM,MAAM,OAAO,OAAO,KAAK,CAAC,EAAE,KAAK;AACxD,eAAO,SAAS;AAAA,MAClB,CAAC;AAAA,IACH,GAAG,KAAK,KAAK;AAAA,EACf;AAAA;AAAA,EAGA,QACE,WACA,UAC2B;AAC3B,UAAM,OAAO;AAAA,MACX,KAAK;AAAA,MACL;AAAA,MACA,UAAU,QAAQ,eAAe,KAAK,OAAO,WAAW;AAAA,MACxD;AAAA,QACE,aAAa,UAAU;AAAA,QACvB,UAAU,UAAU;AAAA,QACpB,aAAa,UAAU;AAAA,MACzB;AAAA,IACF;AACA,WAAO,IAAI,iBAA0B,OAAO,OAAO,YAAY;AAC7D,YAAM,QAAQ,MAAM,KAAK,QAAQ,OAAO,OAAO;AAC/C,aAAO,QAAQ,SAAS,KAAK,MAAM,MAAM,UAAU,QAAQ,OAAO,KAAK,CAAC,CAAC;AAAA,IAC3E,GAAG,KAAK,KAAK;AAAA,EACf;AAAA;AAAA,EAGA,QAA0C;AACxC,UAAM,QAAQ,eAAe,KAAK,KAAK;AACvC,WAAO,IAAI;AAAA,MACT,CAAC,OAAO,YAAY,KAAK,QAAQ,OAAO,OAAO;AAAA,MAC/C;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,QAAQ,OAAc,SAAuD;AACzF,WAAQ,MAAM,KAAK,SAAS,OAAO,OAAO;AAAA,EAC5C;AACF;AAEA,SAAS,SAAY,OAAa;AAChC,SAAO;AACT;AAEA,SAAS,YAAY,OAAoC;AACvD,SAAO,iBAAiB,EAAE;AAC5B;AAEA,SAAS,eAAe,OAA+C;AACrE,MAAI,OAAO,UAAU,YAAY,UAAU,QAAQ,MAAM,QAAQ,KAAK,GAAG;AACvE,WAAO;AAAA,EACT;AACA,QAAM,YAAY;AAClB,SACE,WAAW,aACX,oBAAoB,aACpB,qBAAqB,aACrB,mBAAmB,aACnB,mBAAmB;AAEvB;AAEA,SAAS,mBAAmB,OAA2C;AACrE,MAAI,OAAO,UAAU,YAAY,UAAU,QAAQ,MAAM,QAAQ,KAAK,GAAG;AACvE,WAAO;AAAA,EACT;AACA,QAAM,YAAY;AAClB,SACE,QAAQ,aACR,UAAU,aACV,iBAAiB,aACjB,cAAc;AAElB;","names":[]}

package/dist/skills/index.d.ts

@@ -1,6 +1,6 @@
-import { b as SkillLoader, a as SkillSet } from '../types-BfEFm6zu.js';
-export { c as Skill, S as SkillValidationError, d as SkillValidationIssue } from '../types-BfEFm6zu.js';
-import '../tool-Bwj_zoTx.js';
+import { b as SkillLoader, a as SkillSet } from '../types-D-YTJdLj.js';
+export { c as Skill, S as SkillValidationError, d as SkillValidationIssue } from '../types-D-YTJdLj.js';
+import '../tool-BLypI31e.js';
 import '../types-C8loR4il.js';
 
 declare function loadSkills(loaders: SkillLoader | SkillLoader[]): Promise<SkillSet>;

package/dist/skills/index.js

@@ -2,15 +2,15 @@ import {
   SkillValidationError,
   loadSkills,
   skill
-} from "../chunk-ZT2YH2GA.js";
+} from "../chunk-ITYXP4CL.js";
 import "../chunk-YK4WAAS4.js";
-import "../chunk-5OGAUWM5.js";
-import "../chunk-66IJDP6L.js";
+import "../chunk-USWULPZJ.js";
+import "../chunk-VQE5PAAT.js";
 import "../chunk-MMHG7WAM.js";
-import "../chunk-WJO6NVE2.js";
-import "../chunk-WQKHFADH.js";
 import "../chunk-FTO3W4UP.js";
 import "../chunk-OIMLU4SF.js";
+import "../chunk-WJO6NVE2.js";
+import "../chunk-WQKHFADH.js";
 export {
   SkillValidationError,
   loadSkills,

package/dist/{think-tool-DQw-Oz9J.d.ts → think-tool-B8Rqw2i_.d.ts}

@@ -1,6 +1,6 @@
 import { z } from 'zod';
 import { Z as ZodSchema } from './zod-schema-C7F4clpm.js';
-import { b as ToolApprovalPolicy, c as ToolCallContext, T as Tool } from './tool-Bwj_zoTx.js';
+import { c as ToolApprovalPolicy, f as ToolCallContext, T as Tool } from './tool-BLypI31e.js';
 
 type CreateToolOptions<InputSchema extends ZodSchema, OutputSchema extends ZodSchema | undefined = undefined, Output = unknown> = {
     name: string;

package/dist/tool/index.d.ts

@@ -1,6 +1,6 @@
-export { b as CreateThinkToolOptions, C as CreateToolOptions, c as createThinkTool, a as createTool } from '../think-tool-DQw-Oz9J.js';
-export { A as AgentMiddleware, C as CompletionRequestMiddlewareArgs, a as CompletionRequestMiddlewareResult, b as CompletionResponseMiddlewareArgs, c as CompletionResponseMiddlewareResult, D as DynamicToolIndex, E as EmbedToolsOptions, T as ToolInputMiddlewareArgs, d as ToolInputMiddlewareResult, e as ToolMiddleware, f as ToolOutputMiddlewareArgs, g as ToolOutputMiddlewareResult, h as ToolResultMiddlewareArgs, k as ToolSearchDocument, l as ToolSet, i as createMiddleware, m as createToolIndex, j as createToolMiddleware, n as embedTools, o as isDynamicToolIndex } from '../middleware-C0eXGwoh.js';
-export { A as AnyTool, N as NormalizedToolOutput, T as Tool, a as ToolApprovalContext, b as ToolApprovalPolicy, e as ToolApprovalRunContext, c as ToolCallContext, d as ToolCallStreamEvent, f as ToolOutput, n as normalizeToolResultOutput, p as parseToolArgs, t as toolResultContentToText } from '../tool-Bwj_zoTx.js';
+export { b as CreateThinkToolOptions, C as CreateToolOptions, c as createThinkTool, a as createTool } from '../think-tool-B8Rqw2i_.js';
+export { A as AgentMiddleware, C as CompletionRequestMiddlewareArgs, a as CompletionRequestMiddlewareResult, b as CompletionResponseMiddlewareArgs, c as CompletionResponseMiddlewareResult, D as DynamicToolIndex, E as EmbedToolsOptions, T as ToolInputMiddlewareArgs, d as ToolInputMiddlewareResult, e as ToolMiddleware, f as ToolOutputMiddlewareArgs, g as ToolOutputMiddlewareResult, h as ToolResultMiddlewareArgs, k as ToolSearchDocument, l as ToolSet, i as createMiddleware, m as createToolIndex, j as createToolMiddleware, n as embedTools, o as isDynamicToolIndex } from '../middleware-CZocYAtZ.js';
+export { A as AnyTool, N as NormalizedToolOutput, T as Tool, a as ToolApprovalContext, b as ToolApprovalDecision, c as ToolApprovalPolicy, d as ToolApprovalRequest, h as ToolApprovalRunContext, e as ToolApprovalsOptions, f as ToolCallContext, g as ToolCallStreamEvent, i as ToolOutput, n as normalizeToolResultOutput, p as parseToolArgs, t as toolResultContentToText } from '../tool-BLypI31e.js';
 export { q as isToolResultContentArray, s as serializeToolOutput } from '../types-C8loR4il.js';
 import 'zod';
 import '../zod-schema-C7F4clpm.js';

package/dist/tool/index.js

@@ -13,18 +13,18 @@ import {
   normalizeToolResultOutput,
   parseToolArgs,
   toolResultContentToText
-} from "../chunk-5OGAUWM5.js";
+} from "../chunk-USWULPZJ.js";
 import {
   createTool
-} from "../chunk-66IJDP6L.js";
+} from "../chunk-VQE5PAAT.js";
 import "../chunk-MMHG7WAM.js";
+import "../chunk-FTO3W4UP.js";
+import "../chunk-OIMLU4SF.js";
 import {
   isToolResultContentArray,
   serializeToolResultOutput
 } from "../chunk-WJO6NVE2.js";
 import "../chunk-WQKHFADH.js";
-import "../chunk-FTO3W4UP.js";
-import "../chunk-OIMLU4SF.js";
 export {
   ToolCallError,
   ToolJsonError,

package/dist/{tool-Bwj_zoTx.d.ts → tool-BLypI31e.d.ts}

@@ -19,6 +19,21 @@ type ToolApprovalPolicy<Args = unknown> = {
     reason?: string | ((ctx: ToolApprovalContext<Args>) => string | Promise<string>);
     rejectMessage?: string | ((ctx: ToolApprovalContext<Args>) => string | Promise<string>);
 };
+type ToolApprovalRequest<Args = unknown> = ToolApprovalContext<Args> & {
+    reason?: string;
+    rejectMessage?: string;
+};
+type ToolApprovalDecision = boolean | {
+    approved: true;
+    reason?: string;
+} | {
+    approved: false;
+    reason?: string;
+    rejectMessage?: string;
+};
+type ToolApprovalsOptions = {
+    handler(request: ToolApprovalRequest): ToolApprovalDecision | Promise<ToolApprovalDecision>;
+};
 type ToolCallStreamEvent = {
     agentId: string;
     agentName?: string | undefined;
@@ -46,4 +61,4 @@ declare function normalizeToolResultOutput(output: unknown): NormalizedToolOutpu
 declare function toolResultContentToText(content: ToolResultContent[]): string;
 declare function parseToolArgs(args: string): JsonValue;
 
-export { type AnyTool as A, type NormalizedToolOutput as N, type Tool as T, type ToolApprovalContext as a, type ToolApprovalPolicy as b, type ToolCallContext as c, type ToolCallStreamEvent as d, type ToolApprovalRunContext as e, ToolOutput as f, normalizeToolResultOutput as n, parseToolArgs as p, toolResultContentToText as t };
+export { type AnyTool as A, type NormalizedToolOutput as N, type Tool as T, type ToolApprovalContext as a, type ToolApprovalDecision as b, type ToolApprovalPolicy as c, type ToolApprovalRequest as d, type ToolApprovalsOptions as e, type ToolCallContext as f, type ToolCallStreamEvent as g, type ToolApprovalRunContext as h, ToolOutput as i, normalizeToolResultOutput as n, parseToolArgs as p, toolResultContentToText as t };

package/dist/{types-BLyF-Tub.d.ts → types-CIwm9p4w.d.ts}

@@ -2,7 +2,7 @@ import { SSEClientTransportOptions } from '@modelcontextprotocol/sdk/client/sse.
 import { StdioServerParameters } from '@modelcontextprotocol/sdk/client/stdio.js';
 import { StreamableHTTPClientTransportOptions } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
 import { J as JsonObject } from './types-C8loR4il.js';
-import { A as AnyTool } from './tool-Bwj_zoTx.js';
+import { A as AnyTool } from './tool-BLypI31e.js';
 
 type McpToolDefinition = {
     name: string;

package/dist/{types-BfEFm6zu.d.ts → types-D-YTJdLj.d.ts}

@@ -1,4 +1,4 @@
-import { A as AnyTool } from './tool-Bwj_zoTx.js';
+import { A as AnyTool } from './tool-BLypI31e.js';
 
 type Skill = {
     readonly name: string;

package/dist/vector-store/index.d.ts

@@ -1,5 +1,5 @@
 import { d as VectorMetadataValue, V as VectorMetadata, E as EmbeddingModel, b as EmbeddedDocument, c as Embedding } from '../types-IB2e9u5M.js';
-import { T as Tool } from '../tool-Bwj_zoTx.js';
+import { T as Tool } from '../tool-BLypI31e.js';
 import '../types-C8loR4il.js';
 
 type VectorFilter = {

package/dist/vector-store/index.js

@@ -3,11 +3,11 @@ import {
   InMemoryVectorStore,
   createVectorSearchTool,
   vectorFilter
-} from "../chunk-66IJDP6L.js";
+} from "../chunk-VQE5PAAT.js";
 import "../chunk-MMHG7WAM.js";
-import "../chunk-WQKHFADH.js";
 import "../chunk-FTO3W4UP.js";
 import "../chunk-OIMLU4SF.js";
+import "../chunk-WQKHFADH.js";
 export {
   InMemoryVectorIndex,
   InMemoryVectorStore,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@anvia/core",
-  "version": "0.6.3",
+  "version": "0.7.1",
   "description": "Core runtime primitives for context-aware Anvia agents.",
   "author": "anvia",
   "maintainer": "Indra Zulfi",

package/dist/chunk-5OGAUWM5.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/tool/errors.ts","../src/tool/tool.ts","../src/tool/tool-set.ts","../src/tool/dynamic-tools.ts","../src/tool/middleware.ts","../src/tool/think-tool.ts"],"sourcesContent":["export class ToolCallError extends Error {\n  constructor(\n    message: string,\n    readonly cause?: unknown,\n  ) {\n    super(message);\n    this.name = \"ToolCallError\";\n  }\n}\n\nexport class ToolNotFoundError extends Error {\n  constructor(readonly toolName: string) {\n    super(`Tool not found: ${toolName}`);\n    this.name = \"ToolNotFoundError\";\n  }\n}\n\nexport class ToolJsonError extends Error {\n  constructor(\n    message: string,\n    readonly cause?: unknown,\n  ) {\n    super(message);\n    this.name = \"ToolJsonError\";\n  }\n}\n","import type { JsonObject, JsonValue, ToolDefinition, ToolResultContent } from \"../completion/types\";\nimport {\n  isToolResultContentArray,\n  serializeToolResultOutput as serializeToolOutput,\n} from \"../completion/types\";\n\nexport type ToolApprovalRunContext = {\n  agentId: string;\n  runId: string;\n  sessionId?: string;\n  metadata?: JsonObject;\n};\n\nexport type ToolApprovalContext<Args = unknown> = {\n  toolName: string;\n  args: Args;\n  rawArgs: string;\n  toolCallId?: string;\n  internalCallId: string;\n  run: ToolApprovalRunContext;\n};\n\nexport type ToolApprovalPolicy<Args = unknown> = {\n  when(ctx: ToolApprovalContext<Args>): boolean | Promise<boolean>;\n  reason?: string | ((ctx: ToolApprovalContext<Args>) => string | Promise<string>);\n  rejectMessage?: string | ((ctx: ToolApprovalContext<Args>) => string | Promise<string>);\n};\n\nexport type ToolCallStreamEvent = {\n  agentId: string;\n  agentName?: string | undefined;\n  event: unknown;\n};\n\nexport type ToolCallContext = {\n  emitStreamEvent?(event: ToolCallStreamEvent): void | Promise<void>;\n};\n\nexport interface Tool<Args = unknown, Output = unknown> {\n  readonly name: string;\n  readonly approval?: ToolApprovalPolicy<Args>;\n  definition(prompt: string): ToolDefinition | Promise<ToolDefinition>;\n  call(args: Args, context?: ToolCallContext): Output | Promise<Output>;\n  parseApprovalArgs?(args: unknown): Args;\n}\n\nexport type AnyTool = Omit<Tool<unknown, unknown>, \"approval\"> & {\n  readonly approval?: unknown;\n};\n\nexport type NormalizedToolOutput = string | ToolResultContent[];\n\nexport const ToolOutput = {\n  content(content: ToolResultContent[]): ToolResultContent[] {\n    return content;\n  },\n};\n\nexport { isToolResultContentArray, serializeToolOutput };\n\nexport function normalizeToolResultOutput(output: unknown): NormalizedToolOutput {\n  return isToolResultContentArray(output) ? output : serializeToolOutput(output);\n}\n\nexport function toolResultContentToText(content: ToolResultContent[]): string {\n  return content\n    .map((item) => (item.type === \"text\" ? item.text : `[image:${item.mediaType ?? \"image/png\"}]`))\n    .join(\"\\n\");\n}\n\nexport function parseToolArgs(args: string): JsonValue {\n  if (args.trim() === \"\") {\n    return {};\n  }\n\n  return JSON.parse(args) as JsonValue;\n}\n","import type { ToolDefinition } from \"../completion/types\";\nimport { ToolCallError, ToolJsonError, ToolNotFoundError } from \"./errors\";\nimport {\n  type AnyTool,\n  type NormalizedToolOutput,\n  normalizeToolResultOutput,\n  parseToolArgs,\n  type ToolCallContext,\n} from \"./tool\";\n\nexport class ToolSet {\n  private readonly tools = new Map<string, AnyTool>();\n\n  static fromTools(tools: AnyTool[]): ToolSet {\n    const toolSet = new ToolSet();\n    for (const tool of tools) {\n      toolSet.addTool(tool);\n    }\n    return toolSet;\n  }\n\n  addTool(tool: AnyTool): this {\n    this.tools.set(tool.name, tool);\n    return this;\n  }\n\n  addTools(tools: AnyTool[] | ToolSet): this {\n    const values = Array.isArray(tools) ? tools : tools.values();\n    for (const tool of values) {\n      this.addTool(tool);\n    }\n    return this;\n  }\n\n  deleteTool(toolName: string): boolean {\n    return this.tools.delete(toolName);\n  }\n\n  contains(toolName: string): boolean {\n    return this.tools.has(toolName);\n  }\n\n  get(toolName: string): AnyTool | undefined {\n    return this.tools.get(toolName);\n  }\n\n  values(): AnyTool[] {\n    return [...this.tools.values()];\n  }\n\n  async getToolDefinitions(prompt = \"\"): Promise<ToolDefinition[]> {\n    const defs: ToolDefinition[] = [];\n    for (const tool of this.tools.values()) {\n      defs.push(await tool.definition(prompt));\n    }\n    return defs;\n  }\n\n  async call(\n    toolName: string,\n    args: string,\n    context?: ToolCallContext,\n  ): Promise<NormalizedToolOutput> {\n    const tool = this.tools.get(toolName);\n    if (tool === undefined) {\n      throw new ToolNotFoundError(toolName);\n    }\n\n    let parsedArgs: unknown;\n    try {\n      parsedArgs = parseToolArgs(args);\n    } catch (error) {\n      throw new ToolJsonError(`Invalid JSON arguments for tool ${toolName}`, error);\n    }\n\n    try {\n      const output = await tool.call(parsedArgs, context);\n      return normalizeToolResultOutput(output);\n    } catch (error) {\n      if (error instanceof Error) {\n        throw new ToolCallError(error.message, error);\n      }\n      throw new ToolCallError(`Tool ${toolName} failed`, error);\n    }\n  }\n}\n","import type { ToolDefinition } from \"../completion\";\nimport type { EmbeddedDocument, EmbeddingModel, VectorMetadata } from \"../embeddings\";\nimport { embedDocuments } from \"../embeddings\";\nimport { compact } from \"../internal/compact\";\nimport type {\n  VectorInspectPage,\n  VectorInspectRequest,\n  VectorSearchResult,\n  VectorSearchToolOptions,\n} from \"../vector-store\";\nimport {\n  InMemoryVectorStore,\n  type VectorSearchIndex,\n  type VectorSearchRequest,\n} from \"../vector-store\";\nimport type { AnyTool, Tool } from \"./tool\";\nimport { ToolSet } from \"./tool-set\";\n\nexport type ToolSearchDocument<Metadata extends VectorMetadata = VectorMetadata> = {\n  toolName: string;\n  definition: ToolDefinition;\n  text: string;\n  metadata?: Metadata | undefined;\n};\n\nexport type EmbedToolsOptions<Metadata extends VectorMetadata = VectorMetadata> = {\n  content?: ((tool: AnyTool, definition: ToolDefinition) => string | string[]) | undefined;\n  metadata?: ((tool: AnyTool, definition: ToolDefinition) => Metadata | undefined) | undefined;\n  concurrency?: number | undefined;\n};\n\nexport interface DynamicToolIndex<Metadata extends VectorMetadata = VectorMetadata>\n  extends VectorSearchIndex<ToolSearchDocument<Metadata>, Metadata> {\n  readonly toolSet: ToolSet;\n}\n\nexport async function embedTools<Metadata extends VectorMetadata = VectorMetadata>(\n  model: EmbeddingModel,\n  tools: AnyTool[] | ToolSet,\n  options: EmbedToolsOptions<Metadata> = {},\n): Promise<Array<EmbeddedDocument<ToolSearchDocument<Metadata>, Metadata>>> {\n  const toolList = Array.isArray(tools) ? tools : tools.values();\n  const definitions = await Promise.all(\n    toolList.map(async (tool) => ({ tool, definition: await tool.definition(\"\") })),\n  );\n  const documents = definitions.map(({ tool, definition }) => {\n    const content = options.content?.(tool, definition) ?? defaultToolEmbeddingText(definition);\n    const texts = Array.isArray(content) ? content : [content];\n    const metadata = options.metadata?.(tool, definition);\n    const document: ToolSearchDocument<Metadata> = compact({\n      toolName: tool.name,\n      definition,\n      text: texts.join(\"\\n\"),\n      metadata,\n    }) as ToolSearchDocument<Metadata>;\n    return { tool, document, texts, metadata };\n  });\n\n  return embedDocuments(model, documents, {\n    id: (item) => item.tool.name,\n    content: (item) => item.texts,\n    metadata: (item) => item.metadata,\n    concurrency: options.concurrency,\n  }).then((embedded) =>\n    embedded.map((item) => ({\n      ...item,\n      document: item.document.document,\n    })),\n  );\n}\n\nexport async function createToolIndex<Metadata extends VectorMetadata = VectorMetadata>(\n  model: EmbeddingModel,\n  tools: AnyTool[] | ToolSet,\n  options: EmbedToolsOptions<Metadata> = {},\n): Promise<DynamicToolIndex<Metadata>> {\n  const toolSet = Array.isArray(tools) ? ToolSet.fromTools(tools) : tools;\n  const embedded = await embedTools(model, toolSet, options);\n  const index = InMemoryVectorStore.fromDocuments(embedded).index(model);\n  return new DynamicToolSearchIndex(index, toolSet);\n}\n\nexport function isDynamicToolIndex(value: unknown): value is DynamicToolIndex {\n  return (\n    typeof value === \"object\" &&\n    value !== null &&\n    \"toolSet\" in value &&\n    (value as { toolSet?: unknown }).toolSet instanceof ToolSet\n  );\n}\n\nclass DynamicToolSearchIndex<Metadata extends VectorMetadata>\n  implements DynamicToolIndex<Metadata>\n{\n  readonly inspect?: (\n    request: VectorInspectRequest,\n  ) => Promise<VectorInspectPage<ToolSearchDocument<Metadata>, Metadata>>;\n\n  constructor(\n    private readonly index: VectorSearchIndex<ToolSearchDocument<Metadata>, Metadata>,\n    readonly toolSet: ToolSet,\n  ) {\n    if (index.inspect !== undefined) {\n      this.inspect = (request) =>\n        index.inspect?.(request) as Promise<\n          VectorInspectPage<ToolSearchDocument<Metadata>, Metadata>\n        >;\n    }\n  }\n\n  search(\n    request: VectorSearchRequest,\n  ): Promise<Array<VectorSearchResult<ToolSearchDocument<Metadata>, Metadata>>> {\n    return this.index.search(request);\n  }\n\n  searchIds(request: VectorSearchRequest): Promise<Array<{ score: number; id: string }>> {\n    return this.index.searchIds(request);\n  }\n\n  asTool(options: VectorSearchToolOptions): Tool<{ query: string; topK?: number }, unknown> {\n    return this.index.asTool(options);\n  }\n}\n\nfunction defaultToolEmbeddingText(definition: ToolDefinition): string[] {\n  return [definition.name, definition.description, JSON.stringify(definition.parameters)];\n}\n","import type {\n  CompletionRequest,\n  CompletionResponse,\n  JsonValue,\n  ToolResultContent,\n} from \"../completion\";\n\ntype MaybePromise<T> = T | Promise<T>;\n\nexport type CompletionRequestMiddlewareArgs = {\n  turn: number;\n  request: CompletionRequest;\n  originalRequest: CompletionRequest;\n};\n\nexport type CompletionRequestMiddlewareResult =\n  | {\n      request: CompletionRequest;\n    }\n  | undefined;\n\nexport type CompletionResponseMiddlewareArgs<RawResponse = unknown> = {\n  turn: number;\n  request: CompletionRequest;\n  response: CompletionResponse<RawResponse>;\n  originalResponse: CompletionResponse<RawResponse>;\n};\n\nexport type CompletionResponseMiddlewareResult<RawResponse = unknown> =\n  | {\n      response: CompletionResponse<RawResponse>;\n    }\n  | undefined;\n\nexport type ToolInputMiddlewareArgs = {\n  toolName: string;\n  args: string;\n  originalArgs: string;\n  turn: number;\n  toolCallId?: string | undefined;\n  internalCallId: string;\n};\n\nexport type ToolInputMiddlewareResult =\n  | {\n      args: JsonValue | string;\n    }\n  | undefined;\n\nexport type ToolResultMiddlewareArgs = {\n  toolName: string;\n  args: string;\n  result: string;\n  originalResult: string;\n  structuredResult?: ToolResultContent[] | undefined;\n  originalStructuredResult?: ToolResultContent[] | undefined;\n  turn: number;\n  toolCallId?: string | undefined;\n  internalCallId: string;\n};\n\nexport type ToolOutputMiddlewareArgs = ToolResultMiddlewareArgs;\n\nexport type ToolOutputMiddlewareResult =\n  | string\n  | {\n      result?: string | undefined;\n      structuredResult?: ToolResultContent[] | undefined;\n    }\n  | undefined;\n\nexport interface AgentMiddleware<RawResponse = unknown> {\n  onCompletionRequest?(\n    args: CompletionRequestMiddlewareArgs,\n  ): MaybePromise<CompletionRequestMiddlewareResult>;\n  onCompletionResponse?(\n    args: CompletionResponseMiddlewareArgs<RawResponse>,\n  ): MaybePromise<CompletionResponseMiddlewareResult<RawResponse>>;\n  onToolInput?(args: ToolInputMiddlewareArgs): MaybePromise<ToolInputMiddlewareResult>;\n  onToolOutput?(args: ToolOutputMiddlewareArgs): MaybePromise<ToolOutputMiddlewareResult>;\n  /**\n   * @deprecated Use `onToolOutput` instead.\n   */\n  onResult?(args: ToolResultMiddlewareArgs): string | undefined | Promise<string | undefined>;\n}\n\n/**\n * @deprecated Use `AgentMiddleware` instead.\n */\nexport type ToolMiddleware<RawResponse = unknown> = AgentMiddleware<RawResponse>;\n\nexport function createMiddleware<RawResponse = unknown>(\n  middleware: AgentMiddleware<RawResponse>,\n): AgentMiddleware<RawResponse> {\n  return middleware;\n}\n\n/**\n * @deprecated Use `createMiddleware` instead.\n */\nexport function createToolMiddleware<RawResponse = unknown>(\n  middleware: ToolMiddleware<RawResponse>,\n): ToolMiddleware<RawResponse> {\n  return middleware;\n}\n","import { z } from \"zod\";\nimport { createTool } from \"./create-tool\";\n\nconst defaultThinkToolDescription =\n  \"Use this tool to record a thought while reasoning through a complex task. It does not retrieve information, store memory, or change external state.\";\n\nconst thinkToolInput = z.object({\n  thought: z.string().describe(\"A thought to record while reasoning through a task.\"),\n});\n\nexport type CreateThinkToolOptions = {\n  name?: string;\n  description?: string;\n};\n\nexport function createThinkTool(options: CreateThinkToolOptions = {}) {\n  return createTool({\n    name: options.name ?? \"think\",\n    description: options.description ?? defaultThinkToolDescription,\n    input: thinkToolInput,\n    output: z.string(),\n    execute: (args) => args.thought,\n  });\n}\n"],"mappings":";;;;;;;;;;;;;;;;AAAO,IAAM,gBAAN,cAA4B,MAAM;AAAA,EACvC,YACE,SACS,OACT;AACA,UAAM,OAAO;AAFJ;AAGT,SAAK,OAAO;AAAA,EACd;AAAA,EAJW;AAKb;AAEO,IAAM,oBAAN,cAAgC,MAAM;AAAA,EAC3C,YAAqB,UAAkB;AACrC,UAAM,mBAAmB,QAAQ,EAAE;AADhB;AAEnB,SAAK,OAAO;AAAA,EACd;AAAA,EAHqB;AAIvB;AAEO,IAAM,gBAAN,cAA4B,MAAM;AAAA,EACvC,YACE,SACS,OACT;AACA,UAAM,OAAO;AAFJ;AAGT,SAAK,OAAO;AAAA,EACd;AAAA,EAJW;AAKb;;;AC2BO,IAAM,aAAa;AAAA,EACxB,QAAQ,SAAmD;AACzD,WAAO;AAAA,EACT;AACF;AAIO,SAAS,0BAA0B,QAAuC;AAC/E,SAAO,yBAAyB,MAAM,IAAI,SAAS,0BAAoB,MAAM;AAC/E;AAEO,SAAS,wBAAwB,SAAsC;AAC5E,SAAO,QACJ,IAAI,CAAC,SAAU,KAAK,SAAS,SAAS,KAAK,OAAO,UAAU,KAAK,aAAa,WAAW,GAAI,EAC7F,KAAK,IAAI;AACd;AAEO,SAAS,cAAc,MAAyB;AACrD,MAAI,KAAK,KAAK,MAAM,IAAI;AACtB,WAAO,CAAC;AAAA,EACV;AAEA,SAAO,KAAK,MAAM,IAAI;AACxB;;;AClEO,IAAM,UAAN,MAAM,SAAQ;AAAA,EACF,QAAQ,oBAAI,IAAqB;AAAA,EAElD,OAAO,UAAU,OAA2B;AAC1C,UAAM,UAAU,IAAI,SAAQ;AAC5B,eAAW,QAAQ,OAAO;AACxB,cAAQ,QAAQ,IAAI;AAAA,IACtB;AACA,WAAO;AAAA,EACT;AAAA,EAEA,QAAQ,MAAqB;AAC3B,SAAK,MAAM,IAAI,KAAK,MAAM,IAAI;AAC9B,WAAO;AAAA,EACT;AAAA,EAEA,SAAS,OAAkC;AACzC,UAAM,SAAS,MAAM,QAAQ,KAAK,IAAI,QAAQ,MAAM,OAAO;AAC3D,eAAW,QAAQ,QAAQ;AACzB,WAAK,QAAQ,IAAI;AAAA,IACnB;AACA,WAAO;AAAA,EACT;AAAA,EAEA,WAAW,UAA2B;AACpC,WAAO,KAAK,MAAM,OAAO,QAAQ;AAAA,EACnC;AAAA,EAEA,SAAS,UAA2B;AAClC,WAAO,KAAK,MAAM,IAAI,QAAQ;AAAA,EAChC;AAAA,EAEA,IAAI,UAAuC;AACzC,WAAO,KAAK,MAAM,IAAI,QAAQ;AAAA,EAChC;AAAA,EAEA,SAAoB;AAClB,WAAO,CAAC,GAAG,KAAK,MAAM,OAAO,CAAC;AAAA,EAChC;AAAA,EAEA,MAAM,mBAAmB,SAAS,IAA+B;AAC/D,UAAM,OAAyB,CAAC;AAChC,eAAW,QAAQ,KAAK,MAAM,OAAO,GAAG;AACtC,WAAK,KAAK,MAAM,KAAK,WAAW,MAAM,CAAC;AAAA,IACzC;AACA,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,KACJ,UACA,MACA,SAC+B;AAC/B,UAAM,OAAO,KAAK,MAAM,IAAI,QAAQ;AACpC,QAAI,SAAS,QAAW;AACtB,YAAM,IAAI,kBAAkB,QAAQ;AAAA,IACtC;AAEA,QAAI;AACJ,QAAI;AACF,mBAAa,cAAc,IAAI;AAAA,IACjC,SAAS,OAAO;AACd,YAAM,IAAI,cAAc,mCAAmC,QAAQ,IAAI,KAAK;AAAA,IAC9E;AAEA,QAAI;AACF,YAAM,SAAS,MAAM,KAAK,KAAK,YAAY,OAAO;AAClD,aAAO,0BAA0B,MAAM;AAAA,IACzC,SAAS,OAAO;AACd,UAAI,iBAAiB,OAAO;AAC1B,cAAM,IAAI,cAAc,MAAM,SAAS,KAAK;AAAA,MAC9C;AACA,YAAM,IAAI,cAAc,QAAQ,QAAQ,WAAW,KAAK;AAAA,IAC1D;AAAA,EACF;AACF;;;ACjDA,eAAsB,WACpB,OACA,OACA,UAAuC,CAAC,GACkC;AAC1E,QAAM,WAAW,MAAM,QAAQ,KAAK,IAAI,QAAQ,MAAM,OAAO;AAC7D,QAAM,cAAc,MAAM,QAAQ;AAAA,IAChC,SAAS,IAAI,OAAO,UAAU,EAAE,MAAM,YAAY,MAAM,KAAK,WAAW,EAAE,EAAE,EAAE;AAAA,EAChF;AACA,QAAM,YAAY,YAAY,IAAI,CAAC,EAAE,MAAM,WAAW,MAAM;AAC1D,UAAM,UAAU,QAAQ,UAAU,MAAM,UAAU,KAAK,yBAAyB,UAAU;AAC1F,UAAM,QAAQ,MAAM,QAAQ,OAAO,IAAI,UAAU,CAAC,OAAO;AACzD,UAAM,WAAW,QAAQ,WAAW,MAAM,UAAU;AACpD,UAAM,WAAyC,QAAQ;AAAA,MACrD,UAAU,KAAK;AAAA,MACf;AAAA,MACA,MAAM,MAAM,KAAK,IAAI;AAAA,MACrB;AAAA,IACF,CAAC;AACD,WAAO,EAAE,MAAM,UAAU,OAAO,SAAS;AAAA,EAC3C,CAAC;AAED,SAAO,eAAe,OAAO,WAAW;AAAA,IACtC,IAAI,CAAC,SAAS,KAAK,KAAK;AAAA,IACxB,SAAS,CAAC,SAAS,KAAK;AAAA,IACxB,UAAU,CAAC,SAAS,KAAK;AAAA,IACzB,aAAa,QAAQ;AAAA,EACvB,CAAC,EAAE;AAAA,IAAK,CAAC,aACP,SAAS,IAAI,CAAC,UAAU;AAAA,MACtB,GAAG;AAAA,MACH,UAAU,KAAK,SAAS;AAAA,IAC1B,EAAE;AAAA,EACJ;AACF;AAEA,eAAsB,gBACpB,OACA,OACA,UAAuC,CAAC,GACH;AACrC,QAAM,UAAU,MAAM,QAAQ,KAAK,IAAI,QAAQ,UAAU,KAAK,IAAI;AAClE,QAAM,WAAW,MAAM,WAAW,OAAO,SAAS,OAAO;AACzD,QAAM,QAAQ,oBAAoB,cAAc,QAAQ,EAAE,MAAM,KAAK;AACrE,SAAO,IAAI,uBAAuB,OAAO,OAAO;AAClD;AAEO,SAAS,mBAAmB,OAA2C;AAC5E,SACE,OAAO,UAAU,YACjB,UAAU,QACV,aAAa,SACZ,MAAgC,mBAAmB;AAExD;AAEA,IAAM,yBAAN,MAEA;AAAA,EAKE,YACmB,OACR,SACT;AAFiB;AACR;AAET,QAAI,MAAM,YAAY,QAAW;AAC/B,WAAK,UAAU,CAAC,YACd,MAAM,UAAU,OAAO;AAAA,IAG3B;AAAA,EACF;AAAA,EATmB;AAAA,EACR;AAAA,EANF;AAAA,EAgBT,OACE,SAC4E;AAC5E,WAAO,KAAK,MAAM,OAAO,OAAO;AAAA,EAClC;AAAA,EAEA,UAAU,SAA6E;AACrF,WAAO,KAAK,MAAM,UAAU,OAAO;AAAA,EACrC;AAAA,EAEA,OAAO,SAAmF;AACxF,WAAO,KAAK,MAAM,OAAO,OAAO;AAAA,EAClC;AACF;AAEA,SAAS,yBAAyB,YAAsC;AACtE,SAAO,CAAC,WAAW,MAAM,WAAW,aAAa,KAAK,UAAU,WAAW,UAAU,CAAC;AACxF;;;ACpCO,SAAS,iBACd,YAC8B;AAC9B,SAAO;AACT;AAKO,SAAS,qBACd,YAC6B;AAC7B,SAAO;AACT;;;ACxGA,SAAS,SAAS;AAGlB,IAAM,8BACJ;AAEF,IAAM,iBAAiB,EAAE,OAAO;AAAA,EAC9B,SAAS,EAAE,OAAO,EAAE,SAAS,qDAAqD;AACpF,CAAC;AAOM,SAAS,gBAAgB,UAAkC,CAAC,GAAG;AACpE,SAAO,WAAW;AAAA,IAChB,MAAM,QAAQ,QAAQ;AAAA,IACtB,aAAa,QAAQ,eAAe;AAAA,IACpC,OAAO;AAAA,IACP,QAAQ,EAAE,OAAO;AAAA,IACjB,SAAS,CAAC,SAAS,KAAK;AAAA,EAC1B,CAAC;AACH;","names":[]}