@amplitude/ai 0.1.0

package/dist/utils/logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.js","names":["defaultLogger: Logger"],"sources":["../../src/utils/logger.ts"],"sourcesContent":["export interface Logger {\n  debug(message: string): void;\n  error(message: string): void;\n  warn(message: string): void;\n  info(message: string): void;\n}\n\nconst defaultLogger: Logger = {\n  debug: () => {},\n  error: (msg) => console.error(`[amplitude-ai] ${msg}`),\n  warn: (msg) => console.warn(`[amplitude-ai] ${msg}`),\n  info: () => {},\n};\n\nexport function getLogger(amplitude?: unknown): Logger {\n  if (amplitude && typeof amplitude === 'object') {\n    const config = (amplitude as Record<string, unknown>).configuration as\n      | Record<string, unknown>\n      | undefined;\n    if (config?.loggerProvider && typeof config.loggerProvider === 'object') {\n      return config.loggerProvider as Logger;\n    }\n  }\n  return defaultLogger;\n}\n"],"mappings":";AAOA,MAAMA,gBAAwB;CAC5B,aAAa;CACb,QAAQ,QAAQ,QAAQ,MAAM,kBAAkB,MAAM;CACtD,OAAO,QAAQ,QAAQ,KAAK,kBAAkB,MAAM;CACpD,YAAY;CACb;AAED,SAAgB,UAAU,WAA6B;AACrD,KAAI,aAAa,OAAO,cAAc,UAAU;EAC9C,MAAM,SAAU,UAAsC;AAGtD,MAAI,QAAQ,kBAAkB,OAAO,OAAO,mBAAmB,SAC7D,QAAO,OAAO;;AAGlB,QAAO"}

package/dist/utils/model-tiers.d.ts

@@ -0,0 +1,8 @@
+//#region src/utils/model-tiers.d.ts
+declare const TIER_FAST = "fast";
+declare const TIER_STANDARD = "standard";
+declare const TIER_REASONING = "reasoning";
+declare function inferModelTier(modelName: string): string;
+//#endregion
+export { TIER_FAST, TIER_REASONING, TIER_STANDARD, inferModelTier };
+//# sourceMappingURL=model-tiers.d.ts.map

package/dist/utils/model-tiers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-tiers.d.ts","names":[],"sources":["../../src/utils/model-tiers.ts"],"sourcesContent":[],"mappings":";cAAa,SAAA;AAAA,cACA,aAAA,GADS,UAAA;AACT,cACA,cAAA,GADa,WAAA;AACb,iBAwEG,cAAA,CAxEW,SAAA,EAAA,MAAA,CAAA,EAAA,MAAA"}

package/dist/utils/model-tiers.js

@@ -0,0 +1,76 @@
+//#region src/utils/model-tiers.ts
+const TIER_FAST = "fast";
+const TIER_STANDARD = "standard";
+const TIER_REASONING = "reasoning";
+const MODEL_TIER_RULES = [
+	["gpt-4o-mini", TIER_FAST],
+	["gpt-4.1-mini", TIER_FAST],
+	["gpt-4.1-nano", TIER_FAST],
+	["gpt-3.5", TIER_FAST],
+	["o1-mini", TIER_FAST],
+	["o3-mini", TIER_FAST],
+	["o4-mini", TIER_FAST],
+	["o1-pro", TIER_REASONING],
+	["o1-preview", TIER_REASONING],
+	["o1", TIER_REASONING],
+	["o3", TIER_REASONING],
+	["o4", TIER_REASONING],
+	["gpt-4o", TIER_STANDARD],
+	["gpt-4.1", TIER_STANDARD],
+	["gpt-4-turbo", TIER_STANDARD],
+	["gpt-4", TIER_STANDARD],
+	["claude-3-haiku", TIER_FAST],
+	["claude-3.5-haiku", TIER_FAST],
+	["claude-haiku", TIER_FAST],
+	["claude-3-opus", TIER_STANDARD],
+	["claude-opus", TIER_STANDARD],
+	["claude-sonnet", TIER_STANDARD],
+	["claude-3.5-sonnet", TIER_STANDARD],
+	["claude-3-sonnet", TIER_STANDARD],
+	["claude-4", TIER_STANDARD],
+	["gemini-2.0-flash", TIER_FAST],
+	["gemini-1.5-flash", TIER_FAST],
+	["gemini-flash", TIER_FAST],
+	["gemini-2.0-pro", TIER_STANDARD],
+	["gemini-1.5-pro", TIER_STANDARD],
+	["gemini-pro", TIER_STANDARD],
+	["gemini-ultra", TIER_REASONING],
+	["mistral-small", TIER_FAST],
+	["mistral-tiny", TIER_FAST],
+	["pixtral", TIER_FAST],
+	["mistral-large", TIER_STANDARD],
+	["mistral-medium", TIER_STANDARD],
+	["mixtral", TIER_STANDARD],
+	["mistral-nemo", TIER_STANDARD],
+	["llama-3.3", TIER_STANDARD],
+	["llama-3.2", TIER_FAST],
+	["llama-3.1-405b", TIER_STANDARD],
+	["llama-3.1-70b", TIER_STANDARD],
+	["llama-3.1-8b", TIER_FAST],
+	["llama-3-70b", TIER_STANDARD],
+	["llama-3-8b", TIER_FAST],
+	["llama", TIER_STANDARD],
+	["command-r-plus", TIER_STANDARD],
+	["command-r", TIER_FAST],
+	["command-light", TIER_FAST],
+	["command", TIER_STANDARD],
+	["deepseek-r1", TIER_REASONING],
+	["deepseek-v3", TIER_STANDARD],
+	["deepseek-chat", TIER_STANDARD],
+	["deepseek-coder", TIER_STANDARD],
+	["deepseek", TIER_STANDARD]
+];
+function inferModelTier(modelName) {
+	const lower = modelName.toLowerCase();
+	const normalized = lower.includes(":") ? lower.split(":", 2)[1] ?? lower : lower;
+	for (const [pattern, tier] of MODEL_TIER_RULES) {
+		if (pattern === "o1" || pattern === "o3" || pattern === "o4") continue;
+		if (normalized.includes(pattern)) return tier;
+	}
+	if (normalized.match(/(?:^|[^a-z0-9])(o[134])(?:$|[^a-z0-9])/) != null) return TIER_REASONING;
+	return TIER_STANDARD;
+}
+
+//#endregion
+export { TIER_FAST, TIER_REASONING, TIER_STANDARD, inferModelTier };
+//# sourceMappingURL=model-tiers.js.map

