@anvia/core 0.26.0 → 1.0.0-rc.1

package/dist/chunk-Q25IWLBT.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/completion/documents.ts","../src/completion/request.ts"],"sourcesContent":["import { type Document, Message, type Message as MessageType } from \"./types\";\n\nexport function normalizeDocuments(documents: Document[]): MessageType | undefined {\n  if (documents.length === 0) {\n    return undefined;\n  }\n\n  return Message.user(documents.map(formatDocument).join(\"\\n\"));\n}\n\nexport function formatDocument(document: Document): string {\n  return `<file id: ${document.id}>\\n${formatDocumentBody(document)}\\n</file>\\n`;\n}\n\nfunction formatDocumentBody(document: Document): string {\n  const metadata = formatMetadata(document.additionalProps);\n  return metadata === undefined ? document.text : `${metadata}\\n${document.text}`;\n}\n\nfunction formatMetadata(additionalProps: Record<string, string> | undefined): string | undefined {\n  if (additionalProps === undefined) {\n    return undefined;\n  }\n\n  const entries = Object.entries(additionalProps).sort(([left], [right]) =>\n    left.localeCompare(right),\n  );\n  if (entries.length === 0) {\n    return undefined;\n  }\n\n  const metadata = entries.map(([key, value]) => `${key}: ${JSON.stringify(value)}`).join(\" \");\n  return `<metadata ${metadata} />`;\n}\n","import type {\n  CompletionModel,\n  CompletionRequest,\n  CompletionResponse,\n  CompletionTool,\n  Document,\n  JsonObject,\n  JsonValue,\n  Message as MessageType,\n  ProviderTool,\n  ToolChoice,\n  ToolDefinition,\n} from \"./types\";\nimport { assertCompletionRequestSupported, isProviderTool } from \"./types\";\n\ntype ModelNameOf<M extends CompletionModel> =\n  M extends CompletionModel<unknown, infer ModelName> ? ModelName : string;\n\ntype RawResponseOf<M extends CompletionModel> =\n  M extends CompletionModel<infer RawResponse, infer _ModelName> ? RawResponse : unknown;\n\nexport class CompletionRequestBuilder<M extends CompletionModel = CompletionModel> {\n  private requestModel: ModelNameOf<M> | undefined;\n  private instructionBlocks: string[] = [];\n  private history: MessageType[] = [];\n  private docs: Document[] = [];\n  private toolDefs: ToolDefinition[] = [];\n  private providerToolDefs: ProviderTool[] = [];\n  private temp: number | undefined;\n  private maxTokenCount: number | undefined;\n  private choice: ToolChoice | undefined;\n  private params: JsonValue | undefined;\n  private schema: JsonObject | undefined;\n\n  constructor(\n    private readonly model: M,\n    private readonly promptMessage: MessageType,\n  ) {}\n\n  modelOverride(model: ModelNameOf<M> | undefined): this {\n    this.requestModel = model;\n    return this;\n  }\n\n  instructions(instructions: string | undefined): this {\n    if (instructions !== undefined && instructions.length > 0) {\n      this.instructionBlocks.push(instructions);\n    }\n    return this;\n  }\n\n  messages(messages: MessageType[]): this {\n    this.history.push(...messages);\n    return this;\n  }\n\n  documents(documents: Document[]): this {\n    this.docs.push(...documents);\n    return this;\n  }\n\n  tools(tools: CompletionTool[]): this {\n    for (const tool of tools) {\n      if (isProviderTool(tool)) {\n        this.providerToolDefs.push(tool);\n      } else {\n        this.toolDefs.push(tool);\n      }\n    }\n    return this;\n  }\n\n  temperature(temperature: number | undefined): this {\n    this.temp = temperature;\n    return this;\n  }\n\n  maxTokens(maxTokens: number | undefined): this {\n    this.maxTokenCount = maxTokens;\n    return this;\n  }\n\n  toolChoice(toolChoice: ToolChoice | undefined): this {\n    this.choice = toolChoice;\n    return this;\n  }\n\n  additionalParams(additionalParams: JsonValue | undefined): this {\n    this.params = additionalParams;\n    return this;\n  }\n\n  outputSchema(outputSchema: JsonObject | undefined): this {\n    this.schema = outputSchema;\n    return this;\n  }\n\n  build(): CompletionRequest<ModelNameOf<M>> {\n    const instructions = this.buildInstructions();\n    const request: CompletionRequest<ModelNameOf<M>> = {\n      chatHistory: [...this.history, this.promptMessage],\n      documents: [...this.docs],\n      tools: [...this.toolDefs],\n    };\n\n    if (this.providerToolDefs.length > 0) request.providerTools = [...this.providerToolDefs];\n    if (this.requestModel !== undefined) request.model = this.requestModel;\n    if (instructions !== undefined) request.instructions = instructions;\n    if (this.temp !== undefined) request.temperature = this.temp;\n    if (this.maxTokenCount !== undefined) request.maxTokens = this.maxTokenCount;\n    if (this.choice !== undefined) request.toolChoice = this.choice;\n    if (this.params !== undefined) request.additionalParams = this.params;\n    if (this.schema !== undefined) request.outputSchema = this.schema;\n\n    return request;\n  }\n\n  async send(): Promise<CompletionResponse<RawResponseOf<M>>> {\n    const request = this.build();\n    assertCompletionRequestSupported(this.model, request);\n    return this.model.completion(request) as Promise<CompletionResponse<RawResponseOf<M>>>;\n  }\n\n  private buildInstructions(): string | undefined {\n    return this.instructionBlocks.length === 0 ? undefined : this.instructionBlocks.join(\"\\n\\n\");\n  }\n}\n"],"mappings":";;;;;;;AAEO,SAAS,mBAAmB,WAAgD;AACjF,MAAI,UAAU,WAAW,GAAG;AAC1B,WAAO;AAAA,EACT;AAEA,SAAO,QAAQ,KAAK,UAAU,IAAI,cAAc,EAAE,KAAK,IAAI,CAAC;AAC9D;AAEO,SAAS,eAAe,UAA4B;AACzD,SAAO,aAAa,SAAS,EAAE;AAAA,EAAM,mBAAmB,QAAQ,CAAC;AAAA;AAAA;AACnE;AAEA,SAAS,mBAAmB,UAA4B;AACtD,QAAM,WAAW,eAAe,SAAS,eAAe;AACxD,SAAO,aAAa,SAAY,SAAS,OAAO,GAAG,QAAQ;AAAA,EAAK,SAAS,IAAI;AAC/E;AAEA,SAAS,eAAe,iBAAyE;AAC/F,MAAI,oBAAoB,QAAW;AACjC,WAAO;AAAA,EACT;AAEA,QAAM,UAAU,OAAO,QAAQ,eAAe,EAAE;AAAA,IAAK,CAAC,CAAC,IAAI,GAAG,CAAC,KAAK,MAClE,KAAK,cAAc,KAAK;AAAA,EAC1B;AACA,MAAI,QAAQ,WAAW,GAAG;AACxB,WAAO;AAAA,EACT;AAEA,QAAM,WAAW,QAAQ,IAAI,CAAC,CAAC,KAAK,KAAK,MAAM,GAAG,GAAG,KAAK,KAAK,UAAU,KAAK,CAAC,EAAE,EAAE,KAAK,GAAG;AAC3F,SAAO,aAAa,QAAQ;AAC9B;;;ACZO,IAAM,2BAAN,MAA4E;AAAA,EAajF,YACmB,OACA,eACjB;AAFiB;AACA;AAAA,EAChB;AAAA,EAFgB;AAAA,EACA;AAAA,EAdX;AAAA,EACA,oBAA8B,CAAC;AAAA,EAC/B,UAAyB,CAAC;AAAA,EAC1B,OAAmB,CAAC;AAAA,EACpB,WAA6B,CAAC;AAAA,EAC9B,mBAAmC,CAAC;AAAA,EACpC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAOR,cAAc,OAAyC;AACrD,SAAK,eAAe;AACpB,WAAO;AAAA,EACT;AAAA,EAEA,aAAa,cAAwC;AACnD,QAAI,iBAAiB,UAAa,aAAa,SAAS,GAAG;AACzD,WAAK,kBAAkB,KAAK,YAAY;AAAA,IAC1C;AACA,WAAO;AAAA,EACT;AAAA,EAEA,SAAS,UAA+B;AACtC,SAAK,QAAQ,KAAK,GAAG,QAAQ;AAC7B,WAAO;AAAA,EACT;AAAA,EAEA,UAAU,WAA6B;AACrC,SAAK,KAAK,KAAK,GAAG,SAAS;AAC3B,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,OAA+B;AACnC,eAAW,QAAQ,OAAO;AACxB,UAAI,eAAe,IAAI,GAAG;AACxB,aAAK,iBAAiB,KAAK,IAAI;AAAA,MACjC,OAAO;AACL,aAAK,SAAS,KAAK,IAAI;AAAA,MACzB;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA,EAEA,YAAY,aAAuC;AACjD,SAAK,OAAO;AACZ,WAAO;AAAA,EACT;AAAA,EAEA,UAAU,WAAqC;AAC7C,SAAK,gBAAgB;AACrB,WAAO;AAAA,EACT;AAAA,EAEA,WAAW,YAA0C;AACnD,SAAK,SAAS;AACd,WAAO;AAAA,EACT;AAAA,EAEA,iBAAiB,kBAA+C;AAC9D,SAAK,SAAS;AACd,WAAO;AAAA,EACT;AAAA,EAEA,aAAa,cAA4C;AACvD,SAAK,SAAS;AACd,WAAO;AAAA,EACT;AAAA,EAEA,QAA2C;AACzC,UAAM,eAAe,KAAK,kBAAkB;AAC5C,UAAM,UAA6C;AAAA,MACjD,aAAa,CAAC,GAAG,KAAK,SAAS,KAAK,aAAa;AAAA,MACjD,WAAW,CAAC,GAAG,KAAK,IAAI;AAAA,MACxB,OAAO,CAAC,GAAG,KAAK,QAAQ;AAAA,IAC1B;AAEA,QAAI,KAAK,iBAAiB,SAAS,EAAG,SAAQ,gBAAgB,CAAC,GAAG,KAAK,gBAAgB;AACvF,QAAI,KAAK,iBAAiB,OAAW,SAAQ,QAAQ,KAAK;AAC1D,QAAI,iBAAiB,OAAW,SAAQ,eAAe;AACvD,QAAI,KAAK,SAAS,OAAW,SAAQ,cAAc,KAAK;AACxD,QAAI,KAAK,kBAAkB,OAAW,SAAQ,YAAY,KAAK;AAC/D,QAAI,KAAK,WAAW,OAAW,SAAQ,aAAa,KAAK;AACzD,QAAI,KAAK,WAAW,OAAW,SAAQ,mBAAmB,KAAK;AAC/D,QAAI,KAAK,WAAW,OAAW,SAAQ,eAAe,KAAK;AAE3D,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,OAAsD;AAC1D,UAAM,UAAU,KAAK,MAAM;AAC3B,qCAAiC,KAAK,OAAO,OAAO;AACpD,WAAO,KAAK,MAAM,WAAW,OAAO;AAAA,EACtC;AAAA,EAEQ,oBAAwC;AAC9C,WAAO,KAAK,kBAAkB,WAAW,IAAI,SAAY,KAAK,kBAAkB,KAAK,MAAM;AAAA,EAC7F;AACF;","names":[]}

