@anvia/core 0.18.0 → 0.21.0

package/dist/{chunk-JNJBKSLA.js → chunk-6X2UTX3L.js}

@@ -1,14 +1,14 @@
 import {
   InMemoryVectorStore,
   createTool
-} from "./chunk-I7TM7TKX.js";
+} from "./chunk-MELYDKWQ.js";
+import {
+  embedDocuments
+} from "./chunk-MNWK2USR.js";
 import {
   isToolResultContentArray,
   serializeToolResultOutput
 } from "./chunk-PJDL5UT2.js";
-import {
-  embedDocuments
-} from "./chunk-YBBD4HZ3.js";
 
 // src/tool/errors.ts
 var ToolCallError = class extends Error {
@@ -188,9 +188,6 @@ function defaultToolEmbeddingText(definition) {
 function createMiddleware(middleware) {
   return middleware;
 }
-function createToolMiddleware(middleware) {
-  return middleware;
-}
 
 // src/tool/think-tool.ts
 import { z } from "zod";
@@ -221,7 +218,6 @@ export {
   createToolIndex,
   isDynamicToolIndex,
   createMiddleware,
-  createToolMiddleware,
   createThinkTool
 };
-//# sourceMappingURL=chunk-JNJBKSLA.js.map
+//# sourceMappingURL=chunk-6X2UTX3L.js.map

package/dist/{chunk-JNJBKSLA.js.map → chunk-6X2UTX3L.js.map}

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

package/dist/{chunk-I7TM7TKX.js → chunk-MELYDKWQ.js}

@@ -1,10 +1,10 @@
-import {
-  toProviderJsonSchema
-} from "./chunk-WQKHFADH.js";
 import {
   cosineSimilarity,
   embedText
-} from "./chunk-YBBD4HZ3.js";
+} from "./chunk-MNWK2USR.js";
+import {
+  toProviderJsonSchema
+} from "./chunk-WQKHFADH.js";
 
 // src/vector-store/index.ts
 import { z } from "zod";
@@ -334,4 +334,4 @@ export {
   InMemoryVectorIndex,
   createVectorSearchTool
 };
-//# sourceMappingURL=chunk-I7TM7TKX.js.map
+//# sourceMappingURL=chunk-MELYDKWQ.js.map

package/dist/{chunk-YBBD4HZ3.js → chunk-MNWK2USR.js}

@@ -73,17 +73,29 @@ async function embedTexts(model, texts) {
   }
   return embeddings;
 }
