@anvia/core 1.0.0-rc.1 → 1.0.0-rc.3

package/dist/chunk-VKYZZXP5.js

@@ -0,0 +1,323 @@
+import {
+  Usage
+} from "./chunk-D2ECOWN5.js";
+
+// src/observability/snapshot.ts
+function observerSnapshot(value) {
+  if (containsError(value, /* @__PURE__ */ new WeakSet())) {
+    return cloneObserverFallback(value, /* @__PURE__ */ new WeakMap());
+  }
+  try {
+    return globalThis.structuredClone(value);
+  } catch {
+    return cloneObserverFallback(value, /* @__PURE__ */ new WeakMap());
+  }
+}
+function cloneObserverFallback(value, seen) {
+  if (typeof value !== "object" || value === null) {
+    return value;
+  }
+  const existing = seen.get(value);
+  if (existing !== void 0) {
+    return existing;
+  }
+  if (Array.isArray(value)) {
+    const clone2 = [];
+    seen.set(value, clone2);
+    clone2.push(...value.map((item) => cloneObserverFallback(item, seen)));
+    return clone2;
+  }
+  if (value instanceof Date) {
+    return new Date(value.getTime());
+  }
+  if (value instanceof Error) {
+    const clone2 = Object.create(Object.getPrototypeOf(value));
+    seen.set(value, clone2);
+    Object.defineProperties(clone2, {
+      name: { configurable: true, writable: true, value: value.name },
+      message: { configurable: true, writable: true, value: value.message }
+    });
+    if (value.stack !== void 0) {
+      Object.defineProperty(clone2, "stack", {
+        configurable: true,
+        writable: true,
+        value: value.stack
+      });
+    }
+    if (value.cause !== void 0) {
+      Object.defineProperty(clone2, "cause", {
+        configurable: true,
+        writable: true,
+        value: cloneObserverFallback(value.cause, seen)
+      });
+    }
+    for (const key of Object.getOwnPropertyNames(value)) {
+      if (key === "name" || key === "message" || key === "stack" || key === "cause") continue;
+      const descriptor = Object.getOwnPropertyDescriptor(value, key);
+      if (descriptor === void 0) continue;
+      if ("value" in descriptor) {
+        descriptor.value = cloneObserverFallback(descriptor.value, seen);
+      }
+      Object.defineProperty(clone2, key, descriptor);
+    }
+    return clone2;
+  }
+  if (Object.getPrototypeOf(value) !== Object.prototype) {
+    return value;
+  }
+  const clone = {};
+  seen.set(value, clone);
+  for (const [key, item] of Object.entries(value)) {
+    clone[key] = cloneObserverFallback(item, seen);
+  }
+  return clone;
+}
+function containsError(value, seen) {
+  if (typeof value !== "object" || value === null) return false;
+  if (value instanceof Error) return true;
+  if (seen.has(value)) return false;
+  seen.add(value);
+  if (Array.isArray(value)) {
+    return value.some((item) => containsError(item, seen));
+  }
+  if (Object.getPrototypeOf(value) !== Object.prototype) return false;
+  return Object.values(value).some((item) => containsError(item, seen));
+}
+
+// src/observability/group.ts
+var AgentObserverDispatchError = class extends AggregateError {
+  phase;
+  failures;
+  constructor(phase, failures) {
+    super(
+      failures.map((failure) => failure.error),
+      `Agent observer ${phase} failed for ${failures.map((failure) => failure.observer).join(", ")}.`
+    );
+    this.name = "AgentObserverDispatchError";
+    this.phase = phase;
+    this.failures = Object.freeze([...failures]);
+  }
+};
+async function startAgentRunObservers(observers, args, options) {
+  const runObservers = [];
+  const failures = [];
+  for (const [name, observer] of Object.entries(observers)) {
+    try {
+      const runObserver = await observer.startRun(observerSnapshot(args));
+      if (runObserver !== void 0) {
+        runObservers.push({ name, observer: runObserver, terminal: false });
+      }
+    } catch (error) {
+      failures.push({ observer: name, error });
+    }
+  }
+  if (options.errorPolicy === "throw" && failures.length > 0) {
+    const startupError = new AgentObserverDispatchError("startRun", failures);
+    const cleanupFailures = await terminateObservers(
+      runObservers,
+      (observer) => observer.error?.(
+        observerSnapshot({
+          error: startupError,
+          usage: Usage.empty(),
+          messages: [...args.history, args.prompt]
+        })
+      )
+    );
+    throw new AgentObserverDispatchError("startRun", [...failures, ...cleanupFailures]);
+  }
+  return new ActiveAgentRunObservers(runObservers, options);
+}
+var ActiveAgentRunObservers = class {
+  constructor(runObservers, options) {
+    this.runObservers = runObservers;
+    this.options = options;
+    const primary = options.primaryTrace === void 0 ? void 0 : runObservers.find((entry) => entry.name === options.primaryTrace);
+    this.trace = primary?.observer.trace === void 0 ? void 0 : Object.freeze({ observer: primary.name, ...primary.observer.trace });
+  }
+  runObservers;
+  options;
+  trace;
+  async startGeneration(args) {
+    const generationObservers = [];
+    const failures = [];
+    for (const entry of this.runObservers) {
+      if (entry.observer.startGeneration === void 0) continue;
+      try {
+        const observer = await entry.observer.startGeneration(observerSnapshot(args));
+        if (observer !== void 0) {
+          generationObservers.push({ name: entry.name, observer, terminal: false });
+        }
+      } catch (error) {
+        failures.push({ observer: entry.name, error });
+      }
+    }
+    if (this.options.errorPolicy === "throw" && failures.length > 0) {
+      const startupError = new AgentObserverDispatchError("startGeneration", failures);
+      const cleanupFailures = await terminateObservers(
+        generationObservers,
+        (observer) => observer.error?.(observerSnapshot({ turn: args.turn, error: startupError }))
+      );
+      throw new AgentObserverDispatchError("startGeneration", [...failures, ...cleanupFailures]);
+    }
+    return new ActiveGenerationObservers(generationObservers, this.options.errorPolicy);
+  }
+  async startTool(args) {
+    const toolObservers = [];
+    const failures = [];
+    for (const entry of this.runObservers) {
+      if (entry.observer.startTool === void 0) continue;
+      try {
+        const observer = await entry.observer.startTool(observerSnapshot(args));
+        if (observer !== void 0) {
+          toolObservers.push({ name: entry.name, observer, terminal: false });
+        }
+      } catch (error) {
+        failures.push({ observer: entry.name, error });
+      }
+    }
+    if (this.options.errorPolicy === "throw" && failures.length > 0) {
+      const startupError = new AgentObserverDispatchError("startTool", failures);
+      const cleanupFailures = await terminateObservers(
+        toolObservers,
+        (observer) => observer.error?.(observerSnapshot({ ...args, error: startupError }))
+      );
+      throw new AgentObserverDispatchError("startTool", [...failures, ...cleanupFailures]);
+    }
+    return new ActiveToolObservers(toolObservers, this.options.errorPolicy);
+  }
+  async end(args) {
+    await dispatchTerminalObservers(
+      this.runObservers,
+      "end",
+      (observer) => observer.end(observerSnapshot(args)),
+      this.options.errorPolicy
+    );
+  }
+  async error(args) {
+    await dispatchTerminalObservers(
+      this.runObservers,
+      "error",
+      (observer) => observer.error?.(observerSnapshot(args)),
+      "ignore"
+    );
+  }
+  async event(args) {
+    await dispatchObservers(
+      this.runObservers,
+      "event",
+      (observer) => observer.event?.(observerSnapshot(args)),
+      this.options.errorPolicy
+    );
+  }
+};
+var ActiveGenerationObservers = class {
+  constructor(observers, errorPolicy) {
+    this.observers = observers;
+    this.errorPolicy = errorPolicy;
+  }
+  observers;
+  errorPolicy;
+  async end(args) {
+    await dispatchTerminalObservers(
+      this.observers,
+      "generation.end",
+      (observer) => observer.end(observerSnapshot(args)),
+      this.errorPolicy
+    );
+  }
+  async error(args) {
+    await dispatchTerminalObservers(
+      this.observers,
+      "generation.error",
+      (observer) => observer.error?.(observerSnapshot(args)),
+      "ignore"
+    );
+  }
+  async update(args) {
+    await dispatchObservers(
+      this.observers,
+      "generation.update",
+      (observer) => observer.update?.(observerSnapshot(args)),
+      this.errorPolicy
+    );
+  }
+};
+var ActiveToolObservers = class {
+  constructor(observers, errorPolicy) {
+    this.observers = observers;
+    this.errorPolicy = errorPolicy;
+  }
+  observers;
+  errorPolicy;
+  async streamEvent(args) {
+    await dispatchObservers(
+      this.observers,
+      "tool.streamEvent",
+      (observer) => observer.streamEvent?.(observerSnapshot(args)),
+      this.errorPolicy
+    );
+  }
+  async end(args) {
+    await dispatchTerminalObservers(
+      this.observers,
+      "tool.end",
+      (observer) => observer.end(observerSnapshot(args)),
+      this.errorPolicy
+    );
+  }
+  async suspend(args) {
+    await dispatchTerminalObservers(
+      this.observers,
+      "tool.suspend",
+      (observer) => observer.suspend?.(observerSnapshot(args)),
+      this.errorPolicy
+    );
+  }
+  async error(args) {
+    await dispatchTerminalObservers(
+      this.observers,
+      "tool.error",
+      (observer) => observer.error?.(observerSnapshot(args)),
+      "ignore"
+    );
+  }
+};
+async function dispatchObservers(observers, phase, dispatch, errorPolicy) {
+  const failures = [];
+  for (const entry of observers) {
+    try {
+      await dispatch(entry.observer);
+    } catch (error) {
+      failures.push({ observer: entry.name, error });
+    }
+  }
+  throwObserverFailures(phase, failures, errorPolicy);
+}
+async function dispatchTerminalObservers(observers, phase, dispatch, errorPolicy) {
+  const failures = await terminateObservers(observers, dispatch);
+  throwObserverFailures(phase, failures, errorPolicy);
+}
+async function terminateObservers(observers, dispatch) {
+  const failures = [];
+  for (const entry of observers) {
+    if (entry.terminal) continue;
+    entry.terminal = true;
+    try {
+      await dispatch(entry.observer);
+    } catch (error) {
+      failures.push({ observer: entry.name, error });
+    }
+  }
+  return failures;
+}
+function throwObserverFailures(phase, failures, errorPolicy) {
+  if (errorPolicy === "throw" && failures.length > 0) {
+    throw new AgentObserverDispatchError(phase, failures);
+  }
+}
+
+export {
+  AgentObserverDispatchError,
+  startAgentRunObservers
+};
+//# sourceMappingURL=chunk-VKYZZXP5.js.map

