@anvia/core 0.1.0

package/dist/chunk-GNWMOSNR.js

@@ -0,0 +1,113 @@
+// src/tool/errors.ts
+var ToolCallError = class extends Error {
+  constructor(message, cause) {
+    super(message);
+    this.cause = cause;
+    this.name = "ToolCallError";
+  }
+  cause;
+};
+var ToolNotFoundError = class extends Error {
+  constructor(toolName) {
+    super(`Tool not found: ${toolName}`);
+    this.toolName = toolName;
+    this.name = "ToolNotFoundError";
+  }
+  toolName;
+};
+var ToolJsonError = class extends Error {
+  constructor(message, cause) {
+    super(message);
+    this.cause = cause;
+    this.name = "ToolJsonError";
+  }
+  cause;
+};
+
+// src/tool/tool.ts
+function serializeToolOutput(output) {
+  if (typeof output === "string") {
+    return output;
+  }
+  const serialized = JSON.stringify(output);
+  return serialized === void 0 ? String(output) : serialized;
+}
+function parseToolArgs(args) {
+  if (args.trim() === "") {
+    return {};
+  }
+  return JSON.parse(args);
+}
+
+// src/tool/tool-set.ts
+var ToolSet = class _ToolSet {
+  tools = /* @__PURE__ */ new Map();
+  static fromTools(tools) {
+    const toolSet = new _ToolSet();
+    for (const tool of tools) {
+      toolSet.addTool(tool);
+    }
+    return toolSet;
+  }
+  addTool(tool) {
+    this.tools.set(tool.name, tool);
+    return this;
+  }
+  addTools(tools) {
+    const values = Array.isArray(tools) ? tools : tools.values();
+    for (const tool of values) {
+      this.addTool(tool);
+    }
+    return this;
+  }
+  deleteTool(toolName) {
+    return this.tools.delete(toolName);
+  }
+  contains(toolName) {
+    return this.tools.has(toolName);
+  }
+  get(toolName) {
+    return this.tools.get(toolName);
+  }
+  values() {
+    return [...this.tools.values()];
+  }
+  async getToolDefinitions(prompt = "") {
+    const defs = [];
+    for (const tool of this.tools.values()) {
+      defs.push(await tool.definition(prompt));
+    }
+    return defs;
+  }
+  async call(toolName, args) {
+    const tool = this.tools.get(toolName);
+    if (tool === void 0) {
+      throw new ToolNotFoundError(toolName);
+    }
+    let parsedArgs;
+    try {
+      parsedArgs = parseToolArgs(args);
+    } catch (error) {
+      throw new ToolJsonError(`Invalid JSON arguments for tool ${toolName}`, error);
+    }
+    try {
+      const output = await tool.call(parsedArgs);
+      return serializeToolOutput(output);
+    } catch (error) {
+      if (error instanceof Error) {
+        throw new ToolCallError(error.message, error);
+      }
+      throw new ToolCallError(`Tool ${toolName} failed`, error);
+    }
+  }
+};
+
+export {
+  ToolCallError,
+  ToolNotFoundError,
+  ToolJsonError,
+  serializeToolOutput,
+  parseToolArgs,
+  ToolSet
+};
+//# sourceMappingURL=chunk-GNWMOSNR.js.map

