@anvia/core 0.26.0 → 1.0.0-rc.1

package/dist/mcp/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/mcp/result.ts","../../src/mcp/tool.ts","../../src/mcp/connect.ts","../../src/mcp/connections.ts","../../src/mcp/url-safety.ts"],"sourcesContent":["import { isRecord } from \"../internal/record\";\nimport type { McpToolCallContent, McpToolCallResult } from \"./types\";\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): string {\n  if (\"toolResult\" in result) {\n    return serializeMcpValue(result.toolResult);\n  }\n\n  if (result.isError === true) {\n    throw new Error(mcpErrorMessage(result.content));\n  }\n\n  return result.content.map(mapMcpContent).join(\"\");\n}\n\nfunction mcpErrorMessage(content: McpToolCallContent[]): 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: McpToolCallContent): string {\n  if (content.type === \"text\") {\n    return content.text;\n  }\n\n  if (content.type === \"image\") {\n    return `data:${content.mimeType};base64,${content.data}`;\n  }\n\n  if (content.type === \"resource\") {\n    const mimeType =\n      content.resource.mimeType === undefined ? \"\" : `data:${content.resource.mimeType};`;\n    if (\"text\" in content.resource) {\n      return `${mimeType}${content.resource.uri}:${content.resource.text}`;\n    }\n\n    return `${mimeType}${content.resource.uri}:${content.resource.blob}`;\n  }\n\n  throw new Error(`Unsupported MCP tool result content: ${serializeMcpValue(content)}`);\n}\n\nfunction serializeMcpValue(value: unknown): string {\n  if (typeof value === \"string\") {\n    return value;\n  }\n\n  const serialized = JSON.stringify(value);\n  return serialized === undefined ? String(value) : serialized;\n}\n","import type { ToolDefinition } from \"../completion/index\";\nimport type { Tool } from \"../tool/index\";\nimport { createCallToolParams, mapMcpToolResult } from \"./result\";\nimport type { McpClient, McpToolDefinition } from \"./types\";\n\nconst MCP_TOOL_METADATA_KEY = Symbol.for(\"anvia.mcp.tool.metadata\");\n\nexport function createMcpTool(\n  definition: McpToolDefinition,\n  client: McpClient,\n  serverName?: string,\n): Tool {\n  const tool: Tool = {\n    name: definition.name,\n    definition(): ToolDefinition {\n      return {\n        name: definition.name,\n        description: definition.description ?? \"\",\n        parameters: definition.inputSchema,\n      };\n    },\n    async call(args): Promise<string> {\n      const result = await client.callTool(createCallToolParams(definition.name, args));\n      return mapMcpToolResult(result);\n    },\n  };\n  if (serverName !== undefined) {\n    Object.defineProperty(tool, MCP_TOOL_METADATA_KEY, {\n      value: { serverName },\n      enumerable: false,\n    });\n  }\n  return tool;\n}\n","import { createMcpTool } from \"./tool\";\nimport type { McpConnection, McpServer } from \"./types\";\n\nexport async function connectMcp(connection: McpConnection): Promise<McpServer> {\n  const client = await connection.connect();\n  let tools: Awaited<ReturnType<typeof client.listTools>>[\"tools\"];\n  try {\n    ({ tools } = await client.listTools());\n  } catch (error) {\n    try {\n      await client.close();\n    } catch {\n      // Preserve the initialization failure so callers see the actionable MCP error.\n    }\n    throw error;\n  }\n\n  return {\n    name: connection.name,\n    tools: tools.map((tool) => createMcpTool(tool, client, connection.name)),\n    close: () => client.close(),\n  };\n}\n","import { readFileSync } from \"node:fs\";\nimport { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport type {\n  McpClient,\n  McpConnection,\n  McpHttpOptions,\n  McpSseOptions,\n  McpStdioOptions,\n} from \"./types\";\nimport { createSafeMcpFetch, parseAndValidateMcpUrl } from \"./url-safety\";\n\nconst CORE_CLIENT_VERSION = readCorePackageVersion();\n\nexport const mcp = {\n  stdio(options: McpStdioOptions): McpConnection {\n    return {\n      name: options.name,\n      async connect(): Promise<McpClient> {\n        const { name: _name, ...server } = options;\n        const client = createSdkClient();\n        await client.connect(asSdkTransport(new StdioClientTransport(server)));\n        return client as McpClient;\n      },\n    };\n  },\n\n  http(options: McpHttpOptions): McpConnection {\n    return {\n      name: options.name,\n      async connect(): Promise<McpClient> {\n        assertNoCustomMcpFetch(options.transport?.fetch);\n        const url = parseAndValidateMcpUrl(options.url);\n        const client = createSdkClient();\n        await client.connect(\n          asSdkTransport(\n            new StreamableHTTPClientTransport(url, {\n              ...options.transport,\n              fetch: createSafeMcpFetch(),\n            }),\n          ),\n        );\n        return client as McpClient;\n      },\n    };\n  },\n\n  sse(options: McpSseOptions): McpConnection {\n    return {\n      name: options.name,\n      async connect(): Promise<McpClient> {\n        assertNoCustomMcpFetch(options.transport?.fetch);\n        const url = parseAndValidateMcpUrl(options.url);\n        const client = createSdkClient();\n        await client.connect(\n          asSdkTransport(\n            new SSEClientTransport(url, {\n              ...options.transport,\n              fetch: createSafeMcpFetch(),\n            }),\n          ),\n        );\n        return client as McpClient;\n      },\n    };\n  },\n};\n\nfunction assertNoCustomMcpFetch(fetch: unknown): void {\n  if (fetch !== undefined) {\n    throw new Error(\n      \"Custom MCP transport fetch implementations are not allowed because they can bypass SSRF protection\",\n    );\n  }\n}\n\nfunction createSdkClient(): Client {\n  return new Client({\n    name: \"@anvia/core\",\n    version: CORE_CLIENT_VERSION,\n  });\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 asSdkTransport(transport: unknown): Parameters<Client[\"connect\"]>[0] {\n  return transport as Parameters<Client[\"connect\"]>[0];\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  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  validateMcpHostname(parsed.hostname);\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 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"],"mappings":";;;;;AAGO,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,QAAmC;AAClE,MAAI,gBAAgB,QAAQ;AAC1B,WAAO,kBAAkB,OAAO,UAAU;AAAA,EAC5C;AAEA,MAAI,OAAO,YAAY,MAAM;AAC3B,UAAM,IAAI,MAAM,gBAAgB,OAAO,OAAO,CAAC;AAAA,EACjD;AAEA,SAAO,OAAO,QAAQ,IAAI,aAAa,EAAE,KAAK,EAAE;AAClD;AAEA,SAAS,gBAAgB,SAAuC;AAC9D,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,SAAqC;AAC1D,MAAI,QAAQ,SAAS,QAAQ;AAC3B,WAAO,QAAQ;AAAA,EACjB;AAEA,MAAI,QAAQ,SAAS,SAAS;AAC5B,WAAO,QAAQ,QAAQ,QAAQ,WAAW,QAAQ,IAAI;AAAA,EACxD;AAEA,MAAI,QAAQ,SAAS,YAAY;AAC/B,UAAM,WACJ,QAAQ,SAAS,aAAa,SAAY,KAAK,QAAQ,QAAQ,SAAS,QAAQ;AAClF,QAAI,UAAU,QAAQ,UAAU;AAC9B,aAAO,GAAG,QAAQ,GAAG,QAAQ,SAAS,GAAG,IAAI,QAAQ,SAAS,IAAI;AAAA,IACpE;AAEA,WAAO,GAAG,QAAQ,GAAG,QAAQ,SAAS,GAAG,IAAI,QAAQ,SAAS,IAAI;AAAA,EACpE;AAEA,QAAM,IAAI,MAAM,wCAAwC,kBAAkB,OAAO,CAAC,EAAE;AACtF;AAEA,SAAS,kBAAkB,OAAwB;AACjD,MAAI,OAAO,UAAU,UAAU;AAC7B,WAAO;AAAA,EACT;AAEA,QAAM,aAAa,KAAK,UAAU,KAAK;AACvC,SAAO,eAAe,SAAY,OAAO,KAAK,IAAI;AACpD;;;AC/DA,IAAM,wBAAwB,uBAAO,IAAI,yBAAyB;AAE3D,SAAS,cACd,YACA,QACA,YACM;AACN,QAAM,OAAa;AAAA,IACjB,MAAM,WAAW;AAAA,IACjB,aAA6B;AAC3B,aAAO;AAAA,QACL,MAAM,WAAW;AAAA,QACjB,aAAa,WAAW,eAAe;AAAA,QACvC,YAAY,WAAW;AAAA,MACzB;AAAA,IACF;AAAA,IACA,MAAM,KAAK,MAAuB;AAChC,YAAM,SAAS,MAAM,OAAO,SAAS,qBAAqB,WAAW,MAAM,IAAI,CAAC;AAChF,aAAO,iBAAiB,MAAM;AAAA,IAChC;AAAA,EACF;AACA,MAAI,eAAe,QAAW;AAC5B,WAAO,eAAe,MAAM,uBAAuB;AAAA,MACjD,OAAO,EAAE,WAAW;AAAA,MACpB,YAAY;AAAA,IACd,CAAC;AAAA,EACH;AACA,SAAO;AACT;;;AC9BA,eAAsB,WAAW,YAA+C;AAC9E,QAAM,SAAS,MAAM,WAAW,QAAQ;AACxC,MAAI;AACJ,MAAI;AACF,KAAC,EAAE,MAAM,IAAI,MAAM,OAAO,UAAU;AAAA,EACtC,SAAS,OAAO;AACd,QAAI;AACF,YAAM,OAAO,MAAM;AAAA,IACrB,QAAQ;AAAA,IAER;AACA,UAAM;AAAA,EACR;AAEA,SAAO;AAAA,IACL,MAAM,WAAW;AAAA,IACjB,OAAO,MAAM,IAAI,CAAC,SAAS,cAAc,MAAM,QAAQ,WAAW,IAAI,CAAC;AAAA,IACvE,OAAO,MAAM,OAAO,MAAM;AAAA,EAC5B;AACF;;;ACtBA,SAAS,oBAAoB;AAC7B,SAAS,cAAc;AACvB,SAAS,0BAA0B;AACnC,SAAS,4BAA4B;AACrC,SAAS,qCAAqC;;;ACH9C,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,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,sBAAoB,OAAO,QAAQ;AACnC,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;;;ADtMD,IAAM,sBAAsB,uBAAuB;AAE5C,IAAM,MAAM;AAAA,EACjB,MAAM,SAAyC;AAC7C,WAAO;AAAA,MACL,MAAM,QAAQ;AAAA,MACd,MAAM,UAA8B;AAClC,cAAM,EAAE,MAAM,OAAO,GAAG,OAAO,IAAI;AACnC,cAAM,SAAS,gBAAgB;AAC/B,cAAM,OAAO,QAAQ,eAAe,IAAI,qBAAqB,MAAM,CAAC,CAAC;AACrE,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AAAA,EAEA,KAAK,SAAwC;AAC3C,WAAO;AAAA,MACL,MAAM,QAAQ;AAAA,MACd,MAAM,UAA8B;AAClC,+BAAuB,QAAQ,WAAW,KAAK;AAC/C,cAAM,MAAM,uBAAuB,QAAQ,GAAG;AAC9C,cAAM,SAAS,gBAAgB;AAC/B,cAAM,OAAO;AAAA,UACX;AAAA,YACE,IAAI,8BAA8B,KAAK;AAAA,cACrC,GAAG,QAAQ;AAAA,cACX,OAAO,mBAAmB;AAAA,YAC5B,CAAC;AAAA,UACH;AAAA,QACF;AACA,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AAAA,EAEA,IAAI,SAAuC;AACzC,WAAO;AAAA,MACL,MAAM,QAAQ;AAAA,MACd,MAAM,UAA8B;AAClC,+BAAuB,QAAQ,WAAW,KAAK;AAC/C,cAAM,MAAM,uBAAuB,QAAQ,GAAG;AAC9C,cAAM,SAAS,gBAAgB;AAC/B,cAAM,OAAO;AAAA,UACX;AAAA,YACE,IAAI,mBAAmB,KAAK;AAAA,cAC1B,GAAG,QAAQ;AAAA,cACX,OAAO,mBAAmB;AAAA,YAC5B,CAAC;AAAA,UACH;AAAA,QACF;AACA,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,uBAAuB,OAAsB;AACpD,MAAI,UAAU,QAAW;AACvB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,kBAA0B;AACjC,SAAO,IAAI,OAAO;AAAA,IAChB,MAAM;AAAA,IACN,SAAS;AAAA,EACX,CAAC;AACH;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,WAAsD;AAC5E,SAAO;AACT;","names":[]}
+{"version":3,"sources":["../../src/mcp/result.ts","../../src/mcp/tool.ts","../../src/mcp/connect.ts","../../src/mcp/connections.ts","../../src/mcp/url-safety.ts"],"sourcesContent":["import { isRecord } from \"../internal/record\";\nimport type { McpToolCallContent, McpToolCallResult } from \"./types\";\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): string {\n  if (\"toolResult\" in result) {\n    return serializeMcpValue(result.toolResult);\n  }\n\n  if (result.isError === true) {\n    throw new Error(mcpErrorMessage(result.content));\n  }\n\n  return result.content.map(mapMcpContent).join(\"\");\n}\n\nfunction mcpErrorMessage(content: McpToolCallContent[]): 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: McpToolCallContent): string {\n  if (content.type === \"text\") {\n    return content.text;\n  }\n\n  if (content.type === \"image\") {\n    return `data:${content.mimeType};base64,${content.data}`;\n  }\n\n  if (content.type === \"resource\") {\n    const mimeType =\n      content.resource.mimeType === undefined ? \"\" : `data:${content.resource.mimeType};`;\n    if (\"text\" in content.resource) {\n      return `${mimeType}${content.resource.uri}:${content.resource.text}`;\n    }\n\n    return `${mimeType}${content.resource.uri}:${content.resource.blob}`;\n  }\n\n  throw new Error(`Unsupported MCP tool result content: ${serializeMcpValue(content)}`);\n}\n\nfunction serializeMcpValue(value: unknown): string {\n  if (typeof value === \"string\") {\n    return value;\n  }\n\n  const serialized = JSON.stringify(value);\n  return serialized === undefined ? String(value) : serialized;\n}\n","import type { ToolDefinition } from \"../completion/index\";\nimport type { Tool } from \"../tool/index\";\nimport { createCallToolParams, mapMcpToolResult } from \"./result\";\nimport type { McpClient, McpToolDefinition } from \"./types\";\n\nconst MCP_TOOL_METADATA_KEY = Symbol.for(\"anvia.mcp.tool.metadata\");\n\nexport function createMcpTool(\n  definition: McpToolDefinition,\n  client: McpClient,\n  serverName?: string,\n): Tool {\n  const tool: Tool = {\n    name: definition.name,\n    definition(): ToolDefinition {\n      return {\n        name: definition.name,\n        description: definition.description ?? \"\",\n        parameters: definition.inputSchema,\n      };\n    },\n    async call(args): Promise<string> {\n      const result = await client.callTool(createCallToolParams(definition.name, args));\n      return mapMcpToolResult(result);\n    },\n  };\n  if (serverName !== undefined) {\n    Object.defineProperty(tool, MCP_TOOL_METADATA_KEY, {\n      value: { serverName },\n      enumerable: false,\n    });\n  }\n  return tool;\n}\n","import { createMcpTool } from \"./tool\";\nimport type { McpConnection, McpServer } from \"./types\";\n\nexport async function connectMcp(connection: McpConnection): Promise<McpServer> {\n  const client = await connection.connect();\n  let tools: Awaited<ReturnType<typeof client.listTools>>[\"tools\"];\n  try {\n    ({ tools } = await client.listTools());\n  } catch (error) {\n    try {\n      await client.close();\n    } catch {\n      // Preserve the initialization failure so callers see the actionable MCP error.\n    }\n    throw error;\n  }\n\n  return {\n    name: connection.name,\n    tools: tools.map((tool) => createMcpTool(tool, client, connection.name)),\n    close: () => client.close(),\n  };\n}\n","import { readFileSync } from \"node:fs\";\nimport { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport type {\n  McpClient,\n  McpConnection,\n  McpHttpOptions,\n  McpSseOptions,\n  McpStdioOptions,\n} from \"./types\";\nimport { createSafeMcpFetch, parseAndValidateMcpUrl } from \"./url-safety\";\n\nconst CORE_CLIENT_VERSION = readCorePackageVersion();\n\nexport const mcp = {\n  stdio(options: McpStdioOptions): McpConnection {\n    return {\n      name: options.name,\n      async connect(): Promise<McpClient> {\n        const { name: _name, ...server } = options;\n        const client = createSdkClient();\n        await client.connect(asSdkTransport(new StdioClientTransport(server)));\n        return client as McpClient;\n      },\n    };\n  },\n\n  http(options: McpHttpOptions): McpConnection {\n    return {\n      name: options.name,\n      async connect(): Promise<McpClient> {\n        assertNoCustomMcpFetch(options.transport?.fetch);\n        const url = parseAndValidateMcpUrl(options.url);\n        const client = createSdkClient();\n        await client.connect(\n          asSdkTransport(\n            new StreamableHTTPClientTransport(url, {\n              ...options.transport,\n              fetch: createSafeMcpFetch(),\n            }),\n          ),\n        );\n        return client as McpClient;\n      },\n    };\n  },\n\n  sse(options: McpSseOptions): McpConnection {\n    return {\n      name: options.name,\n      async connect(): Promise<McpClient> {\n        assertNoCustomMcpFetch(options.transport?.fetch);\n        const url = parseAndValidateMcpUrl(options.url);\n        const client = createSdkClient();\n        await client.connect(\n          asSdkTransport(\n            new SSEClientTransport(url, {\n              ...options.transport,\n              fetch: createSafeMcpFetch(),\n            }),\n          ),\n        );\n        return client as McpClient;\n      },\n    };\n  },\n};\n\nfunction assertNoCustomMcpFetch(fetch: unknown): void {\n  if (fetch !== undefined) {\n    throw new Error(\n      \"Custom MCP transport fetch implementations are not allowed because they can bypass SSRF protection\",\n    );\n  }\n}\n\nfunction createSdkClient(): Client {\n  return new Client({\n    name: \"@anvia/core\",\n    version: CORE_CLIENT_VERSION,\n  });\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 asSdkTransport(transport: unknown): Parameters<Client[\"connect\"]>[0] {\n  return transport as Parameters<Client[\"connect\"]>[0];\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  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  validateMcpHostname(parsed.hostname);\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"],"mappings":";;;;;AAGO,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,QAAmC;AAClE,MAAI,gBAAgB,QAAQ;AAC1B,WAAO,kBAAkB,OAAO,UAAU;AAAA,EAC5C;AAEA,MAAI,OAAO,YAAY,MAAM;AAC3B,UAAM,IAAI,MAAM,gBAAgB,OAAO,OAAO,CAAC;AAAA,EACjD;AAEA,SAAO,OAAO,QAAQ,IAAI,aAAa,EAAE,KAAK,EAAE;AAClD;AAEA,SAAS,gBAAgB,SAAuC;AAC9D,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,SAAqC;AAC1D,MAAI,QAAQ,SAAS,QAAQ;AAC3B,WAAO,QAAQ;AAAA,EACjB;AAEA,MAAI,QAAQ,SAAS,SAAS;AAC5B,WAAO,QAAQ,QAAQ,QAAQ,WAAW,QAAQ,IAAI;AAAA,EACxD;AAEA,MAAI,QAAQ,SAAS,YAAY;AAC/B,UAAM,WACJ,QAAQ,SAAS,aAAa,SAAY,KAAK,QAAQ,QAAQ,SAAS,QAAQ;AAClF,QAAI,UAAU,QAAQ,UAAU;AAC9B,aAAO,GAAG,QAAQ,GAAG,QAAQ,SAAS,GAAG,IAAI,QAAQ,SAAS,IAAI;AAAA,IACpE;AAEA,WAAO,GAAG,QAAQ,GAAG,QAAQ,SAAS,GAAG,IAAI,QAAQ,SAAS,IAAI;AAAA,EACpE;AAEA,QAAM,IAAI,MAAM,wCAAwC,kBAAkB,OAAO,CAAC,EAAE;AACtF;AAEA,SAAS,kBAAkB,OAAwB;AACjD,MAAI,OAAO,UAAU,UAAU;AAC7B,WAAO;AAAA,EACT;AAEA,QAAM,aAAa,KAAK,UAAU,KAAK;AACvC,SAAO,eAAe,SAAY,OAAO,KAAK,IAAI;AACpD;;;AC/DA,IAAM,wBAAwB,uBAAO,IAAI,yBAAyB;AAE3D,SAAS,cACd,YACA,QACA,YACM;AACN,QAAM,OAAa;AAAA,IACjB,MAAM,WAAW;AAAA,IACjB,aAA6B;AAC3B,aAAO;AAAA,QACL,MAAM,WAAW;AAAA,QACjB,aAAa,WAAW,eAAe;AAAA,QACvC,YAAY,WAAW;AAAA,MACzB;AAAA,IACF;AAAA,IACA,MAAM,KAAK,MAAuB;AAChC,YAAM,SAAS,MAAM,OAAO,SAAS,qBAAqB,WAAW,MAAM,IAAI,CAAC;AAChF,aAAO,iBAAiB,MAAM;AAAA,IAChC;AAAA,EACF;AACA,MAAI,eAAe,QAAW;AAC5B,WAAO,eAAe,MAAM,uBAAuB;AAAA,MACjD,OAAO,EAAE,WAAW;AAAA,MACpB,YAAY;AAAA,IACd,CAAC;AAAA,EACH;AACA,SAAO;AACT;;;AC9BA,eAAsB,WAAW,YAA+C;AAC9E,QAAM,SAAS,MAAM,WAAW,QAAQ;AACxC,MAAI;AACJ,MAAI;AACF,KAAC,EAAE,MAAM,IAAI,MAAM,OAAO,UAAU;AAAA,EACtC,SAAS,OAAO;AACd,QAAI;AACF,YAAM,OAAO,MAAM;AAAA,IACrB,QAAQ;AAAA,IAER;AACA,UAAM;AAAA,EACR;AAEA,SAAO;AAAA,IACL,MAAM,WAAW;AAAA,IACjB,OAAO,MAAM,IAAI,CAAC,SAAS,cAAc,MAAM,QAAQ,WAAW,IAAI,CAAC;AAAA,IACvE,OAAO,MAAM,OAAO,MAAM;AAAA,EAC5B;AACF;;;ACtBA,SAAS,oBAAoB;AAC7B,SAAS,cAAc;AACvB,SAAS,0BAA0B;AACnC,SAAS,4BAA4B;AACrC,SAAS,qCAAqC;;;ACH9C,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,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,sBAAoB,OAAO,QAAQ;AACnC,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;;;ADtMD,IAAM,sBAAsB,uBAAuB;AAE5C,IAAM,MAAM;AAAA,EACjB,MAAM,SAAyC;AAC7C,WAAO;AAAA,MACL,MAAM,QAAQ;AAAA,MACd,MAAM,UAA8B;AAClC,cAAM,EAAE,MAAM,OAAO,GAAG,OAAO,IAAI;AACnC,cAAM,SAAS,gBAAgB;AAC/B,cAAM,OAAO,QAAQ,eAAe,IAAI,qBAAqB,MAAM,CAAC,CAAC;AACrE,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AAAA,EAEA,KAAK,SAAwC;AAC3C,WAAO;AAAA,MACL,MAAM,QAAQ;AAAA,MACd,MAAM,UAA8B;AAClC,+BAAuB,QAAQ,WAAW,KAAK;AAC/C,cAAM,MAAM,uBAAuB,QAAQ,GAAG;AAC9C,cAAM,SAAS,gBAAgB;AAC/B,cAAM,OAAO;AAAA,UACX;AAAA,YACE,IAAI,8BAA8B,KAAK;AAAA,cACrC,GAAG,QAAQ;AAAA,cACX,OAAO,mBAAmB;AAAA,YAC5B,CAAC;AAAA,UACH;AAAA,QACF;AACA,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AAAA,EAEA,IAAI,SAAuC;AACzC,WAAO;AAAA,MACL,MAAM,QAAQ;AAAA,MACd,MAAM,UAA8B;AAClC,+BAAuB,QAAQ,WAAW,KAAK;AAC/C,cAAM,MAAM,uBAAuB,QAAQ,GAAG;AAC9C,cAAM,SAAS,gBAAgB;AAC/B,cAAM,OAAO;AAAA,UACX;AAAA,YACE,IAAI,mBAAmB,KAAK;AAAA,cAC1B,GAAG,QAAQ;AAAA,cACX,OAAO,mBAAmB;AAAA,YAC5B,CAAC;AAAA,UACH;AAAA,QACF;AACA,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,uBAAuB,OAAsB;AACpD,MAAI,UAAU,QAAW;AACvB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACF;AAEA,SAAS,kBAA0B;AACjC,SAAO,IAAI,OAAO;AAAA,IAChB,MAAM;AAAA,IACN,SAAS;AAAA,EACX,CAAC;AACH;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,WAAsD;AAC5E,SAAO;AACT;","names":[]}