package/dist/utils/model-tiers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-tiers.js","names":["MODEL_TIER_RULES: Array<[string, string]>"],"sources":["../../src/utils/model-tiers.ts"],"sourcesContent":["export const TIER_FAST = 'fast';\nexport const TIER_STANDARD = 'standard';\nexport const TIER_REASONING = 'reasoning';\n\nconst MODEL_TIER_RULES: Array<[string, string]> = [\n  // OpenAI fast\n  ['gpt-4o-mini', TIER_FAST],\n  ['gpt-4.1-mini', TIER_FAST],\n  ['gpt-4.1-nano', TIER_FAST],\n  ['gpt-3.5', TIER_FAST],\n  // OpenAI reasoning + o-series\n  ['o1-mini', TIER_FAST],\n  ['o3-mini', TIER_FAST],\n  ['o4-mini', TIER_FAST],\n  ['o1-pro', TIER_REASONING],\n  ['o1-preview', TIER_REASONING],\n  ['o1', TIER_REASONING],\n  ['o3', TIER_REASONING],\n  ['o4', TIER_REASONING],\n  // OpenAI standard\n  ['gpt-4o', TIER_STANDARD],\n  ['gpt-4.1', TIER_STANDARD],\n  ['gpt-4-turbo', TIER_STANDARD],\n  ['gpt-4', TIER_STANDARD],\n\n  // Anthropic\n  ['claude-3-haiku', TIER_FAST],\n  ['claude-3.5-haiku', TIER_FAST],\n  ['claude-haiku', TIER_FAST],\n  ['claude-3-opus', TIER_STANDARD],\n  ['claude-opus', TIER_STANDARD],\n  ['claude-sonnet', TIER_STANDARD],\n  ['claude-3.5-sonnet', TIER_STANDARD],\n  ['claude-3-sonnet', TIER_STANDARD],\n  ['claude-4', TIER_STANDARD],\n\n  // Gemini\n  ['gemini-2.0-flash', TIER_FAST],\n  ['gemini-1.5-flash', TIER_FAST],\n  ['gemini-flash', TIER_FAST],\n  ['gemini-2.0-pro', TIER_STANDARD],\n  ['gemini-1.5-pro', TIER_STANDARD],\n  ['gemini-pro', TIER_STANDARD],\n  ['gemini-ultra', TIER_REASONING],\n\n  // Mistral\n  ['mistral-small', TIER_FAST],\n  ['mistral-tiny', TIER_FAST],\n  ['pixtral', TIER_FAST],\n  ['mistral-large', TIER_STANDARD],\n  ['mistral-medium', TIER_STANDARD],\n  ['mixtral', TIER_STANDARD],\n  ['mistral-nemo', TIER_STANDARD],\n\n  // Bedrock/meta + others\n  ['llama-3.3', TIER_STANDARD],\n  ['llama-3.2', TIER_FAST],\n  ['llama-3.1-405b', TIER_STANDARD],\n  ['llama-3.1-70b', TIER_STANDARD],\n  ['llama-3.1-8b', TIER_FAST],\n  ['llama-3-70b', TIER_STANDARD],\n  ['llama-3-8b', TIER_FAST],\n  ['llama', TIER_STANDARD],\n  ['command-r-plus', TIER_STANDARD],\n  ['command-r', TIER_FAST],\n  ['command-light', TIER_FAST],\n  ['command', TIER_STANDARD],\n  ['deepseek-r1', TIER_REASONING],\n  ['deepseek-v3', TIER_STANDARD],\n  ['deepseek-chat', TIER_STANDARD],\n  ['deepseek-coder', TIER_STANDARD],\n  ['deepseek', TIER_STANDARD],\n];\n\nexport function inferModelTier(modelName: string): string {\n  const lower = modelName.toLowerCase();\n  const normalized = lower.includes(':')\n    ? (lower.split(':', 2)[1] ?? lower)\n    : lower;\n\n  for (const [pattern, tier] of MODEL_TIER_RULES) {\n    if (pattern === 'o1' || pattern === 'o3' || pattern === 'o4') {\n      continue;\n    }\n    if (normalized.includes(pattern)) return tier;\n  }\n\n  // Avoid false positives for short o-series identifiers.\n  // Examples that should NOT match:\n  // - \"o100\"\n  // - \"co3-something\"\n  // Examples that SHOULD match:\n  // - \"o1\"\n  // - \"o1-2024-12-17\"\n  // - \"o3-high\"\n  const shortOseriesMatch = normalized.match(\n    /(?:^|[^a-z0-9])(o[134])(?:$|[^a-z0-9])/,\n  );\n  if (shortOseriesMatch != null) {\n    return TIER_REASONING;\n  }\n\n  return TIER_STANDARD;\n}\n"],"mappings":";AAAA,MAAa,YAAY;AACzB,MAAa,gBAAgB;AAC7B,MAAa,iBAAiB;AAE9B,MAAMA,mBAA4C;CAEhD,CAAC,eAAe,UAAU;CAC1B,CAAC,gBAAgB,UAAU;CAC3B,CAAC,gBAAgB,UAAU;CAC3B,CAAC,WAAW,UAAU;CAEtB,CAAC,WAAW,UAAU;CACtB,CAAC,WAAW,UAAU;CACtB,CAAC,WAAW,UAAU;CACtB,CAAC,UAAU,eAAe;CAC1B,CAAC,cAAc,eAAe;CAC9B,CAAC,MAAM,eAAe;CACtB,CAAC,MAAM,eAAe;CACtB,CAAC,MAAM,eAAe;CAEtB,CAAC,UAAU,cAAc;CACzB,CAAC,WAAW,cAAc;CAC1B,CAAC,eAAe,cAAc;CAC9B,CAAC,SAAS,cAAc;CAGxB,CAAC,kBAAkB,UAAU;CAC7B,CAAC,oBAAoB,UAAU;CAC/B,CAAC,gBAAgB,UAAU;CAC3B,CAAC,iBAAiB,cAAc;CAChC,CAAC,eAAe,cAAc;CAC9B,CAAC,iBAAiB,cAAc;CAChC,CAAC,qBAAqB,cAAc;CACpC,CAAC,mBAAmB,cAAc;CAClC,CAAC,YAAY,cAAc;CAG3B,CAAC,oBAAoB,UAAU;CAC/B,CAAC,oBAAoB,UAAU;CAC/B,CAAC,gBAAgB,UAAU;CAC3B,CAAC,kBAAkB,cAAc;CACjC,CAAC,kBAAkB,cAAc;CACjC,CAAC,cAAc,cAAc;CAC7B,CAAC,gBAAgB,eAAe;CAGhC,CAAC,iBAAiB,UAAU;CAC5B,CAAC,gBAAgB,UAAU;CAC3B,CAAC,WAAW,UAAU;CACtB,CAAC,iBAAiB,cAAc;CAChC,CAAC,kBAAkB,cAAc;CACjC,CAAC,WAAW,cAAc;CAC1B,CAAC,gBAAgB,cAAc;CAG/B,CAAC,aAAa,cAAc;CAC5B,CAAC,aAAa,UAAU;CACxB,CAAC,kBAAkB,cAAc;CACjC,CAAC,iBAAiB,cAAc;CAChC,CAAC,gBAAgB,UAAU;CAC3B,CAAC,eAAe,cAAc;CAC9B,CAAC,cAAc,UAAU;CACzB,CAAC,SAAS,cAAc;CACxB,CAAC,kBAAkB,cAAc;CACjC,CAAC,aAAa,UAAU;CACxB,CAAC,iBAAiB,UAAU;CAC5B,CAAC,WAAW,cAAc;CAC1B,CAAC,eAAe,eAAe;CAC/B,CAAC,eAAe,cAAc;CAC9B,CAAC,iBAAiB,cAAc;CAChC,CAAC,kBAAkB,cAAc;CACjC,CAAC,YAAY,cAAc;CAC5B;AAED,SAAgB,eAAe,WAA2B;CACxD,MAAM,QAAQ,UAAU,aAAa;CACrC,MAAM,aAAa,MAAM,SAAS,IAAI,GACjC,MAAM,MAAM,KAAK,EAAE,CAAC,MAAM,QAC3B;AAEJ,MAAK,MAAM,CAAC,SAAS,SAAS,kBAAkB;AAC9C,MAAI,YAAY,QAAQ,YAAY,QAAQ,YAAY,KACtD;AAEF,MAAI,WAAW,SAAS,QAAQ,CAAE,QAAO;;AAc3C,KAH0B,WAAW,MACnC,yCACD,IACwB,KACvB,QAAO;AAGT,QAAO"}

package/dist/utils/providers.d.ts

@@ -0,0 +1,5 @@
+//#region src/utils/providers.d.ts
+declare function inferProviderFromModel(modelName: string): string;
+//#endregion
+export { inferProviderFromModel };
+//# sourceMappingURL=providers.d.ts.map

package/dist/utils/providers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"providers.d.ts","names":[],"sources":["../../src/utils/providers.ts"],"sourcesContent":[],"mappings":";iBAYgB,sBAAA"}