package/dist/chunk-GNWMOSNR.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/tool/errors.ts","../src/tool/tool.ts","../src/tool/tool-set.ts"],"sourcesContent":["export class ToolCallError extends Error {\n  constructor(\n    message: string,\n    readonly cause?: unknown,\n  ) {\n    super(message);\n    this.name = \"ToolCallError\";\n  }\n}\n\nexport class ToolNotFoundError extends Error {\n  constructor(readonly toolName: string) {\n    super(`Tool not found: ${toolName}`);\n    this.name = \"ToolNotFoundError\";\n  }\n}\n\nexport class ToolJsonError extends Error {\n  constructor(\n    message: string,\n    readonly cause?: unknown,\n  ) {\n    super(message);\n    this.name = \"ToolJsonError\";\n  }\n}\n","import type { JsonObject, JsonValue, ToolDefinition } from \"../completion/types\";\n\nexport type ToolApprovalRunContext = {\n  agentId: string;\n  runId: string;\n  sessionId?: string;\n  metadata?: JsonObject;\n};\n\nexport type ToolApprovalContext<Args = unknown> = {\n  toolName: string;\n  args: Args;\n  rawArgs: string;\n  toolCallId?: string;\n  internalCallId: string;\n  run: ToolApprovalRunContext;\n};\n\nexport type ToolApprovalPolicy<Args = unknown> = {\n  when(ctx: ToolApprovalContext<Args>): boolean | Promise<boolean>;\n  reason?: string | ((ctx: ToolApprovalContext<Args>) => string | Promise<string>);\n  rejectMessage?: string | ((ctx: ToolApprovalContext<Args>) => string | Promise<string>);\n};\n\nexport interface Tool<Args = unknown, Output = unknown> {\n  readonly name: string;\n  readonly approval?: ToolApprovalPolicy<Args>;\n  definition(prompt: string): ToolDefinition | Promise<ToolDefinition>;\n  call(args: Args): Output | Promise<Output>;\n  parseApprovalArgs?(args: unknown): Args;\n}\n\nexport type AnyTool = Omit<Tool<unknown, unknown>, \"approval\"> & {\n  readonly approval?: unknown;\n};\n\nexport function serializeToolOutput(output: unknown): string {\n  if (typeof output === \"string\") {\n    return output;\n  }\n\n  const serialized = JSON.stringify(output);\n  return serialized === undefined ? String(output) : serialized;\n}\n\nexport function parseToolArgs(args: string): JsonValue {\n  if (args.trim() === \"\") {\n    return {};\n  }\n\n  return JSON.parse(args) as JsonValue;\n}\n","import type { ToolDefinition } from \"../completion/types\";\nimport { ToolCallError, ToolJsonError, ToolNotFoundError } from \"./errors\";\nimport { type AnyTool, parseToolArgs, serializeToolOutput } from \"./tool\";\n\nexport class ToolSet {\n  private readonly tools = new Map<string, AnyTool>();\n\n  static fromTools(tools: AnyTool[]): ToolSet {\n    const toolSet = new ToolSet();\n    for (const tool of tools) {\n      toolSet.addTool(tool);\n    }\n    return toolSet;\n  }\n\n  addTool(tool: AnyTool): this {\n    this.tools.set(tool.name, tool);\n    return this;\n  }\n\n  addTools(tools: AnyTool[] | ToolSet): this {\n    const values = Array.isArray(tools) ? tools : tools.values();\n    for (const tool of values) {\n      this.addTool(tool);\n    }\n    return this;\n  }\n\n  deleteTool(toolName: string): boolean {\n    return this.tools.delete(toolName);\n  }\n\n  contains(toolName: string): boolean {\n    return this.tools.has(toolName);\n  }\n\n  get(toolName: string): AnyTool | undefined {\n    return this.tools.get(toolName);\n  }\n\n  values(): AnyTool[] {\n    return [...this.tools.values()];\n  }\n\n  async getToolDefinitions(prompt = \"\"): Promise<ToolDefinition[]> {\n    const defs: ToolDefinition[] = [];\n    for (const tool of this.tools.values()) {\n      defs.push(await tool.definition(prompt));\n    }\n    return defs;\n  }\n\n  async call(toolName: string, args: string): Promise<string> {\n    const tool = this.tools.get(toolName);\n    if (tool === undefined) {\n      throw new ToolNotFoundError(toolName);\n    }\n\n    let parsedArgs: unknown;\n    try {\n      parsedArgs = parseToolArgs(args);\n    } catch (error) {\n      throw new ToolJsonError(`Invalid JSON arguments for tool ${toolName}`, error);\n    }\n\n    try {\n      const output = await tool.call(parsedArgs);\n      return serializeToolOutput(output);\n    } catch (error) {\n      if (error instanceof Error) {\n        throw new ToolCallError(error.message, error);\n      }\n      throw new ToolCallError(`Tool ${toolName} failed`, error);\n    }\n  }\n}\n"],"mappings":";AAAO,IAAM,gBAAN,cAA4B,MAAM;AAAA,EACvC,YACE,SACS,OACT;AACA,UAAM,OAAO;AAFJ;AAGT,SAAK,OAAO;AAAA,EACd;AAAA,EAJW;AAKb;AAEO,IAAM,oBAAN,cAAgC,MAAM;AAAA,EAC3C,YAAqB,UAAkB;AACrC,UAAM,mBAAmB,QAAQ,EAAE;AADhB;AAEnB,SAAK,OAAO;AAAA,EACd;AAAA,EAHqB;AAIvB;AAEO,IAAM,gBAAN,cAA4B,MAAM;AAAA,EACvC,YACE,SACS,OACT;AACA,UAAM,OAAO;AAFJ;AAGT,SAAK,OAAO;AAAA,EACd;AAAA,EAJW;AAKb;;;ACWO,SAAS,oBAAoB,QAAyB;AAC3D,MAAI,OAAO,WAAW,UAAU;AAC9B,WAAO;AAAA,EACT;AAEA,QAAM,aAAa,KAAK,UAAU,MAAM;AACxC,SAAO,eAAe,SAAY,OAAO,MAAM,IAAI;AACrD;AAEO,SAAS,cAAc,MAAyB;AACrD,MAAI,KAAK,KAAK,MAAM,IAAI;AACtB,WAAO,CAAC;AAAA,EACV;AAEA,SAAO,KAAK,MAAM,IAAI;AACxB;;;AC/CO,IAAM,UAAN,MAAM,SAAQ;AAAA,EACF,QAAQ,oBAAI,IAAqB;AAAA,EAElD,OAAO,UAAU,OAA2B;AAC1C,UAAM,UAAU,IAAI,SAAQ;AAC5B,eAAW,QAAQ,OAAO;AACxB,cAAQ,QAAQ,IAAI;AAAA,IACtB;AACA,WAAO;AAAA,EACT;AAAA,EAEA,QAAQ,MAAqB;AAC3B,SAAK,MAAM,IAAI,KAAK,MAAM,IAAI;AAC9B,WAAO;AAAA,EACT;AAAA,EAEA,SAAS,OAAkC;AACzC,UAAM,SAAS,MAAM,QAAQ,KAAK,IAAI,QAAQ,MAAM,OAAO;AAC3D,eAAW,QAAQ,QAAQ;AACzB,WAAK,QAAQ,IAAI;AAAA,IACnB;AACA,WAAO;AAAA,EACT;AAAA,EAEA,WAAW,UAA2B;AACpC,WAAO,KAAK,MAAM,OAAO,QAAQ;AAAA,EACnC;AAAA,EAEA,SAAS,UAA2B;AAClC,WAAO,KAAK,MAAM,IAAI,QAAQ;AAAA,EAChC;AAAA,EAEA,IAAI,UAAuC;AACzC,WAAO,KAAK,MAAM,IAAI,QAAQ;AAAA,EAChC;AAAA,EAEA,SAAoB;AAClB,WAAO,CAAC,GAAG,KAAK,MAAM,OAAO,CAAC;AAAA,EAChC;AAAA,EAEA,MAAM,mBAAmB,SAAS,IAA+B;AAC/D,UAAM,OAAyB,CAAC;AAChC,eAAW,QAAQ,KAAK,MAAM,OAAO,GAAG;AACtC,WAAK,KAAK,MAAM,KAAK,WAAW,MAAM,CAAC;AAAA,IACzC;AACA,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,KAAK,UAAkB,MAA+B;AAC1D,UAAM,OAAO,KAAK,MAAM,IAAI,QAAQ;AACpC,QAAI,SAAS,QAAW;AACtB,YAAM,IAAI,kBAAkB,QAAQ;AAAA,IACtC;AAEA,QAAI;AACJ,QAAI;AACF,mBAAa,cAAc,IAAI;AAAA,IACjC,SAAS,OAAO;AACd,YAAM,IAAI,cAAc,mCAAmC,QAAQ,IAAI,KAAK;AAAA,IAC9E;AAEA,QAAI;AACF,YAAM,SAAS,MAAM,KAAK,KAAK,UAAU;AACzC,aAAO,oBAAoB,MAAM;AAAA,IACnC,SAAS,OAAO;AACd,UAAI,iBAAiB,OAAO;AAC1B,cAAM,IAAI,cAAc,MAAM,SAAS,KAAK;AAAA,MAC9C;AACA,YAAM,IAAI,cAAc,QAAQ,QAAQ,WAAW,KAAK;AAAA,IAC1D;AAAA,EACF;AACF;","names":[]}

