@amplitude/ai 0.1.1 → 0.1.2

package/AGENTS.md

@@ -2,7 +2,7 @@
 
 # AGENTS.md
 
-Package: `@amplitude/ai` v0.1.1
+Package: `@amplitude/ai` v0.1.2
 
 ## Install
 

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

@@ -1 +1 @@
-{"version":3,"file":"anthropic.d.ts","names":[],"sources":["../../src/providers/anthropic.ts"],"sourcesContent":[],"mappings":";;;;;;AAwBM,cADO,mBACW,EAAA,OAAM;AAI9B,cAJM,gBAI2B,EAJT,MAKX,CAAA,MAAA,EAAA,OAEK,CAAA,GAAA,IAAA;AAOA,UAVD,gBAAA,CAUC;EAA0B,SAAA,EAT/B,aAS+B;EAGvB,MAAA,CAAA,EAAA,MAAA;EAGE,aAAA,CAAA,EAbL,aAaK,GAAA,IAAA;EAmCP,gBAAA,CAAA,EAAA,OAAA;EAxCN;EAAc,eAAA,CAAA,EAAA,OAAA;AA6CxB;AACa,cAhDA,SAgDA,CAAA,gBA/CK,MA+CL,CAAA,MAAA,EAAA,OAAA,CAAA,GA/C+B,MA+C/B,CAAA,MAAA,EAAA,OAAA,CAAA,CAAA,SA9CH,cAAA,CA8CG;EAQA,QAAA,OAAA;EACE,SAAA,QAAA,EArDM,eAqDN;EACI,QAAA,iBAAA;EAeP,WAAA,CAAA,OAAA,EAlEW,gBAkEX;EACa,IAAA,MAAA,CAAA,CAAA,EAhCT,OAgCS;;AACQ,cA5BpB,eAAA,CA4BoB;EAA5B,SAAA,EA3BQ,MA2BR,CAAA,MAAA,EAAA,OAAA,CAAA;EAAO,QAAA,QAAA;EA+SI,QAAA,UAAA;EAmBA,QAAA,cAAA;EACC,QAAA,iBAAA;EAAN,WAAA,CAAA,MAAA,EAAA,OAAA,EAAA,OAAA,EAtVE,OAsVF,EAAA,SAAA,EArVI,aAqVJ,EAAA,aAAA,EApVQ,aAoVR,GAAA,IAAA,EAAA,gBAAA,EAAA,OAAA;EAIQ,MAAA,CAAA,MAAA,EAzUP,MAyUO,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,kBAAA,CAAA,EAxUM,oBAwUN,CAAA,EAvUd,OAuUc,CAvUN,iBAuUM,GAvUc,aAuUd,CAAA,OAAA,CAAA,CAAA;EAAN,QAAA,WAAA;EAAK,QAAA,mBAAA;;;iBAxBF,4BAAA;iBAmBA,uBAAA,UACL,MAAM;;;aAIJ,MAAM"}
+{"version":3,"file":"anthropic.d.ts","names":[],"sources":["../../src/providers/anthropic.ts"],"sourcesContent":[],"mappings":";;;;;;AAyBM,cADO,mBACW,EAAA,OAAM;AAI9B,cAJM,gBAI2B,EAJT,MAKX,CAAA,MAAA,EAAA,OAEK,CAAA,GAAA,IAAA;AAOA,UAVD,gBAAA,CAUC;EAA0B,SAAA,EAT/B,aAS+B;EAGvB,MAAA,CAAA,EAAA,MAAA;EAGE,aAAA,CAAA,EAbL,aAaK,GAAA,IAAA;EAmCP,gBAAA,CAAA,EAAA,OAAA;EAxCN;EAAc,eAAA,CAAA,EAAA,OAAA;AA6CxB;AACa,cAhDA,SAgDA,CAAA,gBA/CK,MA+CL,CAAA,MAAA,EAAA,OAAA,CAAA,GA/C+B,MA+C/B,CAAA,MAAA,EAAA,OAAA,CAAA,CAAA,SA9CH,cAAA,CA8CG;EAQA,QAAA,OAAA;EACE,SAAA,QAAA,EArDM,eAqDN;EACI,QAAA,iBAAA;EAeP,WAAA,CAAA,OAAA,EAlEW,gBAkEX;EACa,IAAA,MAAA,CAAA,CAAA,EAhCT,OAgCS;;AACQ,cA5BpB,eAAA,CA4BoB;EAA5B,SAAA,EA3BQ,MA2BR,CAAA,MAAA,EAAA,OAAA,CAAA;EAAO,QAAA,QAAA;EAiSI,QAAA,UAAA;EAmBA,QAAA,cAAA;EACC,QAAA,iBAAA;EAAN,WAAA,CAAA,MAAA,EAAA,OAAA,EAAA,OAAA,EAxUE,OAwUF,EAAA,SAAA,EAvUI,aAuUJ,EAAA,aAAA,EAtUQ,aAsUR,GAAA,IAAA,EAAA,gBAAA,EAAA,OAAA;EAIQ,MAAA,CAAA,MAAA,EA3TP,MA2TO,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,kBAAA,CAAA,EA1TM,oBA0TN,CAAA,EAzTd,OAyTc,CAzTN,iBAyTM,GAzTc,aAyTd,CAAA,OAAA,CAAA,CAAA;EAAN,QAAA,WAAA;EAAK,QAAA,mBAAA;;;iBAxBF,4BAAA;iBAmBA,uBAAA,UACL,MAAM;;;aAIJ,MAAM"}

package/dist/providers/anthropic.js

@@ -3,7 +3,7 @@ import { getDefaultPropagateContext, injectContext } from "../propagation.js";
 import { tryRequire } from "../utils/resolve-module.js";
 import { calculateCost } from "../utils/costs.js";
 import { StreamingAccumulator } from "../utils/streaming.js";
-import { BaseAIProvider, applySessionContext } from "./base.js";
+import { BaseAIProvider, applySessionContext, contextFields } from "./base.js";
 
 //#region src/providers/anthropic.ts
 const _resolved = tryRequire("@anthropic-ai/sdk");