+async function embedSparseTexts(model, texts) {
+  if (texts.length === 0) {
+    return [];
+  }
+  const maxBatchSize = Math.max(1, Math.trunc(model.maxBatchSize ?? texts.length));
+  const batches = [];
+  for (let index = 0; index < texts.length; index += maxBatchSize) {
+    batches.push(texts.slice(index, index + maxBatchSize));
+  }
+  const results = await mapWithConcurrency(batches, 1, (batch) => model.embedTexts(batch));
+  const embeddings = results.flat();
+  if (embeddings.length !== texts.length) {
+    throw new Error(
+      `Sparse embedding model returned ${embeddings.length} embeddings for ${texts.length} texts`
+    );
+  }
+  return embeddings;
+}
+async function embedSparseQuery(model, query) {
+  return model.embedQuery(query);
+}
 async function embedDocuments(model, documents, options) {
-  const prepared = documents.map((document, index) => {
-    const content = options.content(document, index);
-    const texts = Array.isArray(content) ? content : [content];
-    return {
-      id: options.id?.(document, index) ?? `doc${index}`,
-      document,
-      metadata: options.metadata?.(document, index),
-      texts
-    };
-  });
+  const prepared = prepareDocuments(documents, options);
   const flatTexts = prepared.flatMap(
     (item, documentIndex) => item.texts.map((text) => ({ documentIndex, text }))
   );
@@ -110,11 +122,100 @@ async function embedDocuments(model, documents, options) {
     byDocument.set(item.documentIndex, list);
   }
   return prepared.map((item, index) => {
-    const embeddings2 = byDocument.get(index) ?? [];
+    const documentEmbeddings = byDocument.get(index) ?? [];
+    if (item.metadata === void 0) {
+      return { id: item.id, document: item.document, embeddings: documentEmbeddings };
+    }
+    return {
+      id: item.id,
+      document: item.document,
+      metadata: item.metadata,
+      embeddings: documentEmbeddings
+    };
+  });
+}
+async function embedHybridDocuments(models, documents, options) {
+  const prepared = prepareDocuments(documents, options);
+  const flatTexts = prepared.flatMap(
+    (item, documentIndex) => item.texts.map((text) => ({ documentIndex, text }))
+  );
+  const concurrency = Math.max(1, Math.trunc(options.concurrency ?? 1));
+  const denseBatches = chunk(
+    flatTexts,
+    Math.max(1, Math.trunc(models.dense.maxBatchSize ?? (flatTexts.length || 1)))
+  );
+  const sparseBatches = chunk(
+    flatTexts,
+    Math.max(1, Math.trunc(models.sparse.maxBatchSize ?? (flatTexts.length || 1)))
+  );
+  const [denseResults, sparseResults] = await Promise.all([
+    mapWithConcurrency(denseBatches, concurrency, async (batch) => {
+      const batchEmbeddings = await models.dense.embedTexts(batch.map((item) => item.text));
+      if (batchEmbeddings.length !== batch.length) {
+        throw new Error(
+          `Embedding model returned ${batchEmbeddings.length} embeddings for ${batch.length} texts`
+        );
+      }
+      return batch.map((item, index) => ({
+        documentIndex: item.documentIndex,
+        embedding: batchEmbeddings[index]
+      }));
+    }),
+    mapWithConcurrency(sparseBatches, concurrency, async (batch) => {
+      const batchEmbeddings = await models.sparse.embedTexts(batch.map((item) => item.text));
+      if (batchEmbeddings.length !== batch.length) {
+        throw new Error(
+          `Sparse embedding model returned ${batchEmbeddings.length} embeddings for ${batch.length} texts`
+        );
+      }
+      return batch.map((item, index) => ({
+        documentIndex: item.documentIndex,
+        embedding: batchEmbeddings[index]
+      }));
+    })
+  ]);
+  const denseByDocument = /* @__PURE__ */ new Map();
+  for (const item of denseResults.flat()) {
+    const list = denseByDocument.get(item.documentIndex) ?? [];
+    list.push(item.embedding);
+    denseByDocument.set(item.documentIndex, list);
+  }
+  const sparseByDocument = /* @__PURE__ */ new Map();
+  for (const item of sparseResults.flat()) {
+    const list = sparseByDocument.get(item.documentIndex) ?? [];
+    list.push(item.embedding);
+    sparseByDocument.set(item.documentIndex, list);
+  }
+  return prepared.map((item, index) => {
+    const embeddings = denseByDocument.get(index) ?? [];
+    const sparseEmbeddings = sparseByDocument.get(index) ?? [];
+    if (embeddings.length !== sparseEmbeddings.length) {
+      throw new Error(
+        `Hybrid embedding produced ${embeddings.length} dense and ${sparseEmbeddings.length} sparse vectors for document ${item.id}`
+      );
+    }
     if (item.metadata === void 0) {
-      return { id: item.id, document: item.document, embeddings: embeddings2 };
+      return { id: item.id, document: item.document, embeddings, sparseEmbeddings };
     }
-    return { id: item.id, document: item.document, metadata: item.metadata, embeddings: embeddings2 };
+    return {
+      id: item.id,
+      document: item.document,
+      metadata: item.metadata,
+      embeddings,
+      sparseEmbeddings
+    };
+  });
+}
+function prepareDocuments(documents, options) {
+  return documents.map((document, index) => {
+    const content = options.content(document, index);
+    const texts = Array.isArray(content) ? content : [content];
+    return {
+      id: options.id?.(document, index) ?? `doc${index}`,
+      document,
+      metadata: options.metadata?.(document, index),
+      texts
+    };
   });
 }
 function chunk(items, size) {
@@ -134,6 +235,9 @@ export {
   chebyshevDistance,
   embedText,
   embedTexts,
-  embedDocuments
+  embedSparseTexts,
+  embedSparseQuery,
+  embedDocuments,
+  embedHybridDocuments
 };
-//# sourceMappingURL=chunk-YBBD4HZ3.js.map
+//# sourceMappingURL=chunk-MNWK2USR.js.map

package/dist/chunk-MNWK2USR.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/embeddings/distance.ts","../src/embeddings/embed.ts"],"sourcesContent":["export function dotProduct(left: number[], right: number[]): number {\n  assertSameDimensions(left, right);\n  return left.reduce((sum, value, index) => sum + value * (right[index] as number), 0);\n}\n\nexport function cosineSimilarity(left: number[], right: number[]): number {\n  assertSameDimensions(left, right);\n  const leftMagnitude = magnitude(left);\n  const rightMagnitude = magnitude(right);\n  if (leftMagnitude === 0 || rightMagnitude === 0) {\n    return 0;\n  }\n  return dotProduct(left, right) / (leftMagnitude * rightMagnitude);\n}\n\nexport function angularDistance(left: number[], right: number[]): number {\n  const similarity = Math.max(-1, Math.min(1, cosineSimilarity(left, right)));\n  return Math.acos(similarity) / Math.PI;\n}\n\nexport function euclideanDistance(left: number[], right: number[]): number {\n  assertSameDimensions(left, right);\n  return Math.sqrt(\n    left.reduce((sum, value, index) => sum + (value - (right[index] as number)) ** 2, 0),\n  );\n}\n\nexport function manhattanDistance(left: number[], right: number[]): number {\n  assertSameDimensions(left, right);\n  return left.reduce((sum, value, index) => sum + Math.abs(value - (right[index] as number)), 0);\n}\n\nexport function chebyshevDistance(left: number[], right: number[]): number {\n  assertSameDimensions(left, right);\n  return left.reduce(\n    (max, value, index) => Math.max(max, Math.abs(value - (right[index] as number))),\n    0,\n  );\n}\n\nfunction magnitude(vector: number[]): number {\n  return Math.sqrt(vector.reduce((sum, value) => sum + value ** 2, 0));\n}\n\nfunction assertSameDimensions(left: number[], right: number[]): void {\n  if (left.length !== right.length) {\n    throw new Error(`Vector dimension mismatch: ${left.length} !== ${right.length}`);\n  }\n}\n","import { mapWithConcurrency } from \"../internal/concurrency\";\nimport type {\n  EmbedDocumentsOptions,\n  EmbeddedDocument,\n  Embedding,\n  EmbeddingModel,\n  EmbedHybridDocumentsOptions,\n  SparseEmbedding,\n  SparseEmbeddingModel,\n  VectorMetadata,\n} from \"./types\";\n\nexport async function embedText(model: EmbeddingModel, text: string): Promise<Embedding> {\n  const embeddings = await embedTexts(model, [text]);\n  const embedding = embeddings[0];\n  if (embedding === undefined) {\n    throw new Error(\"Embedding model returned no embeddings\");\n  }\n  return embedding;\n}\n\nexport async function embedTexts(model: EmbeddingModel, texts: string[]): Promise<Embedding[]> {\n  if (texts.length === 0) {\n    return [];\n  }\n\n  const maxBatchSize = Math.max(1, Math.trunc(model.maxBatchSize ?? texts.length));\n  const batches: string[][] = [];\n  for (let index = 0; index < texts.length; index += maxBatchSize) {\n    batches.push(texts.slice(index, index + maxBatchSize));\n  }\n\n  const results = await mapWithConcurrency(batches, 1, (batch) => model.embedTexts(batch));\n  const embeddings = results.flat();\n  if (embeddings.length !== texts.length) {\n    throw new Error(\n      `Embedding model returned ${embeddings.length} embeddings for ${texts.length} texts`,\n    );\n  }\n  return embeddings;\n}\n\nexport async function embedSparseTexts(\n  model: SparseEmbeddingModel,\n  texts: string[],\n): Promise<SparseEmbedding[]> {\n  if (texts.length === 0) {\n    return [];\n  }\n\n  const maxBatchSize = Math.max(1, Math.trunc(model.maxBatchSize ?? texts.length));\n  const batches: string[][] = [];\n  for (let index = 0; index < texts.length; index += maxBatchSize) {\n    batches.push(texts.slice(index, index + maxBatchSize));\n  }\n\n  const results = await mapWithConcurrency(batches, 1, (batch) => model.embedTexts(batch));\n  const embeddings = results.flat();\n  if (embeddings.length !== texts.length) {\n    throw new Error(\n      `Sparse embedding model returned ${embeddings.length} embeddings for ${texts.length} texts`,\n    );\n  }\n  return embeddings;\n}\n\nexport async function embedSparseQuery(\n  model: SparseEmbeddingModel,\n  query: string,\n): Promise<SparseEmbedding> {\n  return model.embedQuery(query);\n}\n\nexport async function embedDocuments<T, Metadata extends VectorMetadata = VectorMetadata>(\n  model: EmbeddingModel,\n  documents: T[],\n  options: EmbedDocumentsOptions<T, Metadata>,\n): Promise<Array<EmbeddedDocument<T, Metadata>>> {\n  const prepared = prepareDocuments(documents, options);\n\n  const flatTexts = prepared.flatMap((item, documentIndex) =>\n    item.texts.map((text) => ({ documentIndex, text })),\n  );\n  const embeddings = await mapWithConcurrency(\n    chunk(flatTexts, Math.max(1, Math.trunc(model.maxBatchSize ?? (flatTexts.length || 1)))),\n    Math.max(1, Math.trunc(options.concurrency ?? 1)),\n    async (batch) => {\n      const batchEmbeddings = await model.embedTexts(batch.map((item) => item.text));\n      if (batchEmbeddings.length !== batch.length) {\n        throw new Error(\n          `Embedding model returned ${batchEmbeddings.length} embeddings for ${batch.length} texts`,\n        );\n      }\n      return batch.map((item, index) => ({\n        documentIndex: item.documentIndex,\n        embedding: batchEmbeddings[index] as Embedding,\n      }));\n    },\n  );\n\n  const byDocument = new Map<number, Embedding[]>();\n  for (const item of embeddings.flat()) {\n    const list = byDocument.get(item.documentIndex) ?? [];\n    list.push(item.embedding);\n    byDocument.set(item.documentIndex, list);\n  }\n\n  return prepared.map((item, index) => {\n    const documentEmbeddings = byDocument.get(index) ?? [];\n    if (item.metadata === undefined) {\n      return { id: item.id, document: item.document, embeddings: documentEmbeddings };\n    }\n    return {\n      id: item.id,\n      document: item.document,\n      metadata: item.metadata,\n      embeddings: documentEmbeddings,\n    };\n  });\n}\n\nexport async function embedHybridDocuments<T, Metadata extends VectorMetadata = VectorMetadata>(\n  models: EmbedHybridDocumentsOptions,\n  documents: T[],\n  options: EmbedDocumentsOptions<T, Metadata>,\n): Promise<Array<EmbeddedDocument<T, Metadata>>> {\n  const prepared = prepareDocuments(documents, options);\n  const flatTexts = prepared.flatMap((item, documentIndex) =>\n    item.texts.map((text) => ({ documentIndex, text })),\n  );\n  const concurrency = Math.max(1, Math.trunc(options.concurrency ?? 1));\n\n  const denseBatches = chunk(\n    flatTexts,\n    Math.max(1, Math.trunc(models.dense.maxBatchSize ?? (flatTexts.length || 1))),\n  );\n  const sparseBatches = chunk(\n    flatTexts,\n    Math.max(1, Math.trunc(models.sparse.maxBatchSize ?? (flatTexts.length || 1))),\n  );\n\n  const [denseResults, sparseResults] = await Promise.all([\n    mapWithConcurrency(denseBatches, concurrency, async (batch) => {\n      const batchEmbeddings = await models.dense.embedTexts(batch.map((item) => item.text));\n      if (batchEmbeddings.length !== batch.length) {\n        throw new Error(\n          `Embedding model returned ${batchEmbeddings.length} embeddings for ${batch.length} texts`,\n        );\n      }\n      return batch.map((item, index) => ({\n        documentIndex: item.documentIndex,\n        embedding: batchEmbeddings[index] as Embedding,\n      }));\n    }),\n    mapWithConcurrency(sparseBatches, concurrency, async (batch) => {\n      const batchEmbeddings = await models.sparse.embedTexts(batch.map((item) => item.text));\n      if (batchEmbeddings.length !== batch.length) {\n        throw new Error(\n          `Sparse embedding model returned ${batchEmbeddings.length} embeddings for ${batch.length} texts`,\n        );\n      }\n      return batch.map((item, index) => ({\n        documentIndex: item.documentIndex,\n        embedding: batchEmbeddings[index] as SparseEmbedding,\n      }));\n    }),\n  ]);\n\n  const denseByDocument = new Map<number, Embedding[]>();\n  for (const item of denseResults.flat()) {\n    const list = denseByDocument.get(item.documentIndex) ?? [];\n    list.push(item.embedding);\n    denseByDocument.set(item.documentIndex, list);\n  }\n\n  const sparseByDocument = new Map<number, SparseEmbedding[]>();\n  for (const item of sparseResults.flat()) {\n    const list = sparseByDocument.get(item.documentIndex) ?? [];\n    list.push(item.embedding);\n    sparseByDocument.set(item.documentIndex, list);\n  }\n\n  return prepared.map((item, index) => {\n    const embeddings = denseByDocument.get(index) ?? [];\n    const sparseEmbeddings = sparseByDocument.get(index) ?? [];\n    if (embeddings.length !== sparseEmbeddings.length) {\n      throw new Error(\n        `Hybrid embedding produced ${embeddings.length} dense and ${sparseEmbeddings.length} sparse vectors for document ${item.id}`,\n      );\n    }\n    if (item.metadata === undefined) {\n      return { id: item.id, document: item.document, embeddings, sparseEmbeddings };\n    }\n    return {\n      id: item.id,\n      document: item.document,\n      metadata: item.metadata,\n      embeddings,\n      sparseEmbeddings,\n    };\n  });\n}\n\nfunction prepareDocuments<T, Metadata extends VectorMetadata>(\n  documents: T[],\n  options: EmbedDocumentsOptions<T, Metadata>,\n) {\n  return documents.map((document, index) => {\n    const content = options.content(document, index);\n    const texts = Array.isArray(content) ? content : [content];\n    return {\n      id: options.id?.(document, index) ?? `doc${index}`,\n      document,\n      metadata: options.metadata?.(document, index),\n      texts,\n    };\n  });\n}\n\nfunction chunk<T>(items: T[], size: number): T[][] {\n  const chunks: T[][] = [];\n  for (let index = 0; index < items.length; index += size) {\n    chunks.push(items.slice(index, index + size));\n  }\n  return chunks;\n}\n"],"mappings":";;;;;AAAO,SAAS,WAAW,MAAgB,OAAyB;AAClE,uBAAqB,MAAM,KAAK;AAChC,SAAO,KAAK,OAAO,CAAC,KAAK,OAAO,UAAU,MAAM,QAAS,MAAM,KAAK,GAAc,CAAC;AACrF;AAEO,SAAS,iBAAiB,MAAgB,OAAyB;AACxE,uBAAqB,MAAM,KAAK;AAChC,QAAM,gBAAgB,UAAU,IAAI;AACpC,QAAM,iBAAiB,UAAU,KAAK;AACtC,MAAI,kBAAkB,KAAK,mBAAmB,GAAG;AAC/C,WAAO;AAAA,EACT;AACA,SAAO,WAAW,MAAM,KAAK,KAAK,gBAAgB;AACpD;AAEO,SAAS,gBAAgB,MAAgB,OAAyB;AACvE,QAAM,aAAa,KAAK,IAAI,IAAI,KAAK,IAAI,GAAG,iBAAiB,MAAM,KAAK,CAAC,CAAC;AAC1E,SAAO,KAAK,KAAK,UAAU,IAAI,KAAK;AACtC;AAEO,SAAS,kBAAkB,MAAgB,OAAyB;AACzE,uBAAqB,MAAM,KAAK;AAChC,SAAO,KAAK;AAAA,IACV,KAAK,OAAO,CAAC,KAAK,OAAO,UAAU,OAAO,QAAS,MAAM,KAAK,MAAiB,GAAG,CAAC;AAAA,EACrF;AACF;AAEO,SAAS,kBAAkB,MAAgB,OAAyB;AACzE,uBAAqB,MAAM,KAAK;AAChC,SAAO,KAAK,OAAO,CAAC,KAAK,OAAO,UAAU,MAAM,KAAK,IAAI,QAAS,MAAM,KAAK,CAAY,GAAG,CAAC;AAC/F;AAEO,SAAS,kBAAkB,MAAgB,OAAyB;AACzE,uBAAqB,MAAM,KAAK;AAChC,SAAO,KAAK;AAAA,IACV,CAAC,KAAK,OAAO,UAAU,KAAK,IAAI,KAAK,KAAK,IAAI,QAAS,MAAM,KAAK,CAAY,CAAC;AAAA,IAC/E;AAAA,EACF;AACF;AAEA,SAAS,UAAU,QAA0B;AAC3C,SAAO,KAAK,KAAK,OAAO,OAAO,CAAC,KAAK,UAAU,MAAM,SAAS,GAAG,CAAC,CAAC;AACrE;AAEA,SAAS,qBAAqB,MAAgB,OAAuB;AACnE,MAAI,KAAK,WAAW,MAAM,QAAQ;AAChC,UAAM,IAAI,MAAM,8BAA8B,KAAK,MAAM,QAAQ,MAAM,MAAM,EAAE;AAAA,EACjF;AACF;;;ACpCA,eAAsB,UAAU,OAAuB,MAAkC;AACvF,QAAM,aAAa,MAAM,WAAW,OAAO,CAAC,IAAI,CAAC;AACjD,QAAM,YAAY,WAAW,CAAC;AAC9B,MAAI,cAAc,QAAW;AAC3B,UAAM,IAAI,MAAM,wCAAwC;AAAA,EAC1D;AACA,SAAO;AACT;AAEA,eAAsB,WAAW,OAAuB,OAAuC;AAC7F,MAAI,MAAM,WAAW,GAAG;AACtB,WAAO,CAAC;AAAA,EACV;AAEA,QAAM,eAAe,KAAK,IAAI,GAAG,KAAK,MAAM,MAAM,gBAAgB,MAAM,MAAM,CAAC;AAC/E,QAAM,UAAsB,CAAC;AAC7B,WAAS,QAAQ,GAAG,QAAQ,MAAM,QAAQ,SAAS,cAAc;AAC/D,YAAQ,KAAK,MAAM,MAAM,OAAO,QAAQ,YAAY,CAAC;AAAA,EACvD;AAEA,QAAM,UAAU,MAAM,mBAAmB,SAAS,GAAG,CAAC,UAAU,MAAM,WAAW,KAAK,CAAC;AACvF,QAAM,aAAa,QAAQ,KAAK;AAChC,MAAI,WAAW,WAAW,MAAM,QAAQ;AACtC,UAAM,IAAI;AAAA,MACR,4BAA4B,WAAW,MAAM,mBAAmB,MAAM,MAAM;AAAA,IAC9E;AAAA,EACF;AACA,SAAO;AACT;AAEA,eAAsB,iBACpB,OACA,OAC4B;AAC5B,MAAI,MAAM,WAAW,GAAG;AACtB,WAAO,CAAC;AAAA,EACV;AAEA,QAAM,eAAe,KAAK,IAAI,GAAG,KAAK,MAAM,MAAM,gBAAgB,MAAM,MAAM,CAAC;AAC/E,QAAM,UAAsB,CAAC;AAC7B,WAAS,QAAQ,GAAG,QAAQ,MAAM,QAAQ,SAAS,cAAc;AAC/D,YAAQ,KAAK,MAAM,MAAM,OAAO,QAAQ,YAAY,CAAC;AAAA,EACvD;AAEA,QAAM,UAAU,MAAM,mBAAmB,SAAS,GAAG,CAAC,UAAU,MAAM,WAAW,KAAK,CAAC;AACvF,QAAM,aAAa,QAAQ,KAAK;AAChC,MAAI,WAAW,WAAW,MAAM,QAAQ;AACtC,UAAM,IAAI;AAAA,MACR,mCAAmC,WAAW,MAAM,mBAAmB,MAAM,MAAM;AAAA,IACrF;AAAA,EACF;AACA,SAAO;AACT;AAEA,eAAsB,iBACpB,OACA,OAC0B;AAC1B,SAAO,MAAM,WAAW,KAAK;AAC/B;AAEA,eAAsB,eACpB,OACA,WACA,SAC+C;AAC/C,QAAM,WAAW,iBAAiB,WAAW,OAAO;AAEpD,QAAM,YAAY,SAAS;AAAA,IAAQ,CAAC,MAAM,kBACxC,KAAK,MAAM,IAAI,CAAC,UAAU,EAAE,eAAe,KAAK,EAAE;AAAA,EACpD;AACA,QAAM,aAAa,MAAM;AAAA,IACvB,MAAM,WAAW,KAAK,IAAI,GAAG,KAAK,MAAM,MAAM,iBAAiB,UAAU,UAAU,EAAE,CAAC,CAAC;AAAA,IACvF,KAAK,IAAI,GAAG,KAAK,MAAM,QAAQ,eAAe,CAAC,CAAC;AAAA,IAChD,OAAO,UAAU;AACf,YAAM,kBAAkB,MAAM,MAAM,WAAW,MAAM,IAAI,CAAC,SAAS,KAAK,IAAI,CAAC;AAC7E,UAAI,gBAAgB,WAAW,MAAM,QAAQ;AAC3C,cAAM,IAAI;AAAA,UACR,4BAA4B,gBAAgB,MAAM,mBAAmB,MAAM,MAAM;AAAA,QACnF;AAAA,MACF;AACA,aAAO,MAAM,IAAI,CAAC,MAAM,WAAW;AAAA,QACjC,eAAe,KAAK;AAAA,QACpB,WAAW,gBAAgB,KAAK;AAAA,MAClC,EAAE;AAAA,IACJ;AAAA,EACF;AAEA,QAAM,aAAa,oBAAI,IAAyB;AAChD,aAAW,QAAQ,WAAW,KAAK,GAAG;AACpC,UAAM,OAAO,WAAW,IAAI,KAAK,aAAa,KAAK,CAAC;AACpD,SAAK,KAAK,KAAK,SAAS;AACxB,eAAW,IAAI,KAAK,eAAe,IAAI;AAAA,EACzC;AAEA,SAAO,SAAS,IAAI,CAAC,MAAM,UAAU;AACnC,UAAM,qBAAqB,WAAW,IAAI,KAAK,KAAK,CAAC;AACrD,QAAI,KAAK,aAAa,QAAW;AAC/B,aAAO,EAAE,IAAI,KAAK,IAAI,UAAU,KAAK,UAAU,YAAY,mBAAmB;AAAA,IAChF;AACA,WAAO;AAAA,MACL,IAAI,KAAK;AAAA,MACT,UAAU,KAAK;AAAA,MACf,UAAU,KAAK;AAAA,MACf,YAAY;AAAA,IACd;AAAA,EACF,CAAC;AACH;AAEA,eAAsB,qBACpB,QACA,WACA,SAC+C;AAC/C,QAAM,WAAW,iBAAiB,WAAW,OAAO;AACpD,QAAM,YAAY,SAAS;AAAA,IAAQ,CAAC,MAAM,kBACxC,KAAK,MAAM,IAAI,CAAC,UAAU,EAAE,eAAe,KAAK,EAAE;AAAA,EACpD;AACA,QAAM,cAAc,KAAK,IAAI,GAAG,KAAK,MAAM,QAAQ,eAAe,CAAC,CAAC;AAEpE,QAAM,eAAe;AAAA,IACnB;AAAA,IACA,KAAK,IAAI,GAAG,KAAK,MAAM,OAAO,MAAM,iBAAiB,UAAU,UAAU,EAAE,CAAC;AAAA,EAC9E;AACA,QAAM,gBAAgB;AAAA,IACpB;AAAA,IACA,KAAK,IAAI,GAAG,KAAK,MAAM,OAAO,OAAO,iBAAiB,UAAU,UAAU,EAAE,CAAC;AAAA,EAC/E;AAEA,QAAM,CAAC,cAAc,aAAa,IAAI,MAAM,QAAQ,IAAI;AAAA,IACtD,mBAAmB,cAAc,aAAa,OAAO,UAAU;AAC7D,YAAM,kBAAkB,MAAM,OAAO,MAAM,WAAW,MAAM,IAAI,CAAC,SAAS,KAAK,IAAI,CAAC;AACpF,UAAI,gBAAgB,WAAW,MAAM,QAAQ;AAC3C,cAAM,IAAI;AAAA,UACR,4BAA4B,gBAAgB,MAAM,mBAAmB,MAAM,MAAM;AAAA,QACnF;AAAA,MACF;AACA,aAAO,MAAM,IAAI,CAAC,MAAM,WAAW;AAAA,QACjC,eAAe,KAAK;AAAA,QACpB,WAAW,gBAAgB,KAAK;AAAA,MAClC,EAAE;AAAA,IACJ,CAAC;AAAA,IACD,mBAAmB,eAAe,aAAa,OAAO,UAAU;AAC9D,YAAM,kBAAkB,MAAM,OAAO,OAAO,WAAW,MAAM,IAAI,CAAC,SAAS,KAAK,IAAI,CAAC;AACrF,UAAI,gBAAgB,WAAW,MAAM,QAAQ;AAC3C,cAAM,IAAI;AAAA,UACR,mCAAmC,gBAAgB,MAAM,mBAAmB,MAAM,MAAM;AAAA,QAC1F;AAAA,MACF;AACA,aAAO,MAAM,IAAI,CAAC,MAAM,WAAW;AAAA,QACjC,eAAe,KAAK;AAAA,QACpB,WAAW,gBAAgB,KAAK;AAAA,MAClC,EAAE;AAAA,IACJ,CAAC;AAAA,EACH,CAAC;AAED,QAAM,kBAAkB,oBAAI,IAAyB;AACrD,aAAW,QAAQ,aAAa,KAAK,GAAG;AACtC,UAAM,OAAO,gBAAgB,IAAI,KAAK,aAAa,KAAK,CAAC;AACzD,SAAK,KAAK,KAAK,SAAS;AACxB,oBAAgB,IAAI,KAAK,eAAe,IAAI;AAAA,EAC9C;AAEA,QAAM,mBAAmB,oBAAI,IAA+B;AAC5D,aAAW,QAAQ,cAAc,KAAK,GAAG;AACvC,UAAM,OAAO,iBAAiB,IAAI,KAAK,aAAa,KAAK,CAAC;AAC1D,SAAK,KAAK,KAAK,SAAS;AACxB,qBAAiB,IAAI,KAAK,eAAe,IAAI;AAAA,EAC/C;AAEA,SAAO,SAAS,IAAI,CAAC,MAAM,UAAU;AACnC,UAAM,aAAa,gBAAgB,IAAI,KAAK,KAAK,CAAC;AAClD,UAAM,mBAAmB,iBAAiB,IAAI,KAAK,KAAK,CAAC;AACzD,QAAI,WAAW,WAAW,iBAAiB,QAAQ;AACjD,YAAM,IAAI;AAAA,QACR,6BAA6B,WAAW,MAAM,cAAc,iBAAiB,MAAM,gCAAgC,KAAK,EAAE;AAAA,MAC5H;AAAA,IACF;AACA,QAAI,KAAK,aAAa,QAAW;AAC/B,aAAO,EAAE,IAAI,KAAK,IAAI,UAAU,KAAK,UAAU,YAAY,iBAAiB;AAAA,IAC9E;AACA,WAAO;AAAA,MACL,IAAI,KAAK;AAAA,MACT,UAAU,KAAK;AAAA,MACf,UAAU,KAAK;AAAA,MACf;AAAA,MACA;AAAA,IACF;AAAA,EACF,CAAC;AACH;AAEA,SAAS,iBACP,WACA,SACA;AACA,SAAO,UAAU,IAAI,CAAC,UAAU,UAAU;AACxC,UAAM,UAAU,QAAQ,QAAQ,UAAU,KAAK;AAC/C,UAAM,QAAQ,MAAM,QAAQ,OAAO,IAAI,UAAU,CAAC,OAAO;AACzD,WAAO;AAAA,MACL,IAAI,QAAQ,KAAK,UAAU,KAAK,KAAK,MAAM,KAAK;AAAA,MAChD;AAAA,MACA,UAAU,QAAQ,WAAW,UAAU,KAAK;AAAA,MAC5C;AAAA,IACF;AAAA,EACF,CAAC;AACH;AAEA,SAAS,MAAS,OAAY,MAAqB;AACjD,QAAM,SAAgB,CAAC;AACvB,WAAS,QAAQ,GAAG,QAAQ,MAAM,QAAQ,SAAS,MAAM;AACvD,WAAO,KAAK,MAAM,MAAM,OAAO,QAAQ,IAAI,CAAC;AAAA,EAC9C;AACA,SAAO;AACT;","names":[]}