package/dist/chunk-VKYZZXP5.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/observability/snapshot.ts","../src/observability/group.ts"],"sourcesContent":["export function observerSnapshot<T>(value: T): T {\n  if (containsError(value, new WeakSet<object>())) {\n    return cloneObserverFallback(value, new WeakMap<object, object>());\n  }\n  try {\n    return globalThis.structuredClone(value);\n  } catch {\n    return cloneObserverFallback(value, new WeakMap<object, object>());\n  }\n}\n\nfunction cloneObserverFallback<T>(value: T, seen: WeakMap<object, object>): T {\n  if (typeof value !== \"object\" || value === null) {\n    return value;\n  }\n  const existing = seen.get(value);\n  if (existing !== undefined) {\n    return existing as T;\n  }\n  if (Array.isArray(value)) {\n    const clone: unknown[] = [];\n    seen.set(value, clone);\n    clone.push(...value.map((item) => cloneObserverFallback(item, seen)));\n    return clone as T;\n  }\n  if (value instanceof Date) {\n    return new Date(value.getTime()) as T;\n  }\n  if (value instanceof Error) {\n    const clone = Object.create(Object.getPrototypeOf(value)) as Error;\n    seen.set(value, clone);\n    Object.defineProperties(clone, {\n      name: { configurable: true, writable: true, value: value.name },\n      message: { configurable: true, writable: true, value: value.message },\n    });\n    if (value.stack !== undefined) {\n      Object.defineProperty(clone, \"stack\", {\n        configurable: true,\n        writable: true,\n        value: value.stack,\n      });\n    }\n    if (value.cause !== undefined) {\n      Object.defineProperty(clone, \"cause\", {\n        configurable: true,\n        writable: true,\n        value: cloneObserverFallback(value.cause, seen),\n      });\n    }\n    for (const key of Object.getOwnPropertyNames(value)) {\n      if (key === \"name\" || key === \"message\" || key === \"stack\" || key === \"cause\") continue;\n      const descriptor = Object.getOwnPropertyDescriptor(value, key);\n      if (descriptor === undefined) continue;\n      if (\"value\" in descriptor) {\n        descriptor.value = cloneObserverFallback(descriptor.value, seen);\n      }\n      Object.defineProperty(clone, key, descriptor);\n    }\n    return clone as T;\n  }\n  if (Object.getPrototypeOf(value) !== Object.prototype) {\n    return value;\n  }\n  const clone: Record<string, unknown> = {};\n  seen.set(value, clone);\n  for (const [key, item] of Object.entries(value)) {\n    clone[key] = cloneObserverFallback(item, seen);\n  }\n  return clone as T;\n}\n\nfunction containsError(value: unknown, seen: WeakSet<object>): boolean {\n  if (typeof value !== \"object\" || value === null) return false;\n  if (value instanceof Error) return true;\n  if (seen.has(value)) return false;\n  seen.add(value);\n  if (Array.isArray(value)) {\n    return value.some((item) => containsError(item, seen));\n  }\n  if (Object.getPrototypeOf(value) !== Object.prototype) return false;\n  return Object.values(value).some((item) => containsError(item, seen));\n}\n","import { Usage } from \"../completion\";\nimport { observerSnapshot } from \"./snapshot\";\nimport type {\n  AgentGenerationEndArgs,\n  AgentGenerationErrorArgs,\n  AgentGenerationObserver,\n  AgentGenerationStartArgs,\n  AgentGenerationUpdateArgs,\n  AgentObserverErrorPolicy,\n  AgentObserverMap,\n  AgentRunEndArgs,\n  AgentRunErrorArgs,\n  AgentRunEventArgs,\n  AgentRunObserver,\n  AgentRunStartArgs,\n  AgentToolEndArgs,\n  AgentToolErrorArgs,\n  AgentToolObserver,\n  AgentToolStartArgs,\n  AgentToolStreamEventArgs,\n  AgentToolSuspendedArgs,\n  AgentTraceInfo,\n} from \"./types\";\n\nexport type AgentObserverFailure = {\n  readonly observer: string;\n  readonly error: unknown;\n};\n\nexport class AgentObserverDispatchError extends AggregateError {\n  readonly phase: string;\n  readonly failures: readonly AgentObserverFailure[];\n\n  constructor(phase: string, failures: readonly AgentObserverFailure[]) {\n    super(\n      failures.map((failure) => failure.error),\n      `Agent observer ${phase} failed for ${failures.map((failure) => failure.observer).join(\", \")}.`,\n    );\n    this.name = \"AgentObserverDispatchError\";\n    this.phase = phase;\n    this.failures = Object.freeze([...failures]);\n  }\n}\n\ntype NamedObserver<T> = {\n  readonly name: string;\n  readonly observer: T;\n};\n\ntype ActiveNamedObserver<T> = NamedObserver<T> & {\n  terminal: boolean;\n};\n\nexport async function startAgentRunObservers(\n  observers: AgentObserverMap,\n  args: AgentRunStartArgs,\n  options: {\n    readonly primaryTrace?: string | undefined;\n    readonly errorPolicy: AgentObserverErrorPolicy;\n  },\n): Promise<ActiveAgentRunObservers> {\n  const runObservers: ActiveNamedObserver<AgentRunObserver>[] = [];\n  const failures: AgentObserverFailure[] = [];\n  for (const [name, observer] of Object.entries(observers)) {\n    try {\n      const runObserver = await observer.startRun(observerSnapshot(args));\n      if (runObserver !== undefined) {\n        runObservers.push({ name, observer: runObserver, terminal: false });\n      }\n    } catch (error) {\n      failures.push({ observer: name, error });\n    }\n  }\n  if (options.errorPolicy === \"throw\" && failures.length > 0) {\n    const startupError = new AgentObserverDispatchError(\"startRun\", failures);\n    const cleanupFailures = await terminateObservers(runObservers, (observer) =>\n      observer.error?.(\n        observerSnapshot({\n          error: startupError,\n          usage: Usage.empty(),\n          messages: [...args.history, args.prompt],\n        }),\n      ),\n    );\n    throw new AgentObserverDispatchError(\"startRun\", [...failures, ...cleanupFailures]);\n  }\n  return new ActiveAgentRunObservers(runObservers, options);\n}\n\nexport class ActiveAgentRunObservers {\n  readonly trace: AgentTraceInfo | undefined;\n\n  constructor(\n    private readonly runObservers: readonly ActiveNamedObserver<AgentRunObserver>[],\n    private readonly options: {\n      readonly primaryTrace?: string | undefined;\n      readonly errorPolicy: AgentObserverErrorPolicy;\n    },\n  ) {\n    const primary =\n      options.primaryTrace === undefined\n        ? undefined\n        : runObservers.find((entry) => entry.name === options.primaryTrace);\n    this.trace =\n      primary?.observer.trace === undefined\n        ? undefined\n        : Object.freeze({ observer: primary.name, ...primary.observer.trace });\n  }\n\n  async startGeneration(args: AgentGenerationStartArgs): Promise<ActiveGenerationObservers> {\n    const generationObservers: ActiveNamedObserver<AgentGenerationObserver>[] = [];\n    const failures: AgentObserverFailure[] = [];\n    for (const entry of this.runObservers) {\n      if (entry.observer.startGeneration === undefined) continue;\n      try {\n        const observer = await entry.observer.startGeneration(observerSnapshot(args));\n        if (observer !== undefined) {\n          generationObservers.push({ name: entry.name, observer, terminal: false });\n        }\n      } catch (error) {\n        failures.push({ observer: entry.name, error });\n      }\n    }\n    if (this.options.errorPolicy === \"throw\" && failures.length > 0) {\n      const startupError = new AgentObserverDispatchError(\"startGeneration\", failures);\n      const cleanupFailures = await terminateObservers(generationObservers, (observer) =>\n        observer.error?.(observerSnapshot({ turn: args.turn, error: startupError })),\n      );\n      throw new AgentObserverDispatchError(\"startGeneration\", [...failures, ...cleanupFailures]);\n    }\n    return new ActiveGenerationObservers(generationObservers, this.options.errorPolicy);\n  }\n\n  async startTool(args: AgentToolStartArgs): Promise<ActiveToolObservers> {\n    const toolObservers: ActiveNamedObserver<AgentToolObserver>[] = [];\n    const failures: AgentObserverFailure[] = [];\n    for (const entry of this.runObservers) {\n      if (entry.observer.startTool === undefined) continue;\n      try {\n        const observer = await entry.observer.startTool(observerSnapshot(args));\n        if (observer !== undefined) {\n          toolObservers.push({ name: entry.name, observer, terminal: false });\n        }\n      } catch (error) {\n        failures.push({ observer: entry.name, error });\n      }\n    }\n    if (this.options.errorPolicy === \"throw\" && failures.length > 0) {\n      const startupError = new AgentObserverDispatchError(\"startTool\", failures);\n      const cleanupFailures = await terminateObservers(toolObservers, (observer) =>\n        observer.error?.(observerSnapshot({ ...args, error: startupError })),\n      );\n      throw new AgentObserverDispatchError(\"startTool\", [...failures, ...cleanupFailures]);\n    }\n    return new ActiveToolObservers(toolObservers, this.options.errorPolicy);\n  }\n\n  async end(args: AgentRunEndArgs): Promise<void> {\n    await dispatchTerminalObservers(\n      this.runObservers,\n      \"end\",\n      (observer) => observer.end(observerSnapshot(args)),\n      this.options.errorPolicy,\n    );\n  }\n\n  async error(args: AgentRunErrorArgs): Promise<void> {\n    await dispatchTerminalObservers(\n      this.runObservers,\n      \"error\",\n      (observer) => observer.error?.(observerSnapshot(args)),\n      \"ignore\",\n    );\n  }\n\n  async event(args: AgentRunEventArgs): Promise<void> {\n    await dispatchObservers(\n      this.runObservers,\n      \"event\",\n      (observer) => observer.event?.(observerSnapshot(args)),\n      this.options.errorPolicy,\n    );\n  }\n}\n\nexport class ActiveGenerationObservers {\n  constructor(\n    private readonly observers: readonly ActiveNamedObserver<AgentGenerationObserver>[],\n    private readonly errorPolicy: AgentObserverErrorPolicy,\n  ) {}\n\n  async end(args: AgentGenerationEndArgs): Promise<void> {\n    await dispatchTerminalObservers(\n      this.observers,\n      \"generation.end\",\n      (observer) => observer.end(observerSnapshot(args)),\n      this.errorPolicy,\n    );\n  }\n\n  async error(args: AgentGenerationErrorArgs): Promise<void> {\n    await dispatchTerminalObservers(\n      this.observers,\n      \"generation.error\",\n      (observer) => observer.error?.(observerSnapshot(args)),\n      \"ignore\",\n    );\n  }\n\n  async update(args: AgentGenerationUpdateArgs): Promise<void> {\n    await dispatchObservers(\n      this.observers,\n      \"generation.update\",\n      (observer) => observer.update?.(observerSnapshot(args)),\n      this.errorPolicy,\n    );\n  }\n}\n\nexport class ActiveToolObservers {\n  constructor(\n    private readonly observers: readonly ActiveNamedObserver<AgentToolObserver>[],\n    private readonly errorPolicy: AgentObserverErrorPolicy,\n  ) {}\n\n  async streamEvent(args: AgentToolStreamEventArgs): Promise<void> {\n    await dispatchObservers(\n      this.observers,\n      \"tool.streamEvent\",\n      (observer) => observer.streamEvent?.(observerSnapshot(args)),\n      this.errorPolicy,\n    );\n  }\n\n  async end(args: AgentToolEndArgs): Promise<void> {\n    await dispatchTerminalObservers(\n      this.observers,\n      \"tool.end\",\n      (observer) => observer.end(observerSnapshot(args)),\n      this.errorPolicy,\n    );\n  }\n\n  async suspend(args: AgentToolSuspendedArgs): Promise<void> {\n    await dispatchTerminalObservers(\n      this.observers,\n      \"tool.suspend\",\n      (observer) => observer.suspend?.(observerSnapshot(args)),\n      this.errorPolicy,\n    );\n  }\n\n  async error(args: AgentToolErrorArgs): Promise<void> {\n    await dispatchTerminalObservers(\n      this.observers,\n      \"tool.error\",\n      (observer) => observer.error?.(observerSnapshot(args)),\n      \"ignore\",\n    );\n  }\n}\n\nasync function dispatchObservers<T>(\n  observers: readonly NamedObserver<T>[],\n  phase: string,\n  dispatch: (observer: T) => void | Promise<void> | undefined,\n  errorPolicy: AgentObserverErrorPolicy,\n): Promise<void> {\n  const failures: AgentObserverFailure[] = [];\n  for (const entry of observers) {\n    try {\n      await dispatch(entry.observer);\n    } catch (error) {\n      failures.push({ observer: entry.name, error });\n    }\n  }\n  throwObserverFailures(phase, failures, errorPolicy);\n}\n\nasync function dispatchTerminalObservers<T>(\n  observers: readonly ActiveNamedObserver<T>[],\n  phase: string,\n  dispatch: (observer: T) => void | Promise<void> | undefined,\n  errorPolicy: AgentObserverErrorPolicy,\n): Promise<void> {\n  const failures = await terminateObservers(observers, dispatch);\n  throwObserverFailures(phase, failures, errorPolicy);\n}\n\nasync function terminateObservers<T>(\n  observers: readonly ActiveNamedObserver<T>[],\n  dispatch: (observer: T) => void | Promise<void> | undefined,\n): Promise<AgentObserverFailure[]> {\n  const failures: AgentObserverFailure[] = [];\n  for (const entry of observers) {\n    if (entry.terminal) continue;\n    entry.terminal = true;\n    try {\n      await dispatch(entry.observer);\n    } catch (error) {\n      failures.push({ observer: entry.name, error });\n    }\n  }\n  return failures;\n}\n\nfunction throwObserverFailures(\n  phase: string,\n  failures: readonly AgentObserverFailure[],\n  errorPolicy: AgentObserverErrorPolicy,\n): void {\n  if (errorPolicy === \"throw\" && failures.length > 0) {\n    throw new AgentObserverDispatchError(phase, failures);\n  }\n}\n"],"mappings":";;;;;AAAO,SAAS,iBAAoB,OAAa;AAC/C,MAAI,cAAc,OAAO,oBAAI,QAAgB,CAAC,GAAG;AAC/C,WAAO,sBAAsB,OAAO,oBAAI,QAAwB,CAAC;AAAA,EACnE;AACA,MAAI;AACF,WAAO,WAAW,gBAAgB,KAAK;AAAA,EACzC,QAAQ;AACN,WAAO,sBAAsB,OAAO,oBAAI,QAAwB,CAAC;AAAA,EACnE;AACF;AAEA,SAAS,sBAAyB,OAAU,MAAkC;AAC5E,MAAI,OAAO,UAAU,YAAY,UAAU,MAAM;AAC/C,WAAO;AAAA,EACT;AACA,QAAM,WAAW,KAAK,IAAI,KAAK;AAC/B,MAAI,aAAa,QAAW;AAC1B,WAAO;AAAA,EACT;AACA,MAAI,MAAM,QAAQ,KAAK,GAAG;AACxB,UAAMA,SAAmB,CAAC;AAC1B,SAAK,IAAI,OAAOA,MAAK;AACrB,IAAAA,OAAM,KAAK,GAAG,MAAM,IAAI,CAAC,SAAS,sBAAsB,MAAM,IAAI,CAAC,CAAC;AACpE,WAAOA;AAAA,EACT;AACA,MAAI,iBAAiB,MAAM;AACzB,WAAO,IAAI,KAAK,MAAM,QAAQ,CAAC;AAAA,EACjC;AACA,MAAI,iBAAiB,OAAO;AAC1B,UAAMA,SAAQ,OAAO,OAAO,OAAO,eAAe,KAAK,CAAC;AACxD,SAAK,IAAI,OAAOA,MAAK;AACrB,WAAO,iBAAiBA,QAAO;AAAA,MAC7B,MAAM,EAAE,cAAc,MAAM,UAAU,MAAM,OAAO,MAAM,KAAK;AAAA,MAC9D,SAAS,EAAE,cAAc,MAAM,UAAU,MAAM,OAAO,MAAM,QAAQ;AAAA,IACtE,CAAC;AACD,QAAI,MAAM,UAAU,QAAW;AAC7B,aAAO,eAAeA,QAAO,SAAS;AAAA,QACpC,cAAc;AAAA,QACd,UAAU;AAAA,QACV,OAAO,MAAM;AAAA,MACf,CAAC;AAAA,IACH;AACA,QAAI,MAAM,UAAU,QAAW;AAC7B,aAAO,eAAeA,QAAO,SAAS;AAAA,QACpC,cAAc;AAAA,QACd,UAAU;AAAA,QACV,OAAO,sBAAsB,MAAM,OAAO,IAAI;AAAA,MAChD,CAAC;AAAA,IACH;AACA,eAAW,OAAO,OAAO,oBAAoB,KAAK,GAAG;AACnD,UAAI,QAAQ,UAAU,QAAQ,aAAa,QAAQ,WAAW,QAAQ,QAAS;AAC/E,YAAM,aAAa,OAAO,yBAAyB,OAAO,GAAG;AAC7D,UAAI,eAAe,OAAW;AAC9B,UAAI,WAAW,YAAY;AACzB,mBAAW,QAAQ,sBAAsB,WAAW,OAAO,IAAI;AAAA,MACjE;AACA,aAAO,eAAeA,QAAO,KAAK,UAAU;AAAA,IAC9C;AACA,WAAOA;AAAA,EACT;AACA,MAAI,OAAO,eAAe,KAAK,MAAM,OAAO,WAAW;AACrD,WAAO;AAAA,EACT;AACA,QAAM,QAAiC,CAAC;AACxC,OAAK,IAAI,OAAO,KAAK;AACrB,aAAW,CAAC,KAAK,IAAI,KAAK,OAAO,QAAQ,KAAK,GAAG;AAC/C,UAAM,GAAG,IAAI,sBAAsB,MAAM,IAAI;AAAA,EAC/C;AACA,SAAO;AACT;AAEA,SAAS,cAAc,OAAgB,MAAgC;AACrE,MAAI,OAAO,UAAU,YAAY,UAAU,KAAM,QAAO;AACxD,MAAI,iBAAiB,MAAO,QAAO;AACnC,MAAI,KAAK,IAAI,KAAK,EAAG,QAAO;AAC5B,OAAK,IAAI,KAAK;AACd,MAAI,MAAM,QAAQ,KAAK,GAAG;AACxB,WAAO,MAAM,KAAK,CAAC,SAAS,cAAc,MAAM,IAAI,CAAC;AAAA,EACvD;AACA,MAAI,OAAO,eAAe,KAAK,MAAM,OAAO,UAAW,QAAO;AAC9D,SAAO,OAAO,OAAO,KAAK,EAAE,KAAK,CAAC,SAAS,cAAc,MAAM,IAAI,CAAC;AACtE;;;ACpDO,IAAM,6BAAN,cAAyC,eAAe;AAAA,EACpD;AAAA,EACA;AAAA,EAET,YAAY,OAAe,UAA2C;AACpE;AAAA,MACE,SAAS,IAAI,CAAC,YAAY,QAAQ,KAAK;AAAA,MACvC,kBAAkB,KAAK,eAAe,SAAS,IAAI,CAAC,YAAY,QAAQ,QAAQ,EAAE,KAAK,IAAI,CAAC;AAAA,IAC9F;AACA,SAAK,OAAO;AACZ,SAAK,QAAQ;AACb,SAAK,WAAW,OAAO,OAAO,CAAC,GAAG,QAAQ,CAAC;AAAA,EAC7C;AACF;AAWA,eAAsB,uBACpB,WACA,MACA,SAIkC;AAClC,QAAM,eAAwD,CAAC;AAC/D,QAAM,WAAmC,CAAC;AAC1C,aAAW,CAAC,MAAM,QAAQ,KAAK,OAAO,QAAQ,SAAS,GAAG;AACxD,QAAI;AACF,YAAM,cAAc,MAAM,SAAS,SAAS,iBAAiB,IAAI,CAAC;AAClE,UAAI,gBAAgB,QAAW;AAC7B,qBAAa,KAAK,EAAE,MAAM,UAAU,aAAa,UAAU,MAAM,CAAC;AAAA,MACpE;AAAA,IACF,SAAS,OAAO;AACd,eAAS,KAAK,EAAE,UAAU,MAAM,MAAM,CAAC;AAAA,IACzC;AAAA,EACF;AACA,MAAI,QAAQ,gBAAgB,WAAW,SAAS,SAAS,GAAG;AAC1D,UAAM,eAAe,IAAI,2BAA2B,YAAY,QAAQ;AACxE,UAAM,kBAAkB,MAAM;AAAA,MAAmB;AAAA,MAAc,CAAC,aAC9D,SAAS;AAAA,QACP,iBAAiB;AAAA,UACf,OAAO;AAAA,UACP,OAAO,MAAM,MAAM;AAAA,UACnB,UAAU,CAAC,GAAG,KAAK,SAAS,KAAK,MAAM;AAAA,QACzC,CAAC;AAAA,MACH;AAAA,IACF;AACA,UAAM,IAAI,2BAA2B,YAAY,CAAC,GAAG,UAAU,GAAG,eAAe,CAAC;AAAA,EACpF;AACA,SAAO,IAAI,wBAAwB,cAAc,OAAO;AAC1D;AAEO,IAAM,0BAAN,MAA8B;AAAA,EAGnC,YACmB,cACA,SAIjB;AALiB;AACA;AAKjB,UAAM,UACJ,QAAQ,iBAAiB,SACrB,SACA,aAAa,KAAK,CAAC,UAAU,MAAM,SAAS,QAAQ,YAAY;AACtE,SAAK,QACH,SAAS,SAAS,UAAU,SACxB,SACA,OAAO,OAAO,EAAE,UAAU,QAAQ,MAAM,GAAG,QAAQ,SAAS,MAAM,CAAC;AAAA,EAC3E;AAAA,EAdmB;AAAA,EACA;AAAA,EAJV;AAAA,EAmBT,MAAM,gBAAgB,MAAoE;AACxF,UAAM,sBAAsE,CAAC;AAC7E,UAAM,WAAmC,CAAC;AAC1C,eAAW,SAAS,KAAK,cAAc;AACrC,UAAI,MAAM,SAAS,oBAAoB,OAAW;AAClD,UAAI;AACF,cAAM,WAAW,MAAM,MAAM,SAAS,gBAAgB,iBAAiB,IAAI,CAAC;AAC5E,YAAI,aAAa,QAAW;AAC1B,8BAAoB,KAAK,EAAE,MAAM,MAAM,MAAM,UAAU,UAAU,MAAM,CAAC;AAAA,QAC1E;AAAA,MACF,SAAS,OAAO;AACd,iBAAS,KAAK,EAAE,UAAU,MAAM,MAAM,MAAM,CAAC;AAAA,MAC/C;AAAA,IACF;AACA,QAAI,KAAK,QAAQ,gBAAgB,WAAW,SAAS,SAAS,GAAG;AAC/D,YAAM,eAAe,IAAI,2BAA2B,mBAAmB,QAAQ;AAC/E,YAAM,kBAAkB,MAAM;AAAA,QAAmB;AAAA,QAAqB,CAAC,aACrE,SAAS,QAAQ,iBAAiB,EAAE,MAAM,KAAK,MAAM,OAAO,aAAa,CAAC,CAAC;AAAA,MAC7E;AACA,YAAM,IAAI,2BAA2B,mBAAmB,CAAC,GAAG,UAAU,GAAG,eAAe,CAAC;AAAA,IAC3F;AACA,WAAO,IAAI,0BAA0B,qBAAqB,KAAK,QAAQ,WAAW;AAAA,EACpF;AAAA,EAEA,MAAM,UAAU,MAAwD;AACtE,UAAM,gBAA0D,CAAC;AACjE,UAAM,WAAmC,CAAC;AAC1C,eAAW,SAAS,KAAK,cAAc;AACrC,UAAI,MAAM,SAAS,cAAc,OAAW;AAC5C,UAAI;AACF,cAAM,WAAW,MAAM,MAAM,SAAS,UAAU,iBAAiB,IAAI,CAAC;AACtE,YAAI,aAAa,QAAW;AAC1B,wBAAc,KAAK,EAAE,MAAM,MAAM,MAAM,UAAU,UAAU,MAAM,CAAC;AAAA,QACpE;AAAA,MACF,SAAS,OAAO;AACd,iBAAS,KAAK,EAAE,UAAU,MAAM,MAAM,MAAM,CAAC;AAAA,MAC/C;AAAA,IACF;AACA,QAAI,KAAK,QAAQ,gBAAgB,WAAW,SAAS,SAAS,GAAG;AAC/D,YAAM,eAAe,IAAI,2BAA2B,aAAa,QAAQ;AACzE,YAAM,kBAAkB,MAAM;AAAA,QAAmB;AAAA,QAAe,CAAC,aAC/D,SAAS,QAAQ,iBAAiB,EAAE,GAAG,MAAM,OAAO,aAAa,CAAC,CAAC;AAAA,MACrE;AACA,YAAM,IAAI,2BAA2B,aAAa,CAAC,GAAG,UAAU,GAAG,eAAe,CAAC;AAAA,IACrF;AACA,WAAO,IAAI,oBAAoB,eAAe,KAAK,QAAQ,WAAW;AAAA,EACxE;AAAA,EAEA,MAAM,IAAI,MAAsC;AAC9C,UAAM;AAAA,MACJ,KAAK;AAAA,MACL;AAAA,MACA,CAAC,aAAa,SAAS,IAAI,iBAAiB,IAAI,CAAC;AAAA,MACjD,KAAK,QAAQ;AAAA,IACf;AAAA,EACF;AAAA,EAEA,MAAM,MAAM,MAAwC;AAClD,UAAM;AAAA,MACJ,KAAK;AAAA,MACL;AAAA,MACA,CAAC,aAAa,SAAS,QAAQ,iBAAiB,IAAI,CAAC;AAAA,MACrD;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,MAAM,MAAwC;AAClD,UAAM;AAAA,MACJ,KAAK;AAAA,MACL;AAAA,MACA,CAAC,aAAa,SAAS,QAAQ,iBAAiB,IAAI,CAAC;AAAA,MACrD,KAAK,QAAQ;AAAA,IACf;AAAA,EACF;AACF;AAEO,IAAM,4BAAN,MAAgC;AAAA,EACrC,YACmB,WACA,aACjB;AAFiB;AACA;AAAA,EAChB;AAAA,EAFgB;AAAA,EACA;AAAA,EAGnB,MAAM,IAAI,MAA6C;AACrD,UAAM;AAAA,MACJ,KAAK;AAAA,MACL;AAAA,MACA,CAAC,aAAa,SAAS,IAAI,iBAAiB,IAAI,CAAC;AAAA,MACjD,KAAK;AAAA,IACP;AAAA,EACF;AAAA,EAEA,MAAM,MAAM,MAA+C;AACzD,UAAM;AAAA,MACJ,KAAK;AAAA,MACL;AAAA,MACA,CAAC,aAAa,SAAS,QAAQ,iBAAiB,IAAI,CAAC;AAAA,MACrD;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,OAAO,MAAgD;AAC3D,UAAM;AAAA,MACJ,KAAK;AAAA,MACL;AAAA,MACA,CAAC,aAAa,SAAS,SAAS,iBAAiB,IAAI,CAAC;AAAA,MACtD,KAAK;AAAA,IACP;AAAA,EACF;AACF;AAEO,IAAM,sBAAN,MAA0B;AAAA,EAC/B,YACmB,WACA,aACjB;AAFiB;AACA;AAAA,EAChB;AAAA,EAFgB;AAAA,EACA;AAAA,EAGnB,MAAM,YAAY,MAA+C;AAC/D,UAAM;AAAA,MACJ,KAAK;AAAA,MACL;AAAA,MACA,CAAC,aAAa,SAAS,cAAc,iBAAiB,IAAI,CAAC;AAAA,MAC3D,KAAK;AAAA,IACP;AAAA,EACF;AAAA,EAEA,MAAM,IAAI,MAAuC;AAC/C,UAAM;AAAA,MACJ,KAAK;AAAA,MACL;AAAA,MACA,CAAC,aAAa,SAAS,IAAI,iBAAiB,IAAI,CAAC;AAAA,MACjD,KAAK;AAAA,IACP;AAAA,EACF;AAAA,EAEA,MAAM,QAAQ,MAA6C;AACzD,UAAM;AAAA,MACJ,KAAK;AAAA,MACL;AAAA,MACA,CAAC,aAAa,SAAS,UAAU,iBAAiB,IAAI,CAAC;AAAA,MACvD,KAAK;AAAA,IACP;AAAA,EACF;AAAA,EAEA,MAAM,MAAM,MAAyC;AACnD,UAAM;AAAA,MACJ,KAAK;AAAA,MACL;AAAA,MACA,CAAC,aAAa,SAAS,QAAQ,iBAAiB,IAAI,CAAC;AAAA,MACrD;AAAA,IACF;AAAA,EACF;AACF;AAEA,eAAe,kBACb,WACA,OACA,UACA,aACe;AACf,QAAM,WAAmC,CAAC;AAC1C,aAAW,SAAS,WAAW;AAC7B,QAAI;AACF,YAAM,SAAS,MAAM,QAAQ;AAAA,IAC/B,SAAS,OAAO;AACd,eAAS,KAAK,EAAE,UAAU,MAAM,MAAM,MAAM,CAAC;AAAA,IAC/C;AAAA,EACF;AACA,wBAAsB,OAAO,UAAU,WAAW;AACpD;AAEA,eAAe,0BACb,WACA,OACA,UACA,aACe;AACf,QAAM,WAAW,MAAM,mBAAmB,WAAW,QAAQ;AAC7D,wBAAsB,OAAO,UAAU,WAAW;AACpD;AAEA,eAAe,mBACb,WACA,UACiC;AACjC,QAAM,WAAmC,CAAC;AAC1C,aAAW,SAAS,WAAW;AAC7B,QAAI,MAAM,SAAU;AACpB,UAAM,WAAW;AACjB,QAAI;AACF,YAAM,SAAS,MAAM,QAAQ;AAAA,IAC/B,SAAS,OAAO;AACd,eAAS,KAAK,EAAE,UAAU,MAAM,MAAM,MAAM,CAAC;AAAA,IAC/C;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,sBACP,OACA,UACA,aACM;AACN,MAAI,gBAAgB,WAAW,SAAS,SAAS,GAAG;AAClD,UAAM,IAAI,2BAA2B,OAAO,QAAQ;AAAA,EACtD;AACF;","names":["clone"]}