package/dist/memory/index.d.ts

@@ -1,7 +1,7 @@
-export { M as MemoryCompactionConflictError, a as MemoryCompactionError, c as createSummaryMemoryCompactor, i as isMemoryCompactionSummary } from '../errors-2moLDuuN.js';
-import { n as MemoryOptions, o as ResolvedMemoryOptions } from '../types-BrLbKnEz.js';
-export { p as MemoryAppendInput, M as MemoryCompactionCommitInput, a as MemoryCompactionCommitResult, b as MemoryCompactionOptions, c as MemoryCompactionSnapshot, d as MemoryCompactionStore, e as MemoryCompactor, f as MemoryCompactorInput, g as MemoryCompactorResult, q as MemoryContext, h as MemoryConversation, i as MemoryConversationListOptions, j as MemoryConversationMessage, k as MemoryConversationSummary, r as MemoryErrorInput, l as MemoryInspector, s as MemoryRegistration, t as MemorySavePolicy, m as MemoryStore, R as ResolvedMemoryCompactionOptions, u as SessionOptions, S as SummaryMemoryCompactorOptions } from '../types-BrLbKnEz.js';
-import '../types-G12i56s2.js';
+export { M as MemoryCompactionConflictError, a as MemoryCompactionError, c as createSummaryMemoryCompactor, i as isMemoryCompactionSummary } from '../errors-CYKUcU89.js';
+import { n as MemoryOptions, o as ResolvedMemoryOptions } from '../types-BBEGKB9m.js';
+export { p as MemoryAppendInput, M as MemoryCompactionCommitInput, a as MemoryCompactionCommitResult, b as MemoryCompactionOptions, c as MemoryCompactionSnapshot, d as MemoryCompactionStore, e as MemoryCompactor, f as MemoryCompactorInput, g as MemoryCompactorResult, q as MemoryContext, h as MemoryConversation, i as MemoryConversationListOptions, j as MemoryConversationMessage, k as MemoryConversationSummary, r as MemoryErrorInput, l as MemoryInspector, s as MemoryRegistration, t as MemorySavePolicy, m as MemoryStore, R as ResolvedMemoryCompactionOptions, u as SessionOptions, S as SummaryMemoryCompactorOptions } from '../types-BBEGKB9m.js';
+import '../types-B3A4YxaA.js';
 
 declare function resolveMemoryOptions(options?: MemoryOptions): ResolvedMemoryOptions;
 

