@anvia/core 0.26.0 → 1.0.0-rc.10

package/dist/chunk-EMZKPRE6.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/vector-store/index.ts","../src/internal/vector-search-options.ts","../src/internal/agent-runtime/approval-requirement.ts","../src/tool/errors.ts","../src/tool/tool.ts","../src/internal/agent-runtime/prepared-tool-call.ts","../src/tool/create-tool.ts","../src/vector-store/filter.ts","../src/vector-store/lsh.ts","../src/vector-store/retrieve.ts"],"sourcesContent":["import { z } from \"zod\";\nimport {\n  cosineSimilarity,\n  type EmbeddedDocument,\n  type Embedding,\n  type VectorMetadata,\n} from \"../embeddings\";\nimport { assertFiniteMinScore, assertPositiveSearchLimit } from \"../internal/vector-search-options\";\nimport { createTool } from \"../tool/create-tool\";\nimport type { Tool } from \"../tool/tool\";\nimport { matchesVectorFilter } from \"./filter\";\nimport { LshIndex } from \"./lsh\";\nimport { retrieveDocuments } from \"./retrieve\";\nimport type {\n  IndexStrategy,\n  VectorInspectItem,\n  VectorInspectPage,\n  VectorInspectRequest,\n  VectorSearchRequest,\n  VectorSearchResult,\n  VectorSearchToolOptions,\n  VectorStore,\n  VectorStoreUpsertOptions,\n} from \"./types\";\n\nexport { matchesVectorFilter, vectorFilter } from \"./filter\";\nexport { retrieveDocuments } from \"./retrieve\";\nexport type * from \"./types\";\n\ntype StoredDocument<T, Metadata extends VectorMetadata> = EmbeddedDocument<T, Metadata>;\n\nexport type InMemoryVectorStoreOptions = {\n  dimensions?: number | undefined;\n  index?: IndexStrategy | undefined;\n};\n\nexport type InMemoryVectorStoreFromDocumentsOptions<\n  T,\n  Metadata extends VectorMetadata = VectorMetadata,\n> = InMemoryVectorStoreOptions & {\n  documents: Array<EmbeddedDocument<T, Metadata>>;\n};\n\nexport class InMemoryVectorStore<T, Metadata extends VectorMetadata = VectorMetadata>\n  implements VectorStore<T, Metadata>\n{\n  private readonly documents = new Map<string, StoredDocument<T, Metadata>>();\n  private readonly indexStrategy: IndexStrategy;\n  private lshIndex: LshIndex | undefined;\n  private embeddingDimension: number | undefined;\n\n  constructor(options: InMemoryVectorStoreOptions = {}) {\n    this.indexStrategy = options.index ?? { type: \"bruteForce\" };\n    if (\n      options.dimensions !== undefined &&\n      (!Number.isSafeInteger(options.dimensions) || options.dimensions < 1)\n    ) {\n      throw new RangeError(\"Vector dimensions must be a positive safe integer.\");\n    }\n    this.embeddingDimension = options.dimensions;\n  }\n\n  static fromDocuments<T, Metadata extends VectorMetadata = VectorMetadata>(\n    options: InMemoryVectorStoreFromDocumentsOptions<T, Metadata>,\n  ): InMemoryVectorStore<T, Metadata> {\n    const store = new InMemoryVectorStore<T, Metadata>(options);\n    store.replaceDocuments(options.documents);\n    return store;\n  }\n\n  async ensure(): Promise<void> {}\n  async validate(): Promise<void> {}\n\n  async upsert(options: VectorStoreUpsertOptions<T, Metadata>): Promise<void> {\n    this.replaceDocuments(options.documents);\n  }\n\n  get(options: { id: string }): StoredDocument<T, Metadata> | undefined {\n    return this.documents.get(options.id);\n  }\n\n  values(): Array<StoredDocument<T, Metadata>> {\n    return [...this.documents.values()];\n  }\n\n  len(): number {\n    return this.documents.size;\n  }\n\n  isEmpty(): boolean {\n    return this.documents.size === 0;\n  }\n\n  async search(request: VectorSearchRequest): Promise<Array<VectorSearchResult<T, Metadata>>> {\n    throwIfAborted(request.abortSignal);\n    assertPositiveSearchLimit(request.topK);\n    assertFiniteMinScore(request.minScore);\n    const queryEmbedding: Embedding = { document: \"\", vector: request.vector };\n    const results = this.candidates(queryEmbedding)\n      .filter((document) => matchesVectorFilter(document.metadata, request.filter))\n      .flatMap((document) => {\n        const score = bestScore(queryEmbedding, document.embeddings);\n        if (score === undefined || (request.minScore !== undefined && score < request.minScore)) {\n          return [];\n        }\n        let result: VectorSearchResult<T, Metadata> = {\n          score,\n          id: document.id,\n          document: document.document,\n        };\n        if (document.metadata !== undefined) {\n          result = { ...result, metadata: document.metadata };\n        }\n        return [result];\n      })\n      .sort((left, right) => right.score - left.score)\n      .slice(0, request.topK);\n    throwIfAborted(request.abortSignal);\n    return results;\n  }\n\n  async inspect(request: VectorInspectRequest): Promise<VectorInspectPage<T, Metadata>> {\n    throwIfAborted(request.abortSignal);\n    const limit = assertPositiveSearchLimit(request.limit);\n    const start = Math.max(0, Math.trunc(Number(request.cursor ?? \"0\")));\n    const documents = this.values().filter((document) =>\n      matchesVectorFilter(document.metadata, request.filter),\n    );\n    const page = documents.slice(start, start + limit);\n    const nextOffset = start + page.length;\n    const result: VectorInspectPage<T, Metadata> = {\n      items: page.map((document): VectorInspectItem<T, Metadata> => {\n        let item: VectorInspectItem<T, Metadata> = {\n          id: document.id,\n          document: document.document,\n        };\n        if (document.metadata !== undefined) {\n          item = { ...item, metadata: document.metadata };\n        }\n        return item;\n      }),\n      totalCount: documents.length,\n    };\n    if (nextOffset < documents.length) result.nextCursor = String(nextOffset);\n    throwIfAborted(request.abortSignal);\n    return result;\n  }\n\n  private replaceDocuments(documents: Array<EmbeddedDocument<T, Metadata>>): void {\n    const ids = new Set<string>();\n    for (const document of documents) {\n      if (ids.has(document.id)) throw new TypeError(`Duplicate vector document id: ${document.id}`);\n      ids.add(document.id);\n      if (document.embeddings.length === 0) {\n        throw new TypeError(`Vector document ${document.id} must contain at least one embedding.`);\n      }\n      for (const embedding of document.embeddings)\n        assertFiniteVector(embedding.vector, document.id);\n    }\n    this.validateDocumentDimensions(documents);\n    for (const document of documents) this.documents.set(document.id, document);\n    this.rebuildLshIndex();\n  }\n\n  private candidates(queryEmbedding: Embedding): Array<StoredDocument<T, Metadata>> {\n    this.validateQueryDimension(queryEmbedding);\n    if (this.indexStrategy.type !== \"lsh\" || this.lshIndex === undefined) return this.values();\n    const candidateIds = this.lshIndex.query(queryEmbedding.vector);\n    if (candidateIds.size === 0) return this.values();\n    return [...candidateIds].flatMap((id) => {\n      const document = this.documents.get(id);\n      return document === undefined ? [] : [document];\n    });\n  }\n\n  private rebuildLshIndex(): void {\n    if (this.indexStrategy.type !== \"lsh\") {\n      this.lshIndex = undefined;\n      return;\n    }\n    const firstEmbedding = this.values().flatMap((document) => document.embeddings)[0];\n    if (firstEmbedding === undefined) {\n      this.lshIndex = undefined;\n      return;\n    }\n    const index = new LshIndex(firstEmbedding.vector.length, this.indexStrategy);\n    for (const document of this.documents.values()) {\n      for (const embedding of document.embeddings) index.insert(document.id, embedding.vector);\n    }\n    this.lshIndex = index;\n  }\n\n  private validateDocumentDimensions(documents: Array<EmbeddedDocument<T, Metadata>>): void {\n    let dimension = this.embeddingDimension;\n    for (const document of documents) {\n      for (const embedding of document.embeddings) {\n        dimension = validateEmbeddingDimension(dimension, embedding, document.id);\n      }\n    }\n    this.embeddingDimension = dimension;\n  }\n\n  private validateQueryDimension(queryEmbedding: Embedding): void {\n    if (this.embeddingDimension !== undefined) {\n      validateEmbeddingDimension(this.embeddingDimension, queryEmbedding, \"query\");\n    }\n  }\n}\n\nfunction throwIfAborted(signal: AbortSignal | undefined): void {\n  if (signal?.aborted) {\n    const error = new Error(\"The operation was aborted.\");\n    error.name = \"AbortError\";\n    throw error;\n  }\n}\n\nexport function createVectorSearchTool<T, Metadata extends VectorMetadata>(\n  options: VectorSearchToolOptions<T, Metadata>,\n): Tool<{ query: string; topK?: number }, Array<VectorSearchResult<T, Metadata>>> {\n  const configuredTopK = assertPositiveSearchLimit(options.topK ?? 5);\n  assertFiniteMinScore(options.minScore);\n  return createTool({\n    name: options.name,\n    description:\n      options.description ?? \"Search a vector store for documents relevant to the provided query.\",\n    inputSchema: z.object({\n      query: z.string().min(1).describe(\"The query string to search for relevant documents.\"),\n      topK: z.number().int().positive().optional().describe(\"The maximum number of results.\"),\n    }),\n    outputSchema: z.array(\n      z.object({\n        score: z.number(),\n        id: z.string(),\n        document: z.any(),\n        metadata: z\n          .record(z.string(), z.union([z.string(), z.number(), z.boolean(), z.null()]))\n          .optional(),\n      }),\n    ),\n    execute: ({ query, topK }, context) => {\n      const request = {\n        query,\n        topK: topK ?? configuredTopK,\n        minScore: options.minScore,\n        filter: options.filter,\n        retries: options.retries,\n        abortSignal: context.abortSignal,\n      };\n      return \"models\" in options && options.models !== undefined\n        ? retrieveDocuments({\n            ...request,\n            store: options.store,\n            models: options.models,\n            fusion: options.fusion,\n          })\n        : retrieveDocuments({ ...request, store: options.store, model: options.model });\n    },\n  }) as Tool<{ query: string; topK?: number }, Array<VectorSearchResult<T, Metadata>>>;\n}\n\nfunction bestScore(queryEmbedding: Embedding, embeddings: Embedding[]): number | undefined {\n  let best: number | undefined;\n  for (const embedding of embeddings) {\n    const score = cosineSimilarity(queryEmbedding.vector, embedding.vector);\n    best = best === undefined ? score : Math.max(best, score);\n  }\n  return best;\n}\n\nfunction validateEmbeddingDimension(\n  expectedDimension: number | undefined,\n  embedding: Embedding,\n  id: string,\n): number {\n  assertFiniteVector(embedding.vector, id);\n  if (expectedDimension === undefined) return embedding.vector.length;\n  if (embedding.vector.length !== expectedDimension) {\n    throw new Error(\n      `Vector dimension mismatch: expected ${expectedDimension} dimensions but received ${embedding.vector.length} for ${id}`,\n    );\n  }\n  return expectedDimension;\n}\n\nfunction assertFiniteVector(vector: number[], id: string): void {\n  if (vector.length === 0) throw new TypeError(`Vector for ${id} must not be empty.`);\n  if (!vector.every(Number.isFinite)) {\n    throw new TypeError(`Vector for ${id} must contain only finite numbers.`);\n  }\n}\n","export function assertPositiveSearchLimit(value: number, name = \"topK\"): number {\n  if (!Number.isSafeInteger(value) || value <= 0) {\n    throw new RangeError(`${name} must be a positive safe integer.`);\n  }\n  return value;\n}\n\nexport function assertFiniteMinScore(\n  value: number | undefined,\n  name = \"minScore\",\n): number | undefined {\n  if (value !== undefined && !Number.isFinite(value)) {\n    throw new RangeError(`${name} must be a finite number.`);\n  }\n  return value;\n}\n","import type { ToolApprovalRequirement } from \"../../tool/tool\";\n\nexport function toolMayRequireApproval(requirement: unknown): boolean {\n  return requirement !== undefined && requirement !== false;\n}\n\nexport function assertToolApprovalRequirement(\n  requirement: unknown,\n  options: { allowFunction: boolean },\n): asserts requirement is boolean | ToolApprovalRequirement | ((...args: never[]) => unknown) {\n  if (typeof requirement === \"boolean\") {\n    return;\n  }\n  if (options.allowFunction && typeof requirement === \"function\") {\n    return;\n  }\n  if (typeof requirement !== \"object\" || requirement === null || Array.isArray(requirement)) {\n    throw new TypeError(\n      'Tool \"requiresApproval\" must be a boolean, a function, or an object with an optional string reason.',\n    );\n  }\n  const reason = (requirement as { reason?: unknown }).reason;\n  if (reason !== undefined && typeof reason !== \"string\") {\n    throw new TypeError('Tool \"requiresApproval.reason\" must be a string.');\n  }\n}\n","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 { isJsonValue } from \"../completion/json\";\nimport { parseMessage } from \"../completion/message-schema\";\nimport type {\n  JsonObject,\n  JsonValue,\n  ToolDefinition,\n  ToolResultContentPart,\n  ToolResultOutput,\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  callId?: string;\n  internalCallId: string;\n  run: ToolApprovalRunContext;\n};\n\nexport type ToolApprovalRequirement = {\n  reason?: string | undefined;\n};\n\nexport type ToolRequiresApproval<Args = unknown> =\n  | boolean\n  | ToolApprovalRequirement\n  | ((\n      args: Args,\n      context: ToolApprovalContext<Args>,\n    ) => boolean | ToolApprovalRequirement | Promise<boolean | ToolApprovalRequirement>);\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  abortSignal?: AbortSignal | undefined;\n};\n\nexport interface Tool<Args = unknown, Output = unknown> {\n  readonly name: string;\n  readonly requiresApproval?: ToolRequiresApproval<Args>;\n  definition(prompt: string): ToolDefinition | Promise<ToolDefinition>;\n  call(args: Args, context?: ToolCallContext): Output | Promise<Output>;\n  parseInput?(args: JsonValue): Args;\n}\n\nexport type AnyTool = Omit<Tool<unknown, unknown>, \"requiresApproval\"> & {\n  readonly requiresApproval?: unknown;\n};\n\nconst richToolOutput = Symbol(\"anvia.tool-output.content\");\n\nexport type RichToolOutput = Readonly<{\n  [richToolOutput]: true;\n  content: readonly ToolResultContentPart[];\n}>;\n\nexport type NormalizedToolOutput = ToolResultOutput;\n\nexport class ToolResultSerializationError extends TypeError {\n  constructor(readonly output: unknown) {\n    super(\"Tool output must be a string, a strict JSON value, or ToolOutput.content(...).\");\n    this.name = \"ToolResultSerializationError\";\n  }\n}\n\nexport const ToolOutput = {\n  content(content: readonly ToolResultContentPart[]): RichToolOutput {\n    return { [richToolOutput]: true, content };\n  },\n};\n\nexport function normalizeToolResultOutput(output: unknown): NormalizedToolOutput {\n  if (typeof output === \"string\") {\n    return { type: \"text\", value: output };\n  }\n  if (isRichToolOutput(output)) {\n    try {\n      const message = parseMessage({\n        role: \"tool\",\n        content: [\n          {\n            type: \"tool-result\",\n            toolCallId: \"validation\",\n            toolName: \"validation\",\n            output: { type: \"content\", value: output.content },\n          },\n        ],\n      });\n      if (message.role !== \"tool\") throw new TypeError(\"Unexpected message role\");\n      const result = message.content[0];\n      if (result?.type !== \"tool-result\") throw new TypeError(\"Unexpected tool result part\");\n      if (result?.output.type !== \"content\") throw new TypeError(\"Unexpected tool output\");\n      return result.output;\n    } catch {\n      throw new ToolResultSerializationError(output);\n    }\n  }\n  if (isJsonValue(output)) {\n    return { type: \"json\", value: output };\n  }\n  throw new ToolResultSerializationError(output);\n}\n\nexport function toolResultContentToText(content: readonly ToolResultContentPart[]): string {\n  return content\n    .map((item) => (item.type === \"text\" ? item.text : `[file:${item.mediaType}]`))\n    .join(\"\\n\");\n}\n\nfunction isRichToolOutput(value: unknown): value is RichToolOutput {\n  return typeof value === \"object\" && value !== null && richToolOutput in value;\n}\n\nexport function parseToolArgs(args: string): JsonValue {\n  const value: unknown = JSON.parse(args);\n  if (!isJsonValue(value)) {\n    throw new TypeError(\"Tool arguments must be a JSON value.\");\n  }\n  return value;\n}\n","import { ToolCallError, ToolJsonError } from \"../../tool/errors\";\nimport {\n  type AnyTool,\n  type NormalizedToolOutput,\n  normalizeToolResultOutput,\n  parseToolArgs,\n  type ToolCallContext,\n} from \"../../tool/tool\";\n\nconst preparedToolInputSymbol = Symbol(\"preparedToolInput\");\nconst preparedToolOwnerSymbol = Symbol(\"preparedToolOwner\");\n\nexport type PreparedToolCall = {\n  input: unknown;\n  call(context: ToolCallContext): Promise<NormalizedToolOutput>;\n};\n\ntype PreparedToolCallContext = ToolCallContext & {\n  [preparedToolInputSymbol]?: { owner: object; input: unknown };\n};\n\ntype ToolWithPreparedOwner = AnyTool & {\n  [preparedToolOwnerSymbol]?: object;\n};\n\nexport function attachPreparedToolOwner<T extends AnyTool>(tool: T, owner: object): T {\n  Object.defineProperty(tool, preparedToolOwnerSymbol, {\n    configurable: false,\n    enumerable: true,\n    value: owner,\n    writable: false,\n  });\n  return tool;\n}\n\nexport function preparedToolInput(\n  context: ToolCallContext | undefined,\n  owner: object,\n): { input: unknown } | undefined {\n  if (context === undefined || !(preparedToolInputSymbol in context)) {\n    return undefined;\n  }\n  const prepared = (context as PreparedToolCallContext)[preparedToolInputSymbol];\n  return prepared?.owner === owner ? { input: prepared.input } : undefined;\n}\n\nexport function withoutPreparedToolInput(context: ToolCallContext): ToolCallContext {\n  if (!(preparedToolInputSymbol in context)) {\n    return context;\n  }\n  const { [preparedToolInputSymbol]: _prepared, ...publicContext } =\n    context as PreparedToolCallContext;\n  return publicContext;\n}\n\nexport function prepareToolCall(tool: AnyTool, args: string): PreparedToolCall {\n  let rawInput: ReturnType<typeof parseToolArgs>;\n  try {\n    rawInput = parseToolArgs(args);\n  } catch (error) {\n    throw new ToolJsonError(`Invalid JSON arguments for tool ${tool.name}`, error);\n  }\n\n  let input: unknown;\n  try {\n    input = tool.parseInput === undefined ? rawInput : tool.parseInput(rawInput);\n  } catch (error) {\n    throw asToolCallError(tool.name, error);\n  }\n\n  return prepareToolCallFromInput(tool, input);\n}\n\nexport function prepareToolCallFromInput(tool: AnyTool, input: unknown): PreparedToolCall {\n  return {\n    input,\n    async call(context) {\n      try {\n        const owner = (tool as ToolWithPreparedOwner)[preparedToolOwnerSymbol];\n        return normalizeToolResultOutput(\n          await tool.call(\n            input,\n            owner === undefined ? context : withPreparedToolInput(context, owner, input),\n          ),\n        );\n      } catch (error) {\n        throw asToolCallError(tool.name, error);\n      }\n    },\n  };\n}\n\nfunction withPreparedToolInput(\n  context: ToolCallContext,\n  owner: object,\n  input: unknown,\n): ToolCallContext {\n  const preparedContext: PreparedToolCallContext = { ...context };\n  Object.defineProperty(preparedContext, preparedToolInputSymbol, {\n    configurable: false,\n    enumerable: true,\n    value: { owner, input },\n    writable: false,\n  });\n  return preparedContext;\n}\n\nfunction asToolCallError(toolName: string, error: unknown): ToolCallError {\n  if (error instanceof ToolCallError) {\n    return error;\n  }\n  return error instanceof Error\n    ? new ToolCallError(error.message, error)\n    : new ToolCallError(`Tool ${toolName} failed`, error);\n}\n","import type { z } from \"zod\";\nimport { assertToolApprovalRequirement } from \"../internal/agent-runtime/approval-requirement\";\nimport {\n  attachPreparedToolOwner,\n  preparedToolInput,\n  withoutPreparedToolInput,\n} from \"../internal/agent-runtime/prepared-tool-call\";\nimport { toProviderJsonSchema, type ZodSchema } from \"../schema/zod-schema\";\nimport type { Tool, ToolCallContext, ToolRequiresApproval } from \"./tool\";\n\nexport type CreateToolOptions<\n  InputSchema extends ZodSchema,\n  OutputSchema extends ZodSchema | undefined = undefined,\n  Output = unknown,\n> = {\n  name: string;\n  description: string;\n  inputSchema: InputSchema;\n  outputSchema?: OutputSchema;\n  requiresApproval?: ToolRequiresApproval<z.output<InputSchema>>;\n  execute(\n    args: z.output<InputSchema>,\n    context: ToolCallContext,\n  ): OutputSchema extends ZodSchema\n    ? z.input<OutputSchema> | Promise<z.input<OutputSchema>>\n    : Output | Promise<Output>;\n};\n\ntype CreateToolOutput<\n  OutputSchema extends ZodSchema | undefined,\n  Output,\n> = OutputSchema extends ZodSchema ? z.output<OutputSchema> : Output;\n\nexport function createTool<InputSchema extends ZodSchema, Output = unknown>(\n  options: CreateToolOptions<InputSchema, undefined, Output> & { outputSchema?: undefined },\n): Tool<z.output<InputSchema>, Output>;\n\nexport function createTool<InputSchema extends ZodSchema, OutputSchema extends ZodSchema>(\n  options: CreateToolOptions<InputSchema, OutputSchema>,\n): Tool<z.output<InputSchema>, z.output<OutputSchema>>;\n\nexport function createTool<\n  InputSchema extends ZodSchema,\n  OutputSchema extends ZodSchema | undefined = undefined,\n  Output = unknown,\n>(\n  options: CreateToolOptions<InputSchema, OutputSchema, Output>,\n): Tool<z.output<InputSchema>, CreateToolOutput<OutputSchema, Output>> {\n  const { name, description, inputSchema, outputSchema, execute } = options;\n  const requiresApproval = snapshotApprovalRequirement(options.requiresApproval);\n  if (requiresApproval !== undefined) {\n    assertToolApprovalRequirement(requiresApproval, { allowFunction: true });\n  }\n  const parameters = toProviderJsonSchema(inputSchema);\n  const preparedInputOwner = {};\n  const definition = () => ({\n    name,\n    description,\n    parameters: globalThis.structuredClone(parameters),\n  });\n  const call = async (\n    args: z.output<InputSchema>,\n    context: ToolCallContext = {},\n  ): Promise<CreateToolOutput<OutputSchema, Output>> => {\n    const prepared = preparedToolInput(context, preparedInputOwner);\n    const parsedArgs =\n      prepared === undefined ? inputSchema.parse(args) : (prepared.input as z.output<InputSchema>);\n    const executionContext = prepared === undefined ? context : withoutPreparedToolInput(context);\n    const result = await execute(parsedArgs, executionContext);\n    return (outputSchema === undefined ? result : outputSchema.parse(result)) as CreateToolOutput<\n      OutputSchema,\n      Output\n    >;\n  };\n  const parseInput = (args: unknown): z.output<InputSchema> => inputSchema.parse(args);\n\n  const tool: Tool<z.output<InputSchema>, CreateToolOutput<OutputSchema, Output>> = {\n    name,\n    definition,\n    call,\n    parseInput,\n  };\n  if (requiresApproval !== undefined) {\n    Object.defineProperty(tool, \"requiresApproval\", {\n      configurable: false,\n      enumerable: true,\n      value: requiresApproval,\n      writable: false,\n    });\n  }\n  return attachPreparedToolOwner(tool, preparedInputOwner);\n}\n\nfunction snapshotApprovalRequirement<Args>(\n  requirement: ToolRequiresApproval<Args> | undefined,\n): ToolRequiresApproval<Args> | undefined {\n  return typeof requirement === \"object\" && requirement !== null\n    ? Object.freeze({ ...requirement })\n    : requirement;\n}\n","import type { VectorMetadata, VectorMetadataValue } from \"../embeddings\";\nimport type { VectorFilter } from \"./types\";\n\nexport type { VectorFilter } from \"./types\";\n\nexport const vectorFilter = {\n  eq(key: string, value: VectorMetadataValue): VectorFilter {\n    return { type: \"eq\", key, value };\n  },\n  gt(key: string, value: VectorMetadataValue): VectorFilter {\n    return { type: \"gt\", key, value };\n  },\n  lt(key: string, value: VectorMetadataValue): VectorFilter {\n    return { type: \"lt\", key, value };\n  },\n  and(left: VectorFilter, right: VectorFilter): VectorFilter {\n    return { type: \"and\", filters: [left, right] };\n  },\n  or(left: VectorFilter, right: VectorFilter): VectorFilter {\n    return { type: \"or\", filters: [left, right] };\n  },\n};\n\nexport function matchesVectorFilter(\n  metadata: VectorMetadata | undefined,\n  filter: VectorFilter | undefined,\n): boolean {\n  if (filter === undefined) {\n    return true;\n  }\n  if (metadata === undefined) {\n    return false;\n  }\n\n  switch (filter.type) {\n    case \"eq\":\n      return metadata[filter.key] === filter.value;\n    case \"gt\":\n      return compare(metadata[filter.key], filter.value) > 0;\n    case \"lt\":\n      return compare(metadata[filter.key], filter.value) < 0;\n    case \"and\":\n      return (\n        matchesVectorFilter(metadata, filter.filters[0]) &&\n        matchesVectorFilter(metadata, filter.filters[1])\n      );\n    case \"or\":\n      return (\n        matchesVectorFilter(metadata, filter.filters[0]) ||\n        matchesVectorFilter(metadata, filter.filters[1])\n      );\n  }\n}\n\nfunction compare(left: VectorMetadataValue | undefined, right: VectorMetadataValue): number {\n  if (typeof left === \"number\" && typeof right === \"number\") {\n    return left - right;\n  }\n  if (typeof left === \"string\" && typeof right === \"string\") {\n    return left.localeCompare(right);\n  }\n  if (typeof left === \"boolean\" && typeof right === \"boolean\") {\n    return Number(left) - Number(right);\n  }\n  return 0;\n}\n","import type { LshOptions } from \"./types\";\n\nexport type { LshOptions } from \"./types\";\n\nexport class LshIndex {\n  private readonly hyperplanes: number[][];\n  private readonly tables: Array<Map<string, Set<string>>>;\n\n  constructor(\n    dimensions: number,\n    private readonly options: LshOptions,\n  ) {\n    const rng = seededRandom(options.seed ?? 42);\n    this.hyperplanes = [];\n    for (let index = 0; index < options.numTables * options.numHyperplanes; index += 1) {\n      const plane = Array.from({ length: dimensions }, () => rng() * 2 - 1);\n      const norm = Math.sqrt(plane.reduce((sum, value) => sum + value ** 2, 0));\n      this.hyperplanes.push(norm === 0 ? plane : plane.map((value) => value / norm));\n    }\n    this.tables = Array.from({ length: options.numTables }, () => new Map());\n  }\n\n  insert(id: string, vector: number[]): void {\n    for (let table = 0; table < this.options.numTables; table += 1) {\n      const hash = this.hash(vector, table);\n      const bucket = this.tables[table]?.get(hash) ?? new Set<string>();\n      bucket.add(id);\n      this.tables[table]?.set(hash, bucket);\n    }\n  }\n\n  query(vector: number[]): Set<string> {\n    const candidates = new Set<string>();\n    for (let table = 0; table < this.options.numTables; table += 1) {\n      const hash = this.hash(vector, table);\n      for (const id of this.tables[table]?.get(hash) ?? []) {\n        candidates.add(id);\n      }\n    }\n    return candidates;\n  }\n\n  private hash(vector: number[], table: number): string {\n    let hash = \"\";\n    const start = table * this.options.numHyperplanes;\n    for (let offset = 0; offset < this.options.numHyperplanes; offset += 1) {\n      const plane = this.hyperplanes[start + offset] as number[];\n      const dot = vector.reduce((sum, value, index) => sum + value * (plane[index] ?? 0), 0);\n      hash += dot >= 0 ? \"1\" : \"0\";\n    }\n    return hash;\n  }\n}\n\nfunction seededRandom(seed: number): () => number {\n  let state = seed >>> 0;\n  return () => {\n    state = (state * 1664525 + 1013904223) >>> 0;\n    return state / 0x100000000;\n  };\n}\n","import { embedSparseQuery, embedText, type VectorMetadata } from \"../embeddings\";\nimport { throwIfAborted } from \"../internal/abort\";\nimport { assertFiniteMinScore, assertPositiveSearchLimit } from \"../internal/vector-search-options\";\nimport { type ResolvedRetryOptions, resolveRetryOptions, runWithRetries } from \"../retry\";\nimport type {\n  RetrieveDocumentsOptions,\n  RetrieveHybridDocumentsOptions,\n  VectorSearchResult,\n} from \"./types\";\n\nexport function retrieveDocuments<T, Metadata extends VectorMetadata = VectorMetadata>(\n  options: RetrieveDocumentsOptions<T, Metadata>,\n): Promise<Array<VectorSearchResult<T, Metadata>>>;\nexport function retrieveDocuments<T, Metadata extends VectorMetadata = VectorMetadata>(\n  options: RetrieveHybridDocumentsOptions<T, Metadata>,\n): Promise<Array<VectorSearchResult<T, Metadata>>>;\nexport async function retrieveDocuments<T, Metadata extends VectorMetadata = VectorMetadata>(\n  options: RetrieveDocumentsOptions<T, Metadata> | RetrieveHybridDocumentsOptions<T, Metadata>,\n): Promise<Array<VectorSearchResult<T, Metadata>>> {\n  assertPositiveSearchLimit(options.topK);\n  assertFiniteMinScore(options.minScore);\n  throwIfAborted(options.abortSignal);\n  const retries = resolveRetries(options.retries);\n  const search = (operation: () => Promise<Array<VectorSearchResult<T, Metadata>>>) =>\n    runWithRetries(operation, retries, {\n      streaming: false,\n      abortSignal: options.abortSignal,\n    });\n\n  if (\"model\" in options && options.model !== undefined) {\n    const { embedding } = await embedText({\n      model: options.model,\n      text: options.query,\n      retries: options.retries,\n      abortSignal: options.abortSignal,\n    });\n    return deduplicateResults(\n      await search(() =>\n        options.store.search({\n          vector: embedding.vector,\n          topK: options.topK,\n          minScore: options.minScore,\n          filter: options.filter,\n          abortSignal: options.abortSignal,\n        }),\n      ),\n    );\n  }\n\n  const [{ embedding: dense }, { embedding: sparse }] = await Promise.all([\n    embedText({\n      model: options.models.dense,\n      text: options.query,\n      retries: options.retries,\n      abortSignal: options.abortSignal,\n    }),\n    embedSparseQuery({\n      model: options.models.sparse,\n      query: options.query,\n      retries: options.retries,\n      abortSignal: options.abortSignal,\n    }),\n  ]);\n  return deduplicateResults(\n    await search(() =>\n      options.store.searchHybrid({\n        vector: dense.vector,\n        sparseVector: sparse.vector,\n        fusion: options.fusion,\n        topK: options.topK,\n        minScore: options.minScore,\n        filter: options.filter,\n        abortSignal: options.abortSignal,\n      }),\n    ),\n  );\n}\n\nfunction deduplicateResults<T, Metadata extends VectorMetadata>(\n  results: Array<VectorSearchResult<T, Metadata>>,\n): Array<VectorSearchResult<T, Metadata>> {\n  const bestById = new Map<string, VectorSearchResult<T, Metadata>>();\n  for (const result of results) {\n    const current = bestById.get(result.id);\n    if (current === undefined || result.score > current.score) bestById.set(result.id, result);\n  }\n  return [...bestById.values()].sort((left, right) => right.score - left.score);\n}\n\nfunction resolveRetries(\n  setting: RetrieveDocumentsOptions<unknown>[\"retries\"],\n): ResolvedRetryOptions | undefined {\n  return setting === undefined || setting === false ? undefined : resolveRetryOptions(setting);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;AAAA,SAAS,SAAS;;;ACAX,SAAS,0BAA0B,OAAe,OAAO,QAAgB;AAC9E,MAAI,CAAC,OAAO,cAAc,KAAK,KAAK,SAAS,GAAG;AAC9C,UAAM,IAAI,WAAW,GAAG,IAAI,mCAAmC;AAAA,EACjE;AACA,SAAO;AACT;AAEO,SAAS,qBACd,OACA,OAAO,YACa;AACpB,MAAI,UAAU,UAAa,CAAC,OAAO,SAAS,KAAK,GAAG;AAClD,UAAM,IAAI,WAAW,GAAG,IAAI,2BAA2B;AAAA,EACzD;AACA,SAAO;AACT;;;ACbO,SAAS,uBAAuB,aAA+B;AACpE,SAAO,gBAAgB,UAAa,gBAAgB;AACtD;AAEO,SAAS,8BACd,aACA,SAC4F;AAC5F,MAAI,OAAO,gBAAgB,WAAW;AACpC;AAAA,EACF;AACA,MAAI,QAAQ,iBAAiB,OAAO,gBAAgB,YAAY;AAC9D;AAAA,EACF;AACA,MAAI,OAAO,gBAAgB,YAAY,gBAAgB,QAAQ,MAAM,QAAQ,WAAW,GAAG;AACzF,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACA,QAAM,SAAU,YAAqC;AACrD,MAAI,WAAW,UAAa,OAAO,WAAW,UAAU;AACtD,UAAM,IAAI,UAAU,kDAAkD;AAAA,EACxE;AACF;;;ACzBO,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;;;ACqCA,IAAM,iBAAiB,uBAAO,2BAA2B;AASlD,IAAM,+BAAN,cAA2C,UAAU;AAAA,EAC1D,YAAqB,QAAiB;AACpC,UAAM,gFAAgF;AADnE;AAEnB,SAAK,OAAO;AAAA,EACd;AAAA,EAHqB;AAIvB;AAEO,IAAM,aAAa;AAAA,EACxB,QAAQ,SAA2D;AACjE,WAAO,EAAE,CAAC,cAAc,GAAG,MAAM,QAAQ;AAAA,EAC3C;AACF;AAEO,SAAS,0BAA0B,QAAuC;AAC/E,MAAI,OAAO,WAAW,UAAU;AAC9B,WAAO,EAAE,MAAM,QAAQ,OAAO,OAAO;AAAA,EACvC;AACA,MAAI,iBAAiB,MAAM,GAAG;AAC5B,QAAI;AACF,YAAM,UAAU,aAAa;AAAA,QAC3B,MAAM;AAAA,QACN,SAAS;AAAA,UACP;AAAA,YACE,MAAM;AAAA,YACN,YAAY;AAAA,YACZ,UAAU;AAAA,YACV,QAAQ,EAAE,MAAM,WAAW,OAAO,OAAO,QAAQ;AAAA,UACnD;AAAA,QACF;AAAA,MACF,CAAC;AACD,UAAI,QAAQ,SAAS,OAAQ,OAAM,IAAI,UAAU,yBAAyB;AAC1E,YAAM,SAAS,QAAQ,QAAQ,CAAC;AAChC,UAAI,QAAQ,SAAS,cAAe,OAAM,IAAI,UAAU,6BAA6B;AACrF,UAAI,QAAQ,OAAO,SAAS,UAAW,OAAM,IAAI,UAAU,wBAAwB;AACnF,aAAO,OAAO;AAAA,IAChB,QAAQ;AACN,YAAM,IAAI,6BAA6B,MAAM;AAAA,IAC/C;AAAA,EACF;AACA,MAAI,YAAY,MAAM,GAAG;AACvB,WAAO,EAAE,MAAM,QAAQ,OAAO,OAAO;AAAA,EACvC;AACA,QAAM,IAAI,6BAA6B,MAAM;AAC/C;AAEO,SAAS,wBAAwB,SAAmD;AACzF,SAAO,QACJ,IAAI,CAAC,SAAU,KAAK,SAAS,SAAS,KAAK,OAAO,SAAS,KAAK,SAAS,GAAI,EAC7E,KAAK,IAAI;AACd;AAEA,SAAS,iBAAiB,OAAyC;AACjE,SAAO,OAAO,UAAU,YAAY,UAAU,QAAQ,kBAAkB;AAC1E;AAEO,SAAS,cAAc,MAAyB;AACrD,QAAM,QAAiB,KAAK,MAAM,IAAI;AACtC,MAAI,CAAC,YAAY,KAAK,GAAG;AACvB,UAAM,IAAI,UAAU,sCAAsC;AAAA,EAC5D;AACA,SAAO;AACT;;;AC3HA,IAAM,0BAA0B,uBAAO,mBAAmB;AAC1D,IAAM,0BAA0B,uBAAO,mBAAmB;AAenD,SAAS,wBAA2C,MAAS,OAAkB;AACpF,SAAO,eAAe,MAAM,yBAAyB;AAAA,IACnD,cAAc;AAAA,IACd,YAAY;AAAA,IACZ,OAAO;AAAA,IACP,UAAU;AAAA,EACZ,CAAC;AACD,SAAO;AACT;AAEO,SAAS,kBACd,SACA,OACgC;AAChC,MAAI,YAAY,UAAa,EAAE,2BAA2B,UAAU;AAClE,WAAO;AAAA,EACT;AACA,QAAM,WAAY,QAAoC,uBAAuB;AAC7E,SAAO,UAAU,UAAU,QAAQ,EAAE,OAAO,SAAS,MAAM,IAAI;AACjE;AAEO,SAAS,yBAAyB,SAA2C;AAClF,MAAI,EAAE,2BAA2B,UAAU;AACzC,WAAO;AAAA,EACT;AACA,QAAM,EAAE,CAAC,uBAAuB,GAAG,WAAW,GAAG,cAAc,IAC7D;AACF,SAAO;AACT;AAEO,SAAS,gBAAgB,MAAe,MAAgC;AAC7E,MAAI;AACJ,MAAI;AACF,eAAW,cAAc,IAAI;AAAA,EAC/B,SAAS,OAAO;AACd,UAAM,IAAI,cAAc,mCAAmC,KAAK,IAAI,IAAI,KAAK;AAAA,EAC/E;AAEA,MAAI;AACJ,MAAI;AACF,YAAQ,KAAK,eAAe,SAAY,WAAW,KAAK,WAAW,QAAQ;AAAA,EAC7E,SAAS,OAAO;AACd,UAAM,gBAAgB,KAAK,MAAM,KAAK;AAAA,EACxC;AAEA,SAAO,yBAAyB,MAAM,KAAK;AAC7C;AAEO,SAAS,yBAAyB,MAAe,OAAkC;AACxF,SAAO;AAAA,IACL;AAAA,IACA,MAAM,KAAK,SAAS;AAClB,UAAI;AACF,cAAM,QAAS,KAA+B,uBAAuB;AACrE,eAAO;AAAA,UACL,MAAM,KAAK;AAAA,YACT;AAAA,YACA,UAAU,SAAY,UAAU,sBAAsB,SAAS,OAAO,KAAK;AAAA,UAC7E;AAAA,QACF;AAAA,MACF,SAAS,OAAO;AACd,cAAM,gBAAgB,KAAK,MAAM,KAAK;AAAA,MACxC;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,sBACP,SACA,OACA,OACiB;AACjB,QAAM,kBAA2C,EAAE,GAAG,QAAQ;AAC9D,SAAO,eAAe,iBAAiB,yBAAyB;AAAA,IAC9D,cAAc;AAAA,IACd,YAAY;AAAA,IACZ,OAAO,EAAE,OAAO,MAAM;AAAA,IACtB,UAAU;AAAA,EACZ,CAAC;AACD,SAAO;AACT;AAEA,SAAS,gBAAgB,UAAkB,OAA+B;AACxE,MAAI,iBAAiB,eAAe;AAClC,WAAO;AAAA,EACT;AACA,SAAO,iBAAiB,QACpB,IAAI,cAAc,MAAM,SAAS,KAAK,IACtC,IAAI,cAAc,QAAQ,QAAQ,WAAW,KAAK;AACxD;;;ACzEO,SAAS,WAKd,SACqE;AACrE,QAAM,EAAE,MAAM,aAAa,aAAa,cAAc,QAAQ,IAAI;AAClE,QAAM,mBAAmB,4BAA4B,QAAQ,gBAAgB;AAC7E,MAAI,qBAAqB,QAAW;AAClC,kCAA8B,kBAAkB,EAAE,eAAe,KAAK,CAAC;AAAA,EACzE;AACA,QAAM,aAAa,qBAAqB,WAAW;AACnD,QAAM,qBAAqB,CAAC;AAC5B,QAAM,aAAa,OAAO;AAAA,IACxB;AAAA,IACA;AAAA,IACA,YAAY,WAAW,gBAAgB,UAAU;AAAA,EACnD;AACA,QAAM,OAAO,OACX,MACA,UAA2B,CAAC,MACwB;AACpD,UAAM,WAAW,kBAAkB,SAAS,kBAAkB;AAC9D,UAAM,aACJ,aAAa,SAAY,YAAY,MAAM,IAAI,IAAK,SAAS;AAC/D,UAAM,mBAAmB,aAAa,SAAY,UAAU,yBAAyB,OAAO;AAC5F,UAAM,SAAS,MAAM,QAAQ,YAAY,gBAAgB;AACzD,WAAQ,iBAAiB,SAAY,SAAS,aAAa,MAAM,MAAM;AAAA,EAIzE;AACA,QAAM,aAAa,CAAC,SAAyC,YAAY,MAAM,IAAI;AAEnF,QAAM,OAA4E;AAAA,IAChF;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AACA,MAAI,qBAAqB,QAAW;AAClC,WAAO,eAAe,MAAM,oBAAoB;AAAA,MAC9C,cAAc;AAAA,MACd,YAAY;AAAA,MACZ,OAAO;AAAA,MACP,UAAU;AAAA,IACZ,CAAC;AAAA,EACH;AACA,SAAO,wBAAwB,MAAM,kBAAkB;AACzD;AAEA,SAAS,4BACP,aACwC;AACxC,SAAO,OAAO,gBAAgB,YAAY,gBAAgB,OACtD,OAAO,OAAO,EAAE,GAAG,YAAY,CAAC,IAChC;AACN;;;AC9FO,IAAM,eAAe;AAAA,EAC1B,GAAG,KAAa,OAA0C;AACxD,WAAO,EAAE,MAAM,MAAM,KAAK,MAAM;AAAA,EAClC;AAAA,EACA,GAAG,KAAa,OAA0C;AACxD,WAAO,EAAE,MAAM,MAAM,KAAK,MAAM;AAAA,EAClC;AAAA,EACA,GAAG,KAAa,OAA0C;AACxD,WAAO,EAAE,MAAM,MAAM,KAAK,MAAM;AAAA,EAClC;AAAA,EACA,IAAI,MAAoB,OAAmC;AACzD,WAAO,EAAE,MAAM,OAAO,SAAS,CAAC,MAAM,KAAK,EAAE;AAAA,EAC/C;AAAA,EACA,GAAG,MAAoB,OAAmC;AACxD,WAAO,EAAE,MAAM,MAAM,SAAS,CAAC,MAAM,KAAK,EAAE;AAAA,EAC9C;AACF;AAEO,SAAS,oBACd,UACA,QACS;AACT,MAAI,WAAW,QAAW;AACxB,WAAO;AAAA,EACT;AACA,MAAI,aAAa,QAAW;AAC1B,WAAO;AAAA,EACT;AAEA,UAAQ,OAAO,MAAM;AAAA,IACnB,KAAK;AACH,aAAO,SAAS,OAAO,GAAG,MAAM,OAAO;AAAA,IACzC,KAAK;AACH,aAAO,QAAQ,SAAS,OAAO,GAAG,GAAG,OAAO,KAAK,IAAI;AAAA,IACvD,KAAK;AACH,aAAO,QAAQ,SAAS,OAAO,GAAG,GAAG,OAAO,KAAK,IAAI;AAAA,IACvD,KAAK;AACH,aACE,oBAAoB,UAAU,OAAO,QAAQ,CAAC,CAAC,KAC/C,oBAAoB,UAAU,OAAO,QAAQ,CAAC,CAAC;AAAA,IAEnD,KAAK;AACH,aACE,oBAAoB,UAAU,OAAO,QAAQ,CAAC,CAAC,KAC/C,oBAAoB,UAAU,OAAO,QAAQ,CAAC,CAAC;AAAA,EAErD;AACF;AAEA,SAAS,QAAQ,MAAuC,OAAoC;AAC1F,MAAI,OAAO,SAAS,YAAY,OAAO,UAAU,UAAU;AACzD,WAAO,OAAO;AAAA,EAChB;AACA,MAAI,OAAO,SAAS,YAAY,OAAO,UAAU,UAAU;AACzD,WAAO,KAAK,cAAc,KAAK;AAAA,EACjC;AACA,MAAI,OAAO,SAAS,aAAa,OAAO,UAAU,WAAW;AAC3D,WAAO,OAAO,IAAI,IAAI,OAAO,KAAK;AAAA,EACpC;AACA,SAAO;AACT;;;AC7DO,IAAM,WAAN,MAAe;AAAA,EAIpB,YACE,YACiB,SACjB;AADiB;AAEjB,UAAM,MAAM,aAAa,QAAQ,QAAQ,EAAE;AAC3C,SAAK,cAAc,CAAC;AACpB,aAAS,QAAQ,GAAG,QAAQ,QAAQ,YAAY,QAAQ,gBAAgB,SAAS,GAAG;AAClF,YAAM,QAAQ,MAAM,KAAK,EAAE,QAAQ,WAAW,GAAG,MAAM,IAAI,IAAI,IAAI,CAAC;AACpE,YAAM,OAAO,KAAK,KAAK,MAAM,OAAO,CAAC,KAAK,UAAU,MAAM,SAAS,GAAG,CAAC,CAAC;AACxE,WAAK,YAAY,KAAK,SAAS,IAAI,QAAQ,MAAM,IAAI,CAAC,UAAU,QAAQ,IAAI,CAAC;AAAA,IAC/E;AACA,SAAK,SAAS,MAAM,KAAK,EAAE,QAAQ,QAAQ,UAAU,GAAG,MAAM,oBAAI,IAAI,CAAC;AAAA,EACzE;AAAA,EAVmB;AAAA,EALF;AAAA,EACA;AAAA,EAgBjB,OAAO,IAAY,QAAwB;AACzC,aAAS,QAAQ,GAAG,QAAQ,KAAK,QAAQ,WAAW,SAAS,GAAG;AAC9D,YAAM,OAAO,KAAK,KAAK,QAAQ,KAAK;AACpC,YAAM,SAAS,KAAK,OAAO,KAAK,GAAG,IAAI,IAAI,KAAK,oBAAI,IAAY;AAChE,aAAO,IAAI,EAAE;AACb,WAAK,OAAO,KAAK,GAAG,IAAI,MAAM,MAAM;AAAA,IACtC;AAAA,EACF;AAAA,EAEA,MAAM,QAA+B;AACnC,UAAM,aAAa,oBAAI,IAAY;AACnC,aAAS,QAAQ,GAAG,QAAQ,KAAK,QAAQ,WAAW,SAAS,GAAG;AAC9D,YAAM,OAAO,KAAK,KAAK,QAAQ,KAAK;AACpC,iBAAW,MAAM,KAAK,OAAO,KAAK,GAAG,IAAI,IAAI,KAAK,CAAC,GAAG;AACpD,mBAAW,IAAI,EAAE;AAAA,MACnB;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA,EAEQ,KAAK,QAAkB,OAAuB;AACpD,QAAI,OAAO;AACX,UAAM,QAAQ,QAAQ,KAAK,QAAQ;AACnC,aAAS,SAAS,GAAG,SAAS,KAAK,QAAQ,gBAAgB,UAAU,GAAG;AACtE,YAAM,QAAQ,KAAK,YAAY,QAAQ,MAAM;AAC7C,YAAM,MAAM,OAAO,OAAO,CAAC,KAAK,OAAO,UAAU,MAAM,SAAS,MAAM,KAAK,KAAK,IAAI,CAAC;AACrF,cAAQ,OAAO,IAAI,MAAM;AAAA,IAC3B;AACA,WAAO;AAAA,EACT;AACF;AAEA,SAAS,aAAa,MAA4B;AAChD,MAAI,QAAQ,SAAS;AACrB,SAAO,MAAM;AACX,YAAS,QAAQ,UAAU,eAAgB;AAC3C,WAAO,QAAQ;AAAA,EACjB;AACF;;;AC5CA,eAAsB,kBACpB,SACiD;AACjD,4BAA0B,QAAQ,IAAI;AACtC,uBAAqB,QAAQ,QAAQ;AACrC,iBAAe,QAAQ,WAAW;AAClC,QAAM,UAAU,eAAe,QAAQ,OAAO;AAC9C,QAAM,SAAS,CAAC,cACd,eAAe,WAAW,SAAS;AAAA,IACjC,WAAW;AAAA,IACX,aAAa,QAAQ;AAAA,EACvB,CAAC;AAEH,MAAI,WAAW,WAAW,QAAQ,UAAU,QAAW;AACrD,UAAM,EAAE,UAAU,IAAI,MAAM,UAAU;AAAA,MACpC,OAAO,QAAQ;AAAA,MACf,MAAM,QAAQ;AAAA,MACd,SAAS,QAAQ;AAAA,MACjB,aAAa,QAAQ;AAAA,IACvB,CAAC;AACD,WAAO;AAAA,MACL,MAAM;AAAA,QAAO,MACX,QAAQ,MAAM,OAAO;AAAA,UACnB,QAAQ,UAAU;AAAA,UAClB,MAAM,QAAQ;AAAA,UACd,UAAU,QAAQ;AAAA,UAClB,QAAQ,QAAQ;AAAA,UAChB,aAAa,QAAQ;AAAA,QACvB,CAAC;AAAA,MACH;AAAA,IACF;AAAA,EACF;AAEA,QAAM,CAAC,EAAE,WAAW,MAAM,GAAG,EAAE,WAAW,OAAO,CAAC,IAAI,MAAM,QAAQ,IAAI;AAAA,IACtE,UAAU;AAAA,MACR,OAAO,QAAQ,OAAO;AAAA,MACtB,MAAM,QAAQ;AAAA,MACd,SAAS,QAAQ;AAAA,MACjB,aAAa,QAAQ;AAAA,IACvB,CAAC;AAAA,IACD,iBAAiB;AAAA,MACf,OAAO,QAAQ,OAAO;AAAA,MACtB,OAAO,QAAQ;AAAA,MACf,SAAS,QAAQ;AAAA,MACjB,aAAa,QAAQ;AAAA,IACvB,CAAC;AAAA,EACH,CAAC;AACD,SAAO;AAAA,IACL,MAAM;AAAA,MAAO,MACX,QAAQ,MAAM,aAAa;AAAA,QACzB,QAAQ,MAAM;AAAA,QACd,cAAc,OAAO;AAAA,QACrB,QAAQ,QAAQ;AAAA,QAChB,MAAM,QAAQ;AAAA,QACd,UAAU,QAAQ;AAAA,QAClB,QAAQ,QAAQ;AAAA,QAChB,aAAa,QAAQ;AAAA,MACvB,CAAC;AAAA,IACH;AAAA,EACF;AACF;AAEA,SAAS,mBACP,SACwC;AACxC,QAAM,WAAW,oBAAI,IAA6C;AAClE,aAAW,UAAU,SAAS;AAC5B,UAAM,UAAU,SAAS,IAAI,OAAO,EAAE;AACtC,QAAI,YAAY,UAAa,OAAO,QAAQ,QAAQ,MAAO,UAAS,IAAI,OAAO,IAAI,MAAM;AAAA,EAC3F;AACA,SAAO,CAAC,GAAG,SAAS,OAAO,CAAC,EAAE,KAAK,CAAC,MAAM,UAAU,MAAM,QAAQ,KAAK,KAAK;AAC9E;AAEA,SAAS,eACP,SACkC;AAClC,SAAO,YAAY,UAAa,YAAY,QAAQ,SAAY,oBAAoB,OAAO;AAC7F;;;ATlDO,IAAM,sBAAN,MAAM,qBAEb;AAAA,EACmB,YAAY,oBAAI,IAAyC;AAAA,EACzD;AAAA,EACT;AAAA,EACA;AAAA,EAER,YAAY,UAAsC,CAAC,GAAG;AACpD,SAAK,gBAAgB,QAAQ,SAAS,EAAE,MAAM,aAAa;AAC3D,QACE,QAAQ,eAAe,WACtB,CAAC,OAAO,cAAc,QAAQ,UAAU,KAAK,QAAQ,aAAa,IACnE;AACA,YAAM,IAAI,WAAW,oDAAoD;AAAA,IAC3E;AACA,SAAK,qBAAqB,QAAQ;AAAA,EACpC;AAAA,EAEA,OAAO,cACL,SACkC;AAClC,UAAM,QAAQ,IAAI,qBAAiC,OAAO;AAC1D,UAAM,iBAAiB,QAAQ,SAAS;AACxC,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,SAAwB;AAAA,EAAC;AAAA,EAC/B,MAAM,WAA0B;AAAA,EAAC;AAAA,EAEjC,MAAM,OAAO,SAA+D;AAC1E,SAAK,iBAAiB,QAAQ,SAAS;AAAA,EACzC;AAAA,EAEA,IAAI,SAAkE;AACpE,WAAO,KAAK,UAAU,IAAI,QAAQ,EAAE;AAAA,EACtC;AAAA,EAEA,SAA6C;AAC3C,WAAO,CAAC,GAAG,KAAK,UAAU,OAAO,CAAC;AAAA,EACpC;AAAA,EAEA,MAAc;AACZ,WAAO,KAAK,UAAU;AAAA,EACxB;AAAA,EAEA,UAAmB;AACjB,WAAO,KAAK,UAAU,SAAS;AAAA,EACjC;AAAA,EAEA,MAAM,OAAO,SAA+E;AAC1F,IAAAA,gBAAe,QAAQ,WAAW;AAClC,8BAA0B,QAAQ,IAAI;AACtC,yBAAqB,QAAQ,QAAQ;AACrC,UAAM,iBAA4B,EAAE,UAAU,IAAI,QAAQ,QAAQ,OAAO;AACzE,UAAM,UAAU,KAAK,WAAW,cAAc,EAC3C,OAAO,CAAC,aAAa,oBAAoB,SAAS,UAAU,QAAQ,MAAM,CAAC,EAC3E,QAAQ,CAAC,aAAa;AACrB,YAAM,QAAQ,UAAU,gBAAgB,SAAS,UAAU;AAC3D,UAAI,UAAU,UAAc,QAAQ,aAAa,UAAa,QAAQ,QAAQ,UAAW;AACvF,eAAO,CAAC;AAAA,MACV;AACA,UAAI,SAA0C;AAAA,QAC5C;AAAA,QACA,IAAI,SAAS;AAAA,QACb,UAAU,SAAS;AAAA,MACrB;AACA,UAAI,SAAS,aAAa,QAAW;AACnC,iBAAS,EAAE,GAAG,QAAQ,UAAU,SAAS,SAAS;AAAA,MACpD;AACA,aAAO,CAAC,MAAM;AAAA,IAChB,CAAC,EACA,KAAK,CAAC,MAAM,UAAU,MAAM,QAAQ,KAAK,KAAK,EAC9C,MAAM,GAAG,QAAQ,IAAI;AACxB,IAAAA,gBAAe,QAAQ,WAAW;AAClC,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,QAAQ,SAAwE;AACpF,IAAAA,gBAAe,QAAQ,WAAW;AAClC,UAAM,QAAQ,0BAA0B,QAAQ,KAAK;AACrD,UAAM,QAAQ,KAAK,IAAI,GAAG,KAAK,MAAM,OAAO,QAAQ,UAAU,GAAG,CAAC,CAAC;AACnE,UAAM,YAAY,KAAK,OAAO,EAAE;AAAA,MAAO,CAAC,aACtC,oBAAoB,SAAS,UAAU,QAAQ,MAAM;AAAA,IACvD;AACA,UAAM,OAAO,UAAU,MAAM,OAAO,QAAQ,KAAK;AACjD,UAAM,aAAa,QAAQ,KAAK;AAChC,UAAM,SAAyC;AAAA,MAC7C,OAAO,KAAK,IAAI,CAAC,aAA6C;AAC5D,YAAI,OAAuC;AAAA,UACzC,IAAI,SAAS;AAAA,UACb,UAAU,SAAS;AAAA,QACrB;AACA,YAAI,SAAS,aAAa,QAAW;AACnC,iBAAO,EAAE,GAAG,MAAM,UAAU,SAAS,SAAS;AAAA,QAChD;AACA,eAAO;AAAA,MACT,CAAC;AAAA,MACD,YAAY,UAAU;AAAA,IACxB;AACA,QAAI,aAAa,UAAU,OAAQ,QAAO,aAAa,OAAO,UAAU;AACxE,IAAAA,gBAAe,QAAQ,WAAW;AAClC,WAAO;AAAA,EACT;AAAA,EAEQ,iBAAiB,WAAuD;AAC9E,UAAM,MAAM,oBAAI,IAAY;AAC5B,eAAW,YAAY,WAAW;AAChC,UAAI,IAAI,IAAI,SAAS,EAAE,EAAG,OAAM,IAAI,UAAU,iCAAiC,SAAS,EAAE,EAAE;AAC5F,UAAI,IAAI,SAAS,EAAE;AACnB,UAAI,SAAS,WAAW,WAAW,GAAG;AACpC,cAAM,IAAI,UAAU,mBAAmB,SAAS,EAAE,uCAAuC;AAAA,MAC3F;AACA,iBAAW,aAAa,SAAS;AAC/B,2BAAmB,UAAU,QAAQ,SAAS,EAAE;AAAA,IACpD;AACA,SAAK,2BAA2B,SAAS;AACzC,eAAW,YAAY,UAAW,MAAK,UAAU,IAAI,SAAS,IAAI,QAAQ;AAC1E,SAAK,gBAAgB;AAAA,EACvB;AAAA,EAEQ,WAAW,gBAA+D;AAChF,SAAK,uBAAuB,cAAc;AAC1C,QAAI,KAAK,cAAc,SAAS,SAAS,KAAK,aAAa,OAAW,QAAO,KAAK,OAAO;AACzF,UAAM,eAAe,KAAK,SAAS,MAAM,eAAe,MAAM;AAC9D,QAAI,aAAa,SAAS,EAAG,QAAO,KAAK,OAAO;AAChD,WAAO,CAAC,GAAG,YAAY,EAAE,QAAQ,CAAC,OAAO;AACvC,YAAM,WAAW,KAAK,UAAU,IAAI,EAAE;AACtC,aAAO,aAAa,SAAY,CAAC,IAAI,CAAC,QAAQ;AAAA,IAChD,CAAC;AAAA,EACH;AAAA,EAEQ,kBAAwB;AAC9B,QAAI,KAAK,cAAc,SAAS,OAAO;AACrC,WAAK,WAAW;AAChB;AAAA,IACF;AACA,UAAM,iBAAiB,KAAK,OAAO,EAAE,QAAQ,CAAC,aAAa,SAAS,UAAU,EAAE,CAAC;AACjF,QAAI,mBAAmB,QAAW;AAChC,WAAK,WAAW;AAChB;AAAA,IACF;AACA,UAAM,QAAQ,IAAI,SAAS,eAAe,OAAO,QAAQ,KAAK,aAAa;AAC3E,eAAW,YAAY,KAAK,UAAU,OAAO,GAAG;AAC9C,iBAAW,aAAa,SAAS,WAAY,OAAM,OAAO,SAAS,IAAI,UAAU,MAAM;AAAA,IACzF;AACA,SAAK,WAAW;AAAA,EAClB;AAAA,EAEQ,2BAA2B,WAAuD;AACxF,QAAI,YAAY,KAAK;AACrB,eAAW,YAAY,WAAW;AAChC,iBAAW,aAAa,SAAS,YAAY;AAC3C,oBAAY,2BAA2B,WAAW,WAAW,SAAS,EAAE;AAAA,MAC1E;AAAA,IACF;AACA,SAAK,qBAAqB;AAAA,EAC5B;AAAA,EAEQ,uBAAuB,gBAAiC;AAC9D,QAAI,KAAK,uBAAuB,QAAW;AACzC,iCAA2B,KAAK,oBAAoB,gBAAgB,OAAO;AAAA,IAC7E;AAAA,EACF;AACF;AAEA,SAASA,gBAAe,QAAuC;AAC7D,MAAI,QAAQ,SAAS;AACnB,UAAM,QAAQ,IAAI,MAAM,4BAA4B;AACpD,UAAM,OAAO;AACb,UAAM;AAAA,EACR;AACF;AAEO,SAAS,uBACd,SACgF;AAChF,QAAM,iBAAiB,0BAA0B,QAAQ,QAAQ,CAAC;AAClE,uBAAqB,QAAQ,QAAQ;AACrC,SAAO,WAAW;AAAA,IAChB,MAAM,QAAQ;AAAA,IACd,aACE,QAAQ,eAAe;AAAA,IACzB,aAAa,EAAE,OAAO;AAAA,MACpB,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,SAAS,oDAAoD;AAAA,MACtF,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,gCAAgC;AAAA,IACxF,CAAC;AAAA,IACD,cAAc,EAAE;AAAA,MACd,EAAE,OAAO;AAAA,QACP,OAAO,EAAE,OAAO;AAAA,QAChB,IAAI,EAAE,OAAO;AAAA,QACb,UAAU,EAAE,IAAI;AAAA,QAChB,UAAU,EACP,OAAO,EAAE,OAAO,GAAG,EAAE,MAAM,CAAC,EAAE,OAAO,GAAG,EAAE,OAAO,GAAG,EAAE,QAAQ,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC,EAC3E,SAAS;AAAA,MACd,CAAC;AAAA,IACH;AAAA,IACA,SAAS,CAAC,EAAE,OAAO,KAAK,GAAG,YAAY;AACrC,YAAM,UAAU;AAAA,QACd;AAAA,QACA,MAAM,QAAQ;AAAA,QACd,UAAU,QAAQ;AAAA,QAClB,QAAQ,QAAQ;AAAA,QAChB,SAAS,QAAQ;AAAA,QACjB,aAAa,QAAQ;AAAA,MACvB;AACA,aAAO,YAAY,WAAW,QAAQ,WAAW,SAC7C,kBAAkB;AAAA,QAChB,GAAG;AAAA,QACH,OAAO,QAAQ;AAAA,QACf,QAAQ,QAAQ;AAAA,QAChB,QAAQ,QAAQ;AAAA,MAClB,CAAC,IACD,kBAAkB,EAAE,GAAG,SAAS,OAAO,QAAQ,OAAO,OAAO,QAAQ,MAAM,CAAC;AAAA,IAClF;AAAA,EACF,CAAC;AACH;AAEA,SAAS,UAAU,gBAA2B,YAA6C;AACzF,MAAI;AACJ,aAAW,aAAa,YAAY;AAClC,UAAM,QAAQ,iBAAiB,eAAe,QAAQ,UAAU,MAAM;AACtE,WAAO,SAAS,SAAY,QAAQ,KAAK,IAAI,MAAM,KAAK;AAAA,EAC1D;AACA,SAAO;AACT;AAEA,SAAS,2BACP,mBACA,WACA,IACQ;AACR,qBAAmB,UAAU,QAAQ,EAAE;AACvC,MAAI,sBAAsB,OAAW,QAAO,UAAU,OAAO;AAC7D,MAAI,UAAU,OAAO,WAAW,mBAAmB;AACjD,UAAM,IAAI;AAAA,MACR,uCAAuC,iBAAiB,4BAA4B,UAAU,OAAO,MAAM,QAAQ,EAAE;AAAA,IACvH;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,mBAAmB,QAAkB,IAAkB;AAC9D,MAAI,OAAO,WAAW,EAAG,OAAM,IAAI,UAAU,cAAc,EAAE,qBAAqB;AAClF,MAAI,CAAC,OAAO,MAAM,OAAO,QAAQ,GAAG;AAClC,UAAM,IAAI,UAAU,cAAc,EAAE,oCAAoC;AAAA,EAC1E;AACF;","names":["throwIfAborted"]}