package/dist/chunk-I77SDTFE.js

@@ -0,0 +1,43 @@
+// src/audio-generation/index.ts
+var AudioGenerationRequestBuilder = class {
+  constructor(model) {
+    this.model = model;
+  }
+  model;
+  request = {
+    text: "",
+    voice: "",
+    speed: 1
+  };
+  text(text) {
+    this.request = { ...this.request, text };
+    return this;
+  }
+  voice(voice) {
+    this.request = { ...this.request, voice };
+    return this;
+  }
+  speed(speed) {
+    this.request = { ...this.request, speed };
+    return this;
+  }
+  additionalParams(additionalParams) {
+    this.request = { ...this.request, additionalParams };
+    return this;
+  }
+  build() {
+    return { ...this.request };
+  }
+  send() {
+    return this.model.audioGeneration(this.build());
+  }
+};
+function audioGenerationRequest(model) {
+  return new AudioGenerationRequestBuilder(model);
+}
+
+export {
+  AudioGenerationRequestBuilder,
+  audioGenerationRequest
+};
+//# sourceMappingURL=chunk-I77SDTFE.js.map

package/dist/chunk-I77SDTFE.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/audio-generation/index.ts"],"sourcesContent":["import type { JsonValue } from \"../completion\";\n\nexport type AudioGenerationRequest = {\n  text: string;\n  voice: string;\n  speed: number;\n  additionalParams?: JsonValue | undefined;\n};\n\nexport type AudioGenerationResponse<RawResponse = unknown> = {\n  audio: Uint8Array;\n  mediaType?: string | undefined;\n  rawResponse: RawResponse;\n};\n\nexport interface AudioGenerationModel<RawResponse = unknown> {\n  readonly provider?: string | undefined;\n  readonly defaultModel?: string | undefined;\n  audioGeneration(request: AudioGenerationRequest): Promise<AudioGenerationResponse<RawResponse>>;\n}\n\nexport class AudioGenerationRequestBuilder<\n  Model extends AudioGenerationModel = AudioGenerationModel,\n> {\n  private request: AudioGenerationRequest = {\n    text: \"\",\n    voice: \"\",\n    speed: 1,\n  };\n\n  constructor(private readonly model: Model) {}\n\n  text(text: string): this {\n    this.request = { ...this.request, text };\n    return this;\n  }\n\n  voice(voice: string): this {\n    this.request = { ...this.request, voice };\n    return this;\n  }\n\n  speed(speed: number): this {\n    this.request = { ...this.request, speed };\n    return this;\n  }\n\n  additionalParams(additionalParams: JsonValue): this {\n    this.request = { ...this.request, additionalParams };\n    return this;\n  }\n\n  build(): AudioGenerationRequest {\n    return { ...this.request };\n  }\n\n  send(): Promise<Awaited<ReturnType<Model[\"audioGeneration\"]>>> {\n    return this.model.audioGeneration(this.build()) as Promise<\n      Awaited<ReturnType<Model[\"audioGeneration\"]>>\n    >;\n  }\n}\n\nexport function audioGenerationRequest<Model extends AudioGenerationModel>(\n  model: Model,\n): AudioGenerationRequestBuilder<Model> {\n  return new AudioGenerationRequestBuilder(model);\n}\n"],"mappings":";AAqBO,IAAM,gCAAN,MAEL;AAAA,EAOA,YAA6B,OAAc;AAAd;AAAA,EAAe;AAAA,EAAf;AAAA,EANrB,UAAkC;AAAA,IACxC,MAAM;AAAA,IACN,OAAO;AAAA,IACP,OAAO;AAAA,EACT;AAAA,EAIA,KAAK,MAAoB;AACvB,SAAK,UAAU,EAAE,GAAG,KAAK,SAAS,KAAK;AACvC,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,OAAqB;AACzB,SAAK,UAAU,EAAE,GAAG,KAAK,SAAS,MAAM;AACxC,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,OAAqB;AACzB,SAAK,UAAU,EAAE,GAAG,KAAK,SAAS,MAAM;AACxC,WAAO;AAAA,EACT;AAAA,EAEA,iBAAiB,kBAAmC;AAClD,SAAK,UAAU,EAAE,GAAG,KAAK,SAAS,iBAAiB;AACnD,WAAO;AAAA,EACT;AAAA,EAEA,QAAgC;AAC9B,WAAO,EAAE,GAAG,KAAK,QAAQ;AAAA,EAC3B;AAAA,EAEA,OAA+D;AAC7D,WAAO,KAAK,MAAM,gBAAgB,KAAK,MAAM,CAAC;AAAA,EAGhD;AACF;AAEO,SAAS,uBACd,OACsC;AACtC,SAAO,IAAI,8BAA8B,KAAK;AAChD;","names":[]}