package/dist/{chunk-QABIBFKE.js → chunk-N3YV3WCO.js}

@@ -1,15 +1,15 @@
 import {
   PromptRequest
-} from "./chunk-7LCKIULT.js";
+} from "./chunk-3IYTCGHS.js";
 import {
   isSkillTool
 } from "./chunk-YK4WAAS4.js";
 import {
   ToolSet
-} from "./chunk-JNJBKSLA.js";
+} from "./chunk-6X2UTX3L.js";
 import {
   createTool
-} from "./chunk-I7TM7TKX.js";
+} from "./chunk-MELYDKWQ.js";
 import {
   isStreamingCompletionModel
 } from "./chunk-F56AMXMQ.js";
@@ -53,10 +53,6 @@ var Agent = class {
   dynamicContexts;
   dynamicTools;
   middlewares;
-  /**
-   * @deprecated Use `middlewares` instead.
-   */
-  toolMiddlewares;
   memory;
   eventStore;
   constructor(options) {
@@ -80,8 +76,7 @@ var Agent = class {
     this.guardrails = [...options.guardrails ?? []];
     this.dynamicContexts = [...options.dynamicContexts ?? []];
     this.dynamicTools = [...options.dynamicTools ?? []];
-    this.middlewares = [...options.middlewares ?? options.toolMiddlewares ?? []];
-    this.toolMiddlewares = this.middlewares;
+    this.middlewares = [...options.middlewares ?? []];
     this.memory = options.memory;
     this.eventStore = options.eventStore;
   }
@@ -210,4 +205,4 @@ export {
   Agent,
   AgentSession
 };
-//# sourceMappingURL=chunk-QABIBFKE.js.map
+//# sourceMappingURL=chunk-N3YV3WCO.js.map

package/dist/chunk-N3YV3WCO.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/agent/agent.ts","../src/agent/ids.ts"],"sourcesContent":["import { z } from \"zod\";\nimport { isStreamingCompletionModel } from \"../completion/create-completion\";\nimport type {\n  CompletionModel,\n  Document,\n  JsonObject,\n  JsonValue,\n  Message as MessageType,\n  ProviderTool,\n  ToolChoice,\n} from \"../completion/index\";\nimport type { GuardrailPolicy } from \"../guardrails\";\nimport type { PromptHook } from \"../hooks\";\nimport type { MemoryContext, MemoryRegistration, SessionOptions } from \"../memory/types\";\nimport type { AgentObserverRegistration } from \"../observability\";\nimport { PromptRequest } from \"../request\";\nimport { createTool } from \"../tool/create-tool\";\nimport type { ToolSearchDocument } from \"../tool/dynamic-tools\";\nimport type { AgentMiddleware } from \"../tool/middleware\";\nimport { isSkillTool } from \"../tool/skill-tool-marker\";\nimport type {\n  AnyTool,\n  NormalizedToolOutput,\n  Tool,\n  ToolApprovalsOptions,\n  ToolCallContext,\n  ToolCallStreamEvent,\n} from \"../tool/tool\";\nimport { ToolSet } from \"../tool/tool-set\";\nimport type { VectorSearchIndex } from \"../vector-store\";\nimport { normalizeAgentId } from \"./ids\";\nimport type {\n  AgentEventStoreRegistration,\n  AgentOptions,\n  AgentToolOptions,\n  DynamicContextRegistration,\n  DynamicToolRegistration,\n} from \"./types\";\n\nexport const DEFAULT_MAX_TURNS = 20;\n\nexport class Agent<M extends CompletionModel = CompletionModel> {\n  readonly id: string;\n  readonly name: string | undefined;\n  readonly description: string | undefined;\n  readonly model: M;\n  readonly instructions: string | undefined;\n  readonly staticContext: Document[];\n  readonly temperature: number | undefined;\n  readonly maxTokens: number | undefined;\n  readonly additionalParams: JsonValue | undefined;\n  readonly toolSet: ToolSet;\n  readonly providerTools: ProviderTool[];\n  readonly toolChoice: ToolChoice | undefined;\n  readonly defaultMaxTurns: number | undefined;\n  readonly hook: PromptHook | undefined;\n  readonly outputSchema: JsonObject | undefined;\n  readonly observers: AgentObserverRegistration[];\n  readonly approvals: ToolApprovalsOptions | undefined;\n  readonly guardrails: GuardrailPolicy[];\n  readonly dynamicContexts: DynamicContextRegistration[];\n  readonly dynamicTools: DynamicToolRegistration[];\n  readonly middlewares: AgentMiddleware[];\n  readonly memory: MemoryRegistration | undefined;\n  readonly eventStore: AgentEventStoreRegistration | undefined;\n\n  constructor(options: AgentOptions<M>) {\n    this.id = normalizeAgentId(options.id);\n    this.name = options.name;\n    this.description = options.description;\n    this.model = options.model;\n    this.instructions = options.instructions;\n    this.staticContext = [...(options.staticContext ?? [])];\n    this.temperature = options.temperature;\n    this.maxTokens = options.maxTokens;\n    this.additionalParams = options.additionalParams;\n    this.toolSet = options.toolSet ?? new ToolSet();\n    this.providerTools = [...(options.providerTools ?? [])];\n    this.toolChoice = options.toolChoice;\n    this.defaultMaxTurns = options.defaultMaxTurns ?? DEFAULT_MAX_TURNS;\n    this.hook = options.hook;\n    this.outputSchema = options.outputSchema;\n    this.observers = [...(options.observers ?? [])];\n    this.approvals = options.approvals;\n    this.guardrails = [...(options.guardrails ?? [])];\n    this.dynamicContexts = [...(options.dynamicContexts ?? [])];\n    this.dynamicTools = [...(options.dynamicTools ?? [])];\n    this.middlewares = [...(options.middlewares ?? [])];\n    this.memory = options.memory;\n    this.eventStore = options.eventStore;\n  }\n\n  prompt(prompt: string | MessageType | MessageType[]): PromptRequest<M> {\n    return PromptRequest.fromAgent(this, prompt);\n  }\n\n  session(sessionId: string, options: SessionOptions = {}): AgentSession<M> {\n    if (this.memory === undefined) {\n      throw new Error(`Agent \"${this.id}\" has no memory store configured.`);\n    }\n    const normalized = sessionId.trim();\n    if (normalized.length === 0) {\n      throw new TypeError(\"Session id must be a non-empty string.\");\n    }\n    const context: MemoryContext = {\n      sessionId: normalized,\n    };\n    if (options.userId !== undefined) {\n      context.userId = options.userId;\n    }\n    if (options.metadata !== undefined) {\n      context.metadata = options.metadata;\n    }\n    return new AgentSession(this, context);\n  }\n\n  asTool(options: AgentToolOptions): Tool<{ prompt: string }, string> {\n    const description =\n      options.description ?? this.description ?? `Prompt the ${options.name} agent.`;\n\n    return createTool({\n      name: options.name,\n      description,\n      input: z.object({\n        prompt: z.string().describe(\"The prompt to send to the agent.\"),\n      }),\n      output: z.string(),\n      execute: async ({ prompt }, context: ToolCallContext) => {\n        const request = this.prompt(prompt);\n        const childRequest =\n          options.maxTurns === undefined ? request : request.maxTurns(options.maxTurns);\n        if (\n          options.stream === true &&\n          context.emitStreamEvent !== undefined &&\n          this.model.capabilities.streaming &&\n          isStreamingCompletionModel(this.model)\n        ) {\n          let output = \"\";\n          const childStream =\n            context.includeToolCallDeltas === false\n              ? childRequest.stream({ includeToolCallDeltas: false })\n              : childRequest.stream();\n          for await (const event of childStream) {\n            const streamEvent: ToolCallStreamEvent = {\n              agentId: this.id,\n              event,\n            };\n            if (this.name !== undefined) {\n              streamEvent.agentName = this.name;\n            }\n            await context.emitStreamEvent(streamEvent);\n            if (event.type === \"final\") {\n              output = event.output;\n            }\n          }\n          return output;\n        }\n        const response = await childRequest.send();\n        return response.output;\n      },\n    });\n  }\n\n  getTool(toolName: string): AnyTool | undefined {\n    const staticTool = this.toolSet.get(toolName);\n    if (staticTool !== undefined) {\n      return staticTool;\n    }\n\n    for (const registration of this.dynamicTools) {\n      const dynamicTool = dynamicToolSetFromIndex(registration.index)?.get(toolName);\n      if (dynamicTool !== undefined) {\n        return dynamicTool;\n      }\n    }\n\n    return undefined;\n  }\n\n  async callTool(\n    toolName: string,\n    args: string,\n    context?: ToolCallContext,\n  ): Promise<NormalizedToolOutput> {\n    if (this.toolSet.contains(toolName)) {\n      return this.toolSet.call(toolName, args, context);\n    }\n\n    for (const registration of this.dynamicTools) {\n      const toolSet = dynamicToolSetFromIndex(registration.index);\n      if (toolSet?.contains(toolName)) {\n        return toolSet.call(toolName, args, context);\n      }\n    }\n\n    return this.toolSet.call(toolName, args, context);\n  }\n\n  shouldApplyToolMiddleware(toolName: string): boolean {\n    return !isSkillTool(this.getTool(toolName));\n  }\n}\n\nexport class AgentSession<M extends CompletionModel = CompletionModel> {\n  constructor(\n    private readonly agent: Agent<M>,\n    private readonly context: {\n      sessionId: string;\n      userId?: string | undefined;\n      metadata?: JsonObject | undefined;\n    },\n  ) {}\n\n  prompt(prompt: string | MessageType): PromptRequest<M> {\n    if (Array.isArray(prompt)) {\n      throw new TypeError(\"AgentSession.prompt does not accept Message[] transcripts.\");\n    }\n    return PromptRequest.fromAgent(this.agent, prompt, { memoryContext: this.context });\n  }\n\n  async messages(): Promise<MessageType[]> {\n    const memory = this.agent.memory;\n    if (memory === undefined) {\n      throw new Error(`Agent \"${this.agent.id}\" has no memory store configured.`);\n    }\n    return memory.store.load(this.context);\n  }\n\n  async clear(): Promise<void> {\n    const memory = this.agent.memory;\n    if (memory === undefined) {\n      throw new Error(`Agent \"${this.agent.id}\" has no memory store configured.`);\n    }\n    await memory.store.clear(this.context);\n  }\n}\n\nfunction dynamicToolSetFromIndex(\n  index: VectorSearchIndex<ToolSearchDocument>,\n): ToolSet | undefined {\n  const maybeIndex = index as { toolSet?: unknown };\n  return maybeIndex.toolSet instanceof ToolSet ? maybeIndex.toolSet : undefined;\n}\n","export function normalizeAgentId(id: string): string {\n  if (typeof id !== \"string\") {\n    throw new TypeError(\"Agent id must be a string.\");\n  }\n\n  const normalized = id.trim();\n  if (normalized.length === 0) {\n    throw new TypeError(\"Agent id must be a non-empty string.\");\n  }\n\n  return normalized;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;AAAA,SAAS,SAAS;;;ACAX,SAAS,iBAAiB,IAAoB;AACnD,MAAI,OAAO,OAAO,UAAU;AAC1B,UAAM,IAAI,UAAU,4BAA4B;AAAA,EAClD;AAEA,QAAM,aAAa,GAAG,KAAK;AAC3B,MAAI,WAAW,WAAW,GAAG;AAC3B,UAAM,IAAI,UAAU,sCAAsC;AAAA,EAC5D;AAEA,SAAO;AACT;;;AD4BO,IAAM,oBAAoB;AAE1B,IAAM,QAAN,MAAyD;AAAA,EACrD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAET,YAAY,SAA0B;AACpC,SAAK,KAAK,iBAAiB,QAAQ,EAAE;AACrC,SAAK,OAAO,QAAQ;AACpB,SAAK,cAAc,QAAQ;AAC3B,SAAK,QAAQ,QAAQ;AACrB,SAAK,eAAe,QAAQ;AAC5B,SAAK,gBAAgB,CAAC,GAAI,QAAQ,iBAAiB,CAAC,CAAE;AACtD,SAAK,cAAc,QAAQ;AAC3B,SAAK,YAAY,QAAQ;AACzB,SAAK,mBAAmB,QAAQ;AAChC,SAAK,UAAU,QAAQ,WAAW,IAAI,QAAQ;AAC9C,SAAK,gBAAgB,CAAC,GAAI,QAAQ,iBAAiB,CAAC,CAAE;AACtD,SAAK,aAAa,QAAQ;AAC1B,SAAK,kBAAkB,QAAQ,mBAAmB;AAClD,SAAK,OAAO,QAAQ;AACpB,SAAK,eAAe,QAAQ;AAC5B,SAAK,YAAY,CAAC,GAAI,QAAQ,aAAa,CAAC,CAAE;AAC9C,SAAK,YAAY,QAAQ;AACzB,SAAK,aAAa,CAAC,GAAI,QAAQ,cAAc,CAAC,CAAE;AAChD,SAAK,kBAAkB,CAAC,GAAI,QAAQ,mBAAmB,CAAC,CAAE;AAC1D,SAAK,eAAe,CAAC,GAAI,QAAQ,gBAAgB,CAAC,CAAE;AACpD,SAAK,cAAc,CAAC,GAAI,QAAQ,eAAe,CAAC,CAAE;AAClD,SAAK,SAAS,QAAQ;AACtB,SAAK,aAAa,QAAQ;AAAA,EAC5B;AAAA,EAEA,OAAO,QAAgE;AACrE,WAAO,cAAc,UAAU,MAAM,MAAM;AAAA,EAC7C;AAAA,EAEA,QAAQ,WAAmB,UAA0B,CAAC,GAAoB;AACxE,QAAI,KAAK,WAAW,QAAW;AAC7B,YAAM,IAAI,MAAM,UAAU,KAAK,EAAE,mCAAmC;AAAA,IACtE;AACA,UAAM,aAAa,UAAU,KAAK;AAClC,QAAI,WAAW,WAAW,GAAG;AAC3B,YAAM,IAAI,UAAU,wCAAwC;AAAA,IAC9D;AACA,UAAM,UAAyB;AAAA,MAC7B,WAAW;AAAA,IACb;AACA,QAAI,QAAQ,WAAW,QAAW;AAChC,cAAQ,SAAS,QAAQ;AAAA,IAC3B;AACA,QAAI,QAAQ,aAAa,QAAW;AAClC,cAAQ,WAAW,QAAQ;AAAA,IAC7B;AACA,WAAO,IAAI,aAAa,MAAM,OAAO;AAAA,EACvC;AAAA,EAEA,OAAO,SAA6D;AAClE,UAAM,cACJ,QAAQ,eAAe,KAAK,eAAe,cAAc,QAAQ,IAAI;AAEvE,WAAO,WAAW;AAAA,MAChB,MAAM,QAAQ;AAAA,MACd;AAAA,MACA,OAAO,EAAE,OAAO;AAAA,QACd,QAAQ,EAAE,OAAO,EAAE,SAAS,kCAAkC;AAAA,MAChE,CAAC;AAAA,MACD,QAAQ,EAAE,OAAO;AAAA,MACjB,SAAS,OAAO,EAAE,OAAO,GAAG,YAA6B;AACvD,cAAM,UAAU,KAAK,OAAO,MAAM;AAClC,cAAM,eACJ,QAAQ,aAAa,SAAY,UAAU,QAAQ,SAAS,QAAQ,QAAQ;AAC9E,YACE,QAAQ,WAAW,QACnB,QAAQ,oBAAoB,UAC5B,KAAK,MAAM,aAAa,aACxB,2BAA2B,KAAK,KAAK,GACrC;AACA,cAAI,SAAS;AACb,gBAAM,cACJ,QAAQ,0BAA0B,QAC9B,aAAa,OAAO,EAAE,uBAAuB,MAAM,CAAC,IACpD,aAAa,OAAO;AAC1B,2BAAiB,SAAS,aAAa;AACrC,kBAAM,cAAmC;AAAA,cACvC,SAAS,KAAK;AAAA,cACd;AAAA,YACF;AACA,gBAAI,KAAK,SAAS,QAAW;AAC3B,0BAAY,YAAY,KAAK;AAAA,YAC/B;AACA,kBAAM,QAAQ,gBAAgB,WAAW;AACzC,gBAAI,MAAM,SAAS,SAAS;AAC1B,uBAAS,MAAM;AAAA,YACjB;AAAA,UACF;AACA,iBAAO;AAAA,QACT;AACA,cAAM,WAAW,MAAM,aAAa,KAAK;AACzC,eAAO,SAAS;AAAA,MAClB;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,QAAQ,UAAuC;AAC7C,UAAM,aAAa,KAAK,QAAQ,IAAI,QAAQ;AAC5C,QAAI,eAAe,QAAW;AAC5B,aAAO;AAAA,IACT;AAEA,eAAW,gBAAgB,KAAK,cAAc;AAC5C,YAAM,cAAc,wBAAwB,aAAa,KAAK,GAAG,IAAI,QAAQ;AAC7E,UAAI,gBAAgB,QAAW;AAC7B,eAAO;AAAA,MACT;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,SACJ,UACA,MACA,SAC+B;AAC/B,QAAI,KAAK,QAAQ,SAAS,QAAQ,GAAG;AACnC,aAAO,KAAK,QAAQ,KAAK,UAAU,MAAM,OAAO;AAAA,IAClD;AAEA,eAAW,gBAAgB,KAAK,cAAc;AAC5C,YAAM,UAAU,wBAAwB,aAAa,KAAK;AAC1D,UAAI,SAAS,SAAS,QAAQ,GAAG;AAC/B,eAAO,QAAQ,KAAK,UAAU,MAAM,OAAO;AAAA,MAC7C;AAAA,IACF;AAEA,WAAO,KAAK,QAAQ,KAAK,UAAU,MAAM,OAAO;AAAA,EAClD;AAAA,EAEA,0BAA0B,UAA2B;AACnD,WAAO,CAAC,YAAY,KAAK,QAAQ,QAAQ,CAAC;AAAA,EAC5C;AACF;AAEO,IAAM,eAAN,MAAgE;AAAA,EACrE,YACmB,OACA,SAKjB;AANiB;AACA;AAAA,EAKhB;AAAA,EANgB;AAAA,EACA;AAAA,EAOnB,OAAO,QAAgD;AACrD,QAAI,MAAM,QAAQ,MAAM,GAAG;AACzB,YAAM,IAAI,UAAU,4DAA4D;AAAA,IAClF;AACA,WAAO,cAAc,UAAU,KAAK,OAAO,QAAQ,EAAE,eAAe,KAAK,QAAQ,CAAC;AAAA,EACpF;AAAA,EAEA,MAAM,WAAmC;AACvC,UAAM,SAAS,KAAK,MAAM;AAC1B,QAAI,WAAW,QAAW;AACxB,YAAM,IAAI,MAAM,UAAU,KAAK,MAAM,EAAE,mCAAmC;AAAA,IAC5E;AACA,WAAO,OAAO,MAAM,KAAK,KAAK,OAAO;AAAA,EACvC;AAAA,EAEA,MAAM,QAAuB;AAC3B,UAAM,SAAS,KAAK,MAAM;AAC1B,QAAI,WAAW,QAAW;AACxB,YAAM,IAAI,MAAM,UAAU,KAAK,MAAM,EAAE,mCAAmC;AAAA,IAC5E;AACA,UAAM,OAAO,MAAM,MAAM,KAAK,OAAO;AAAA,EACvC;AACF;AAEA,SAAS,wBACP,OACqB;AACrB,QAAM,aAAa;AACnB,SAAO,WAAW,mBAAmB,UAAU,WAAW,UAAU;AACtE;","names":[]}

package/dist/{chunk-DC2GJDPR.js → chunk-NCPN4SJR.js}

@@ -6,7 +6,7 @@ import {
 } from "./chunk-YK4WAAS4.js";
 import {
   createTool
-} from "./chunk-I7TM7TKX.js";
+} from "./chunk-MELYDKWQ.js";
 
 // src/skills/instructions.ts
 function skillInstructions(skills) {
@@ -395,4 +395,4 @@ export {
   SkillValidationError,
   skill
 };
-//# sourceMappingURL=chunk-DC2GJDPR.js.map
+//# sourceMappingURL=chunk-NCPN4SJR.js.map

package/dist/{chunk-DPTZ2MK4.js → chunk-WHAPIIEZ.js}

@@ -1,10 +1,10 @@
 import {
   Agent,
   normalizeAgentId
-} from "./chunk-QABIBFKE.js";
+} from "./chunk-N3YV3WCO.js";
 import {
   ToolSet
-} from "./chunk-JNJBKSLA.js";
+} from "./chunk-6X2UTX3L.js";
 import {
   resolveMemoryOptions
 } from "./chunk-ZNTIUOKL.js";