package/dist/chunk-ZCPOIDAQ.js

@@ -0,0 +1,127 @@
+import {
+  CompletionCapabilityError,
+  Usage,
+  generateCompletion
+} from "./chunk-D2ECOWN5.js";
+import {
+  toProviderJsonSchema
+} from "./chunk-PK5LKOFL.js";
+import {
+  isAbortError,
+  resolveRetryOptions,
+  retryDelayMs,
+  retryOptionsForFailure,
+  waitForRetry
+} from "./chunk-FN7HLLAZ.js";
+
+// src/extractor/extractor.ts
+import { z } from "zod";
+var SUBMIT_TOOL_NAME = "submit";
+var DEFAULT_EXTRACTOR_INSTRUCTIONS = "You are an AI assistant whose purpose is to extract structured data from the provided text.\nYou have access to a `submit` function that defines the structure of the data to extract.\nAlways call the `submit` function with the structured data. Use default or null values when information is missing.";
+var ExtractionError = class extends Error {
+  attempts;
+  usage;
+  constructor(message, options) {
+    super(message, options.cause === void 0 ? void 0 : { cause: options.cause });
+    this.name = "ExtractionError";
+    this.attempts = options.attempts;
+    this.usage = options.usage;
+  }
+};
+async function extract(options) {
+  if (!(options.outputSchema instanceof z.ZodType)) {
+    throw new TypeError("extract outputSchema must be a Zod schema.");
+  }
+  if (typeof options.text !== "string") {
+    throw new TypeError("extract text must be a string.");
+  }
+  const retries = resolveExtractionRetries(options.retries);
+  const maxAttempts = retries?.maxAttempts ?? 1;
+  const submitTool = createSubmitTool(options.outputSchema);
+  const instructions = extractionInstructions(options.instructions);
+  let usage = Usage.empty();
+  let lastError;
+  for (let attempt = 1; attempt <= maxAttempts; attempt += 1) {
+    try {
+      const result = await generateCompletion({
+        model: options.model,
+        prompt: options.text,
+        instructions,
+        tools: [globalThis.structuredClone(submitTool)],
+        temperature: options.temperature,
+        maxTokens: options.maxTokens,
+        providerOptions: options.providerOptions,
+        toolChoice: "required",
+        retries: false,
+        abortSignal: options.abortSignal
+      });
+      usage = Usage.add(usage, result.usage);
+      return {
+        ...result,
+        output: extractSubmittedData(result.content, options.outputSchema),
+        usage
+      };
+    } catch (error) {
+      if (error instanceof CompletionCapabilityError || isAbortError(error)) {
+        throw error;
+      }
+      lastError = error;
+      const retryOptions = retryOptionsForFailure(retries, {
+        error,
+        attempt,
+        streaming: false
+      });
+      if (retryOptions === void 0) {
+        throw new ExtractionError("No data extracted", {
+          cause: lastError,
+          attempts: attempt,
+          usage
+        });
+      }
+      await waitForRetry(retryDelayMs(retryOptions, attempt), options.abortSignal);
+    }
+  }
+  throw new ExtractionError("No data extracted", {
+    cause: lastError,
+    attempts: maxAttempts,
+    usage
+  });
+}
+function extractionInstructions(instructions) {
+  return instructions === void 0 || instructions.length === 0 ? DEFAULT_EXTRACTOR_INSTRUCTIONS : `${DEFAULT_EXTRACTOR_INSTRUCTIONS}
+
+${instructions}`;
+}
+function createSubmitTool(outputSchema) {
+  return {
+    name: SUBMIT_TOOL_NAME,
+    description: "Submit the structured data extracted from the provided text.",
+    parameters: toProviderJsonSchema(outputSchema, { io: "input" })
+  };
+}
+function resolveExtractionRetries(options) {
+  if (options === void 0 || options === false) {
+    return void 0;
+  }
+  return resolveRetryOptions({
+    ...options,
+    shouldRetry: options.shouldRetry ?? (() => true)
+  });
+}
+function extractSubmittedData(content, schema) {
+  const submitted = content.filter((item) => item.type === "tool-call").filter((toolCall) => toolCall.toolName === SUBMIT_TOOL_NAME).at(-1);
+  if (submitted === void 0) {
+    throw new Error("The model did not call the submit tool");
+  }
+  try {
+    return schema.parse(submitted.input);
+  } catch (error) {
+    throw new Error("Submitted data failed output schema validation", { cause: error });
+  }
+}
+
+export {
+  ExtractionError,
+  extract
+};
+//# sourceMappingURL=chunk-ZCPOIDAQ.js.map