package/dist/chunk-I3Q7XSBB.js

@@ -0,0 +1,59 @@
+import {
+  assertJsonObject
+} from "./chunk-7JLAIN6E.js";
+import {
+  resolveRetryOptions,
+  runWithRetries,
+  throwIfAborted
+} from "./chunk-3RWESPUG.js";
+
+// src/image-generation/generate-image.ts
+async function generateImage(options) {
+  assertNonEmptyString(options.prompt, "Image prompt");
+  throwIfAborted(options.abortSignal);
+  const width = options.width ?? 1024;
+  const height = options.height ?? 1024;
+  assertPositiveInteger(width, "Image width");
+  assertPositiveInteger(height, "Image height");
+  const request = { prompt: options.prompt, width, height };
+  if (options.providerOptions !== void 0) {
+    assertJsonObject(options.providerOptions, "providerOptions");
+    request.providerOptions = options.providerOptions;
+  }
+  const result = await runWithRetries(
+    () => {
+      throwIfAborted(options.abortSignal);
+      return options.model.imageGeneration(
+        request,
+        modelCallOptions(options.abortSignal)
+      );
+    },
+    resolveRetries(options.retries),
+    { streaming: false, abortSignal: options.abortSignal }
+  );
+  if (!Array.isArray(result.images) || result.images.length === 0) {
+    throw new Error("The image generation model returned no images.");
+  }
+  return result;
+}
+function resolveRetries(setting) {
+  return setting === void 0 || setting === false ? void 0 : resolveRetryOptions(setting);
+}
+function modelCallOptions(abortSignal) {
+  return abortSignal === void 0 ? void 0 : { abortSignal };
+}
+function assertNonEmptyString(value, name) {
+  if (typeof value !== "string" || value.trim().length === 0) {
+    throw new TypeError(`${name} must be a non-empty string.`);
+  }
+}
+function assertPositiveInteger(value, name) {
+  if (!Number.isSafeInteger(value) || value < 1) {
+    throw new RangeError(`${name} must be a positive integer.`);
+  }
+}
+
+export {
+  generateImage
+};
+//# sourceMappingURL=chunk-I3Q7XSBB.js.map

