@anvia/core 0.3.1 → 0.4.1

package/dist/pipeline/index.js

@@ -1,7 +1,338 @@
 import {
-  Pipeline,
-  PipelineBuilder
-} from "../chunk-RTOGIJH2.js";
+  mapWithConcurrency
+} from "../chunk-OIMLU4SF.js";
+
+// src/pipeline/graph.ts
+function initialBuilderState(metadata) {
+  return {
+    graph: initialGraph(metadata),
+    terminalNodeId: "input",
+    terminalNodeIds: ["input"],
+    nextNodeIndex: 1,
+    nextEdgeIndex: 1
+  };
+}
+function initialGraph(metadata) {
+  const id = normalizeId(metadata.id ?? "pipeline");
+  return {
+    id,
+    ...metadata.name === void 0 ? {} : { name: metadata.name },
+    ...metadata.description === void 0 ? {} : { description: metadata.description },
+    ...metadata.metadata === void 0 ? {} : { metadata: metadata.metadata },
+    nodes: [{ id: "input", kind: "input", label: "Input" }],
+    edges: []
+  };
+}
+function appendNode(state, kind, label, options = {}) {
+  const node = graphNode(kind, label, state.nextNodeIndex, {
+    ...options,
+    existingIds: new Set(state.graph.nodes.map((item) => item.id))
+  });
+  return {
+    node,
+    state: appendGraphNode(state, node, activeTerminalNodeIds(state), [node.id])
+  };
+}
+function appendChildNode(state, parentId, kind, label, options = {}) {
+  const node = graphNode(kind, label, state.nextNodeIndex, {
+    ...options,
+    existingIds: new Set(state.graph.nodes.map((item) => item.id))
+  });
+  return {
+    node,
+    state: appendGraphNode(state, node, [parentId], activeTerminalNodeIds(state))
+  };
+}
+function activeTerminalNodeIds(state) {
+  return state.terminalNodeIds.length > 0 ? state.terminalNodeIds : [state.terminalNodeId];
+}
+function withTerminalNodes(state, terminalNodeIds) {
+  return {
+    ...state,
+    terminalNodeId: terminalNodeIds.at(-1) ?? state.terminalNodeId,
+    terminalNodeIds
+  };
+}
+function withOutputNode(state) {
+  const graph = cloneGraph(state.graph);
+  if (graph.nodes.some((node) => node.id === "output")) {
+    return graph;
+  }
+  graph.nodes.push({ id: "output", kind: "output", label: "Output" });
+  graph.edges.push(
+    ...activeTerminalNodeIds(state).map((sourceId, index) => ({
+      id: `edge_${state.nextEdgeIndex + index}`,
+      source: sourceId,
+      target: "output"
+    }))
+  );
+  return graph;
+}
+function nextStageLabel(state, prefix) {
+  return `${prefix} ${state.nextNodeIndex}`;
+}
+function cloneGraph(graph) {
+  return {
+    ...graph,
+    nodes: graph.nodes.map((node) => ({ ...node })),
+    edges: graph.edges.map((edge) => ({ ...edge }))
+  };
+}
+function appendGraphNode(state, node, sourceIds, terminalNodeIds) {
+  const edges = sourceIds.map((sourceId, index) => ({
+    id: `edge_${state.nextEdgeIndex + index}`,
+    source: sourceId,
+    target: node.id
+  }));
+  const terminalNodeId = terminalNodeIds.at(-1) ?? state.terminalNodeId;
+  return {
+    graph: {
+      ...state.graph,
+      nodes: [...state.graph.nodes, node],
+      edges: [...state.graph.edges, ...edges]
+    },
+    terminalNodeId,
+    terminalNodeIds,
+    nextNodeIndex: state.nextNodeIndex + 1,
+    nextEdgeIndex: state.nextEdgeIndex + edges.length
+  };
+}
+function graphNode(kind, label, index, options = {}) {
+  const id = uniqueGraphNodeId(
+    normalizeId(options.preferredId ?? `${kind}_${index}`),
+    options.existingIds ?? /* @__PURE__ */ new Set()
+  );
+  return {
+    id,
+    kind,
+    label,
+    ...options.description === void 0 ? {} : { description: options.description },
+    ...options.metadata === void 0 ? {} : { metadata: options.metadata },
+    ...options.agentId === void 0 ? {} : { agentId: options.agentId },
+    ...options.agentName === void 0 ? {} : { agentName: options.agentName },
+    ...options.pipelineId === void 0 ? {} : { pipelineId: options.pipelineId },
+    ...options.branchKey === void 0 ? {} : { branchKey: options.branchKey }
+  };
+}
+function normalizeId(value) {
+  const normalized = value.trim().toLowerCase().replace(/[^a-z0-9_-]+/g, "_").replace(/^_+|_+$/g, "");
+  return normalized.length === 0 ? "pipeline" : normalized;
+}
+function uniqueGraphNodeId(baseId, existingIds) {
+  let id = baseId;
+  let suffix = 2;
+  while (existingIds.has(id)) {
+    id = `${baseId}_${suffix}`;
+    suffix += 1;
+  }
+  return id;
+}
+
+// src/pipeline/pipeline.ts
+var Pipeline = class {
+  constructor(executor, pipelineGraph = initialGraph({})) {
+    this.executor = executor;
+    this.pipelineGraph = pipelineGraph;
+    this.id = pipelineGraph.id;
+    this.name = pipelineGraph.name;
+    this.description = pipelineGraph.description;
+    this.metadata = pipelineGraph.metadata;
+  }
+  executor;
+  pipelineGraph;
+  id;
+  name;
+  description;
+  metadata;
+  /** Run one input through the built pipeline and return the final stage output. */
+  async run(input, options = {}) {
+    return await this.executor(input, { observer: options.observer });
+  }
+  /** Run many inputs through the same pipeline with bounded concurrency. */
+  async batch(inputs, options) {
+    return mapWithConcurrency([...inputs], options.concurrency, (input) => this.run(input));
+  }
+  graph() {
+    return cloneGraph(this.pipelineGraph);
+  }
+};
+
+// src/pipeline/runtime.ts
+async function runNode(context, node, fn) {
+  const startedAt = Date.now();
+  await context.observer?.onEvent({ type: "stage_started", node });
+  try {
+    const output = await fn();
+    await context.observer?.onEvent({
+      type: "stage_completed",
+      node,
+      durationMs: Date.now() - startedAt
+    });
+    return output;
+  } catch (error) {
+    await context.observer?.onEvent({
+      type: "stage_failed",
+      node,
+      durationMs: Date.now() - startedAt,
+      error
+    });
+    throw error;
+  }
+}
+
+// src/pipeline/builder.ts
+var PipelineBuilder = class _PipelineBuilder {
+  executor;
+  state;
+  constructor(metadataOrExecutor, state) {
+    if (state !== void 0) {
+      this.executor = metadataOrExecutor;
+      this.state = state;
+      return;
+    }
+    if (typeof metadataOrExecutor === "function") {
+      const executor = metadataOrExecutor;
+      this.executor = ((input) => executor(input));
+      this.state = initialBuilderState({});
+      return;
+    }
+    this.executor = identity;
+    this.state = initialBuilderState(metadataOrExecutor ?? {});
+  }
+  /** Add a synchronous or asynchronous transform stage. */
+  step(fn, metadata) {
+    const next = appendNode(
+      this.state,
+      "step",
+      metadata?.name ?? nextStageLabel(this.state, "Step"),
+      {
+        description: metadata?.description,
+        metadata: metadata?.metadata,
+        preferredId: metadata?.id
+      }
+    );
+    return new _PipelineBuilder(
+      async (input, context) => {
+        const value = await this.runStep(input, context);
+        const result = await runNode(context, next.node, () => fn(value));
+        return result;
+      },
+      next.state
+    );
+  }
+  /** Compose another pipeline operation after the current stage. */
+  use(op, metadata) {
+    const nested = op instanceof Pipeline ? op : void 0;
+    const next = appendNode(
+      this.state,
+      nested === void 0 ? "step" : "pipeline",
+      metadata?.name ?? nested?.name ?? nested?.id ?? nextStageLabel(this.state, nested === void 0 ? "Operation" : "Pipeline"),
+      {
+        description: metadata?.description ?? nested?.description,
+        metadata: metadata?.metadata ?? nested?.metadata,
+        preferredId: metadata?.id,
+        pipelineId: nested?.id
+      }
+    );
+    return new _PipelineBuilder(
+      async (input, context) => {
+        const value = await this.runStep(input, context);
+        const result = await runNode(context, next.node, () => op.run(value));
+        return result;
+      },
+      next.state
+    );
+  }
+  /** Run named branch operations concurrently from the current value. */
+  parallel(branches, metadata) {
+    const parallel = appendNode(
+      this.state,
+      "parallel",
+      metadata?.name ?? `${Object.keys(branches).length} parallel branches`,
+      {
+        description: metadata?.description,
+        metadata: metadata?.metadata,
+        preferredId: metadata?.id
+      }
+    );
+    let nextState = parallel.state;
+    const branchNodes = {};
+    for (const key of Object.keys(branches)) {
+      const branch = appendChildNode(nextState, parallel.node.id, "branch", key, {
+        branchKey: key
+      });
+      nextState = branch.state;
+      branchNodes[key] = branch.node;
+    }
+    nextState = withTerminalNodes(
+      nextState,
+      Object.values(branchNodes).map((node) => node.id)
+    );
+    return new _PipelineBuilder(async (input, context) => {
+      const value = await this.runStep(input, context);
+      const entries = await runNode(
+        context,
+        parallel.node,
+        () => Promise.all(
+          Object.entries(branches).map(async ([key, op]) => {
+            const node = branchNodes[key];
+            const output = await runNode(context, node, () => op.run(value));
+            return [key, output];
+          })
+        )
+      );
+      return Object.fromEntries(entries);
+    }, nextState);
+  }
+  /** Send the current value to an agent as text and continue with the agent output. */
+  prompt(agent, metadata) {
+    const next = appendNode(this.state, "agent", metadata?.name ?? agent.name ?? agent.id, {
+      description: metadata?.description ?? agent.description,
+      metadata: metadata?.metadata,
+      preferredId: metadata?.id,
+      agentId: agent.id,
+      agentName: agent.name
+    });
+    return new _PipelineBuilder(async (input, context) => {
+      const value = await this.runStep(input, context);
+      return runNode(context, next.node, async () => {
+        const response = await agent.prompt(String(value)).send();
+        return response.output;
+      });
+    }, next.state);
+  }
+  /** Send the current value to an extractor as text and continue with typed schema data. */
+  extract(extractor, metadata) {
+    const next = appendNode(
+      this.state,
+      "extractor",
+      metadata?.name ?? nextStageLabel(this.state, "Extractor"),
+      {
+        description: metadata?.description,
+        metadata: metadata?.metadata,
+        preferredId: metadata?.id
+      }
+    );
+    return new _PipelineBuilder(async (input, context) => {
+      const value = await this.runStep(input, context);
+      return runNode(context, next.node, () => extractor.extract(String(value)));
+    }, next.state);
+  }
+  /** Finish the builder and return a runnable pipeline. */
+  build() {
+    const graph = withOutputNode(this.state);
+    return new Pipeline(
+      (input, context) => this.runStep(input, context),
+      graph
+    );
+  }
+  async runStep(input, context) {
+    return await this.executor(input, context);
+  }
+};
+function identity(input) {
+  return input;
+}
 export {
   Pipeline,
   PipelineBuilder

package/dist/pipeline/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}
+{"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 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    id,\n    ...(metadata.name === undefined ? {} : { name: metadata.name }),\n    ...(metadata.description === undefined ? {} : { description: metadata.description }),\n    ...(metadata.metadata === undefined ? {} : { metadata: metadata.metadata }),\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 {\n    id,\n    kind,\n    label,\n    ...(options.description === undefined ? {} : { description: options.description }),\n    ...(options.metadata === undefined ? {} : { metadata: options.metadata }),\n    ...(options.agentId === undefined ? {} : { agentId: options.agentId }),\n    ...(options.agentName === undefined ? {} : { agentName: options.agentName }),\n    ...(options.pipelineId === undefined ? {} : { pipelineId: options.pipelineId }),\n    ...(options.branchKey === undefined ? {} : { branchKey: options.branchKey }),\n  };\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":";;;;;AASO,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;AAAA,IACA,GAAI,SAAS,SAAS,SAAY,CAAC,IAAI,EAAE,MAAM,SAAS,KAAK;AAAA,IAC7D,GAAI,SAAS,gBAAgB,SAAY,CAAC,IAAI,EAAE,aAAa,SAAS,YAAY;AAAA,IAClF,GAAI,SAAS,aAAa,SAAY,CAAC,IAAI,EAAE,UAAU,SAAS,SAAS;AAAA,IACzE,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;AAAA,IACL;AAAA,IACA;AAAA,IACA;AAAA,IACA,GAAI,QAAQ,gBAAgB,SAAY,CAAC,IAAI,EAAE,aAAa,QAAQ,YAAY;AAAA,IAChF,GAAI,QAAQ,aAAa,SAAY,CAAC,IAAI,EAAE,UAAU,QAAQ,SAAS;AAAA,IACvE,GAAI,QAAQ,YAAY,SAAY,CAAC,IAAI,EAAE,SAAS,QAAQ,QAAQ;AAAA,IACpE,GAAI,QAAQ,cAAc,SAAY,CAAC,IAAI,EAAE,WAAW,QAAQ,UAAU;AAAA,IAC1E,GAAI,QAAQ,eAAe,SAAY,CAAC,IAAI,EAAE,YAAY,QAAQ,WAAW;AAAA,IAC7E,GAAI,QAAQ,cAAc,SAAY,CAAC,IAAI,EAAE,WAAW,QAAQ,UAAU;AAAA,EAC5E;AACF;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;;;AClLO,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":[]}

