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

package/dist/chunk-5MK3BISB.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/mcp/client.ts","../src/mcp/abort.ts","../src/mcp/result.ts","../src/mcp/tool.ts","../src/mcp/url-safety.ts","../src/mcp/client-group.ts"],"sourcesContent":["import { readFileSync } from \"node:fs\";\nimport { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport type { Transport } from \"@modelcontextprotocol/sdk/shared/transport.js\";\nimport type { JsonObject } from \"../completion/index\";\nimport { abortError, throwIfAborted } from \"../internal/abort\";\nimport { linkAbortSignal } from \"./abort\";\nimport { createMcpTool } from \"./tool\";\nimport type { McpClientOptions, McpConnectOptions, McpServer, McpServerInfo } from \"./types\";\nimport { createSafeMcpFetch, parseAndValidateMcpUrl, parseMcpHttpUrl } from \"./url-safety\";\n\nlet coreClientVersion: string | undefined;\n\ntype ClientResource = {\n  readonly client: Client;\n  closePromise?: Promise<void> | undefined;\n};\n\ntype McpFetch = NonNullable<\n  NonNullable<ConstructorParameters<typeof StreamableHTTPClientTransport>[1]>[\"fetch\"]\n>;\n\nconst transportOwnedHttpHeaders = new Set([\n  \"accept\",\n  \"content-type\",\n  \"last-event-id\",\n  \"mcp-protocol-version\",\n  \"mcp-session-id\",\n]);\n\nexport class McpClient {\n  readonly name: string;\n\n  readonly #options: McpClientOptions;\n  #closed = false;\n  #connectAbortController?: AbortController | undefined;\n  #connecting?: Promise<McpServer> | undefined;\n  #resource?: ClientResource | undefined;\n  #server?: McpServer | undefined;\n\n  constructor(options: McpClientOptions) {\n    if (options.name.trim() === \"\") {\n      throw new TypeError(\"MCP client name must not be empty\");\n    }\n    this.name = options.name;\n    this.#options = options;\n  }\n\n  connect(options: McpConnectOptions = {}): Promise<McpServer> {\n    if (this.#closed) {\n      return Promise.reject(new Error(`MCP client \"${this.name}\" is closed`));\n    }\n    if (this.#server !== undefined) {\n      return Promise.resolve(this.#server);\n    }\n    if (this.#connecting !== undefined) {\n      return this.#connecting;\n    }\n\n    if (options.abortSignal?.aborted === true) {\n      return Promise.reject(abortError(options.abortSignal.reason));\n    }\n\n    const abortController = new AbortController();\n    const unlinkAbortSignal = linkAbortSignal(options.abortSignal, abortController);\n    this.#connectAbortController = abortController;\n    const connecting = this.#initialize(abortController.signal);\n    this.#connecting = connecting;\n    const clearConnecting = () => {\n      unlinkAbortSignal();\n      if (this.#connecting === connecting) {\n        this.#connecting = undefined;\n      }\n      if (this.#connectAbortController === abortController) {\n        this.#connectAbortController = undefined;\n      }\n    };\n    void connecting.then(clearConnecting, clearConnecting);\n    return connecting;\n  }\n\n  async close(): Promise<void> {\n    this.#closed = true;\n    this.#connectAbortController?.abort(\n      abortError(new Error(`MCP client \"${this.name}\" was closed during connection`)),\n    );\n\n    const resource = this.#resource;\n    if (resource !== undefined) {\n      await closeResource(resource);\n    }\n  }\n\n  async #initialize(abortSignal: AbortSignal): Promise<McpServer> {\n    const client = createSdkClient();\n    const resource: ClientResource = { client };\n    this.#resource = resource;\n\n    try {\n      throwIfAborted(abortSignal);\n      const transport = await createTransport(this.#options, { abortSignal });\n      throwIfAborted(abortSignal);\n      await client.connect(transport, requestOptions(abortSignal));\n      throwIfAborted(abortSignal);\n      const definitions = await listAllTools(client, abortSignal);\n      throwIfAborted(abortSignal);\n      const tools = definitions.map((definition) =>\n        createMcpTool({\n          definition,\n          client,\n          serverName: this.name,\n          prefix: this.#options.tools?.prefix,\n        }),\n      );\n      assertUniqueToolNames(tools, this.name);\n\n      if (this.#closed) {\n        throw new Error(`MCP client \"${this.name}\" was closed during connection`);\n      }\n\n      const serverInfo = copyServerInfo(client.getServerVersion());\n      const capabilities = copyJsonObject(client.getServerCapabilities());\n      const instructions = client.getInstructions();\n      let registration: McpServer = {\n        name: this.name,\n        tools,\n      };\n      if (serverInfo !== undefined) registration = { ...registration, serverInfo };\n      if (capabilities !== undefined) registration = { ...registration, capabilities };\n      if (instructions !== undefined) registration = { ...registration, instructions };\n      const server = freezeServer(registration);\n      this.#server = server;\n      return server;\n    } catch (error) {\n      try {\n        await closeResource(resource);\n      } catch {\n        // Preserve the actionable connection or discovery failure.\n      }\n      if (this.#resource === resource) {\n        this.#resource = undefined;\n      }\n      this.#server = undefined;\n      throw error;\n    }\n  }\n}\n\nasync function createTransport(\n  options: McpClientOptions,\n  connectOptions: McpConnectOptions,\n): Promise<Transport> {\n  const transport = options.transport;\n  if (transport.type === \"custom\") {\n    return transport.create({ abortSignal: connectOptions.abortSignal });\n  }\n  if (transport.type === \"stdio\") {\n    let parameters: ConstructorParameters<typeof StdioClientTransport>[0] = {\n      command: transport.command,\n    };\n    if (transport.args !== undefined) parameters = { ...parameters, args: transport.args };\n    if (transport.env !== undefined) parameters = { ...parameters, env: transport.env };\n    if (transport.cwd !== undefined) parameters = { ...parameters, cwd: transport.cwd };\n    if (transport.stderr !== undefined) parameters = { ...parameters, stderr: transport.stderr };\n    if (transport.maxBufferSize !== undefined) {\n      parameters = { ...parameters, maxBufferSize: transport.maxBufferSize };\n    }\n    return new StdioClientTransport(parameters);\n  }\n\n  const ssrfProtection = transport.ssrfProtection ?? \"strict\";\n  if (ssrfProtection !== \"strict\" && ssrfProtection !== \"disabled\") {\n    throw new TypeError(\"MCP Streamable HTTP ssrfProtection must be strict or disabled\");\n  }\n  let parameters: ConstructorParameters<typeof StreamableHTTPClientTransport>[1] = {};\n  let url: URL;\n  let transportFetch: McpFetch | undefined;\n  if (ssrfProtection === \"strict\") {\n    transportFetch = createSafeMcpFetch();\n    url = parseAndValidateMcpUrl(transport.url);\n  } else {\n    url = parseMcpHttpUrl(transport.url);\n  }\n  if (transport.headers !== undefined) {\n    const headers = copyMcpHeaders(transport.headers, transport.authProvider !== undefined);\n    const fetchRequest = transportFetch ?? defaultMcpFetch;\n    transportFetch = createMcpEndpointFetch(url, headers, fetchRequest);\n  }\n  if (transportFetch !== undefined) parameters = { ...parameters, fetch: transportFetch };\n  if (transport.authProvider !== undefined) {\n    parameters = { ...parameters, authProvider: transport.authProvider };\n  }\n  if (transport.reconnectionOptions !== undefined) {\n    parameters = { ...parameters, reconnectionOptions: transport.reconnectionOptions };\n  }\n  if (transport.sessionId !== undefined) {\n    parameters = { ...parameters, sessionId: transport.sessionId };\n  }\n  return asSdkTransport(new StreamableHTTPClientTransport(url, parameters));\n}\n\nfunction copyMcpHeaders(\n  value: Readonly<Record<string, string>>,\n  hasAuthProvider: boolean,\n): Headers {\n  if (typeof value !== \"object\" || value === null || Array.isArray(value)) {\n    throw new TypeError(\"MCP Streamable HTTP headers must be a plain object\");\n  }\n  const prototype = Object.getPrototypeOf(value);\n  if (prototype !== Object.prototype && prototype !== null) {\n    throw new TypeError(\"MCP Streamable HTTP headers must be a plain object\");\n  }\n  const headers = new Headers();\n  for (const [name, headerValue] of Object.entries(value)) {\n    if (typeof headerValue !== \"string\") {\n      throw new TypeError(`MCP Streamable HTTP header ${name} must be a string`);\n    }\n    const normalizedName = name.toLowerCase();\n    if (transportOwnedHttpHeaders.has(normalizedName)) {\n      throw new TypeError(`MCP Streamable HTTP header ${name} is owned by the transport`);\n    }\n    if (hasAuthProvider && normalizedName === \"authorization\") {\n      throw new TypeError(\n        \"MCP Streamable HTTP authorization header cannot be combined with authProvider\",\n      );\n    }\n    headers.set(name, headerValue);\n  }\n  return headers;\n}\n\nfunction createMcpEndpointFetch(\n  endpoint: URL,\n  configuredHeaders: Headers,\n  fetchRequest: McpFetch,\n): McpFetch {\n  return (input, init) => {\n    const requestUrl = input instanceof Request ? new URL(input.url) : new URL(input);\n    if (requestUrl.href !== endpoint.href) return fetchRequest(input, init);\n\n    const requestHeaders = new Headers(\n      init?.headers ?? (input instanceof Request ? input.headers : undefined),\n    );\n    configuredHeaders.forEach((value, name) => {\n      requestHeaders.set(name, value);\n    });\n    return fetchRequest(input, { ...init, headers: requestHeaders, redirect: \"error\" });\n  };\n}\n\nconst defaultMcpFetch: McpFetch = (input, init) => globalThis.fetch(input, init);\n\nasync function listAllTools(\n  client: Client,\n  abortSignal?: AbortSignal | undefined,\n): Promise<Awaited<ReturnType<Client[\"listTools\"]>>[\"tools\"]> {\n  const tools: Awaited<ReturnType<Client[\"listTools\"]>>[\"tools\"] = [];\n  const seenCursors = new Set<string>();\n  let cursor: string | undefined;\n\n  do {\n    const page = await client.listTools(\n      cursor === undefined ? undefined : { cursor },\n      requestOptions(abortSignal),\n    );\n    tools.push(...page.tools);\n    cursor = page.nextCursor;\n    if (cursor !== undefined && seenCursors.has(cursor)) {\n      throw new Error(`MCP server returned a repeated tools cursor: ${cursor}`);\n    }\n    if (cursor !== undefined) {\n      seenCursors.add(cursor);\n    }\n  } while (cursor !== undefined);\n\n  return tools;\n}\n\nfunction assertUniqueToolNames(tools: readonly { name: string }[], serverName: string): void {\n  const names = new Set<string>();\n  for (const tool of tools) {\n    if (names.has(tool.name)) {\n      throw new Error(`Duplicate MCP tool name \"${tool.name}\" from server \"${serverName}\"`);\n    }\n    names.add(tool.name);\n  }\n}\n\nfunction createSdkClient(): Client {\n  return new Client({\n    name: \"@anvia/core\",\n    version: getCoreClientVersion(),\n  });\n}\n\nfunction getCoreClientVersion(): string {\n  if (coreClientVersion === undefined) {\n    coreClientVersion = readCorePackageVersion();\n  }\n  return coreClientVersion;\n}\n\nfunction requestOptions(signal?: AbortSignal | undefined): { signal?: AbortSignal } {\n  return signal === undefined ? {} : { signal };\n}\n\nfunction asSdkTransport(transport: unknown): Transport {\n  return transport as Transport;\n}\n\nfunction closeResource(resource: ClientResource): Promise<void> {\n  resource.closePromise ??= resource.client.close();\n  return resource.closePromise;\n}\n\nfunction readCorePackageVersion(): string {\n  try {\n    const packageJson = JSON.parse(\n      readFileSync(new URL(\"../../package.json\", import.meta.url), \"utf8\"),\n    ) as { version?: unknown };\n    return typeof packageJson.version === \"string\" ? packageJson.version : \"0.0.0\";\n  } catch {\n    return \"0.0.0\";\n  }\n}\n\nfunction copyServerInfo(value: ReturnType<Client[\"getServerVersion\"]>): McpServerInfo | undefined {\n  if (value === undefined) {\n    return undefined;\n  }\n  return deepFreeze(structuredClone(value)) as McpServerInfo;\n}\n\nfunction copyJsonObject(value: unknown): JsonObject | undefined {\n  if (value === undefined) {\n    return undefined;\n  }\n  return deepFreeze(structuredClone(value)) as JsonObject;\n}\n\nfunction freezeServer(server: McpServer): McpServer {\n  Object.freeze(server.tools);\n  return Object.freeze(server);\n}\n\nfunction deepFreeze<T>(value: T): T {\n  if (typeof value !== \"object\" || value === null || Object.isFrozen(value)) {\n    return value;\n  }\n  for (const child of Object.values(value)) {\n    deepFreeze(child);\n  }\n  return Object.freeze(value);\n}\n","export function linkAbortSignal(\n  source: AbortSignal | undefined,\n  target: AbortController,\n): () => void {\n  if (source === undefined) {\n    return () => undefined;\n  }\n  if (source.aborted) {\n    target.abort(source.reason);\n    return () => undefined;\n  }\n\n  const onAbort = () => target.abort(source.reason);\n  source.addEventListener(\"abort\", onAbort, { once: true });\n  return () => source.removeEventListener(\"abort\", onAbort);\n}\n","import type { ToolResultContentPart } from \"../completion/index\";\nimport { isRecord } from \"../internal/record\";\n\ntype McpResultContent =\n  | { type: \"text\"; text: string }\n  | { type: \"image\"; data: string; mimeType: string }\n  | { type: \"audio\"; data: string; mimeType: string }\n  | {\n      type: \"resource\";\n      resource:\n        | { uri: string; text: string; mimeType?: string | undefined }\n        | { uri: string; blob: string; mimeType?: string | undefined };\n    }\n  | {\n      type: \"resource_link\";\n      uri: string;\n      name: string;\n      mimeType?: string | undefined;\n    };\n\nexport type McpToolCallResult =\n  | {\n      content: McpResultContent[];\n      structuredContent?: Record<string, unknown> | undefined;\n      isError?: boolean | undefined;\n    }\n  | {\n      toolResult: unknown;\n    };\n\nexport function createCallToolParams(\n  name: string,\n  args: unknown,\n): { name: string; arguments?: Record<string, unknown> } {\n  if (args === null || args === undefined) {\n    return { name };\n  }\n\n  if (!isRecord(args)) {\n    throw new Error(\"MCP tool arguments must be a JSON object\");\n  }\n\n  return { name, arguments: args };\n}\n\nexport function mapMcpToolResult(result: McpToolCallResult): readonly ToolResultContentPart[] {\n  if (\"toolResult\" in result) {\n    return [{ type: \"text\", text: serializeMcpValue(result.toolResult) }];\n  }\n\n  if (result.isError === true) {\n    throw new Error(mcpErrorMessage(result.content));\n  }\n\n  const content = result.content.map(mapMcpContent);\n  if (content.length === 0 && result.structuredContent !== undefined) {\n    content.push({ type: \"text\", text: serializeMcpValue(result.structuredContent) });\n  }\n  return content;\n}\n\nfunction mcpErrorMessage(content: McpResultContent[]): string {\n  const text = content\n    .map((item) => (item.type === \"text\" ? item.text : undefined))\n    .filter((item): item is string => item !== undefined)\n    .join(\"\\n\");\n\n  return text === \"\" ? \"MCP tool returned an error\" : text;\n}\n\nfunction mapMcpContent(content: McpResultContent): ToolResultContentPart {\n  if (content.type === \"text\") {\n    return { type: \"text\", text: content.text };\n  }\n\n  if (content.type === \"image\") {\n    return {\n      type: \"file\",\n      data: { type: \"data\", data: content.data },\n      mediaType: content.mimeType,\n    };\n  }\n\n  if (content.type === \"resource\") {\n    return { type: \"text\", text: serializeResource(content.resource) };\n  }\n\n  if (content.type === \"audio\") {\n    throw new Error(`Unsupported MCP tool result content type: audio (${content.mimeType})`);\n  }\n\n  throw new Error(`Unsupported MCP tool result content type: resource_link (${content.uri})`);\n}\n\nfunction serializeResource(\n  resource:\n    | { uri: string; text: string; mimeType?: string | undefined }\n    | { uri: string; blob: string; mimeType?: string | undefined },\n): string {\n  const mediaType =\n    resource.mimeType ?? (\"text\" in resource ? \"text/plain\" : \"application/octet-stream\");\n  const encoding = \"text\" in resource ? \"text\" : \"base64\";\n  const data = \"text\" in resource ? resource.text : resource.blob;\n  return `MCP resource (${resource.uri}; ${mediaType}; ${encoding})\\n${data}`;\n}\n\nfunction serializeMcpValue(value: unknown): string {\n  if (typeof value === \"string\") {\n    return value;\n  }\n\n  let serialized: string | undefined;\n  try {\n    serialized = JSON.stringify(value);\n  } catch (error) {\n    throw new TypeError(\"MCP tool results must be JSON-serializable.\", { cause: error });\n  }\n  if (serialized === undefined) {\n    throw new TypeError(\"MCP tool results must be JSON-serializable.\");\n  }\n  return serialized;\n}\n","import type { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport type { JsonObject, ToolDefinition } from \"../completion/index\";\nimport { ToolOutput } from \"../tool\";\nimport { createCallToolParams, mapMcpToolResult } from \"./result\";\nimport type { McpTool } from \"./types\";\n\ntype SdkMcpToolDefinition = Awaited<ReturnType<Client[\"listTools\"]>>[\"tools\"][number];\n\nexport function createMcpTool(options: {\n  definition: SdkMcpToolDefinition;\n  client: Client;\n  serverName: string;\n  prefix?: string | undefined;\n}): McpTool {\n  const { definition, client, serverName, prefix = \"\" } = options;\n  const exposedName = `${prefix}${definition.name}`;\n  const provenance = Object.freeze({\n    serverName,\n    remoteName: definition.name,\n  });\n  const toolDefinition = Object.freeze<ToolDefinition>({\n    name: exposedName,\n    description: definition.description ?? \"\",\n    parameters: deepFreeze(structuredClone(definition.inputSchema)) as unknown as JsonObject,\n  });\n  const tool: McpTool = {\n    name: exposedName,\n    mcp: provenance,\n    definition(): ToolDefinition {\n      return toolDefinition;\n    },\n    async call(args, context) {\n      const result = await client.callTool(\n        createCallToolParams(definition.name, args),\n        undefined,\n        context?.abortSignal === undefined ? {} : { signal: context.abortSignal },\n      );\n      return ToolOutput.content(mapMcpToolResult(result));\n    },\n  };\n  return Object.freeze(tool);\n}\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\nfunction deepFreeze<T>(value: T): T {\n  if (typeof value !== \"object\" || value === null || Object.isFrozen(value)) {\n    return value;\n  }\n  for (const child of Object.values(value)) {\n    deepFreeze(child);\n  }\n  return Object.freeze(value);\n}\n","import type { LookupAddress } from \"node:dns\";\nimport { lookup } from \"node:dns/promises\";\nimport type { LookupFunction } from \"node:net\";\nimport { BlockList, isIP } from \"node:net\";\nimport type { StreamableHTTPClientTransportOptions } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport { Agent, Pool, fetch as undiciFetch } from \"undici\";\n\ntype FetchLike = NonNullable<StreamableHTTPClientTransportOptions[\"fetch\"]>;\n\nexport type McpAddressResolver = (hostname: string) => Promise<LookupAddress[]>;\n\nconst blockedIpv4Addresses = new BlockList();\nconst blockedIpv6Addresses = new BlockList();\n\nfor (const [network, prefix] of [\n  [\"0.0.0.0\", 8],\n  [\"10.0.0.0\", 8],\n  [\"100.64.0.0\", 10],\n  [\"127.0.0.0\", 8],\n  [\"169.254.0.0\", 16],\n  [\"172.16.0.0\", 12],\n  [\"192.0.0.0\", 24],\n  [\"192.0.2.0\", 24],\n  [\"192.88.99.0\", 24],\n  [\"192.168.0.0\", 16],\n  [\"198.18.0.0\", 15],\n  [\"198.51.100.0\", 24],\n  [\"203.0.113.0\", 24],\n  [\"224.0.0.0\", 4],\n  [\"240.0.0.0\", 4],\n] as const) {\n  blockedIpv4Addresses.addSubnet(network, prefix, \"ipv4\");\n}\n\nfor (const [network, prefix] of [\n  [\"::\", 96],\n  [\"::ffff:0:0\", 96],\n  [\"64:ff9b::\", 96],\n  [\"64:ff9b:1::\", 48],\n  [\"100::\", 64],\n  [\"2001::\", 32],\n  [\"2001:2::\", 48],\n  [\"2001:10::\", 28],\n  [\"2001:20::\", 28],\n  [\"2001:db8::\", 32],\n  [\"2002::\", 16],\n  [\"fc00::\", 7],\n  [\"fe80::\", 10],\n  [\"fec0::\", 10],\n  [\"ff00::\", 8],\n] as const) {\n  blockedIpv6Addresses.addSubnet(network, prefix, \"ipv6\");\n}\n\nconst defaultAddressResolver: McpAddressResolver = (hostname) =>\n  lookup(hostname, { all: true, verbatim: true });\n\nexport function parseAndValidateMcpUrl(url: string | URL): URL {\n  const parsed = parseMcpHttpUrl(url);\n  validateMcpHostname(parsed.hostname);\n  return parsed;\n}\n\nexport function parseMcpHttpUrl(url: string | URL): URL {\n  let parsed: URL;\n  try {\n    parsed = new URL(url);\n  } catch {\n    throw new Error(`Invalid MCP URL: ${url}`);\n  }\n\n  if (parsed.protocol !== \"http:\" && parsed.protocol !== \"https:\") {\n    throw new Error(`MCP URL blocked: only HTTP(S) URLs are allowed (${parsed.protocol})`);\n  }\n\n  return parsed;\n}\n\nexport async function resolveAndValidateMcpHostname(\n  hostname: string,\n  resolveAddresses: McpAddressResolver = defaultAddressResolver,\n): Promise<LookupAddress[]> {\n  const normalizedHostname = normalizeHostname(hostname);\n  validateMcpHostname(normalizedHostname);\n\n  const family = isIP(normalizedHostname);\n  if (family !== 0) {\n    return [{ address: normalizedHostname, family }];\n  }\n\n  let addresses: LookupAddress[];\n  try {\n    addresses = await resolveAddresses(normalizedHostname);\n  } catch (cause) {\n    throw new Error(`Unable to resolve MCP URL hostname (${normalizedHostname})`, { cause });\n  }\n\n  if (addresses.length === 0) {\n    throw new Error(`Unable to resolve MCP URL hostname (${normalizedHostname})`);\n  }\n\n  for (const { address } of addresses) {\n    validateMcpHostname(address);\n  }\n\n  return addresses;\n}\n\nexport function createSafeMcpLookup(\n  resolveAddresses: McpAddressResolver = defaultAddressResolver,\n): LookupFunction {\n  return (hostname, options, callback) => {\n    void resolveAndValidateMcpHostname(hostname, resolveAddresses).then(\n      (addresses) => {\n        const requestedFamily = normalizeRequestedFamily(options.family);\n        const matchingAddresses =\n          requestedFamily === 0\n            ? addresses\n            : addresses.filter(({ family }) => family === requestedFamily);\n\n        if (matchingAddresses.length === 0) {\n          const error = new Error(`No matching address family for MCP URL (${hostname})`);\n          Object.assign(error, { code: \"EAI_ADDRFAMILY\" });\n          callback(error, []);\n          return;\n        }\n\n        if (options.all) {\n          callback(null, matchingAddresses);\n          return;\n        }\n\n        const address = matchingAddresses[0];\n        if (!address) {\n          callback(new Error(`Unable to resolve MCP URL hostname (${hostname})`), []);\n          return;\n        }\n        callback(null, address.address, address.family);\n      },\n      (cause: unknown) => {\n        callback(asError(cause), []);\n      },\n    );\n  };\n}\n\nexport function createSafeMcpFetch(): FetchLike {\n  return (url, init) => {\n    parseAndValidateMcpUrl(url);\n    return undiciFetch(url as Parameters<typeof undiciFetch>[0], {\n      ...(init as Parameters<typeof undiciFetch>[1]),\n      dispatcher: safeMcpDispatcher,\n    }) as unknown as ReturnType<FetchLike>;\n  };\n}\n\nfunction validateMcpHostname(hostname: string): void {\n  const normalizedHostname = normalizeHostname(hostname);\n\n  if (normalizedHostname === \"localhost\" || normalizedHostname.endsWith(\".localhost\")) {\n    throw new Error(`MCP URL blocked: localhost not allowed (${normalizedHostname})`);\n  }\n\n  const family = isIP(normalizedHostname);\n  if (family === 0) {\n    return;\n  }\n\n  if (\n    (family === 4 && normalizedHostname.startsWith(\"127.\")) ||\n    (family === 6 && normalizedHostname === \"::1\")\n  ) {\n    throw new Error(`MCP URL blocked: localhost not allowed (${normalizedHostname})`);\n  }\n\n  if (normalizedHostname === \"169.254.169.254\" || normalizedHostname === \"fd00:ec2::254\") {\n    throw new Error(`MCP URL blocked: cloud metadata endpoint not allowed (${normalizedHostname})`);\n  }\n\n  if (\n    (family === 4 && blockedIpv4Addresses.check(normalizedHostname, \"ipv4\")) ||\n    (family === 6 && blockedIpv6Addresses.check(normalizedHostname, \"ipv6\"))\n  ) {\n    const range = family === 4 ? \"IP\" : \"IPv6\";\n    throw new Error(`MCP URL blocked: private ${range} range not allowed (${normalizedHostname})`);\n  }\n}\n\nfunction normalizeHostname(hostname: string): string {\n  const withoutBrackets =\n    hostname.startsWith(\"[\") && hostname.endsWith(\"]\") ? hostname.slice(1, -1) : hostname;\n  return withoutBrackets.toLowerCase().replace(/\\.+$/, \"\");\n}\n\nfunction normalizeRequestedFamily(family: number | string | undefined): 0 | 4 | 6 {\n  if (family === 4 || family === \"IPv4\") {\n    return 4;\n  }\n  if (family === 6 || family === \"IPv6\") {\n    return 6;\n  }\n  return 0;\n}\n\nfunction asError(cause: unknown): Error {\n  return cause instanceof Error ? cause : new Error(String(cause));\n}\n\nconst safeMcpDispatcher = new Agent({\n  autoSelectFamily: true,\n  connect: {\n    lookup: createSafeMcpLookup(),\n  },\n  factory(origin, options) {\n    parseAndValidateMcpUrl(origin);\n    return new Pool(origin, options as Pool.Options);\n  },\n});\n","import { linkAbortSignal } from \"./abort\";\nimport type { McpClient } from \"./client\";\nimport type { McpClientGroupConnectOptions, McpServer } from \"./types\";\n\nexport class McpClientGroup {\n  readonly clients: readonly McpClient[];\n  readonly servers: readonly McpServer[];\n\n  #closePromise?: Promise<void> | undefined;\n\n  private constructor(clients: readonly McpClient[], servers: readonly McpServer[]) {\n    this.clients = Object.freeze([...clients]);\n    this.servers = Object.freeze([...servers]);\n  }\n\n  static async connect(options: McpClientGroupConnectOptions): Promise<McpClientGroup> {\n    const clients = [...options.clients];\n    assertUniqueClientNames(clients);\n\n    const abortController = new AbortController();\n    const unlinkAbortSignal = linkAbortSignal(options.abortSignal, abortController);\n    let firstConnectionFailure: unknown;\n    let connectionFailed = false;\n    const connections = clients.map((client) =>\n      client.connect({ abortSignal: abortController.signal }).catch((error: unknown) => {\n        if (!connectionFailed) {\n          connectionFailed = true;\n          firstConnectionFailure = error;\n          abortController.abort(error);\n        }\n        throw error;\n      }),\n    );\n    const settled = await Promise.allSettled(connections).finally(unlinkAbortSignal);\n    const firstFailure = settled.find(\n      (result): result is PromiseRejectedResult => result.status === \"rejected\",\n    );\n    if (firstFailure !== undefined) {\n      await rollbackSuccessfulClients(clients, settled);\n      throw connectionFailed ? firstConnectionFailure : firstFailure.reason;\n    }\n\n    const servers = settled.map((result) => (result as PromiseFulfilledResult<McpServer>).value);\n    try {\n      assertUniqueServerAndToolNames(servers);\n    } catch (error) {\n      try {\n        await closeAll(clients);\n      } catch {\n        // Preserve the registration validation failure that caused rollback.\n      }\n      throw error;\n    }\n\n    return new McpClientGroup(clients, servers);\n  }\n\n  close(): Promise<void> {\n    this.#closePromise ??= closeAll(this.clients);\n    return this.#closePromise;\n  }\n}\n\nfunction assertUniqueClientNames(clients: readonly McpClient[]): void {\n  const names = new Set<string>();\n  for (const client of clients) {\n    if (names.has(client.name)) {\n      throw new Error(`Duplicate MCP client name \"${client.name}\"`);\n    }\n    names.add(client.name);\n  }\n}\n\nfunction assertUniqueServerAndToolNames(servers: readonly McpServer[]): void {\n  const serverNames = new Set<string>();\n  const toolNames = new Map<string, string>();\n  for (const server of servers) {\n    if (serverNames.has(server.name)) {\n      throw new Error(`Duplicate MCP server name \"${server.name}\"`);\n    }\n    serverNames.add(server.name);\n\n    for (const tool of server.tools) {\n      const existing = toolNames.get(tool.name);\n      if (existing !== undefined) {\n        throw new Error(\n          `Duplicate MCP tool name \"${tool.name}\" from servers \"${existing}\" and \"${server.name}\"`,\n        );\n      }\n      toolNames.set(tool.name, server.name);\n    }\n  }\n}\n\nasync function rollbackSuccessfulClients(\n  clients: readonly McpClient[],\n  settled: readonly PromiseSettledResult<McpServer>[],\n): Promise<void> {\n  const successful = clients.filter((_, index) => settled[index]?.status === \"fulfilled\");\n  try {\n    await closeAll(successful);\n  } catch {\n    // Preserve the connection failure that caused rollback.\n  }\n}\n\nasync function closeAll(clients: readonly McpClient[]): Promise<void> {\n  const settled = await Promise.allSettled(clients.map((client) => client.close()));\n  const errors = settled.flatMap((result) => (result.status === \"rejected\" ? [result.reason] : []));\n  if (errors.length > 0) {\n    throw new AggregateError(errors, \"Failed to close one or more MCP clients\");\n  }\n}\n"],"mappings":";;;;;;;;;;;;AAAA,SAAS,oBAAoB;AAC7B,SAAS,cAAc;AACvB,SAAS,4BAA4B;AACrC,SAAS,qCAAqC;;;ACHvC,SAAS,gBACd,QACA,QACY;AACZ,MAAI,WAAW,QAAW;AACxB,WAAO,MAAM;AAAA,EACf;AACA,MAAI,OAAO,SAAS;AAClB,WAAO,MAAM,OAAO,MAAM;AAC1B,WAAO,MAAM;AAAA,EACf;AAEA,QAAM,UAAU,MAAM,OAAO,MAAM,OAAO,MAAM;AAChD,SAAO,iBAAiB,SAAS,SAAS,EAAE,MAAM,KAAK,CAAC;AACxD,SAAO,MAAM,OAAO,oBAAoB,SAAS,OAAO;AAC1D;;;ACeO,SAAS,qBACd,MACA,MACuD;AACvD,MAAI,SAAS,QAAQ,SAAS,QAAW;AACvC,WAAO,EAAE,KAAK;AAAA,EAChB;AAEA,MAAI,CAAC,SAAS,IAAI,GAAG;AACnB,UAAM,IAAI,MAAM,0CAA0C;AAAA,EAC5D;AAEA,SAAO,EAAE,MAAM,WAAW,KAAK;AACjC;AAEO,SAAS,iBAAiB,QAA6D;AAC5F,MAAI,gBAAgB,QAAQ;AAC1B,WAAO,CAAC,EAAE,MAAM,QAAQ,MAAM,kBAAkB,OAAO,UAAU,EAAE,CAAC;AAAA,EACtE;AAEA,MAAI,OAAO,YAAY,MAAM;AAC3B,UAAM,IAAI,MAAM,gBAAgB,OAAO,OAAO,CAAC;AAAA,EACjD;AAEA,QAAM,UAAU,OAAO,QAAQ,IAAI,aAAa;AAChD,MAAI,QAAQ,WAAW,KAAK,OAAO,sBAAsB,QAAW;AAClE,YAAQ,KAAK,EAAE,MAAM,QAAQ,MAAM,kBAAkB,OAAO,iBAAiB,EAAE,CAAC;AAAA,EAClF;AACA,SAAO;AACT;AAEA,SAAS,gBAAgB,SAAqC;AAC5D,QAAM,OAAO,QACV,IAAI,CAAC,SAAU,KAAK,SAAS,SAAS,KAAK,OAAO,MAAU,EAC5D,OAAO,CAAC,SAAyB,SAAS,MAAS,EACnD,KAAK,IAAI;AAEZ,SAAO,SAAS,KAAK,+BAA+B;AACtD;AAEA,SAAS,cAAc,SAAkD;AACvE,MAAI,QAAQ,SAAS,QAAQ;AAC3B,WAAO,EAAE,MAAM,QAAQ,MAAM,QAAQ,KAAK;AAAA,EAC5C;AAEA,MAAI,QAAQ,SAAS,SAAS;AAC5B,WAAO;AAAA,MACL,MAAM;AAAA,MACN,MAAM,EAAE,MAAM,QAAQ,MAAM,QAAQ,KAAK;AAAA,MACzC,WAAW,QAAQ;AAAA,IACrB;AAAA,EACF;AAEA,MAAI,QAAQ,SAAS,YAAY;AAC/B,WAAO,EAAE,MAAM,QAAQ,MAAM,kBAAkB,QAAQ,QAAQ,EAAE;AAAA,EACnE;AAEA,MAAI,QAAQ,SAAS,SAAS;AAC5B,UAAM,IAAI,MAAM,oDAAoD,QAAQ,QAAQ,GAAG;AAAA,EACzF;AAEA,QAAM,IAAI,MAAM,4DAA4D,QAAQ,GAAG,GAAG;AAC5F;AAEA,SAAS,kBACP,UAGQ;AACR,QAAM,YACJ,SAAS,aAAa,UAAU,WAAW,eAAe;AAC5D,QAAM,WAAW,UAAU,WAAW,SAAS;AAC/C,QAAM,OAAO,UAAU,WAAW,SAAS,OAAO,SAAS;AAC3D,SAAO,iBAAiB,SAAS,GAAG,KAAK,SAAS,KAAK,QAAQ;AAAA,EAAM,IAAI;AAC3E;AAEA,SAAS,kBAAkB,OAAwB;AACjD,MAAI,OAAO,UAAU,UAAU;AAC7B,WAAO;AAAA,EACT;AAEA,MAAI;AACJ,MAAI;AACF,iBAAa,KAAK,UAAU,KAAK;AAAA,EACnC,SAAS,OAAO;AACd,UAAM,IAAI,UAAU,+CAA+C,EAAE,OAAO,MAAM,CAAC;AAAA,EACrF;AACA,MAAI,eAAe,QAAW;AAC5B,UAAM,IAAI,UAAU,6CAA6C;AAAA,EACnE;AACA,SAAO;AACT;;;ACjHO,SAAS,cAAc,SAKlB;AACV,QAAM,EAAE,YAAY,QAAQ,YAAY,SAAS,GAAG,IAAI;AACxD,QAAM,cAAc,GAAG,MAAM,GAAG,WAAW,IAAI;AAC/C,QAAM,aAAa,OAAO,OAAO;AAAA,IAC/B;AAAA,IACA,YAAY,WAAW;AAAA,EACzB,CAAC;AACD,QAAM,iBAAiB,OAAO,OAAuB;AAAA,IACnD,MAAM;AAAA,IACN,aAAa,WAAW,eAAe;AAAA,IACvC,YAAY,WAAW,gBAAgB,WAAW,WAAW,CAAC;AAAA,EAChE,CAAC;AACD,QAAM,OAAgB;AAAA,IACpB,MAAM;AAAA,IACN,KAAK;AAAA,IACL,aAA6B;AAC3B,aAAO;AAAA,IACT;AAAA,IACA,MAAM,KAAK,MAAM,SAAS;AACxB,YAAM,SAAS,MAAM,OAAO;AAAA,QAC1B,qBAAqB,WAAW,MAAM,IAAI;AAAA,QAC1C;AAAA,QACA,SAAS,gBAAgB,SAAY,CAAC,IAAI,EAAE,QAAQ,QAAQ,YAAY;AAAA,MAC1E;AACA,aAAO,WAAW,QAAQ,iBAAiB,MAAM,CAAC;AAAA,IACpD;AAAA,EACF;AACA,SAAO,OAAO,OAAO,IAAI;AAC3B;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;AAEA,SAAS,WAAc,OAAa;AAClC,MAAI,OAAO,UAAU,YAAY,UAAU,QAAQ,OAAO,SAAS,KAAK,GAAG;AACzE,WAAO;AAAA,EACT;AACA,aAAW,SAAS,OAAO,OAAO,KAAK,GAAG;AACxC,eAAW,KAAK;AAAA,EAClB;AACA,SAAO,OAAO,OAAO,KAAK;AAC5B;;;ACnEA,SAAS,cAAc;AAEvB,SAAS,WAAW,YAAY;AAEhC,SAAS,OAAO,MAAM,SAAS,mBAAmB;AAMlD,IAAM,uBAAuB,IAAI,UAAU;AAC3C,IAAM,uBAAuB,IAAI,UAAU;AAE3C,WAAW,CAAC,SAAS,MAAM,KAAK;AAAA,EAC9B,CAAC,WAAW,CAAC;AAAA,EACb,CAAC,YAAY,CAAC;AAAA,EACd,CAAC,cAAc,EAAE;AAAA,EACjB,CAAC,aAAa,CAAC;AAAA,EACf,CAAC,eAAe,EAAE;AAAA,EAClB,CAAC,cAAc,EAAE;AAAA,EACjB,CAAC,aAAa,EAAE;AAAA,EAChB,CAAC,aAAa,EAAE;AAAA,EAChB,CAAC,eAAe,EAAE;AAAA,EAClB,CAAC,eAAe,EAAE;AAAA,EAClB,CAAC,cAAc,EAAE;AAAA,EACjB,CAAC,gBAAgB,EAAE;AAAA,EACnB,CAAC,eAAe,EAAE;AAAA,EAClB,CAAC,aAAa,CAAC;AAAA,EACf,CAAC,aAAa,CAAC;AACjB,GAAY;AACV,uBAAqB,UAAU,SAAS,QAAQ,MAAM;AACxD;AAEA,WAAW,CAAC,SAAS,MAAM,KAAK;AAAA,EAC9B,CAAC,MAAM,EAAE;AAAA,EACT,CAAC,cAAc,EAAE;AAAA,EACjB,CAAC,aAAa,EAAE;AAAA,EAChB,CAAC,eAAe,EAAE;AAAA,EAClB,CAAC,SAAS,EAAE;AAAA,EACZ,CAAC,UAAU,EAAE;AAAA,EACb,CAAC,YAAY,EAAE;AAAA,EACf,CAAC,aAAa,EAAE;AAAA,EAChB,CAAC,aAAa,EAAE;AAAA,EAChB,CAAC,cAAc,EAAE;AAAA,EACjB,CAAC,UAAU,EAAE;AAAA,EACb,CAAC,UAAU,CAAC;AAAA,EACZ,CAAC,UAAU,EAAE;AAAA,EACb,CAAC,UAAU,EAAE;AAAA,EACb,CAAC,UAAU,CAAC;AACd,GAAY;AACV,uBAAqB,UAAU,SAAS,QAAQ,MAAM;AACxD;AAEA,IAAM,yBAA6C,CAAC,aAClD,OAAO,UAAU,EAAE,KAAK,MAAM,UAAU,KAAK,CAAC;AAEzC,SAAS,uBAAuB,KAAwB;AAC7D,QAAM,SAAS,gBAAgB,GAAG;AAClC,sBAAoB,OAAO,QAAQ;AACnC,SAAO;AACT;AAEO,SAAS,gBAAgB,KAAwB;AACtD,MAAI;AACJ,MAAI;AACF,aAAS,IAAI,IAAI,GAAG;AAAA,EACtB,QAAQ;AACN,UAAM,IAAI,MAAM,oBAAoB,GAAG,EAAE;AAAA,EAC3C;AAEA,MAAI,OAAO,aAAa,WAAW,OAAO,aAAa,UAAU;AAC/D,UAAM,IAAI,MAAM,mDAAmD,OAAO,QAAQ,GAAG;AAAA,EACvF;AAEA,SAAO;AACT;AAEA,eAAsB,8BACpB,UACA,mBAAuC,wBACb;AAC1B,QAAM,qBAAqB,kBAAkB,QAAQ;AACrD,sBAAoB,kBAAkB;AAEtC,QAAM,SAAS,KAAK,kBAAkB;AACtC,MAAI,WAAW,GAAG;AAChB,WAAO,CAAC,EAAE,SAAS,oBAAoB,OAAO,CAAC;AAAA,EACjD;AAEA,MAAI;AACJ,MAAI;AACF,gBAAY,MAAM,iBAAiB,kBAAkB;AAAA,EACvD,SAAS,OAAO;AACd,UAAM,IAAI,MAAM,uCAAuC,kBAAkB,KAAK,EAAE,MAAM,CAAC;AAAA,EACzF;AAEA,MAAI,UAAU,WAAW,GAAG;AAC1B,UAAM,IAAI,MAAM,uCAAuC,kBAAkB,GAAG;AAAA,EAC9E;AAEA,aAAW,EAAE,QAAQ,KAAK,WAAW;AACnC,wBAAoB,OAAO;AAAA,EAC7B;AAEA,SAAO;AACT;AAEO,SAAS,oBACd,mBAAuC,wBACvB;AAChB,SAAO,CAAC,UAAU,SAAS,aAAa;AACtC,SAAK,8BAA8B,UAAU,gBAAgB,EAAE;AAAA,MAC7D,CAAC,cAAc;AACb,cAAM,kBAAkB,yBAAyB,QAAQ,MAAM;AAC/D,cAAM,oBACJ,oBAAoB,IAChB,YACA,UAAU,OAAO,CAAC,EAAE,OAAO,MAAM,WAAW,eAAe;AAEjE,YAAI,kBAAkB,WAAW,GAAG;AAClC,gBAAM,QAAQ,IAAI,MAAM,2CAA2C,QAAQ,GAAG;AAC9E,iBAAO,OAAO,OAAO,EAAE,MAAM,iBAAiB,CAAC;AAC/C,mBAAS,OAAO,CAAC,CAAC;AAClB;AAAA,QACF;AAEA,YAAI,QAAQ,KAAK;AACf,mBAAS,MAAM,iBAAiB;AAChC;AAAA,QACF;AAEA,cAAM,UAAU,kBAAkB,CAAC;AACnC,YAAI,CAAC,SAAS;AACZ,mBAAS,IAAI,MAAM,uCAAuC,QAAQ,GAAG,GAAG,CAAC,CAAC;AAC1E;AAAA,QACF;AACA,iBAAS,MAAM,QAAQ,SAAS,QAAQ,MAAM;AAAA,MAChD;AAAA,MACA,CAAC,UAAmB;AAClB,iBAAS,QAAQ,KAAK,GAAG,CAAC,CAAC;AAAA,MAC7B;AAAA,IACF;AAAA,EACF;AACF;AAEO,SAAS,qBAAgC;AAC9C,SAAO,CAAC,KAAK,SAAS;AACpB,2BAAuB,GAAG;AAC1B,WAAO,YAAY,KAA0C;AAAA,MAC3D,GAAI;AAAA,MACJ,YAAY;AAAA,IACd,CAAC;AAAA,EACH;AACF;AAEA,SAAS,oBAAoB,UAAwB;AACnD,QAAM,qBAAqB,kBAAkB,QAAQ;AAErD,MAAI,uBAAuB,eAAe,mBAAmB,SAAS,YAAY,GAAG;AACnF,UAAM,IAAI,MAAM,2CAA2C,kBAAkB,GAAG;AAAA,EAClF;AAEA,QAAM,SAAS,KAAK,kBAAkB;AACtC,MAAI,WAAW,GAAG;AAChB;AAAA,EACF;AAEA,MACG,WAAW,KAAK,mBAAmB,WAAW,MAAM,KACpD,WAAW,KAAK,uBAAuB,OACxC;AACA,UAAM,IAAI,MAAM,2CAA2C,kBAAkB,GAAG;AAAA,EAClF;AAEA,MAAI,uBAAuB,qBAAqB,uBAAuB,iBAAiB;AACtF,UAAM,IAAI,MAAM,yDAAyD,kBAAkB,GAAG;AAAA,EAChG;AAEA,MACG,WAAW,KAAK,qBAAqB,MAAM,oBAAoB,MAAM,KACrE,WAAW,KAAK,qBAAqB,MAAM,oBAAoB,MAAM,GACtE;AACA,UAAM,QAAQ,WAAW,IAAI,OAAO;AACpC,UAAM,IAAI,MAAM,4BAA4B,KAAK,uBAAuB,kBAAkB,GAAG;AAAA,EAC/F;AACF;AAEA,SAAS,kBAAkB,UAA0B;AACnD,QAAM,kBACJ,SAAS,WAAW,GAAG,KAAK,SAAS,SAAS,GAAG,IAAI,SAAS,MAAM,GAAG,EAAE,IAAI;AAC/E,SAAO,gBAAgB,YAAY,EAAE,QAAQ,QAAQ,EAAE;AACzD;AAEA,SAAS,yBAAyB,QAAgD;AAChF,MAAI,WAAW,KAAK,WAAW,QAAQ;AACrC,WAAO;AAAA,EACT;AACA,MAAI,WAAW,KAAK,WAAW,QAAQ;AACrC,WAAO;AAAA,EACT;AACA,SAAO;AACT;AAEA,SAAS,QAAQ,OAAuB;AACtC,SAAO,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC;AACjE;AAEA,IAAM,oBAAoB,IAAI,MAAM;AAAA,EAClC,kBAAkB;AAAA,EAClB,SAAS;AAAA,IACP,QAAQ,oBAAoB;AAAA,EAC9B;AAAA,EACA,QAAQ,QAAQ,SAAS;AACvB,2BAAuB,MAAM;AAC7B,WAAO,IAAI,KAAK,QAAQ,OAAuB;AAAA,EACjD;AACF,CAAC;;;AJ7MD,IAAI;AAWJ,IAAM,4BAA4B,oBAAI,IAAI;AAAA,EACxC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,CAAC;AAEM,IAAM,YAAN,MAAgB;AAAA,EACZ;AAAA,EAEA;AAAA,EACT,UAAU;AAAA,EACV;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EAEA,YAAY,SAA2B;AACrC,QAAI,QAAQ,KAAK,KAAK,MAAM,IAAI;AAC9B,YAAM,IAAI,UAAU,mCAAmC;AAAA,IACzD;AACA,SAAK,OAAO,QAAQ;AACpB,SAAK,WAAW;AAAA,EAClB;AAAA,EAEA,QAAQ,UAA6B,CAAC,GAAuB;AAC3D,QAAI,KAAK,SAAS;AAChB,aAAO,QAAQ,OAAO,IAAI,MAAM,eAAe,KAAK,IAAI,aAAa,CAAC;AAAA,IACxE;AACA,QAAI,KAAK,YAAY,QAAW;AAC9B,aAAO,QAAQ,QAAQ,KAAK,OAAO;AAAA,IACrC;AACA,QAAI,KAAK,gBAAgB,QAAW;AAClC,aAAO,KAAK;AAAA,IACd;AAEA,QAAI,QAAQ,aAAa,YAAY,MAAM;AACzC,aAAO,QAAQ,OAAO,WAAW,QAAQ,YAAY,MAAM,CAAC;AAAA,IAC9D;AAEA,UAAM,kBAAkB,IAAI,gBAAgB;AAC5C,UAAM,oBAAoB,gBAAgB,QAAQ,aAAa,eAAe;AAC9E,SAAK,0BAA0B;AAC/B,UAAM,aAAa,KAAK,YAAY,gBAAgB,MAAM;AAC1D,SAAK,cAAc;AACnB,UAAM,kBAAkB,MAAM;AAC5B,wBAAkB;AAClB,UAAI,KAAK,gBAAgB,YAAY;AACnC,aAAK,cAAc;AAAA,MACrB;AACA,UAAI,KAAK,4BAA4B,iBAAiB;AACpD,aAAK,0BAA0B;AAAA,MACjC;AAAA,IACF;AACA,SAAK,WAAW,KAAK,iBAAiB,eAAe;AACrD,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,QAAuB;AAC3B,SAAK,UAAU;AACf,SAAK,yBAAyB;AAAA,MAC5B,WAAW,IAAI,MAAM,eAAe,KAAK,IAAI,gCAAgC,CAAC;AAAA,IAChF;AAEA,UAAM,WAAW,KAAK;AACtB,QAAI,aAAa,QAAW;AAC1B,YAAM,cAAc,QAAQ;AAAA,IAC9B;AAAA,EACF;AAAA,EAEA,MAAM,YAAY,aAA8C;AAC9D,UAAM,SAAS,gBAAgB;AAC/B,UAAM,WAA2B,EAAE,OAAO;AAC1C,SAAK,YAAY;AAEjB,QAAI;AACF,qBAAe,WAAW;AAC1B,YAAM,YAAY,MAAM,gBAAgB,KAAK,UAAU,EAAE,YAAY,CAAC;AACtE,qBAAe,WAAW;AAC1B,YAAM,OAAO,QAAQ,WAAW,eAAe,WAAW,CAAC;AAC3D,qBAAe,WAAW;AAC1B,YAAM,cAAc,MAAM,aAAa,QAAQ,WAAW;AAC1D,qBAAe,WAAW;AAC1B,YAAM,QAAQ,YAAY;AAAA,QAAI,CAAC,eAC7B,cAAc;AAAA,UACZ;AAAA,UACA;AAAA,UACA,YAAY,KAAK;AAAA,UACjB,QAAQ,KAAK,SAAS,OAAO;AAAA,QAC/B,CAAC;AAAA,MACH;AACA,4BAAsB,OAAO,KAAK,IAAI;AAEtC,UAAI,KAAK,SAAS;AAChB,cAAM,IAAI,MAAM,eAAe,KAAK,IAAI,gCAAgC;AAAA,MAC1E;AAEA,YAAM,aAAa,eAAe,OAAO,iBAAiB,CAAC;AAC3D,YAAM,eAAe,eAAe,OAAO,sBAAsB,CAAC;AAClE,YAAM,eAAe,OAAO,gBAAgB;AAC5C,UAAI,eAA0B;AAAA,QAC5B,MAAM,KAAK;AAAA,QACX;AAAA,MACF;AACA,UAAI,eAAe,OAAW,gBAAe,EAAE,GAAG,cAAc,WAAW;AAC3E,UAAI,iBAAiB,OAAW,gBAAe,EAAE,GAAG,cAAc,aAAa;AAC/E,UAAI,iBAAiB,OAAW,gBAAe,EAAE,GAAG,cAAc,aAAa;AAC/E,YAAM,SAAS,aAAa,YAAY;AACxC,WAAK,UAAU;AACf,aAAO;AAAA,IACT,SAAS,OAAO;AACd,UAAI;AACF,cAAM,cAAc,QAAQ;AAAA,MAC9B,QAAQ;AAAA,MAER;AACA,UAAI,KAAK,cAAc,UAAU;AAC/B,aAAK,YAAY;AAAA,MACnB;AACA,WAAK,UAAU;AACf,YAAM;AAAA,IACR;AAAA,EACF;AACF;AAEA,eAAe,gBACb,SACA,gBACoB;AACpB,QAAM,YAAY,QAAQ;AAC1B,MAAI,UAAU,SAAS,UAAU;AAC/B,WAAO,UAAU,OAAO,EAAE,aAAa,eAAe,YAAY,CAAC;AAAA,EACrE;AACA,MAAI,UAAU,SAAS,SAAS;AAC9B,QAAIA,cAAoE;AAAA,MACtE,SAAS,UAAU;AAAA,IACrB;AACA,QAAI,UAAU,SAAS,OAAW,CAAAA,cAAa,EAAE,GAAGA,aAAY,MAAM,UAAU,KAAK;AACrF,QAAI,UAAU,QAAQ,OAAW,CAAAA,cAAa,EAAE,GAAGA,aAAY,KAAK,UAAU,IAAI;AAClF,QAAI,UAAU,QAAQ,OAAW,CAAAA,cAAa,EAAE,GAAGA,aAAY,KAAK,UAAU,IAAI;AAClF,QAAI,UAAU,WAAW,OAAW,CAAAA,cAAa,EAAE,GAAGA,aAAY,QAAQ,UAAU,OAAO;AAC3F,QAAI,UAAU,kBAAkB,QAAW;AACzC,MAAAA,cAAa,EAAE,GAAGA,aAAY,eAAe,UAAU,cAAc;AAAA,IACvE;AACA,WAAO,IAAI,qBAAqBA,WAAU;AAAA,EAC5C;AAEA,QAAM,iBAAiB,UAAU,kBAAkB;AACnD,MAAI,mBAAmB,YAAY,mBAAmB,YAAY;AAChE,UAAM,IAAI,UAAU,+DAA+D;AAAA,EACrF;AACA,MAAI,aAA6E,CAAC;AAClF,MAAI;AACJ,MAAI;AACJ,MAAI,mBAAmB,UAAU;AAC/B,qBAAiB,mBAAmB;AACpC,UAAM,uBAAuB,UAAU,GAAG;AAAA,EAC5C,OAAO;AACL,UAAM,gBAAgB,UAAU,GAAG;AAAA,EACrC;AACA,MAAI,UAAU,YAAY,QAAW;AACnC,UAAM,UAAU,eAAe,UAAU,SAAS,UAAU,iBAAiB,MAAS;AACtF,UAAM,eAAe,kBAAkB;AACvC,qBAAiB,uBAAuB,KAAK,SAAS,YAAY;AAAA,EACpE;AACA,MAAI,mBAAmB,OAAW,cAAa,EAAE,GAAG,YAAY,OAAO,eAAe;AACtF,MAAI,UAAU,iBAAiB,QAAW;AACxC,iBAAa,EAAE,GAAG,YAAY,cAAc,UAAU,aAAa;AAAA,EACrE;AACA,MAAI,UAAU,wBAAwB,QAAW;AAC/C,iBAAa,EAAE,GAAG,YAAY,qBAAqB,UAAU,oBAAoB;AAAA,EACnF;AACA,MAAI,UAAU,cAAc,QAAW;AACrC,iBAAa,EAAE,GAAG,YAAY,WAAW,UAAU,UAAU;AAAA,EAC/D;AACA,SAAO,eAAe,IAAI,8BAA8B,KAAK,UAAU,CAAC;AAC1E;AAEA,SAAS,eACP,OACA,iBACS;AACT,MAAI,OAAO,UAAU,YAAY,UAAU,QAAQ,MAAM,QAAQ,KAAK,GAAG;AACvE,UAAM,IAAI,UAAU,oDAAoD;AAAA,EAC1E;AACA,QAAM,YAAY,OAAO,eAAe,KAAK;AAC7C,MAAI,cAAc,OAAO,aAAa,cAAc,MAAM;AACxD,UAAM,IAAI,UAAU,oDAAoD;AAAA,EAC1E;AACA,QAAM,UAAU,IAAI,QAAQ;AAC5B,aAAW,CAAC,MAAM,WAAW,KAAK,OAAO,QAAQ,KAAK,GAAG;AACvD,QAAI,OAAO,gBAAgB,UAAU;AACnC,YAAM,IAAI,UAAU,8BAA8B,IAAI,mBAAmB;AAAA,IAC3E;AACA,UAAM,iBAAiB,KAAK,YAAY;AACxC,QAAI,0BAA0B,IAAI,cAAc,GAAG;AACjD,YAAM,IAAI,UAAU,8BAA8B,IAAI,4BAA4B;AAAA,IACpF;AACA,QAAI,mBAAmB,mBAAmB,iBAAiB;AACzD,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AACA,YAAQ,IAAI,MAAM,WAAW;AAAA,EAC/B;AACA,SAAO;AACT;AAEA,SAAS,uBACP,UACA,mBACA,cACU;AACV,SAAO,CAAC,OAAO,SAAS;AACtB,UAAM,aAAa,iBAAiB,UAAU,IAAI,IAAI,MAAM,GAAG,IAAI,IAAI,IAAI,KAAK;AAChF,QAAI,WAAW,SAAS,SAAS,KAAM,QAAO,aAAa,OAAO,IAAI;AAEtE,UAAM,iBAAiB,IAAI;AAAA,MACzB,MAAM,YAAY,iBAAiB,UAAU,MAAM,UAAU;AAAA,IAC/D;AACA,sBAAkB,QAAQ,CAAC,OAAO,SAAS;AACzC,qBAAe,IAAI,MAAM,KAAK;AAAA,IAChC,CAAC;AACD,WAAO,aAAa,OAAO,EAAE,GAAG,MAAM,SAAS,gBAAgB,UAAU,QAAQ,CAAC;AAAA,EACpF;AACF;AAEA,IAAM,kBAA4B,CAAC,OAAO,SAAS,WAAW,MAAM,OAAO,IAAI;AAE/E,eAAe,aACb,QACA,aAC4D;AAC5D,QAAM,QAA2D,CAAC;AAClE,QAAM,cAAc,oBAAI,IAAY;AACpC,MAAI;AAEJ,KAAG;AACD,UAAM,OAAO,MAAM,OAAO;AAAA,MACxB,WAAW,SAAY,SAAY,EAAE,OAAO;AAAA,MAC5C,eAAe,WAAW;AAAA,IAC5B;AACA,UAAM,KAAK,GAAG,KAAK,KAAK;AACxB,aAAS,KAAK;AACd,QAAI,WAAW,UAAa,YAAY,IAAI,MAAM,GAAG;AACnD,YAAM,IAAI,MAAM,gDAAgD,MAAM,EAAE;AAAA,IAC1E;AACA,QAAI,WAAW,QAAW;AACxB,kBAAY,IAAI,MAAM;AAAA,IACxB;AAAA,EACF,SAAS,WAAW;AAEpB,SAAO;AACT;AAEA,SAAS,sBAAsB,OAAoC,YAA0B;AAC3F,QAAM,QAAQ,oBAAI,IAAY;AAC9B,aAAW,QAAQ,OAAO;AACxB,QAAI,MAAM,IAAI,KAAK,IAAI,GAAG;AACxB,YAAM,IAAI,MAAM,4BAA4B,KAAK,IAAI,kBAAkB,UAAU,GAAG;AAAA,IACtF;AACA,UAAM,IAAI,KAAK,IAAI;AAAA,EACrB;AACF;AAEA,SAAS,kBAA0B;AACjC,SAAO,IAAI,OAAO;AAAA,IAChB,MAAM;AAAA,IACN,SAAS,qBAAqB;AAAA,EAChC,CAAC;AACH;AAEA,SAAS,uBAA+B;AACtC,MAAI,sBAAsB,QAAW;AACnC,wBAAoB,uBAAuB;AAAA,EAC7C;AACA,SAAO;AACT;AAEA,SAAS,eAAe,QAA4D;AAClF,SAAO,WAAW,SAAY,CAAC,IAAI,EAAE,OAAO;AAC9C;AAEA,SAAS,eAAe,WAA+B;AACrD,SAAO;AACT;AAEA,SAAS,cAAc,UAAyC;AAC9D,WAAS,iBAAiB,SAAS,OAAO,MAAM;AAChD,SAAO,SAAS;AAClB;AAEA,SAAS,yBAAiC;AACxC,MAAI;AACF,UAAM,cAAc,KAAK;AAAA,MACvB,aAAa,IAAI,IAAI,sBAAsB,YAAY,GAAG,GAAG,MAAM;AAAA,IACrE;AACA,WAAO,OAAO,YAAY,YAAY,WAAW,YAAY,UAAU;AAAA,EACzE,QAAQ;AACN,WAAO;AAAA,EACT;AACF;AAEA,SAAS,eAAe,OAA0E;AAChG,MAAI,UAAU,QAAW;AACvB,WAAO;AAAA,EACT;AACA,SAAOC,YAAW,gBAAgB,KAAK,CAAC;AAC1C;AAEA,SAAS,eAAe,OAAwC;AAC9D,MAAI,UAAU,QAAW;AACvB,WAAO;AAAA,EACT;AACA,SAAOA,YAAW,gBAAgB,KAAK,CAAC;AAC1C;AAEA,SAAS,aAAa,QAA8B;AAClD,SAAO,OAAO,OAAO,KAAK;AAC1B,SAAO,OAAO,OAAO,MAAM;AAC7B;AAEA,SAASA,YAAc,OAAa;AAClC,MAAI,OAAO,UAAU,YAAY,UAAU,QAAQ,OAAO,SAAS,KAAK,GAAG;AACzE,WAAO;AAAA,EACT;AACA,aAAW,SAAS,OAAO,OAAO,KAAK,GAAG;AACxC,IAAAA,YAAW,KAAK;AAAA,EAClB;AACA,SAAO,OAAO,OAAO,KAAK;AAC5B;;;AK9VO,IAAM,iBAAN,MAAM,gBAAe;AAAA,EACjB;AAAA,EACA;AAAA,EAET;AAAA,EAEQ,YAAY,SAA+B,SAA+B;AAChF,SAAK,UAAU,OAAO,OAAO,CAAC,GAAG,OAAO,CAAC;AACzC,SAAK,UAAU,OAAO,OAAO,CAAC,GAAG,OAAO,CAAC;AAAA,EAC3C;AAAA,EAEA,aAAa,QAAQ,SAAgE;AACnF,UAAM,UAAU,CAAC,GAAG,QAAQ,OAAO;AACnC,4BAAwB,OAAO;AAE/B,UAAM,kBAAkB,IAAI,gBAAgB;AAC5C,UAAM,oBAAoB,gBAAgB,QAAQ,aAAa,eAAe;AAC9E,QAAI;AACJ,QAAI,mBAAmB;AACvB,UAAM,cAAc,QAAQ;AAAA,MAAI,CAAC,WAC/B,OAAO,QAAQ,EAAE,aAAa,gBAAgB,OAAO,CAAC,EAAE,MAAM,CAAC,UAAmB;AAChF,YAAI,CAAC,kBAAkB;AACrB,6BAAmB;AACnB,mCAAyB;AACzB,0BAAgB,MAAM,KAAK;AAAA,QAC7B;AACA,cAAM;AAAA,MACR,CAAC;AAAA,IACH;AACA,UAAM,UAAU,MAAM,QAAQ,WAAW,WAAW,EAAE,QAAQ,iBAAiB;AAC/E,UAAM,eAAe,QAAQ;AAAA,MAC3B,CAAC,WAA4C,OAAO,WAAW;AAAA,IACjE;AACA,QAAI,iBAAiB,QAAW;AAC9B,YAAM,0BAA0B,SAAS,OAAO;AAChD,YAAM,mBAAmB,yBAAyB,aAAa;AAAA,IACjE;AAEA,UAAM,UAAU,QAAQ,IAAI,CAAC,WAAY,OAA6C,KAAK;AAC3F,QAAI;AACF,qCAA+B,OAAO;AAAA,IACxC,SAAS,OAAO;AACd,UAAI;AACF,cAAM,SAAS,OAAO;AAAA,MACxB,QAAQ;AAAA,MAER;AACA,YAAM;AAAA,IACR;AAEA,WAAO,IAAI,gBAAe,SAAS,OAAO;AAAA,EAC5C;AAAA,EAEA,QAAuB;AACrB,SAAK,kBAAkB,SAAS,KAAK,OAAO;AAC5C,WAAO,KAAK;AAAA,EACd;AACF;AAEA,SAAS,wBAAwB,SAAqC;AACpE,QAAM,QAAQ,oBAAI,IAAY;AAC9B,aAAW,UAAU,SAAS;AAC5B,QAAI,MAAM,IAAI,OAAO,IAAI,GAAG;AAC1B,YAAM,IAAI,MAAM,8BAA8B,OAAO,IAAI,GAAG;AAAA,IAC9D;AACA,UAAM,IAAI,OAAO,IAAI;AAAA,EACvB;AACF;AAEA,SAAS,+BAA+B,SAAqC;AAC3E,QAAM,cAAc,oBAAI,IAAY;AACpC,QAAM,YAAY,oBAAI,IAAoB;AAC1C,aAAW,UAAU,SAAS;AAC5B,QAAI,YAAY,IAAI,OAAO,IAAI,GAAG;AAChC,YAAM,IAAI,MAAM,8BAA8B,OAAO,IAAI,GAAG;AAAA,IAC9D;AACA,gBAAY,IAAI,OAAO,IAAI;AAE3B,eAAW,QAAQ,OAAO,OAAO;AAC/B,YAAM,WAAW,UAAU,IAAI,KAAK,IAAI;AACxC,UAAI,aAAa,QAAW;AAC1B,cAAM,IAAI;AAAA,UACR,4BAA4B,KAAK,IAAI,mBAAmB,QAAQ,UAAU,OAAO,IAAI;AAAA,QACvF;AAAA,MACF;AACA,gBAAU,IAAI,KAAK,MAAM,OAAO,IAAI;AAAA,IACtC;AAAA,EACF;AACF;AAEA,eAAe,0BACb,SACA,SACe;AACf,QAAM,aAAa,QAAQ,OAAO,CAAC,GAAG,UAAU,QAAQ,KAAK,GAAG,WAAW,WAAW;AACtF,MAAI;AACF,UAAM,SAAS,UAAU;AAAA,EAC3B,QAAQ;AAAA,EAER;AACF;AAEA,eAAe,SAAS,SAA8C;AACpE,QAAM,UAAU,MAAM,QAAQ,WAAW,QAAQ,IAAI,CAAC,WAAW,OAAO,MAAM,CAAC,CAAC;AAChF,QAAM,SAAS,QAAQ,QAAQ,CAAC,WAAY,OAAO,WAAW,aAAa,CAAC,OAAO,MAAM,IAAI,CAAC,CAAE;AAChG,MAAI,OAAO,SAAS,GAAG;AACrB,UAAM,IAAI,eAAe,QAAQ,yCAAyC;AAAA,EAC5E;AACF;","names":["parameters","deepFreeze"]}