package/dist/chunk-I3Q7XSBB.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/image-generation/generate-image.ts"],"sourcesContent":["import type { JsonObject } from \"../completion/types\";\nimport { throwIfAborted } from \"../internal/abort\";\nimport { assertJsonObject } from \"../internal/json-object\";\nimport type { ModelCallOptions } from \"../model-call-options\";\nimport {\n  type ResolvedRetryOptions,\n  type RetrySetting,\n  resolveRetryOptions,\n  runWithRetries,\n} from \"../retry\";\nimport type { ImageGenerationModel, ImageGenerationRequest, ImageGenerationResult } from \"./types\";\n\nexport type GenerateImageOptions<Model extends ImageGenerationModel = ImageGenerationModel> = {\n  model: Model;\n  prompt: string;\n  width?: number | undefined;\n  height?: number | undefined;\n  providerOptions?: JsonObject | undefined;\n  retries?: RetrySetting | undefined;\n  abortSignal?: AbortSignal | undefined;\n};\n\ntype RawResponseOf<Model> =\n  Model extends ImageGenerationModel<infer RawResponse> ? RawResponse : unknown;\n\nexport async function generateImage<Model extends ImageGenerationModel>(\n  options: GenerateImageOptions<Model>,\n): Promise<ImageGenerationResult<RawResponseOf<Model>>> {\n  assertNonEmptyString(options.prompt, \"Image prompt\");\n  throwIfAborted(options.abortSignal);\n  const width = options.width ?? 1024;\n  const height = options.height ?? 1024;\n  assertPositiveInteger(width, \"Image width\");\n  assertPositiveInteger(height, \"Image height\");\n\n  const request: ImageGenerationRequest = { prompt: options.prompt, width, height };\n  if (options.providerOptions !== undefined) {\n    assertJsonObject(options.providerOptions, \"providerOptions\");\n    request.providerOptions = options.providerOptions;\n  }\n  const result = await runWithRetries(\n    () => {\n      throwIfAborted(options.abortSignal);\n      return options.model.imageGeneration(\n        request,\n        modelCallOptions(options.abortSignal),\n      ) as Promise<ImageGenerationResult<RawResponseOf<Model>>>;\n    },\n    resolveRetries(options.retries),\n    { streaming: false, abortSignal: options.abortSignal },\n  );\n  if (!Array.isArray(result.images) || result.images.length === 0) {\n    throw new Error(\"The image generation model returned no images.\");\n  }\n  return result;\n}\n\nfunction resolveRetries(setting: RetrySetting | undefined): ResolvedRetryOptions | undefined {\n  return setting === undefined || setting === false ? undefined : resolveRetryOptions(setting);\n}\n\nfunction modelCallOptions(abortSignal: AbortSignal | undefined): ModelCallOptions | undefined {\n  return abortSignal === undefined ? undefined : { abortSignal };\n}\n\nfunction assertNonEmptyString(value: string, name: string): void {\n  if (typeof value !== \"string\" || value.trim().length === 0) {\n    throw new TypeError(`${name} must be a non-empty string.`);\n  }\n}\n\nfunction assertPositiveInteger(value: number, name: string): void {\n  if (!Number.isSafeInteger(value) || value < 1) {\n    throw new RangeError(`${name} must be a positive integer.`);\n  }\n}\n"],"mappings":";;;;;;;;;;AAyBA,eAAsB,cACpB,SACsD;AACtD,uBAAqB,QAAQ,QAAQ,cAAc;AACnD,iBAAe,QAAQ,WAAW;AAClC,QAAM,QAAQ,QAAQ,SAAS;AAC/B,QAAM,SAAS,QAAQ,UAAU;AACjC,wBAAsB,OAAO,aAAa;AAC1C,wBAAsB,QAAQ,cAAc;AAE5C,QAAM,UAAkC,EAAE,QAAQ,QAAQ,QAAQ,OAAO,OAAO;AAChF,MAAI,QAAQ,oBAAoB,QAAW;AACzC,qBAAiB,QAAQ,iBAAiB,iBAAiB;AAC3D,YAAQ,kBAAkB,QAAQ;AAAA,EACpC;AACA,QAAM,SAAS,MAAM;AAAA,IACnB,MAAM;AACJ,qBAAe,QAAQ,WAAW;AAClC,aAAO,QAAQ,MAAM;AAAA,QACnB;AAAA,QACA,iBAAiB,QAAQ,WAAW;AAAA,MACtC;AAAA,IACF;AAAA,IACA,eAAe,QAAQ,OAAO;AAAA,IAC9B,EAAE,WAAW,OAAO,aAAa,QAAQ,YAAY;AAAA,EACvD;AACA,MAAI,CAAC,MAAM,QAAQ,OAAO,MAAM,KAAK,OAAO,OAAO,WAAW,GAAG;AAC/D,UAAM,IAAI,MAAM,gDAAgD;AAAA,EAClE;AACA,SAAO;AACT;AAEA,SAAS,eAAe,SAAqE;AAC3F,SAAO,YAAY,UAAa,YAAY,QAAQ,SAAY,oBAAoB,OAAO;AAC7F;AAEA,SAAS,iBAAiB,aAAoE;AAC5F,SAAO,gBAAgB,SAAY,SAAY,EAAE,YAAY;AAC/D;AAEA,SAAS,qBAAqB,OAAe,MAAoB;AAC/D,MAAI,OAAO,UAAU,YAAY,MAAM,KAAK,EAAE,WAAW,GAAG;AAC1D,UAAM,IAAI,UAAU,GAAG,IAAI,8BAA8B;AAAA,EAC3D;AACF;AAEA,SAAS,sBAAsB,OAAe,MAAoB;AAChE,MAAI,CAAC,OAAO,cAAc,KAAK,KAAK,QAAQ,GAAG;AAC7C,UAAM,IAAI,WAAW,GAAG,IAAI,8BAA8B;AAAA,EAC5D;AACF;","names":[]}