package/dist/chunk-ZCPOIDAQ.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/extractor/extractor.ts"],"sourcesContent":["import { z } from \"zod\";\nimport {\n  type AssistantContentPart,\n  CompletionCapabilityError,\n  type CompletionModel,\n  type CompletionResult,\n  generateCompletion,\n  type JsonObject,\n  type ToolDefinition,\n  Usage,\n} from \"../completion/index\";\nimport { isAbortError } from \"../internal/abort\";\nimport {\n  type ResolvedRetryOptions,\n  type RetrySetting,\n  resolveRetryOptions,\n  retryDelayMs,\n  retryOptionsForFailure,\n  waitForRetry,\n} from \"../retry\";\nimport { toProviderJsonSchema, type ZodSchema } from \"../schema/zod-schema\";\n\nconst SUBMIT_TOOL_NAME = \"submit\";\n\nconst DEFAULT_EXTRACTOR_INSTRUCTIONS =\n  \"You are an AI assistant whose purpose is to extract structured data from the provided text.\\n\" +\n  \"You have access to a `submit` function that defines the structure of the data to extract.\\n\" +\n  \"Always call the `submit` function with the structured data. Use default or null values when information is missing.\";\n\ntype RawResponseOf<Model> =\n  Model extends CompletionModel<infer RawResponse> ? RawResponse : unknown;\n\nexport type ExtractOptions<Output, Model extends CompletionModel = CompletionModel> = {\n  model: Model;\n  text: string;\n  outputSchema: ZodSchema<Output>;\n  instructions?: string | undefined;\n  retries?: RetrySetting | undefined;\n  temperature?: number | undefined;\n  maxTokens?: number | undefined;\n  providerOptions?: JsonObject | undefined;\n  abortSignal?: AbortSignal | undefined;\n};\n\nexport type ExtractionResult<Output, RawResponse = unknown> = Omit<\n  CompletionResult<string, RawResponse>,\n  \"output\"\n> & {\n  output: Output;\n};\n\nexport class ExtractionError extends Error {\n  readonly attempts: number;\n  readonly usage: Usage;\n\n  constructor(\n    message: string,\n    options: {\n      cause?: unknown;\n      attempts: number;\n      usage: Usage;\n    },\n  ) {\n    super(message, options.cause === undefined ? undefined : { cause: options.cause });\n    this.name = \"ExtractionError\";\n    this.attempts = options.attempts;\n    this.usage = options.usage;\n  }\n}\n\n/** Extract structured data from text using a required, schema-backed submit tool. */\nexport async function extract<Output, Model extends CompletionModel>(\n  options: ExtractOptions<Output, Model>,\n): Promise<ExtractionResult<Output, RawResponseOf<Model>>> {\n  if (!(options.outputSchema instanceof z.ZodType)) {\n    throw new TypeError(\"extract outputSchema must be a Zod schema.\");\n  }\n  if (typeof options.text !== \"string\") {\n    throw new TypeError(\"extract text must be a string.\");\n  }\n\n  const retries = resolveExtractionRetries(options.retries);\n  const maxAttempts = retries?.maxAttempts ?? 1;\n  const submitTool = createSubmitTool(options.outputSchema);\n  const instructions = extractionInstructions(options.instructions);\n  let usage = Usage.empty();\n  let lastError: unknown;\n\n  for (let attempt = 1; attempt <= maxAttempts; attempt += 1) {\n    try {\n      const result = await generateCompletion({\n        model: options.model,\n        prompt: options.text,\n        instructions,\n        tools: [globalThis.structuredClone(submitTool)],\n        temperature: options.temperature,\n        maxTokens: options.maxTokens,\n        providerOptions: options.providerOptions,\n        toolChoice: \"required\",\n        retries: false,\n        abortSignal: options.abortSignal,\n      });\n      usage = Usage.add(usage, result.usage);\n      return {\n        ...result,\n        output: extractSubmittedData(result.content, options.outputSchema),\n        usage,\n      };\n    } catch (error) {\n      if (error instanceof CompletionCapabilityError || isAbortError(error)) {\n        throw error;\n      }\n      lastError = error;\n      const retryOptions = retryOptionsForFailure(retries, {\n        error,\n        attempt,\n        streaming: false,\n      });\n      if (retryOptions === undefined) {\n        throw new ExtractionError(\"No data extracted\", {\n          cause: lastError,\n          attempts: attempt,\n          usage,\n        });\n      }\n      await waitForRetry(retryDelayMs(retryOptions, attempt), options.abortSignal);\n    }\n  }\n\n  throw new ExtractionError(\"No data extracted\", {\n    cause: lastError,\n    attempts: maxAttempts,\n    usage,\n  });\n}\n\nfunction extractionInstructions(instructions: string | undefined): string {\n  return instructions === undefined || instructions.length === 0\n    ? DEFAULT_EXTRACTOR_INSTRUCTIONS\n    : `${DEFAULT_EXTRACTOR_INSTRUCTIONS}\\n\\n${instructions}`;\n}\n\nfunction createSubmitTool<Output>(outputSchema: ZodSchema<Output>): ToolDefinition {\n  return {\n    name: SUBMIT_TOOL_NAME,\n    description: \"Submit the structured data extracted from the provided text.\",\n    parameters: toProviderJsonSchema(outputSchema, { io: \"input\" }),\n  };\n}\n\nfunction resolveExtractionRetries(\n  options: RetrySetting | undefined,\n): ResolvedRetryOptions | undefined {\n  if (options === undefined || options === false) {\n    return undefined;\n  }\n  return resolveRetryOptions({\n    ...options,\n    shouldRetry: options.shouldRetry ?? (() => true),\n  });\n}\n\nfunction extractSubmittedData<Output>(\n  content: readonly AssistantContentPart[],\n  schema: ZodSchema<Output>,\n): Output {\n  const submitted = content\n    .filter((item) => item.type === \"tool-call\")\n    .filter((toolCall) => toolCall.toolName === SUBMIT_TOOL_NAME)\n    .at(-1);\n\n  if (submitted === undefined) {\n    throw new Error(\"The model did not call the submit tool\");\n  }\n\n  try {\n    return schema.parse(submitted.input);\n  } catch (error) {\n    throw new Error(\"Submitted data failed output schema validation\", { cause: error });\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;AAAA,SAAS,SAAS;AAsBlB,IAAM,mBAAmB;AAEzB,IAAM,iCACJ;AA0BK,IAAM,kBAAN,cAA8B,MAAM;AAAA,EAChC;AAAA,EACA;AAAA,EAET,YACE,SACA,SAKA;AACA,UAAM,SAAS,QAAQ,UAAU,SAAY,SAAY,EAAE,OAAO,QAAQ,MAAM,CAAC;AACjF,SAAK,OAAO;AACZ,SAAK,WAAW,QAAQ;AACxB,SAAK,QAAQ,QAAQ;AAAA,EACvB;AACF;AAGA,eAAsB,QACpB,SACyD;AACzD,MAAI,EAAE,QAAQ,wBAAwB,EAAE,UAAU;AAChD,UAAM,IAAI,UAAU,4CAA4C;AAAA,EAClE;AACA,MAAI,OAAO,QAAQ,SAAS,UAAU;AACpC,UAAM,IAAI,UAAU,gCAAgC;AAAA,EACtD;AAEA,QAAM,UAAU,yBAAyB,QAAQ,OAAO;AACxD,QAAM,cAAc,SAAS,eAAe;AAC5C,QAAM,aAAa,iBAAiB,QAAQ,YAAY;AACxD,QAAM,eAAe,uBAAuB,QAAQ,YAAY;AAChE,MAAI,QAAQ,MAAM,MAAM;AACxB,MAAI;AAEJ,WAAS,UAAU,GAAG,WAAW,aAAa,WAAW,GAAG;AAC1D,QAAI;AACF,YAAM,SAAS,MAAM,mBAAmB;AAAA,QACtC,OAAO,QAAQ;AAAA,QACf,QAAQ,QAAQ;AAAA,QAChB;AAAA,QACA,OAAO,CAAC,WAAW,gBAAgB,UAAU,CAAC;AAAA,QAC9C,aAAa,QAAQ;AAAA,QACrB,WAAW,QAAQ;AAAA,QACnB,iBAAiB,QAAQ;AAAA,QACzB,YAAY;AAAA,QACZ,SAAS;AAAA,QACT,aAAa,QAAQ;AAAA,MACvB,CAAC;AACD,cAAQ,MAAM,IAAI,OAAO,OAAO,KAAK;AACrC,aAAO;AAAA,QACL,GAAG;AAAA,QACH,QAAQ,qBAAqB,OAAO,SAAS,QAAQ,YAAY;AAAA,QACjE;AAAA,MACF;AAAA,IACF,SAAS,OAAO;AACd,UAAI,iBAAiB,6BAA6B,aAAa,KAAK,GAAG;AACrE,cAAM;AAAA,MACR;AACA,kBAAY;AACZ,YAAM,eAAe,uBAAuB,SAAS;AAAA,QACnD;AAAA,QACA;AAAA,QACA,WAAW;AAAA,MACb,CAAC;AACD,UAAI,iBAAiB,QAAW;AAC9B,cAAM,IAAI,gBAAgB,qBAAqB;AAAA,UAC7C,OAAO;AAAA,UACP,UAAU;AAAA,UACV;AAAA,QACF,CAAC;AAAA,MACH;AACA,YAAM,aAAa,aAAa,cAAc,OAAO,GAAG,QAAQ,WAAW;AAAA,IAC7E;AAAA,EACF;AAEA,QAAM,IAAI,gBAAgB,qBAAqB;AAAA,IAC7C,OAAO;AAAA,IACP,UAAU;AAAA,IACV;AAAA,EACF,CAAC;AACH;AAEA,SAAS,uBAAuB,cAA0C;AACxE,SAAO,iBAAiB,UAAa,aAAa,WAAW,IACzD,iCACA,GAAG,8BAA8B;AAAA;AAAA,EAAO,YAAY;AAC1D;AAEA,SAAS,iBAAyB,cAAiD;AACjF,SAAO;AAAA,IACL,MAAM;AAAA,IACN,aAAa;AAAA,IACb,YAAY,qBAAqB,cAAc,EAAE,IAAI,QAAQ,CAAC;AAAA,EAChE;AACF;AAEA,SAAS,yBACP,SACkC;AAClC,MAAI,YAAY,UAAa,YAAY,OAAO;AAC9C,WAAO;AAAA,EACT;AACA,SAAO,oBAAoB;AAAA,IACzB,GAAG;AAAA,IACH,aAAa,QAAQ,gBAAgB,MAAM;AAAA,EAC7C,CAAC;AACH;AAEA,SAAS,qBACP,SACA,QACQ;AACR,QAAM,YAAY,QACf,OAAO,CAAC,SAAS,KAAK,SAAS,WAAW,EAC1C,OAAO,CAAC,aAAa,SAAS,aAAa,gBAAgB,EAC3D,GAAG,EAAE;AAER,MAAI,cAAc,QAAW;AAC3B,UAAM,IAAI,MAAM,wCAAwC;AAAA,EAC1D;AAEA,MAAI;AACF,WAAO,OAAO,MAAM,UAAU,KAAK;AAAA,EACrC,SAAS,OAAO;AACd,UAAM,IAAI,MAAM,kDAAkD,EAAE,OAAO,MAAM,CAAC;AAAA,EACpF;AACF;","names":[]}