package/dist/memory/index.js

@@ -1,16 +1,15 @@
 import "../chunk-XWUC7CIT.js";
-import {
-  resolveMemoryOptions
-} from "../chunk-UQX6EXWG.js";
 import {
   MemoryCompactionConflictError,
   MemoryCompactionError,
   createSummaryMemoryCompactor,
-  isMemoryCompactionSummary
-} from "../chunk-MRQLEK2B.js";
-import "../chunk-MC3CCKEB.js";
+  isMemoryCompactionSummary,
+  resolveMemoryOptions
+} from "../chunk-5XKDW36Z.js";
+import "../chunk-M3IOWB4Y.js";
+import "../chunk-RFDVKCIE.js";
 import "../chunk-ADH7NNCS.js";
-import "../chunk-WQKHFADH.js";
+import "../chunk-A3UBYTJT.js";
 export {
   MemoryCompactionConflictError,
   MemoryCompactionError,

package/dist/middleware-KynLg_RW.d.ts

@@ -0,0 +1,56 @@
+import { f as CompletionRequest, g as CompletionResponse, l as JsonValue, u as ToolResultContent } from './types-B3A4YxaA.js';
+import { M as MaybePromise } from './type-utils-CtHVDRn_.js';
+
+type CompletionRequestMiddlewareArgs = {
+    turn: number;
+    request: CompletionRequest;
+    originalRequest: CompletionRequest;
+};
+type CompletionRequestMiddlewareResult = {
+    request: CompletionRequest;
+} | undefined;
+type CompletionResponseMiddlewareArgs<RawResponse = unknown> = {
+    turn: number;
+    request: CompletionRequest;
+    response: CompletionResponse<RawResponse>;
+    originalResponse: CompletionResponse<RawResponse>;
+};
+type CompletionResponseMiddlewareResult<RawResponse = unknown> = {
+    response: CompletionResponse<RawResponse>;
+} | undefined;
+type ToolInputMiddlewareArgs = {
+    toolName: string;
+    args: string;
+    originalArgs: string;
+    turn: number;
+    toolCallId?: string | undefined;
+    internalCallId: string;
+};
+type ToolInputMiddlewareResult = {
+    args: JsonValue | string;
+} | undefined;
+type ToolResultMiddlewareArgs = {
+    toolName: string;
+    args: string;
+    result: string;
+    originalResult: string;
+    structuredResult?: ToolResultContent[] | undefined;
+    originalStructuredResult?: ToolResultContent[] | undefined;
+    turn: number;
+    toolCallId?: string | undefined;
+    internalCallId: string;
+};
+type ToolOutputMiddlewareArgs = ToolResultMiddlewareArgs;
+type ToolOutputMiddlewareResult = string | {
+    result?: string | undefined;
+    structuredResult?: ToolResultContent[] | undefined;
+} | undefined;
+interface AgentMiddleware<RawResponse = unknown> {
+    onCompletionRequest?(args: CompletionRequestMiddlewareArgs): MaybePromise<CompletionRequestMiddlewareResult>;
+    onCompletionResponse?(args: CompletionResponseMiddlewareArgs<RawResponse>): MaybePromise<CompletionResponseMiddlewareResult<RawResponse>>;
+    onToolInput?(args: ToolInputMiddlewareArgs): MaybePromise<ToolInputMiddlewareResult>;
+    onToolOutput?(args: ToolOutputMiddlewareArgs): MaybePromise<ToolOutputMiddlewareResult>;
+}
+declare function createMiddleware<RawResponse = unknown>(middleware: AgentMiddleware<RawResponse>): AgentMiddleware<RawResponse>;
+
+export { type AgentMiddleware as A, type CompletionRequestMiddlewareArgs as C, type ToolInputMiddlewareArgs as T, type CompletionRequestMiddlewareResult as a, type CompletionResponseMiddlewareArgs as b, type CompletionResponseMiddlewareResult as c, type ToolInputMiddlewareResult as d, type ToolOutputMiddlewareArgs as e, type ToolOutputMiddlewareResult as f, type ToolResultMiddlewareArgs as g, createMiddleware as h };

package/dist/observability/index.d.ts

@@ -1,4 +1,7 @@
-export { m as AgentGenerationEndArgs, n as AgentGenerationErrorArgs, o as AgentGenerationModelInfo, p as AgentGenerationObserver, q as AgentGenerationStartArgs, r as AgentGenerationUpdateArgs, i as AgentObserver, k as AgentObserverRegistration, s as AgentRunEndArgs, t as AgentRunErrorArgs, u as AgentRunEventArgs, v as AgentRunObserver, w as AgentRunPromptRef, x as AgentRunStartArgs, y as AgentToolEndArgs, z as AgentToolErrorArgs, B as AgentToolObserver, C as AgentToolStartArgs, D as AgentToolStreamEventArgs, E as AgentTraceInfo, j as AgentTraceOptions, O as ObserveOptions, F as createObserver } from '../index-BxSbmOMr.js';
-import '../types-G12i56s2.js';
+export { C as AgentGenerationEndArgs, D as AgentGenerationErrorArgs, E as AgentGenerationModelInfo, F as AgentGenerationObserver, G as AgentGenerationStartArgs, H as AgentGenerationUpdateArgs, z as AgentObserver, B as AgentObserverRegistration, I as AgentRunEndArgs, J as AgentRunErrorArgs, K as AgentRunEventArgs, L as AgentRunObserver, M as AgentRunPromptRef, N as AgentRunStartArgs, P as AgentToolEndArgs, Q as AgentToolErrorArgs, R as AgentToolObserver, S as AgentToolStartArgs, T as AgentToolStreamEventArgs, U as AgentTraceInfo, V as AgentTraceOptions, O as ObserveOptions, W as createObserver } from '../index-DxB74ybJ.js';
+import '../retry-D3Ruy-ba.js';
+import '../types-B3A4YxaA.js';
 import '../guardrails/index.js';
-import '../tool-Cohdv5GC.js';
+import '../type-utils-CtHVDRn_.js';
+import '../middleware-KynLg_RW.js';
+import '../tool-C_8GudXn.js';

package/dist/observability/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/observability/types.ts"],"sourcesContent":["import type {\n  CompletionModelCapabilities,\n  CompletionRequest,\n  CompletionResponse,\n  CompletionSource,\n  JsonObject,\n  JsonValue,\n  Message,\n  ProviderToolCall,\n  ToolCall,\n  ToolDefinition,\n  ToolResultContent,\n  Usage,\n} from \"../completion\";\nimport type { AgentDeltaEvent } from \"../request/types\";\nimport type { ToolCallStreamEvent } from \"../tool\";\n\nexport type AgentTraceInfo = {\n  traceId?: string | undefined;\n  observationId?: string | undefined;\n};\n\nexport type AgentTraceOptions = {\n  name?: string | undefined;\n  userId?: string | undefined;\n  sessionId?: string | undefined;\n  metadata?: Record<string, unknown> | undefined;\n  tags?: string[] | undefined;\n  version?: string | undefined;\n  traceId?: string | undefined;\n  failOnObserverError?: boolean | undefined;\n  promptRef?: AgentRunPromptRef | undefined;\n};\n\nexport type AgentRunPromptRef = {\n  name: string;\n  version?: number | undefined;\n};\n\nexport type AgentRunStartArgs = {\n  agentName?: string | undefined;\n  agentDescription?: string | undefined;\n  instructions?: string | undefined;\n  trace?: AgentTraceOptions | undefined;\n  prompt: Message;\n  promptRef?: AgentRunPromptRef | undefined;\n  history: Message[];\n  maxTurns: number;\n};\n\nexport type AgentRunEndArgs = {\n  output: string;\n  usage: Usage;\n  messages: Message[];\n  sources?: CompletionSource[] | undefined;\n  providerToolCalls?: ProviderToolCall[] | undefined;\n};\n\nexport type AgentRunErrorArgs = {\n  error: unknown;\n  usage: Usage;\n  messages: Message[];\n};\n\nexport type AgentGenerationModelInfo = {\n  provider: string;\n  defaultModel: string;\n  capabilities?: CompletionModelCapabilities | undefined;\n};\n\nexport type AgentGenerationStartArgs = {\n  turn: number;\n  request: CompletionRequest;\n  providerRequest?: JsonObject | undefined;\n  modelInfo?: AgentGenerationModelInfo | undefined;\n};\n\nexport type AgentGenerationEndArgs<RawResponse = unknown> = {\n  turn: number;\n  response: CompletionResponse<RawResponse>;\n  firstDeltaMs?: number | undefined;\n};\n\nexport type AgentGenerationErrorArgs = {\n  turn: number;\n  error: unknown;\n};\n\nexport type AgentGenerationUpdateArgs = {\n  turn: number;\n  delta: AgentDeltaEvent;\n};\n\nexport type AgentRunEventArgs = {\n  name: string;\n  attributes?: Record<string, JsonValue | undefined> | undefined;\n  level?: \"DEFAULT\" | \"WARNING\" | \"ERROR\" | undefined;\n  timestamp?: Date | string | undefined;\n};\n\nexport type AgentToolStartArgs = {\n  turn: number;\n  toolCall: ToolCall;\n  toolName: string;\n  args: string;\n  internalCallId: string;\n  toolCallId?: string | undefined;\n  toolDefinition?: ToolDefinition | undefined;\n  toolMetadata?: JsonObject | undefined;\n};\n\nexport type AgentToolEndArgs = AgentToolStartArgs & {\n  result: string;\n  structuredResult?: ToolResultContent[] | undefined;\n  skipped: boolean;\n};\n\nexport type AgentToolErrorArgs = AgentToolStartArgs & {\n  error: unknown;\n};\n\nexport type AgentToolStreamEventArgs = AgentToolStartArgs & {\n  event: ToolCallStreamEvent;\n};\n\nexport interface AgentGenerationObserver {\n  end(args: AgentGenerationEndArgs): void | Promise<void>;\n  error?(args: AgentGenerationErrorArgs): void | Promise<void>;\n  update?(args: AgentGenerationUpdateArgs): void | Promise<void>;\n}\n\nexport interface AgentToolObserver {\n  streamEvent?(args: AgentToolStreamEventArgs): void | Promise<void>;\n  end(args: AgentToolEndArgs): void | Promise<void>;\n  error?(args: AgentToolErrorArgs): void | Promise<void>;\n}\n\nexport interface AgentRunObserver {\n  readonly trace?: AgentTraceInfo | undefined;\n  startGeneration?(\n    args: AgentGenerationStartArgs,\n  ): AgentGenerationObserver | undefined | Promise<AgentGenerationObserver | undefined>;\n  startTool?(\n    args: AgentToolStartArgs,\n  ): AgentToolObserver | undefined | Promise<AgentToolObserver | undefined>;\n  end(args: AgentRunEndArgs): void | Promise<void>;\n  error?(args: AgentRunErrorArgs): void | Promise<void>;\n  event?(args: AgentRunEventArgs): void | Promise<void>;\n}\n\nexport interface AgentObserver {\n  startRun(\n    args: AgentRunStartArgs,\n  ): AgentRunObserver | undefined | Promise<AgentRunObserver | undefined>;\n  flush?(): void | Promise<void>;\n  shutdown?(): void | Promise<void>;\n}\n\nexport type AgentObserverRegistration = {\n  observer: AgentObserver;\n  failOnObserverError?: boolean | undefined;\n};\n\nexport type ObserveOptions = {\n  failOnObserverError?: boolean | undefined;\n};\n\nexport function createObserver(observer: AgentObserver): AgentObserver {\n  return observer;\n}\n"],"mappings":";AAuKO,SAAS,eAAe,UAAwC;AACrE,SAAO;AACT;","names":[]}
+{"version":3,"sources":["../../src/observability/types.ts"],"sourcesContent":["import type { AgentDeltaEvent } from \"../agent/run-types\";\nimport type {\n  CompletionModelCapabilities,\n  CompletionRequest,\n  CompletionResponse,\n  CompletionSource,\n  JsonObject,\n  JsonValue,\n  Message,\n  ProviderToolCall,\n  ToolCall,\n  ToolDefinition,\n  ToolResultContent,\n  Usage,\n} from \"../completion\";\nimport type { DeepReadonly } from \"../internal/type-utils\";\nimport type { ToolCallStreamEvent } from \"../tool\";\n\nexport type AgentTraceInfo = {\n  readonly traceId?: string | undefined;\n  readonly observationId?: string | undefined;\n};\n\nexport type AgentTraceOptions = {\n  readonly name?: string | undefined;\n  readonly userId?: string | undefined;\n  readonly sessionId?: string | undefined;\n  readonly metadata?: Readonly<Record<string, unknown>> | undefined;\n  readonly tags?: readonly string[] | undefined;\n  readonly version?: string | undefined;\n  readonly traceId?: string | undefined;\n  readonly failOnObserverError?: boolean | undefined;\n  readonly promptRef?: AgentRunPromptRef | undefined;\n};\n\nexport type AgentRunPromptRef = {\n  readonly name: string;\n  readonly version?: number | undefined;\n};\n\nexport type AgentRunStartArgs = {\n  readonly runId: string;\n  readonly agentName?: string | undefined;\n  readonly agentDescription?: string | undefined;\n  readonly instructions?: string | undefined;\n  readonly trace?: DeepReadonly<AgentTraceOptions> | undefined;\n  readonly prompt: DeepReadonly<Message>;\n  readonly promptRef?: DeepReadonly<AgentRunPromptRef> | undefined;\n  readonly history: readonly DeepReadonly<Message>[];\n  readonly maxTurns: number;\n};\n\nexport type AgentRunEndArgs = {\n  readonly output: string;\n  readonly usage: DeepReadonly<Usage>;\n  readonly messages: readonly DeepReadonly<Message>[];\n  readonly sources?: readonly DeepReadonly<CompletionSource>[] | undefined;\n  readonly providerToolCalls?: readonly DeepReadonly<ProviderToolCall>[] | undefined;\n};\n\nexport type AgentRunErrorArgs = {\n  readonly error: unknown;\n  readonly usage: DeepReadonly<Usage>;\n  readonly messages: readonly DeepReadonly<Message>[];\n};\n\nexport type AgentGenerationModelInfo = {\n  readonly provider: string;\n  readonly defaultModel: string;\n  readonly capabilities?: DeepReadonly<CompletionModelCapabilities> | undefined;\n};\n\nexport type AgentGenerationStartArgs = {\n  readonly turn: number;\n  readonly request: DeepReadonly<CompletionRequest>;\n  readonly providerRequest?: DeepReadonly<JsonObject> | undefined;\n  readonly modelInfo?: DeepReadonly<AgentGenerationModelInfo> | undefined;\n};\n\nexport type AgentGenerationEndArgs<RawResponse = unknown> = {\n  readonly turn: number;\n  readonly response: DeepReadonly<CompletionResponse<RawResponse>>;\n  readonly firstDeltaMs?: number | undefined;\n};\n\nexport type AgentGenerationErrorArgs = {\n  readonly turn: number;\n  readonly error: unknown;\n};\n\nexport type AgentGenerationUpdateArgs = {\n  readonly turn: number;\n  readonly delta: DeepReadonly<AgentDeltaEvent>;\n};\n\nexport type AgentRunEventArgs = {\n  readonly name: string;\n  readonly attributes?: Readonly<Record<string, DeepReadonly<JsonValue> | undefined>> | undefined;\n  readonly level?: \"DEFAULT\" | \"WARNING\" | \"ERROR\" | undefined;\n  readonly timestamp?: Date | string | undefined;\n};\n\nexport type AgentToolStartArgs = {\n  readonly turn: number;\n  readonly toolCall: DeepReadonly<ToolCall>;\n  readonly toolName: string;\n  readonly args: string;\n  readonly internalCallId: string;\n  readonly toolCallId?: string | undefined;\n  readonly toolDefinition?: DeepReadonly<ToolDefinition> | undefined;\n  readonly toolMetadata?: DeepReadonly<JsonObject> | undefined;\n};\n\nexport type AgentToolEndArgs = AgentToolStartArgs & {\n  readonly result: string;\n  readonly structuredResult?: readonly DeepReadonly<ToolResultContent>[] | undefined;\n  readonly skipped: boolean;\n};\n\nexport type AgentToolErrorArgs = AgentToolStartArgs & {\n  readonly error: unknown;\n};\n\nexport type AgentToolStreamEventArgs = AgentToolStartArgs & {\n  readonly event: DeepReadonly<ToolCallStreamEvent>;\n};\n\nexport interface AgentGenerationObserver {\n  end(args: AgentGenerationEndArgs): void | Promise<void>;\n  error?(args: AgentGenerationErrorArgs): void | Promise<void>;\n  update?(args: AgentGenerationUpdateArgs): void | Promise<void>;\n}\n\nexport interface AgentToolObserver {\n  streamEvent?(args: AgentToolStreamEventArgs): void | Promise<void>;\n  end(args: AgentToolEndArgs): void | Promise<void>;\n  error?(args: AgentToolErrorArgs): void | Promise<void>;\n}\n\nexport interface AgentRunObserver {\n  readonly trace?: AgentTraceInfo | undefined;\n  startGeneration?(\n    args: AgentGenerationStartArgs,\n  ): AgentGenerationObserver | undefined | Promise<AgentGenerationObserver | undefined>;\n  startTool?(\n    args: AgentToolStartArgs,\n  ): AgentToolObserver | undefined | Promise<AgentToolObserver | undefined>;\n  end(args: AgentRunEndArgs): void | Promise<void>;\n  error?(args: AgentRunErrorArgs): void | Promise<void>;\n  event?(args: AgentRunEventArgs): void | Promise<void>;\n}\n\nexport interface AgentObserver {\n  startRun(\n    args: AgentRunStartArgs,\n  ): AgentRunObserver | undefined | Promise<AgentRunObserver | undefined>;\n  flush?(): void | Promise<void>;\n  shutdown?(): void | Promise<void>;\n}\n\nexport type AgentObserverRegistration = {\n  readonly observer: AgentObserver;\n  readonly failOnObserverError?: boolean | undefined;\n};\n\nexport type ObserveOptions = {\n  readonly failOnObserverError?: boolean | undefined;\n};\n\nexport function createObserver(observer: AgentObserver): AgentObserver {\n  return observer;\n}\n"],"mappings":";AAyKO,SAAS,eAAe,UAAwC;AACrE,SAAO;AACT;","names":[]}