package/dist/chunk-IA76K5UX.js

@@ -0,0 +1,142 @@
+import {
+  AgentBuilder
+} from "./chunk-IQBY2GCF.js";
+import {
+  createTool
+} from "./chunk-B4QHQN5K.js";
+import {
+  CompletionCapabilityError,
+  CompletionRequestBuilder,
+  Message,
+  Usage
+} from "./chunk-CP47FBJV.js";
+
+// src/extractor/extractor.ts
+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 {
+  constructor(message, cause) {
+    super(message);
+    this.cause = cause;
+    this.name = "ExtractionError";
+  }
+  cause;
+};
+var Extractor = class {
+  constructor(agent, schema, retryCount) {
+    this.agent = agent;
+    this.schema = schema;
+    this.retryCount = retryCount;
+  }
+  agent;
+  schema;
+  retryCount;
+  async extract(text) {
+    return (await this.extractWithUsage(text)).data;
+  }
+  async extractWithUsage(text) {
+    return this.run(text);
+  }
+  async extractWithHistory(text, history) {
+    return (await this.run(text, history)).data;
+  }
+  getInner() {
+    return this.agent;
+  }
+  async run(text, history) {
+    let usage = Usage.empty();
+    let lastError;
+    const prompt = typeof text === "string" ? Message.user(text) : text;
+    for (let attempt = 0; attempt <= this.retryCount; attempt += 1) {
+      try {
+        const toolDefs = await this.agent.toolSet.getToolDefinitions(extractRagText(prompt));
+        const response = await new CompletionRequestBuilder(this.agent.model, prompt).instructions(this.agent.instructions).messages(history ?? []).documents(this.agent.staticContext).tools(toolDefs).temperature(this.agent.temperature).maxTokens(this.agent.maxTokens).additionalParams(this.agent.additionalParams).toolChoice(this.agent.toolChoice).send();
+        usage = Usage.add(usage, response.usage);
+        const data = extractSubmittedData(response, this.schema);
+        return {
+          data,
+          usage,
+          messages: [
+            ...history ?? [],
+            prompt,
+            Message.assistant(response.choice, response.messageId)
+          ]
+        };
+      } catch (error) {
+        if (error instanceof CompletionCapabilityError) {
+          throw error;
+        }
+        lastError = error;
+      }
+    }
+    throw new ExtractionError("No data extracted", lastError);
+  }
+};
+var ExtractorBuilder = class {
+  constructor(model, schema) {
+    this.schema = schema;
+    this.agentBuilder = new AgentBuilder("extractor", model).instructions(DEFAULT_EXTRACTOR_INSTRUCTIONS).tool(
+      createTool({
+        name: SUBMIT_TOOL_NAME,
+        description: "Submit the structured data extracted from the provided text.",
+        input: schema,
+        output: schema,
+        execute: (args) => args
+      })
+    ).toolChoice("required");
+  }
+  schema;
+  agentBuilder;
+  retryCount = 0;
+  instructions(instructions) {
+    this.agentBuilder.instructions(instructions);
+    return this;
+  }
+  context(text, id) {
+    this.agentBuilder.context(text, id);
+    return this;
+  }
+  temperature(temperature) {
+    this.agentBuilder.temperature(temperature);
+    return this;
+  }
+  maxTokens(maxTokens) {
+    this.agentBuilder.maxTokens(maxTokens);
+    return this;
+  }
+  additionalParams(params) {
+    this.agentBuilder.additionalParams(params);
+    return this;
+  }
+  toolChoice(toolChoice) {
+    this.agentBuilder.toolChoice(toolChoice);
+    return this;
+  }
+  retries(retries) {
+    this.retryCount = Math.max(0, Math.trunc(retries));
+    return this;
+  }
+  build() {
+    return new Extractor(this.agentBuilder.build(), this.schema, this.retryCount);
+  }
+};
+function extractSubmittedData(response, schema) {
+  const submitted = response.choice.filter((content) => content.type === "tool_call").filter((toolCall) => toolCall.function.name === SUBMIT_TOOL_NAME).at(-1);
+  if (submitted === void 0) {
+    throw new ExtractionError("The model did not call the submit tool");
+  }
+  return schema.parse(submitted.function.arguments);
+}
+function extractRagText(message) {
+  if (message.role === "user") {
+    return message.content.flatMap((item) => item.type === "text" ? [item.text] : []).join("\n");
+  }
+  return void 0;
+}
+
+export {
+  ExtractionError,
+  Extractor,
+  ExtractorBuilder
+};
+//# sourceMappingURL=chunk-IA76K5UX.js.map