package/dist/skills/index.d.ts

@@ -1,7 +1,7 @@
-import { a as SkillLoader, b as SkillSet } from '../types-B0iXLkqi.js';
-export { S as Skill, c as SkillValidationError, d as SkillValidationIssue } from '../types-B0iXLkqi.js';
-import '../tool-ClZYES-Z.js';
-import '../types-BM8-Y8Hy.js';
+import { b as SkillLoader, a as SkillSet } from '../types-T9rlOIUc.js';
+export { c as Skill, S as SkillValidationError, d as SkillValidationIssue } from '../types-T9rlOIUc.js';
+import '../tool-C3ciF-VG.js';
+import '../types-C54aNoCd.js';
 
 declare function loadSkills(loaders: SkillLoader | SkillLoader[]): Promise<SkillSet>;
 

package/dist/skills/index.js

@@ -2,11 +2,12 @@ import {
   SkillValidationError,
   loadSkills,
   skill
-} from "../chunk-TILE6Z2N.js";
+} from "../chunk-L6JCKFAX.js";
 import "../chunk-YK4WAAS4.js";
-import "../chunk-I2HOMD3R.js";
-import "../chunk-65QV627O.js";
-import "../chunk-A7VDIZQN.js";
+import "../chunk-CY7SB5FG.js";
+import "../chunk-OIA4CN5V.js";
+import "../chunk-EFGX3EX5.js";
+import "../chunk-OIMLU4SF.js";
 export {
   SkillValidationError,
   loadSkills,

package/dist/think-tool-ByArsrxe.d.ts

@@ -0,0 +1,26 @@
+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-C3ciF-VG.js';
+
+type CreateToolOptions<InputSchema extends ZodSchema, OutputSchema extends ZodSchema | undefined = undefined, Output = unknown> = {
+    name: string;
+    description: string;
+    input: InputSchema;
+    output?: OutputSchema;
+    approval?: ToolApprovalPolicy<z.output<InputSchema>>;
+    execute(args: z.output<InputSchema>, context: ToolCallContext): OutputSchema extends ZodSchema ? z.input<OutputSchema> | Promise<z.input<OutputSchema>> : Output | Promise<Output>;
+};
+declare function createTool<InputSchema extends ZodSchema, Output = unknown>(options: CreateToolOptions<InputSchema, undefined, Output> & {
+    output?: undefined;
+}): Tool<z.output<InputSchema>, Output>;
+declare function createTool<InputSchema extends ZodSchema, OutputSchema extends ZodSchema>(options: CreateToolOptions<InputSchema, OutputSchema>): Tool<z.output<InputSchema>, z.output<OutputSchema>>;
+
+type CreateThinkToolOptions = {
+    name?: string;
+    description?: string;
+};
+declare function createThinkTool(options?: CreateThinkToolOptions): Tool<{
+    thought: string;
+}, string>;
+
+export { type CreateToolOptions as C, createTool as a, type CreateThinkToolOptions as b, createThinkTool as c };

package/dist/tool/index.d.ts

@@ -1,25 +1,12 @@
-import { z } from 'zod';
-import { Z as ZodSchema } from '../middleware-2FR89muX.js';
-export { D as DynamicToolIndex, E as EmbedToolsOptions, T as ToolMiddleware, a as ToolResultMiddlewareArgs, b as ToolSearchDocument, c as ToolSet, d as createToolIndex, e as createToolMiddleware, f as embedTools, i as isDynamicToolIndex } from '../middleware-2FR89muX.js';
-import { b as ToolApprovalPolicy, d as ToolCallContext, T as Tool } from '../tool-ClZYES-Z.js';
-export { A as AnyTool, N as NormalizedToolOutput, a as ToolApprovalContext, c as ToolApprovalRunContext, e as ToolCallStreamEvent, f as ToolOutput, i as isToolResultContentArray, n as normalizeToolResultOutput, p as parseToolArgs, s as serializeToolOutput, t as toolResultContentToText } from '../tool-ClZYES-Z.js';
-import '../types-BM8-Y8Hy.js';
-import '../embeddings/index.js';
+export { b as CreateThinkToolOptions, C as CreateToolOptions, c as createThinkTool, a as createTool } from '../think-tool-ByArsrxe.js';
+export { D as DynamicToolIndex, E as EmbedToolsOptions, T as ToolMiddleware, a as ToolResultMiddlewareArgs, b as ToolSearchDocument, c as ToolSet, d as createToolIndex, e as createToolMiddleware, f as embedTools, i as isDynamicToolIndex } from '../middleware-CGiEIaBx.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, i as isToolResultContentArray, n as normalizeToolResultOutput, p as parseToolArgs, s as serializeToolOutput, t as toolResultContentToText } from '../tool-C3ciF-VG.js';
+import 'zod';
+import '../zod-schema-C7F4clpm.js';
+import '../types-C54aNoCd.js';
+import '../types-IB2e9u5M.js';
 import '../vector-store/index.js';
 
-type CreateToolOptions<InputSchema extends ZodSchema, OutputSchema extends ZodSchema | undefined = undefined, Output = unknown> = {
-    name: string;
-    description: string;
-    input: InputSchema;
-    output?: OutputSchema;
-    approval?: ToolApprovalPolicy<z.output<InputSchema>>;
-    execute(args: z.output<InputSchema>, context: ToolCallContext): OutputSchema extends ZodSchema ? z.input<OutputSchema> | Promise<z.input<OutputSchema>> : Output | Promise<Output>;
-};
-declare function createTool<InputSchema extends ZodSchema, Output = unknown>(options: CreateToolOptions<InputSchema, undefined, Output> & {
-    output?: undefined;
-}): Tool<z.output<InputSchema>, Output>;
-declare function createTool<InputSchema extends ZodSchema, OutputSchema extends ZodSchema>(options: CreateToolOptions<InputSchema, OutputSchema>): Tool<z.output<InputSchema>, z.output<OutputSchema>>;
-
 declare class ToolCallError extends Error {
     readonly cause?: unknown | undefined;
     constructor(message: string, cause?: unknown | undefined);
@@ -33,12 +20,4 @@ declare class ToolJsonError extends Error {
     constructor(message: string, cause?: unknown | undefined);
 }
 
-type CreateThinkToolOptions = {
-    name?: string;
-    description?: string;
-};
-declare function createThinkTool(options?: CreateThinkToolOptions): Tool<{
-    thought: string;
-}, string>;
-
-export { type CreateThinkToolOptions, type CreateToolOptions, Tool, ToolApprovalPolicy, ToolCallContext, ToolCallError, ToolJsonError, ToolNotFoundError, createThinkTool, createTool };
+export { ToolCallError, ToolJsonError, ToolNotFoundError };

package/dist/tool/index.js

@@ -14,11 +14,12 @@ import {
   parseToolArgs,
   serializeToolOutput,
   toolResultContentToText
-} from "../chunk-I2HOMD3R.js";
+} from "../chunk-CY7SB5FG.js";
 import {
   createTool
-} from "../chunk-65QV627O.js";
-import "../chunk-A7VDIZQN.js";
+} from "../chunk-OIA4CN5V.js";
+import "../chunk-EFGX3EX5.js";
+import "../chunk-OIMLU4SF.js";
 export {
   ToolCallError,
   ToolJsonError,

package/dist/{tool-ClZYES-Z.d.ts → tool-C3ciF-VG.d.ts}

@@ -1,4 +1,4 @@
-import { J as JsonObject, q as ToolDefinition, u as ToolResultContent, j as JsonValue } from './types-BM8-Y8Hy.js';
+import { J as JsonObject, g as ToolDefinition, j as ToolResultContent, e as JsonValue } from './types-C54aNoCd.js';
 
 type ToolApprovalRunContext = {
     agentId: string;
@@ -47,4 +47,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 ToolApprovalRunContext as c, type ToolCallContext as d, type ToolCallStreamEvent as e, ToolOutput as f, isToolResultContentArray as i, normalizeToolResultOutput as n, parseToolArgs as p, serializeToolOutput as s, toolResultContentToText as t };
+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, isToolResultContentArray as i, normalizeToolResultOutput as n, parseToolArgs as p, serializeToolOutput as s, toolResultContentToText as t };