package/dist/chunk-J6LVLV6P.js

@@ -0,0 +1,13 @@
+// src/mcp/tool.ts
+function isMcpTool(tool) {
+  if (typeof tool !== "object" || tool === null || !("mcp" in tool)) {
+    return false;
+  }
+  const mcp = tool.mcp;
+  return typeof mcp === "object" && mcp !== null && "serverName" in mcp && typeof mcp.serverName === "string" && mcp.serverName.length > 0 && "remoteName" in mcp && typeof mcp.remoteName === "string" && mcp.remoteName.length > 0;
+}
+
+export {
+  isMcpTool
+};
+//# sourceMappingURL=chunk-J6LVLV6P.js.map

package/dist/chunk-J6LVLV6P.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/mcp/tool.ts"],"sourcesContent":["import type { McpTool } from \"./types\";\n\nexport function isMcpTool(tool: unknown): tool is McpTool {\n  if (typeof tool !== \"object\" || tool === null || !(\"mcp\" in tool)) {\n    return false;\n  }\n  const mcp = tool.mcp;\n  return (\n    typeof mcp === \"object\" &&\n    mcp !== null &&\n    \"serverName\" in mcp &&\n    typeof mcp.serverName === \"string\" &&\n    mcp.serverName.length > 0 &&\n    \"remoteName\" in mcp &&\n    typeof mcp.remoteName === \"string\" &&\n    mcp.remoteName.length > 0\n  );\n}\n"],"mappings":";AAEO,SAAS,UAAU,MAAgC;AACxD,MAAI,OAAO,SAAS,YAAY,SAAS,QAAQ,EAAE,SAAS,OAAO;AACjE,WAAO;AAAA,EACT;AACA,QAAM,MAAM,KAAK;AACjB,SACE,OAAO,QAAQ,YACf,QAAQ,QACR,gBAAgB,OAChB,OAAO,IAAI,eAAe,YAC1B,IAAI,WAAW,SAAS,KACxB,gBAAgB,OAChB,OAAO,IAAI,eAAe,YAC1B,IAAI,WAAW,SAAS;AAE5B;","names":[]}