package/dist/chunk-UQX6EXWG.js

@@ -1,33 +0,0 @@
-import {
-  assertNonnegativeInteger,
-  assertPositiveInteger
-} from "./chunk-MRQLEK2B.js";
-
-// src/memory/options.ts
-function resolveMemoryOptions(options = {}) {
-  const resolved = {
-    savePolicy: options.savePolicy ?? "message"
-  };
-  if (options.compaction !== void 0) {
-    assertPositiveInteger(options.compaction.maxMessages, "compaction.maxMessages");
-    const keepRecentUserTurns = options.compaction.keepRecentUserTurns ?? 4;
-    const conflictRetries = options.compaction.conflictRetries ?? 1;
-    assertPositiveInteger(keepRecentUserTurns, "compaction.keepRecentUserTurns");
-    assertNonnegativeInteger(conflictRetries, "compaction.conflictRetries");
-    if (typeof options.compaction.compactor !== "function") {
-      throw new TypeError("compaction.compactor must be a function.");
-    }
-    resolved.compaction = {
-      maxMessages: options.compaction.maxMessages,
-      keepRecentUserTurns,
-      compactor: options.compaction.compactor,
-      conflictRetries
-    };
-  }
-  return resolved;
-}
-
-export {
-  resolveMemoryOptions
-};
-//# sourceMappingURL=chunk-UQX6EXWG.js.map