package/dist/transcription/index.d.ts

@@ -1,4 +1,4 @@
-import { j as JsonValue } from '../types-BM8-Y8Hy.js';
+import { e as JsonValue } from '../types-C54aNoCd.js';
 
 type TranscriptionRequest = {
     data: Uint8Array;

package/dist/transcription/index.js

@@ -1,7 +1,58 @@
-import {
-  TranscriptionRequestBuilder,
-  transcriptionRequest
-} from "../chunk-7QI6ZAFI.js";
+// src/transcription/index.ts
+var TranscriptionRequestBuilder = class {
+  constructor(model) {
+    this.model = model;
+  }
+  model;
+  request = {
+    data: new Uint8Array(),
+    filename: "file"
+  };
+  data(data) {
+    this.request = { ...this.request, data: toUint8Array(data) };
+    return this;
+  }
+  filename(filename) {
+    this.request = { ...this.request, filename };
+    return this;
+  }
+  language(language) {
+    this.request = { ...this.request, language };
+    return this;
+  }
+  prompt(prompt) {
+    this.request = { ...this.request, prompt };
+    return this;
+  }
+  temperature(temperature) {
+    this.request = { ...this.request, temperature };
+    return this;
+  }
+  additionalParams(additionalParams) {
+    this.request = { ...this.request, additionalParams };
+    return this;
+  }
+  build() {
+    if (this.request.data.byteLength === 0) {
+      throw new Error("Transcription data cannot be empty.");
+    }
+    return { ...this.request, data: toUint8Array(this.request.data) };
+  }
+  send() {
+    return this.model.transcription(this.build());
+  }
+};
+function transcriptionRequest(model) {
+  return new TranscriptionRequestBuilder(model);
+}
+function toUint8Array(bytes) {
+  if (bytes instanceof Uint8Array) {
+    return new Uint8Array(
+      bytes.buffer.slice(bytes.byteOffset, bytes.byteOffset + bytes.byteLength)
+    );
+  }
+  return new Uint8Array(bytes);
+}
 export {
   TranscriptionRequestBuilder,
   transcriptionRequest

package/dist/transcription/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}
+{"version":3,"sources":["../../src/transcription/index.ts"],"sourcesContent":["import type { JsonValue } from \"../completion\";\n\nexport type TranscriptionRequest = {\n  data: Uint8Array;\n  filename: string;\n  language?: string | undefined;\n  prompt?: string | undefined;\n  temperature?: number | undefined;\n  additionalParams?: JsonValue | undefined;\n};\n\nexport type TranscriptionResponse<RawResponse = unknown> = {\n  text: string;\n  rawResponse: RawResponse;\n};\n\nexport interface TranscriptionModel<RawResponse = unknown> {\n  readonly provider?: string | undefined;\n  readonly defaultModel?: string | undefined;\n  transcription(request: TranscriptionRequest): Promise<TranscriptionResponse<RawResponse>>;\n}\n\nexport class TranscriptionRequestBuilder<Model extends TranscriptionModel = TranscriptionModel> {\n  private request: TranscriptionRequest = {\n    data: new Uint8Array(),\n    filename: \"file\",\n  };\n\n  constructor(private readonly model: Model) {}\n\n  data(data: Uint8Array | ArrayBuffer): this {\n    this.request = { ...this.request, data: toUint8Array(data) };\n    return this;\n  }\n\n  filename(filename: string): this {\n    this.request = { ...this.request, filename };\n    return this;\n  }\n\n  language(language: string): this {\n    this.request = { ...this.request, language };\n    return this;\n  }\n\n  prompt(prompt: string): this {\n    this.request = { ...this.request, prompt };\n    return this;\n  }\n\n  temperature(temperature: number): this {\n    this.request = { ...this.request, temperature };\n    return this;\n  }\n\n  additionalParams(additionalParams: JsonValue): this {\n    this.request = { ...this.request, additionalParams };\n    return this;\n  }\n\n  build(): TranscriptionRequest {\n    if (this.request.data.byteLength === 0) {\n      throw new Error(\"Transcription data cannot be empty.\");\n    }\n    return { ...this.request, data: toUint8Array(this.request.data) };\n  }\n\n  send(): Promise<Awaited<ReturnType<Model[\"transcription\"]>>> {\n    return this.model.transcription(this.build()) as Promise<\n      Awaited<ReturnType<Model[\"transcription\"]>>\n    >;\n  }\n}\n\nexport function transcriptionRequest<Model extends TranscriptionModel>(\n  model: Model,\n): TranscriptionRequestBuilder<Model> {\n  return new TranscriptionRequestBuilder(model);\n}\n\nfunction toUint8Array(bytes: Uint8Array | ArrayBuffer): Uint8Array {\n  if (bytes instanceof Uint8Array) {\n    return new Uint8Array(\n      bytes.buffer.slice(bytes.byteOffset, bytes.byteOffset + bytes.byteLength),\n    );\n  }\n  return new Uint8Array(bytes);\n}\n"],"mappings":";AAsBO,IAAM,8BAAN,MAAyF;AAAA,EAM9F,YAA6B,OAAc;AAAd;AAAA,EAAe;AAAA,EAAf;AAAA,EALrB,UAAgC;AAAA,IACtC,MAAM,IAAI,WAAW;AAAA,IACrB,UAAU;AAAA,EACZ;AAAA,EAIA,KAAK,MAAsC;AACzC,SAAK,UAAU,EAAE,GAAG,KAAK,SAAS,MAAM,aAAa,IAAI,EAAE;AAC3D,WAAO;AAAA,EACT;AAAA,EAEA,SAAS,UAAwB;AAC/B,SAAK,UAAU,EAAE,GAAG,KAAK,SAAS,SAAS;AAC3C,WAAO;AAAA,EACT;AAAA,EAEA,SAAS,UAAwB;AAC/B,SAAK,UAAU,EAAE,GAAG,KAAK,SAAS,SAAS;AAC3C,WAAO;AAAA,EACT;AAAA,EAEA,OAAO,QAAsB;AAC3B,SAAK,UAAU,EAAE,GAAG,KAAK,SAAS,OAAO;AACzC,WAAO;AAAA,EACT;AAAA,EAEA,YAAY,aAA2B;AACrC,SAAK,UAAU,EAAE,GAAG,KAAK,SAAS,YAAY;AAC9C,WAAO;AAAA,EACT;AAAA,EAEA,iBAAiB,kBAAmC;AAClD,SAAK,UAAU,EAAE,GAAG,KAAK,SAAS,iBAAiB;AACnD,WAAO;AAAA,EACT;AAAA,EAEA,QAA8B;AAC5B,QAAI,KAAK,QAAQ,KAAK,eAAe,GAAG;AACtC,YAAM,IAAI,MAAM,qCAAqC;AAAA,IACvD;AACA,WAAO,EAAE,GAAG,KAAK,SAAS,MAAM,aAAa,KAAK,QAAQ,IAAI,EAAE;AAAA,EAClE;AAAA,EAEA,OAA6D;AAC3D,WAAO,KAAK,MAAM,cAAc,KAAK,MAAM,CAAC;AAAA,EAG9C;AACF;AAEO,SAAS,qBACd,OACoC;AACpC,SAAO,IAAI,4BAA4B,KAAK;AAC9C;AAEA,SAAS,aAAa,OAA6C;AACjE,MAAI,iBAAiB,YAAY;AAC/B,WAAO,IAAI;AAAA,MACT,MAAM,OAAO,MAAM,MAAM,YAAY,MAAM,aAAa,MAAM,UAAU;AAAA,IAC1E;AAAA,EACF;AACA,SAAO,IAAI,WAAW,KAAK;AAC7B;","names":[]}

package/dist/{types-BM8-Y8Hy.d.ts → types-C54aNoCd.d.ts}

@@ -250,4 +250,4 @@ declare function assertCompletionRequestSupported(model: CompletionModel, reques
 }): void;
 declare function textFromAssistantContent(content: AssistantContent[]): string;
 