package/dist/chunk-LTXABCOO.js

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

package/dist/chunk-LTXABCOO.js.map

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

package/dist/chunk-OJBFDBLG.js

@@ -0,0 +1,164 @@
+import {
+  isJsonValue
+} from "./chunk-3XQGVDU5.js";
+
+// src/agent/interactions.ts
+import { z } from "zod";
+var nonblank = z.string().trim().min(1);
+var jsonValueSchema = z.custom(isJsonValue, {
+  message: "Expected a strict JSON value"
+});
+var jsonObjectSchema = z.custom(
+  (value) => typeof value === "object" && value !== null && !Array.isArray(value) && isJsonValue(value),
+  { message: "Expected a strict JSON object" }
+);
+var questionChoiceSchema = z.object({
+  label: nonblank,
+  value: nonblank
+}).strict();
+var questionPromptSchema = z.object({
+  id: nonblank,
+  text: nonblank,
+  choices: z.array(questionChoiceSchema).min(1).optional(),
+  allowCustom: z.boolean().optional()
+}).strict().superRefine((question, context) => {
+  if (question.choices === void 0 && question.allowCustom === false) {
+    context.addIssue({
+      code: "custom",
+      message: "A free-text question cannot disable custom answers.",
+      path: ["allowCustom"]
+    });
+  }
+  const values = /* @__PURE__ */ new Set();
+  for (const [index, choice] of (question.choices ?? []).entries()) {
+    if (values.has(choice.value)) {
+      context.addIssue({
+        code: "custom",
+        message: "Question choice values must be unique.",
+        path: ["choices", index, "value"]
+      });
+    }
+    values.add(choice.value);
+  }
+});
+var questionAnswerSchema = z.object({
+  questionId: nonblank,
+  value: nonblank
+}).strict();
+var interactionBase = {
+  id: nonblank,
+  toolName: nonblank,
+  toolCallId: nonblank,
+  callId: z.string().optional(),
+  internalCallId: nonblank
+};
+var approvalRequestSchema = z.object({
+  type: z.literal("tool-approval"),
+  ...interactionBase,
+  input: jsonValueSchema,
+  reason: z.string().optional()
+}).strict();
+var questionRequestSchema = z.object({
+  type: z.literal("tool-question"),
+  ...interactionBase,
+  questions: z.array(questionPromptSchema).min(1)
+}).strict().superRefine((request, context) => {
+  const ids = /* @__PURE__ */ new Set();
+  for (const [index, question] of request.questions.entries()) {
+    if (ids.has(question.id)) {
+      context.addIssue({
+        code: "custom",
+        message: "Question IDs must be unique.",
+        path: ["questions", index, "id"]
+      });
+    }
+    ids.add(question.id);
+  }
+});
+var agentInteractionRequestSchema = z.discriminatedUnion("type", [
+  approvalRequestSchema,
+  questionRequestSchema
+]);
+var agentInteractionResponseSchema = z.discriminatedUnion("type", [
+  z.object({
+    type: z.literal("tool-approval"),
+    approved: z.boolean(),
+    reason: z.string().optional()
+  }).strict(),
+  z.object({
+    type: z.literal("tool-question"),
+    answers: z.array(questionAnswerSchema)
+  }).strict()
+]);
+var agentContinuationSchema = z.object({
+  version: z.literal(1),
+  agentId: nonblank,
+  sourceRunId: nonblank,
+  interaction: agentInteractionRequestSchema,
+  state: jsonObjectSchema
+}).strict();
+function parseAgentInteractionRequest(value) {
+  return freeze(agentInteractionRequestSchema.parse(value));
+}
+function parseAgentInteractionResponse(value) {
+  return freeze(agentInteractionResponseSchema.parse(value));
+}
+function parseAgentQuestionPrompts(value) {
+  const request = questionRequestSchema.parse({
+    type: "tool-question",
+    id: "validation",
+    toolName: "validation",
+    toolCallId: "validation",
+    internalCallId: "validation",
+    questions: value
+  });
+  return freeze(request.questions);
+}
+function parseAgentContinuation(value) {
+  return freeze(agentContinuationSchema.parse(value));
+}
+function assertAgentInteractionResponse(request, response) {
+  if (request.type !== response.type) {
+    throw new TypeError("Interaction response type does not match the pending request.");
+  }
+  if (request.type === "tool-question" && response.type === "tool-question") {
+    assertQuestionAnswers(request, response.answers);
+  }
+}
+function assertQuestionAnswers(request, answers) {
+  const parsed = answers.map((answer) => questionAnswerSchema.parse(answer));
+  const byId = new Map(parsed.map((answer) => [answer.questionId, answer]));
+  if (byId.size !== parsed.length) {
+    throw new TypeError("Question answers must not contain duplicate question IDs.");
+  }
+  if (byId.size !== request.questions.length) {
+    throw new TypeError("Question responses must answer every question exactly once.");
+  }
+  for (const question of request.questions) {
+    const answer = byId.get(question.id);
+    if (answer === void 0) {
+      throw new TypeError(`Question response is missing an answer for "${question.id}".`);
+    }
+    const choices = question.choices;
+    if (choices !== void 0 && question.allowCustom !== true && !choices.some((choice) => choice.value === answer.value)) {
+      throw new TypeError(`Question "${question.id}" requires one of its configured choices.`);
+    }
+  }
+}
+function freeze(value) {
+  if (typeof value !== "object" || value === null || Object.isFrozen(value)) return value;
+  for (const child of Object.values(value)) freeze(child);
+  return Object.freeze(value);
+}
+
+export {
+  agentInteractionRequestSchema,
+  agentInteractionResponseSchema,
+  agentContinuationSchema,
+  parseAgentInteractionRequest,
+  parseAgentInteractionResponse,
+  parseAgentQuestionPrompts,
+  parseAgentContinuation,
+  assertAgentInteractionResponse
+};
+//# sourceMappingURL=chunk-OJBFDBLG.js.map