package/dist/utils/providers.js

@@ -0,0 +1,20 @@
+//#region src/utils/providers.ts
+const PROVIDER_PATTERNS = [
+	[/^gpt-|^o[1-9]|^dall-e|^text-embedding|^whisper|^tts/i, "openai"],
+	[/^claude/i, "anthropic"],
+	[/^gemini|^palm/i, "gemini"],
+	[/^mistral|^codestral|^open-mistral|^pixtral/i, "mistral"],
+	[/^command|^embed-/i, "cohere"],
+	[/^amazon\.|^titan/i, "amazon"],
+	[/^llama|^meta\./i, "meta"],
+	[/^jamba|^ai21\./i, "ai21"],
+	[/^deepseek/i, "deepseek"]
+];
+function inferProviderFromModel(modelName) {
+	for (const [pattern, provider] of PROVIDER_PATTERNS) if (pattern.test(modelName)) return provider;
+	return "openai";
+}
+
+//#endregion
+export { inferProviderFromModel };
+//# sourceMappingURL=providers.js.map

package/dist/utils/providers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"providers.js","names":["PROVIDER_PATTERNS: Array<[RegExp, string]>"],"sources":["../../src/utils/providers.ts"],"sourcesContent":["const PROVIDER_PATTERNS: Array<[RegExp, string]> = [\n  [/^gpt-|^o[1-9]|^dall-e|^text-embedding|^whisper|^tts/i, 'openai'],\n  [/^claude/i, 'anthropic'],\n  [/^gemini|^palm/i, 'gemini'],\n  [/^mistral|^codestral|^open-mistral|^pixtral/i, 'mistral'],\n  [/^command|^embed-/i, 'cohere'],\n  [/^amazon\\.|^titan/i, 'amazon'],\n  [/^llama|^meta\\./i, 'meta'],\n  [/^jamba|^ai21\\./i, 'ai21'],\n  [/^deepseek/i, 'deepseek'],\n];\n\nexport function inferProviderFromModel(modelName: string): string {\n  for (const [pattern, provider] of PROVIDER_PATTERNS) {\n    if (pattern.test(modelName)) return provider;\n  }\n  return 'openai';\n}\n"],"mappings":";AAAA,MAAMA,oBAA6C;CACjD,CAAC,wDAAwD,SAAS;CAClE,CAAC,YAAY,YAAY;CACzB,CAAC,kBAAkB,SAAS;CAC5B,CAAC,+CAA+C,UAAU;CAC1D,CAAC,qBAAqB,SAAS;CAC/B,CAAC,qBAAqB,SAAS;CAC/B,CAAC,mBAAmB,OAAO;CAC3B,CAAC,mBAAmB,OAAO;CAC3B,CAAC,cAAc,WAAW;CAC3B;AAED,SAAgB,uBAAuB,WAA2B;AAChE,MAAK,MAAM,CAAC,SAAS,aAAa,kBAChC,KAAI,QAAQ,KAAK,UAAU,CAAE,QAAO;AAEtC,QAAO"}

package/dist/utils/resolve-module.d.ts

@@ -0,0 +1,5 @@
+//#region src/utils/resolve-module.d.ts
+declare function tryRequire(name: string): Record<string, unknown> | null;
+//#endregion
+export { tryRequire };
+//# sourceMappingURL=resolve-module.d.ts.map

package/dist/utils/resolve-module.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"resolve-module.d.ts","names":[],"sources":["../../src/utils/resolve-module.ts"],"sourcesContent":[],"mappings":";iBAmBgB,UAAA,gBAA0B"}

package/dist/utils/resolve-module.js

@@ -0,0 +1,29 @@
+import { createRequire } from "node:module";
+
+//#region src/utils/resolve-module.ts
+/**
+* Wrapper around `createRequire()` for optional dependency loading.
+*
+* This package ships as ESM, so bare `require()` is not available at runtime.
+* Keeping this in a single module lets tests vi.mock() optional resolution.
+*
+* In bundler environments (Turbopack, Webpack, etc.) `createRequire` or
+* `import.meta.url` may be rewritten to nonsensical values, so we wrap
+* the initialisation in a try-catch and degrade gracefully to `null`.
+*/
+let localRequire = null;
+try {
+	localRequire = createRequire(import.meta.url);
+} catch {}
+function tryRequire(name) {
+	if (localRequire == null) return null;
+	try {
+		return localRequire(name);
+	} catch {
+		return null;
+	}
+}
+
+//#endregion
+export { tryRequire };
+//# sourceMappingURL=resolve-module.js.map

package/dist/utils/resolve-module.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"resolve-module.js","names":["localRequire: NodeRequire | null"],"sources":["../../src/utils/resolve-module.ts"],"sourcesContent":["/**\n * Wrapper around `createRequire()` for optional dependency loading.\n *\n * This package ships as ESM, so bare `require()` is not available at runtime.\n * Keeping this in a single module lets tests vi.mock() optional resolution.\n *\n * In bundler environments (Turbopack, Webpack, etc.) `createRequire` or\n * `import.meta.url` may be rewritten to nonsensical values, so we wrap\n * the initialisation in a try-catch and degrade gracefully to `null`.\n */\nimport { createRequire } from 'node:module';\n\nlet localRequire: NodeRequire | null = null;\ntry {\n  localRequire = createRequire(import.meta.url);\n} catch {\n  // Bundler environment — import.meta.url is invalid or createRequire unavailable.\n}\n\nexport function tryRequire(name: string): Record<string, unknown> | null {\n  if (localRequire == null) return null;\n  try {\n    return localRequire(name) as Record<string, unknown>;\n  } catch {\n    return null;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;AAYA,IAAIA,eAAmC;AACvC,IAAI;AACF,gBAAe,cAAc,OAAO,KAAK,IAAI;QACvC;AAIR,SAAgB,WAAW,MAA8C;AACvE,KAAI,gBAAgB,KAAM,QAAO;AACjC,KAAI;AACF,SAAO,aAAa,KAAK;SACnB;AACN,SAAO"}

package/dist/utils/streaming.d.ts