package/dist/chunk-UQX6EXWG.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/memory/options.ts"],"sourcesContent":["import { assertNonnegativeInteger, assertPositiveInteger } from \"./assert\";\nimport type { MemoryOptions, ResolvedMemoryOptions } from \"./types\";\n\nexport function resolveMemoryOptions(options: MemoryOptions = {}): ResolvedMemoryOptions {\n  const resolved: ResolvedMemoryOptions = {\n    savePolicy: options.savePolicy ?? \"message\",\n  };\n  if (options.compaction !== undefined) {\n    assertPositiveInteger(options.compaction.maxMessages, \"compaction.maxMessages\");\n    const keepRecentUserTurns = options.compaction.keepRecentUserTurns ?? 4;\n    const conflictRetries = options.compaction.conflictRetries ?? 1;\n    assertPositiveInteger(keepRecentUserTurns, \"compaction.keepRecentUserTurns\");\n    assertNonnegativeInteger(conflictRetries, \"compaction.conflictRetries\");\n    if (typeof options.compaction.compactor !== \"function\") {\n      throw new TypeError(\"compaction.compactor must be a function.\");\n    }\n    resolved.compaction = {\n      maxMessages: options.compaction.maxMessages,\n      keepRecentUserTurns,\n      compactor: options.compaction.compactor,\n      conflictRetries,\n    };\n  }\n  return resolved;\n}\n"],"mappings":";;;;;;AAGO,SAAS,qBAAqB,UAAyB,CAAC,GAA0B;AACvF,QAAM,WAAkC;AAAA,IACtC,YAAY,QAAQ,cAAc;AAAA,EACpC;AACA,MAAI,QAAQ,eAAe,QAAW;AACpC,0BAAsB,QAAQ,WAAW,aAAa,wBAAwB;AAC9E,UAAM,sBAAsB,QAAQ,WAAW,uBAAuB;AACtE,UAAM,kBAAkB,QAAQ,WAAW,mBAAmB;AAC9D,0BAAsB,qBAAqB,gCAAgC;AAC3E,6BAAyB,iBAAiB,4BAA4B;AACtE,QAAI,OAAO,QAAQ,WAAW,cAAc,YAAY;AACtD,YAAM,IAAI,UAAU,0CAA0C;AAAA,IAChE;AACA,aAAS,aAAa;AAAA,MACpB,aAAa,QAAQ,WAAW;AAAA,MAChC;AAAA,MACA,WAAW,QAAQ,WAAW;AAAA,MAC9B;AAAA,IACF;AAAA,EACF;AACA,SAAO;AACT;","names":[]}