-export { AssistantContent as A, CompletionCapabilityError as C, type Document as D, type ImageContent as I, type JsonObject as J, Message as M, type Reasoning as R, type StreamingCompletionModel as S, type Text as T, Usage as U, type AssistantMessage as a, type CompletionModel as b, type CompletionModelCapabilities as c, type CompletionRequest as d, type CompletionResponse as e, type CompletionStreamEvent as f, type DocumentContent as g, type ImageDetail as h, type JsonPrimitive as i, type JsonValue as j, type ReasoningContent as k, type ReasoningContentType as l, type SystemMessage as m, type ToolCall as n, type ToolChoice as o, ToolContent as p, type ToolDefinition as q, type ToolFunction as r, type ToolMessage as s, type ToolResult as t, type ToolResultContent as u, UserContent as v, type UserMessage as w, assertCompletionRequestSupported as x, reasoningDisplayText as y, textFromAssistantContent as z };
+export { AssistantContent as A, type CompletionModel as C, type Document as D, type ImageContent as I, type JsonObject as J, Message as M, type Reasoning as R, type SystemMessage as S, type Text as T, Usage as U, type AssistantMessage as a, type CompletionRequest as b, type CompletionResponse as c, type JsonPrimitive as d, type JsonValue as e, type ToolCall as f, type ToolDefinition as g, type ToolMessage as h, type ToolResult as i, type ToolResultContent as j, UserContent as k, type UserMessage as l, type ToolChoice as m, CompletionCapabilityError as n, type CompletionModelCapabilities as o, type CompletionStreamEvent as p, type DocumentContent as q, type ImageDetail as r, type ReasoningContent as s, type ReasoningContentType as t, type StreamingCompletionModel as u, ToolContent as v, type ToolFunction as w, assertCompletionRequestSupported as x, reasoningDisplayText as y, textFromAssistantContent as z };

package/dist/types-IB2e9u5M.d.ts

@@ -0,0 +1,25 @@
+type Embedding = {
+    document: string;
+    vector: number[];
+};
+interface EmbeddingModel {
+    readonly dimensions?: number | undefined;
+    readonly maxBatchSize?: number | undefined;
+    embedTexts(texts: string[]): Promise<Embedding[]>;
+}
+type EmbeddedDocument<T, Metadata extends VectorMetadata = VectorMetadata> = {
+    id: string;
+    document: T;
+    metadata?: Metadata | undefined;
+    embeddings: Embedding[];
+};
+type VectorMetadataValue = string | number | boolean | null;
+type VectorMetadata = Record<string, VectorMetadataValue>;
+type EmbedDocumentsOptions<T, Metadata extends VectorMetadata = VectorMetadata> = {
+    id?: ((document: T, index: number) => string) | undefined;
+    content(document: T, index: number): string | string[];
+    metadata?: ((document: T, index: number) => Metadata | undefined) | undefined;
+    concurrency?: number | undefined;
+};
+
+export type { EmbeddingModel as E, VectorMetadata as V, EmbedDocumentsOptions as a, EmbeddedDocument as b, Embedding as c, VectorMetadataValue as d };