@@ -0,0 +1,55 @@
+//#region src/utils/streaming.d.ts
+interface StreamingAccumulatorState {
+  content: string;
+  inputTokens: number | null;
+  outputTokens: number | null;
+  totalTokens: number | null;
+  reasoningTokens: number | null;
+  cacheReadTokens: number | null;
+  cacheCreationTokens: number | null;
+  finishReason: string | null;
+  toolCalls: Array<Record<string, unknown>>;
+  model: string | null;
+  ttfbMs: number | null;
+  isError: boolean;
+  errorMessage: string | null;
+}
+declare class StreamingAccumulator {
+  content: string;
+  inputTokens: number | null;
+  outputTokens: number | null;
+  totalTokens: number | null;
+  reasoningTokens: number | null;
+  cacheReadTokens: number | null;
+  cacheCreationTokens: number | null;
+  finishReason: string | null;
+  toolCalls: Array<Record<string, unknown>>;
+  model: string | null;
+  ttfbMs: number | null;
+  isError: boolean;
+  errorMessage: string | null;
+  private _startTime;
+  private _firstChunkReceived;
+  constructor();
+  addContent(chunk: string): void;
+  /**
+   * Set (overwrite) token usage fields. Only non-null values are written,
+   * allowing incremental updates where different streaming events provide
+   * different fields (e.g. Anthropic message_start vs message_delta).
+   */
+  setUsage(usage: {
+    inputTokens?: number;
+    outputTokens?: number;
+    totalTokens?: number;
+    reasoningTokens?: number;
+    cacheReadTokens?: number;
+    cacheCreationTokens?: number;
+  }): void;
+  addToolCall(toolCall: Record<string, unknown>): void;
+  setError(message: string): void;
+  getState(): StreamingAccumulatorState;
+  get elapsedMs(): number;
+}
+//#endregion
+export { StreamingAccumulator, StreamingAccumulatorState };
+//# sourceMappingURL=streaming.d.ts.map

package/dist/utils/streaming.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming.d.ts","names":[],"sources":["../../src/utils/streaming.ts"],"sourcesContent":[],"mappings":";UAAiB,yBAAA;EAAA,OAAA,EAAA,MAAA;EAgBJ,WAAA,EAAA,MAAA,GAAA,IAAoB;EASd,YAAA,EAAA,MAAA,GAAA,IAAA;EAAN,WAAA,EAAA,MAAA,GAAA,IAAA;EA4CW,eAAA,EAAA,MAAA,GAAA,IAAA;EASV,eAAA,EAAA,MAAA,GAAA,IAAA;EAAyB,mBAAA,EAAA,MAAA,GAAA,IAAA;;aArE1B,MAAM;;;;;;cAON,oBAAA;;;;;;;;;aASA,MAAM;;;;;;;;;;;;;;;;;;;;;;wBA4CK;;cASV"}

package/dist/utils/streaming.js

@@ -0,0 +1,72 @@
+//#region src/utils/streaming.ts
+var StreamingAccumulator = class {
+	content = "";
+	inputTokens = null;
+	outputTokens = null;
+	totalTokens = null;
+	reasoningTokens = null;
+	cacheReadTokens = null;
+	cacheCreationTokens = null;
+	finishReason = null;
+	toolCalls = [];
+	model = null;
+	ttfbMs = null;
+	isError = false;
+	errorMessage = null;
+	_startTime;
+	_firstChunkReceived = false;
+	constructor() {
+		this._startTime = performance.now();
+	}
+	addContent(chunk) {
+		if (!this._firstChunkReceived) {
+			this.ttfbMs = performance.now() - this._startTime;
+			this._firstChunkReceived = true;
+		}
+		this.content += chunk;
+	}
+	/**
+	* Set (overwrite) token usage fields. Only non-null values are written,
+	* allowing incremental updates where different streaming events provide
+	* different fields (e.g. Anthropic message_start vs message_delta).
+	*/
+	setUsage(usage) {
+		if (usage.inputTokens != null) this.inputTokens = usage.inputTokens;
+		if (usage.outputTokens != null) this.outputTokens = usage.outputTokens;
+		if (usage.totalTokens != null) this.totalTokens = usage.totalTokens;
+		if (usage.reasoningTokens != null) this.reasoningTokens = usage.reasoningTokens;
+		if (usage.cacheReadTokens != null) this.cacheReadTokens = usage.cacheReadTokens;
+		if (usage.cacheCreationTokens != null) this.cacheCreationTokens = usage.cacheCreationTokens;
+	}
+	addToolCall(toolCall) {
+		this.toolCalls.push(toolCall);
+	}
+	setError(message) {
+		this.isError = true;
+		this.errorMessage = message;
+	}
+	getState() {
+		return {
+			content: this.content,
+			inputTokens: this.inputTokens,
+			outputTokens: this.outputTokens,
+			totalTokens: this.totalTokens,
+			reasoningTokens: this.reasoningTokens,
+			cacheReadTokens: this.cacheReadTokens,
+			cacheCreationTokens: this.cacheCreationTokens,
+			finishReason: this.finishReason,
+			toolCalls: this.toolCalls,
+			model: this.model,
+			ttfbMs: this.ttfbMs,
+			isError: this.isError,
+			errorMessage: this.errorMessage
+		};
+	}
+	get elapsedMs() {
+		return performance.now() - this._startTime;
+	}
+};
+
+//#endregion
+export { StreamingAccumulator };
+//# sourceMappingURL=streaming.js.map