@@ -138,18 +138,6 @@ var AgentBuilder = class {
     this.middlewareRegistrations.push(...middlewares);
     return this;
   }
-  /**
-   * @deprecated Use `middleware` instead.
-   */
-  toolMiddleware(middleware) {
-    return this.middleware(middleware);
-  }
-  /**
-   * @deprecated Use `middlewares` instead.
-   */
-  toolMiddlewares(middlewares) {
-    return this.middlewares(middlewares);
-  }
   observe(observer, options = {}) {
     this.observerRegistrations.push({
       observer,
@@ -229,4 +217,4 @@ var AgentBuilder = class {
 export {
   AgentBuilder
 };
-//# sourceMappingURL=chunk-DPTZ2MK4.js.map
+//# sourceMappingURL=chunk-WHAPIIEZ.js.map

package/dist/chunk-WHAPIIEZ.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/agent/builder.ts"],"sourcesContent":["import {\n  type CompletionModel,\n  type Document,\n  isProviderTool,\n  type JsonObject,\n  type JsonValue,\n  type ProviderTool,\n  type ToolChoice,\n} from \"../completion\";\nimport type { GuardrailPolicy, GuardrailPolicyInput } from \"../guardrails\";\nimport { appendGuardrailPolicies } from \"../guardrails\";\nimport type { PromptHook } from \"../hooks\";\nimport type { McpServer } from \"../mcp\";\nimport { resolveMemoryOptions } from \"../memory/options\";\nimport type { MemoryOptions, MemoryRegistration, MemoryStore } from \"../memory/types\";\nimport type { AgentObserver, AgentObserverRegistration, ObserveOptions } from \"../observability\";\nimport { toProviderJsonSchema, type ZodSchema } from \"../schema/zod-schema\";\nimport type { SkillSet } from \"../skills\";\nimport type { ToolSearchDocument } from \"../tool/dynamic-tools\";\nimport type { AgentMiddleware } from \"../tool/middleware\";\nimport type { AnyTool, ToolApprovalsOptions } from \"../tool/tool\";\nimport { ToolSet } from \"../tool/tool-set\";\nimport type { VectorSearchIndex } from \"../vector-store\";\nimport { Agent } from \"./agent\";\nimport { normalizeAgentId } from \"./ids\";\nimport type {\n  AgentEventStore,\n  AgentEventStoreOptions,\n  AgentEventStoreRegistration,\n  DynamicContextOptions,\n  DynamicContextRegistration,\n  DynamicToolOptions,\n  DynamicToolRegistration,\n} from \"./types\";\n\nexport class AgentBuilder<M extends CompletionModel = CompletionModel> {\n  private readonly agentId: string;\n  private agentName: string | undefined;\n  private agentDescription: string | undefined;\n  private instructionBlocks: string[] = [];\n  private contextDocs: Document[] = [];\n  private temp: number | undefined;\n  private maxTokenCount: number | undefined;\n  private params: JsonValue | undefined;\n  private choice: ToolChoice | undefined;\n  private turns: number | undefined;\n  private requestHook: PromptHook | undefined;\n  private schema: JsonObject | undefined;\n  private approvalOptions: ToolApprovalsOptions | undefined;\n  private guardrailPolicies: GuardrailPolicy[] = [];\n  private skillInstructionBlocks: string[] = [];\n  private observerRegistrations: AgentObserverRegistration[] = [];\n  private dynamicContextRegistrations: DynamicContextRegistration[] = [];\n  private dynamicToolRegistrations: DynamicToolRegistration[] = [];\n  private middlewareRegistrations: AgentMiddleware[] = [];\n  private memoryRegistration: MemoryRegistration | undefined;\n  private eventStoreRegistration: AgentEventStoreRegistration | undefined;\n  private activeToolSet = new ToolSet();\n  private providerToolDefs: ProviderTool[] = [];\n\n  constructor(\n    agentId: string,\n    private readonly completionModel: M,\n  ) {\n    this.agentId = normalizeAgentId(agentId);\n  }\n\n  name(name: string): this {\n    this.agentName = name;\n    return this;\n  }\n\n  description(description: string): this {\n    this.agentDescription = description;\n    return this;\n  }\n\n  instructions(instructions: string): this {\n    if (instructions.length > 0) {\n      this.instructionBlocks.push(instructions);\n    }\n    return this;\n  }\n\n  context(text: string, id = `static_doc_${this.contextDocs.length}`): this {\n    this.contextDocs.push({ id, text });\n    return this;\n  }\n\n  dynamicContext<T>(index: VectorSearchIndex<T>, options: DynamicContextOptions<T>): this {\n    this.dynamicContextRegistrations.push({ index, options } as DynamicContextRegistration);\n    return this;\n  }\n\n  dynamicTools(index: VectorSearchIndex<ToolSearchDocument>, options: DynamicToolOptions): this {\n    this.dynamicToolRegistrations.push({ index, options });\n    return this;\n  }\n\n  tool(tool: AnyTool | ProviderTool): this {\n    if (isProviderTool(tool)) {\n      this.providerToolDefs.push(tool);\n    } else {\n      this.activeToolSet.addTool(tool);\n    }\n    return this;\n  }\n\n  tools(tools: Array<AnyTool | ProviderTool>): this {\n    for (const tool of tools) {\n      this.tool(tool);\n    }\n    return this;\n  }\n\n  mcp(servers: McpServer[]): this {\n    for (const server of servers) {\n      this.activeToolSet.addTools(server.tools);\n    }\n    return this;\n  }\n\n  skills(skillSet: SkillSet): this {\n    if (skillSet.instructions.length > 0) {\n      this.skillInstructionBlocks.push(skillSet.instructions);\n    }\n    this.activeToolSet.addTools(skillSet.tools);\n    return this;\n  }\n\n  useToolSet(toolSet: ToolSet): this {\n    toolSet.addTools(this.activeToolSet);\n    this.activeToolSet = toolSet;\n    return this;\n  }\n\n  temperature(temperature: number): this {\n    this.temp = temperature;\n    return this;\n  }\n\n  maxTokens(maxTokens: number): this {\n    this.maxTokenCount = maxTokens;\n    return this;\n  }\n\n  additionalParams(params: JsonValue): this {\n    this.params = params;\n    return this;\n  }\n\n  toolChoice(toolChoice: ToolChoice): this {\n    this.choice = toolChoice;\n    return this;\n  }\n\n  defaultMaxTurns(defaultMaxTurns: number): this {\n    this.turns = defaultMaxTurns;\n    return this;\n  }\n\n  hook(hook: PromptHook): this {\n    this.requestHook = hook;\n    return this;\n  }\n\n  middleware(middleware: AgentMiddleware): this {\n    this.middlewareRegistrations.push(middleware);\n    return this;\n  }\n\n  middlewares(middlewares: AgentMiddleware[]): this {\n    this.middlewareRegistrations.push(...middlewares);\n    return this;\n  }\n\n  observe(observer: AgentObserver, options: ObserveOptions = {}): this {\n    this.observerRegistrations.push({\n      observer,\n      failOnObserverError: options.failOnObserverError,\n    });\n    return this;\n  }\n\n  approvals(options: ToolApprovalsOptions): this {\n    this.approvalOptions = options;\n    return this;\n  }\n\n  guardrails(policies: GuardrailPolicyInput): this {\n    this.guardrailPolicies = appendGuardrailPolicies(this.guardrailPolicies, policies);\n    return this;\n  }\n\n  memory(store: MemoryStore, options: MemoryOptions = {}): this {\n    const resolvedOptions = resolveMemoryOptions(options);\n    if (resolvedOptions.compaction !== undefined && store.compaction === undefined) {\n      throw new TypeError(\n        \"Memory compaction requires a store with the optional compaction capability.\",\n      );\n    }\n    this.memoryRegistration = {\n      store,\n      options: resolvedOptions,\n    };\n    return this;\n  }\n\n  eventStore(store: AgentEventStore, options: AgentEventStoreOptions = {}): this {\n    this.eventStoreRegistration = {\n      store,\n      options: {\n        include: options.include ?? \"all\",\n      },\n    };\n    return this;\n  }\n\n  outputSchema(schema: ZodSchema): this {\n    this.schema = toProviderJsonSchema(schema);\n    return this;\n  }\n\n  build(): Agent<M> {\n    return new Agent({\n      id: this.agentId,\n      name: this.agentName,\n      description: this.agentDescription,\n      model: this.completionModel,\n      instructions: this.buildInstructions(),\n      staticContext: this.contextDocs,\n      temperature: this.temp,\n      maxTokens: this.maxTokenCount,\n      additionalParams: this.params,\n      toolSet: this.activeToolSet,\n      providerTools: this.providerToolDefs,\n      toolChoice: this.choice,\n      defaultMaxTurns: this.turns,\n      hook: this.requestHook,\n      outputSchema: this.schema,\n      observers: this.observerRegistrations,\n      approvals: this.approvalOptions,\n      guardrails: this.guardrailPolicies,\n      dynamicContexts: this.dynamicContextRegistrations,\n      dynamicTools: this.dynamicToolRegistrations,\n      middlewares: this.middlewareRegistrations,\n      memory: this.memoryRegistration,\n      eventStore: this.eventStoreRegistration,\n    });\n  }\n\n  private buildInstructions(): string | undefined {\n    const parts = [...this.instructionBlocks, ...this.skillInstructionBlocks].filter(\n      (part): part is string => part !== undefined && part.length > 0,\n    );\n    return parts.length === 0 ? undefined : parts.join(\"\\n\\n\");\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAmCO,IAAM,eAAN,MAAgE;AAAA,EAyBrE,YACE,SACiB,iBACjB;AADiB;AAEjB,SAAK,UAAU,iBAAiB,OAAO;AAAA,EACzC;AAAA,EAHmB;AAAA,EA1BF;AAAA,EACT;AAAA,EACA;AAAA,EACA,oBAA8B,CAAC;AAAA,EAC/B,cAA0B,CAAC;AAAA,EAC3B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA,oBAAuC,CAAC;AAAA,EACxC,yBAAmC,CAAC;AAAA,EACpC,wBAAqD,CAAC;AAAA,EACtD,8BAA4D,CAAC;AAAA,EAC7D,2BAAsD,CAAC;AAAA,EACvD,0BAA6C,CAAC;AAAA,EAC9C;AAAA,EACA;AAAA,EACA,gBAAgB,IAAI,QAAQ;AAAA,EAC5B,mBAAmC,CAAC;AAAA,EAS5C,KAAK,MAAoB;AACvB,SAAK,YAAY;AACjB,WAAO;AAAA,EACT;AAAA,EAEA,YAAY,aAA2B;AACrC,SAAK,mBAAmB;AACxB,WAAO;AAAA,EACT;AAAA,EAEA,aAAa,cAA4B;AACvC,QAAI,aAAa,SAAS,GAAG;AAC3B,WAAK,kBAAkB,KAAK,YAAY;AAAA,IAC1C;AACA,WAAO;AAAA,EACT;AAAA,EAEA,QAAQ,MAAc,KAAK,cAAc,KAAK,YAAY,MAAM,IAAU;AACxE,SAAK,YAAY,KAAK,EAAE,IAAI,KAAK,CAAC;AAClC,WAAO;AAAA,EACT;AAAA,EAEA,eAAkB,OAA6B,SAAyC;AACtF,SAAK,4BAA4B,KAAK,EAAE,OAAO,QAAQ,CAA+B;AACtF,WAAO;AAAA,EACT;AAAA,EAEA,aAAa,OAA8C,SAAmC;AAC5F,SAAK,yBAAyB,KAAK,EAAE,OAAO,QAAQ,CAAC;AACrD,WAAO;AAAA,EACT;AAAA,EAEA,KAAK,MAAoC;AACvC,QAAI,eAAe,IAAI,GAAG;AACxB,WAAK,iBAAiB,KAAK,IAAI;AAAA,IACjC,OAAO;AACL,WAAK,cAAc,QAAQ,IAAI;AAAA,IACjC;AACA,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,OAA4C;AAChD,eAAW,QAAQ,OAAO;AACxB,WAAK,KAAK,IAAI;AAAA,IAChB;AACA,WAAO;AAAA,EACT;AAAA,EAEA,IAAI,SAA4B;AAC9B,eAAW,UAAU,SAAS;AAC5B,WAAK,cAAc,SAAS,OAAO,KAAK;AAAA,IAC1C;AACA,WAAO;AAAA,EACT;AAAA,EAEA,OAAO,UAA0B;AAC/B,QAAI,SAAS,aAAa,SAAS,GAAG;AACpC,WAAK,uBAAuB,KAAK,SAAS,YAAY;AAAA,IACxD;AACA,SAAK,cAAc,SAAS,SAAS,KAAK;AAC1C,WAAO;AAAA,EACT;AAAA,EAEA,WAAW,SAAwB;AACjC,YAAQ,SAAS,KAAK,aAAa;AACnC,SAAK,gBAAgB;AACrB,WAAO;AAAA,EACT;AAAA,EAEA,YAAY,aAA2B;AACrC,SAAK,OAAO;AACZ,WAAO;AAAA,EACT;AAAA,EAEA,UAAU,WAAyB;AACjC,SAAK,gBAAgB;AACrB,WAAO;AAAA,EACT;AAAA,EAEA,iBAAiB,QAAyB;AACxC,SAAK,SAAS;AACd,WAAO;AAAA,EACT;AAAA,EAEA,WAAW,YAA8B;AACvC,SAAK,SAAS;AACd,WAAO;AAAA,EACT;AAAA,EAEA,gBAAgB,iBAA+B;AAC7C,SAAK,QAAQ;AACb,WAAO;AAAA,EACT;AAAA,EAEA,KAAK,MAAwB;AAC3B,SAAK,cAAc;AACnB,WAAO;AAAA,EACT;AAAA,EAEA,WAAW,YAAmC;AAC5C,SAAK,wBAAwB,KAAK,UAAU;AAC5C,WAAO;AAAA,EACT;AAAA,EAEA,YAAY,aAAsC;AAChD,SAAK,wBAAwB,KAAK,GAAG,WAAW;AAChD,WAAO;AAAA,EACT;AAAA,EAEA,QAAQ,UAAyB,UAA0B,CAAC,GAAS;AACnE,SAAK,sBAAsB,KAAK;AAAA,MAC9B;AAAA,MACA,qBAAqB,QAAQ;AAAA,IAC/B,CAAC;AACD,WAAO;AAAA,EACT;AAAA,EAEA,UAAU,SAAqC;AAC7C,SAAK,kBAAkB;AACvB,WAAO;AAAA,EACT;AAAA,EAEA,WAAW,UAAsC;AAC/C,SAAK,oBAAoB,wBAAwB,KAAK,mBAAmB,QAAQ;AACjF,WAAO;AAAA,EACT;AAAA,EAEA,OAAO,OAAoB,UAAyB,CAAC,GAAS;AAC5D,UAAM,kBAAkB,qBAAqB,OAAO;AACpD,QAAI,gBAAgB,eAAe,UAAa,MAAM,eAAe,QAAW;AAC9E,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AACA,SAAK,qBAAqB;AAAA,MACxB;AAAA,MACA,SAAS;AAAA,IACX;AACA,WAAO;AAAA,EACT;AAAA,EAEA,WAAW,OAAwB,UAAkC,CAAC,GAAS;AAC7E,SAAK,yBAAyB;AAAA,MAC5B;AAAA,MACA,SAAS;AAAA,QACP,SAAS,QAAQ,WAAW;AAAA,MAC9B;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA,EAEA,aAAa,QAAyB;AACpC,SAAK,SAAS,qBAAqB,MAAM;AACzC,WAAO;AAAA,EACT;AAAA,EAEA,QAAkB;AAChB,WAAO,IAAI,MAAM;AAAA,MACf,IAAI,KAAK;AAAA,MACT,MAAM,KAAK;AAAA,MACX,aAAa,KAAK;AAAA,MAClB,OAAO,KAAK;AAAA,MACZ,cAAc,KAAK,kBAAkB;AAAA,MACrC,eAAe,KAAK;AAAA,MACpB,aAAa,KAAK;AAAA,MAClB,WAAW,KAAK;AAAA,MAChB,kBAAkB,KAAK;AAAA,MACvB,SAAS,KAAK;AAAA,MACd,eAAe,KAAK;AAAA,MACpB,YAAY,KAAK;AAAA,MACjB,iBAAiB,KAAK;AAAA,MACtB,MAAM,KAAK;AAAA,MACX,cAAc,KAAK;AAAA,MACnB,WAAW,KAAK;AAAA,MAChB,WAAW,KAAK;AAAA,MAChB,YAAY,KAAK;AAAA,MACjB,iBAAiB,KAAK;AAAA,MACtB,cAAc,KAAK;AAAA,MACnB,aAAa,KAAK;AAAA,MAClB,QAAQ,KAAK;AAAA,MACb,YAAY,KAAK;AAAA,IACnB,CAAC;AAAA,EACH;AAAA,EAEQ,oBAAwC;AAC9C,UAAM,QAAQ,CAAC,GAAG,KAAK,mBAAmB,GAAG,KAAK,sBAAsB,EAAE;AAAA,MACxE,CAAC,SAAyB,SAAS,UAAa,KAAK,SAAS;AAAA,IAChE;AACA,WAAO,MAAM,WAAW,IAAI,SAAY,MAAM,KAAK,MAAM;AAAA,EAC3D;AACF;","names":[]}

package/dist/{chunk-3YJXLOGO.js → chunk-ZSTUMG7G.js}

@@ -1,12 +1,12 @@
 import {
   AgentBuilder
-} from "./chunk-DPTZ2MK4.js";
+} from "./chunk-WHAPIIEZ.js";
 import {
   extractRagText
-} from "./chunk-7LCKIULT.js";
+} from "./chunk-3IYTCGHS.js";
 import {
   createTool
-} from "./chunk-I7TM7TKX.js";
+} from "./chunk-MELYDKWQ.js";
 import {
   CompletionRequestBuilder
 } from "./chunk-MTGNA4OS.js";