package/dist/client-DvUatElp.d.ts

@@ -0,0 +1,84 @@
+import { IOType } from 'node:child_process';
+import { Stream } from 'node:stream';
+import { OAuthClientProvider } from '@modelcontextprotocol/sdk/client/auth.js';
+import { StreamableHTTPReconnectionOptions } from '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import { Transport } from '@modelcontextprotocol/sdk/shared/transport.js';
+import { J as JsonObject } from './types-D9rKQ_as.js';
+import { A as AnyTool } from './tool-r62gf3QY.js';
+
+type McpStdioTransport = {
+    readonly type: "stdio";
+    readonly command: string;
+    readonly args?: string[] | undefined;
+    readonly env?: Record<string, string> | undefined;
+    readonly cwd?: string | undefined;
+    readonly stderr?: IOType | Stream | number | undefined;
+    readonly maxBufferSize?: number | undefined;
+};
+type McpStreamableHttpTransport = {
+    readonly type: "streamableHttp";
+    readonly url: string | URL;
+    readonly ssrfProtection?: "strict" | "disabled" | undefined;
+    readonly headers?: Readonly<Record<string, string>> | undefined;
+    readonly authProvider?: OAuthClientProvider | undefined;
+    readonly reconnectionOptions?: StreamableHTTPReconnectionOptions | undefined;
+    readonly sessionId?: string | undefined;
+};
+type McpTransport = Transport;
+type McpCustomTransport = {
+    readonly type: "custom";
+    readonly create: (options: {
+        abortSignal?: AbortSignal | undefined;
+    }) => McpTransport | Promise<McpTransport>;
+};
+type McpClientTransport = McpStdioTransport | McpStreamableHttpTransport | McpCustomTransport;
+type McpClientOptions = {
+    readonly name: string;
+    readonly transport: McpClientTransport;
+    readonly tools?: {
+        readonly prefix?: string | undefined;
+    };
+};
+type McpConnectOptions = {
+    readonly abortSignal?: AbortSignal | undefined;
+};
+type McpServerInfo = {
+    readonly name: string;
+    readonly version: string;
+    readonly title?: string | undefined;
+    readonly description?: string | undefined;
+    readonly websiteUrl?: string | undefined;
+    readonly icons?: readonly {
+        readonly src: string;
+        readonly mimeType?: string | undefined;
+        readonly sizes?: readonly string[] | undefined;
+        readonly theme?: "light" | "dark" | undefined;
+    }[];
+};
+type McpTool = AnyTool & {
+    readonly mcp: {
+        readonly serverName: string;
+        readonly remoteName: string;
+    };
+};
+type McpServer = {
+    readonly name: string;
+    readonly tools: readonly McpTool[];
+    readonly serverInfo?: McpServerInfo | undefined;
+    readonly capabilities?: JsonObject | undefined;
+    readonly instructions?: string | undefined;
+};
+type McpClientGroupConnectOptions = {
+    readonly clients: readonly McpClient[];
+    readonly abortSignal?: AbortSignal | undefined;
+};
+
+declare class McpClient {
+    #private;
+    readonly name: string;
+    constructor(options: McpClientOptions);
+    connect(options?: McpConnectOptions): Promise<McpServer>;
+    close(): Promise<void>;
+}
+
+export { McpClient as M, type McpServer as a, type McpClientGroupConnectOptions as b, type McpTool as c, type McpClientOptions as d, type McpClientTransport as e, type McpConnectOptions as f, type McpCustomTransport as g, type McpServerInfo as h, type McpStdioTransport as i, type McpStreamableHttpTransport as j, type McpTransport as k };