package/dist/utils/streaming.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming.js","names":[],"sources":["../../src/utils/streaming.ts"],"sourcesContent":["export interface StreamingAccumulatorState {\n  content: string;\n  inputTokens: number | null;\n  outputTokens: number | null;\n  totalTokens: number | null;\n  reasoningTokens: number | null;\n  cacheReadTokens: number | null;\n  cacheCreationTokens: number | null;\n  finishReason: string | null;\n  toolCalls: Array<Record<string, unknown>>;\n  model: string | null;\n  ttfbMs: number | null;\n  isError: boolean;\n  errorMessage: string | null;\n}\n\nexport class StreamingAccumulator {\n  content = '';\n  inputTokens: number | null = null;\n  outputTokens: number | null = null;\n  totalTokens: number | null = null;\n  reasoningTokens: number | null = null;\n  cacheReadTokens: number | null = null;\n  cacheCreationTokens: number | null = null;\n  finishReason: string | null = null;\n  toolCalls: Array<Record<string, unknown>> = [];\n  model: string | null = null;\n  ttfbMs: number | null = null;\n  isError = false;\n  errorMessage: string | null = null;\n  private _startTime: number;\n  private _firstChunkReceived = false;\n\n  constructor() {\n    this._startTime = performance.now();\n  }\n\n  addContent(chunk: string): void {\n    if (!this._firstChunkReceived) {\n      this.ttfbMs = performance.now() - this._startTime;\n      this._firstChunkReceived = true;\n    }\n    this.content += chunk;\n  }\n\n  /**\n   * Set (overwrite) token usage fields. Only non-null values are written,\n   * allowing incremental updates where different streaming events provide\n   * different fields (e.g. Anthropic message_start vs message_delta).\n   */\n  setUsage(usage: {\n    inputTokens?: number;\n    outputTokens?: number;\n    totalTokens?: number;\n    reasoningTokens?: number;\n    cacheReadTokens?: number;\n    cacheCreationTokens?: number;\n  }): void {\n    if (usage.inputTokens != null) this.inputTokens = usage.inputTokens;\n    if (usage.outputTokens != null) this.outputTokens = usage.outputTokens;\n    if (usage.totalTokens != null) this.totalTokens = usage.totalTokens;\n    if (usage.reasoningTokens != null)\n      this.reasoningTokens = usage.reasoningTokens;\n    if (usage.cacheReadTokens != null)\n      this.cacheReadTokens = usage.cacheReadTokens;\n    if (usage.cacheCreationTokens != null)\n      this.cacheCreationTokens = usage.cacheCreationTokens;\n  }\n\n  addToolCall(toolCall: Record<string, unknown>): void {\n    this.toolCalls.push(toolCall);\n  }\n\n  setError(message: string): void {\n    this.isError = true;\n    this.errorMessage = message;\n  }\n\n  getState(): StreamingAccumulatorState {\n    return {\n      content: this.content,\n      inputTokens: this.inputTokens,\n      outputTokens: this.outputTokens,\n      totalTokens: this.totalTokens,\n      reasoningTokens: this.reasoningTokens,\n      cacheReadTokens: this.cacheReadTokens,\n      cacheCreationTokens: this.cacheCreationTokens,\n      finishReason: this.finishReason,\n      toolCalls: this.toolCalls,\n      model: this.model,\n      ttfbMs: this.ttfbMs,\n      isError: this.isError,\n      errorMessage: this.errorMessage,\n    };\n  }\n\n  get elapsedMs(): number {\n    return performance.now() - this._startTime;\n  }\n}\n"],"mappings":";AAgBA,IAAa,uBAAb,MAAkC;CAChC,UAAU;CACV,cAA6B;CAC7B,eAA8B;CAC9B,cAA6B;CAC7B,kBAAiC;CACjC,kBAAiC;CACjC,sBAAqC;CACrC,eAA8B;CAC9B,YAA4C,EAAE;CAC9C,QAAuB;CACvB,SAAwB;CACxB,UAAU;CACV,eAA8B;CAC9B,AAAQ;CACR,AAAQ,sBAAsB;CAE9B,cAAc;AACZ,OAAK,aAAa,YAAY,KAAK;;CAGrC,WAAW,OAAqB;AAC9B,MAAI,CAAC,KAAK,qBAAqB;AAC7B,QAAK,SAAS,YAAY,KAAK,GAAG,KAAK;AACvC,QAAK,sBAAsB;;AAE7B,OAAK,WAAW;;;;;;;CAQlB,SAAS,OAOA;AACP,MAAI,MAAM,eAAe,KAAM,MAAK,cAAc,MAAM;AACxD,MAAI,MAAM,gBAAgB,KAAM,MAAK,eAAe,MAAM;AAC1D,MAAI,MAAM,eAAe,KAAM,MAAK,cAAc,MAAM;AACxD,MAAI,MAAM,mBAAmB,KAC3B,MAAK,kBAAkB,MAAM;AAC/B,MAAI,MAAM,mBAAmB,KAC3B,MAAK,kBAAkB,MAAM;AAC/B,MAAI,MAAM,uBAAuB,KAC/B,MAAK,sBAAsB,MAAM;;CAGrC,YAAY,UAAyC;AACnD,OAAK,UAAU,KAAK,SAAS;;CAG/B,SAAS,SAAuB;AAC9B,OAAK,UAAU;AACf,OAAK,eAAe;;CAGtB,WAAsC;AACpC,SAAO;GACL,SAAS,KAAK;GACd,aAAa,KAAK;GAClB,cAAc,KAAK;GACnB,aAAa,KAAK;GAClB,iBAAiB,KAAK;GACtB,iBAAiB,KAAK;GACtB,qBAAqB,KAAK;GAC1B,cAAc,KAAK;GACnB,WAAW,KAAK;GAChB,OAAO,KAAK;GACZ,QAAQ,KAAK;GACb,SAAS,KAAK;GACd,cAAc,KAAK;GACpB;;CAGH,IAAI,YAAoB;AACtB,SAAO,YAAY,KAAK,GAAG,KAAK"}

package/dist/utils/tokens.d.ts

@@ -0,0 +1,16 @@
+//#region src/utils/tokens.d.ts
+/**
+ * Token counting utilities.
+ *
+ * Uses tiktoken when available for accurate counts.
+ * Falls back to a heuristic (~3.5 chars per token + word count adjustment).
+ */
+declare function countTokens(text: string, model?: string): number;
+declare function estimateTokens(text: string): number;
+declare function countMessageTokens(messages: Array<{
+  role?: string;
+  content?: unknown;
+}>, model?: string): number;
+//#endregion
+export { countMessageTokens, countTokens, estimateTokens };
+//# sourceMappingURL=tokens.d.ts.map

package/dist/utils/tokens.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tokens.d.ts","names":[],"sources":["../../src/utils/tokens.ts"],"sourcesContent":[],"mappings":";;AA6CA;AAkBA;AA6BA;;;iBA/CgB,WAAA;iBAkBA,cAAA;iBA6BA,kBAAA,WACJ"}

package/dist/utils/tokens.js

@@ -0,0 +1,80 @@
+import { tryRequire } from "./resolve-module.js";
+
+//#region src/utils/tokens.ts
+/**
+* Token counting utilities.
+*
+* Uses tiktoken when available for accurate counts.
+* Falls back to a heuristic (~3.5 chars per token + word count adjustment).
+*/
+const tiktokenModule = tryRequire("tiktoken") ?? tryRequire("js-tiktoken");
+const encodingCache = /* @__PURE__ */ new Map();
+function getEncoding(modelName) {
+	const key = modelName ?? "cl100k_base";
+	if (encodingCache.has(key)) return encodingCache.get(key);
+	if (tiktokenModule == null) return null;
+	try {
+		const mod = tiktokenModule;
+		let encoding;
+		if (typeof mod.encoding_for_model === "function" && modelName) try {
+			encoding = mod.encoding_for_model(modelName);
+		} catch {
+			if (typeof mod.get_encoding === "function") try {
+				encoding = mod.get_encoding("o200k_base");
+			} catch {
+				encoding = mod.get_encoding("cl100k_base");
+			}
+		}
+		else if (typeof mod.get_encoding === "function") encoding = mod.get_encoding("cl100k_base");
+		if (encoding) encodingCache.set(key, encoding);
+		return encoding ?? null;
+	} catch {
+		return null;
+	}
+}
+function countTokens(text, model) {
+	const encoding = getEncoding(model);
+	if (encoding != null && typeof encoding.encode === "function") try {
+		return encoding.encode(text).length;
+	} catch {}
+	return estimateTokens(text);
+}
+function estimateTokens(text) {
+	const charEstimate = text.length / 3.5;
+	const wordEstimate = text.split(/\s+/).filter(Boolean).length * .1;
+	return Math.max(1, Math.ceil(charEstimate + wordEstimate));
+}
+/**
+* Extracts text from a message content field that may be either a string
+* or a list of content blocks (multimodal format).
+*/
+function extractMessageText(content) {
+	if (typeof content === "string") return content;
+	if (Array.isArray(content)) {
+		const parts = [];
+		for (const item of content) if (typeof item === "object" && item != null && item.type === "text") {
+			const text = item.text;
+			if (typeof text === "string") parts.push(text);
+		}
+		return parts.join("");
+	}
+	return "";
+}
+function countMessageTokens(messages, model) {
+	if (messages.length === 0) return 0;
+	const TOKENS_PER_MESSAGE = 3;
+	const PRIMING_TOKENS = 3;
+	let total = 0;
+	for (const msg of messages) {
+		total += TOKENS_PER_MESSAGE;
+		const text = extractMessageText(msg.content);
+		const exact = countTokens(text, model);
+		total += exact ?? estimateTokens(text);
+	}
+	total += PRIMING_TOKENS;
+	return total;
+}
+
+//#endregion
+export { countMessageTokens, countTokens, estimateTokens };
+//# sourceMappingURL=tokens.js.map