package/dist/chunk-IA76K5UX.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/extractor/extractor.ts"],"sourcesContent":["import { type Agent, AgentBuilder } from \"../agent/index\";\nimport {\n  CompletionCapabilityError,\n  type CompletionModel,\n  CompletionRequestBuilder,\n  type CompletionResponse,\n  type JsonValue,\n  type Message,\n  Message as MessageFactory,\n  type ToolChoice,\n  Usage,\n} from \"../completion/index\";\nimport type { ZodSchema } from \"../schema/zod-schema\";\nimport { createTool } from \"../tool/index\";\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\nexport type ExtractionResponse<T> = {\n  data: T;\n  usage: Usage;\n  messages: Message[];\n};\n\nexport class ExtractionError extends Error {\n  constructor(\n    message: string,\n    readonly cause?: unknown,\n  ) {\n    super(message);\n    this.name = \"ExtractionError\";\n  }\n}\n\nexport class Extractor<T, M extends CompletionModel = CompletionModel> {\n  constructor(\n    private readonly agent: Agent<M>,\n    private readonly schema: ZodSchema<T>,\n    private readonly retryCount: number,\n  ) {}\n\n  async extract(text: string | Message): Promise<T> {\n    return (await this.extractWithUsage(text)).data;\n  }\n\n  async extractWithUsage(text: string | Message): Promise<ExtractionResponse<T>> {\n    return this.run(text);\n  }\n\n  async extractWithHistory(text: string | Message, history: Message[]): Promise<T> {\n    return (await this.run(text, history)).data;\n  }\n\n  getInner(): Agent<M> {\n    return this.agent;\n  }\n\n  private async run(text: string | Message, history?: Message[]): Promise<ExtractionResponse<T>> {\n    let usage = Usage.empty();\n    let lastError: unknown;\n    const prompt = typeof text === \"string\" ? MessageFactory.user(text) : text;\n\n    for (let attempt = 0; attempt <= this.retryCount; attempt += 1) {\n      try {\n        const toolDefs = await this.agent.toolSet.getToolDefinitions(extractRagText(prompt));\n        const response = await new CompletionRequestBuilder(this.agent.model, prompt)\n          .instructions(this.agent.instructions)\n          .messages(history ?? [])\n          .documents(this.agent.staticContext)\n          .tools(toolDefs)\n          .temperature(this.agent.temperature)\n          .maxTokens(this.agent.maxTokens)\n          .additionalParams(this.agent.additionalParams)\n          .toolChoice(this.agent.toolChoice)\n          .send();\n        usage = Usage.add(usage, response.usage);\n        const data = extractSubmittedData(response, this.schema);\n        return {\n          data,\n          usage,\n          messages: [\n            ...(history ?? []),\n            prompt,\n            MessageFactory.assistant(response.choice, response.messageId),\n          ],\n        };\n      } catch (error) {\n        if (error instanceof CompletionCapabilityError) {\n          throw error;\n        }\n        lastError = error;\n      }\n    }\n\n    throw new ExtractionError(\"No data extracted\", lastError);\n  }\n}\n\nexport class ExtractorBuilder<T, M extends CompletionModel = CompletionModel> {\n  private readonly agentBuilder: AgentBuilder<M>;\n  private retryCount = 0;\n\n  constructor(\n    model: M,\n    private readonly schema: ZodSchema<T>,\n  ) {\n    this.agentBuilder = new AgentBuilder(\"extractor\", model)\n      .instructions(DEFAULT_EXTRACTOR_INSTRUCTIONS)\n      .tool(\n        createTool({\n          name: SUBMIT_TOOL_NAME,\n          description: \"Submit the structured data extracted from the provided text.\",\n          input: schema,\n          output: schema,\n          execute: (args) => args,\n        }),\n      )\n      .toolChoice(\"required\");\n  }\n\n  instructions(instructions: string): this {\n    this.agentBuilder.instructions(instructions);\n    return this;\n  }\n\n  context(text: string, id?: string): this {\n    this.agentBuilder.context(text, id);\n    return this;\n  }\n\n  temperature(temperature: number): this {\n    this.agentBuilder.temperature(temperature);\n    return this;\n  }\n\n  maxTokens(maxTokens: number): this {\n    this.agentBuilder.maxTokens(maxTokens);\n    return this;\n  }\n\n  additionalParams(params: JsonValue): this {\n    this.agentBuilder.additionalParams(params);\n    return this;\n  }\n\n  toolChoice(toolChoice: ToolChoice): this {\n    this.agentBuilder.toolChoice(toolChoice);\n    return this;\n  }\n\n  retries(retries: number): this {\n    this.retryCount = Math.max(0, Math.trunc(retries));\n    return this;\n  }\n\n  build(): Extractor<T, M> {\n    return new Extractor(this.agentBuilder.build(), this.schema, this.retryCount);\n  }\n}\n\nfunction extractSubmittedData<T>(response: CompletionResponse, schema: ZodSchema<T>): T {\n  const submitted = response.choice\n    .filter((content) => content.type === \"tool_call\")\n    .filter((toolCall) => toolCall.function.name === SUBMIT_TOOL_NAME)\n    .at(-1);\n\n  if (submitted === undefined) {\n    throw new ExtractionError(\"The model did not call the submit tool\");\n  }\n\n  return schema.parse(submitted.function.arguments);\n}\n\nfunction extractRagText(message: Message): string | undefined {\n  if (message.role === \"user\") {\n    return message.content.flatMap((item) => (item.type === \"text\" ? [item.text] : [])).join(\"\\n\");\n  }\n\n  return undefined;\n}\n"],"mappings":";;;;;;;;;;;;;;AAeA,IAAM,mBAAmB;AAEzB,IAAM,iCACJ;AAUK,IAAM,kBAAN,cAA8B,MAAM;AAAA,EACzC,YACE,SACS,OACT;AACA,UAAM,OAAO;AAFJ;AAGT,SAAK,OAAO;AAAA,EACd;AAAA,EAJW;AAKb;AAEO,IAAM,YAAN,MAAgE;AAAA,EACrE,YACmB,OACA,QACA,YACjB;AAHiB;AACA;AACA;AAAA,EAChB;AAAA,EAHgB;AAAA,EACA;AAAA,EACA;AAAA,EAGnB,MAAM,QAAQ,MAAoC;AAChD,YAAQ,MAAM,KAAK,iBAAiB,IAAI,GAAG;AAAA,EAC7C;AAAA,EAEA,MAAM,iBAAiB,MAAwD;AAC7E,WAAO,KAAK,IAAI,IAAI;AAAA,EACtB;AAAA,EAEA,MAAM,mBAAmB,MAAwB,SAAgC;AAC/E,YAAQ,MAAM,KAAK,IAAI,MAAM,OAAO,GAAG;AAAA,EACzC;AAAA,EAEA,WAAqB;AACnB,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,MAAc,IAAI,MAAwB,SAAqD;AAC7F,QAAI,QAAQ,MAAM,MAAM;AACxB,QAAI;AACJ,UAAM,SAAS,OAAO,SAAS,WAAW,QAAe,KAAK,IAAI,IAAI;AAEtE,aAAS,UAAU,GAAG,WAAW,KAAK,YAAY,WAAW,GAAG;AAC9D,UAAI;AACF,cAAM,WAAW,MAAM,KAAK,MAAM,QAAQ,mBAAmB,eAAe,MAAM,CAAC;AACnF,cAAM,WAAW,MAAM,IAAI,yBAAyB,KAAK,MAAM,OAAO,MAAM,EACzE,aAAa,KAAK,MAAM,YAAY,EACpC,SAAS,WAAW,CAAC,CAAC,EACtB,UAAU,KAAK,MAAM,aAAa,EAClC,MAAM,QAAQ,EACd,YAAY,KAAK,MAAM,WAAW,EAClC,UAAU,KAAK,MAAM,SAAS,EAC9B,iBAAiB,KAAK,MAAM,gBAAgB,EAC5C,WAAW,KAAK,MAAM,UAAU,EAChC,KAAK;AACR,gBAAQ,MAAM,IAAI,OAAO,SAAS,KAAK;AACvC,cAAM,OAAO,qBAAqB,UAAU,KAAK,MAAM;AACvD,eAAO;AAAA,UACL;AAAA,UACA;AAAA,UACA,UAAU;AAAA,YACR,GAAI,WAAW,CAAC;AAAA,YAChB;AAAA,YACA,QAAe,UAAU,SAAS,QAAQ,SAAS,SAAS;AAAA,UAC9D;AAAA,QACF;AAAA,MACF,SAAS,OAAO;AACd,YAAI,iBAAiB,2BAA2B;AAC9C,gBAAM;AAAA,QACR;AACA,oBAAY;AAAA,MACd;AAAA,IACF;AAEA,UAAM,IAAI,gBAAgB,qBAAqB,SAAS;AAAA,EAC1D;AACF;AAEO,IAAM,mBAAN,MAAuE;AAAA,EAI5E,YACE,OACiB,QACjB;AADiB;AAEjB,SAAK,eAAe,IAAI,aAAa,aAAa,KAAK,EACpD,aAAa,8BAA8B,EAC3C;AAAA,MACC,WAAW;AAAA,QACT,MAAM;AAAA,QACN,aAAa;AAAA,QACb,OAAO;AAAA,QACP,QAAQ;AAAA,QACR,SAAS,CAAC,SAAS;AAAA,MACrB,CAAC;AAAA,IACH,EACC,WAAW,UAAU;AAAA,EAC1B;AAAA,EAdmB;AAAA,EALF;AAAA,EACT,aAAa;AAAA,EAoBrB,aAAa,cAA4B;AACvC,SAAK,aAAa,aAAa,YAAY;AAC3C,WAAO;AAAA,EACT;AAAA,EAEA,QAAQ,MAAc,IAAmB;AACvC,SAAK,aAAa,QAAQ,MAAM,EAAE;AAClC,WAAO;AAAA,EACT;AAAA,EAEA,YAAY,aAA2B;AACrC,SAAK,aAAa,YAAY,WAAW;AACzC,WAAO;AAAA,EACT;AAAA,EAEA,UAAU,WAAyB;AACjC,SAAK,aAAa,UAAU,SAAS;AACrC,WAAO;AAAA,EACT;AAAA,EAEA,iBAAiB,QAAyB;AACxC,SAAK,aAAa,iBAAiB,MAAM;AACzC,WAAO;AAAA,EACT;AAAA,EAEA,WAAW,YAA8B;AACvC,SAAK,aAAa,WAAW,UAAU;AACvC,WAAO;AAAA,EACT;AAAA,EAEA,QAAQ,SAAuB;AAC7B,SAAK,aAAa,KAAK,IAAI,GAAG,KAAK,MAAM,OAAO,CAAC;AACjD,WAAO;AAAA,EACT;AAAA,EAEA,QAAyB;AACvB,WAAO,IAAI,UAAU,KAAK,aAAa,MAAM,GAAG,KAAK,QAAQ,KAAK,UAAU;AAAA,EAC9E;AACF;AAEA,SAAS,qBAAwB,UAA8B,QAAyB;AACtF,QAAM,YAAY,SAAS,OACxB,OAAO,CAAC,YAAY,QAAQ,SAAS,WAAW,EAChD,OAAO,CAAC,aAAa,SAAS,SAAS,SAAS,gBAAgB,EAChE,GAAG,EAAE;AAER,MAAI,cAAc,QAAW;AAC3B,UAAM,IAAI,gBAAgB,wCAAwC;AAAA,EACpE;AAEA,SAAO,OAAO,MAAM,UAAU,SAAS,SAAS;AAClD;AAEA,SAAS,eAAe,SAAsC;AAC5D,MAAI,QAAQ,SAAS,QAAQ;AAC3B,WAAO,QAAQ,QAAQ,QAAQ,CAAC,SAAU,KAAK,SAAS,SAAS,CAAC,KAAK,IAAI,IAAI,CAAC,CAAE,EAAE,KAAK,IAAI;AAAA,EAC/F;AAEA,SAAO;AACT;","names":[]}