@@ -138,4 +138,4 @@ export {
   Extractor,
   ExtractorBuilder
 };
-//# sourceMappingURL=chunk-3YJXLOGO.js.map
+//# sourceMappingURL=chunk-ZSTUMG7G.js.map

package/dist/embeddings/index.d.ts

@@ -1,5 +1,5 @@
-import { V as VectorMetadata, E as EmbeddingModel, b as EmbedDocumentsOptions, c as EmbeddedDocument, d as Embedding } from '../types-DHHZztGO.js';
-export { a as VectorMetadataValue } from '../types-DHHZztGO.js';
+import { V as VectorMetadata, E as EmbeddingModel, b as EmbedDocumentsOptions, c as EmbeddedDocument, d as EmbedHybridDocumentsOptions, S as SparseEmbeddingModel, e as SparseEmbedding, f as Embedding } from '../types-BCTRUGex.js';
+export { g as SparseVector, a as VectorMetadataValue } from '../types-BCTRUGex.js';
 
 declare function dotProduct(left: number[], right: number[]): number;
 declare function cosineSimilarity(left: number[], right: number[]): number;
@@ -10,6 +10,9 @@ declare function chebyshevDistance(left: number[], right: number[]): number;
 
 declare function embedText(model: EmbeddingModel, text: string): Promise<Embedding>;
 declare function embedTexts(model: EmbeddingModel, texts: string[]): Promise<Embedding[]>;