package/dist/utils/tokens.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tokens.js","names":["tiktokenModule: Record<string, unknown> | null","encoding: unknown","parts: string[]"],"sources":["../../src/utils/tokens.ts"],"sourcesContent":["/**\n * Token counting utilities.\n *\n * Uses tiktoken when available for accurate counts.\n * Falls back to a heuristic (~3.5 chars per token + word count adjustment).\n */\n\nimport { tryRequire } from './resolve-module.js';\n\nconst tiktokenModule: Record<string, unknown> | null =\n  tryRequire('tiktoken') ?? tryRequire('js-tiktoken');\n\nconst encodingCache = new Map<string, unknown>();\n\nfunction getEncoding(modelName?: string): unknown {\n  const key = modelName ?? 'cl100k_base';\n  if (encodingCache.has(key)) return encodingCache.get(key);\n\n  if (tiktokenModule == null) return null;\n\n  try {\n    const mod = tiktokenModule as Record<string, unknown>;\n    let encoding: unknown;\n    if (typeof mod.encoding_for_model === 'function' && modelName) {\n      try {\n        encoding = mod.encoding_for_model(modelName);\n      } catch {\n        if (typeof mod.get_encoding === 'function') {\n          try {\n            encoding = mod.get_encoding('o200k_base');\n          } catch {\n            encoding = mod.get_encoding('cl100k_base');\n          }\n        }\n      }\n    } else if (typeof mod.get_encoding === 'function') {\n      encoding = mod.get_encoding('cl100k_base');\n    }\n    if (encoding) encodingCache.set(key, encoding);\n    return encoding ?? null;\n  } catch {\n    return null;\n  }\n}\n\nexport function countTokens(text: string, model?: string): number {\n  const encoding = getEncoding(model);\n  if (\n    encoding != null &&\n    typeof (encoding as Record<string, unknown>).encode === 'function'\n  ) {\n    try {\n      const tokens = (\n        encoding as { encode: (text: string) => unknown[] }\n      ).encode(text);\n      return tokens.length;\n    } catch {\n      // Fall through to heuristic\n    }\n  }\n  return estimateTokens(text);\n}\n\nexport function estimateTokens(text: string): number {\n  const charEstimate = text.length / 3.5;\n  const wordEstimate = text.split(/\\s+/).filter(Boolean).length * 0.1;\n  return Math.max(1, Math.ceil(charEstimate + wordEstimate));\n}\n\n/**\n * Extracts text from a message content field that may be either a string\n * or a list of content blocks (multimodal format).\n */\nfunction extractMessageText(content: unknown): string {\n  if (typeof content === 'string') return content;\n  if (Array.isArray(content)) {\n    const parts: string[] = [];\n    for (const item of content) {\n      if (\n        typeof item === 'object' &&\n        item != null &&\n        (item as Record<string, unknown>).type === 'text'\n      ) {\n        const text = (item as Record<string, unknown>).text;\n        if (typeof text === 'string') parts.push(text);\n      }\n    }\n    return parts.join('');\n  }\n  return '';\n}\n\nexport function countMessageTokens(\n  messages: Array<{ role?: string; content?: unknown }>,\n  model?: string,\n): number {\n  if (messages.length === 0) return 0;\n\n  const TOKENS_PER_MESSAGE = 3;\n  const PRIMING_TOKENS = 3;\n  let total = 0;\n  for (const msg of messages) {\n    total += TOKENS_PER_MESSAGE;\n    const text = extractMessageText(msg.content);\n    const exact = countTokens(text, model);\n    total += exact ?? estimateTokens(text);\n  }\n  total += PRIMING_TOKENS;\n  return total;\n}\n"],"mappings":";;;;;;;;;AASA,MAAMA,iBACJ,WAAW,WAAW,IAAI,WAAW,cAAc;AAErD,MAAM,gCAAgB,IAAI,KAAsB;AAEhD,SAAS,YAAY,WAA6B;CAChD,MAAM,MAAM,aAAa;AACzB,KAAI,cAAc,IAAI,IAAI,CAAE,QAAO,cAAc,IAAI,IAAI;AAEzD,KAAI,kBAAkB,KAAM,QAAO;AAEnC,KAAI;EACF,MAAM,MAAM;EACZ,IAAIC;AACJ,MAAI,OAAO,IAAI,uBAAuB,cAAc,UAClD,KAAI;AACF,cAAW,IAAI,mBAAmB,UAAU;UACtC;AACN,OAAI,OAAO,IAAI,iBAAiB,WAC9B,KAAI;AACF,eAAW,IAAI,aAAa,aAAa;WACnC;AACN,eAAW,IAAI,aAAa,cAAc;;;WAIvC,OAAO,IAAI,iBAAiB,WACrC,YAAW,IAAI,aAAa,cAAc;AAE5C,MAAI,SAAU,eAAc,IAAI,KAAK,SAAS;AAC9C,SAAO,YAAY;SACb;AACN,SAAO;;;AAIX,SAAgB,YAAY,MAAc,OAAwB;CAChE,MAAM,WAAW,YAAY,MAAM;AACnC,KACE,YAAY,QACZ,OAAQ,SAAqC,WAAW,WAExD,KAAI;AAIF,SAFE,SACA,OAAO,KAAK,CACA;SACR;AAIV,QAAO,eAAe,KAAK;;AAG7B,SAAgB,eAAe,MAAsB;CACnD,MAAM,eAAe,KAAK,SAAS;CACnC,MAAM,eAAe,KAAK,MAAM,MAAM,CAAC,OAAO,QAAQ,CAAC,SAAS;AAChE,QAAO,KAAK,IAAI,GAAG,KAAK,KAAK,eAAe,aAAa,CAAC;;;;;;AAO5D,SAAS,mBAAmB,SAA0B;AACpD,KAAI,OAAO,YAAY,SAAU,QAAO;AACxC,KAAI,MAAM,QAAQ,QAAQ,EAAE;EAC1B,MAAMC,QAAkB,EAAE;AAC1B,OAAK,MAAM,QAAQ,QACjB,KACE,OAAO,SAAS,YAChB,QAAQ,QACP,KAAiC,SAAS,QAC3C;GACA,MAAM,OAAQ,KAAiC;AAC/C,OAAI,OAAO,SAAS,SAAU,OAAM,KAAK,KAAK;;AAGlD,SAAO,MAAM,KAAK,GAAG;;AAEvB,QAAO;;AAGT,SAAgB,mBACd,UACA,OACQ;AACR,KAAI,SAAS,WAAW,EAAG,QAAO;CAElC,MAAM,qBAAqB;CAC3B,MAAM,iBAAiB;CACvB,IAAI,QAAQ;AACZ,MAAK,MAAM,OAAO,UAAU;AAC1B,WAAS;EACT,MAAM,OAAO,mBAAmB,IAAI,QAAQ;EAC5C,MAAM,QAAQ,YAAY,MAAM,MAAM;AACtC,WAAS,SAAS,eAAe,KAAK;;AAExC,UAAS;AACT,QAAO"}

package/dist/wrappers.d.ts

@@ -0,0 +1,37 @@
+import { AmplitudeOrAI } from "./types.js";
+import { AmplitudeAIError } from "./exceptions.js";
+import { Anthropic } from "./providers/anthropic.js";
+import { OpenAI } from "./providers/openai.js";
+import { AzureOpenAI } from "./providers/azure-openai.js";
+import * as openai0 from "openai";
+import * as _anthropic_ai_sdk0 from "@anthropic-ai/sdk";
+
+//#region src/wrappers.d.ts
+
+declare class AmplitudeAIWrapError extends AmplitudeAIError {
+  constructor(message: string);
+}
+interface WrapOpts {
+  propagateContext?: boolean;
+  /**
+   * Pass the provider SDK module directly to bypass `tryRequire`.
+   * Required in bundler environments (Turbopack, Webpack, etc.) where
+   * `createRequire` is unavailable.
+   *
+   * For OpenAI/Azure clients, pass the `openai` module default export.
+   * For Anthropic clients, pass the `@anthropic-ai/sdk` module default export.
+   */
+  providerModule?: unknown;
+  [key: string]: unknown;
+}
+/**
+ * Type-preserving overloads: when TypeScript can narrow the client type,
+ * the return type is the corresponding instrumented wrapper.
+ */
+declare function wrap(client: InstanceType<typeof openai0.AzureOpenAI>, amplitude: AmplitudeOrAI, opts?: WrapOpts): AzureOpenAI;
+declare function wrap(client: InstanceType<typeof openai0.OpenAI>, amplitude: AmplitudeOrAI, opts?: WrapOpts): OpenAI;
+declare function wrap(client: InstanceType<typeof _anthropic_ai_sdk0.Anthropic>, amplitude: AmplitudeOrAI, opts?: WrapOpts): Anthropic;
+declare function wrap(client: unknown, amplitude: AmplitudeOrAI, opts?: WrapOpts): OpenAI | AzureOpenAI | Anthropic;
+//#endregion
+export { AmplitudeAIWrapError, wrap };
+//# sourceMappingURL=wrappers.d.ts.map

package/dist/wrappers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrappers.d.ts","names":[],"sources":["../src/wrappers.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;AA4Ca,cA3BA,oBAAA,SAA6B,gBAAA,CA2B7B;EACJ,WAAA,CAAA,OAAA,EAAA,MAAA;;UArBC,QAAA,CAsBO;EACD,gBAAI,CAAA,EAAA,OAAA;EAAA;;;;;;AAKpB;;EACU,cAAA,CAAA,EAAA,OAAA;EACG,CAAA,GAAA,EAAA,MAAA,CAAA,EAAA,OAAA;;;;AAGb;;AAGS,iBAlBO,IAAA,CAkBP,MAAA,EAjBC,YAiBD,CAAA,OAlBW,OAAA,CAC2B,WAiBtC,CAAA,EAAA,SAAA,EAhBI,aAgBJ,EAAA,IAAA,CAAA,EAfA,QAeA,CAAA,EAdN,WAcM;AACN,iBAda,IAAA,CAcb,MAAA,EAbO,YAaP,CAAA,OAdiB,OAAA,CAC2B,MAa5C,CAAA,EAAA,SAAA,EAZU,aAYV,EAAA,IAAA,CAAA,EAXM,QAWN,CAAA,EAVA,MAUA;AAAY,iBATC,IAAA,CASD,MAAA,EARL,YAQK,CAAA,OATK,kBAAA,CACsC,SAQ3C,CAAA,EAAA,SAAA,EAPF,aAOE,EAAA,IAAA,CAAA,EANN,QAMM,CAAA,EALZ,SAKY;AAAiB,iBAJhB,IAAA,CAIgB,MAAA,EAAA,OAAA,EAAA,SAAA,EAFnB,aAEmB,EAAA,IAAA,CAAA,EADvB,QACuB,CAAA,EAA7B,MAA6B,GAAjB,WAAiB,GAAA,SAAA"}

package/dist/wrappers.js

@@ -0,0 +1,61 @@
+import { AmplitudeAIError } from "./exceptions.js";
+import { tryRequire } from "./utils/resolve-module.js";
+import { Anthropic } from "./providers/anthropic.js";
+import { OpenAI } from "./providers/openai.js";
+import { AzureOpenAI } from "./providers/azure-openai.js";
+
+//#region src/wrappers.ts
+/**
+* Convenience wrap() function for dependency injection.
+*
+* When the customer has already created a provider client, wrap()
+* extracts credentials and returns an instrumented SDK wrapper.
+*
+* Supported providers: OpenAI, Anthropic, AzureOpenAI.
+* Unsupported types raise AmplitudeAIWrapError.
+*/
+var AmplitudeAIWrapError = class extends AmplitudeAIError {
+	constructor(message) {
+		super(message);
+		this.name = "AmplitudeAIWrapError";
+	}
+};
+function wrap(client, amplitude, opts) {
+	const clientObj = client;
+	const clientName = (client?.constructor)?.name ?? "unknown";
+	const explicitModule = opts?.providerModule;
+	const openaiModule = explicitModule && "OpenAI" in explicitModule ? explicitModule : tryRequire("openai");
+	if (openaiModule != null) {
+		const OpenAIClass = openaiModule.OpenAI;
+		const AzureOpenAIClass = openaiModule.AzureOpenAI;
+		if (AzureOpenAIClass && client instanceof AzureOpenAIClass) return new AzureOpenAI({
+			amplitude,
+			apiKey: clientObj.apiKey,
+			azureEndpoint: String(clientObj.baseURL ?? ""),
+			propagateContext: opts?.propagateContext,
+			openaiModule
+		});
+		if (OpenAIClass && client instanceof OpenAIClass) return new OpenAI({
+			amplitude,
+			apiKey: clientObj.apiKey,
+			baseUrl: clientObj.baseURL != null ? String(clientObj.baseURL) : void 0,
+			propagateContext: opts?.propagateContext,
+			openaiModule
+		});
+	}
+	const anthropicModule = explicitModule && "Anthropic" in explicitModule ? explicitModule : tryRequire("@anthropic-ai/sdk");
+	if (anthropicModule != null) {
+		const AnthropicClass = anthropicModule.Anthropic;
+		if (AnthropicClass && client instanceof AnthropicClass) return new Anthropic({
+			amplitude,
+			apiKey: clientObj.apiKey,
+			propagateContext: opts?.propagateContext,
+			anthropicModule
+		});
+	}
+	throw new AmplitudeAIWrapError(`wrap() does not support ${clientName}. Supported types: openai.OpenAI, openai.AzureOpenAI, @anthropic-ai/sdk.Anthropic. For Gemini, Bedrock, and Mistral, use the SDK's provider classes directly.`);
+}
+
+//#endregion
+export { AmplitudeAIWrapError, wrap };
+//# sourceMappingURL=wrappers.js.map

package/dist/wrappers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"wrappers.js","names":["AmpAzureOpenAI","AmpOpenAI","AmpAnthropic"],"sources":["../src/wrappers.ts"],"sourcesContent":["/**\n * Convenience wrap() function for dependency injection.\n *\n * When the customer has already created a provider client, wrap()\n * extracts credentials and returns an instrumented SDK wrapper.\n *\n * Supported providers: OpenAI, Anthropic, AzureOpenAI.\n * Unsupported types raise AmplitudeAIWrapError.\n */\n\nimport { AmplitudeAIError } from './exceptions.js';\nimport { Anthropic as AmpAnthropic } from './providers/anthropic.js';\nimport { AzureOpenAI as AmpAzureOpenAI } from './providers/azure-openai.js';\nimport { OpenAI as AmpOpenAI } from './providers/openai.js';\nimport type { AmplitudeOrAI } from './types.js';\nimport { tryRequire } from './utils/resolve-module.js';\n\nexport class AmplitudeAIWrapError extends AmplitudeAIError {\n  constructor(message: string) {\n    super(message);\n    this.name = 'AmplitudeAIWrapError';\n  }\n}\n\ninterface WrapOpts {\n  propagateContext?: boolean;\n  /**\n   * Pass the provider SDK module directly to bypass `tryRequire`.\n   * Required in bundler environments (Turbopack, Webpack, etc.) where\n   * `createRequire` is unavailable.\n   *\n   * For OpenAI/Azure clients, pass the `openai` module default export.\n   * For Anthropic clients, pass the `@anthropic-ai/sdk` module default export.\n   */\n  providerModule?: unknown;\n  [key: string]: unknown;\n}\n\n/**\n * Type-preserving overloads: when TypeScript can narrow the client type,\n * the return type is the corresponding instrumented wrapper.\n */\nexport function wrap(\n  client: InstanceType<typeof import('openai').AzureOpenAI>,\n  amplitude: AmplitudeOrAI,\n  opts?: WrapOpts,\n): AmpAzureOpenAI;\nexport function wrap(\n  client: InstanceType<typeof import('openai').OpenAI>,\n  amplitude: AmplitudeOrAI,\n  opts?: WrapOpts,\n): AmpOpenAI;\nexport function wrap(\n  client: InstanceType<typeof import('@anthropic-ai/sdk').Anthropic>,\n  amplitude: AmplitudeOrAI,\n  opts?: WrapOpts,\n): AmpAnthropic;\nexport function wrap(\n  client: unknown,\n  amplitude: AmplitudeOrAI,\n  opts?: WrapOpts,\n): AmpOpenAI | AmpAzureOpenAI | AmpAnthropic;\nexport function wrap(\n  client: unknown,\n  amplitude: AmplitudeOrAI,\n  opts?: WrapOpts,\n): AmpOpenAI | AmpAzureOpenAI | AmpAnthropic {\n  const clientObj = client as Record<string, unknown>;\n  const clientConstructor = (client as { constructor?: { name?: string } })\n    ?.constructor;\n  const clientName = clientConstructor?.name ?? 'unknown';\n  const explicitModule = opts?.providerModule as\n    | Record<string, unknown>\n    | undefined;\n\n  // OpenAI check — use explicit module only if it exports OpenAI\n  const openaiModule =\n    explicitModule && 'OpenAI' in explicitModule\n      ? explicitModule\n      : tryRequire('openai');\n  if (openaiModule != null) {\n    const OpenAIClass = openaiModule.OpenAI as\n      | (new (...args: unknown[]) => unknown)\n      | undefined;\n    const AzureOpenAIClass = openaiModule.AzureOpenAI as\n      | (new (...args: unknown[]) => unknown)\n      | undefined;\n\n    if (AzureOpenAIClass && client instanceof AzureOpenAIClass) {\n      return new AmpAzureOpenAI({\n        amplitude,\n        apiKey: clientObj.apiKey as string | undefined,\n        azureEndpoint: String(clientObj.baseURL ?? ''),\n        propagateContext: opts?.propagateContext as boolean | undefined,\n        openaiModule,\n      });\n    }\n\n    if (OpenAIClass && client instanceof OpenAIClass) {\n      return new AmpOpenAI({\n        amplitude,\n        apiKey: clientObj.apiKey as string | undefined,\n        baseUrl:\n          clientObj.baseURL != null ? String(clientObj.baseURL) : undefined,\n        propagateContext: opts?.propagateContext as boolean | undefined,\n        openaiModule,\n      });\n    }\n  }\n\n  // Anthropic check — use explicit module only if it exports Anthropic\n  const anthropicModule =\n    explicitModule && 'Anthropic' in explicitModule\n      ? explicitModule\n      : tryRequire('@anthropic-ai/sdk');\n  if (anthropicModule != null) {\n    const AnthropicClass = anthropicModule.Anthropic as\n      | (new (...args: unknown[]) => unknown)\n      | undefined;\n\n    if (AnthropicClass && client instanceof AnthropicClass) {\n      return new AmpAnthropic({\n        amplitude,\n        apiKey: clientObj.apiKey as string | undefined,\n        propagateContext: opts?.propagateContext as boolean | undefined,\n        anthropicModule,\n      });\n    }\n  }\n\n  const supported =\n    'openai.OpenAI, openai.AzureOpenAI, @anthropic-ai/sdk.Anthropic';\n  throw new AmplitudeAIWrapError(\n    `wrap() does not support ${clientName}. Supported types: ${supported}. For Gemini, Bedrock, and Mistral, use the SDK's provider classes directly.`,\n  );\n}\n"],"mappings":";;;;;;;;;;;;;;;;AAiBA,IAAa,uBAAb,cAA0C,iBAAiB;CACzD,YAAY,SAAiB;AAC3B,QAAM,QAAQ;AACd,OAAK,OAAO;;;AA0ChB,SAAgB,KACd,QACA,WACA,MAC2C;CAC3C,MAAM,YAAY;CAGlB,MAAM,cAFqB,QACvB,cACkC,QAAQ;CAC9C,MAAM,iBAAiB,MAAM;CAK7B,MAAM,eACJ,kBAAkB,YAAY,iBAC1B,iBACA,WAAW,SAAS;AAC1B,KAAI,gBAAgB,MAAM;EACxB,MAAM,cAAc,aAAa;EAGjC,MAAM,mBAAmB,aAAa;AAItC,MAAI,oBAAoB,kBAAkB,iBACxC,QAAO,IAAIA,YAAe;GACxB;GACA,QAAQ,UAAU;GAClB,eAAe,OAAO,UAAU,WAAW,GAAG;GAC9C,kBAAkB,MAAM;GACxB;GACD,CAAC;AAGJ,MAAI,eAAe,kBAAkB,YACnC,QAAO,IAAIC,OAAU;GACnB;GACA,QAAQ,UAAU;GAClB,SACE,UAAU,WAAW,OAAO,OAAO,UAAU,QAAQ,GAAG;GAC1D,kBAAkB,MAAM;GACxB;GACD,CAAC;;CAKN,MAAM,kBACJ,kBAAkB,eAAe,iBAC7B,iBACA,WAAW,oBAAoB;AACrC,KAAI,mBAAmB,MAAM;EAC3B,MAAM,iBAAiB,gBAAgB;AAIvC,MAAI,kBAAkB,kBAAkB,eACtC,QAAO,IAAIC,UAAa;GACtB;GACA,QAAQ,UAAU;GAClB,kBAAkB,MAAM;GACxB;GACD,CAAC;;AAMN,OAAM,IAAI,qBACR,2BAA2B,WAAW,+JACvC"}