package/dist/chunk-WQKHFADH.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/schema/zod-schema.ts"],"sourcesContent":["import { z } from \"zod\";\nimport type { JsonObject } from \"../completion/index\";\n\nexport type ZodSchema<T = unknown> = z.ZodType<T>;\n\nexport function toProviderJsonSchema(schema: z.ZodType): JsonObject {\n  const jsonSchema = z.toJSONSchema(schema) as JsonObject;\n  const { $schema: _schema, ...providerSchema } = jsonSchema;\n  return providerSchema;\n}\n"],"mappings":";AAAA,SAAS,SAAS;AAKX,SAAS,qBAAqB,QAA+B;AAClE,QAAM,aAAa,EAAE,aAAa,MAAM;AACxC,QAAM,EAAE,SAAS,SAAS,GAAG,eAAe,IAAI;AAChD,SAAO;AACT;","names":[]}

package/dist/chunk-XUUY2L2D.js

@@ -1,42 +0,0 @@
-// src/streaming/readable-stream.ts
-function toReadableStream(events, _options = {}) {
-  const encoder = new TextEncoder();
-  const iterator = events[Symbol.asyncIterator]();
-  return new ReadableStream({
-    async pull(controller) {
-      try {
-        const next = await iterator.next();
-        if (next.done === true) {
-          controller.close();
-          return;
-        }
-        controller.enqueue(encoder.encode(`${JSON.stringify(next.value)}
-`));
-      } catch (error) {
-        controller.enqueue(
-          encoder.encode(`${JSON.stringify({ type: "error", error: serializeError(error) })}
-`)
-        );
-        controller.close();
-      }
-    },
-    async cancel() {
-      await iterator.return?.();
-    }
-  });
-}
-function serializeError(error) {
-  if (error instanceof Error) {
-    return {
-      name: error.name,
-      message: error.message,
-      stack: error.stack
-    };
-  }
-  return error;
-}
-
-export {
-  toReadableStream
-};
-//# sourceMappingURL=chunk-XUUY2L2D.js.map

package/dist/chunk-XUUY2L2D.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/streaming/readable-stream.ts"],"sourcesContent":["export type ReadableStreamOptions = {\n  format?: \"jsonl\";\n};\n\nexport function toReadableStream<T>(\n  events: AsyncIterable<T>,\n  _options: ReadableStreamOptions = {},\n): ReadableStream<Uint8Array> {\n  const encoder = new TextEncoder();\n  const iterator = events[Symbol.asyncIterator]();\n\n  return new ReadableStream<Uint8Array>({\n    async pull(controller) {\n      try {\n        const next = await iterator.next();\n        if (next.done === true) {\n          controller.close();\n          return;\n        }\n\n        controller.enqueue(encoder.encode(`${JSON.stringify(next.value)}\\n`));\n      } catch (error) {\n        controller.enqueue(\n          encoder.encode(`${JSON.stringify({ type: \"error\", error: serializeError(error) })}\\n`),\n        );\n        controller.close();\n      }\n    },\n    async cancel() {\n      await iterator.return?.();\n    },\n  });\n}\n\nfunction serializeError(error: unknown): unknown {\n  if (error instanceof Error) {\n    return {\n      name: error.name,\n      message: error.message,\n      stack: error.stack,\n    };\n  }\n\n  return error;\n}\n"],"mappings":";AAIO,SAAS,iBACd,QACA,WAAkC,CAAC,GACP;AAC5B,QAAM,UAAU,IAAI,YAAY;AAChC,QAAM,WAAW,OAAO,OAAO,aAAa,EAAE;AAE9C,SAAO,IAAI,eAA2B;AAAA,IACpC,MAAM,KAAK,YAAY;AACrB,UAAI;AACF,cAAM,OAAO,MAAM,SAAS,KAAK;AACjC,YAAI,KAAK,SAAS,MAAM;AACtB,qBAAW,MAAM;AACjB;AAAA,QACF;AAEA,mBAAW,QAAQ,QAAQ,OAAO,GAAG,KAAK,UAAU,KAAK,KAAK,CAAC;AAAA,CAAI,CAAC;AAAA,MACtE,SAAS,OAAO;AACd,mBAAW;AAAA,UACT,QAAQ,OAAO,GAAG,KAAK,UAAU,EAAE,MAAM,SAAS,OAAO,eAAe,KAAK,EAAE,CAAC,CAAC;AAAA,CAAI;AAAA,QACvF;AACA,mBAAW,MAAM;AAAA,MACnB;AAAA,IACF;AAAA,IACA,MAAM,SAAS;AACb,YAAM,SAAS,SAAS;AAAA,IAC1B;AAAA,EACF,CAAC;AACH;AAEA,SAAS,eAAe,OAAyB;AAC/C,MAAI,iBAAiB,OAAO;AAC1B,WAAO;AAAA,MACL,MAAM,MAAM;AAAA,MACZ,SAAS,MAAM;AAAA,MACf,OAAO,MAAM;AAAA,IACf;AAAA,EACF;AAEA,SAAO;AACT;","names":[]}