package/dist/chunk-OJBFDBLG.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/agent/interactions.ts"],"sourcesContent":["import { z } from \"zod\";\nimport { isJsonValue } from \"../completion/json\";\nimport type { JsonObject, JsonValue, ToolQuestionAnswer } from \"../completion/types\";\n\nexport type AgentQuestionChoice = Readonly<{\n  label: string;\n  value: string;\n}>;\n\nexport type AgentQuestionPrompt = Readonly<{\n  id: string;\n  text: string;\n  choices?: readonly AgentQuestionChoice[];\n  allowCustom?: boolean;\n}>;\n\nexport type AgentQuestionAnswer = ToolQuestionAnswer;\n\ntype AgentInteractionRequestBase = Readonly<{\n  id: string;\n  toolName: string;\n  toolCallId: string;\n  callId?: string;\n  internalCallId: string;\n}>;\n\nexport type AgentToolApprovalRequest = AgentInteractionRequestBase &\n  Readonly<{\n    type: \"tool-approval\";\n    input: JsonValue;\n    reason?: string;\n  }>;\n\nexport type AgentToolQuestionRequest = AgentInteractionRequestBase &\n  Readonly<{\n    type: \"tool-question\";\n    questions: readonly AgentQuestionPrompt[];\n  }>;\n\nexport type AgentInteractionRequest = AgentToolApprovalRequest | AgentToolQuestionRequest;\n\nexport type AgentInteractionResponse =\n  | Readonly<{\n      type: \"tool-approval\";\n      approved: boolean;\n      reason?: string;\n    }>\n  | Readonly<{\n      type: \"tool-question\";\n      answers: readonly AgentQuestionAnswer[];\n    }>;\n\nexport type AgentContinuation = Readonly<{\n  version: 1;\n  agentId: string;\n  sourceRunId: string;\n  interaction: AgentInteractionRequest;\n  state: JsonObject;\n}>;\n\nconst nonblank = z.string().trim().min(1);\nconst jsonValueSchema = z.custom<JsonValue>(isJsonValue, {\n  message: \"Expected a strict JSON value\",\n});\nconst jsonObjectSchema = z.custom<JsonObject>(\n  (value) =>\n    typeof value === \"object\" && value !== null && !Array.isArray(value) && isJsonValue(value),\n  { message: \"Expected a strict JSON object\" },\n);\n\nconst questionChoiceSchema = z\n  .object({\n    label: nonblank,\n    value: nonblank,\n  })\n  .strict();\n\nconst questionPromptSchema = z\n  .object({\n    id: nonblank,\n    text: nonblank,\n    choices: z.array(questionChoiceSchema).min(1).optional(),\n    allowCustom: z.boolean().optional(),\n  })\n  .strict()\n  .superRefine((question, context) => {\n    if (question.choices === undefined && question.allowCustom === false) {\n      context.addIssue({\n        code: \"custom\",\n        message: \"A free-text question cannot disable custom answers.\",\n        path: [\"allowCustom\"],\n      });\n    }\n    const values = new Set<string>();\n    for (const [index, choice] of (question.choices ?? []).entries()) {\n      if (values.has(choice.value)) {\n        context.addIssue({\n          code: \"custom\",\n          message: \"Question choice values must be unique.\",\n          path: [\"choices\", index, \"value\"],\n        });\n      }\n      values.add(choice.value);\n    }\n  });\n\nconst questionAnswerSchema = z\n  .object({\n    questionId: nonblank,\n    value: nonblank,\n  })\n  .strict();\n\nconst interactionBase = {\n  id: nonblank,\n  toolName: nonblank,\n  toolCallId: nonblank,\n  callId: z.string().optional(),\n  internalCallId: nonblank,\n};\n\nconst approvalRequestSchema = z\n  .object({\n    type: z.literal(\"tool-approval\"),\n    ...interactionBase,\n    input: jsonValueSchema,\n    reason: z.string().optional(),\n  })\n  .strict();\n\nconst questionRequestSchema = z\n  .object({\n    type: z.literal(\"tool-question\"),\n    ...interactionBase,\n    questions: z.array(questionPromptSchema).min(1),\n  })\n  .strict()\n  .superRefine((request, context) => {\n    const ids = new Set<string>();\n    for (const [index, question] of request.questions.entries()) {\n      if (ids.has(question.id)) {\n        context.addIssue({\n          code: \"custom\",\n          message: \"Question IDs must be unique.\",\n          path: [\"questions\", index, \"id\"],\n        });\n      }\n      ids.add(question.id);\n    }\n  });\n\nexport const agentInteractionRequestSchema = z.discriminatedUnion(\"type\", [\n  approvalRequestSchema,\n  questionRequestSchema,\n]);\n\nexport const agentInteractionResponseSchema = z.discriminatedUnion(\"type\", [\n  z\n    .object({\n      type: z.literal(\"tool-approval\"),\n      approved: z.boolean(),\n      reason: z.string().optional(),\n    })\n    .strict(),\n  z\n    .object({\n      type: z.literal(\"tool-question\"),\n      answers: z.array(questionAnswerSchema),\n    })\n    .strict(),\n]);\n\nexport const agentContinuationSchema = z\n  .object({\n    version: z.literal(1),\n    agentId: nonblank,\n    sourceRunId: nonblank,\n    interaction: agentInteractionRequestSchema,\n    state: jsonObjectSchema,\n  })\n  .strict();\n\nexport function parseAgentInteractionRequest(value: unknown): AgentInteractionRequest {\n  return freeze(agentInteractionRequestSchema.parse(value)) as AgentInteractionRequest;\n}\n\nexport function parseAgentInteractionResponse(value: unknown): AgentInteractionResponse {\n  return freeze(agentInteractionResponseSchema.parse(value)) as AgentInteractionResponse;\n}\n\nexport function parseAgentQuestionPrompts(value: unknown): readonly AgentQuestionPrompt[] {\n  const request = questionRequestSchema.parse({\n    type: \"tool-question\",\n    id: \"validation\",\n    toolName: \"validation\",\n    toolCallId: \"validation\",\n    internalCallId: \"validation\",\n    questions: value,\n  });\n  return freeze(request.questions) as readonly AgentQuestionPrompt[];\n}\n\nexport function parseAgentContinuation(value: unknown): AgentContinuation {\n  return freeze(agentContinuationSchema.parse(value)) as AgentContinuation;\n}\n\nexport function assertAgentInteractionResponse(\n  request: AgentInteractionRequest,\n  response: AgentInteractionResponse,\n): void {\n  if (request.type !== response.type) {\n    throw new TypeError(\"Interaction response type does not match the pending request.\");\n  }\n  if (request.type === \"tool-question\" && response.type === \"tool-question\") {\n    assertQuestionAnswers(request, response.answers);\n  }\n}\n\nfunction assertQuestionAnswers(\n  request: AgentToolQuestionRequest,\n  answers: readonly AgentQuestionAnswer[],\n): void {\n  const parsed = answers.map((answer) => questionAnswerSchema.parse(answer));\n  const byId = new Map(parsed.map((answer) => [answer.questionId, answer]));\n  if (byId.size !== parsed.length) {\n    throw new TypeError(\"Question answers must not contain duplicate question IDs.\");\n  }\n  if (byId.size !== request.questions.length) {\n    throw new TypeError(\"Question responses must answer every question exactly once.\");\n  }\n  for (const question of request.questions) {\n    const answer = byId.get(question.id);\n    if (answer === undefined) {\n      throw new TypeError(`Question response is missing an answer for \"${question.id}\".`);\n    }\n    const choices = question.choices;\n    if (\n      choices !== undefined &&\n      question.allowCustom !== true &&\n      !choices.some((choice) => choice.value === answer.value)\n    ) {\n      throw new TypeError(`Question \"${question.id}\" requires one of its configured choices.`);\n    }\n  }\n}\n\nfunction freeze<T>(value: T): T {\n  if (typeof value !== \"object\" || value === null || Object.isFrozen(value)) return value;\n  for (const child of Object.values(value)) freeze(child);\n  return Object.freeze(value);\n}\n"],"mappings":";;;;;AAAA,SAAS,SAAS;AA4DlB,IAAM,WAAW,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC;AACxC,IAAM,kBAAkB,EAAE,OAAkB,aAAa;AAAA,EACvD,SAAS;AACX,CAAC;AACD,IAAM,mBAAmB,EAAE;AAAA,EACzB,CAAC,UACC,OAAO,UAAU,YAAY,UAAU,QAAQ,CAAC,MAAM,QAAQ,KAAK,KAAK,YAAY,KAAK;AAAA,EAC3F,EAAE,SAAS,gCAAgC;AAC7C;AAEA,IAAM,uBAAuB,EAC1B,OAAO;AAAA,EACN,OAAO;AAAA,EACP,OAAO;AACT,CAAC,EACA,OAAO;AAEV,IAAM,uBAAuB,EAC1B,OAAO;AAAA,EACN,IAAI;AAAA,EACJ,MAAM;AAAA,EACN,SAAS,EAAE,MAAM,oBAAoB,EAAE,IAAI,CAAC,EAAE,SAAS;AAAA,EACvD,aAAa,EAAE,QAAQ,EAAE,SAAS;AACpC,CAAC,EACA,OAAO,EACP,YAAY,CAAC,UAAU,YAAY;AAClC,MAAI,SAAS,YAAY,UAAa,SAAS,gBAAgB,OAAO;AACpE,YAAQ,SAAS;AAAA,MACf,MAAM;AAAA,MACN,SAAS;AAAA,MACT,MAAM,CAAC,aAAa;AAAA,IACtB,CAAC;AAAA,EACH;AACA,QAAM,SAAS,oBAAI,IAAY;AAC/B,aAAW,CAAC,OAAO,MAAM,MAAM,SAAS,WAAW,CAAC,GAAG,QAAQ,GAAG;AAChE,QAAI,OAAO,IAAI,OAAO,KAAK,GAAG;AAC5B,cAAQ,SAAS;AAAA,QACf,MAAM;AAAA,QACN,SAAS;AAAA,QACT,MAAM,CAAC,WAAW,OAAO,OAAO;AAAA,MAClC,CAAC;AAAA,IACH;AACA,WAAO,IAAI,OAAO,KAAK;AAAA,EACzB;AACF,CAAC;AAEH,IAAM,uBAAuB,EAC1B,OAAO;AAAA,EACN,YAAY;AAAA,EACZ,OAAO;AACT,CAAC,EACA,OAAO;AAEV,IAAM,kBAAkB;AAAA,EACtB,IAAI;AAAA,EACJ,UAAU;AAAA,EACV,YAAY;AAAA,EACZ,QAAQ,EAAE,OAAO,EAAE,SAAS;AAAA,EAC5B,gBAAgB;AAClB;AAEA,IAAM,wBAAwB,EAC3B,OAAO;AAAA,EACN,MAAM,EAAE,QAAQ,eAAe;AAAA,EAC/B,GAAG;AAAA,EACH,OAAO;AAAA,EACP,QAAQ,EAAE,OAAO,EAAE,SAAS;AAC9B,CAAC,EACA,OAAO;AAEV,IAAM,wBAAwB,EAC3B,OAAO;AAAA,EACN,MAAM,EAAE,QAAQ,eAAe;AAAA,EAC/B,GAAG;AAAA,EACH,WAAW,EAAE,MAAM,oBAAoB,EAAE,IAAI,CAAC;AAChD,CAAC,EACA,OAAO,EACP,YAAY,CAAC,SAAS,YAAY;AACjC,QAAM,MAAM,oBAAI,IAAY;AAC5B,aAAW,CAAC,OAAO,QAAQ,KAAK,QAAQ,UAAU,QAAQ,GAAG;AAC3D,QAAI,IAAI,IAAI,SAAS,EAAE,GAAG;AACxB,cAAQ,SAAS;AAAA,QACf,MAAM;AAAA,QACN,SAAS;AAAA,QACT,MAAM,CAAC,aAAa,OAAO,IAAI;AAAA,MACjC,CAAC;AAAA,IACH;AACA,QAAI,IAAI,SAAS,EAAE;AAAA,EACrB;AACF,CAAC;AAEI,IAAM,gCAAgC,EAAE,mBAAmB,QAAQ;AAAA,EACxE;AAAA,EACA;AACF,CAAC;AAEM,IAAM,iCAAiC,EAAE,mBAAmB,QAAQ;AAAA,EACzE,EACG,OAAO;AAAA,IACN,MAAM,EAAE,QAAQ,eAAe;AAAA,IAC/B,UAAU,EAAE,QAAQ;AAAA,IACpB,QAAQ,EAAE,OAAO,EAAE,SAAS;AAAA,EAC9B,CAAC,EACA,OAAO;AAAA,EACV,EACG,OAAO;AAAA,IACN,MAAM,EAAE,QAAQ,eAAe;AAAA,IAC/B,SAAS,EAAE,MAAM,oBAAoB;AAAA,EACvC,CAAC,EACA,OAAO;AACZ,CAAC;AAEM,IAAM,0BAA0B,EACpC,OAAO;AAAA,EACN,SAAS,EAAE,QAAQ,CAAC;AAAA,EACpB,SAAS;AAAA,EACT,aAAa;AAAA,EACb,aAAa;AAAA,EACb,OAAO;AACT,CAAC,EACA,OAAO;AAEH,SAAS,6BAA6B,OAAyC;AACpF,SAAO,OAAO,8BAA8B,MAAM,KAAK,CAAC;AAC1D;AAEO,SAAS,8BAA8B,OAA0C;AACtF,SAAO,OAAO,+BAA+B,MAAM,KAAK,CAAC;AAC3D;AAEO,SAAS,0BAA0B,OAAgD;AACxF,QAAM,UAAU,sBAAsB,MAAM;AAAA,IAC1C,MAAM;AAAA,IACN,IAAI;AAAA,IACJ,UAAU;AAAA,IACV,YAAY;AAAA,IACZ,gBAAgB;AAAA,IAChB,WAAW;AAAA,EACb,CAAC;AACD,SAAO,OAAO,QAAQ,SAAS;AACjC;AAEO,SAAS,uBAAuB,OAAmC;AACxE,SAAO,OAAO,wBAAwB,MAAM,KAAK,CAAC;AACpD;AAEO,SAAS,+BACd,SACA,UACM;AACN,MAAI,QAAQ,SAAS,SAAS,MAAM;AAClC,UAAM,IAAI,UAAU,+DAA+D;AAAA,EACrF;AACA,MAAI,QAAQ,SAAS,mBAAmB,SAAS,SAAS,iBAAiB;AACzE,0BAAsB,SAAS,SAAS,OAAO;AAAA,EACjD;AACF;AAEA,SAAS,sBACP,SACA,SACM;AACN,QAAM,SAAS,QAAQ,IAAI,CAAC,WAAW,qBAAqB,MAAM,MAAM,CAAC;AACzE,QAAM,OAAO,IAAI,IAAI,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,YAAY,MAAM,CAAC,CAAC;AACxE,MAAI,KAAK,SAAS,OAAO,QAAQ;AAC/B,UAAM,IAAI,UAAU,2DAA2D;AAAA,EACjF;AACA,MAAI,KAAK,SAAS,QAAQ,UAAU,QAAQ;AAC1C,UAAM,IAAI,UAAU,6DAA6D;AAAA,EACnF;AACA,aAAW,YAAY,QAAQ,WAAW;AACxC,UAAM,SAAS,KAAK,IAAI,SAAS,EAAE;AACnC,QAAI,WAAW,QAAW;AACxB,YAAM,IAAI,UAAU,+CAA+C,SAAS,EAAE,IAAI;AAAA,IACpF;AACA,UAAM,UAAU,SAAS;AACzB,QACE,YAAY,UACZ,SAAS,gBAAgB,QACzB,CAAC,QAAQ,KAAK,CAAC,WAAW,OAAO,UAAU,OAAO,KAAK,GACvD;AACA,YAAM,IAAI,UAAU,aAAa,SAAS,EAAE,2CAA2C;AAAA,IACzF;AAAA,EACF;AACF;AAEA,SAAS,OAAU,OAAa;AAC9B,MAAI,OAAO,UAAU,YAAY,UAAU,QAAQ,OAAO,SAAS,KAAK,EAAG,QAAO;AAClF,aAAW,SAAS,OAAO,OAAO,KAAK,EAAG,QAAO,KAAK;AACtD,SAAO,OAAO,OAAO,KAAK;AAC5B;","names":[]}