package/dist/pipeline/index.d.ts

@@ -1,16 +1,23 @@
-import { z } from 'zod';
-import { A as Agent } from '../agent-x3YTyv5i.js';
-import { J as JsonObject, C as CompletionModel } from '../types-G12i56s2.js';
+import { A as Agent } from '../agent-DGYY_onS.js';
+import { J as JsonObject, C as CompletionModel } from '../types-B3A4YxaA.js';
 import { Extractor } from '../extractor/index.js';
+import { z } from 'zod';
 import '../guardrails/index.js';
-import '../types-D9JpeAPF.js';
-import '../types-BrLbKnEz.js';
-import '../index-BxSbmOMr.js';
-import '../tool-Cohdv5GC.js';
-import '../middleware-Dp3d_pxi.js';
-import '../types-BCTRUGex.js';
-import '../types-O-abTIin.js';
+import '../type-utils-CtHVDRn_.js';
+import '../types-BBEGKB9m.js';
+import '../index-DxB74ybJ.js';
+import '../retry-D3Ruy-ba.js';
+import '../middleware-KynLg_RW.js';
+import '../tool-C_8GudXn.js';
 import '../zod-schema-C7F4clpm.js';
+import '../types-DulRCYdB.js';
+import '@modelcontextprotocol/sdk/client/sse.js';
+import '@modelcontextprotocol/sdk/client/stdio.js';
+import '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import '../types-DhQfwiZn.js';
+import '../dynamic-tools-i9woysFa.js';
+import '../types-p55Jc-sr.js';
+import '../types-DT3nEemY.js';
 
 /** Minimal interface for anything that can run as a pipeline stage. */
 interface PipelineOp<Input = unknown, Output = unknown> {
@@ -30,6 +37,13 @@ type PipelineMetadata = {
     description?: string | undefined;
     metadata?: JsonObject | undefined;
 };
+type PipelineOptions<Input, Parsed = Input> = {
+    id: string;
+    inputSchema: z.ZodType<Parsed, Input>;
+    name?: string | undefined;
+    description?: string | undefined;
+    metadata?: JsonObject | undefined;
+};
 type PipelineStageMetadata = {
     id?: string | undefined;
     name?: string | undefined;
@@ -78,55 +92,34 @@ type PipelineRunObserver = {
 type PipelineRunOptions = {
     observer?: PipelineRunObserver | undefined;
 };
-type PipelineRunContext = {
-    observer?: PipelineRunObserver | undefined;
-};
-type PipelineExecutor<Input, Output> = (input: Input, context: PipelineRunContext) => Output | Promise<Output>;
-type PipelineBuilderState = {
-    graph: PipelineGraph;
-    terminalNodeId: string;
-    terminalNodeIds: string[];
-    nextNodeIndex: number;
-    nextEdgeIndex: number;
-};
 
-/** Runnable pipeline returned by `PipelineBuilder.build()`. */
-declare class Pipeline<Input, Output> implements PipelineOp<Input, Awaited<Output>> {
-    private readonly executor;
-    private readonly pipelineGraph;
+/** Immutable, typed pipeline that can be composed and run directly. */
+declare class Pipeline<Input, Output = Input> implements PipelineOp<Input, Awaited<Output>> {
     readonly id: string;
     readonly name: string | undefined;
     readonly description: string | undefined;
     readonly metadata: JsonObject | undefined;
-    constructor(executor: PipelineExecutor<Input, Output>, pipelineGraph?: PipelineGraph);
-    /** Run one input through the built pipeline and return the final stage output. */
-    run(input: Input, options?: PipelineRunOptions): Promise<Awaited<Output>>;
-    /** Run many inputs through the same pipeline with bounded concurrency. */
-    batch<I extends Iterable<Input>>(inputs: I, options: PipelineBatchOptions): Promise<Array<Awaited<Output>>>;
-    graph(): PipelineGraph;
-}
-
-/** Builds a typed pipeline from an original input type to an inferred output type. */
-declare class PipelineBuilder<Input, Output = Input> {
     private readonly executor;
     private readonly state;
-    constructor(schema: z.ZodType<Output, Input>);
-    constructor(schema: z.ZodType<Output, Input>, metadata: PipelineMetadata);
-    constructor(executor: (input: Input) => Output | Promise<Output>);
-    constructor(executor: PipelineExecutor<Input, Output>, state: PipelineBuilderState);
+    constructor(options: PipelineOptions<Input, Output>);
     /** Add a synchronous or asynchronous transform stage. */
-    step<Next>(fn: (input: Awaited<Output>) => Next | Promise<Next>, metadata?: PipelineStageMetadata): PipelineBuilder<Input, Awaited<Next>>;
+    step<Next>(fn: (input: Awaited<Output>) => Next | Promise<Next>, metadata?: PipelineStageMetadata): Pipeline<Input, Awaited<Next>>;
     /** Compose another pipeline operation after the current stage. */
-    use<Next>(op: PipelineOp<Awaited<Output>, Next>, metadata?: PipelineStageMetadata): PipelineBuilder<Input, Awaited<Next>>;
+    use<Next>(op: PipelineOp<Awaited<Output>, Next>, metadata?: PipelineStageMetadata): Pipeline<Input, Awaited<Next>>;
     /** Run named branch operations concurrently from the current value. */
-    parallel<Branches extends Record<string, PipelineOp<Awaited<Output>, unknown>>>(branches: Branches, metadata?: PipelineStageMetadata): PipelineBuilder<Input, ParallelOutput<Branches>>;
+    parallel<Branches extends Record<string, PipelineOp<Awaited<Output>, unknown>>>(branches: Branches, metadata?: PipelineStageMetadata): Pipeline<Input, ParallelOutput<Branches>>;
     /** Send the current value to an agent as text and continue with the agent output. */
-    prompt(agent: Agent<CompletionModel>, metadata?: PipelineStageMetadata): PipelineBuilder<Input, string>;
+    agent(agent: Agent<CompletionModel>, metadata?: PipelineStageMetadata): Pipeline<Input, string>;
     /** Send the current value to an extractor as text and continue with typed schema data. */
-    extract<T>(extractor: Extractor<T, CompletionModel>, metadata?: PipelineStageMetadata): PipelineBuilder<Input, T>;
-    /** Finish the builder and return a runnable pipeline. */
-    build(): Pipeline<Input, Awaited<Output>>;
+    extract<T>(extractor: Extractor<T, CompletionModel>, metadata?: PipelineStageMetadata): Pipeline<Input, T>;
+    /** Run one input through the pipeline and return the final stage output. */
+    run(input: Input, options?: PipelineRunOptions): Promise<Awaited<Output>>;
+    /** Run many inputs through the same pipeline with bounded concurrency. */
+    batch<I extends Iterable<Input>>(inputs: I, options: PipelineBatchOptions): Promise<Array<Awaited<Output>>>;
+    /** Return an isolated graph snapshot for the current pipeline stages. */
+    graph(): PipelineGraph;
+    private derive;
     private runStep;
 }
 
-export { Pipeline, type PipelineBatchOptions, PipelineBuilder, type PipelineGraph, type PipelineGraphEdge, type PipelineGraphNode, type PipelineMetadata, type PipelineOp, type PipelineRunEvent, type PipelineRunObserver, type PipelineRunOptions, type PipelineStageKind, type PipelineStageMetadata };
+export { Pipeline, type PipelineBatchOptions, type PipelineGraph, type PipelineGraphEdge, type PipelineGraphNode, type PipelineMetadata, type PipelineOp, type PipelineOptions, type PipelineRunEvent, type PipelineRunObserver, type PipelineRunOptions, type PipelineStageKind, type PipelineStageMetadata };