package/dist/chunk-YLMDBQO4.js

@@ -1,141 +0,0 @@
-import {
-  AgentBuilder
-} from "./chunk-JI5YZWNT.js";
-import {
-  extractRagText
-} from "./chunk-BALBBVI3.js";
-import {
-  createTool
-} from "./chunk-MELYDKWQ.js";
-import {
-  CompletionRequestBuilder
-} from "./chunk-Q25IWLBT.js";
-import {
-  CompletionCapabilityError,
-  Message,
-  Usage
-} from "./chunk-ADH7NNCS.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, ...this.agent.providerTools]).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);
-}
-
-export {
-  ExtractionError,
-  Extractor,
-  ExtractorBuilder
-};
-//# sourceMappingURL=chunk-YLMDBQO4.js.map

package/dist/chunk-YLMDBQO4.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/extractor/extractor.ts"],"sourcesContent":["import type { Agent } from \"../agent/agent\";\nimport { AgentBuilder } from \"../agent/builder\";\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 { extractRagText } from \"../internal/rag-text\";\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, ...this.agent.providerTools])\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"],"mappings":";;;;;;;;;;;;;;;;;;;AAiBA,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,CAAC,GAAG,UAAU,GAAG,KAAK,MAAM,aAAa,CAAC,EAChD,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;","names":[]}

package/dist/hooks/index.d.ts

@@ -1,12 +0,0 @@
-import { H as HookAction, P as PromptHook, T as ToolApprovalRequestOptions, a as ToolCallHookAction, R as RunControl, b as ToolCallControl } from '../types-D9JpeAPF.js';
-export { C as CompletionCallHookArgs, c as CompletionErrorHookArgs, d as CompletionResponseHookArgs, e as HookResult, f as RunEndHookArgs, g as RunErrorHookArgs, h as RunStartHookArgs, i as ToolCallHookArgs, j as ToolCallHookResult, k as ToolErrorHookArgs, l as ToolHookArgs, m as ToolResultHookArgs, n as TurnEndHookArgs, o as TurnStartHookArgs } from '../types-D9JpeAPF.js';
-import '../types-G12i56s2.js';
-
-declare function createHook<RawResponse = unknown>(hook: PromptHook<RawResponse>): PromptHook<RawResponse>;
-declare function cancelPrompt(reason: string): HookAction;
-declare function skipTool(reason: string): ToolCallHookAction;
-declare function requestToolApproval(options?: ToolApprovalRequestOptions): ToolCallHookAction;
-declare const runControl: RunControl;
-declare const toolCallControl: ToolCallControl;
-
-export { HookAction, PromptHook, RunControl, ToolApprovalRequestOptions, ToolCallControl, ToolCallHookAction, cancelPrompt, createHook, requestToolApproval, runControl, skipTool, toolCallControl };

package/dist/hooks/index.js

@@ -1,17 +0,0 @@
-import {
-  cancelPrompt,
-  createHook,
-  requestToolApproval,
-  runControl,
-  skipTool,
-  toolCallControl
-} from "../chunk-2ODTMRHP.js";
-export {
-  cancelPrompt,
-  createHook,
-  requestToolApproval,
-  runControl,
-  skipTool,
-  toolCallControl
-};
-//# sourceMappingURL=index.js.map

package/dist/hooks/index.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/index-BxSbmOMr.d.ts