package/dist/completion/index.d.ts

@@ -1,7 +1,8 @@
-export { R as RetryContext, a as RetryOptions } from '../retry-D3Ruy-ba.js';
-export { C as CreateCompletionBaseOptions, a as CreateCompletionInput, b as CreateCompletionOptions, c as CreateCompletionResult, d as CreateCompletionStreamOptions, e as CreateParsedCompletionOptions, f as CreateParsedCompletionResult, g as createCompletion, h as createCompletionStream, i as createParsedCompletion, j as isJsonValue, k as isStreamingCompletionModel } from '../json-Zo7K3xmt.js';
-import { D as Document, M as Message } from '../types-B3A4YxaA.js';
-export { A as AssistantContent, a as AssistantGenerationMetadata, b as AssistantMessage, c as AssistantMessageOptions, H as CompletionCapabilityError, C as CompletionModel, K as CompletionModelCapabilities, d as CompletionModelInfo, e as CompletionModelMetadataOptions, f as CompletionRequest, g as CompletionResponse, h as CompletionSource, L as CompletionStreamEvent, i as CompletionTool, j as ContextUsage, N as DocumentContent, I as ImageContent, O as ImageDetail, J as JsonObject, k as JsonPrimitive, l as JsonValue, m as MessageOptions, n as ModelContextLimits, P as ProviderTool, o as ProviderToolCall, R as Reasoning, Q as ReasoningContent, V as ReasoningContentType, W as StreamingCompletionModel, S as SystemMessage, T as Text, p as ToolCall, X as ToolCallArgumentsMode, Y as ToolChoice, q as ToolContent, r as ToolDefinition, Z as ToolFunction, s as ToolMessage, t as ToolResult, u as ToolResultContent, v as ToolResultMessageOptions, w as ToolResultOptions, U as Usage, _ as UsageDetails, x as UserContent, y as UserMessage, $ as assertCompletionRequestSupported, z as calculateContextUsage, B as getAssistantGenerationMetadata, E as isProviderTool, a0 as isToolResultContentArray, a1 as reasoningDisplayText, F as resolveCompletionModelInfo, a2 as serializeToolResultOutput, a3 as textFromAssistantContent, G as withContextUsage } from '../types-B3A4YxaA.js';
+export { M as ModelCallOptions } from '../model-call-options-CZkSw_xN.js';
+export { R as RetryContext, a as RetryOptions, b as RetrySetting } from '../retry-CjvSlKGW.js';
+import { D as Document, M as Message } from '../types-D9rKQ_as.js';
+export { A as AssistantContentPart, a as AssistantGenerationMetadata, b as AssistantMessage, O as CompletionCapabilityError, C as CompletionModel, Q as CompletionModelCapabilities, c as CompletionModelInfo, d as CompletionModelStreamEvent, e as CompletionRequest, f as CompletionResponse, g as CompletionResult, h as CompletionSource, i as CompletionStreamEvent, V as CompletionStreamPart, j as CompletionTool, k as ContextUsage, F as FileData, l as FilePart, I as ImageDetail, m as ImagePart, J as JsonObject, n as JsonPrimitive, o as JsonValue, p as ModelContextLimits, P as ProviderTool, q as ProviderToolCall, R as ReasoningContentType, r as ReasoningDetail, s as ReasoningPart, W as StreamingCompletionModel, S as SystemMessage, T as TextPart, X as ToolApprovalResponsePart, Y as ToolCallArgumentsMode, t as ToolCallPart, u as ToolChoice, v as ToolDefinition, Z as ToolInteractionResponsePart, w as ToolMessage, _ as ToolQuestionAnswer, $ as ToolQuestionResponsePart, x as ToolResultContentPart, y as ToolResultOutput, z as ToolResultPart, U as Usage, a0 as UsageDetails, B as UserContentPart, E as UserMessage, a1 as assertCompletionRequestSupported, G as calculateContextUsage, H as getAssistantGenerationMetadata, K as isProviderTool, a2 as reasoningDisplayText, L as resolveModelContextLimits, a3 as textFromAssistantContent, N as withContextUsage } from '../types-D9rKQ_as.js';
+export { C as CompletionBaseOptions, a as CompletionInput, G as GenerateCompletionOptions, b as GenerateStructuredCompletionOptions, M as MessageMetadataSchema, S as StreamCompletionOptions, c as StreamStructuredCompletionOptions, d as createMessageSchema, g as generateCompletion, i as isJsonValue, e as isMessage, j as isStreamingCompletionModel, m as messageSchema, f as messagesSchema, p as parseMessage, h as parseMessages, s as streamCompletion } from '../message-schema-B2M7AYzR.js';
 import '../zod-schema-C7F4clpm.js';
 import 'zod';