@@ -75,17 +75,12 @@ var WrappedMessages = class {
 				});
 			} catch {}
 			this._trackFn({
-				userId: ctx.userId ?? "unknown",
+				...contextFields(ctx),
 				modelName,
 				provider: "anthropic",
 				responseContent: String(firstTextBlock?.text ?? ""),
 				reasoningContent: extracted.reasoning,
 				latencyMs,
-				sessionId: ctx.sessionId,
-				traceId: ctx.traceId,
-				turnId: ctx.turnId ?? void 0,
-				agentId: ctx.agentId,
-				env: ctx.env,
 				inputTokens: normalizedInput || void 0,
 				outputTokens: usage?.output_tokens,
 				cacheReadInputTokens: cacheRead || void 0,
@@ -103,15 +98,11 @@ var WrappedMessages = class {
 		} catch (error) {
 			const latencyMs = performance.now() - startTime;
 			this._trackFn({
-				userId: ctx.userId ?? "unknown",
+				...contextFields(ctx),
 				modelName: String(requestParams.model ?? "unknown"),
 				provider: "anthropic",
 				responseContent: "",
 				latencyMs,
-				sessionId: ctx.sessionId,
-				traceId: ctx.traceId,
-				agentId: ctx.agentId,
-				env: ctx.env,
 				isError: true,
 				errorMessage: error instanceof Error ? error.message : String(error)
 			});
@@ -178,16 +169,11 @@ var WrappedMessages = class {
 				});
 			} catch {}
 			this._trackFn({
-				userId: sessionCtx.userId ?? "unknown",
+				...contextFields(sessionCtx),
 				modelName,
 				provider: "anthropic",
 				responseContent: state.content,
 				latencyMs: accumulator.elapsedMs,
-				sessionId: sessionCtx.sessionId,
-				traceId: sessionCtx.traceId,
-				turnId: sessionCtx.turnId ?? void 0,
-				agentId: sessionCtx.agentId,
-				env: sessionCtx.env,
 				inputTokens: streamNormalizedInput || void 0,
 				outputTokens: state.outputTokens,
 				cacheReadInputTokens: streamCacheRead || void 0,

package/dist/providers/anthropic.js.map

@@ -1 +1 @@
-{"version":3,"file":"anthropic.js","names":["_AnthropicModule: Record<string, unknown> | null","clientOpts: Record<string, unknown>","costUsd: number | null","reasoning: string | undefined","toolCalls: Array<Record<string, unknown>>"],"sources":["../../src/providers/anthropic.ts"],"sourcesContent":["/**\n * Anthropic provider wrapper with automatic tracking.\n */\n\nimport type { PrivacyConfig } from '../core/privacy.js';\nimport { trackUserMessage } from '../core/tracking.js';\nimport { getDefaultPropagateContext, injectContext } from '../propagation.js';\nimport type {\n  AmplitudeLike,\n  AmplitudeOrAI,\n  AnthropicResponse,\n  TrackFn,\n} from '../types.js';\nimport { calculateCost } from '../utils/costs.js';\nimport { tryRequire } from '../utils/resolve-module.js';\nimport { StreamingAccumulator } from '../utils/streaming.js';\nimport {\n  applySessionContext,\n  BaseAIProvider,\n  type ProviderTrackOptions,\n} from './base.js';\n\nconst _resolved = tryRequire('@anthropic-ai/sdk');\nexport const ANTHROPIC_AVAILABLE = _resolved != null;\nconst _AnthropicModule: Record<string, unknown> | null = _resolved;\n\nexport { _AnthropicModule };\n\nexport interface AnthropicOptions {\n  amplitude: AmplitudeOrAI;\n  apiKey?: string;\n  privacyConfig?: PrivacyConfig | null;\n  propagateContext?: boolean;\n  /** Pass the `@anthropic-ai/sdk` module directly to bypass `tryRequire` (required in bundler environments). */\n  anthropicModule?: unknown;\n}\n\nexport class Anthropic<\n  TClient extends Record<string, unknown> = Record<string, unknown>,\n> extends BaseAIProvider {\n  private _client: TClient;\n  readonly messages: WrappedMessages;\n  private _propagateContext: boolean;\n\n  constructor(options: AnthropicOptions) {\n    super({\n      amplitude: options.amplitude,\n      privacyConfig: options.privacyConfig,\n      providerName: 'anthropic',\n    });\n\n    const mod =\n      (options.anthropicModule as Record<string, unknown> | null) ??\n      _AnthropicModule;\n    if (mod == null) {\n      throw new Error(\n        '@anthropic-ai/sdk package is required. Install it with: npm install @anthropic-ai/sdk — or pass the module directly via the anthropicModule option.',\n      );\n    }\n\n    const AnthropicSDK = mod.Anthropic as new (\n      opts: Record<string, unknown>,\n    ) => unknown;\n\n    const clientOpts: Record<string, unknown> = {};\n    if (options.apiKey) clientOpts.apiKey = options.apiKey;\n\n    this._client = new AnthropicSDK(clientOpts) as TClient;\n    this._propagateContext =\n      options.propagateContext ?? getDefaultPropagateContext();\n    this.messages = new WrappedMessages(\n      this._client,\n      this.trackFn(),\n      this._amplitude,\n      this._privacyConfig,\n      this._propagateContext,\n    );\n  }\n\n  get client(): TClient {\n    return this._client;\n  }\n}\n\nexport class WrappedMessages {\n  _original: Record<string, unknown>;\n  private _trackFn: TrackFn;\n  private _amplitude: AmplitudeLike;\n  private _privacyConfig: PrivacyConfig | null;\n  private _propagateContext: boolean;\n\n  constructor(\n    client: unknown,\n    trackFn: TrackFn,\n    amplitude: AmplitudeLike,\n    privacyConfig: PrivacyConfig | null,\n    propagateContext: boolean,\n  ) {\n    const clientObj = client as Record<string, unknown>;\n    this._original = clientObj.messages as Record<string, unknown>;\n    this._trackFn =\n      typeof trackFn === 'function'\n        ? trackFn\n        : (trackFn as unknown as { trackFn(): TrackFn }).trackFn();\n    this._amplitude = amplitude;\n    this._privacyConfig = privacyConfig;\n    this._propagateContext = propagateContext;\n  }\n\n  async create(\n    params: Record<string, unknown>,\n    amplitudeOverrides?: ProviderTrackOptions,\n  ): Promise<AnthropicResponse | AsyncIterable<unknown>> {\n    const createFn = this._original.create as (\n      ...args: unknown[]\n    ) => Promise<unknown>;\n    const startTime = performance.now();\n    const requestParams = this._withContextHeaders(params);\n    const ctx = applySessionContext(amplitudeOverrides);\n\n    try {\n      this._trackInputMessages(\n        requestParams.messages as unknown,\n        ctx,\n        amplitudeOverrides?.trackInputMessages ?? true,\n      );\n      const response = await createFn.call(this._original, requestParams);\n\n      if (requestParams.stream === true && _isAsyncIterable(response)) {\n        return this._wrapStream(\n          response as AsyncIterable<unknown>,\n          requestParams,\n          startTime,\n          ctx,\n        );\n      }\n\n      const latencyMs = performance.now() - startTime;\n\n      const resp = response as AnthropicResponse;\n      const usage = resp.usage;\n      const extracted = extractAnthropicContent(\n        resp.content as unknown as Array<Record<string, unknown>> | undefined,\n      );\n      const firstTextBlock = resp.content?.find((b) => b.type === 'text');\n      const modelName = String(resp.model ?? requestParams.model ?? 'unknown');\n\n      const cacheRead = usage?.cache_read_input_tokens ?? 0;\n      const cacheCreation = usage?.cache_creation_input_tokens ?? 0;\n      const rawInput = usage?.input_tokens ?? 0;\n      const normalizedInput =\n        cacheRead || cacheCreation\n          ? rawInput + cacheRead + cacheCreation\n          : rawInput;\n\n      let costUsd: number | null = null;\n      if (usage?.input_tokens != null && usage?.output_tokens != null) {\n        try {\n          costUsd = calculateCost({\n            modelName,\n            inputTokens: normalizedInput,\n            outputTokens: usage.output_tokens,\n            cacheReadInputTokens: cacheRead,\n            cacheCreationInputTokens: cacheCreation,\n          });\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      this._trackFn({\n        userId: ctx.userId ?? 'unknown',\n        modelName,\n        provider: 'anthropic',\n        responseContent: String(firstTextBlock?.text ?? ''),\n        reasoningContent: extracted.reasoning,\n        latencyMs,\n        sessionId: ctx.sessionId,\n        traceId: ctx.traceId,\n        turnId: ctx.turnId ?? undefined,\n        agentId: ctx.agentId,\n        env: ctx.env,\n        inputTokens: normalizedInput || undefined,\n        outputTokens: usage?.output_tokens,\n        cacheReadInputTokens: cacheRead || undefined,\n        cacheCreationInputTokens: cacheCreation || undefined,\n        totalCostUsd: costUsd,\n        finishReason: resp.stop_reason,\n        toolCalls:\n          extracted.toolCalls.length > 0 ? extracted.toolCalls : undefined,\n        isStreaming: false,\n        systemPrompt: extractAnthropicSystemPrompt(requestParams.system),\n        temperature: requestParams.temperature as number | undefined,\n        maxOutputTokens: requestParams.max_tokens as number | undefined,\n        topP: requestParams.top_p as number | undefined,\n      });\n\n      return response as AnthropicResponse;\n    } catch (error) {\n      const latencyMs = performance.now() - startTime;\n      this._trackFn({\n        userId: ctx.userId ?? 'unknown',\n        modelName: String(requestParams.model ?? 'unknown'),\n        provider: 'anthropic',\n        responseContent: '',\n        latencyMs,\n        sessionId: ctx.sessionId,\n        traceId: ctx.traceId,\n        agentId: ctx.agentId,\n        env: ctx.env,\n        isError: true,\n        errorMessage: error instanceof Error ? error.message : String(error),\n      });\n\n      throw error;\n    }\n  }\n\n  private async *_wrapStream(\n    stream: AsyncIterable<unknown>,\n    params: Record<string, unknown>,\n    _startTime: number,\n    sessionCtx: ProviderTrackOptions,\n  ): AsyncGenerator<unknown> {\n    const accumulator = new StreamingAccumulator();\n    accumulator.model = String(params.model ?? 'unknown');\n    let reasoningContent = '';\n\n    try {\n      for await (const event of stream) {\n        const evt = event as Record<string, unknown>;\n        const type = evt.type as string | undefined;\n\n        if (type === 'content_block_delta') {\n          const delta = evt.delta as Record<string, unknown> | undefined;\n          if (delta?.type === 'text_delta' && delta.text != null) {\n            accumulator.addContent(String(delta.text));\n          } else if (\n            delta?.type === 'thinking_delta' &&\n            delta.thinking != null\n          ) {\n            reasoningContent += String(delta.thinking);\n          }\n        } else if (type === 'content_block_start') {\n          const block = evt.content_block as\n            | Record<string, unknown>\n            | undefined;\n          if (block?.type === 'tool_use') {\n            accumulator.addToolCall({\n              type: 'function',\n              id: block.id,\n              function: {\n                name: String(block.name ?? ''),\n                arguments:\n                  typeof block.input === 'string'\n                    ? block.input\n                    : JSON.stringify(block.input ?? {}),\n              },\n            });\n          }\n        } else if (type === 'message_delta') {\n          const delta = evt.delta as Record<string, unknown> | undefined;\n          if (delta?.stop_reason != null) {\n            accumulator.finishReason = String(delta.stop_reason);\n          }\n          const usage = evt.usage as Record<string, number> | undefined;\n          if (usage != null) {\n            accumulator.setUsage({\n              outputTokens: usage.output_tokens,\n            });\n          }\n        } else if (type === 'message_start') {\n          const message = evt.message as Record<string, unknown> | undefined;\n          if (message?.model != null) {\n            accumulator.model = String(message.model);\n          }\n          const usage = message?.usage as Record<string, number> | undefined;\n          if (usage != null) {\n            accumulator.setUsage({\n              inputTokens: usage.input_tokens,\n              cacheReadTokens: usage.cache_read_input_tokens,\n              cacheCreationTokens: usage.cache_creation_input_tokens,\n            });\n          }\n        }\n\n        yield event;\n      }\n    } catch (error) {\n      accumulator.setError(\n        error instanceof Error ? error.message : String(error),\n      );\n      throw error;\n    } finally {\n      const state = accumulator.getState();\n      const modelName = String(accumulator.model ?? params.model ?? 'unknown');\n\n      const streamCacheRead = state.cacheReadTokens ?? 0;\n      const streamCacheCreation = state.cacheCreationTokens ?? 0;\n      const streamRawInput = state.inputTokens ?? 0;\n      const streamNormalizedInput =\n        streamCacheRead || streamCacheCreation\n          ? streamRawInput + streamCacheRead + streamCacheCreation\n          : streamRawInput;\n\n      let costUsd: number | null = null;\n      if (state.inputTokens != null && state.outputTokens != null) {\n        try {\n          costUsd = calculateCost({\n            modelName,\n            inputTokens: streamNormalizedInput,\n            outputTokens: state.outputTokens,\n            cacheReadInputTokens: streamCacheRead,\n            cacheCreationInputTokens: streamCacheCreation,\n          });\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      this._trackFn({\n        userId: sessionCtx.userId ?? 'unknown',\n        modelName,\n        provider: 'anthropic',\n        responseContent: state.content,\n        latencyMs: accumulator.elapsedMs,\n        sessionId: sessionCtx.sessionId,\n        traceId: sessionCtx.traceId,\n        turnId: sessionCtx.turnId ?? undefined,\n        agentId: sessionCtx.agentId,\n        env: sessionCtx.env,\n        inputTokens: streamNormalizedInput || undefined,\n        outputTokens: state.outputTokens,\n        cacheReadInputTokens: streamCacheRead || undefined,\n        cacheCreationInputTokens: streamCacheCreation || undefined,\n        totalCostUsd: costUsd,\n        finishReason: state.finishReason,\n        toolCalls: state.toolCalls.length > 0 ? state.toolCalls : undefined,\n        providerTtfbMs: state.ttfbMs,\n        isStreaming: true,\n        isError: state.isError,\n        errorMessage: state.errorMessage,\n        reasoningContent: reasoningContent || undefined,\n        systemPrompt: extractAnthropicSystemPrompt(params.system),\n        temperature: params.temperature as number | undefined,\n        maxOutputTokens: params.max_tokens as number | undefined,\n        topP: params.top_p as number | undefined,\n      });\n    }\n  }\n\n  private _withContextHeaders(\n    params: Record<string, unknown>,\n  ): Record<string, unknown> {\n    if (!this._propagateContext) return params;\n    const existing = (params.extra_headers ?? params.headers) as\n      | Record<string, string>\n      | undefined;\n    const injected = injectContext(existing);\n    return {\n      ...params,\n      extra_headers: injected,\n    };\n  }\n\n  private _trackInputMessages(\n    messages: unknown,\n    ctx: ProviderTrackOptions,\n    shouldTrackInputMessages: boolean,\n  ): void {\n    if (!shouldTrackInputMessages) return;\n    if (ctx.userId == null || ctx.sessionId == null) return;\n    if (!Array.isArray(messages)) return;\n    for (const msg of messages) {\n      const role = (msg as Record<string, unknown>)?.role;\n      if (role !== 'user') continue;\n      const rawContent = (msg as Record<string, unknown>)?.content;\n      const content = Array.isArray(rawContent)\n        ? rawContent\n            .map((part) => {\n              if (typeof part === 'string') return part;\n              const text = (part as Record<string, unknown>)?.text;\n              return typeof text === 'string' ? text : '';\n            })\n            .join('')\n        : typeof rawContent === 'string'\n          ? rawContent\n          : '';\n      if (!content) continue;\n      trackUserMessage({\n        amplitude: this._amplitude,\n        userId: ctx.userId,\n        messageContent: content,\n        sessionId: ctx.sessionId,\n        traceId: ctx.traceId,\n        turnId: ctx.turnId ?? undefined,\n        agentId: ctx.agentId,\n        parentAgentId: ctx.parentAgentId,\n        customerOrgId: ctx.customerOrgId,\n        env: ctx.env,\n        groups: ctx.groups,\n        privacyConfig: this._privacyConfig,\n      });\n    }\n  }\n}\n\nfunction _isAsyncIterable(value: unknown): value is AsyncIterable<unknown> {\n  return (\n    value != null &&\n    typeof (value as Record<symbol, unknown>)[Symbol.asyncIterator] ===\n      'function'\n  );\n}\n\nexport function extractAnthropicSystemPrompt(\n  system: unknown,\n): string | undefined {\n  if (typeof system === 'string') return system;\n  if (Array.isArray(system)) {\n    return system\n      .map((block) => {\n        if (typeof block === 'string') return block;\n        if (typeof block === 'object' && block != null) {\n          return String((block as Record<string, unknown>).text ?? '');\n        }\n        return '';\n      })\n      .filter(Boolean)\n      .join('\\n');\n  }\n  return undefined;\n}\n\nexport function extractAnthropicContent(\n  content: Array<Record<string, unknown>> | undefined,\n): {\n  text: string;\n  reasoning: string | undefined;\n  toolCalls: Array<Record<string, unknown>>;\n} {\n  let text = '';\n  let reasoning: string | undefined;\n  const toolCalls: Array<Record<string, unknown>> = [];\n\n  if (!content) return { text, reasoning, toolCalls };\n\n  for (const block of content) {\n    if (block.type === 'text') {\n      text += String(block.text ?? '');\n    } else if (block.type === 'thinking') {\n      reasoning = String(block.thinking ?? '');\n    } else if (block.type === 'tool_use') {\n      toolCalls.push({\n        type: 'function',\n        id: block.id,\n        function: {\n          name: String(block.name ?? ''),\n          arguments:\n            typeof block.input === 'string'\n              ? block.input\n              : JSON.stringify(block.input ?? {}),\n        },\n      });\n    }\n  }\n\n  return { text, reasoning, toolCalls };\n}\n"],"mappings":";;;;;;;;AAsBA,MAAM,YAAY,WAAW,oBAAoB;AACjD,MAAa,sBAAsB,aAAa;AAChD,MAAMA,mBAAmD;AAazD,IAAa,YAAb,cAEU,eAAe;CACvB,AAAQ;CACR,AAAS;CACT,AAAQ;CAER,YAAY,SAA2B;AACrC,QAAM;GACJ,WAAW,QAAQ;GACnB,eAAe,QAAQ;GACvB,cAAc;GACf,CAAC;EAEF,MAAM,MACH,QAAQ,mBACT;AACF,MAAI,OAAO,KACT,OAAM,IAAI,MACR,sJACD;EAGH,MAAM,eAAe,IAAI;EAIzB,MAAMC,aAAsC,EAAE;AAC9C,MAAI,QAAQ,OAAQ,YAAW,SAAS,QAAQ;AAEhD,OAAK,UAAU,IAAI,aAAa,WAAW;AAC3C,OAAK,oBACH,QAAQ,oBAAoB,4BAA4B;AAC1D,OAAK,WAAW,IAAI,gBAClB,KAAK,SACL,KAAK,SAAS,EACd,KAAK,YACL,KAAK,gBACL,KAAK,kBACN;;CAGH,IAAI,SAAkB;AACpB,SAAO,KAAK;;;AAIhB,IAAa,kBAAb,MAA6B;CAC3B;CACA,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CAER,YACE,QACA,SACA,WACA,eACA,kBACA;AAEA,OAAK,YADa,OACS;AAC3B,OAAK,WACH,OAAO,YAAY,aACf,UACC,QAA8C,SAAS;AAC9D,OAAK,aAAa;AAClB,OAAK,iBAAiB;AACtB,OAAK,oBAAoB;;CAG3B,MAAM,OACJ,QACA,oBACqD;EACrD,MAAM,WAAW,KAAK,UAAU;EAGhC,MAAM,YAAY,YAAY,KAAK;EACnC,MAAM,gBAAgB,KAAK,oBAAoB,OAAO;EACtD,MAAM,MAAM,oBAAoB,mBAAmB;AAEnD,MAAI;AACF,QAAK,oBACH,cAAc,UACd,KACA,oBAAoB,sBAAsB,KAC3C;GACD,MAAM,WAAW,MAAM,SAAS,KAAK,KAAK,WAAW,cAAc;AAEnE,OAAI,cAAc,WAAW,QAAQ,iBAAiB,SAAS,CAC7D,QAAO,KAAK,YACV,UACA,eACA,WACA,IACD;GAGH,MAAM,YAAY,YAAY,KAAK,GAAG;GAEtC,MAAM,OAAO;GACb,MAAM,QAAQ,KAAK;GACnB,MAAM,YAAY,wBAChB,KAAK,QACN;GACD,MAAM,iBAAiB,KAAK,SAAS,MAAM,MAAM,EAAE,SAAS,OAAO;GACnE,MAAM,YAAY,OAAO,KAAK,SAAS,cAAc,SAAS,UAAU;GAExE,MAAM,YAAY,OAAO,2BAA2B;GACpD,MAAM,gBAAgB,OAAO,+BAA+B;GAC5D,MAAM,WAAW,OAAO,gBAAgB;GACxC,MAAM,kBACJ,aAAa,gBACT,WAAW,YAAY,gBACvB;GAEN,IAAIC,UAAyB;AAC7B,OAAI,OAAO,gBAAgB,QAAQ,OAAO,iBAAiB,KACzD,KAAI;AACF,cAAU,cAAc;KACtB;KACA,aAAa;KACb,cAAc,MAAM;KACpB,sBAAsB;KACtB,0BAA0B;KAC3B,CAAC;WACI;AAKV,QAAK,SAAS;IACZ,QAAQ,IAAI,UAAU;IACtB;IACA,UAAU;IACV,iBAAiB,OAAO,gBAAgB,QAAQ,GAAG;IACnD,kBAAkB,UAAU;IAC5B;IACA,WAAW,IAAI;IACf,SAAS,IAAI;IACb,QAAQ,IAAI,UAAU;IACtB,SAAS,IAAI;IACb,KAAK,IAAI;IACT,aAAa,mBAAmB;IAChC,cAAc,OAAO;IACrB,sBAAsB,aAAa;IACnC,0BAA0B,iBAAiB;IAC3C,cAAc;IACd,cAAc,KAAK;IACnB,WACE,UAAU,UAAU,SAAS,IAAI,UAAU,YAAY;IACzD,aAAa;IACb,cAAc,6BAA6B,cAAc,OAAO;IAChE,aAAa,cAAc;IAC3B,iBAAiB,cAAc;IAC/B,MAAM,cAAc;IACrB,CAAC;AAEF,UAAO;WACA,OAAO;GACd,MAAM,YAAY,YAAY,KAAK,GAAG;AACtC,QAAK,SAAS;IACZ,QAAQ,IAAI,UAAU;IACtB,WAAW,OAAO,cAAc,SAAS,UAAU;IACnD,UAAU;IACV,iBAAiB;IACjB;IACA,WAAW,IAAI;IACf,SAAS,IAAI;IACb,SAAS,IAAI;IACb,KAAK,IAAI;IACT,SAAS;IACT,cAAc,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IACrE,CAAC;AAEF,SAAM;;;CAIV,OAAe,YACb,QACA,QACA,YACA,YACyB;EACzB,MAAM,cAAc,IAAI,sBAAsB;AAC9C,cAAY,QAAQ,OAAO,OAAO,SAAS,UAAU;EACrD,IAAI,mBAAmB;AAEvB,MAAI;AACF,cAAW,MAAM,SAAS,QAAQ;IAChC,MAAM,MAAM;IACZ,MAAM,OAAO,IAAI;AAEjB,QAAI,SAAS,uBAAuB;KAClC,MAAM,QAAQ,IAAI;AAClB,SAAI,OAAO,SAAS,gBAAgB,MAAM,QAAQ,KAChD,aAAY,WAAW,OAAO,MAAM,KAAK,CAAC;cAE1C,OAAO,SAAS,oBAChB,MAAM,YAAY,KAElB,qBAAoB,OAAO,MAAM,SAAS;eAEnC,SAAS,uBAAuB;KACzC,MAAM,QAAQ,IAAI;AAGlB,SAAI,OAAO,SAAS,WAClB,aAAY,YAAY;MACtB,MAAM;MACN,IAAI,MAAM;MACV,UAAU;OACR,MAAM,OAAO,MAAM,QAAQ,GAAG;OAC9B,WACE,OAAO,MAAM,UAAU,WACnB,MAAM,QACN,KAAK,UAAU,MAAM,SAAS,EAAE,CAAC;OACxC;MACF,CAAC;eAEK,SAAS,iBAAiB;KACnC,MAAM,QAAQ,IAAI;AAClB,SAAI,OAAO,eAAe,KACxB,aAAY,eAAe,OAAO,MAAM,YAAY;KAEtD,MAAM,QAAQ,IAAI;AAClB,SAAI,SAAS,KACX,aAAY,SAAS,EACnB,cAAc,MAAM,eACrB,CAAC;eAEK,SAAS,iBAAiB;KACnC,MAAM,UAAU,IAAI;AACpB,SAAI,SAAS,SAAS,KACpB,aAAY,QAAQ,OAAO,QAAQ,MAAM;KAE3C,MAAM,QAAQ,SAAS;AACvB,SAAI,SAAS,KACX,aAAY,SAAS;MACnB,aAAa,MAAM;MACnB,iBAAiB,MAAM;MACvB,qBAAqB,MAAM;MAC5B,CAAC;;AAIN,UAAM;;WAED,OAAO;AACd,eAAY,SACV,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,CACvD;AACD,SAAM;YACE;GACR,MAAM,QAAQ,YAAY,UAAU;GACpC,MAAM,YAAY,OAAO,YAAY,SAAS,OAAO,SAAS,UAAU;GAExE,MAAM,kBAAkB,MAAM,mBAAmB;GACjD,MAAM,sBAAsB,MAAM,uBAAuB;GACzD,MAAM,iBAAiB,MAAM,eAAe;GAC5C,MAAM,wBACJ,mBAAmB,sBACf,iBAAiB,kBAAkB,sBACnC;GAEN,IAAIA,UAAyB;AAC7B,OAAI,MAAM,eAAe,QAAQ,MAAM,gBAAgB,KACrD,KAAI;AACF,cAAU,cAAc;KACtB;KACA,aAAa;KACb,cAAc,MAAM;KACpB,sBAAsB;KACtB,0BAA0B;KAC3B,CAAC;WACI;AAKV,QAAK,SAAS;IACZ,QAAQ,WAAW,UAAU;IAC7B;IACA,UAAU;IACV,iBAAiB,MAAM;IACvB,WAAW,YAAY;IACvB,WAAW,WAAW;IACtB,SAAS,WAAW;IACpB,QAAQ,WAAW,UAAU;IAC7B,SAAS,WAAW;IACpB,KAAK,WAAW;IAChB,aAAa,yBAAyB;IACtC,cAAc,MAAM;IACpB,sBAAsB,mBAAmB;IACzC,0BAA0B,uBAAuB;IACjD,cAAc;IACd,cAAc,MAAM;IACpB,WAAW,MAAM,UAAU,SAAS,IAAI,MAAM,YAAY;IAC1D,gBAAgB,MAAM;IACtB,aAAa;IACb,SAAS,MAAM;IACf,cAAc,MAAM;IACpB,kBAAkB,oBAAoB;IACtC,cAAc,6BAA6B,OAAO,OAAO;IACzD,aAAa,OAAO;IACpB,iBAAiB,OAAO;IACxB,MAAM,OAAO;IACd,CAAC;;;CAIN,AAAQ,oBACN,QACyB;AACzB,MAAI,CAAC,KAAK,kBAAmB,QAAO;EAIpC,MAAM,WAAW,cAHC,OAAO,iBAAiB,OAAO,QAGT;AACxC,SAAO;GACL,GAAG;GACH,eAAe;GAChB;;CAGH,AAAQ,oBACN,UACA,KACA,0BACM;AACN,MAAI,CAAC,yBAA0B;AAC/B,MAAI,IAAI,UAAU,QAAQ,IAAI,aAAa,KAAM;AACjD,MAAI,CAAC,MAAM,QAAQ,SAAS,CAAE;AAC9B,OAAK,MAAM,OAAO,UAAU;AAE1B,OADc,KAAiC,SAClC,OAAQ;GACrB,MAAM,aAAc,KAAiC;GACrD,MAAM,UAAU,MAAM,QAAQ,WAAW,GACrC,WACG,KAAK,SAAS;AACb,QAAI,OAAO,SAAS,SAAU,QAAO;IACrC,MAAM,OAAQ,MAAkC;AAChD,WAAO,OAAO,SAAS,WAAW,OAAO;KACzC,CACD,KAAK,GAAG,GACX,OAAO,eAAe,WACpB,aACA;AACN,OAAI,CAAC,QAAS;AACd,oBAAiB;IACf,WAAW,KAAK;IAChB,QAAQ,IAAI;IACZ,gBAAgB;IAChB,WAAW,IAAI;IACf,SAAS,IAAI;IACb,QAAQ,IAAI,UAAU;IACtB,SAAS,IAAI;IACb,eAAe,IAAI;IACnB,eAAe,IAAI;IACnB,KAAK,IAAI;IACT,QAAQ,IAAI;IACZ,eAAe,KAAK;IACrB,CAAC;;;;AAKR,SAAS,iBAAiB,OAAiD;AACzE,QACE,SAAS,QACT,OAAQ,MAAkC,OAAO,mBAC/C;;AAIN,SAAgB,6BACd,QACoB;AACpB,KAAI,OAAO,WAAW,SAAU,QAAO;AACvC,KAAI,MAAM,QAAQ,OAAO,CACvB,QAAO,OACJ,KAAK,UAAU;AACd,MAAI,OAAO,UAAU,SAAU,QAAO;AACtC,MAAI,OAAO,UAAU,YAAY,SAAS,KACxC,QAAO,OAAQ,MAAkC,QAAQ,GAAG;AAE9D,SAAO;GACP,CACD,OAAO,QAAQ,CACf,KAAK,KAAK;;AAKjB,SAAgB,wBACd,SAKA;CACA,IAAI,OAAO;CACX,IAAIC;CACJ,MAAMC,YAA4C,EAAE;AAEpD,KAAI,CAAC,QAAS,QAAO;EAAE;EAAM;EAAW;EAAW;AAEnD,MAAK,MAAM,SAAS,QAClB,KAAI,MAAM,SAAS,OACjB,SAAQ,OAAO,MAAM,QAAQ,GAAG;UACvB,MAAM,SAAS,WACxB,aAAY,OAAO,MAAM,YAAY,GAAG;UAC/B,MAAM,SAAS,WACxB,WAAU,KAAK;EACb,MAAM;EACN,IAAI,MAAM;EACV,UAAU;GACR,MAAM,OAAO,MAAM,QAAQ,GAAG;GAC9B,WACE,OAAO,MAAM,UAAU,WACnB,MAAM,QACN,KAAK,UAAU,MAAM,SAAS,EAAE,CAAC;GACxC;EACF,CAAC;AAIN,QAAO;EAAE;EAAM;EAAW;EAAW"}
+{"version":3,"file":"anthropic.js","names":["_AnthropicModule: Record<string, unknown> | null","clientOpts: Record<string, unknown>","costUsd: number | null","reasoning: string | undefined","toolCalls: Array<Record<string, unknown>>"],"sources":["../../src/providers/anthropic.ts"],"sourcesContent":["/**\n * Anthropic provider wrapper with automatic tracking.\n */\n\nimport type { PrivacyConfig } from '../core/privacy.js';\nimport { trackUserMessage } from '../core/tracking.js';\nimport { getDefaultPropagateContext, injectContext } from '../propagation.js';\nimport type {\n  AmplitudeLike,\n  AmplitudeOrAI,\n  AnthropicResponse,\n  TrackFn,\n} from '../types.js';\nimport { calculateCost } from '../utils/costs.js';\nimport { tryRequire } from '../utils/resolve-module.js';\nimport { StreamingAccumulator } from '../utils/streaming.js';\nimport {\n  applySessionContext,\n  BaseAIProvider,\n  contextFields,\n  type ProviderTrackOptions,\n} from './base.js';\n\nconst _resolved = tryRequire('@anthropic-ai/sdk');\nexport const ANTHROPIC_AVAILABLE = _resolved != null;\nconst _AnthropicModule: Record<string, unknown> | null = _resolved;\n\nexport { _AnthropicModule };\n\nexport interface AnthropicOptions {\n  amplitude: AmplitudeOrAI;\n  apiKey?: string;\n  privacyConfig?: PrivacyConfig | null;\n  propagateContext?: boolean;\n  /** Pass the `@anthropic-ai/sdk` module directly to bypass `tryRequire` (required in bundler environments). */\n  anthropicModule?: unknown;\n}\n\nexport class Anthropic<\n  TClient extends Record<string, unknown> = Record<string, unknown>,\n> extends BaseAIProvider {\n  private _client: TClient;\n  readonly messages: WrappedMessages;\n  private _propagateContext: boolean;\n\n  constructor(options: AnthropicOptions) {\n    super({\n      amplitude: options.amplitude,\n      privacyConfig: options.privacyConfig,\n      providerName: 'anthropic',\n    });\n\n    const mod =\n      (options.anthropicModule as Record<string, unknown> | null) ??\n      _AnthropicModule;\n    if (mod == null) {\n      throw new Error(\n        '@anthropic-ai/sdk package is required. Install it with: npm install @anthropic-ai/sdk — or pass the module directly via the anthropicModule option.',\n      );\n    }\n\n    const AnthropicSDK = mod.Anthropic as new (\n      opts: Record<string, unknown>,\n    ) => unknown;\n\n    const clientOpts: Record<string, unknown> = {};\n    if (options.apiKey) clientOpts.apiKey = options.apiKey;\n\n    this._client = new AnthropicSDK(clientOpts) as TClient;\n    this._propagateContext =\n      options.propagateContext ?? getDefaultPropagateContext();\n    this.messages = new WrappedMessages(\n      this._client,\n      this.trackFn(),\n      this._amplitude,\n      this._privacyConfig,\n      this._propagateContext,\n    );\n  }\n\n  get client(): TClient {\n    return this._client;\n  }\n}\n\nexport class WrappedMessages {\n  _original: Record<string, unknown>;\n  private _trackFn: TrackFn;\n  private _amplitude: AmplitudeLike;\n  private _privacyConfig: PrivacyConfig | null;\n  private _propagateContext: boolean;\n\n  constructor(\n    client: unknown,\n    trackFn: TrackFn,\n    amplitude: AmplitudeLike,\n    privacyConfig: PrivacyConfig | null,\n    propagateContext: boolean,\n  ) {\n    const clientObj = client as Record<string, unknown>;\n    this._original = clientObj.messages as Record<string, unknown>;\n    this._trackFn =\n      typeof trackFn === 'function'\n        ? trackFn\n        : (trackFn as unknown as { trackFn(): TrackFn }).trackFn();\n    this._amplitude = amplitude;\n    this._privacyConfig = privacyConfig;\n    this._propagateContext = propagateContext;\n  }\n\n  async create(\n    params: Record<string, unknown>,\n    amplitudeOverrides?: ProviderTrackOptions,\n  ): Promise<AnthropicResponse | AsyncIterable<unknown>> {\n    const createFn = this._original.create as (\n      ...args: unknown[]\n    ) => Promise<unknown>;\n    const startTime = performance.now();\n    const requestParams = this._withContextHeaders(params);\n    const ctx = applySessionContext(amplitudeOverrides);\n\n    try {\n      this._trackInputMessages(\n        requestParams.messages as unknown,\n        ctx,\n        amplitudeOverrides?.trackInputMessages ?? true,\n      );\n      const response = await createFn.call(this._original, requestParams);\n\n      if (requestParams.stream === true && _isAsyncIterable(response)) {\n        return this._wrapStream(\n          response as AsyncIterable<unknown>,\n          requestParams,\n          startTime,\n          ctx,\n        );\n      }\n\n      const latencyMs = performance.now() - startTime;\n\n      const resp = response as AnthropicResponse;\n      const usage = resp.usage;\n      const extracted = extractAnthropicContent(\n        resp.content as unknown as Array<Record<string, unknown>> | undefined,\n      );\n      const firstTextBlock = resp.content?.find((b) => b.type === 'text');\n      const modelName = String(resp.model ?? requestParams.model ?? 'unknown');\n\n      const cacheRead = usage?.cache_read_input_tokens ?? 0;\n      const cacheCreation = usage?.cache_creation_input_tokens ?? 0;\n      const rawInput = usage?.input_tokens ?? 0;\n      const normalizedInput =\n        cacheRead || cacheCreation\n          ? rawInput + cacheRead + cacheCreation\n          : rawInput;\n\n      let costUsd: number | null = null;\n      if (usage?.input_tokens != null && usage?.output_tokens != null) {\n        try {\n          costUsd = calculateCost({\n            modelName,\n            inputTokens: normalizedInput,\n            outputTokens: usage.output_tokens,\n            cacheReadInputTokens: cacheRead,\n            cacheCreationInputTokens: cacheCreation,\n          });\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      this._trackFn({\n        ...contextFields(ctx),\n        modelName,\n        provider: 'anthropic',\n        responseContent: String(firstTextBlock?.text ?? ''),\n        reasoningContent: extracted.reasoning,\n        latencyMs,\n        inputTokens: normalizedInput || undefined,\n        outputTokens: usage?.output_tokens,\n        cacheReadInputTokens: cacheRead || undefined,\n        cacheCreationInputTokens: cacheCreation || undefined,\n        totalCostUsd: costUsd,\n        finishReason: resp.stop_reason,\n        toolCalls:\n          extracted.toolCalls.length > 0 ? extracted.toolCalls : undefined,\n        isStreaming: false,\n        systemPrompt: extractAnthropicSystemPrompt(requestParams.system),\n        temperature: requestParams.temperature as number | undefined,\n        maxOutputTokens: requestParams.max_tokens as number | undefined,\n        topP: requestParams.top_p as number | undefined,\n      });\n\n      return response as AnthropicResponse;\n    } catch (error) {\n      const latencyMs = performance.now() - startTime;\n      this._trackFn({\n        ...contextFields(ctx),\n        modelName: String(requestParams.model ?? 'unknown'),\n        provider: 'anthropic',\n        responseContent: '',\n        latencyMs,\n        isError: true,\n        errorMessage: error instanceof Error ? error.message : String(error),\n      });\n\n      throw error;\n    }\n  }\n\n  private async *_wrapStream(\n    stream: AsyncIterable<unknown>,\n    params: Record<string, unknown>,\n    _startTime: number,\n    sessionCtx: ProviderTrackOptions,\n  ): AsyncGenerator<unknown> {\n    const accumulator = new StreamingAccumulator();\n    accumulator.model = String(params.model ?? 'unknown');\n    let reasoningContent = '';\n\n    try {\n      for await (const event of stream) {\n        const evt = event as Record<string, unknown>;\n        const type = evt.type as string | undefined;\n\n        if (type === 'content_block_delta') {\n          const delta = evt.delta as Record<string, unknown> | undefined;\n          if (delta?.type === 'text_delta' && delta.text != null) {\n            accumulator.addContent(String(delta.text));\n          } else if (\n            delta?.type === 'thinking_delta' &&\n            delta.thinking != null\n          ) {\n            reasoningContent += String(delta.thinking);\n          }\n        } else if (type === 'content_block_start') {\n          const block = evt.content_block as\n            | Record<string, unknown>\n            | undefined;\n          if (block?.type === 'tool_use') {\n            accumulator.addToolCall({\n              type: 'function',\n              id: block.id,\n              function: {\n                name: String(block.name ?? ''),\n                arguments:\n                  typeof block.input === 'string'\n                    ? block.input\n                    : JSON.stringify(block.input ?? {}),\n              },\n            });\n          }\n        } else if (type === 'message_delta') {\n          const delta = evt.delta as Record<string, unknown> | undefined;\n          if (delta?.stop_reason != null) {\n            accumulator.finishReason = String(delta.stop_reason);\n          }\n          const usage = evt.usage as Record<string, number> | undefined;\n          if (usage != null) {\n            accumulator.setUsage({\n              outputTokens: usage.output_tokens,\n            });\n          }\n        } else if (type === 'message_start') {\n          const message = evt.message as Record<string, unknown> | undefined;\n          if (message?.model != null) {\n            accumulator.model = String(message.model);\n          }\n          const usage = message?.usage as Record<string, number> | undefined;\n          if (usage != null) {\n            accumulator.setUsage({\n              inputTokens: usage.input_tokens,\n              cacheReadTokens: usage.cache_read_input_tokens,\n              cacheCreationTokens: usage.cache_creation_input_tokens,\n            });\n          }\n        }\n\n        yield event;\n      }\n    } catch (error) {\n      accumulator.setError(\n        error instanceof Error ? error.message : String(error),\n      );\n      throw error;\n    } finally {\n      const state = accumulator.getState();\n      const modelName = String(accumulator.model ?? params.model ?? 'unknown');\n\n      const streamCacheRead = state.cacheReadTokens ?? 0;\n      const streamCacheCreation = state.cacheCreationTokens ?? 0;\n      const streamRawInput = state.inputTokens ?? 0;\n      const streamNormalizedInput =\n        streamCacheRead || streamCacheCreation\n          ? streamRawInput + streamCacheRead + streamCacheCreation\n          : streamRawInput;\n\n      let costUsd: number | null = null;\n      if (state.inputTokens != null && state.outputTokens != null) {\n        try {\n          costUsd = calculateCost({\n            modelName,\n            inputTokens: streamNormalizedInput,\n            outputTokens: state.outputTokens,\n            cacheReadInputTokens: streamCacheRead,\n            cacheCreationInputTokens: streamCacheCreation,\n          });\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      this._trackFn({\n        ...contextFields(sessionCtx),\n        modelName,\n        provider: 'anthropic',\n        responseContent: state.content,\n        latencyMs: accumulator.elapsedMs,\n        inputTokens: streamNormalizedInput || undefined,\n        outputTokens: state.outputTokens,\n        cacheReadInputTokens: streamCacheRead || undefined,\n        cacheCreationInputTokens: streamCacheCreation || undefined,\n        totalCostUsd: costUsd,\n        finishReason: state.finishReason,\n        toolCalls: state.toolCalls.length > 0 ? state.toolCalls : undefined,\n        providerTtfbMs: state.ttfbMs,\n        isStreaming: true,\n        isError: state.isError,\n        errorMessage: state.errorMessage,\n        reasoningContent: reasoningContent || undefined,\n        systemPrompt: extractAnthropicSystemPrompt(params.system),\n        temperature: params.temperature as number | undefined,\n        maxOutputTokens: params.max_tokens as number | undefined,\n        topP: params.top_p as number | undefined,\n      });\n    }\n  }\n\n  private _withContextHeaders(\n    params: Record<string, unknown>,\n  ): Record<string, unknown> {\n    if (!this._propagateContext) return params;\n    const existing = (params.extra_headers ?? params.headers) as\n      | Record<string, string>\n      | undefined;\n    const injected = injectContext(existing);\n    return {\n      ...params,\n      extra_headers: injected,\n    };\n  }\n\n  private _trackInputMessages(\n    messages: unknown,\n    ctx: ProviderTrackOptions,\n    shouldTrackInputMessages: boolean,\n  ): void {\n    if (!shouldTrackInputMessages) return;\n    if (ctx.userId == null || ctx.sessionId == null) return;\n    if (!Array.isArray(messages)) return;\n    for (const msg of messages) {\n      const role = (msg as Record<string, unknown>)?.role;\n      if (role !== 'user') continue;\n      const rawContent = (msg as Record<string, unknown>)?.content;\n      const content = Array.isArray(rawContent)\n        ? rawContent\n            .map((part) => {\n              if (typeof part === 'string') return part;\n              const text = (part as Record<string, unknown>)?.text;\n              return typeof text === 'string' ? text : '';\n            })\n            .join('')\n        : typeof rawContent === 'string'\n          ? rawContent\n          : '';\n      if (!content) continue;\n      trackUserMessage({\n        amplitude: this._amplitude,\n        userId: ctx.userId,\n        messageContent: content,\n        sessionId: ctx.sessionId,\n        traceId: ctx.traceId,\n        turnId: ctx.turnId ?? undefined,\n        agentId: ctx.agentId,\n        parentAgentId: ctx.parentAgentId,\n        customerOrgId: ctx.customerOrgId,\n        env: ctx.env,\n        groups: ctx.groups,\n        privacyConfig: this._privacyConfig,\n      });\n    }\n  }\n}\n\nfunction _isAsyncIterable(value: unknown): value is AsyncIterable<unknown> {\n  return (\n    value != null &&\n    typeof (value as Record<symbol, unknown>)[Symbol.asyncIterator] ===\n      'function'\n  );\n}\n\nexport function extractAnthropicSystemPrompt(\n  system: unknown,\n): string | undefined {\n  if (typeof system === 'string') return system;\n  if (Array.isArray(system)) {\n    return system\n      .map((block) => {\n        if (typeof block === 'string') return block;\n        if (typeof block === 'object' && block != null) {\n          return String((block as Record<string, unknown>).text ?? '');\n        }\n        return '';\n      })\n      .filter(Boolean)\n      .join('\\n');\n  }\n  return undefined;\n}\n\nexport function extractAnthropicContent(\n  content: Array<Record<string, unknown>> | undefined,\n): {\n  text: string;\n  reasoning: string | undefined;\n  toolCalls: Array<Record<string, unknown>>;\n} {\n  let text = '';\n  let reasoning: string | undefined;\n  const toolCalls: Array<Record<string, unknown>> = [];\n\n  if (!content) return { text, reasoning, toolCalls };\n\n  for (const block of content) {\n    if (block.type === 'text') {\n      text += String(block.text ?? '');\n    } else if (block.type === 'thinking') {\n      reasoning = String(block.thinking ?? '');\n    } else if (block.type === 'tool_use') {\n      toolCalls.push({\n        type: 'function',\n        id: block.id,\n        function: {\n          name: String(block.name ?? ''),\n          arguments:\n            typeof block.input === 'string'\n              ? block.input\n              : JSON.stringify(block.input ?? {}),\n        },\n      });\n    }\n  }\n\n  return { text, reasoning, toolCalls };\n}\n"],"mappings":";;;;;;;;AAuBA,MAAM,YAAY,WAAW,oBAAoB;AACjD,MAAa,sBAAsB,aAAa;AAChD,MAAMA,mBAAmD;AAazD,IAAa,YAAb,cAEU,eAAe;CACvB,AAAQ;CACR,AAAS;CACT,AAAQ;CAER,YAAY,SAA2B;AACrC,QAAM;GACJ,WAAW,QAAQ;GACnB,eAAe,QAAQ;GACvB,cAAc;GACf,CAAC;EAEF,MAAM,MACH,QAAQ,mBACT;AACF,MAAI,OAAO,KACT,OAAM,IAAI,MACR,sJACD;EAGH,MAAM,eAAe,IAAI;EAIzB,MAAMC,aAAsC,EAAE;AAC9C,MAAI,QAAQ,OAAQ,YAAW,SAAS,QAAQ;AAEhD,OAAK,UAAU,IAAI,aAAa,WAAW;AAC3C,OAAK,oBACH,QAAQ,oBAAoB,4BAA4B;AAC1D,OAAK,WAAW,IAAI,gBAClB,KAAK,SACL,KAAK,SAAS,EACd,KAAK,YACL,KAAK,gBACL,KAAK,kBACN;;CAGH,IAAI,SAAkB;AACpB,SAAO,KAAK;;;AAIhB,IAAa,kBAAb,MAA6B;CAC3B;CACA,AAAQ;CACR,AAAQ;CACR,AAAQ;CACR,AAAQ;CAER,YACE,QACA,SACA,WACA,eACA,kBACA;AAEA,OAAK,YADa,OACS;AAC3B,OAAK,WACH,OAAO,YAAY,aACf,UACC,QAA8C,SAAS;AAC9D,OAAK,aAAa;AAClB,OAAK,iBAAiB;AACtB,OAAK,oBAAoB;;CAG3B,MAAM,OACJ,QACA,oBACqD;EACrD,MAAM,WAAW,KAAK,UAAU;EAGhC,MAAM,YAAY,YAAY,KAAK;EACnC,MAAM,gBAAgB,KAAK,oBAAoB,OAAO;EACtD,MAAM,MAAM,oBAAoB,mBAAmB;AAEnD,MAAI;AACF,QAAK,oBACH,cAAc,UACd,KACA,oBAAoB,sBAAsB,KAC3C;GACD,MAAM,WAAW,MAAM,SAAS,KAAK,KAAK,WAAW,cAAc;AAEnE,OAAI,cAAc,WAAW,QAAQ,iBAAiB,SAAS,CAC7D,QAAO,KAAK,YACV,UACA,eACA,WACA,IACD;GAGH,MAAM,YAAY,YAAY,KAAK,GAAG;GAEtC,MAAM,OAAO;GACb,MAAM,QAAQ,KAAK;GACnB,MAAM,YAAY,wBAChB,KAAK,QACN;GACD,MAAM,iBAAiB,KAAK,SAAS,MAAM,MAAM,EAAE,SAAS,OAAO;GACnE,MAAM,YAAY,OAAO,KAAK,SAAS,cAAc,SAAS,UAAU;GAExE,MAAM,YAAY,OAAO,2BAA2B;GACpD,MAAM,gBAAgB,OAAO,+BAA+B;GAC5D,MAAM,WAAW,OAAO,gBAAgB;GACxC,MAAM,kBACJ,aAAa,gBACT,WAAW,YAAY,gBACvB;GAEN,IAAIC,UAAyB;AAC7B,OAAI,OAAO,gBAAgB,QAAQ,OAAO,iBAAiB,KACzD,KAAI;AACF,cAAU,cAAc;KACtB;KACA,aAAa;KACb,cAAc,MAAM;KACpB,sBAAsB;KACtB,0BAA0B;KAC3B,CAAC;WACI;AAKV,QAAK,SAAS;IACZ,GAAG,cAAc,IAAI;IACrB;IACA,UAAU;IACV,iBAAiB,OAAO,gBAAgB,QAAQ,GAAG;IACnD,kBAAkB,UAAU;IAC5B;IACA,aAAa,mBAAmB;IAChC,cAAc,OAAO;IACrB,sBAAsB,aAAa;IACnC,0BAA0B,iBAAiB;IAC3C,cAAc;IACd,cAAc,KAAK;IACnB,WACE,UAAU,UAAU,SAAS,IAAI,UAAU,YAAY;IACzD,aAAa;IACb,cAAc,6BAA6B,cAAc,OAAO;IAChE,aAAa,cAAc;IAC3B,iBAAiB,cAAc;IAC/B,MAAM,cAAc;IACrB,CAAC;AAEF,UAAO;WACA,OAAO;GACd,MAAM,YAAY,YAAY,KAAK,GAAG;AACtC,QAAK,SAAS;IACZ,GAAG,cAAc,IAAI;IACrB,WAAW,OAAO,cAAc,SAAS,UAAU;IACnD,UAAU;IACV,iBAAiB;IACjB;IACA,SAAS;IACT,cAAc,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IACrE,CAAC;AAEF,SAAM;;;CAIV,OAAe,YACb,QACA,QACA,YACA,YACyB;EACzB,MAAM,cAAc,IAAI,sBAAsB;AAC9C,cAAY,QAAQ,OAAO,OAAO,SAAS,UAAU;EACrD,IAAI,mBAAmB;AAEvB,MAAI;AACF,cAAW,MAAM,SAAS,QAAQ;IAChC,MAAM,MAAM;IACZ,MAAM,OAAO,IAAI;AAEjB,QAAI,SAAS,uBAAuB;KAClC,MAAM,QAAQ,IAAI;AAClB,SAAI,OAAO,SAAS,gBAAgB,MAAM,QAAQ,KAChD,aAAY,WAAW,OAAO,MAAM,KAAK,CAAC;cAE1C,OAAO,SAAS,oBAChB,MAAM,YAAY,KAElB,qBAAoB,OAAO,MAAM,SAAS;eAEnC,SAAS,uBAAuB;KACzC,MAAM,QAAQ,IAAI;AAGlB,SAAI,OAAO,SAAS,WAClB,aAAY,YAAY;MACtB,MAAM;MACN,IAAI,MAAM;MACV,UAAU;OACR,MAAM,OAAO,MAAM,QAAQ,GAAG;OAC9B,WACE,OAAO,MAAM,UAAU,WACnB,MAAM,QACN,KAAK,UAAU,MAAM,SAAS,EAAE,CAAC;OACxC;MACF,CAAC;eAEK,SAAS,iBAAiB;KACnC,MAAM,QAAQ,IAAI;AAClB,SAAI,OAAO,eAAe,KACxB,aAAY,eAAe,OAAO,MAAM,YAAY;KAEtD,MAAM,QAAQ,IAAI;AAClB,SAAI,SAAS,KACX,aAAY,SAAS,EACnB,cAAc,MAAM,eACrB,CAAC;eAEK,SAAS,iBAAiB;KACnC,MAAM,UAAU,IAAI;AACpB,SAAI,SAAS,SAAS,KACpB,aAAY,QAAQ,OAAO,QAAQ,MAAM;KAE3C,MAAM,QAAQ,SAAS;AACvB,SAAI,SAAS,KACX,aAAY,SAAS;MACnB,aAAa,MAAM;MACnB,iBAAiB,MAAM;MACvB,qBAAqB,MAAM;MAC5B,CAAC;;AAIN,UAAM;;WAED,OAAO;AACd,eAAY,SACV,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,CACvD;AACD,SAAM;YACE;GACR,MAAM,QAAQ,YAAY,UAAU;GACpC,MAAM,YAAY,OAAO,YAAY,SAAS,OAAO,SAAS,UAAU;GAExE,MAAM,kBAAkB,MAAM,mBAAmB;GACjD,MAAM,sBAAsB,MAAM,uBAAuB;GACzD,MAAM,iBAAiB,MAAM,eAAe;GAC5C,MAAM,wBACJ,mBAAmB,sBACf,iBAAiB,kBAAkB,sBACnC;GAEN,IAAIA,UAAyB;AAC7B,OAAI,MAAM,eAAe,QAAQ,MAAM,gBAAgB,KACrD,KAAI;AACF,cAAU,cAAc;KACtB;KACA,aAAa;KACb,cAAc,MAAM;KACpB,sBAAsB;KACtB,0BAA0B;KAC3B,CAAC;WACI;AAKV,QAAK,SAAS;IACZ,GAAG,cAAc,WAAW;IAC5B;IACA,UAAU;IACV,iBAAiB,MAAM;IACvB,WAAW,YAAY;IACvB,aAAa,yBAAyB;IACtC,cAAc,MAAM;IACpB,sBAAsB,mBAAmB;IACzC,0BAA0B,uBAAuB;IACjD,cAAc;IACd,cAAc,MAAM;IACpB,WAAW,MAAM,UAAU,SAAS,IAAI,MAAM,YAAY;IAC1D,gBAAgB,MAAM;IACtB,aAAa;IACb,SAAS,MAAM;IACf,cAAc,MAAM;IACpB,kBAAkB,oBAAoB;IACtC,cAAc,6BAA6B,OAAO,OAAO;IACzD,aAAa,OAAO;IACpB,iBAAiB,OAAO;IACxB,MAAM,OAAO;IACd,CAAC;;;CAIN,AAAQ,oBACN,QACyB;AACzB,MAAI,CAAC,KAAK,kBAAmB,QAAO;EAIpC,MAAM,WAAW,cAHC,OAAO,iBAAiB,OAAO,QAGT;AACxC,SAAO;GACL,GAAG;GACH,eAAe;GAChB;;CAGH,AAAQ,oBACN,UACA,KACA,0BACM;AACN,MAAI,CAAC,yBAA0B;AAC/B,MAAI,IAAI,UAAU,QAAQ,IAAI,aAAa,KAAM;AACjD,MAAI,CAAC,MAAM,QAAQ,SAAS,CAAE;AAC9B,OAAK,MAAM,OAAO,UAAU;AAE1B,OADc,KAAiC,SAClC,OAAQ;GACrB,MAAM,aAAc,KAAiC;GACrD,MAAM,UAAU,MAAM,QAAQ,WAAW,GACrC,WACG,KAAK,SAAS;AACb,QAAI,OAAO,SAAS,SAAU,QAAO;IACrC,MAAM,OAAQ,MAAkC;AAChD,WAAO,OAAO,SAAS,WAAW,OAAO;KACzC,CACD,KAAK,GAAG,GACX,OAAO,eAAe,WACpB,aACA;AACN,OAAI,CAAC,QAAS;AACd,oBAAiB;IACf,WAAW,KAAK;IAChB,QAAQ,IAAI;IACZ,gBAAgB;IAChB,WAAW,IAAI;IACf,SAAS,IAAI;IACb,QAAQ,IAAI,UAAU;IACtB,SAAS,IAAI;IACb,eAAe,IAAI;IACnB,eAAe,IAAI;IACnB,KAAK,IAAI;IACT,QAAQ,IAAI;IACZ,eAAe,KAAK;IACrB,CAAC;;;;AAKR,SAAS,iBAAiB,OAAiD;AACzE,QACE,SAAS,QACT,OAAQ,MAAkC,OAAO,mBAC/C;;AAIN,SAAgB,6BACd,QACoB;AACpB,KAAI,OAAO,WAAW,SAAU,QAAO;AACvC,KAAI,MAAM,QAAQ,OAAO,CACvB,QAAO,OACJ,KAAK,UAAU;AACd,MAAI,OAAO,UAAU,SAAU,QAAO;AACtC,MAAI,OAAO,UAAU,YAAY,SAAS,KACxC,QAAO,OAAQ,MAAkC,QAAQ,GAAG;AAE9D,SAAO;GACP,CACD,OAAO,QAAQ,CACf,KAAK,KAAK;;AAKjB,SAAgB,wBACd,SAKA;CACA,IAAI,OAAO;CACX,IAAIC;CACJ,MAAMC,YAA4C,EAAE;AAEpD,KAAI,CAAC,QAAS,QAAO;EAAE;EAAM;EAAW;EAAW;AAEnD,MAAK,MAAM,SAAS,QAClB,KAAI,MAAM,SAAS,OACjB,SAAQ,OAAO,MAAM,QAAQ,GAAG;UACvB,MAAM,SAAS,WACxB,aAAY,OAAO,MAAM,YAAY,GAAG;UAC/B,MAAM,SAAS,WACxB,WAAU,KAAK;EACb,MAAM;EACN,IAAI,MAAM;EACV,UAAU;GACR,MAAM,OAAO,MAAM,QAAQ,GAAG;GAC9B,WACE,OAAO,MAAM,UAAU,WACnB,MAAM,QACN,KAAK,UAAU,MAAM,SAAS,EAAE,CAAC;GACxC;EACF,CAAC;AAIN,QAAO;EAAE;EAAM;EAAW;EAAW"}

package/dist/providers/azure-openai.d.ts

@@ -1,6 +1,6 @@
 import { PrivacyConfig } from "../core/privacy.js";
 import { AmplitudeOrAI } from "../types.js";
-import { BaseAIProvider } from "./base.js";
+import { BaseAIProvider, ProviderTrackOptions } from "./base.js";
 import { OPENAI_AVAILABLE } from "./openai.js";
 
 //#region src/providers/azure-openai.d.ts
@@ -20,8 +20,8 @@ declare class AzureOpenAI extends BaseAIProvider {
   private _propagateContext;
   readonly chat: {
     completions: {
-      create: (params: Record<string, unknown>) => Promise<unknown>;
-      parse: (params: Record<string, unknown>) => Promise<unknown>;
+      create: (params: Record<string, unknown>, amplitudeOverrides?: ProviderTrackOptions) => Promise<unknown>;
+      parse: (params: Record<string, unknown>, amplitudeOverrides?: ProviderTrackOptions) => Promise<unknown>;
     };
   };
   constructor(options: AzureOpenAIOptions);

package/dist/providers/azure-openai.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"azure-openai.d.ts","names":[],"sources":["../../src/providers/azure-openai.ts"],"sourcesContent":[],"mappings":";;;;;;;AAgCuB,UAhBN,kBAAA,CAgBM;EAA4B,SAAA,EAftC,aAesC;EAC7B,MAAA,CAAA,EAAA,MAAA;EAA4B,aAAA,CAAA,EAAA,MAAA;EAI3B,UAAA,CAAA,EAAA,MAAA;EAVU,aAAA,CAAA,EANf,aAMe,GAAA,IAAA;EAAc,gBAAA,CAAA,EAAA,OAAA;;;;cAAlC,WAAA,SAAoB,cAAA;;;;;uBAKV,4BAA4B;sBAC7B,4BAA4B;;;uBAI3B"}
+{"version":3,"file":"azure-openai.d.ts","names":[],"sources":["../../src/providers/azure-openai.ts"],"sourcesContent":[],"mappings":";;;;;;;AAgCuB,UAhBN,kBAAA,CAgBM;EAA8C,SAAA,EAfxD,aAewD;EAAyB,MAAA,CAAA,EAAA,MAAA;EACxE,aAAA,CAAA,EAAA,MAAA;EAA8C,UAAA,CAAA,EAAA,MAAA;EAAyB,aAAA,CAAA,EAZ3E,aAY2E,GAAA,IAAA;EAItE,gBAAA,CAAA,EAAA,OAAA;EAVU;EAAc,YAAA,CAAA,EAAA,OAAA;;cAAlC,WAAA,SAAoB,cAAA;;;;;uBAKV,8CAA8C,yBAAyB;sBACxE,8CAA8C,yBAAyB;;;uBAItE"}

package/dist/providers/azure-openai.js

@@ -25,8 +25,8 @@ var AzureOpenAI = class extends BaseAIProvider {
 		const originalCompletions = this._client.chat.completions;
 		const wrappedCompletions = new WrappedCompletions(originalCompletions, this.trackFn(), this._amplitude, this._privacyConfig, this._propagateContext, "azure-openai");
 		this.chat = { completions: {
-			create: (params) => wrappedCompletions.create(params),
-			parse: (params) => wrappedCompletions.parse(params)
+			create: (params, amplitudeOverrides) => wrappedCompletions.create(params, amplitudeOverrides),
+			parse: (params, amplitudeOverrides) => wrappedCompletions.parse(params, amplitudeOverrides)
 		} };
 	}
 	get client() {

package/dist/providers/azure-openai.js.map

@@ -1 +1 @@
-{"version":3,"file":"azure-openai.js","names":["clientOpts: Record<string, unknown>"],"sources":["../../src/providers/azure-openai.ts"],"sourcesContent":["/**\n * Azure OpenAI provider wrapper — reuses OpenAI wrapper logic.\n */\n\nimport type { PrivacyConfig } from '../core/privacy.js';\nimport { getDefaultPropagateContext } from '../propagation.js';\nimport type { AmplitudeOrAI } from '../types.js';\nimport { BaseAIProvider } from './base.js';\nimport {\n  _OpenAIModule,\n  OPENAI_AVAILABLE,\n  WrappedCompletions,\n} from './openai.js';\n\nexport { OPENAI_AVAILABLE as AZURE_OPENAI_AVAILABLE };\n\nexport interface AzureOpenAIOptions {\n  amplitude: AmplitudeOrAI;\n  apiKey?: string;\n  azureEndpoint?: string;\n  apiVersion?: string;\n  privacyConfig?: PrivacyConfig | null;\n  propagateContext?: boolean;\n  /** Pass the `openai` module directly to bypass `tryRequire` (required in bundler environments). */\n  openaiModule?: unknown;\n}\n\nexport class AzureOpenAI extends BaseAIProvider {\n  private _client: unknown;\n  private _propagateContext: boolean;\n  readonly chat: {\n    completions: {\n      create: (params: Record<string, unknown>) => Promise<unknown>;\n      parse: (params: Record<string, unknown>) => Promise<unknown>;\n    };\n  };\n\n  constructor(options: AzureOpenAIOptions) {\n    super({\n      amplitude: options.amplitude,\n      privacyConfig: options.privacyConfig,\n      providerName: 'azure-openai',\n    });\n\n    const mod =\n      (options.openaiModule as Record<string, unknown> | null) ?? _OpenAIModule;\n    if (mod == null) {\n      throw new Error(\n        'openai package is required for Azure OpenAI. Install it with: npm install openai — or pass the module directly via the openaiModule option.',\n      );\n    }\n\n    const AzureOpenAISDK = mod.AzureOpenAI as new (\n      opts: Record<string, unknown>,\n    ) => unknown;\n\n    const clientOpts: Record<string, unknown> = {};\n    if (options.apiKey) clientOpts.apiKey = options.apiKey;\n    if (options.azureEndpoint) clientOpts.baseURL = options.azureEndpoint;\n    if (options.apiVersion) {\n      clientOpts.defaultQuery = { 'api-version': options.apiVersion };\n    }\n\n    this._client = new AzureOpenAISDK(clientOpts);\n    this._propagateContext =\n      options.propagateContext ?? getDefaultPropagateContext();\n    const clientObj = this._client as Record<string, unknown>;\n    const originalChat = clientObj.chat as Record<string, unknown>;\n    const originalCompletions = originalChat.completions as Record<\n      string,\n      unknown\n    >;\n    const wrappedCompletions = new WrappedCompletions(\n      originalCompletions,\n      this.trackFn(),\n      this._amplitude,\n      this._privacyConfig,\n      this._propagateContext,\n      'azure-openai',\n    );\n\n    this.chat = {\n      completions: {\n        create: (params: Record<string, unknown>): Promise<unknown> =>\n          wrappedCompletions.create(params),\n        parse: (params: Record<string, unknown>): Promise<unknown> =>\n          wrappedCompletions.parse(params),\n      },\n    };\n  }\n\n  get client(): unknown {\n    return this._client;\n  }\n}\n"],"mappings":";;;;;AA2BA,IAAa,cAAb,cAAiC,eAAe;CAC9C,AAAQ;CACR,AAAQ;CACR,AAAS;CAOT,YAAY,SAA6B;AACvC,QAAM;GACJ,WAAW,QAAQ;GACnB,eAAe,QAAQ;GACvB,cAAc;GACf,CAAC;EAEF,MAAM,MACH,QAAQ,gBAAmD;AAC9D,MAAI,OAAO,KACT,OAAM,IAAI,MACR,8IACD;EAGH,MAAM,iBAAiB,IAAI;EAI3B,MAAMA,aAAsC,EAAE;AAC9C,MAAI,QAAQ,OAAQ,YAAW,SAAS,QAAQ;AAChD,MAAI,QAAQ,cAAe,YAAW,UAAU,QAAQ;AACxD,MAAI,QAAQ,WACV,YAAW,eAAe,EAAE,eAAe,QAAQ,YAAY;AAGjE,OAAK,UAAU,IAAI,eAAe,WAAW;AAC7C,OAAK,oBACH,QAAQ,oBAAoB,4BAA4B;EAG1D,MAAM,sBAFY,KAAK,QACQ,KACU;EAIzC,MAAM,qBAAqB,IAAI,mBAC7B,qBACA,KAAK,SAAS,EACd,KAAK,YACL,KAAK,gBACL,KAAK,mBACL,eACD;AAED,OAAK,OAAO,EACV,aAAa;GACX,SAAS,WACP,mBAAmB,OAAO,OAAO;GACnC,QAAQ,WACN,mBAAmB,MAAM,OAAO;GACnC,EACF;;CAGH,IAAI,SAAkB;AACpB,SAAO,KAAK"}
+{"version":3,"file":"azure-openai.js","names":["clientOpts: Record<string, unknown>"],"sources":["../../src/providers/azure-openai.ts"],"sourcesContent":["/**\n * Azure OpenAI provider wrapper — reuses OpenAI wrapper logic.\n */\n\nimport type { PrivacyConfig } from '../core/privacy.js';\nimport { getDefaultPropagateContext } from '../propagation.js';\nimport type { AmplitudeOrAI } from '../types.js';\nimport { BaseAIProvider, type ProviderTrackOptions } from './base.js';\nimport {\n  _OpenAIModule,\n  OPENAI_AVAILABLE,\n  WrappedCompletions,\n} from './openai.js';\n\nexport { OPENAI_AVAILABLE as AZURE_OPENAI_AVAILABLE };\n\nexport interface AzureOpenAIOptions {\n  amplitude: AmplitudeOrAI;\n  apiKey?: string;\n  azureEndpoint?: string;\n  apiVersion?: string;\n  privacyConfig?: PrivacyConfig | null;\n  propagateContext?: boolean;\n  /** Pass the `openai` module directly to bypass `tryRequire` (required in bundler environments). */\n  openaiModule?: unknown;\n}\n\nexport class AzureOpenAI extends BaseAIProvider {\n  private _client: unknown;\n  private _propagateContext: boolean;\n  readonly chat: {\n    completions: {\n      create: (params: Record<string, unknown>, amplitudeOverrides?: ProviderTrackOptions) => Promise<unknown>;\n      parse: (params: Record<string, unknown>, amplitudeOverrides?: ProviderTrackOptions) => Promise<unknown>;\n    };\n  };\n\n  constructor(options: AzureOpenAIOptions) {\n    super({\n      amplitude: options.amplitude,\n      privacyConfig: options.privacyConfig,\n      providerName: 'azure-openai',\n    });\n\n    const mod =\n      (options.openaiModule as Record<string, unknown> | null) ?? _OpenAIModule;\n    if (mod == null) {\n      throw new Error(\n        'openai package is required for Azure OpenAI. Install it with: npm install openai — or pass the module directly via the openaiModule option.',\n      );\n    }\n\n    const AzureOpenAISDK = mod.AzureOpenAI as new (\n      opts: Record<string, unknown>,\n    ) => unknown;\n\n    const clientOpts: Record<string, unknown> = {};\n    if (options.apiKey) clientOpts.apiKey = options.apiKey;\n    if (options.azureEndpoint) clientOpts.baseURL = options.azureEndpoint;\n    if (options.apiVersion) {\n      clientOpts.defaultQuery = { 'api-version': options.apiVersion };\n    }\n\n    this._client = new AzureOpenAISDK(clientOpts);\n    this._propagateContext =\n      options.propagateContext ?? getDefaultPropagateContext();\n    const clientObj = this._client as Record<string, unknown>;\n    const originalChat = clientObj.chat as Record<string, unknown>;\n    const originalCompletions = originalChat.completions as Record<\n      string,\n      unknown\n    >;\n    const wrappedCompletions = new WrappedCompletions(\n      originalCompletions,\n      this.trackFn(),\n      this._amplitude,\n      this._privacyConfig,\n      this._propagateContext,\n      'azure-openai',\n    );\n\n    this.chat = {\n      completions: {\n        create: (params: Record<string, unknown>, amplitudeOverrides?: ProviderTrackOptions): Promise<unknown> =>\n          wrappedCompletions.create(params, amplitudeOverrides),\n        parse: (params: Record<string, unknown>, amplitudeOverrides?: ProviderTrackOptions): Promise<unknown> =>\n          wrappedCompletions.parse(params, amplitudeOverrides),\n      },\n    };\n  }\n\n  get client(): unknown {\n    return this._client;\n  }\n}\n"],"mappings":";;;;;AA2BA,IAAa,cAAb,cAAiC,eAAe;CAC9C,AAAQ;CACR,AAAQ;CACR,AAAS;CAOT,YAAY,SAA6B;AACvC,QAAM;GACJ,WAAW,QAAQ;GACnB,eAAe,QAAQ;GACvB,cAAc;GACf,CAAC;EAEF,MAAM,MACH,QAAQ,gBAAmD;AAC9D,MAAI,OAAO,KACT,OAAM,IAAI,MACR,8IACD;EAGH,MAAM,iBAAiB,IAAI;EAI3B,MAAMA,aAAsC,EAAE;AAC9C,MAAI,QAAQ,OAAQ,YAAW,SAAS,QAAQ;AAChD,MAAI,QAAQ,cAAe,YAAW,UAAU,QAAQ;AACxD,MAAI,QAAQ,WACV,YAAW,eAAe,EAAE,eAAe,QAAQ,YAAY;AAGjE,OAAK,UAAU,IAAI,eAAe,WAAW;AAC7C,OAAK,oBACH,QAAQ,oBAAoB,4BAA4B;EAG1D,MAAM,sBAFY,KAAK,QACQ,KACU;EAIzC,MAAM,qBAAqB,IAAI,mBAC7B,qBACA,KAAK,SAAS,EACd,KAAK,YACL,KAAK,gBACL,KAAK,mBACL,eACD;AAED,OAAK,OAAO,EACV,aAAa;GACX,SAAS,QAAiC,uBACxC,mBAAmB,OAAO,QAAQ,mBAAmB;GACvD,QAAQ,QAAiC,uBACvC,mBAAmB,MAAM,QAAQ,mBAAmB;GACvD,EACF;;CAGH,IAAI,SAAkB;AACpB,SAAO,KAAK"}

package/dist/providers/base.d.ts

@@ -1,5 +1,5 @@
 import { PrivacyConfig } from "../core/privacy.js";
-import { AmplitudeLike, AmplitudeOrAI, TrackFn } from "../types.js";
+import { AmplitudeLike, AmplitudeOrAI, TrackCallOptions, TrackFn } from "../types.js";
 import { TrackAiMessageOptions } from "../core/tracking.js";
 import { StreamingAccumulator } from "../utils/streaming.js";
 
@@ -22,6 +22,8 @@ interface ProviderTrackOptions {
   agentId?: string | null;
   parentAgentId?: string | null;
   customerOrgId?: string | null;
+  agentVersion?: string | null;
+  context?: Record<string, unknown> | null;
   env?: string | null;
   groups?: Record<string, unknown> | null;
   eventProperties?: Record<string, unknown> | null;
@@ -39,6 +41,13 @@ interface ProviderTrackOptions {
 declare function applySessionContext(overrides?: ProviderTrackOptions): ProviderTrackOptions & {
   userId: string;
 };
+/**
+ * Extract all context fields from a resolved ProviderTrackOptions into a
+ * flat object suitable for spreading into _trackFn() / _track() calls.
+ * Ensures all 13 context fields propagate consistently.
+ */
+type TrackContextFields = Pick<TrackCallOptions, 'userId' | 'sessionId' | 'traceId' | 'turnId' | 'agentId' | 'parentAgentId' | 'customerOrgId' | 'agentVersion' | 'context' | 'env' | 'groups' | 'eventProperties'>;
+declare function contextFields(ctx: ProviderTrackOptions): TrackContextFields;
 declare abstract class BaseAIProvider {
   protected _amplitude: AmplitudeLike;
   protected _privacyConfig: PrivacyConfig | null;
@@ -66,5 +75,5 @@ declare class SimpleStreamingTracker {
   finalize(overrides?: ProviderTrackOptions): string;
 }
 //#endregion
-export { BaseAIProvider, ProviderTrackOptions, SimpleStreamingTracker, applySessionContext };
+export { BaseAIProvider, ProviderTrackOptions, SimpleStreamingTracker, TrackContextFields, applySessionContext, contextFields };
 //# sourceMappingURL=base.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"base.d.ts","names":[],"sources":["../../src/providers/base.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;;;;AAsJa,UApHI,oBAAA,CAoHkB;EAEX,MAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAIA,SAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAeK,OAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAAX,MAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAQM,OAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAIF,aAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAAyB,aAAA,CAAA,EAAA,MAAA,GAAA,IAAA;;WA5IpC;oBACS;;;;;;;;;;;;iBAaJ,mBAAA,aACH,uBACV;;;uBAqCmB,cAAA;wBACE;4BACI;;;eAIb;oBACK;;;yBAQK,KAAK;aA6BjB;4BAKe;;cAKf,sBAAA;;wBAEW;;;wBAIA;;;kBAeN,WAAW;;wBAQL;uBAIF"}
+{"version":3,"file":"base.d.ts","names":[],"sources":["../../src/providers/base.ts"],"sourcesContent":[],"mappings":";;;;;;;;;AA2DA;AAgDA;AAgBA;AAiBA;;;;AAOoB,UAjHH,oBAAA,CAiHG;EAQU,MAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAAL,SAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAmCZ,OAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAKe,MAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAAsB,OAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAKrC,aAAA,CAAA,EAAA,MAAA,GAAsB,IAAA;EAEX,aAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAIA,YAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAeK,OAAA,CAAA,EAlLjB,MAkLiB,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;EAAX,GAAA,CAAA,EAAA,MAAA,GAAA,IAAA;EAQM,MAAA,CAAA,EAxLb,MAwLa,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;EAIF,eAAA,CAAA,EA3LF,MA2LE,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;EAAyB;;;;;;;;;;;iBA9K/B,mBAAA,aACH,uBACV;;;;;;;;KA8CS,kBAAA,GAAqB,KAC/B;iBAec,aAAA,MAAmB,uBAAuB;uBAiBpC,cAAA;wBACE;4BACI;;;eAIb;oBACK;;;yBAQK,KAAK;aAmCjB;4BAKe;;cAKf,sBAAA;;wBAEW;;;wBAIA;;;kBAeN,WAAW;;wBAQL;uBAIF"}

package/dist/providers/base.js

@@ -25,10 +25,14 @@ function applySessionContext(overrides = {}) {
 		if (!result.agentId) result.agentId = ctx.agentId;
 		if (!result.parentAgentId) result.parentAgentId = ctx.parentAgentId;
 		if (!result.customerOrgId) result.customerOrgId = ctx.customerOrgId;
+		if (!result.agentVersion) result.agentVersion = ctx.agentVersion;
+		if (!result.context) result.context = ctx.context;
 		if (!result.env) result.env = ctx.env;
 		if (!result.groups) result.groups = ctx.groups;
-		const turnId = ctx.nextTurnId();
-		if (turnId != null && result.turnId == null) result.turnId = turnId;
+		if (result.turnId == null) {
+			const turnId = ctx.nextTurnId();
+			if (turnId != null) result.turnId = turnId;
+		}
 		const existingEp = result.eventProperties;
 		const ep = existingEp != null ? { ...existingEp } : {};
 		if (ctx.idleTimeoutMinutes != null && !(PROP_IDLE_TIMEOUT_MINUTES in ep)) ep[PROP_IDLE_TIMEOUT_MINUTES] = ctx.idleTimeoutMinutes;
@@ -37,6 +41,22 @@ function applySessionContext(overrides = {}) {
 	}
 	return result;
 }
+function contextFields(ctx) {
+	return {
+		userId: ctx.userId ?? "unknown",
+		sessionId: ctx.sessionId,
+		traceId: ctx.traceId,
+		turnId: ctx.turnId ?? void 0,
+		agentId: ctx.agentId,
+		parentAgentId: ctx.parentAgentId,
+		customerOrgId: ctx.customerOrgId,
+		agentVersion: ctx.agentVersion,
+		context: ctx.context,
+		env: ctx.env,
+		groups: ctx.groups,
+		eventProperties: ctx.eventProperties
+	};
+}
 var BaseAIProvider = class {
 	_amplitude;
 	_privacyConfig;
@@ -55,8 +75,11 @@ var BaseAIProvider = class {
 			agentId: opts.agentId,
 			parentAgentId: opts.parentAgentId,
 			customerOrgId: opts.customerOrgId,
+			agentVersion: opts.agentVersion,
+			context: opts.context,
 			env: opts.env,
-			groups: opts.groups
+			groups: opts.groups,
+			eventProperties: opts.eventProperties
 		});
 		return trackAiMessage({
 			...opts,
@@ -68,8 +91,11 @@ var BaseAIProvider = class {
 			agentId: merged.agentId ?? opts.agentId,
 			parentAgentId: merged.parentAgentId ?? opts.parentAgentId,
 			customerOrgId: merged.customerOrgId ?? opts.customerOrgId,
+			agentVersion: merged.agentVersion ?? opts.agentVersion,
+			context: merged.context ?? opts.context,
 			env: merged.env ?? opts.env,
 			groups: merged.groups ?? opts.groups,
+			eventProperties: merged.eventProperties ?? opts.eventProperties,
 			privacyConfig: this._privacyConfig
 		});
 	}
@@ -109,7 +135,7 @@ var SimpleStreamingTracker = class {
 	finalize(overrides = {}) {
 		const state = this.accumulator.getState();
 		return this._trackFn({
-			userId: overrides.userId ?? "unknown",
+			...contextFields(overrides),
 			modelName: this._modelName,
 			provider: this._providerName,
 			responseContent: state.content,
@@ -123,19 +149,11 @@ var SimpleStreamingTracker = class {
 			finishReason: state.finishReason,
 			toolCalls: state.toolCalls.length > 0 ? state.toolCalls : null,
 			providerTtfbMs: state.ttfbMs,
-			isStreaming: true,
-			sessionId: overrides.sessionId,
-			traceId: overrides.traceId,
-			turnId: overrides.turnId ?? void 0,
-			agentId: overrides.agentId,
-			parentAgentId: overrides.parentAgentId,
-			customerOrgId: overrides.customerOrgId,
-			env: overrides.env,
-			groups: overrides.groups
+			isStreaming: true
 		});
 	}
 };
 
 //#endregion
-export { BaseAIProvider, SimpleStreamingTracker, applySessionContext };
+export { BaseAIProvider, SimpleStreamingTracker, applySessionContext, contextFields };
 //# sourceMappingURL=base.js.map

package/dist/providers/base.js.map

@@ -1 +1 @@
-{"version":3,"file":"base.js","names":["result: Record<string, unknown>"],"sources":["../../src/providers/base.ts"],"sourcesContent":["/**\n * Base class for AI provider wrappers.\n *\n * Provides shared tracking logic and session context integration.\n */\n\nimport { getActiveContext } from '../context.js';\nimport {\n  PROP_IDLE_TIMEOUT_MINUTES,\n  PROP_SESSION_REPLAY_ID,\n} from '../core/constants.js';\nimport type { PrivacyConfig } from '../core/privacy.js';\nimport {\n  trackAiMessage,\n  type TrackAiMessageOptions,\n} from '../core/tracking.js';\nimport {\n  resolveAmplitude,\n  type AmplitudeLike,\n  type AmplitudeOrAI,\n  type TrackCallOptions,\n  type TrackFn,\n} from '../types.js';\nimport { StreamingAccumulator } from '../utils/streaming.js';\n\n/**\n * Per-call context overrides for provider wrappers.\n *\n * Pass as the second argument to wrapped provider methods\n * (e.g., `openai.chat.completions.create(params, overrides)`)\n * to set Amplitude tracking context for that specific call.\n * Any fields left `null`/`undefined` are filled from the\n * active `SessionContext` via `AsyncLocalStorage`.\n */\nexport interface ProviderTrackOptions {\n  userId?: string | null;\n  sessionId?: string | null;\n  traceId?: string | null;\n  turnId?: number | null;\n  agentId?: string | null;\n  parentAgentId?: string | null;\n  customerOrgId?: string | null;\n  env?: string | null;\n  groups?: Record<string, unknown> | null;\n  eventProperties?: Record<string, unknown> | null;\n  /**\n   * Controls whether provider wrappers auto-track user input payloads.\n   * Set to false when you already call `trackUserMessage()` explicitly.\n   */\n  trackInputMessages?: boolean;\n}\n\n/**\n * Apply session context fields from AsyncLocalStorage to tracking options.\n * Returns a merged set of fields with explicit values taking precedence.\n * Also injects idle_timeout_minutes and session_replay_id from the context.\n */\nexport function applySessionContext(\n  overrides: ProviderTrackOptions = {},\n): ProviderTrackOptions & { userId: string } {\n  const ctx = getActiveContext();\n  const result: Record<string, unknown> = { ...overrides };\n\n  if (ctx != null) {\n    if (!result.userId) result.userId = ctx.userId;\n    if (!result.sessionId) result.sessionId = ctx.sessionId;\n    if (!result.traceId) result.traceId = ctx.traceId;\n    if (!result.agentId) result.agentId = ctx.agentId;\n    if (!result.parentAgentId) result.parentAgentId = ctx.parentAgentId;\n    if (!result.customerOrgId) result.customerOrgId = ctx.customerOrgId;\n    if (!result.env) result.env = ctx.env;\n    if (!result.groups) result.groups = ctx.groups;\n\n    const turnId = ctx.nextTurnId();\n    if (turnId != null && result.turnId == null) result.turnId = turnId;\n\n    const existingEp = result.eventProperties as Record<string, unknown> | null;\n    const ep = existingEp != null ? { ...existingEp } : {};\n    if (ctx.idleTimeoutMinutes != null && !(PROP_IDLE_TIMEOUT_MINUTES in ep)) {\n      ep[PROP_IDLE_TIMEOUT_MINUTES] = ctx.idleTimeoutMinutes;\n    }\n    if (\n      ctx.deviceId &&\n      ctx.browserSessionId &&\n      !(PROP_SESSION_REPLAY_ID in ep)\n    ) {\n      ep[PROP_SESSION_REPLAY_ID] = `${ctx.deviceId}/${ctx.browserSessionId}`;\n    }\n    if (Object.keys(ep).length > 0) {\n      result.eventProperties = ep;\n    }\n  }\n\n  return result as unknown as ProviderTrackOptions & { userId: string };\n}\n\nexport abstract class BaseAIProvider {\n  protected _amplitude: AmplitudeLike;\n  protected _privacyConfig: PrivacyConfig | null;\n  readonly _providerName: string;\n\n  constructor(options: {\n    amplitude: AmplitudeOrAI;\n    privacyConfig?: PrivacyConfig | null;\n    providerName: string;\n  }) {\n    this._amplitude = resolveAmplitude(options.amplitude);\n    this._privacyConfig = options.privacyConfig ?? null;\n    this._providerName = options.providerName;\n  }\n\n  protected _track(opts: Omit<TrackAiMessageOptions, 'amplitude'>): string {\n    const merged = applySessionContext({\n      userId: opts.userId,\n      sessionId: opts.sessionId,\n      traceId: opts.traceId,\n      turnId: opts.turnId,\n      agentId: opts.agentId,\n      parentAgentId: opts.parentAgentId,\n      customerOrgId: opts.customerOrgId,\n      env: opts.env,\n      groups: opts.groups,\n    });\n\n    return trackAiMessage({\n      ...opts,\n      amplitude: this._amplitude,\n      userId: merged.userId ?? opts.userId,\n      sessionId: merged.sessionId ?? opts.sessionId,\n      traceId: merged.traceId ?? opts.traceId,\n      turnId: merged.turnId ?? opts.turnId,\n      agentId: merged.agentId ?? opts.agentId,\n      parentAgentId: merged.parentAgentId ?? opts.parentAgentId,\n      customerOrgId: merged.customerOrgId ?? opts.customerOrgId,\n      env: merged.env ?? opts.env,\n      groups: merged.groups ?? opts.groups,\n      privacyConfig: this._privacyConfig,\n    });\n  }\n\n  trackFn(): TrackFn {\n    return (opts: TrackCallOptions) =>\n      this._track(opts as Omit<TrackAiMessageOptions, 'amplitude'>);\n  }\n\n  createStreamingTracker(): SimpleStreamingTracker {\n    return new SimpleStreamingTracker(this);\n  }\n}\n\nexport class SimpleStreamingTracker {\n  private _trackFn: TrackFn;\n  readonly accumulator: StreamingAccumulator;\n  private _modelName = 'unknown';\n  private _providerName: string;\n\n  constructor(provider: BaseAIProvider) {\n    this._trackFn = provider.trackFn();\n    this._providerName = provider._providerName;\n    this.accumulator = new StreamingAccumulator();\n  }\n\n  setModel(model: string): void {\n    this._modelName = model;\n    this.accumulator.model = model;\n  }\n\n  addContent(chunk: string): void {\n    this.accumulator.addContent(chunk);\n  }\n\n  setUsage(usage: Parameters<StreamingAccumulator['setUsage']>[0]): void {\n    this.accumulator.setUsage(usage);\n  }\n\n  setFinishReason(reason: string): void {\n    this.accumulator.finishReason = reason;\n  }\n\n  addToolCall(toolCall: Record<string, unknown>): void {\n    this.accumulator.addToolCall(toolCall);\n  }\n\n  finalize(overrides: ProviderTrackOptions = {}): string {\n    const state = this.accumulator.getState();\n\n    return this._trackFn({\n      userId: overrides.userId ?? 'unknown',\n      modelName: this._modelName,\n      provider: this._providerName,\n      responseContent: state.content,\n      latencyMs: this.accumulator.elapsedMs,\n      inputTokens: state.inputTokens,\n      outputTokens: state.outputTokens,\n      totalTokens: state.totalTokens,\n      reasoningTokens: state.reasoningTokens,\n      cacheReadInputTokens: state.cacheReadTokens,\n      cacheCreationInputTokens: state.cacheCreationTokens,\n      finishReason: state.finishReason,\n      toolCalls: state.toolCalls.length > 0 ? state.toolCalls : null,\n      providerTtfbMs: state.ttfbMs,\n      isStreaming: true,\n      sessionId: overrides.sessionId,\n      traceId: overrides.traceId,\n      turnId: overrides.turnId ?? undefined,\n      agentId: overrides.agentId,\n      parentAgentId: overrides.parentAgentId,\n      customerOrgId: overrides.customerOrgId,\n      env: overrides.env,\n      groups: overrides.groups,\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;AAyDA,SAAgB,oBACd,YAAkC,EAAE,EACO;CAC3C,MAAM,MAAM,kBAAkB;CAC9B,MAAMA,SAAkC,EAAE,GAAG,WAAW;AAExD,KAAI,OAAO,MAAM;AACf,MAAI,CAAC,OAAO,OAAQ,QAAO,SAAS,IAAI;AACxC,MAAI,CAAC,OAAO,UAAW,QAAO,YAAY,IAAI;AAC9C,MAAI,CAAC,OAAO,QAAS,QAAO,UAAU,IAAI;AAC1C,MAAI,CAAC,OAAO,QAAS,QAAO,UAAU,IAAI;AAC1C,MAAI,CAAC,OAAO,cAAe,QAAO,gBAAgB,IAAI;AACtD,MAAI,CAAC,OAAO,cAAe,QAAO,gBAAgB,IAAI;AACtD,MAAI,CAAC,OAAO,IAAK,QAAO,MAAM,IAAI;AAClC,MAAI,CAAC,OAAO,OAAQ,QAAO,SAAS,IAAI;EAExC,MAAM,SAAS,IAAI,YAAY;AAC/B,MAAI,UAAU,QAAQ,OAAO,UAAU,KAAM,QAAO,SAAS;EAE7D,MAAM,aAAa,OAAO;EAC1B,MAAM,KAAK,cAAc,OAAO,EAAE,GAAG,YAAY,GAAG,EAAE;AACtD,MAAI,IAAI,sBAAsB,QAAQ,EAAE,6BAA6B,IACnE,IAAG,6BAA6B,IAAI;AAEtC,MACE,IAAI,YACJ,IAAI,oBACJ,EAAE,0BAA0B,IAE5B,IAAG,0BAA0B,GAAG,IAAI,SAAS,GAAG,IAAI;AAEtD,MAAI,OAAO,KAAK,GAAG,CAAC,SAAS,EAC3B,QAAO,kBAAkB;;AAI7B,QAAO;;AAGT,IAAsB,iBAAtB,MAAqC;CACnC,AAAU;CACV,AAAU;CACV,AAAS;CAET,YAAY,SAIT;AACD,OAAK,aAAa,iBAAiB,QAAQ,UAAU;AACrD,OAAK,iBAAiB,QAAQ,iBAAiB;AAC/C,OAAK,gBAAgB,QAAQ;;CAG/B,AAAU,OAAO,MAAwD;EACvE,MAAM,SAAS,oBAAoB;GACjC,QAAQ,KAAK;GACb,WAAW,KAAK;GAChB,SAAS,KAAK;GACd,QAAQ,KAAK;GACb,SAAS,KAAK;GACd,eAAe,KAAK;GACpB,eAAe,KAAK;GACpB,KAAK,KAAK;GACV,QAAQ,KAAK;GACd,CAAC;AAEF,SAAO,eAAe;GACpB,GAAG;GACH,WAAW,KAAK;GAChB,QAAQ,OAAO,UAAU,KAAK;GAC9B,WAAW,OAAO,aAAa,KAAK;GACpC,SAAS,OAAO,WAAW,KAAK;GAChC,QAAQ,OAAO,UAAU,KAAK;GAC9B,SAAS,OAAO,WAAW,KAAK;GAChC,eAAe,OAAO,iBAAiB,KAAK;GAC5C,eAAe,OAAO,iBAAiB,KAAK;GAC5C,KAAK,OAAO,OAAO,KAAK;GACxB,QAAQ,OAAO,UAAU,KAAK;GAC9B,eAAe,KAAK;GACrB,CAAC;;CAGJ,UAAmB;AACjB,UAAQ,SACN,KAAK,OAAO,KAAiD;;CAGjE,yBAAiD;AAC/C,SAAO,IAAI,uBAAuB,KAAK;;;AAI3C,IAAa,yBAAb,MAAoC;CAClC,AAAQ;CACR,AAAS;CACT,AAAQ,aAAa;CACrB,AAAQ;CAER,YAAY,UAA0B;AACpC,OAAK,WAAW,SAAS,SAAS;AAClC,OAAK,gBAAgB,SAAS;AAC9B,OAAK,cAAc,IAAI,sBAAsB;;CAG/C,SAAS,OAAqB;AAC5B,OAAK,aAAa;AAClB,OAAK,YAAY,QAAQ;;CAG3B,WAAW,OAAqB;AAC9B,OAAK,YAAY,WAAW,MAAM;;CAGpC,SAAS,OAA8D;AACrE,OAAK,YAAY,SAAS,MAAM;;CAGlC,gBAAgB,QAAsB;AACpC,OAAK,YAAY,eAAe;;CAGlC,YAAY,UAAyC;AACnD,OAAK,YAAY,YAAY,SAAS;;CAGxC,SAAS,YAAkC,EAAE,EAAU;EACrD,MAAM,QAAQ,KAAK,YAAY,UAAU;AAEzC,SAAO,KAAK,SAAS;GACnB,QAAQ,UAAU,UAAU;GAC5B,WAAW,KAAK;GAChB,UAAU,KAAK;GACf,iBAAiB,MAAM;GACvB,WAAW,KAAK,YAAY;GAC5B,aAAa,MAAM;GACnB,cAAc,MAAM;GACpB,aAAa,MAAM;GACnB,iBAAiB,MAAM;GACvB,sBAAsB,MAAM;GAC5B,0BAA0B,MAAM;GAChC,cAAc,MAAM;GACpB,WAAW,MAAM,UAAU,SAAS,IAAI,MAAM,YAAY;GAC1D,gBAAgB,MAAM;GACtB,aAAa;GACb,WAAW,UAAU;GACrB,SAAS,UAAU;GACnB,QAAQ,UAAU,UAAU;GAC5B,SAAS,UAAU;GACnB,eAAe,UAAU;GACzB,eAAe,UAAU;GACzB,KAAK,UAAU;GACf,QAAQ,UAAU;GACnB,CAAC"}
+{"version":3,"file":"base.js","names":["result: Record<string, unknown>"],"sources":["../../src/providers/base.ts"],"sourcesContent":["/**\n * Base class for AI provider wrappers.\n *\n * Provides shared tracking logic and session context integration.\n */\n\nimport { getActiveContext } from '../context.js';\nimport {\n  PROP_IDLE_TIMEOUT_MINUTES,\n  PROP_SESSION_REPLAY_ID,\n} from '../core/constants.js';\nimport type { PrivacyConfig } from '../core/privacy.js';\nimport {\n  trackAiMessage,\n  type TrackAiMessageOptions,\n} from '../core/tracking.js';\nimport {\n  resolveAmplitude,\n  type AmplitudeLike,\n  type AmplitudeOrAI,\n  type TrackCallOptions,\n  type TrackFn,\n} from '../types.js';\nimport { StreamingAccumulator } from '../utils/streaming.js';\n\n/**\n * Per-call context overrides for provider wrappers.\n *\n * Pass as the second argument to wrapped provider methods\n * (e.g., `openai.chat.completions.create(params, overrides)`)\n * to set Amplitude tracking context for that specific call.\n * Any fields left `null`/`undefined` are filled from the\n * active `SessionContext` via `AsyncLocalStorage`.\n */\nexport interface ProviderTrackOptions {\n  userId?: string | null;\n  sessionId?: string | null;\n  traceId?: string | null;\n  turnId?: number | null;\n  agentId?: string | null;\n  parentAgentId?: string | null;\n  customerOrgId?: string | null;\n  agentVersion?: string | null;\n  context?: Record<string, unknown> | null;\n  env?: string | null;\n  groups?: Record<string, unknown> | null;\n  eventProperties?: Record<string, unknown> | null;\n  /**\n   * Controls whether provider wrappers auto-track user input payloads.\n   * Set to false when you already call `trackUserMessage()` explicitly.\n   */\n  trackInputMessages?: boolean;\n}\n\n/**\n * Apply session context fields from AsyncLocalStorage to tracking options.\n * Returns a merged set of fields with explicit values taking precedence.\n * Also injects idle_timeout_minutes and session_replay_id from the context.\n */\nexport function applySessionContext(\n  overrides: ProviderTrackOptions = {},\n): ProviderTrackOptions & { userId: string } {\n  const ctx = getActiveContext();\n  const result: Record<string, unknown> = { ...overrides };\n\n  if (ctx != null) {\n    if (!result.userId) result.userId = ctx.userId;\n    if (!result.sessionId) result.sessionId = ctx.sessionId;\n    if (!result.traceId) result.traceId = ctx.traceId;\n    if (!result.agentId) result.agentId = ctx.agentId;\n    if (!result.parentAgentId) result.parentAgentId = ctx.parentAgentId;\n    if (!result.customerOrgId) result.customerOrgId = ctx.customerOrgId;\n    if (!result.agentVersion) result.agentVersion = ctx.agentVersion;\n    if (!result.context) result.context = ctx.context;\n    if (!result.env) result.env = ctx.env;\n    if (!result.groups) result.groups = ctx.groups;\n\n    if (result.turnId == null) {\n      const turnId = ctx.nextTurnId();\n      if (turnId != null) result.turnId = turnId;\n    }\n\n    const existingEp = result.eventProperties as Record<string, unknown> | null;\n    const ep = existingEp != null ? { ...existingEp } : {};\n    if (ctx.idleTimeoutMinutes != null && !(PROP_IDLE_TIMEOUT_MINUTES in ep)) {\n      ep[PROP_IDLE_TIMEOUT_MINUTES] = ctx.idleTimeoutMinutes;\n    }\n    if (\n      ctx.deviceId &&\n      ctx.browserSessionId &&\n      !(PROP_SESSION_REPLAY_ID in ep)\n    ) {\n      ep[PROP_SESSION_REPLAY_ID] = `${ctx.deviceId}/${ctx.browserSessionId}`;\n    }\n    if (Object.keys(ep).length > 0) {\n      result.eventProperties = ep;\n    }\n  }\n\n  return result as unknown as ProviderTrackOptions & { userId: string };\n}\n\n/**\n * Extract all context fields from a resolved ProviderTrackOptions into a\n * flat object suitable for spreading into _trackFn() / _track() calls.\n * Ensures all 13 context fields propagate consistently.\n */\nexport type TrackContextFields = Pick<\n  TrackCallOptions,\n  | 'userId'\n  | 'sessionId'\n  | 'traceId'\n  | 'turnId'\n  | 'agentId'\n  | 'parentAgentId'\n  | 'customerOrgId'\n  | 'agentVersion'\n  | 'context'\n  | 'env'\n  | 'groups'\n  | 'eventProperties'\n>;\n\nexport function contextFields(ctx: ProviderTrackOptions): TrackContextFields {\n  return {\n    userId: ctx.userId ?? 'unknown',\n    sessionId: ctx.sessionId,\n    traceId: ctx.traceId,\n    turnId: ctx.turnId ?? undefined,\n    agentId: ctx.agentId,\n    parentAgentId: ctx.parentAgentId,\n    customerOrgId: ctx.customerOrgId,\n    agentVersion: ctx.agentVersion,\n    context: ctx.context,\n    env: ctx.env,\n    groups: ctx.groups,\n    eventProperties: ctx.eventProperties,\n  };\n}\n\nexport abstract class BaseAIProvider {\n  protected _amplitude: AmplitudeLike;\n  protected _privacyConfig: PrivacyConfig | null;\n  readonly _providerName: string;\n\n  constructor(options: {\n    amplitude: AmplitudeOrAI;\n    privacyConfig?: PrivacyConfig | null;\n    providerName: string;\n  }) {\n    this._amplitude = resolveAmplitude(options.amplitude);\n    this._privacyConfig = options.privacyConfig ?? null;\n    this._providerName = options.providerName;\n  }\n\n  protected _track(opts: Omit<TrackAiMessageOptions, 'amplitude'>): string {\n    const merged = applySessionContext({\n      userId: opts.userId,\n      sessionId: opts.sessionId,\n      traceId: opts.traceId,\n      turnId: opts.turnId,\n      agentId: opts.agentId,\n      parentAgentId: opts.parentAgentId,\n      customerOrgId: opts.customerOrgId,\n      agentVersion: opts.agentVersion,\n      context: opts.context,\n      env: opts.env,\n      groups: opts.groups,\n      eventProperties: opts.eventProperties,\n    });\n\n    return trackAiMessage({\n      ...opts,\n      amplitude: this._amplitude,\n      userId: merged.userId ?? opts.userId,\n      sessionId: merged.sessionId ?? opts.sessionId,\n      traceId: merged.traceId ?? opts.traceId,\n      turnId: merged.turnId ?? opts.turnId,\n      agentId: merged.agentId ?? opts.agentId,\n      parentAgentId: merged.parentAgentId ?? opts.parentAgentId,\n      customerOrgId: merged.customerOrgId ?? opts.customerOrgId,\n      agentVersion: merged.agentVersion ?? opts.agentVersion,\n      context: merged.context ?? opts.context,\n      env: merged.env ?? opts.env,\n      groups: merged.groups ?? opts.groups,\n      eventProperties: merged.eventProperties ?? opts.eventProperties,\n      privacyConfig: this._privacyConfig,\n    });\n  }\n\n  trackFn(): TrackFn {\n    return (opts: TrackCallOptions) =>\n      this._track(opts as Omit<TrackAiMessageOptions, 'amplitude'>);\n  }\n\n  createStreamingTracker(): SimpleStreamingTracker {\n    return new SimpleStreamingTracker(this);\n  }\n}\n\nexport class SimpleStreamingTracker {\n  private _trackFn: TrackFn;\n  readonly accumulator: StreamingAccumulator;\n  private _modelName = 'unknown';\n  private _providerName: string;\n\n  constructor(provider: BaseAIProvider) {\n    this._trackFn = provider.trackFn();\n    this._providerName = provider._providerName;\n    this.accumulator = new StreamingAccumulator();\n  }\n\n  setModel(model: string): void {\n    this._modelName = model;\n    this.accumulator.model = model;\n  }\n\n  addContent(chunk: string): void {\n    this.accumulator.addContent(chunk);\n  }\n\n  setUsage(usage: Parameters<StreamingAccumulator['setUsage']>[0]): void {\n    this.accumulator.setUsage(usage);\n  }\n\n  setFinishReason(reason: string): void {\n    this.accumulator.finishReason = reason;\n  }\n\n  addToolCall(toolCall: Record<string, unknown>): void {\n    this.accumulator.addToolCall(toolCall);\n  }\n\n  finalize(overrides: ProviderTrackOptions = {}): string {\n    const state = this.accumulator.getState();\n\n    return this._trackFn({\n      ...contextFields(overrides),\n      modelName: this._modelName,\n      provider: this._providerName,\n      responseContent: state.content,\n      latencyMs: this.accumulator.elapsedMs,\n      inputTokens: state.inputTokens,\n      outputTokens: state.outputTokens,\n      totalTokens: state.totalTokens,\n      reasoningTokens: state.reasoningTokens,\n      cacheReadInputTokens: state.cacheReadTokens,\n      cacheCreationInputTokens: state.cacheCreationTokens,\n      finishReason: state.finishReason,\n      toolCalls: state.toolCalls.length > 0 ? state.toolCalls : null,\n      providerTtfbMs: state.ttfbMs,\n      isStreaming: true,\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;AA2DA,SAAgB,oBACd,YAAkC,EAAE,EACO;CAC3C,MAAM,MAAM,kBAAkB;CAC9B,MAAMA,SAAkC,EAAE,GAAG,WAAW;AAExD,KAAI,OAAO,MAAM;AACf,MAAI,CAAC,OAAO,OAAQ,QAAO,SAAS,IAAI;AACxC,MAAI,CAAC,OAAO,UAAW,QAAO,YAAY,IAAI;AAC9C,MAAI,CAAC,OAAO,QAAS,QAAO,UAAU,IAAI;AAC1C,MAAI,CAAC,OAAO,QAAS,QAAO,UAAU,IAAI;AAC1C,MAAI,CAAC,OAAO,cAAe,QAAO,gBAAgB,IAAI;AACtD,MAAI,CAAC,OAAO,cAAe,QAAO,gBAAgB,IAAI;AACtD,MAAI,CAAC,OAAO,aAAc,QAAO,eAAe,IAAI;AACpD,MAAI,CAAC,OAAO,QAAS,QAAO,UAAU,IAAI;AAC1C,MAAI,CAAC,OAAO,IAAK,QAAO,MAAM,IAAI;AAClC,MAAI,CAAC,OAAO,OAAQ,QAAO,SAAS,IAAI;AAExC,MAAI,OAAO,UAAU,MAAM;GACzB,MAAM,SAAS,IAAI,YAAY;AAC/B,OAAI,UAAU,KAAM,QAAO,SAAS;;EAGtC,MAAM,aAAa,OAAO;EAC1B,MAAM,KAAK,cAAc,OAAO,EAAE,GAAG,YAAY,GAAG,EAAE;AACtD,MAAI,IAAI,sBAAsB,QAAQ,EAAE,6BAA6B,IACnE,IAAG,6BAA6B,IAAI;AAEtC,MACE,IAAI,YACJ,IAAI,oBACJ,EAAE,0BAA0B,IAE5B,IAAG,0BAA0B,GAAG,IAAI,SAAS,GAAG,IAAI;AAEtD,MAAI,OAAO,KAAK,GAAG,CAAC,SAAS,EAC3B,QAAO,kBAAkB;;AAI7B,QAAO;;AAwBT,SAAgB,cAAc,KAA+C;AAC3E,QAAO;EACL,QAAQ,IAAI,UAAU;EACtB,WAAW,IAAI;EACf,SAAS,IAAI;EACb,QAAQ,IAAI,UAAU;EACtB,SAAS,IAAI;EACb,eAAe,IAAI;EACnB,eAAe,IAAI;EACnB,cAAc,IAAI;EAClB,SAAS,IAAI;EACb,KAAK,IAAI;EACT,QAAQ,IAAI;EACZ,iBAAiB,IAAI;EACtB;;AAGH,IAAsB,iBAAtB,MAAqC;CACnC,AAAU;CACV,AAAU;CACV,AAAS;CAET,YAAY,SAIT;AACD,OAAK,aAAa,iBAAiB,QAAQ,UAAU;AACrD,OAAK,iBAAiB,QAAQ,iBAAiB;AAC/C,OAAK,gBAAgB,QAAQ;;CAG/B,AAAU,OAAO,MAAwD;EACvE,MAAM,SAAS,oBAAoB;GACjC,QAAQ,KAAK;GACb,WAAW,KAAK;GAChB,SAAS,KAAK;GACd,QAAQ,KAAK;GACb,SAAS,KAAK;GACd,eAAe,KAAK;GACpB,eAAe,KAAK;GACpB,cAAc,KAAK;GACnB,SAAS,KAAK;GACd,KAAK,KAAK;GACV,QAAQ,KAAK;GACb,iBAAiB,KAAK;GACvB,CAAC;AAEF,SAAO,eAAe;GACpB,GAAG;GACH,WAAW,KAAK;GAChB,QAAQ,OAAO,UAAU,KAAK;GAC9B,WAAW,OAAO,aAAa,KAAK;GACpC,SAAS,OAAO,WAAW,KAAK;GAChC,QAAQ,OAAO,UAAU,KAAK;GAC9B,SAAS,OAAO,WAAW,KAAK;GAChC,eAAe,OAAO,iBAAiB,KAAK;GAC5C,eAAe,OAAO,iBAAiB,KAAK;GAC5C,cAAc,OAAO,gBAAgB,KAAK;GAC1C,SAAS,OAAO,WAAW,KAAK;GAChC,KAAK,OAAO,OAAO,KAAK;GACxB,QAAQ,OAAO,UAAU,KAAK;GAC9B,iBAAiB,OAAO,mBAAmB,KAAK;GAChD,eAAe,KAAK;GACrB,CAAC;;CAGJ,UAAmB;AACjB,UAAQ,SACN,KAAK,OAAO,KAAiD;;CAGjE,yBAAiD;AAC/C,SAAO,IAAI,uBAAuB,KAAK;;;AAI3C,IAAa,yBAAb,MAAoC;CAClC,AAAQ;CACR,AAAS;CACT,AAAQ,aAAa;CACrB,AAAQ;CAER,YAAY,UAA0B;AACpC,OAAK,WAAW,SAAS,SAAS;AAClC,OAAK,gBAAgB,SAAS;AAC9B,OAAK,cAAc,IAAI,sBAAsB;;CAG/C,SAAS,OAAqB;AAC5B,OAAK,aAAa;AAClB,OAAK,YAAY,QAAQ;;CAG3B,WAAW,OAAqB;AAC9B,OAAK,YAAY,WAAW,MAAM;;CAGpC,SAAS,OAA8D;AACrE,OAAK,YAAY,SAAS,MAAM;;CAGlC,gBAAgB,QAAsB;AACpC,OAAK,YAAY,eAAe;;CAGlC,YAAY,UAAyC;AACnD,OAAK,YAAY,YAAY,SAAS;;CAGxC,SAAS,YAAkC,EAAE,EAAU;EACrD,MAAM,QAAQ,KAAK,YAAY,UAAU;AAEzC,SAAO,KAAK,SAAS;GACnB,GAAG,cAAc,UAAU;GAC3B,WAAW,KAAK;GAChB,UAAU,KAAK;GACf,iBAAiB,MAAM;GACvB,WAAW,KAAK,YAAY;GAC5B,aAAa,MAAM;GACnB,cAAc,MAAM;GACpB,aAAa,MAAM;GACnB,iBAAiB,MAAM;GACvB,sBAAsB,MAAM;GAC5B,0BAA0B,MAAM;GAChC,cAAc,MAAM;GACpB,WAAW,MAAM,UAAU,SAAS,IAAI,MAAM,YAAY;GAC1D,gBAAgB,MAAM;GACtB,aAAa;GACd,CAAC"}

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

@@ -1 +1 @@
-{"version":3,"file":"bedrock.d.ts","names":[],"sources":["../../src/providers/bedrock.ts"],"sourcesContent":[],"mappings":";;;;;;AA2Ba,cAbA,iBAaQ,EAAA,OAAA;cAZf,cAgBiB,EAhBD,MAgBC,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;AAY4B,UAxBlC,cAAA,CAwBkC;EAwFpB,SAAA,EA/GlB,aA+GkB;EAA0B,MAAA,EAAA,OAAA;EAxG5B,aAAA,CAAA,EALX,aAKW,GAAA,IAAA;EAAc;EA+Q3B,aAAA,CAAA,EAAA,OAAA;;cA/QH,OAAA,SAAgB,cAAA;;;uBAIN;mBAYE,0BAA0B;yBAwFpB,0BAA0B;;;;iBAuKzC,sBAAA;;;;;;;;aAQH,MAAM"}
+{"version":3,"file":"bedrock.d.ts","names":[],"sources":["../../src/providers/bedrock.ts"],"sourcesContent":[],"mappings":";;;;;;AA2Ba,cAbA,iBAaQ,EAAA,OAAA;cAZf,cAgBiB,EAhBD,MAgBC,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;AAY4B,UAxBlC,cAAA,CAwBkC;EA+EpB,SAAA,EAtGlB,aAsGkB;EAA0B,MAAA,EAAA,OAAA;EA/F5B,aAAA,CAAA,EALX,aAKW,GAAA,IAAA;EAAc;EAiQ3B,aAAA,CAAA,EAAA,OAAA;;cAjQH,OAAA,SAAgB,cAAA;;;uBAIN;mBAYE,0BAA0B;yBA+EpB,0BAA0B;;;;iBAkKzC,sBAAA;;;;;;;;aAQH,MAAM"}

package/dist/providers/bedrock.js

@@ -1,7 +1,7 @@
 import { tryRequire } from "../utils/resolve-module.js";
 import { calculateCost } from "../utils/costs.js";
 import { StreamingAccumulator } from "../utils/streaming.js";
-import { BaseAIProvider, applySessionContext } from "./base.js";
+import { BaseAIProvider, applySessionContext, contextFields } from "./base.js";
 
 //#region src/providers/bedrock.ts
 const _resolved = tryRequire("@aws-sdk/client-bedrock-runtime");
@@ -42,16 +42,11 @@ var Bedrock = class extends BaseAIProvider {
 			} catch {}
 			const ctx = applySessionContext();
 			this._track({
-				userId: ctx.userId ?? "unknown",
+				...contextFields(ctx),
 				modelName: modelId,
 				provider: "bedrock",
 				responseContent: extracted.text,
 				latencyMs,
-				sessionId: ctx.sessionId,
-				traceId: ctx.traceId,
-				turnId: ctx.turnId ?? void 0,
-				agentId: ctx.agentId,
-				env: ctx.env,
 				inputTokens: extracted.inputTokens,
 				outputTokens: extracted.outputTokens,
 				totalTokens: extracted.totalTokens,
@@ -69,15 +64,11 @@ var Bedrock = class extends BaseAIProvider {
 			const latencyMs = performance.now() - startTime;
 			const ctx = applySessionContext();
 			this._track({
-				userId: ctx.userId ?? "unknown",
+				...contextFields(ctx),
 				modelName: modelId,
 				provider: "bedrock",
 				responseContent: "",
 				latencyMs,
-				sessionId: ctx.sessionId,
-				traceId: ctx.traceId,
-				agentId: ctx.agentId,
-				env: ctx.env,
 				isError: true,
 				errorMessage: error instanceof Error ? error.message : String(error)
 			});
@@ -99,19 +90,15 @@ var Bedrock = class extends BaseAIProvider {
 			if (!_isAsyncIterable(stream)) throw new Error("Bedrock stream response is not AsyncIterable");
 			return {
 				...response,
-				stream: this._wrapConverseStream(modelId, params, stream)
+				stream: this._wrapConverseStream(modelId, params, stream, ctx)
 			};
 		} catch (error) {
 			this._track({
-				userId: ctx.userId ?? "unknown",
+				...contextFields(ctx),
 				modelName: modelId,
 				provider: "bedrock",
 				responseContent: "",
 				latencyMs: performance.now() - startTime,
-				sessionId: ctx.sessionId,
-				traceId: ctx.traceId,
-				agentId: ctx.agentId,
-				env: ctx.env,
 				isError: true,
 				errorMessage: error instanceof Error ? error.message : String(error),
 				isStreaming: true
@@ -122,10 +109,9 @@ var Bedrock = class extends BaseAIProvider {
 	get client() {
 		return this._client;
 	}
-	async *_wrapConverseStream(modelId, params, stream) {
+	async *_wrapConverseStream(modelId, params, stream, ctx) {
 		const accumulator = new StreamingAccumulator();
 		accumulator.model = modelId;
-		const ctx = applySessionContext();
 		try {
 			for await (const rawEvent of stream) {
 				const event = rawEvent;
@@ -141,7 +127,9 @@ var Bedrock = class extends BaseAIProvider {
 				accumulator.setUsage({
 					inputTokens: usage?.inputTokens,
 					outputTokens: usage?.outputTokens,
-					totalTokens: usage?.totalTokens
+					totalTokens: usage?.totalTokens,
+					cacheReadTokens: usage?.cacheReadInputTokens,
+					cacheCreationTokens: usage?.cacheWriteInputTokens
 				});
 				yield rawEvent;
 			}
@@ -156,20 +144,17 @@ var Bedrock = class extends BaseAIProvider {
 				costUsd = calculateCost({
 					modelName,
 					inputTokens: state.inputTokens,
-					outputTokens: state.outputTokens
+					outputTokens: state.outputTokens,
+					cacheReadInputTokens: state.cacheReadTokens ?? 0,
+					cacheCreationInputTokens: state.cacheCreationTokens ?? 0
 				});
 			} catch {}
 			this._track({
-				userId: ctx.userId ?? "unknown",
+				...contextFields(ctx),
 				modelName,
 				provider: "bedrock",
 				responseContent: state.content,
 				latencyMs: accumulator.elapsedMs,
-				sessionId: ctx.sessionId,
-				traceId: ctx.traceId,
-				turnId: ctx.turnId ?? void 0,
-				agentId: ctx.agentId,
-				env: ctx.env,
 				inputTokens: state.inputTokens,
 				outputTokens: state.outputTokens,
 				totalTokens: state.totalTokens,