@@ -1,257 +0,0 @@
-import { Y as ReasoningContentType, p as ToolCall, h as CompletionSource, o as ProviderToolCall, U as Usage, j as ContextUsage, M as Message, f as CompletionRequest, u as ToolResultContent, g as CompletionResponse, _ as ToolCallArgumentsMode, J as JsonObject, O as CompletionModelCapabilities, r as ToolDefinition, l as JsonValue } from './types-G12i56s2.js';
-import { GuardrailDecisionRecord } from './guardrails/index.js';
-import { g as ToolCallStreamEvent } from './tool-Cohdv5GC.js';
-
-type PromptResponse = {
-    output: string;
-    usage: Usage;
-    contextUsage?: ContextUsage | undefined;
-    messages: Message[];
-    trace?: AgentTraceInfo | undefined;
-    guardrails?: GuardrailDecisionRecord[] | undefined;
-    sources?: CompletionSource[] | undefined;
-    providerToolCalls?: ProviderToolCall[] | undefined;
-};
-type AgentDeltaEvent = {
-    type: "text_delta";
-    delta: string;
-} | {
-    type: "reasoning_delta";
-    delta: string;
-    id?: string;
-    contentType?: ReasoningContentType;
-    signature?: string;
-} | {
-    type: "tool_call";
-    toolCall: ToolCall;
-} | {
-    type: "source";
-    source: CompletionSource;
-} | {
-    type: "provider_tool_call";
-    toolCall: ProviderToolCall;
-};
-type AgentErrorStreamEvent = {
-    type: "error";
-    error: unknown;
-    usage: Usage;
-};
-type AgentStreamOptions = {
-    /** @default true */
-    includeToolCallDeltas?: boolean;
-};
-type AgentToolCallDeltaEvent = {
-    type: "tool_call_delta";
-    turn: number;
-    id: string;
-    callId?: string;
-    name?: string;
-    argumentsDelta?: string;
-    argumentsMode?: ToolCallArgumentsMode;
-    signature?: string;
-};
-type AgentChildStreamEventBase<RawResponse = unknown> = {
-    type: "turn_start";
-    turn: number;
-    prompt: Message;
-    history: Message[];
-} | {
-    type: "generation_start";
-    turn: number;
-    request: CompletionRequest;
-    modelInfo: AgentGenerationModelInfo;
-} | {
-    type: "text_delta";
-    turn: number;
-    delta: string;
-} | {
-    type: "reasoning_delta";
-    turn: number;
-    delta: string;
-    id?: string;
-    contentType?: ReasoningContentType;
-    signature?: string;
-} | {
-    type: "tool_call";
-    turn: number;
-    toolCall: ToolCall;
-} | {
-    type: "source";
-    turn: number;
-    source: CompletionSource;
-} | {
-    type: "provider_tool_call";
-    turn: number;
-    toolCall: ProviderToolCall;
-} | {
-    type: "tool_result";
-    turn: number;
-    toolName: string;
-    toolCallId?: string;
-    internalCallId: string;
-    args: string;
-    result: string;
-    structuredResult?: ToolResultContent[] | undefined;
-} | {
-    type: "turn_end";
-    turn: number;
-    response: CompletionResponse<RawResponse>;
-    firstDeltaMs?: number | undefined;
-} | {
-    type: "guardrail_decision";
-    turn?: number | undefined;
-    decision: GuardrailDecisionRecord;
-} | {
-    type: "final";
-    runId: string;
-    output: string;
-    usage: Usage;
-    contextUsage?: ContextUsage | undefined;
-    messages: Message[];
-    trace?: AgentTraceInfo | undefined;
-    guardrails?: GuardrailDecisionRecord[] | undefined;
-    sources?: CompletionSource[] | undefined;
-    providerToolCalls?: ProviderToolCall[] | undefined;
-} | AgentErrorStreamEvent;
-type AgentChildStreamEventWithoutToolCallDeltas<RawResponse = unknown> = AgentChildStreamEventBase<RawResponse>;
-type AgentChildStreamEvent<RawResponse = unknown> = AgentChildStreamEventBase<RawResponse> | AgentToolCallDeltaEvent;
-type AgentChildStreamEventWithToolCallDeltas<RawResponse = unknown> = AgentChildStreamEvent<RawResponse>;
-type AgentToolStreamEvent<ChildEvent> = {
-    type: "agent_tool_event";
-    turn: number;
-    toolName: string;
-    toolCallId?: string;
-    internalCallId: string;
-    agentId: string;
-    agentName?: string;
-    event: ChildEvent;
-};
-type AgentStreamEventWithoutToolCallDeltas<RawResponse = unknown> = AgentChildStreamEventWithoutToolCallDeltas<RawResponse> | AgentToolStreamEvent<AgentChildStreamEventWithoutToolCallDeltas<RawResponse>>;
-type AgentStreamEvent<RawResponse = unknown> = AgentChildStreamEvent<RawResponse> | AgentToolStreamEvent<AgentChildStreamEvent<RawResponse>>;
-type AgentStreamEventWithToolCallDeltas<RawResponse = unknown> = AgentStreamEvent<RawResponse>;
-
-type AgentTraceInfo = {
-    traceId?: string | undefined;
-    observationId?: string | undefined;
-};
-type AgentTraceOptions = {
-    name?: string | undefined;
-    userId?: string | undefined;
-    sessionId?: string | undefined;
-    metadata?: Record<string, unknown> | undefined;
-    tags?: string[] | undefined;
-    version?: string | undefined;
-    traceId?: string | undefined;
-    failOnObserverError?: boolean | undefined;
-    promptRef?: AgentRunPromptRef | undefined;
-};
-type AgentRunPromptRef = {
-    name: string;
-    version?: number | undefined;
-};
-type AgentRunStartArgs = {
-    agentName?: string | undefined;
-    agentDescription?: string | undefined;
-    instructions?: string | undefined;
-    trace?: AgentTraceOptions | undefined;
-    prompt: Message;
-    promptRef?: AgentRunPromptRef | undefined;
-    history: Message[];
-    maxTurns: number;
-};
-type AgentRunEndArgs = {
-    output: string;
-    usage: Usage;
-    messages: Message[];
-    sources?: CompletionSource[] | undefined;
-    providerToolCalls?: ProviderToolCall[] | undefined;
-};
-type AgentRunErrorArgs = {
-    error: unknown;
-    usage: Usage;
-    messages: Message[];
-};
-type AgentGenerationModelInfo = {
-    provider: string;
-    defaultModel: string;
-    capabilities?: CompletionModelCapabilities | undefined;
-};
-type AgentGenerationStartArgs = {
-    turn: number;
-    request: CompletionRequest;
-    providerRequest?: JsonObject | undefined;
-    modelInfo?: AgentGenerationModelInfo | undefined;
-};
-type AgentGenerationEndArgs<RawResponse = unknown> = {
-    turn: number;
-    response: CompletionResponse<RawResponse>;
-    firstDeltaMs?: number | undefined;
-};
-type AgentGenerationErrorArgs = {
-    turn: number;
-    error: unknown;
-};
-type AgentGenerationUpdateArgs = {
-    turn: number;
-    delta: AgentDeltaEvent;
-};
-type AgentRunEventArgs = {
-    name: string;
-    attributes?: Record<string, JsonValue | undefined> | undefined;
-    level?: "DEFAULT" | "WARNING" | "ERROR" | undefined;
-    timestamp?: Date | string | undefined;
-};
-type AgentToolStartArgs = {
-    turn: number;
-    toolCall: ToolCall;
-    toolName: string;
-    args: string;
-    internalCallId: string;
-    toolCallId?: string | undefined;
-    toolDefinition?: ToolDefinition | undefined;
-    toolMetadata?: JsonObject | undefined;
-};
-type AgentToolEndArgs = AgentToolStartArgs & {
-    result: string;
-    structuredResult?: ToolResultContent[] | undefined;
-    skipped: boolean;
-};
-type AgentToolErrorArgs = AgentToolStartArgs & {
-    error: unknown;
-};
-type AgentToolStreamEventArgs = AgentToolStartArgs & {
-    event: ToolCallStreamEvent;
-};
-interface AgentGenerationObserver {
-    end(args: AgentGenerationEndArgs): void | Promise<void>;
-    error?(args: AgentGenerationErrorArgs): void | Promise<void>;
-    update?(args: AgentGenerationUpdateArgs): void | Promise<void>;
-}
-interface AgentToolObserver {
-    streamEvent?(args: AgentToolStreamEventArgs): void | Promise<void>;
-    end(args: AgentToolEndArgs): void | Promise<void>;
-    error?(args: AgentToolErrorArgs): void | Promise<void>;
-}
-interface AgentRunObserver {
-    readonly trace?: AgentTraceInfo | undefined;
-    startGeneration?(args: AgentGenerationStartArgs): AgentGenerationObserver | undefined | Promise<AgentGenerationObserver | undefined>;
-    startTool?(args: AgentToolStartArgs): AgentToolObserver | undefined | Promise<AgentToolObserver | undefined>;
-    end(args: AgentRunEndArgs): void | Promise<void>;
-    error?(args: AgentRunErrorArgs): void | Promise<void>;
-    event?(args: AgentRunEventArgs): void | Promise<void>;
-}
-interface AgentObserver {
-    startRun(args: AgentRunStartArgs): AgentRunObserver | undefined | Promise<AgentRunObserver | undefined>;
-    flush?(): void | Promise<void>;
-    shutdown?(): void | Promise<void>;
-}
-type AgentObserverRegistration = {
-    observer: AgentObserver;
-    failOnObserverError?: boolean | undefined;
-};
-type ObserveOptions = {
-    failOnObserverError?: boolean | undefined;
-};
-declare function createObserver(observer: AgentObserver): AgentObserver;
-
-export { type AgentChildStreamEvent as A, type AgentToolObserver as B, type AgentToolStartArgs as C, type AgentToolStreamEventArgs as D, type AgentTraceInfo as E, createObserver as F, type ObserveOptions as O, type PromptResponse as P, type AgentChildStreamEventWithToolCallDeltas as a, type AgentChildStreamEventWithoutToolCallDeltas as b, type AgentErrorStreamEvent as c, type AgentStreamEvent as d, type AgentStreamEventWithToolCallDeltas as e, type AgentStreamEventWithoutToolCallDeltas as f, type AgentStreamOptions as g, type AgentToolCallDeltaEvent as h, type AgentObserver as i, type AgentTraceOptions as j, type AgentObserverRegistration as k, type AgentDeltaEvent as l, type AgentGenerationEndArgs as m, type AgentGenerationErrorArgs as n, type AgentGenerationModelInfo as o, type AgentGenerationObserver as p, type AgentGenerationStartArgs as q, type AgentGenerationUpdateArgs as r, type AgentRunEndArgs as s, type AgentRunErrorArgs as t, type AgentRunEventArgs as u, type AgentRunObserver as v, type AgentRunPromptRef as w, type AgentRunStartArgs as x, type AgentToolEndArgs as y, type AgentToolErrorArgs as z };