+declare function embedSparseTexts(model: SparseEmbeddingModel, texts: string[]): Promise<SparseEmbedding[]>;
+declare function embedSparseQuery(model: SparseEmbeddingModel, query: string): Promise<SparseEmbedding>;
 declare function embedDocuments<T, Metadata extends VectorMetadata = VectorMetadata>(model: EmbeddingModel, documents: T[], options: EmbedDocumentsOptions<T, Metadata>): Promise<Array<EmbeddedDocument<T, Metadata>>>;
+declare function embedHybridDocuments<T, Metadata extends VectorMetadata = VectorMetadata>(models: EmbedHybridDocumentsOptions, documents: T[], options: EmbedDocumentsOptions<T, Metadata>): Promise<Array<EmbeddedDocument<T, Metadata>>>;
 
-export { EmbedDocumentsOptions, EmbeddedDocument, Embedding, EmbeddingModel, VectorMetadata, angularDistance, chebyshevDistance, cosineSimilarity, dotProduct, embedDocuments, embedText, embedTexts, euclideanDistance, manhattanDistance };
+export { EmbedDocumentsOptions, EmbedHybridDocumentsOptions, EmbeddedDocument, Embedding, EmbeddingModel, SparseEmbedding, SparseEmbeddingModel, VectorMetadata, angularDistance, chebyshevDistance, cosineSimilarity, dotProduct, embedDocuments, embedHybridDocuments, embedSparseQuery, embedSparseTexts, embedText, embedTexts, euclideanDistance, manhattanDistance };