package/dist/chunk-AI5JMUW7.js

@@ -0,0 +1,45 @@
+import {
+  resolveRetryOptions,
+  runWithRetries,
+  throwIfAborted
+} from "./chunk-FN7HLLAZ.js";
+
+// src/speech-generation/generate-speech.ts
+async function generateSpeech(options) {
+  assertNonEmptyString(options.text, "Speech text");
+  assertNonEmptyString(options.voice, "Speech voice");
+  throwIfAborted(options.abortSignal);
+  const speed = options.speed ?? 1;
+  if (!Number.isFinite(speed) || speed <= 0) {
+    throw new RangeError("Speech speed must be a positive finite number.");
+  }
+  const request = { text: options.text, voice: options.voice, speed };
+  if (options.providerOptions !== void 0) request.providerOptions = options.providerOptions;
+  return runWithRetries(
+    () => {
+      throwIfAborted(options.abortSignal);
+      return options.model.speechGeneration(
+        request,
+        modelCallOptions(options.abortSignal)
+      );
+    },
+    resolveRetries(options.retries),
+    { streaming: false, abortSignal: options.abortSignal }
+  );
+}
+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.`);
+  }
+}
+
+export {
+  generateSpeech
+};
+//# sourceMappingURL=chunk-AI5JMUW7.js.map

package/dist/chunk-AI5JMUW7.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/speech-generation/generate-speech.ts"],"sourcesContent":["import type { JsonObject } from \"../completion/types\";\nimport { throwIfAborted } from \"../internal/abort\";\nimport type { ModelCallOptions } from \"../model-call-options\";\nimport {\n  type ResolvedRetryOptions,\n  type RetrySetting,\n  resolveRetryOptions,\n  runWithRetries,\n} from \"../retry\";\nimport type {\n  SpeechGenerationModel,\n  SpeechGenerationRequest,\n  SpeechGenerationResult,\n} from \"./types\";\n\nexport type GenerateSpeechOptions<Model extends SpeechGenerationModel = SpeechGenerationModel> = {\n  model: Model;\n  text: string;\n  voice: string;\n  speed?: number | undefined;\n  providerOptions?: JsonObject | undefined;\n  retries?: RetrySetting | undefined;\n  abortSignal?: AbortSignal | undefined;\n};\n\ntype RawResponseOf<Model> =\n  Model extends SpeechGenerationModel<infer RawResponse> ? RawResponse : unknown;\n\nexport async function generateSpeech<Model extends SpeechGenerationModel>(\n  options: GenerateSpeechOptions<Model>,\n): Promise<SpeechGenerationResult<RawResponseOf<Model>>> {\n  assertNonEmptyString(options.text, \"Speech text\");\n  assertNonEmptyString(options.voice, \"Speech voice\");\n  throwIfAborted(options.abortSignal);\n  const speed = options.speed ?? 1;\n  if (!Number.isFinite(speed) || speed <= 0) {\n    throw new RangeError(\"Speech speed must be a positive finite number.\");\n  }\n\n  const request: SpeechGenerationRequest = { text: options.text, voice: options.voice, speed };\n  if (options.providerOptions !== undefined) request.providerOptions = options.providerOptions;\n  return runWithRetries(\n    () => {\n      throwIfAborted(options.abortSignal);\n      return options.model.speechGeneration(\n        request,\n        modelCallOptions(options.abortSignal),\n      ) as Promise<SpeechGenerationResult<RawResponseOf<Model>>>;\n    },\n    resolveRetries(options.retries),\n    { streaming: false, abortSignal: options.abortSignal },\n  );\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"],"mappings":";;;;;;;AA4BA,eAAsB,eACpB,SACuD;AACvD,uBAAqB,QAAQ,MAAM,aAAa;AAChD,uBAAqB,QAAQ,OAAO,cAAc;AAClD,iBAAe,QAAQ,WAAW;AAClC,QAAM,QAAQ,QAAQ,SAAS;AAC/B,MAAI,CAAC,OAAO,SAAS,KAAK,KAAK,SAAS,GAAG;AACzC,UAAM,IAAI,WAAW,gDAAgD;AAAA,EACvE;AAEA,QAAM,UAAmC,EAAE,MAAM,QAAQ,MAAM,OAAO,QAAQ,OAAO,MAAM;AAC3F,MAAI,QAAQ,oBAAoB,OAAW,SAAQ,kBAAkB,QAAQ;AAC7E,SAAO;AAAA,IACL,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;AACF;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;","names":[]}

package/dist/{chunk-FNQB2JEH.js → chunk-AZB6N7P4.js}

@@ -1,4 +1,28 @@
-// src/guardrails/index.ts
+// src/guardrails/actions.ts
+function allow(options = {}) {
+  return { action: "allow", ...options };
+}
+function block(options) {
+  return { action: "block", ...options };
+}
+var commonActions = {
+  allow,
+  block
+};
+var inputActions = {
+  ...commonActions,
+  rewrite(options) {
+    return { action: "rewrite", ...options };
+  }
+};
+var outputActions = {
+  ...commonActions,
+  rewrite(options) {
+    return { action: "rewrite", ...options };
+  }
+};
+
+// src/guardrails/policy.ts
 function defineGuardrailPolicy(options) {
   return {
     id: options.id,
@@ -13,16 +37,6 @@ function defineInputGuardrail(guardrail) {
 function defineOutputGuardrail(guardrail) {
   return guardrail;
 }
-function allow(options = {}) {
-  return { action: "allow", ...options };
-}
-function block(options) {
-  return { action: "block", ...options };
-}
-var guardrails = {
-  blockText,
-  redactText
-};
 function normalizeGuardrailPolicies(policies) {
   if (policies === void 0) {
     return [];
@@ -35,6 +49,65 @@ function appendGuardrailPolicies(current, next) {
 function hasEnforcedOutputGuardrails(policies) {
   return policies.some((policy) => policy.mode === "enforce" && policy.output.length > 0);
 }
+
+// src/guardrails/message.ts
+function textFromMessage(message) {
+  if (message.role === "system") {
+    return message.content;
+  }
+  if (typeof message.content === "string") {
+    return message.content;
+  }
+  return message.content.flatMap((content) => {
+    if (content.type === "text") {
+      return [content.text];
+    }
+    if (content.type === "file" && content.data.type === "text") {
+      return [content.data.text];
+    }
+    if (content.type === "tool-result") {
+      return textFromToolResult(content.output);
+    }
+    return [];
+  }).join("\n");
+}
+function rewriteMessageText(message, text) {
+  if (message.role === "system") {
+    return { ...message, content: text };
+  }
+  if (message.role === "user") {
+    const content = typeof message.content === "string" ? [] : message.content;
+    return {
+      ...message,
+      content: [
+        { type: "text", text },
+        ...content.filter(
+          (item) => item.type !== "text" && !(item.type === "file" && item.data.type === "text")
+        )
+      ]
+    };
+  }
+  if (message.role === "assistant") {
+    return { ...message, content: [{ type: "text", text }] };
+  }
+  return message;
+}
+function textFromToolResult(output) {
+  switch (output.type) {
+    case "text":
+    case "error-text":
+      return [output.value];
+    case "json":
+    case "error-json":
+      return [JSON.stringify(output.value)];
+    case "content":
+      return output.value.flatMap((part) => part.type === "text" ? [part.text] : []);
+    case "execution-denied":
+      return output.reason === void 0 ? [] : [output.reason];
+  }
+}
+
+// src/guardrails/runtime.ts
 async function runInputGuardrails(policies, context) {
   let prompt = context.prompt;
   let inputText = context.inputText;
@@ -157,21 +230,11 @@ function decisionRecord(args) {
   if (args.action.metadata !== void 0) record.metadata = args.action.metadata;
   return record;
 }
-var commonActions = {
-  allow,
-  block
-};
-var inputActions = {
-  ...commonActions,
-  rewrite(options) {
-    return { action: "rewrite", ...options };
-  }
-};
-var outputActions = {
-  ...commonActions,
-  rewrite(options) {
-    return { action: "rewrite", ...options };
-  }
+
+// src/guardrails/text.ts
+var guardrails = {
+  blockText,
+  redactText
 };
 function blockText(options) {
   return textPatternGuardrail(options, "block");
@@ -240,52 +303,18 @@ function replaceTextPattern(text, pattern, replacement) {
   const globalFlags = flags.includes("g") ? flags : `${flags}g`;
   return text.replace(new RegExp(pattern.source, globalFlags), replacement);
 }
-function textFromMessage(message) {
-  if (message.role === "system") {
-    return message.content;
-  }
-  return message.content.flatMap((content) => {
-    if (content.type === "text") {
-      return [content.text];
-    }
-    if (content.type === "document" && content.source.type === "text") {
-      return [content.source.text];
-    }
-    return [];
-  }).join("\n");
-}
-function rewriteMessageText(message, text) {
-  if (message.role === "system") {
-    return { ...message, content: text };
-  }
-  if (message.role === "user") {
-    return {
-      ...message,
-      content: [
-        { type: "text", text },
-        ...message.content.filter(
-          (item) => item.type !== "text" && !(item.type === "document" && item.source.type === "text")
-        )
-      ]
-    };
-  }
-  if (message.role === "assistant") {
-    return { ...message, content: [{ type: "text", text }] };
-  }
-  return message;
-}
 
 export {
+  allow,
+  block,
   defineGuardrailPolicy,
   defineInputGuardrail,
   defineOutputGuardrail,
-  allow,
-  block,
-  guardrails,
   normalizeGuardrailPolicies,
   appendGuardrailPolicies,
   hasEnforcedOutputGuardrails,
   runInputGuardrails,
-  runOutputGuardrails
+  runOutputGuardrails,
+  guardrails
 };
-//# sourceMappingURL=chunk-FNQB2JEH.js.map
+//# sourceMappingURL=chunk-AZB6N7P4.js.map

package/dist/chunk-AZB6N7P4.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/guardrails/actions.ts","../src/guardrails/policy.ts","../src/guardrails/message.ts","../src/guardrails/runtime.ts","../src/guardrails/text.ts"],"sourcesContent":["import type {\n  GuardrailActionBase,\n  GuardrailAllow,\n  GuardrailBlock,\n  GuardrailCommonActions,\n  InputGuardrailActions,\n  OutputGuardrailActions,\n} from \"./types\";\n\nexport function allow(options: GuardrailActionBase = {}): GuardrailAllow {\n  return { action: \"allow\", ...options };\n}\n\nexport function block(options: Omit<GuardrailBlock, \"action\">): GuardrailBlock {\n  return { action: \"block\", ...options };\n}\n\nconst commonActions: GuardrailCommonActions = {\n  allow,\n  block,\n};\n\nexport const inputActions: InputGuardrailActions = {\n  ...commonActions,\n  rewrite(options) {\n    return { action: \"rewrite\", ...options };\n  },\n};\n\nexport const outputActions: OutputGuardrailActions = {\n  ...commonActions,\n  rewrite(options) {\n    return { action: \"rewrite\", ...options };\n  },\n};\n","import type {\n  GuardrailPolicy,\n  GuardrailPolicyInput,\n  GuardrailPolicyOptions,\n  InputGuardrail,\n  OutputGuardrail,\n} from \"./types\";\n\nexport function defineGuardrailPolicy(options: GuardrailPolicyOptions): GuardrailPolicy {\n  return {\n    id: options.id,\n    mode: options.mode ?? \"enforce\",\n    input: options.input ?? [],\n    output: options.output ?? [],\n  };\n}\n\nexport function defineInputGuardrail(guardrail: InputGuardrail): InputGuardrail {\n  return guardrail;\n}\n\nexport function defineOutputGuardrail(guardrail: OutputGuardrail): OutputGuardrail {\n  return guardrail;\n}\n\nexport function normalizeGuardrailPolicies(\n  policies: GuardrailPolicyInput | undefined,\n): GuardrailPolicy[] {\n  if (policies === undefined) {\n    return [];\n  }\n  return Array.isArray(policies) ? policies : [policies];\n}\n\nexport function appendGuardrailPolicies(\n  current: GuardrailPolicy[],\n  next: GuardrailPolicyInput,\n): GuardrailPolicy[] {\n  return [...current, ...normalizeGuardrailPolicies(next)];\n}\n\nexport function hasEnforcedOutputGuardrails(policies: GuardrailPolicy[]): boolean {\n  return policies.some((policy) => policy.mode === \"enforce\" && policy.output.length > 0);\n}\n","import type { Message, ToolResultPart } from \"../completion\";\n\nexport function textFromMessage(message: Message): string {\n  if (message.role === \"system\") {\n    return message.content;\n  }\n  if (typeof message.content === \"string\") {\n    return message.content;\n  }\n  return message.content\n    .flatMap((content) => {\n      if (content.type === \"text\") {\n        return [content.text];\n      }\n      if (content.type === \"file\" && content.data.type === \"text\") {\n        return [content.data.text];\n      }\n      if (content.type === \"tool-result\") {\n        return textFromToolResult(content.output);\n      }\n      return [];\n    })\n    .join(\"\\n\");\n}\n\nexport function rewriteMessageText(message: Message, text: string): Message {\n  if (message.role === \"system\") {\n    return { ...message, content: text };\n  }\n  if (message.role === \"user\") {\n    const content = typeof message.content === \"string\" ? [] : message.content;\n    return {\n      ...message,\n      content: [\n        { type: \"text\", text },\n        ...content.filter(\n          (item) => item.type !== \"text\" && !(item.type === \"file\" && item.data.type === \"text\"),\n        ),\n      ],\n    };\n  }\n  if (message.role === \"assistant\") {\n    return { ...message, content: [{ type: \"text\", text }] };\n  }\n  return message;\n}\n\nfunction textFromToolResult(output: ToolResultPart[\"output\"]): string[] {\n  switch (output.type) {\n    case \"text\":\n    case \"error-text\":\n      return [output.value];\n    case \"json\":\n    case \"error-json\":\n      return [JSON.stringify(output.value)];\n    case \"content\":\n      return output.value.flatMap((part) => (part.type === \"text\" ? [part.text] : []));\n    case \"execution-denied\":\n      return output.reason === undefined ? [] : [output.reason];\n  }\n}\n","import type { MaybePromise } from \"../internal/type-utils\";\nimport { allow, inputActions, outputActions } from \"./actions\";\nimport { rewriteMessageText, textFromMessage } from \"./message\";\nimport type {\n  GuardrailActionBase,\n  GuardrailActionName,\n  GuardrailAllow,\n  GuardrailBlock,\n  GuardrailBoundary,\n  GuardrailDecisionRecord,\n  GuardrailPolicy,\n  InputGuardrailContext,\n  InputGuardrailRunResult,\n  OutputGuardrailContext,\n  OutputGuardrailRunResult,\n} from \"./types\";\n\nexport async function runInputGuardrails(\n  policies: GuardrailPolicy[],\n  context: InputGuardrailContext,\n): Promise<InputGuardrailRunResult> {\n  let prompt = context.prompt;\n  let inputText = context.inputText;\n  const decisions: GuardrailDecisionRecord[] = [];\n\n  for (const policy of policies) {\n    for (const guardrail of policy.input) {\n      const currentContext = { ...context, prompt, inputText };\n      const result = await runGuardrail(policy, guardrail.id, \"input\", () =>\n        guardrail.check(currentContext, inputActions),\n      );\n      decisions.push(result.decision);\n\n      if (policy.mode === \"observe\") {\n        continue;\n      }\n      if (result.action.action === \"block\") {\n        return {\n          prompt,\n          inputText,\n          blocked: true,\n          message: result.action.message,\n          decisions,\n        };\n      }\n      if (result.action.action === \"rewrite\") {\n        if (result.action.prompt !== undefined) {\n          prompt = result.action.prompt;\n          inputText = textFromMessage(prompt);\n        } else if (result.action.inputText !== undefined) {\n          inputText = result.action.inputText;\n          prompt = rewriteMessageText(prompt, inputText);\n        }\n      }\n    }\n  }\n\n  return { prompt, inputText, blocked: false, decisions };\n}\n\nexport async function runOutputGuardrails(\n  policies: GuardrailPolicy[],\n  context: OutputGuardrailContext,\n): Promise<OutputGuardrailRunResult> {\n  let outputText = context.outputText;\n  const decisions: GuardrailDecisionRecord[] = [];\n\n  for (const policy of policies) {\n    for (const guardrail of policy.output) {\n      const currentContext = { ...context, outputText };\n      const result = await runGuardrail(policy, guardrail.id, \"output\", () =>\n        guardrail.check(currentContext, outputActions),\n      );\n      decisions.push(result.decision);\n\n      if (policy.mode === \"observe\") {\n        continue;\n      }\n      if (result.action.action === \"block\") {\n        return {\n          outputText,\n          blocked: true,\n          message: result.action.message,\n          decisions,\n        };\n      }\n      if (result.action.action === \"rewrite\") {\n        outputText = result.action.outputText;\n      }\n    }\n  }\n\n  return { outputText, blocked: false, decisions };\n}\n\ntype RunGuardrailResult<TAction extends { action: GuardrailActionName }> = {\n  action: TAction;\n  decision: GuardrailDecisionRecord;\n};\n\nasync function runGuardrail<TAction extends { action: GuardrailActionName } & GuardrailActionBase>(\n  policy: GuardrailPolicy,\n  guardrailId: string,\n  boundary: GuardrailBoundary,\n  run: () => MaybePromise<TAction | undefined>,\n): Promise<RunGuardrailResult<TAction | GuardrailAllow>> {\n  const startedAt = Date.now();\n  try {\n    const action = (await run()) ?? allow();\n    return {\n      action,\n      decision: decisionRecord({\n        policy,\n        guardrailId,\n        boundary,\n        action,\n        latencyMs: Date.now() - startedAt,\n      }),\n    };\n  } catch (error) {\n    const message = error instanceof Error ? error.message : String(error);\n    if (policy.mode === \"observe\") {\n      return {\n        action: allow(),\n        decision: {\n          policyId: policy.id,\n          guardrailId,\n          boundary,\n          mode: policy.mode,\n          action: \"error\",\n          applied: false,\n          reason: message,\n          latencyMs: Date.now() - startedAt,\n        },\n      };\n    }\n    throw error;\n  }\n}\n\nfunction decisionRecord(args: {\n  policy: GuardrailPolicy;\n  guardrailId: string;\n  boundary: GuardrailBoundary;\n  action: GuardrailAllow | GuardrailBlock | ({ action: GuardrailActionName } & GuardrailActionBase);\n  latencyMs: number;\n}): GuardrailDecisionRecord {\n  const applied = args.policy.mode === \"enforce\" && args.action.action !== \"allow\";\n  const record: GuardrailDecisionRecord = {\n    policyId: args.policy.id,\n    guardrailId: args.guardrailId,\n    boundary: args.boundary,\n    mode: args.policy.mode,\n    action: args.action.action,\n    applied,\n    latencyMs: args.latencyMs,\n  };\n  if (args.action.reason !== undefined) record.reason = args.action.reason;\n  if (\"message\" in args.action && typeof args.action.message === \"string\") {\n    record.message = args.action.message;\n  }\n  if (args.action.metadata !== undefined) record.metadata = args.action.metadata;\n  return record;\n}\n","import { allow, block } from \"./actions\";\nimport { defineInputGuardrail, defineOutputGuardrail } from \"./policy\";\nimport type { GuardrailAllow, GuardrailBlock, InputGuardrail, OutputGuardrail } from \"./types\";\n\nexport const guardrails = {\n  blockText,\n  redactText,\n};\n\ntype TextPatternBoundary = \"input\" | \"output\";\n\ntype TextPatternGuardrailFor<Boundary extends TextPatternBoundary> = Boundary extends \"input\"\n  ? InputGuardrail\n  : OutputGuardrail;\n\ntype TextPatternGuardrailOptions<Boundary extends TextPatternBoundary = TextPatternBoundary> = {\n  id: string;\n  boundary: Boundary;\n  patterns: Array<string | RegExp>;\n  reason: string;\n  message?: string | undefined;\n};\n\ntype TextPatternRedactOptions<Boundary extends TextPatternBoundary = TextPatternBoundary> =\n  TextPatternGuardrailOptions<Boundary> & { replacement?: string | undefined };\n\nfunction blockText<Boundary extends TextPatternBoundary>(\n  options: TextPatternGuardrailOptions<Boundary>,\n): TextPatternGuardrailFor<Boundary> {\n  return textPatternGuardrail(options, \"block\");\n}\n\nfunction redactText<Boundary extends TextPatternBoundary>(\n  options: TextPatternRedactOptions<Boundary>,\n): TextPatternGuardrailFor<Boundary> {\n  return textPatternGuardrail(options, \"rewrite\");\n}\n\nfunction textPatternGuardrail<Boundary extends TextPatternBoundary>(\n  options: TextPatternRedactOptions<Boundary>,\n  action: \"block\" | \"rewrite\",\n): TextPatternGuardrailFor<Boundary> {\n  if (options.boundary === \"input\") {\n    return defineInputGuardrail({\n      id: options.id,\n      check(ctx, actions) {\n        return textPatternAction(ctx.inputText, options, action, (value) =>\n          actions.rewrite({ inputText: value, reason: options.reason }),\n        );\n      },\n    }) as TextPatternGuardrailFor<Boundary>;\n  }\n  return defineOutputGuardrail({\n    id: options.id,\n    check(ctx, actions) {\n      return textPatternAction(ctx.outputText, options, action, (value) =>\n        actions.rewrite({ outputText: value, reason: options.reason }),\n      );\n    },\n  }) as TextPatternGuardrailFor<Boundary>;\n}\n\nfunction textPatternAction<TRewrite>(\n  text: string,\n  options: TextPatternGuardrailOptions & { replacement?: string | undefined },\n  action: \"block\" | \"rewrite\",\n  rewrite: (value: string) => TRewrite,\n): GuardrailAllow | GuardrailBlock | TRewrite {\n  const matched = options.patterns.some((pattern) => textPatternMatches(text, pattern));\n  if (!matched) {\n    return allow();\n  }\n  if (action === \"block\") {\n    return block({\n      reason: options.reason,\n      message: options.message,\n    });\n  }\n  let current = text;\n  for (const pattern of options.patterns) {\n    current = replaceTextPattern(current, pattern, options.replacement ?? \"[redacted]\");\n  }\n  return rewrite(current);\n}\n\nfunction textPatternMatches(text: string, pattern: string | RegExp): boolean {\n  if (typeof pattern === \"string\") {\n    return text.includes(pattern);\n  }\n  pattern.lastIndex = 0;\n  const matched = pattern.test(text);\n  pattern.lastIndex = 0;\n  return matched;\n}\n\nfunction replaceTextPattern(text: string, pattern: string | RegExp, replacement: string): string {\n  if (typeof pattern === \"string\") {\n    return text.split(pattern).join(replacement);\n  }\n  pattern.lastIndex = 0;\n  const flags = pattern.flags.replace(\"y\", \"\");\n  const globalFlags = flags.includes(\"g\") ? flags : `${flags}g`;\n  return text.replace(new RegExp(pattern.source, globalFlags), replacement);\n}\n"],"mappings":";AASO,SAAS,MAAM,UAA+B,CAAC,GAAmB;AACvE,SAAO,EAAE,QAAQ,SAAS,GAAG,QAAQ;AACvC;AAEO,SAAS,MAAM,SAAyD;AAC7E,SAAO,EAAE,QAAQ,SAAS,GAAG,QAAQ;AACvC;AAEA,IAAM,gBAAwC;AAAA,EAC5C;AAAA,EACA;AACF;AAEO,IAAM,eAAsC;AAAA,EACjD,GAAG;AAAA,EACH,QAAQ,SAAS;AACf,WAAO,EAAE,QAAQ,WAAW,GAAG,QAAQ;AAAA,EACzC;AACF;AAEO,IAAM,gBAAwC;AAAA,EACnD,GAAG;AAAA,EACH,QAAQ,SAAS;AACf,WAAO,EAAE,QAAQ,WAAW,GAAG,QAAQ;AAAA,EACzC;AACF;;;AC1BO,SAAS,sBAAsB,SAAkD;AACtF,SAAO;AAAA,IACL,IAAI,QAAQ;AAAA,IACZ,MAAM,QAAQ,QAAQ;AAAA,IACtB,OAAO,QAAQ,SAAS,CAAC;AAAA,IACzB,QAAQ,QAAQ,UAAU,CAAC;AAAA,EAC7B;AACF;AAEO,SAAS,qBAAqB,WAA2C;AAC9E,SAAO;AACT;AAEO,SAAS,sBAAsB,WAA6C;AACjF,SAAO;AACT;AAEO,SAAS,2BACd,UACmB;AACnB,MAAI,aAAa,QAAW;AAC1B,WAAO,CAAC;AAAA,EACV;AACA,SAAO,MAAM,QAAQ,QAAQ,IAAI,WAAW,CAAC,QAAQ;AACvD;AAEO,SAAS,wBACd,SACA,MACmB;AACnB,SAAO,CAAC,GAAG,SAAS,GAAG,2BAA2B,IAAI,CAAC;AACzD;AAEO,SAAS,4BAA4B,UAAsC;AAChF,SAAO,SAAS,KAAK,CAAC,WAAW,OAAO,SAAS,aAAa,OAAO,OAAO,SAAS,CAAC;AACxF;;;ACzCO,SAAS,gBAAgB,SAA0B;AACxD,MAAI,QAAQ,SAAS,UAAU;AAC7B,WAAO,QAAQ;AAAA,EACjB;AACA,MAAI,OAAO,QAAQ,YAAY,UAAU;AACvC,WAAO,QAAQ;AAAA,EACjB;AACA,SAAO,QAAQ,QACZ,QAAQ,CAAC,YAAY;AACpB,QAAI,QAAQ,SAAS,QAAQ;AAC3B,aAAO,CAAC,QAAQ,IAAI;AAAA,IACtB;AACA,QAAI,QAAQ,SAAS,UAAU,QAAQ,KAAK,SAAS,QAAQ;AAC3D,aAAO,CAAC,QAAQ,KAAK,IAAI;AAAA,IAC3B;AACA,QAAI,QAAQ,SAAS,eAAe;AAClC,aAAO,mBAAmB,QAAQ,MAAM;AAAA,IAC1C;AACA,WAAO,CAAC;AAAA,EACV,CAAC,EACA,KAAK,IAAI;AACd;AAEO,SAAS,mBAAmB,SAAkB,MAAuB;AAC1E,MAAI,QAAQ,SAAS,UAAU;AAC7B,WAAO,EAAE,GAAG,SAAS,SAAS,KAAK;AAAA,EACrC;AACA,MAAI,QAAQ,SAAS,QAAQ;AAC3B,UAAM,UAAU,OAAO,QAAQ,YAAY,WAAW,CAAC,IAAI,QAAQ;AACnE,WAAO;AAAA,MACL,GAAG;AAAA,MACH,SAAS;AAAA,QACP,EAAE,MAAM,QAAQ,KAAK;AAAA,QACrB,GAAG,QAAQ;AAAA,UACT,CAAC,SAAS,KAAK,SAAS,UAAU,EAAE,KAAK,SAAS,UAAU,KAAK,KAAK,SAAS;AAAA,QACjF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACA,MAAI,QAAQ,SAAS,aAAa;AAChC,WAAO,EAAE,GAAG,SAAS,SAAS,CAAC,EAAE,MAAM,QAAQ,KAAK,CAAC,EAAE;AAAA,EACzD;AACA,SAAO;AACT;AAEA,SAAS,mBAAmB,QAA4C;AACtE,UAAQ,OAAO,MAAM;AAAA,IACnB,KAAK;AAAA,IACL,KAAK;AACH,aAAO,CAAC,OAAO,KAAK;AAAA,IACtB,KAAK;AAAA,IACL,KAAK;AACH,aAAO,CAAC,KAAK,UAAU,OAAO,KAAK,CAAC;AAAA,IACtC,KAAK;AACH,aAAO,OAAO,MAAM,QAAQ,CAAC,SAAU,KAAK,SAAS,SAAS,CAAC,KAAK,IAAI,IAAI,CAAC,CAAE;AAAA,IACjF,KAAK;AACH,aAAO,OAAO,WAAW,SAAY,CAAC,IAAI,CAAC,OAAO,MAAM;AAAA,EAC5D;AACF;;;AC3CA,eAAsB,mBACpB,UACA,SACkC;AAClC,MAAI,SAAS,QAAQ;AACrB,MAAI,YAAY,QAAQ;AACxB,QAAM,YAAuC,CAAC;AAE9C,aAAW,UAAU,UAAU;AAC7B,eAAW,aAAa,OAAO,OAAO;AACpC,YAAM,iBAAiB,EAAE,GAAG,SAAS,QAAQ,UAAU;AACvD,YAAM,SAAS,MAAM;AAAA,QAAa;AAAA,QAAQ,UAAU;AAAA,QAAI;AAAA,QAAS,MAC/D,UAAU,MAAM,gBAAgB,YAAY;AAAA,MAC9C;AACA,gBAAU,KAAK,OAAO,QAAQ;AAE9B,UAAI,OAAO,SAAS,WAAW;AAC7B;AAAA,MACF;AACA,UAAI,OAAO,OAAO,WAAW,SAAS;AACpC,eAAO;AAAA,UACL;AAAA,UACA;AAAA,UACA,SAAS;AAAA,UACT,SAAS,OAAO,OAAO;AAAA,UACvB;AAAA,QACF;AAAA,MACF;AACA,UAAI,OAAO,OAAO,WAAW,WAAW;AACtC,YAAI,OAAO,OAAO,WAAW,QAAW;AACtC,mBAAS,OAAO,OAAO;AACvB,sBAAY,gBAAgB,MAAM;AAAA,QACpC,WAAW,OAAO,OAAO,cAAc,QAAW;AAChD,sBAAY,OAAO,OAAO;AAC1B,mBAAS,mBAAmB,QAAQ,SAAS;AAAA,QAC/C;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,SAAO,EAAE,QAAQ,WAAW,SAAS,OAAO,UAAU;AACxD;AAEA,eAAsB,oBACpB,UACA,SACmC;AACnC,MAAI,aAAa,QAAQ;AACzB,QAAM,YAAuC,CAAC;AAE9C,aAAW,UAAU,UAAU;AAC7B,eAAW,aAAa,OAAO,QAAQ;AACrC,YAAM,iBAAiB,EAAE,GAAG,SAAS,WAAW;AAChD,YAAM,SAAS,MAAM;AAAA,QAAa;AAAA,QAAQ,UAAU;AAAA,QAAI;AAAA,QAAU,MAChE,UAAU,MAAM,gBAAgB,aAAa;AAAA,MAC/C;AACA,gBAAU,KAAK,OAAO,QAAQ;AAE9B,UAAI,OAAO,SAAS,WAAW;AAC7B;AAAA,MACF;AACA,UAAI,OAAO,OAAO,WAAW,SAAS;AACpC,eAAO;AAAA,UACL;AAAA,UACA,SAAS;AAAA,UACT,SAAS,OAAO,OAAO;AAAA,UACvB;AAAA,QACF;AAAA,MACF;AACA,UAAI,OAAO,OAAO,WAAW,WAAW;AACtC,qBAAa,OAAO,OAAO;AAAA,MAC7B;AAAA,IACF;AAAA,EACF;AAEA,SAAO,EAAE,YAAY,SAAS,OAAO,UAAU;AACjD;AAOA,eAAe,aACb,QACA,aACA,UACA,KACuD;AACvD,QAAM,YAAY,KAAK,IAAI;AAC3B,MAAI;AACF,UAAM,SAAU,MAAM,IAAI,KAAM,MAAM;AACtC,WAAO;AAAA,MACL;AAAA,MACA,UAAU,eAAe;AAAA,QACvB;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA,WAAW,KAAK,IAAI,IAAI;AAAA,MAC1B,CAAC;AAAA,IACH;AAAA,EACF,SAAS,OAAO;AACd,UAAM,UAAU,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AACrE,QAAI,OAAO,SAAS,WAAW;AAC7B,aAAO;AAAA,QACL,QAAQ,MAAM;AAAA,QACd,UAAU;AAAA,UACR,UAAU,OAAO;AAAA,UACjB;AAAA,UACA;AAAA,UACA,MAAM,OAAO;AAAA,UACb,QAAQ;AAAA,UACR,SAAS;AAAA,UACT,QAAQ;AAAA,UACR,WAAW,KAAK,IAAI,IAAI;AAAA,QAC1B;AAAA,MACF;AAAA,IACF;AACA,UAAM;AAAA,EACR;AACF;AAEA,SAAS,eAAe,MAMI;AAC1B,QAAM,UAAU,KAAK,OAAO,SAAS,aAAa,KAAK,OAAO,WAAW;AACzE,QAAM,SAAkC;AAAA,IACtC,UAAU,KAAK,OAAO;AAAA,IACtB,aAAa,KAAK;AAAA,IAClB,UAAU,KAAK;AAAA,IACf,MAAM,KAAK,OAAO;AAAA,IAClB,QAAQ,KAAK,OAAO;AAAA,IACpB;AAAA,IACA,WAAW,KAAK;AAAA,EAClB;AACA,MAAI,KAAK,OAAO,WAAW,OAAW,QAAO,SAAS,KAAK,OAAO;AAClE,MAAI,aAAa,KAAK,UAAU,OAAO,KAAK,OAAO,YAAY,UAAU;AACvE,WAAO,UAAU,KAAK,OAAO;AAAA,EAC/B;AACA,MAAI,KAAK,OAAO,aAAa,OAAW,QAAO,WAAW,KAAK,OAAO;AACtE,SAAO;AACT;;;AC/JO,IAAM,aAAa;AAAA,EACxB;AAAA,EACA;AACF;AAmBA,SAAS,UACP,SACmC;AACnC,SAAO,qBAAqB,SAAS,OAAO;AAC9C;AAEA,SAAS,WACP,SACmC;AACnC,SAAO,qBAAqB,SAAS,SAAS;AAChD;AAEA,SAAS,qBACP,SACA,QACmC;AACnC,MAAI,QAAQ,aAAa,SAAS;AAChC,WAAO,qBAAqB;AAAA,MAC1B,IAAI,QAAQ;AAAA,MACZ,MAAM,KAAK,SAAS;AAClB,eAAO;AAAA,UAAkB,IAAI;AAAA,UAAW;AAAA,UAAS;AAAA,UAAQ,CAAC,UACxD,QAAQ,QAAQ,EAAE,WAAW,OAAO,QAAQ,QAAQ,OAAO,CAAC;AAAA,QAC9D;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AACA,SAAO,sBAAsB;AAAA,IAC3B,IAAI,QAAQ;AAAA,IACZ,MAAM,KAAK,SAAS;AAClB,aAAO;AAAA,QAAkB,IAAI;AAAA,QAAY;AAAA,QAAS;AAAA,QAAQ,CAAC,UACzD,QAAQ,QAAQ,EAAE,YAAY,OAAO,QAAQ,QAAQ,OAAO,CAAC;AAAA,MAC/D;AAAA,IACF;AAAA,EACF,CAAC;AACH;AAEA,SAAS,kBACP,MACA,SACA,QACA,SAC4C;AAC5C,QAAM,UAAU,QAAQ,SAAS,KAAK,CAAC,YAAY,mBAAmB,MAAM,OAAO,CAAC;AACpF,MAAI,CAAC,SAAS;AACZ,WAAO,MAAM;AAAA,EACf;AACA,MAAI,WAAW,SAAS;AACtB,WAAO,MAAM;AAAA,MACX,QAAQ,QAAQ;AAAA,MAChB,SAAS,QAAQ;AAAA,IACnB,CAAC;AAAA,EACH;AACA,MAAI,UAAU;AACd,aAAW,WAAW,QAAQ,UAAU;AACtC,cAAU,mBAAmB,SAAS,SAAS,QAAQ,eAAe,YAAY;AAAA,EACpF;AACA,SAAO,QAAQ,OAAO;AACxB;AAEA,SAAS,mBAAmB,MAAc,SAAmC;AAC3E,MAAI,OAAO,YAAY,UAAU;AAC/B,WAAO,KAAK,SAAS,OAAO;AAAA,EAC9B;AACA,UAAQ,YAAY;AACpB,QAAM,UAAU,QAAQ,KAAK,IAAI;AACjC,UAAQ,YAAY;AACpB,SAAO;AACT;AAEA,SAAS,mBAAmB,MAAc,SAA0B,aAA6B;AAC/F,MAAI,OAAO,YAAY,UAAU;AAC/B,WAAO,KAAK,MAAM,OAAO,EAAE,KAAK,WAAW;AAAA,EAC7C;AACA,UAAQ,YAAY;AACpB,QAAM,QAAQ,QAAQ,MAAM,QAAQ,KAAK,EAAE;AAC3C,QAAM,cAAc,MAAM,SAAS,GAAG,IAAI,QAAQ,GAAG,KAAK;AAC1D,SAAO,KAAK,QAAQ,IAAI,OAAO,QAAQ,QAAQ,WAAW,GAAG,WAAW;AAC1E;","names":[]}

package/dist/chunk-BEYBU7VJ.js

@@ -0,0 +1,53 @@
+import {
+  resolveRetryOptions,
+  runWithRetries,
+  throwIfAborted
+} from "./chunk-FN7HLLAZ.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) 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-BEYBU7VJ.js.map

package/dist/chunk-BEYBU7VJ.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 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) request.providerOptions = options.providerOptions;\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":";;;;;;;AAwBA,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,OAAW,SAAQ,kBAAkB,QAAQ;AAC7E,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":[]}