package/dist/middleware-Dp3d_pxi.d.ts

@@ -1,90 +0,0 @@
-import { r as ToolDefinition, f as CompletionRequest, g as CompletionResponse, l as JsonValue, u as ToolResultContent } from './types-G12i56s2.js';
-import { V as VectorMetadata, E as EmbeddingModel, c as EmbeddedDocument } from './types-BCTRUGex.js';
-import { A as AnyTool, f as ToolCallContext, N as NormalizedToolOutput } from './tool-Cohdv5GC.js';
-import { a as VectorSearchIndex } from './types-O-abTIin.js';
-
-declare class ToolSet {
-    private readonly tools;
-    static fromTools(tools: AnyTool[]): ToolSet;
-    addTool(tool: AnyTool): this;
-    addTools(tools: AnyTool[] | ToolSet): this;
-    deleteTool(toolName: string): boolean;
-    contains(toolName: string): boolean;
-    get(toolName: string): AnyTool | undefined;
-    values(): AnyTool[];
-    getToolDefinitions(prompt?: string): Promise<ToolDefinition[]>;
-    call(toolName: string, args: string, context?: ToolCallContext): Promise<NormalizedToolOutput>;
-}
-
-type ToolSearchDocument<Metadata extends VectorMetadata = VectorMetadata> = {
-    toolName: string;
-    definition: ToolDefinition;
-    text: string;
-    metadata?: Metadata | undefined;
-};
-type EmbedToolsOptions<Metadata extends VectorMetadata = VectorMetadata> = {
-    content?: ((tool: AnyTool, definition: ToolDefinition) => string | string[]) | undefined;
-    metadata?: ((tool: AnyTool, definition: ToolDefinition) => Metadata | undefined) | undefined;
-    concurrency?: number | undefined;
-};
-interface DynamicToolIndex<Metadata extends VectorMetadata = VectorMetadata> extends VectorSearchIndex<ToolSearchDocument<Metadata>, Metadata> {
-    readonly toolSet: ToolSet;
-}
-declare function embedTools<Metadata extends VectorMetadata = VectorMetadata>(model: EmbeddingModel, tools: AnyTool[] | ToolSet, options?: EmbedToolsOptions<Metadata>): Promise<Array<EmbeddedDocument<ToolSearchDocument<Metadata>, Metadata>>>;
-declare function createToolIndex<Metadata extends VectorMetadata = VectorMetadata>(model: EmbeddingModel, tools: AnyTool[] | ToolSet, options?: EmbedToolsOptions<Metadata>): Promise<DynamicToolIndex<Metadata>>;
-declare function isDynamicToolIndex(value: unknown): value is DynamicToolIndex;
-
-type MaybePromise<T> = T | Promise<T>;
-type CompletionRequestMiddlewareArgs = {
-    turn: number;
-    request: CompletionRequest;
-    originalRequest: CompletionRequest;
-};
-type CompletionRequestMiddlewareResult = {
-    request: CompletionRequest;
-} | undefined;
-type CompletionResponseMiddlewareArgs<RawResponse = unknown> = {
-    turn: number;
-    request: CompletionRequest;
-    response: CompletionResponse<RawResponse>;
-    originalResponse: CompletionResponse<RawResponse>;
-};
-type CompletionResponseMiddlewareResult<RawResponse = unknown> = {
-    response: CompletionResponse<RawResponse>;
-} | undefined;
-type ToolInputMiddlewareArgs = {
-    toolName: string;
-    args: string;
-    originalArgs: string;
-    turn: number;
-    toolCallId?: string | undefined;
-    internalCallId: string;
-};
-type ToolInputMiddlewareResult = {
-    args: JsonValue | string;
-} | undefined;
-type ToolResultMiddlewareArgs = {
-    toolName: string;
-    args: string;
-    result: string;
-    originalResult: string;
-    structuredResult?: ToolResultContent[] | undefined;
-    originalStructuredResult?: ToolResultContent[] | undefined;
-    turn: number;
-    toolCallId?: string | undefined;
-    internalCallId: string;
-};
-type ToolOutputMiddlewareArgs = ToolResultMiddlewareArgs;
-type ToolOutputMiddlewareResult = string | {
-    result?: string | undefined;
-    structuredResult?: ToolResultContent[] | undefined;
-} | undefined;
-interface AgentMiddleware<RawResponse = unknown> {
-    onCompletionRequest?(args: CompletionRequestMiddlewareArgs): MaybePromise<CompletionRequestMiddlewareResult>;
-    onCompletionResponse?(args: CompletionResponseMiddlewareArgs<RawResponse>): MaybePromise<CompletionResponseMiddlewareResult<RawResponse>>;
-    onToolInput?(args: ToolInputMiddlewareArgs): MaybePromise<ToolInputMiddlewareResult>;
-    onToolOutput?(args: ToolOutputMiddlewareArgs): MaybePromise<ToolOutputMiddlewareResult>;
-}
-declare function createMiddleware<RawResponse = unknown>(middleware: AgentMiddleware<RawResponse>): AgentMiddleware<RawResponse>;
-
-export { type AgentMiddleware as A, type CompletionRequestMiddlewareArgs as C, type DynamicToolIndex as D, type EmbedToolsOptions as E, type ToolInputMiddlewareArgs as T, type CompletionRequestMiddlewareResult as a, type CompletionResponseMiddlewareArgs as b, type CompletionResponseMiddlewareResult as c, type ToolInputMiddlewareResult as d, type ToolOutputMiddlewareArgs as e, type ToolOutputMiddlewareResult as f, type ToolResultMiddlewareArgs as g, createMiddleware as h, type ToolSearchDocument as i, ToolSet as j, createToolIndex as k, embedTools as l, isDynamicToolIndex as m };