package/dist/embeddings/index.js

@@ -4,11 +4,14 @@ import {
   cosineSimilarity,
   dotProduct,
   embedDocuments,
+  embedHybridDocuments,
+  embedSparseQuery,
+  embedSparseTexts,
   embedText,
   embedTexts,
   euclideanDistance,
   manhattanDistance
-} from "../chunk-YBBD4HZ3.js";
+} from "../chunk-MNWK2USR.js";
 import "../chunk-OIMLU4SF.js";
 export {
   angularDistance,
@@ -16,6 +19,9 @@ export {
   cosineSimilarity,
   dotProduct,
   embedDocuments,
+  embedHybridDocuments,
+  embedSparseQuery,
+  embedSparseTexts,
   embedText,
   embedTexts,
   euclideanDistance,

package/dist/evals/index.d.ts

@@ -1,14 +1,14 @@
-import { A as Agent } from '../agent-CBz3CRxa.js';
+import { A as Agent } from '../agent-Q7FT1xmG.js';
 import { i as JsonValue, M as Message, C as CompletionModel } from '../types-XjIh933x.js';
 import { P as PromptResponse } from '../index-XTUEMWhU.js';
 import { Z as ZodSchema } from '../zod-schema-C7F4clpm.js';
-import { E as EmbeddingModel } from '../types-DHHZztGO.js';
+import { E as EmbeddingModel } from '../types-BCTRUGex.js';
 import '../guardrails/index.js';
 import '../types-BCA8p0sb.js';
 import '../types-DBHzPsbx.js';
 import '../tool-Dhwg__1L.js';
-import '../middleware-Cp0tm6zE.js';
-import '../types-DMGvEFRf.js';
+import '../middleware-DfMc30in.js';
+import '../types-BrxxAtcJ.js';
 import 'zod';
 
 type EvalOutcome<Score = unknown> = {

package/dist/evals/index.js

@@ -1,27 +1,27 @@
 import {
   ExtractorBuilder
-} from "../chunk-3YJXLOGO.js";
-import "../chunk-DPTZ2MK4.js";
-import "../chunk-QABIBFKE.js";
-import "../chunk-7LCKIULT.js";
+} from "../chunk-ZSTUMG7G.js";
+import "../chunk-WHAPIIEZ.js";
+import "../chunk-N3YV3WCO.js";
+import "../chunk-3IYTCGHS.js";
 import "../chunk-YK4WAAS4.js";
 import "../chunk-XUUY2L2D.js";
-import "../chunk-JNJBKSLA.js";
-import "../chunk-I7TM7TKX.js";
+import "../chunk-6X2UTX3L.js";
+import "../chunk-MELYDKWQ.js";
 import "../chunk-2ODTMRHP.js";
 import "../chunk-ZNTIUOKL.js";
 import "../chunk-TGLNXVII.js";
-import "../chunk-MTGNA4OS.js";
-import "../chunk-F56AMXMQ.js";
-import "../chunk-PJDL5UT2.js";
-import "../chunk-WQKHFADH.js";
 import {
   cosineSimilarity,
   embedText
-} from "../chunk-YBBD4HZ3.js";
+} from "../chunk-MNWK2USR.js";
 import {
   mapWithConcurrency
 } from "../chunk-OIMLU4SF.js";
+import "../chunk-MTGNA4OS.js";
+import "../chunk-F56AMXMQ.js";
+import "../chunk-PJDL5UT2.js";
+import "../chunk-WQKHFADH.js";
 import "../chunk-CWUJUSOS.js";
 
 // src/evals/agent-target.ts