@amplitude/ai 0.3.4 → 0.3.6

package/dist/providers/bedrock.js.map

@@ -1 +1 @@
-{"version":3,"file":"bedrock.js","names":["_BedrockModule: Record<string, unknown> | null","costUsd: number | null"],"sources":["../../src/providers/bedrock.ts"],"sourcesContent":["/**\n * AWS Bedrock provider wrapper with automatic tracking.\n *\n * Wraps the AWS SDK BedrockRuntimeClient's converse command.\n */\n\nimport type { PrivacyConfig } from '../core/privacy.js';\nimport type { AmplitudeOrAI, BedrockConverseResponse } from '../types.js';\nimport { calculateCost } from '../utils/costs.js';\nimport { tryRequire } from '../utils/resolve-module.js';\nimport { StreamingAccumulator } from '../utils/streaming.js';\nimport { applySessionContext, BaseAIProvider, contextFields } from './base.js';\n\nconst _resolved = tryRequire('@aws-sdk/client-bedrock-runtime');\nexport const BEDROCK_AVAILABLE = _resolved != null;\nconst _BedrockModule: Record<string, unknown> | null = _resolved;\n\nexport { _BedrockModule };\n\nexport interface BedrockOptions {\n  amplitude: AmplitudeOrAI;\n  client: unknown;\n  privacyConfig?: PrivacyConfig | null;\n  /** Pass the `@aws-sdk/client-bedrock-runtime` module directly to bypass `tryRequire` (required in bundler environments). */\n  bedrockModule?: unknown;\n}\n\nexport class Bedrock extends BaseAIProvider {\n  private _client: unknown;\n  private _bedrockMod: Record<string, unknown> | null;\n\n  constructor(options: BedrockOptions) {\n    super({\n      amplitude: options.amplitude,\n      privacyConfig: options.privacyConfig,\n      providerName: 'bedrock',\n    });\n    this._client = options.client;\n    this._bedrockMod =\n      (options.bedrockModule as Record<string, unknown> | null) ??\n      _BedrockModule;\n  }\n\n  async converse(params: Record<string, unknown>): Promise<unknown> {\n    const client = this._client as {\n      send: (command: unknown) => Promise<unknown>;\n    };\n    const modelId = String(params.modelId ?? 'unknown');\n\n    if (this._bedrockMod == null) {\n      throw new Error(\n        '@aws-sdk/client-bedrock-runtime is required. Install it with: npm install @aws-sdk/client-bedrock-runtime — or pass the module directly via the bedrockModule option.',\n      );\n    }\n\n    const ConverseCommand = this._bedrockMod.ConverseCommand as new (\n      opts: Record<string, unknown>,\n    ) => unknown;\n\n    const command = new ConverseCommand(params);\n    const startTime = performance.now();\n\n    try {\n      const response = await client.send(command);\n      const latencyMs = performance.now() - startTime;\n\n      const extracted = extractBedrockResponse(response);\n      let costUsd: number | null = null;\n      if (extracted.inputTokens != null && extracted.outputTokens != null) {\n        try {\n          costUsd = calculateCost({\n            modelName: modelId,\n            inputTokens: extracted.inputTokens,\n            outputTokens: extracted.outputTokens,\n            cacheReadInputTokens: extracted.cacheReadTokens ?? 0,\n            cacheCreationInputTokens: extracted.cacheWriteTokens ?? 0,\n            defaultProvider: 'bedrock',\n          });\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      const ctx = applySessionContext();\n      this._track({\n        ...contextFields(ctx),\n        modelName: modelId,\n        provider: 'bedrock',\n        responseContent: extracted.text,\n        latencyMs,\n        inputTokens: extracted.inputTokens,\n        outputTokens: extracted.outputTokens,\n        totalTokens: extracted.totalTokens,\n        totalCostUsd: costUsd,\n        finishReason: extracted.stopReason,\n        toolCalls:\n          extracted.toolCalls.length > 0 ? extracted.toolCalls : undefined,\n        systemPrompt: extracted.systemPrompt,\n        temperature: extracted.temperature,\n        topP: extracted.topP,\n        maxOutputTokens: extracted.maxOutputTokens,\n        isStreaming: false,\n      });\n\n      return response;\n    } catch (error) {\n      const latencyMs = performance.now() - startTime;\n      const ctx = applySessionContext();\n\n      this._track({\n        ...contextFields(ctx),\n        modelName: modelId,\n        provider: 'bedrock',\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  async converseStream(params: Record<string, unknown>): Promise<unknown> {\n    const modelId = String(params.modelId ?? 'unknown');\n    const ctx = applySessionContext();\n    const startTime = performance.now();\n    try {\n      const client = this._client as {\n        send: (command: unknown) => Promise<unknown>;\n      };\n      if (this._bedrockMod == null) {\n        throw new Error(\n          '@aws-sdk/client-bedrock-runtime is required. Install it with: npm install @aws-sdk/client-bedrock-runtime — or pass the module directly via the bedrockModule option.',\n        );\n      }\n\n      const ConverseStreamCommand = this._bedrockMod.ConverseStreamCommand as\n        | (new (opts: Record<string, unknown>) => unknown)\n        | undefined;\n      if (ConverseStreamCommand == null) {\n        throw new Error('Bedrock SDK does not expose ConverseStreamCommand');\n      }\n\n      const command = new ConverseStreamCommand(params);\n      const response = (await client.send(command)) as Record<string, unknown>;\n      const stream = response.stream as AsyncIterable<unknown> | undefined;\n      if (!_isAsyncIterable(stream)) {\n        throw new Error('Bedrock stream response is not AsyncIterable');\n      }\n\n      return {\n        ...response,\n        stream: this._wrapConverseStream(modelId, params, stream, ctx),\n      };\n    } catch (error) {\n      this._track({\n        ...contextFields(ctx),\n        modelName: modelId,\n        provider: 'bedrock',\n        responseContent: '',\n        latencyMs: performance.now() - startTime,\n        isError: true,\n        errorMessage: error instanceof Error ? error.message : String(error),\n        isStreaming: true,\n      });\n      throw error;\n    }\n  }\n\n  get client(): unknown {\n    return this._client;\n  }\n\n  private async *_wrapConverseStream(\n    modelId: string,\n    params: Record<string, unknown>,\n    stream: AsyncIterable<unknown>,\n    ctx: ReturnType<typeof applySessionContext>,\n  ): AsyncGenerator<unknown> {\n    const accumulator = new StreamingAccumulator();\n    accumulator.model = modelId;\n\n    try {\n      for await (const rawEvent of stream) {\n        const event = rawEvent as Record<string, unknown>;\n        const contentBlockDelta = event.contentBlockDelta as\n          | Record<string, unknown>\n          | undefined;\n        const delta = contentBlockDelta?.delta as\n          | Record<string, unknown>\n          | undefined;\n        if (delta?.text != null) {\n          accumulator.addContent(String(delta.text));\n        }\n\n        const contentBlockStart = event.contentBlockStart as\n          | Record<string, unknown>\n          | undefined;\n        const start = contentBlockStart?.start as\n          | Record<string, unknown>\n          | undefined;\n        if (start?.toolUse != null) {\n          accumulator.addToolCall(start.toolUse as Record<string, unknown>);\n        }\n\n        const messageStart = event.messageStart as\n          | Record<string, unknown>\n          | undefined;\n        if (messageStart?.model != null) {\n          accumulator.model = String(messageStart.model);\n        }\n\n        const messageStop = event.messageStop as\n          | Record<string, unknown>\n          | undefined;\n        if (messageStop?.stopReason != null) {\n          accumulator.finishReason = String(messageStop.stopReason);\n        }\n\n        const metadata = event.metadata as Record<string, unknown> | undefined;\n        const usage = metadata?.usage as Record<string, unknown> | undefined;\n        accumulator.setUsage({\n          inputTokens: usage?.inputTokens as number | undefined,\n          outputTokens: usage?.outputTokens as number | undefined,\n          totalTokens: usage?.totalTokens as number | undefined,\n          cacheReadTokens: usage?.cacheReadInputTokens as number | undefined,\n          cacheCreationTokens: usage?.cacheWriteInputTokens as number | undefined,\n        });\n\n        yield rawEvent;\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 ?? modelId);\n      let costUsd: number | null = null;\n      if (state.inputTokens != null && state.outputTokens != null) {\n        try {\n          costUsd = calculateCost({\n            modelName,\n            inputTokens: state.inputTokens,\n            outputTokens: state.outputTokens,\n            cacheReadInputTokens: state.cacheReadTokens ?? 0,\n            cacheCreationInputTokens: state.cacheCreationTokens ?? 0,\n            defaultProvider: 'bedrock',\n          });\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      this._track({\n        ...contextFields(ctx),\n        modelName,\n        provider: 'bedrock',\n        responseContent: state.content,\n        latencyMs: accumulator.elapsedMs,\n        inputTokens: state.inputTokens,\n        outputTokens: state.outputTokens,\n        totalTokens: state.totalTokens,\n        totalCostUsd: costUsd,\n        finishReason: state.finishReason,\n        toolCalls: state.toolCalls.length > 0 ? state.toolCalls : undefined,\n        systemPrompt: extractSystemPromptFromParams(params),\n        temperature: (\n          params.inferenceConfig as Record<string, unknown> | undefined\n        )?.temperature as number | undefined,\n        topP: (params.inferenceConfig as Record<string, unknown> | undefined)\n          ?.topP as number | undefined,\n        maxOutputTokens: (\n          params.inferenceConfig as Record<string, unknown> | undefined\n        )?.maxTokens as number | undefined,\n        providerTtfbMs: state.ttfbMs,\n        isStreaming: true,\n        isError: state.isError,\n        errorMessage: state.errorMessage,\n      });\n    }\n  }\n}\n\nexport function extractBedrockResponse(response: unknown): {\n  text: string;\n  inputTokens?: number;\n  outputTokens?: number;\n  totalTokens?: number;\n  cacheReadTokens?: number;\n  cacheWriteTokens?: number;\n  stopReason?: string;\n  toolCalls: Array<Record<string, unknown>>;\n  systemPrompt?: string;\n  temperature?: number;\n  topP?: number;\n  maxOutputTokens?: number;\n} {\n  const resp = response as BedrockConverseResponse;\n  const content = resp.output?.message?.content;\n  const textBlock = content?.find(\n    (b) => (b as Record<string, unknown>).text != null,\n  ) as { text?: string } | undefined;\n  const toolCalls = content\n    ?.filter((b) => (b as Record<string, unknown>).toolUse != null)\n    .map(\n      (b) => (b as Record<string, unknown>).toolUse as Record<string, unknown>,\n    );\n  const usage = resp.usage;\n  const respAny = resp as Record<string, unknown>;\n  const metrics = respAny.metrics as Record<string, unknown> | undefined;\n  const additionalModelResponseFields =\n    respAny.additionalModelResponseFields as\n      | Record<string, unknown>\n      | undefined;\n  const inferenceConfig = respAny.inferenceConfig as\n    | Record<string, unknown>\n    | undefined;\n  const system = respAny.system as Array<Record<string, unknown>> | undefined;\n  const systemPrompt =\n    Array.isArray(system) && system.length > 0\n      ? system\n          .map((s) => String((s as Record<string, unknown>).text ?? ''))\n          .join('')\n      : undefined;\n\n  return {\n    text: String(textBlock?.text ?? ''),\n    inputTokens: usage?.inputTokens,\n    outputTokens: usage?.outputTokens,\n    totalTokens: usage?.totalTokens,\n    cacheReadTokens: (usage as Record<string, unknown> | undefined)\n      ?.cacheReadInputTokens as number | undefined,\n    cacheWriteTokens: (usage as Record<string, unknown> | undefined)\n      ?.cacheWriteInputTokens as number | undefined,\n    stopReason: resp.stopReason,\n    toolCalls: toolCalls ?? [],\n    systemPrompt,\n    temperature:\n      (inferenceConfig?.temperature as number | undefined) ??\n      (additionalModelResponseFields?.temperature as number | undefined),\n    topP:\n      (inferenceConfig?.topP as number | undefined) ??\n      (additionalModelResponseFields?.topP as number | undefined),\n    maxOutputTokens:\n      (inferenceConfig?.maxTokens as number | undefined) ??\n      (metrics?.maxOutputTokens as number | undefined),\n  };\n}\n\nfunction extractSystemPromptFromParams(\n  params: Record<string, unknown>,\n): string | undefined {\n  const system = params.system as Array<Record<string, unknown>> | undefined;\n  if (!Array.isArray(system) || system.length === 0) return undefined;\n  return system.map((s) => String(s.text ?? '')).join('');\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"],"mappings":";;;;;;AAaA,MAAM,YAAY,WAAW,kCAAkC;AAC/D,MAAa,oBAAoB,aAAa;AAC9C,MAAMA,iBAAiD;AAYvD,IAAa,UAAb,cAA6B,eAAe;CAC1C,AAAQ;CACR,AAAQ;CAER,YAAY,SAAyB;AACnC,QAAM;GACJ,WAAW,QAAQ;GACnB,eAAe,QAAQ;GACvB,cAAc;GACf,CAAC;AACF,OAAK,UAAU,QAAQ;AACvB,OAAK,cACF,QAAQ,iBACT;;CAGJ,MAAM,SAAS,QAAmD;EAChE,MAAM,SAAS,KAAK;EAGpB,MAAM,UAAU,OAAO,OAAO,WAAW,UAAU;AAEnD,MAAI,KAAK,eAAe,KACtB,OAAM,IAAI,MACR,wKACD;EAGH,MAAM,kBAAkB,KAAK,YAAY;EAIzC,MAAM,UAAU,IAAI,gBAAgB,OAAO;EAC3C,MAAM,YAAY,YAAY,KAAK;AAEnC,MAAI;GACF,MAAM,WAAW,MAAM,OAAO,KAAK,QAAQ;GAC3C,MAAM,YAAY,YAAY,KAAK,GAAG;GAEtC,MAAM,YAAY,uBAAuB,SAAS;GAClD,IAAIC,UAAyB;AAC7B,OAAI,UAAU,eAAe,QAAQ,UAAU,gBAAgB,KAC7D,KAAI;AACF,cAAU,cAAc;KACtB,WAAW;KACX,aAAa,UAAU;KACvB,cAAc,UAAU;KACxB,sBAAsB,UAAU,mBAAmB;KACnD,0BAA0B,UAAU,oBAAoB;KACxD,iBAAiB;KAClB,CAAC;WACI;GAKV,MAAM,MAAM,qBAAqB;AACjC,QAAK,OAAO;IACV,GAAG,cAAc,IAAI;IACrB,WAAW;IACX,UAAU;IACV,iBAAiB,UAAU;IAC3B;IACA,aAAa,UAAU;IACvB,cAAc,UAAU;IACxB,aAAa,UAAU;IACvB,cAAc;IACd,cAAc,UAAU;IACxB,WACE,UAAU,UAAU,SAAS,IAAI,UAAU,YAAY;IACzD,cAAc,UAAU;IACxB,aAAa,UAAU;IACvB,MAAM,UAAU;IAChB,iBAAiB,UAAU;IAC3B,aAAa;IACd,CAAC;AAEF,UAAO;WACA,OAAO;GACd,MAAM,YAAY,YAAY,KAAK,GAAG;GACtC,MAAM,MAAM,qBAAqB;AAEjC,QAAK,OAAO;IACV,GAAG,cAAc,IAAI;IACrB,WAAW;IACX,UAAU;IACV,iBAAiB;IACjB;IACA,SAAS;IACT,cAAc,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IACrE,CAAC;AAEF,SAAM;;;CAIV,MAAM,eAAe,QAAmD;EACtE,MAAM,UAAU,OAAO,OAAO,WAAW,UAAU;EACnD,MAAM,MAAM,qBAAqB;EACjC,MAAM,YAAY,YAAY,KAAK;AACnC,MAAI;GACF,MAAM,SAAS,KAAK;AAGpB,OAAI,KAAK,eAAe,KACtB,OAAM,IAAI,MACR,wKACD;GAGH,MAAM,wBAAwB,KAAK,YAAY;AAG/C,OAAI,yBAAyB,KAC3B,OAAM,IAAI,MAAM,oDAAoD;GAGtE,MAAM,UAAU,IAAI,sBAAsB,OAAO;GACjD,MAAM,WAAY,MAAM,OAAO,KAAK,QAAQ;GAC5C,MAAM,SAAS,SAAS;AACxB,OAAI,CAAC,iBAAiB,OAAO,CAC3B,OAAM,IAAI,MAAM,+CAA+C;AAGjE,UAAO;IACL,GAAG;IACH,QAAQ,KAAK,oBAAoB,SAAS,QAAQ,QAAQ,IAAI;IAC/D;WACM,OAAO;AACd,QAAK,OAAO;IACV,GAAG,cAAc,IAAI;IACrB,WAAW;IACX,UAAU;IACV,iBAAiB;IACjB,WAAW,YAAY,KAAK,GAAG;IAC/B,SAAS;IACT,cAAc,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IACpE,aAAa;IACd,CAAC;AACF,SAAM;;;CAIV,IAAI,SAAkB;AACpB,SAAO,KAAK;;CAGd,OAAe,oBACb,SACA,QACA,QACA,KACyB;EACzB,MAAM,cAAc,IAAI,sBAAsB;AAC9C,cAAY,QAAQ;AAEpB,MAAI;AACF,cAAW,MAAM,YAAY,QAAQ;IACnC,MAAM,QAAQ;IAId,MAAM,QAHoB,MAAM,mBAGC;AAGjC,QAAI,OAAO,QAAQ,KACjB,aAAY,WAAW,OAAO,MAAM,KAAK,CAAC;IAM5C,MAAM,QAHoB,MAAM,mBAGC;AAGjC,QAAI,OAAO,WAAW,KACpB,aAAY,YAAY,MAAM,QAAmC;IAGnE,MAAM,eAAe,MAAM;AAG3B,QAAI,cAAc,SAAS,KACzB,aAAY,QAAQ,OAAO,aAAa,MAAM;IAGhD,MAAM,cAAc,MAAM;AAG1B,QAAI,aAAa,cAAc,KAC7B,aAAY,eAAe,OAAO,YAAY,WAAW;IAI3D,MAAM,QADW,MAAM,UACC;AACxB,gBAAY,SAAS;KACnB,aAAa,OAAO;KACpB,cAAc,OAAO;KACrB,aAAa,OAAO;KACpB,iBAAiB,OAAO;KACxB,qBAAqB,OAAO;KAC7B,CAAC;AAEF,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,QAAQ;GACtD,IAAIA,UAAyB;AAC7B,OAAI,MAAM,eAAe,QAAQ,MAAM,gBAAgB,KACrD,KAAI;AACF,cAAU,cAAc;KACtB;KACA,aAAa,MAAM;KACnB,cAAc,MAAM;KACpB,sBAAsB,MAAM,mBAAmB;KAC/C,0BAA0B,MAAM,uBAAuB;KACvD,iBAAiB;KAClB,CAAC;WACI;AAKV,QAAK,OAAO;IACV,GAAG,cAAc,IAAI;IACrB;IACA,UAAU;IACV,iBAAiB,MAAM;IACvB,WAAW,YAAY;IACvB,aAAa,MAAM;IACnB,cAAc,MAAM;IACpB,aAAa,MAAM;IACnB,cAAc;IACd,cAAc,MAAM;IACpB,WAAW,MAAM,UAAU,SAAS,IAAI,MAAM,YAAY;IAC1D,cAAc,8BAA8B,OAAO;IACnD,aACE,OAAO,iBACN;IACH,MAAO,OAAO,iBACV;IACJ,iBACE,OAAO,iBACN;IACH,gBAAgB,MAAM;IACtB,aAAa;IACb,SAAS,MAAM;IACf,cAAc,MAAM;IACrB,CAAC;;;;AAKR,SAAgB,uBAAuB,UAarC;CACA,MAAM,OAAO;CACb,MAAM,UAAU,KAAK,QAAQ,SAAS;CACtC,MAAM,YAAY,SAAS,MACxB,MAAO,EAA8B,QAAQ,KAC/C;CACD,MAAM,YAAY,SACd,QAAQ,MAAO,EAA8B,WAAW,KAAK,CAC9D,KACE,MAAO,EAA8B,QACvC;CACH,MAAM,QAAQ,KAAK;CACnB,MAAM,UAAU;CAChB,MAAM,UAAU,QAAQ;CACxB,MAAM,gCACJ,QAAQ;CAGV,MAAM,kBAAkB,QAAQ;CAGhC,MAAM,SAAS,QAAQ;CACvB,MAAM,eACJ,MAAM,QAAQ,OAAO,IAAI,OAAO,SAAS,IACrC,OACG,KAAK,MAAM,OAAQ,EAA8B,QAAQ,GAAG,CAAC,CAC7D,KAAK,GAAG,GACX;AAEN,QAAO;EACL,MAAM,OAAO,WAAW,QAAQ,GAAG;EACnC,aAAa,OAAO;EACpB,cAAc,OAAO;EACrB,aAAa,OAAO;EACpB,iBAAkB,OACd;EACJ,kBAAmB,OACf;EACJ,YAAY,KAAK;EACjB,WAAW,aAAa,EAAE;EAC1B;EACA,aACG,iBAAiB,eACjB,+BAA+B;EAClC,MACG,iBAAiB,QACjB,+BAA+B;EAClC,iBACG,iBAAiB,aACjB,SAAS;EACb;;AAGH,SAAS,8BACP,QACoB;CACpB,MAAM,SAAS,OAAO;AACtB,KAAI,CAAC,MAAM,QAAQ,OAAO,IAAI,OAAO,WAAW,EAAG,QAAO;AAC1D,QAAO,OAAO,KAAK,MAAM,OAAO,EAAE,QAAQ,GAAG,CAAC,CAAC,KAAK,GAAG;;AAGzD,SAAS,iBAAiB,OAAiD;AACzE,QACE,SAAS,QACT,OAAQ,MAAkC,OAAO,mBAC/C"}
+{"version":3,"file":"bedrock.js","names":["_BedrockModule: Record<string, unknown> | null","costUsd: number | null"],"sources":["../../src/providers/bedrock.ts"],"sourcesContent":["/**\n * AWS Bedrock provider wrapper with automatic tracking.\n *\n * Wraps the AWS SDK BedrockRuntimeClient's converse command.\n */\n\nimport type { PrivacyConfig } from '../core/privacy.js';\nimport type { AmplitudeOrAI, BedrockConverseResponse } from '../types.js';\nimport { calculateCost } from '../utils/costs.js';\nimport { tryRequire } from '../utils/resolve-module.js';\nimport { StreamingAccumulator } from '../utils/streaming.js';\nimport { applySessionContext, BaseAIProvider, contextFields } from './base.js';\n\nconst _resolved = tryRequire('@aws-sdk/client-bedrock-runtime');\nexport const BEDROCK_AVAILABLE = _resolved != null;\nconst _BedrockModule: Record<string, unknown> | null = _resolved;\n\nexport { _BedrockModule };\n\nexport interface BedrockOptions {\n  amplitude: AmplitudeOrAI;\n  client: unknown;\n  privacyConfig?: PrivacyConfig | null;\n  /** Pass the `@aws-sdk/client-bedrock-runtime` module directly to bypass `tryRequire` (required in bundler environments). */\n  bedrockModule?: unknown;\n}\n\nexport class Bedrock extends BaseAIProvider {\n  private _client: unknown;\n  private _bedrockMod: Record<string, unknown> | null;\n\n  constructor(options: BedrockOptions) {\n    super({\n      amplitude: options.amplitude,\n      privacyConfig: options.privacyConfig,\n      providerName: 'bedrock',\n    });\n    this._client = options.client;\n    this._bedrockMod =\n      (options.bedrockModule as Record<string, unknown> | null) ??\n      _BedrockModule;\n  }\n\n  async converse(params: Record<string, unknown>): Promise<unknown> {\n    const client = this._client as {\n      send: (command: unknown) => Promise<unknown>;\n    };\n    const modelId = String(params.modelId ?? 'unknown');\n\n    if (this._bedrockMod == null) {\n      throw new Error(\n        '@aws-sdk/client-bedrock-runtime is required. Install it with: npm install @aws-sdk/client-bedrock-runtime — or pass the module directly via the bedrockModule option.',\n      );\n    }\n\n    const ConverseCommand = this._bedrockMod.ConverseCommand as new (\n      opts: Record<string, unknown>,\n    ) => unknown;\n\n    const command = new ConverseCommand(params);\n    const startTime = performance.now();\n\n    try {\n      const response = await client.send(command);\n      const latencyMs = performance.now() - startTime;\n\n      const extracted = extractBedrockResponse(response);\n      let costUsd: number | null = null;\n      if (extracted.inputTokens != null && extracted.outputTokens != null) {\n        try {\n          costUsd = calculateCost({\n            modelName: modelId,\n            inputTokens: extracted.inputTokens,\n            outputTokens: extracted.outputTokens,\n            cacheReadInputTokens: extracted.cacheReadTokens ?? 0,\n            cacheCreationInputTokens: extracted.cacheWriteTokens ?? 0,\n            defaultProvider: 'bedrock',\n          });\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      const ctx = applySessionContext();\n      this._track({\n        ...contextFields(ctx),\n        modelName: modelId,\n        provider: 'bedrock',\n        responseContent: extracted.text,\n        latencyMs,\n        inputTokens: extracted.inputTokens,\n        outputTokens: extracted.outputTokens,\n        totalTokens: extracted.totalTokens,\n        totalCostUsd: costUsd,\n        finishReason: extracted.stopReason,\n        toolCalls:\n          extracted.toolCalls.length > 0 ? extracted.toolCalls : undefined,\n        toolDefinitions: extractBedrockToolDefinitions(params),\n        systemPrompt: extracted.systemPrompt,\n        temperature: extracted.temperature,\n        topP: extracted.topP,\n        maxOutputTokens: extracted.maxOutputTokens,\n        isStreaming: false,\n      });\n\n      return response;\n    } catch (error) {\n      const latencyMs = performance.now() - startTime;\n      const ctx = applySessionContext();\n\n      this._track({\n        ...contextFields(ctx),\n        modelName: modelId,\n        provider: 'bedrock',\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  async converseStream(params: Record<string, unknown>): Promise<unknown> {\n    const modelId = String(params.modelId ?? 'unknown');\n    const ctx = applySessionContext();\n    const startTime = performance.now();\n    try {\n      const client = this._client as {\n        send: (command: unknown) => Promise<unknown>;\n      };\n      if (this._bedrockMod == null) {\n        throw new Error(\n          '@aws-sdk/client-bedrock-runtime is required. Install it with: npm install @aws-sdk/client-bedrock-runtime — or pass the module directly via the bedrockModule option.',\n        );\n      }\n\n      const ConverseStreamCommand = this._bedrockMod.ConverseStreamCommand as\n        | (new (opts: Record<string, unknown>) => unknown)\n        | undefined;\n      if (ConverseStreamCommand == null) {\n        throw new Error('Bedrock SDK does not expose ConverseStreamCommand');\n      }\n\n      const command = new ConverseStreamCommand(params);\n      const response = (await client.send(command)) as Record<string, unknown>;\n      const stream = response.stream as AsyncIterable<unknown> | undefined;\n      if (!_isAsyncIterable(stream)) {\n        throw new Error('Bedrock stream response is not AsyncIterable');\n      }\n\n      return {\n        ...response,\n        stream: this._wrapConverseStream(modelId, params, stream, ctx),\n      };\n    } catch (error) {\n      this._track({\n        ...contextFields(ctx),\n        modelName: modelId,\n        provider: 'bedrock',\n        responseContent: '',\n        latencyMs: performance.now() - startTime,\n        isError: true,\n        errorMessage: error instanceof Error ? error.message : String(error),\n        isStreaming: true,\n      });\n      throw error;\n    }\n  }\n\n  get client(): unknown {\n    return this._client;\n  }\n\n  private async *_wrapConverseStream(\n    modelId: string,\n    params: Record<string, unknown>,\n    stream: AsyncIterable<unknown>,\n    ctx: ReturnType<typeof applySessionContext>,\n  ): AsyncGenerator<unknown> {\n    const accumulator = new StreamingAccumulator();\n    accumulator.model = modelId;\n\n    try {\n      for await (const rawEvent of stream) {\n        const event = rawEvent as Record<string, unknown>;\n        const contentBlockDelta = event.contentBlockDelta as\n          | Record<string, unknown>\n          | undefined;\n        const delta = contentBlockDelta?.delta as\n          | Record<string, unknown>\n          | undefined;\n        if (delta?.text != null) {\n          accumulator.addContent(String(delta.text));\n        }\n\n        const contentBlockStart = event.contentBlockStart as\n          | Record<string, unknown>\n          | undefined;\n        const start = contentBlockStart?.start as\n          | Record<string, unknown>\n          | undefined;\n        if (start?.toolUse != null) {\n          accumulator.addToolCall(start.toolUse as Record<string, unknown>);\n        }\n\n        const messageStart = event.messageStart as\n          | Record<string, unknown>\n          | undefined;\n        if (messageStart?.model != null) {\n          accumulator.model = String(messageStart.model);\n        }\n\n        const messageStop = event.messageStop as\n          | Record<string, unknown>\n          | undefined;\n        if (messageStop?.stopReason != null) {\n          accumulator.finishReason = String(messageStop.stopReason);\n        }\n\n        const metadata = event.metadata as Record<string, unknown> | undefined;\n        const usage = metadata?.usage as Record<string, unknown> | undefined;\n        accumulator.setUsage({\n          inputTokens: usage?.inputTokens as number | undefined,\n          outputTokens: usage?.outputTokens as number | undefined,\n          totalTokens: usage?.totalTokens as number | undefined,\n          cacheReadTokens: usage?.cacheReadInputTokens as number | undefined,\n          cacheCreationTokens: usage?.cacheWriteInputTokens as number | undefined,\n        });\n\n        yield rawEvent;\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 ?? modelId);\n      let costUsd: number | null = null;\n      if (state.inputTokens != null && state.outputTokens != null) {\n        try {\n          costUsd = calculateCost({\n            modelName,\n            inputTokens: state.inputTokens,\n            outputTokens: state.outputTokens,\n            cacheReadInputTokens: state.cacheReadTokens ?? 0,\n            cacheCreationInputTokens: state.cacheCreationTokens ?? 0,\n            defaultProvider: 'bedrock',\n          });\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      this._track({\n        ...contextFields(ctx),\n        modelName,\n        provider: 'bedrock',\n        responseContent: state.content,\n        latencyMs: accumulator.elapsedMs,\n        inputTokens: state.inputTokens,\n        outputTokens: state.outputTokens,\n        totalTokens: state.totalTokens,\n        totalCostUsd: costUsd,\n        finishReason: state.finishReason,\n        toolCalls: state.toolCalls.length > 0 ? state.toolCalls : undefined,\n        toolDefinitions: extractBedrockToolDefinitions(params),\n        systemPrompt: extractSystemPromptFromParams(params),\n        temperature: (\n          params.inferenceConfig as Record<string, unknown> | undefined\n        )?.temperature as number | undefined,\n        topP: (params.inferenceConfig as Record<string, unknown> | undefined)\n          ?.topP as number | undefined,\n        maxOutputTokens: (\n          params.inferenceConfig as Record<string, unknown> | undefined\n        )?.maxTokens as number | undefined,\n        providerTtfbMs: state.ttfbMs,\n        isStreaming: true,\n        isError: state.isError,\n        errorMessage: state.errorMessage,\n      });\n    }\n  }\n}\n\nexport function extractBedrockResponse(response: unknown): {\n  text: string;\n  inputTokens?: number;\n  outputTokens?: number;\n  totalTokens?: number;\n  cacheReadTokens?: number;\n  cacheWriteTokens?: number;\n  stopReason?: string;\n  toolCalls: Array<Record<string, unknown>>;\n  systemPrompt?: string;\n  temperature?: number;\n  topP?: number;\n  maxOutputTokens?: number;\n} {\n  const resp = response as BedrockConverseResponse;\n  const content = resp.output?.message?.content;\n  const textBlock = content?.find(\n    (b) => (b as Record<string, unknown>).text != null,\n  ) as { text?: string } | undefined;\n  const toolCalls = content\n    ?.filter((b) => (b as Record<string, unknown>).toolUse != null)\n    .map(\n      (b) => (b as Record<string, unknown>).toolUse as Record<string, unknown>,\n    );\n  const usage = resp.usage;\n  const respAny = resp as Record<string, unknown>;\n  const metrics = respAny.metrics as Record<string, unknown> | undefined;\n  const additionalModelResponseFields =\n    respAny.additionalModelResponseFields as\n      | Record<string, unknown>\n      | undefined;\n  const inferenceConfig = respAny.inferenceConfig as\n    | Record<string, unknown>\n    | undefined;\n  const system = respAny.system as Array<Record<string, unknown>> | undefined;\n  const systemPrompt =\n    Array.isArray(system) && system.length > 0\n      ? system\n          .map((s) => String((s as Record<string, unknown>).text ?? ''))\n          .join('')\n      : undefined;\n\n  return {\n    text: String(textBlock?.text ?? ''),\n    inputTokens: usage?.inputTokens,\n    outputTokens: usage?.outputTokens,\n    totalTokens: usage?.totalTokens,\n    cacheReadTokens: (usage as Record<string, unknown> | undefined)\n      ?.cacheReadInputTokens as number | undefined,\n    cacheWriteTokens: (usage as Record<string, unknown> | undefined)\n      ?.cacheWriteInputTokens as number | undefined,\n    stopReason: resp.stopReason,\n    toolCalls: toolCalls ?? [],\n    systemPrompt,\n    temperature:\n      (inferenceConfig?.temperature as number | undefined) ??\n      (additionalModelResponseFields?.temperature as number | undefined),\n    topP:\n      (inferenceConfig?.topP as number | undefined) ??\n      (additionalModelResponseFields?.topP as number | undefined),\n    maxOutputTokens:\n      (inferenceConfig?.maxTokens as number | undefined) ??\n      (metrics?.maxOutputTokens as number | undefined),\n  };\n}\n\nfunction extractSystemPromptFromParams(\n  params: Record<string, unknown>,\n): string | undefined {\n  const system = params.system as Array<Record<string, unknown>> | undefined;\n  if (!Array.isArray(system) || system.length === 0) return undefined;\n  return system.map((s) => String(s.text ?? '')).join('');\n}\n\nfunction extractBedrockToolDefinitions(\n  params: Record<string, unknown>,\n): Array<Record<string, unknown>> | undefined {\n  const toolConfig = params.toolConfig as Record<string, unknown> | undefined;\n  if (toolConfig == null) return undefined;\n  const tools = toolConfig.tools;\n  return Array.isArray(tools) && tools.length > 0\n    ? (tools as Array<Record<string, unknown>>)\n    : undefined;\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"],"mappings":";;;;;;AAaA,MAAM,YAAY,WAAW,kCAAkC;AAC/D,MAAa,oBAAoB,aAAa;AAC9C,MAAMA,iBAAiD;AAYvD,IAAa,UAAb,cAA6B,eAAe;CAC1C,AAAQ;CACR,AAAQ;CAER,YAAY,SAAyB;AACnC,QAAM;GACJ,WAAW,QAAQ;GACnB,eAAe,QAAQ;GACvB,cAAc;GACf,CAAC;AACF,OAAK,UAAU,QAAQ;AACvB,OAAK,cACF,QAAQ,iBACT;;CAGJ,MAAM,SAAS,QAAmD;EAChE,MAAM,SAAS,KAAK;EAGpB,MAAM,UAAU,OAAO,OAAO,WAAW,UAAU;AAEnD,MAAI,KAAK,eAAe,KACtB,OAAM,IAAI,MACR,wKACD;EAGH,MAAM,kBAAkB,KAAK,YAAY;EAIzC,MAAM,UAAU,IAAI,gBAAgB,OAAO;EAC3C,MAAM,YAAY,YAAY,KAAK;AAEnC,MAAI;GACF,MAAM,WAAW,MAAM,OAAO,KAAK,QAAQ;GAC3C,MAAM,YAAY,YAAY,KAAK,GAAG;GAEtC,MAAM,YAAY,uBAAuB,SAAS;GAClD,IAAIC,UAAyB;AAC7B,OAAI,UAAU,eAAe,QAAQ,UAAU,gBAAgB,KAC7D,KAAI;AACF,cAAU,cAAc;KACtB,WAAW;KACX,aAAa,UAAU;KACvB,cAAc,UAAU;KACxB,sBAAsB,UAAU,mBAAmB;KACnD,0BAA0B,UAAU,oBAAoB;KACxD,iBAAiB;KAClB,CAAC;WACI;GAKV,MAAM,MAAM,qBAAqB;AACjC,QAAK,OAAO;IACV,GAAG,cAAc,IAAI;IACrB,WAAW;IACX,UAAU;IACV,iBAAiB,UAAU;IAC3B;IACA,aAAa,UAAU;IACvB,cAAc,UAAU;IACxB,aAAa,UAAU;IACvB,cAAc;IACd,cAAc,UAAU;IACxB,WACE,UAAU,UAAU,SAAS,IAAI,UAAU,YAAY;IACzD,iBAAiB,8BAA8B,OAAO;IACtD,cAAc,UAAU;IACxB,aAAa,UAAU;IACvB,MAAM,UAAU;IAChB,iBAAiB,UAAU;IAC3B,aAAa;IACd,CAAC;AAEF,UAAO;WACA,OAAO;GACd,MAAM,YAAY,YAAY,KAAK,GAAG;GACtC,MAAM,MAAM,qBAAqB;AAEjC,QAAK,OAAO;IACV,GAAG,cAAc,IAAI;IACrB,WAAW;IACX,UAAU;IACV,iBAAiB;IACjB;IACA,SAAS;IACT,cAAc,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IACrE,CAAC;AAEF,SAAM;;;CAIV,MAAM,eAAe,QAAmD;EACtE,MAAM,UAAU,OAAO,OAAO,WAAW,UAAU;EACnD,MAAM,MAAM,qBAAqB;EACjC,MAAM,YAAY,YAAY,KAAK;AACnC,MAAI;GACF,MAAM,SAAS,KAAK;AAGpB,OAAI,KAAK,eAAe,KACtB,OAAM,IAAI,MACR,wKACD;GAGH,MAAM,wBAAwB,KAAK,YAAY;AAG/C,OAAI,yBAAyB,KAC3B,OAAM,IAAI,MAAM,oDAAoD;GAGtE,MAAM,UAAU,IAAI,sBAAsB,OAAO;GACjD,MAAM,WAAY,MAAM,OAAO,KAAK,QAAQ;GAC5C,MAAM,SAAS,SAAS;AACxB,OAAI,CAAC,iBAAiB,OAAO,CAC3B,OAAM,IAAI,MAAM,+CAA+C;AAGjE,UAAO;IACL,GAAG;IACH,QAAQ,KAAK,oBAAoB,SAAS,QAAQ,QAAQ,IAAI;IAC/D;WACM,OAAO;AACd,QAAK,OAAO;IACV,GAAG,cAAc,IAAI;IACrB,WAAW;IACX,UAAU;IACV,iBAAiB;IACjB,WAAW,YAAY,KAAK,GAAG;IAC/B,SAAS;IACT,cAAc,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IACpE,aAAa;IACd,CAAC;AACF,SAAM;;;CAIV,IAAI,SAAkB;AACpB,SAAO,KAAK;;CAGd,OAAe,oBACb,SACA,QACA,QACA,KACyB;EACzB,MAAM,cAAc,IAAI,sBAAsB;AAC9C,cAAY,QAAQ;AAEpB,MAAI;AACF,cAAW,MAAM,YAAY,QAAQ;IACnC,MAAM,QAAQ;IAId,MAAM,QAHoB,MAAM,mBAGC;AAGjC,QAAI,OAAO,QAAQ,KACjB,aAAY,WAAW,OAAO,MAAM,KAAK,CAAC;IAM5C,MAAM,QAHoB,MAAM,mBAGC;AAGjC,QAAI,OAAO,WAAW,KACpB,aAAY,YAAY,MAAM,QAAmC;IAGnE,MAAM,eAAe,MAAM;AAG3B,QAAI,cAAc,SAAS,KACzB,aAAY,QAAQ,OAAO,aAAa,MAAM;IAGhD,MAAM,cAAc,MAAM;AAG1B,QAAI,aAAa,cAAc,KAC7B,aAAY,eAAe,OAAO,YAAY,WAAW;IAI3D,MAAM,QADW,MAAM,UACC;AACxB,gBAAY,SAAS;KACnB,aAAa,OAAO;KACpB,cAAc,OAAO;KACrB,aAAa,OAAO;KACpB,iBAAiB,OAAO;KACxB,qBAAqB,OAAO;KAC7B,CAAC;AAEF,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,QAAQ;GACtD,IAAIA,UAAyB;AAC7B,OAAI,MAAM,eAAe,QAAQ,MAAM,gBAAgB,KACrD,KAAI;AACF,cAAU,cAAc;KACtB;KACA,aAAa,MAAM;KACnB,cAAc,MAAM;KACpB,sBAAsB,MAAM,mBAAmB;KAC/C,0BAA0B,MAAM,uBAAuB;KACvD,iBAAiB;KAClB,CAAC;WACI;AAKV,QAAK,OAAO;IACV,GAAG,cAAc,IAAI;IACrB;IACA,UAAU;IACV,iBAAiB,MAAM;IACvB,WAAW,YAAY;IACvB,aAAa,MAAM;IACnB,cAAc,MAAM;IACpB,aAAa,MAAM;IACnB,cAAc;IACd,cAAc,MAAM;IACpB,WAAW,MAAM,UAAU,SAAS,IAAI,MAAM,YAAY;IAC1D,iBAAiB,8BAA8B,OAAO;IACtD,cAAc,8BAA8B,OAAO;IACnD,aACE,OAAO,iBACN;IACH,MAAO,OAAO,iBACV;IACJ,iBACE,OAAO,iBACN;IACH,gBAAgB,MAAM;IACtB,aAAa;IACb,SAAS,MAAM;IACf,cAAc,MAAM;IACrB,CAAC;;;;AAKR,SAAgB,uBAAuB,UAarC;CACA,MAAM,OAAO;CACb,MAAM,UAAU,KAAK,QAAQ,SAAS;CACtC,MAAM,YAAY,SAAS,MACxB,MAAO,EAA8B,QAAQ,KAC/C;CACD,MAAM,YAAY,SACd,QAAQ,MAAO,EAA8B,WAAW,KAAK,CAC9D,KACE,MAAO,EAA8B,QACvC;CACH,MAAM,QAAQ,KAAK;CACnB,MAAM,UAAU;CAChB,MAAM,UAAU,QAAQ;CACxB,MAAM,gCACJ,QAAQ;CAGV,MAAM,kBAAkB,QAAQ;CAGhC,MAAM,SAAS,QAAQ;CACvB,MAAM,eACJ,MAAM,QAAQ,OAAO,IAAI,OAAO,SAAS,IACrC,OACG,KAAK,MAAM,OAAQ,EAA8B,QAAQ,GAAG,CAAC,CAC7D,KAAK,GAAG,GACX;AAEN,QAAO;EACL,MAAM,OAAO,WAAW,QAAQ,GAAG;EACnC,aAAa,OAAO;EACpB,cAAc,OAAO;EACrB,aAAa,OAAO;EACpB,iBAAkB,OACd;EACJ,kBAAmB,OACf;EACJ,YAAY,KAAK;EACjB,WAAW,aAAa,EAAE;EAC1B;EACA,aACG,iBAAiB,eACjB,+BAA+B;EAClC,MACG,iBAAiB,QACjB,+BAA+B;EAClC,iBACG,iBAAiB,aACjB,SAAS;EACb;;AAGH,SAAS,8BACP,QACoB;CACpB,MAAM,SAAS,OAAO;AACtB,KAAI,CAAC,MAAM,QAAQ,OAAO,IAAI,OAAO,WAAW,EAAG,QAAO;AAC1D,QAAO,OAAO,KAAK,MAAM,OAAO,EAAE,QAAQ,GAAG,CAAC,CAAC,KAAK,GAAG;;AAGzD,SAAS,8BACP,QAC4C;CAC5C,MAAM,aAAa,OAAO;AAC1B,KAAI,cAAc,KAAM,QAAO;CAC/B,MAAM,QAAQ,WAAW;AACzB,QAAO,MAAM,QAAQ,MAAM,IAAI,MAAM,SAAS,IACzC,QACD;;AAGN,SAAS,iBAAiB,OAAiD;AACzE,QACE,SAAS,QACT,OAAQ,MAAkC,OAAO,mBAC/C"}

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

@@ -1 +1 @@
-{"version":3,"file":"gemini.d.ts","names":[],"sources":["../../src/providers/gemini.ts"],"sourcesContent":[],"mappings":";;;;;;AAaM,cADO,gBACQ,EAAM,OAAA;AAI3B,cAJM,aAIwB,EAJT,MAKR,CAAA,MAAA,EAAA,OAEK,CAAA,GAAA,IAAA;AAQK,UAXN,aAAA,CAWM;EAuBX,SAAA,EAjCC,aAiCD;EACP,MAAA,CAAA,EAAA,MAAA;EA0EO,aAAA,CAAA,EA1GM,aA0GN,GAAA,IAAA;EACP;EAtGuB,YAAA,CAAA,EAAA,OAAA;;AA0PZ,cA1PH,MAAA,SAAe,cAAA,CAgQV;;uBA7PK;yCAuBX,0BACP;+CA0EO,0BACP;;;;iBAoJW,qBAAA;;;;;;kBAME,MAAM"}
+{"version":3,"file":"gemini.d.ts","names":[],"sources":["../../src/providers/gemini.ts"],"sourcesContent":[],"mappings":";;;;;;AAaM,cADO,gBACQ,EAAM,OAAA;AAI3B,cAJM,aAIwB,EAJT,MAKR,CAAA,MAAA,EAAA,OAEK,CAAA,GAAA,IAAA;AAQK,UAXN,aAAA,CAWM;EAuBX,SAAA,EAjCC,aAiCD;EACP,MAAA,CAAA,EAAA,MAAA;EA2EO,aAAA,CAAA,EA3GM,aA2GN,GAAA,IAAA;EACP;EAvGuB,YAAA,CAAA,EAAA,OAAA;;AA4PZ,cA5PH,MAAA,SAAe,cAAA,CAkQV;;uBA/PK;yCAuBX,0BACP;+CA2EO,0BACP;;;;iBAqJW,qBAAA;;;;;;kBAME,MAAM"}

package/dist/providers/gemini.js

@@ -50,6 +50,7 @@ var Gemini = class extends BaseAIProvider {
 				totalCostUsd: costUsd,
 				finishReason: extracted.finishReason,
 				toolCalls: extracted.functionCalls?.length ? extracted.functionCalls : void 0,
+				toolDefinitions: extractGeminiToolDefinitions(params),
 				systemPrompt: extractGeminiSystemPrompt(params),
 				temperature: extractGeminiTemperature(params),
 				topP: extractGeminiTopP(params),
@@ -155,6 +156,7 @@ var Gemini = class extends BaseAIProvider {
 				totalCostUsd: costUsd,
 				finishReason: state.finishReason,
 				toolCalls: state.toolCalls.length > 0 ? state.toolCalls : void 0,
+				toolDefinitions: extractGeminiToolDefinitions(params),
 				systemPrompt: extractGeminiSystemPrompt(params),
 				temperature: extractGeminiTemperature(params),
 				topP: extractGeminiTopP(params),
@@ -199,6 +201,10 @@ function extractGeminiTopP(params) {
 function extractGeminiMaxOutputTokens(params) {
 	return params.generationConfig?.maxOutputTokens;
 }
+function extractGeminiToolDefinitions(params) {
+	const tools = params.tools;
+	return Array.isArray(tools) && tools.length > 0 ? tools : void 0;
+}
 function _isAsyncIterable(value) {
 	return value != null && typeof value[Symbol.asyncIterator] === "function";
 }

package/dist/providers/gemini.js.map

@@ -1 +1 @@
-{"version":3,"file":"gemini.js","names":["_GeminiModule: Record<string, unknown> | null","costUsd: number | null"],"sources":["../../src/providers/gemini.ts"],"sourcesContent":["/**\n * Google Gemini provider wrapper with automatic tracking.\n */\n\nimport type { PrivacyConfig } from '../core/privacy.js';\nimport type { AmplitudeOrAI, GeminiResponse } from '../types.js';\nimport { calculateCost } from '../utils/costs.js';\nimport { tryRequire } from '../utils/resolve-module.js';\nimport { StreamingAccumulator } from '../utils/streaming.js';\nimport { applySessionContext, BaseAIProvider, contextFields } from './base.js';\n\nconst _resolved = tryRequire('@google/generative-ai');\nexport const GEMINI_AVAILABLE = _resolved != null;\nconst _GeminiModule: Record<string, unknown> | null = _resolved;\n\nexport { _GeminiModule };\n\nexport interface GeminiOptions {\n  amplitude: AmplitudeOrAI;\n  apiKey?: string;\n  privacyConfig?: PrivacyConfig | null;\n  /** Pass the `@google/generative-ai` module directly to bypass `tryRequire` (required in bundler environments). */\n  geminiModule?: unknown;\n}\n\nexport class Gemini extends BaseAIProvider {\n  private _client: unknown;\n\n  constructor(options: GeminiOptions) {\n    super({\n      amplitude: options.amplitude,\n      privacyConfig: options.privacyConfig,\n      providerName: 'gemini',\n    });\n\n    const mod =\n      (options.geminiModule as Record<string, unknown> | null) ?? _GeminiModule;\n    if (mod == null) {\n      throw new Error(\n        '@google/generative-ai package is required. Install it with: npm install @google/generative-ai — or pass the module directly via the geminiModule option.',\n      );\n    }\n\n    const GoogleGenAI = mod.GoogleGenerativeAI as new (\n      apiKey: string,\n    ) => unknown;\n    this._client = new GoogleGenAI(options.apiKey ?? '');\n  }\n\n  async generateContent(\n    model: string,\n    params: Record<string, unknown>,\n  ): Promise<unknown> {\n    const clientObj = this._client as Record<string, unknown>;\n    const getModel = clientObj.getGenerativeModel as (\n      opts: Record<string, unknown>,\n    ) => Record<string, unknown>;\n    const genModel = getModel.call(this._client, { model });\n    const generateFn = genModel.generateContent as (\n      ...args: unknown[]\n    ) => Promise<unknown>;\n\n    const startTime = performance.now();\n\n    try {\n      const response = await generateFn.call(genModel, params);\n      const latencyMs = performance.now() - startTime;\n\n      const extracted = extractGeminiResponse(response);\n      let costUsd: number | null = null;\n      if (extracted.inputTokens != null && extracted.outputTokens != null) {\n        try {\n          costUsd = calculateCost({\n            modelName: model,\n            inputTokens: extracted.inputTokens,\n            outputTokens: extracted.outputTokens,\n            defaultProvider: 'google',\n          });\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      const ctx = applySessionContext();\n      this._track({\n        ...contextFields(ctx),\n        modelName: model,\n        provider: 'gemini',\n        responseContent: extracted.text,\n        latencyMs,\n        inputTokens: extracted.inputTokens,\n        outputTokens: extracted.outputTokens,\n        totalTokens: extracted.totalTokens,\n        totalCostUsd: costUsd,\n        finishReason: extracted.finishReason,\n        toolCalls: extracted.functionCalls?.length\n          ? extracted.functionCalls\n          : undefined,\n        systemPrompt: extractGeminiSystemPrompt(params),\n        temperature: extractGeminiTemperature(params),\n        topP: extractGeminiTopP(params),\n        maxOutputTokens: extractGeminiMaxOutputTokens(params),\n        isStreaming: false,\n      });\n\n      return response;\n    } catch (error) {\n      const latencyMs = performance.now() - startTime;\n      const ctx = applySessionContext();\n\n      this._track({\n        ...contextFields(ctx),\n        modelName: model,\n        provider: 'gemini',\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  async generateContentStream(\n    model: string,\n    params: Record<string, unknown>,\n  ): Promise<unknown> {\n    const ctx = applySessionContext();\n    const startTime = performance.now();\n    try {\n      const clientObj = this._client as Record<string, unknown>;\n      const getModel = clientObj.getGenerativeModel as (\n        opts: Record<string, unknown>,\n      ) => Record<string, unknown>;\n      const genModel = getModel.call(this._client, { model });\n      const streamFn = genModel.generateContentStream as\n        | ((...args: unknown[]) => Promise<unknown>)\n        | undefined;\n      if (typeof streamFn !== 'function') {\n        throw new Error('Gemini SDK does not expose generateContentStream');\n      }\n\n      const response = await streamFn.call(genModel, params);\n      const streamResponse = response as Record<string, unknown>;\n      const stream = streamResponse.stream as\n        | AsyncIterable<unknown>\n        | undefined;\n      const finalResponse = streamResponse.response as\n        | Promise<unknown>\n        | undefined;\n\n      if (!_isAsyncIterable(stream)) {\n        throw new Error('Gemini stream response is not AsyncIterable');\n      }\n\n      return {\n        ...streamResponse,\n        stream: this._wrapStream(model, params, stream, finalResponse, ctx),\n      };\n    } catch (error) {\n      this._track({\n        ...contextFields(ctx),\n        modelName: model,\n        provider: 'gemini',\n        responseContent: '',\n        latencyMs: performance.now() - startTime,\n        isError: true,\n        errorMessage: error instanceof Error ? error.message : String(error),\n        isStreaming: true,\n      });\n      throw error;\n    }\n  }\n\n  get client(): unknown {\n    return this._client;\n  }\n\n  private async *_wrapStream(\n    model: string,\n    params: Record<string, unknown>,\n    stream: AsyncIterable<unknown>,\n    finalResponse: Promise<unknown> | undefined,\n    ctx: ReturnType<typeof applySessionContext>,\n  ): AsyncGenerator<unknown> {\n    const accumulator = new StreamingAccumulator();\n\n    try {\n      for await (const chunk of stream) {\n        const extracted = extractGeminiResponse(chunk);\n        if (extracted.text) accumulator.addContent(extracted.text);\n        if (Array.isArray(extracted.functionCalls)) {\n          for (const fc of extracted.functionCalls) accumulator.addToolCall(fc);\n        }\n        if (extracted.finishReason != null) {\n          accumulator.finishReason = String(extracted.finishReason);\n        }\n        accumulator.setUsage({\n          inputTokens: extracted.inputTokens,\n          outputTokens: extracted.outputTokens,\n          totalTokens: extracted.totalTokens,\n        });\n        yield chunk;\n      }\n    } catch (error) {\n      accumulator.setError(\n        error instanceof Error ? error.message : String(error),\n      );\n      throw error;\n    } finally {\n      if (finalResponse != null) {\n        try {\n          const extractedFinal = extractGeminiResponse(await finalResponse);\n          accumulator.setUsage({\n            inputTokens: extractedFinal.inputTokens,\n            outputTokens: extractedFinal.outputTokens,\n            totalTokens: extractedFinal.totalTokens,\n          });\n          if (extractedFinal.finishReason != null) {\n            accumulator.finishReason = String(extractedFinal.finishReason);\n          }\n          if (\n            Array.isArray(extractedFinal.functionCalls) &&\n            accumulator.toolCalls.length === 0\n          ) {\n            for (const fc of extractedFinal.functionCalls) {\n              accumulator.addToolCall(fc);\n            }\n          }\n        } catch {\n          // best-effort final response extraction\n        }\n      }\n\n      const state = accumulator.getState();\n      let costUsd: number | null = null;\n      if (state.inputTokens != null && state.outputTokens != null) {\n        try {\n          costUsd = calculateCost({\n            modelName: model,\n            inputTokens: state.inputTokens,\n            outputTokens: state.outputTokens,\n            defaultProvider: 'google',\n          });\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      this._track({\n        ...contextFields(ctx),\n        modelName: model,\n        provider: 'gemini',\n        responseContent: state.content,\n        latencyMs: accumulator.elapsedMs,\n        inputTokens: state.inputTokens,\n        outputTokens: state.outputTokens,\n        totalTokens: state.totalTokens,\n        totalCostUsd: costUsd,\n        finishReason: state.finishReason,\n        toolCalls: state.toolCalls.length > 0 ? state.toolCalls : undefined,\n        systemPrompt: extractGeminiSystemPrompt(params),\n        temperature: extractGeminiTemperature(params),\n        topP: extractGeminiTopP(params),\n        maxOutputTokens: extractGeminiMaxOutputTokens(params),\n        providerTtfbMs: state.ttfbMs,\n        isStreaming: true,\n        isError: state.isError,\n        errorMessage: state.errorMessage,\n      });\n    }\n  }\n}\n\nexport function extractGeminiResponse(response: unknown): {\n  text: string;\n  inputTokens?: number;\n  outputTokens?: number;\n  totalTokens?: number;\n  finishReason?: string;\n  functionCalls?: Array<Record<string, unknown>>;\n} {\n  const resp = response as GeminiResponse;\n  const respObj = resp.response ?? resp;\n  let text = '';\n  if (typeof respObj.text === 'function') {\n    try {\n      text = String(respObj.text());\n    } catch {\n      // text() throws when the response has no candidates (e.g. safety block)\n    }\n  }\n  const usage = respObj.usageMetadata;\n\n  const candidate = respObj.candidates?.[0];\n  const finishReason = candidate?.finishReason;\n\n  const parts = candidate?.content?.parts;\n  const functionCalls = parts\n    ?.filter((p) => p.functionCall != null)\n    .map((p) => p.functionCall as Record<string, unknown>);\n\n  return {\n    text,\n    inputTokens: usage?.promptTokenCount,\n    outputTokens: usage?.candidatesTokenCount,\n    totalTokens: usage?.totalTokenCount,\n    finishReason,\n    functionCalls: functionCalls?.length ? functionCalls : undefined,\n  };\n}\n\nfunction extractGeminiSystemPrompt(\n  params: Record<string, unknown>,\n): string | undefined {\n  const systemInstruction =\n    (params.systemInstruction as string | undefined) ??\n    ((params.generationConfig as Record<string, unknown> | undefined)\n      ?.systemInstruction as string | undefined);\n  return systemInstruction;\n}\n\nfunction extractGeminiTemperature(\n  params: Record<string, unknown>,\n): number | undefined {\n  return (params.generationConfig as Record<string, unknown> | undefined)\n    ?.temperature as number | undefined;\n}\n\nfunction extractGeminiTopP(\n  params: Record<string, unknown>,\n): number | undefined {\n  return (params.generationConfig as Record<string, unknown> | undefined)\n    ?.topP as number | undefined;\n}\n\nfunction extractGeminiMaxOutputTokens(\n  params: Record<string, unknown>,\n): number | undefined {\n  return (params.generationConfig as Record<string, unknown> | undefined)\n    ?.maxOutputTokens as number | undefined;\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"],"mappings":";;;;;;AAWA,MAAM,YAAY,WAAW,wBAAwB;AACrD,MAAa,mBAAmB,aAAa;AAC7C,MAAMA,gBAAgD;AAYtD,IAAa,SAAb,cAA4B,eAAe;CACzC,AAAQ;CAER,YAAY,SAAwB;AAClC,QAAM;GACJ,WAAW,QAAQ;GACnB,eAAe,QAAQ;GACvB,cAAc;GACf,CAAC;EAEF,MAAM,MACH,QAAQ,gBAAmD;AAC9D,MAAI,OAAO,KACT,OAAM,IAAI,MACR,2JACD;EAGH,MAAM,cAAc,IAAI;AAGxB,OAAK,UAAU,IAAI,YAAY,QAAQ,UAAU,GAAG;;CAGtD,MAAM,gBACJ,OACA,QACkB;EAKlB,MAAM,WAJY,KAAK,QACI,mBAGD,KAAK,KAAK,SAAS,EAAE,OAAO,CAAC;EACvD,MAAM,aAAa,SAAS;EAI5B,MAAM,YAAY,YAAY,KAAK;AAEnC,MAAI;GACF,MAAM,WAAW,MAAM,WAAW,KAAK,UAAU,OAAO;GACxD,MAAM,YAAY,YAAY,KAAK,GAAG;GAEtC,MAAM,YAAY,sBAAsB,SAAS;GACjD,IAAIC,UAAyB;AAC7B,OAAI,UAAU,eAAe,QAAQ,UAAU,gBAAgB,KAC7D,KAAI;AACF,cAAU,cAAc;KACtB,WAAW;KACX,aAAa,UAAU;KACvB,cAAc,UAAU;KACxB,iBAAiB;KAClB,CAAC;WACI;GAKV,MAAM,MAAM,qBAAqB;AACjC,QAAK,OAAO;IACV,GAAG,cAAc,IAAI;IACrB,WAAW;IACX,UAAU;IACV,iBAAiB,UAAU;IAC3B;IACA,aAAa,UAAU;IACvB,cAAc,UAAU;IACxB,aAAa,UAAU;IACvB,cAAc;IACd,cAAc,UAAU;IACxB,WAAW,UAAU,eAAe,SAChC,UAAU,gBACV;IACJ,cAAc,0BAA0B,OAAO;IAC/C,aAAa,yBAAyB,OAAO;IAC7C,MAAM,kBAAkB,OAAO;IAC/B,iBAAiB,6BAA6B,OAAO;IACrD,aAAa;IACd,CAAC;AAEF,UAAO;WACA,OAAO;GACd,MAAM,YAAY,YAAY,KAAK,GAAG;GACtC,MAAM,MAAM,qBAAqB;AAEjC,QAAK,OAAO;IACV,GAAG,cAAc,IAAI;IACrB,WAAW;IACX,UAAU;IACV,iBAAiB;IACjB;IACA,SAAS;IACT,cAAc,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IACrE,CAAC;AAEF,SAAM;;;CAIV,MAAM,sBACJ,OACA,QACkB;EAClB,MAAM,MAAM,qBAAqB;EACjC,MAAM,YAAY,YAAY,KAAK;AACnC,MAAI;GAKF,MAAM,WAJY,KAAK,QACI,mBAGD,KAAK,KAAK,SAAS,EAAE,OAAO,CAAC;GACvD,MAAM,WAAW,SAAS;AAG1B,OAAI,OAAO,aAAa,WACtB,OAAM,IAAI,MAAM,mDAAmD;GAIrE,MAAM,iBADW,MAAM,SAAS,KAAK,UAAU,OAAO;GAEtD,MAAM,SAAS,eAAe;GAG9B,MAAM,gBAAgB,eAAe;AAIrC,OAAI,CAAC,iBAAiB,OAAO,CAC3B,OAAM,IAAI,MAAM,8CAA8C;AAGhE,UAAO;IACL,GAAG;IACH,QAAQ,KAAK,YAAY,OAAO,QAAQ,QAAQ,eAAe,IAAI;IACpE;WACM,OAAO;AACd,QAAK,OAAO;IACV,GAAG,cAAc,IAAI;IACrB,WAAW;IACX,UAAU;IACV,iBAAiB;IACjB,WAAW,YAAY,KAAK,GAAG;IAC/B,SAAS;IACT,cAAc,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IACpE,aAAa;IACd,CAAC;AACF,SAAM;;;CAIV,IAAI,SAAkB;AACpB,SAAO,KAAK;;CAGd,OAAe,YACb,OACA,QACA,QACA,eACA,KACyB;EACzB,MAAM,cAAc,IAAI,sBAAsB;AAE9C,MAAI;AACF,cAAW,MAAM,SAAS,QAAQ;IAChC,MAAM,YAAY,sBAAsB,MAAM;AAC9C,QAAI,UAAU,KAAM,aAAY,WAAW,UAAU,KAAK;AAC1D,QAAI,MAAM,QAAQ,UAAU,cAAc,CACxC,MAAK,MAAM,MAAM,UAAU,cAAe,aAAY,YAAY,GAAG;AAEvE,QAAI,UAAU,gBAAgB,KAC5B,aAAY,eAAe,OAAO,UAAU,aAAa;AAE3D,gBAAY,SAAS;KACnB,aAAa,UAAU;KACvB,cAAc,UAAU;KACxB,aAAa,UAAU;KACxB,CAAC;AACF,UAAM;;WAED,OAAO;AACd,eAAY,SACV,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,CACvD;AACD,SAAM;YACE;AACR,OAAI,iBAAiB,KACnB,KAAI;IACF,MAAM,iBAAiB,sBAAsB,MAAM,cAAc;AACjE,gBAAY,SAAS;KACnB,aAAa,eAAe;KAC5B,cAAc,eAAe;KAC7B,aAAa,eAAe;KAC7B,CAAC;AACF,QAAI,eAAe,gBAAgB,KACjC,aAAY,eAAe,OAAO,eAAe,aAAa;AAEhE,QACE,MAAM,QAAQ,eAAe,cAAc,IAC3C,YAAY,UAAU,WAAW,EAEjC,MAAK,MAAM,MAAM,eAAe,cAC9B,aAAY,YAAY,GAAG;WAGzB;GAKV,MAAM,QAAQ,YAAY,UAAU;GACpC,IAAIA,UAAyB;AAC7B,OAAI,MAAM,eAAe,QAAQ,MAAM,gBAAgB,KACrD,KAAI;AACF,cAAU,cAAc;KACtB,WAAW;KACX,aAAa,MAAM;KACnB,cAAc,MAAM;KACpB,iBAAiB;KAClB,CAAC;WACI;AAKV,QAAK,OAAO;IACV,GAAG,cAAc,IAAI;IACrB,WAAW;IACX,UAAU;IACV,iBAAiB,MAAM;IACvB,WAAW,YAAY;IACvB,aAAa,MAAM;IACnB,cAAc,MAAM;IACpB,aAAa,MAAM;IACnB,cAAc;IACd,cAAc,MAAM;IACpB,WAAW,MAAM,UAAU,SAAS,IAAI,MAAM,YAAY;IAC1D,cAAc,0BAA0B,OAAO;IAC/C,aAAa,yBAAyB,OAAO;IAC7C,MAAM,kBAAkB,OAAO;IAC/B,iBAAiB,6BAA6B,OAAO;IACrD,gBAAgB,MAAM;IACtB,aAAa;IACb,SAAS,MAAM;IACf,cAAc,MAAM;IACrB,CAAC;;;;AAKR,SAAgB,sBAAsB,UAOpC;CACA,MAAM,OAAO;CACb,MAAM,UAAU,KAAK,YAAY;CACjC,IAAI,OAAO;AACX,KAAI,OAAO,QAAQ,SAAS,WAC1B,KAAI;AACF,SAAO,OAAO,QAAQ,MAAM,CAAC;SACvB;CAIV,MAAM,QAAQ,QAAQ;CAEtB,MAAM,YAAY,QAAQ,aAAa;CACvC,MAAM,eAAe,WAAW;CAGhC,MAAM,iBADQ,WAAW,SAAS,QAE9B,QAAQ,MAAM,EAAE,gBAAgB,KAAK,CACtC,KAAK,MAAM,EAAE,aAAwC;AAExD,QAAO;EACL;EACA,aAAa,OAAO;EACpB,cAAc,OAAO;EACrB,aAAa,OAAO;EACpB;EACA,eAAe,eAAe,SAAS,gBAAgB;EACxD;;AAGH,SAAS,0BACP,QACoB;AAKpB,QAHG,OAAO,qBACN,OAAO,kBACL;;AAIR,SAAS,yBACP,QACoB;AACpB,QAAQ,OAAO,kBACX;;AAGN,SAAS,kBACP,QACoB;AACpB,QAAQ,OAAO,kBACX;;AAGN,SAAS,6BACP,QACoB;AACpB,QAAQ,OAAO,kBACX;;AAGN,SAAS,iBAAiB,OAAiD;AACzE,QACE,SAAS,QACT,OAAQ,MAAkC,OAAO,mBAC/C"}
+{"version":3,"file":"gemini.js","names":["_GeminiModule: Record<string, unknown> | null","costUsd: number | null"],"sources":["../../src/providers/gemini.ts"],"sourcesContent":["/**\n * Google Gemini provider wrapper with automatic tracking.\n */\n\nimport type { PrivacyConfig } from '../core/privacy.js';\nimport type { AmplitudeOrAI, GeminiResponse } from '../types.js';\nimport { calculateCost } from '../utils/costs.js';\nimport { tryRequire } from '../utils/resolve-module.js';\nimport { StreamingAccumulator } from '../utils/streaming.js';\nimport { applySessionContext, BaseAIProvider, contextFields } from './base.js';\n\nconst _resolved = tryRequire('@google/generative-ai');\nexport const GEMINI_AVAILABLE = _resolved != null;\nconst _GeminiModule: Record<string, unknown> | null = _resolved;\n\nexport { _GeminiModule };\n\nexport interface GeminiOptions {\n  amplitude: AmplitudeOrAI;\n  apiKey?: string;\n  privacyConfig?: PrivacyConfig | null;\n  /** Pass the `@google/generative-ai` module directly to bypass `tryRequire` (required in bundler environments). */\n  geminiModule?: unknown;\n}\n\nexport class Gemini extends BaseAIProvider {\n  private _client: unknown;\n\n  constructor(options: GeminiOptions) {\n    super({\n      amplitude: options.amplitude,\n      privacyConfig: options.privacyConfig,\n      providerName: 'gemini',\n    });\n\n    const mod =\n      (options.geminiModule as Record<string, unknown> | null) ?? _GeminiModule;\n    if (mod == null) {\n      throw new Error(\n        '@google/generative-ai package is required. Install it with: npm install @google/generative-ai — or pass the module directly via the geminiModule option.',\n      );\n    }\n\n    const GoogleGenAI = mod.GoogleGenerativeAI as new (\n      apiKey: string,\n    ) => unknown;\n    this._client = new GoogleGenAI(options.apiKey ?? '');\n  }\n\n  async generateContent(\n    model: string,\n    params: Record<string, unknown>,\n  ): Promise<unknown> {\n    const clientObj = this._client as Record<string, unknown>;\n    const getModel = clientObj.getGenerativeModel as (\n      opts: Record<string, unknown>,\n    ) => Record<string, unknown>;\n    const genModel = getModel.call(this._client, { model });\n    const generateFn = genModel.generateContent as (\n      ...args: unknown[]\n    ) => Promise<unknown>;\n\n    const startTime = performance.now();\n\n    try {\n      const response = await generateFn.call(genModel, params);\n      const latencyMs = performance.now() - startTime;\n\n      const extracted = extractGeminiResponse(response);\n      let costUsd: number | null = null;\n      if (extracted.inputTokens != null && extracted.outputTokens != null) {\n        try {\n          costUsd = calculateCost({\n            modelName: model,\n            inputTokens: extracted.inputTokens,\n            outputTokens: extracted.outputTokens,\n            defaultProvider: 'google',\n          });\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      const ctx = applySessionContext();\n      this._track({\n        ...contextFields(ctx),\n        modelName: model,\n        provider: 'gemini',\n        responseContent: extracted.text,\n        latencyMs,\n        inputTokens: extracted.inputTokens,\n        outputTokens: extracted.outputTokens,\n        totalTokens: extracted.totalTokens,\n        totalCostUsd: costUsd,\n        finishReason: extracted.finishReason,\n        toolCalls: extracted.functionCalls?.length\n          ? extracted.functionCalls\n          : undefined,\n        toolDefinitions: extractGeminiToolDefinitions(params),\n        systemPrompt: extractGeminiSystemPrompt(params),\n        temperature: extractGeminiTemperature(params),\n        topP: extractGeminiTopP(params),\n        maxOutputTokens: extractGeminiMaxOutputTokens(params),\n        isStreaming: false,\n      });\n\n      return response;\n    } catch (error) {\n      const latencyMs = performance.now() - startTime;\n      const ctx = applySessionContext();\n\n      this._track({\n        ...contextFields(ctx),\n        modelName: model,\n        provider: 'gemini',\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  async generateContentStream(\n    model: string,\n    params: Record<string, unknown>,\n  ): Promise<unknown> {\n    const ctx = applySessionContext();\n    const startTime = performance.now();\n    try {\n      const clientObj = this._client as Record<string, unknown>;\n      const getModel = clientObj.getGenerativeModel as (\n        opts: Record<string, unknown>,\n      ) => Record<string, unknown>;\n      const genModel = getModel.call(this._client, { model });\n      const streamFn = genModel.generateContentStream as\n        | ((...args: unknown[]) => Promise<unknown>)\n        | undefined;\n      if (typeof streamFn !== 'function') {\n        throw new Error('Gemini SDK does not expose generateContentStream');\n      }\n\n      const response = await streamFn.call(genModel, params);\n      const streamResponse = response as Record<string, unknown>;\n      const stream = streamResponse.stream as\n        | AsyncIterable<unknown>\n        | undefined;\n      const finalResponse = streamResponse.response as\n        | Promise<unknown>\n        | undefined;\n\n      if (!_isAsyncIterable(stream)) {\n        throw new Error('Gemini stream response is not AsyncIterable');\n      }\n\n      return {\n        ...streamResponse,\n        stream: this._wrapStream(model, params, stream, finalResponse, ctx),\n      };\n    } catch (error) {\n      this._track({\n        ...contextFields(ctx),\n        modelName: model,\n        provider: 'gemini',\n        responseContent: '',\n        latencyMs: performance.now() - startTime,\n        isError: true,\n        errorMessage: error instanceof Error ? error.message : String(error),\n        isStreaming: true,\n      });\n      throw error;\n    }\n  }\n\n  get client(): unknown {\n    return this._client;\n  }\n\n  private async *_wrapStream(\n    model: string,\n    params: Record<string, unknown>,\n    stream: AsyncIterable<unknown>,\n    finalResponse: Promise<unknown> | undefined,\n    ctx: ReturnType<typeof applySessionContext>,\n  ): AsyncGenerator<unknown> {\n    const accumulator = new StreamingAccumulator();\n\n    try {\n      for await (const chunk of stream) {\n        const extracted = extractGeminiResponse(chunk);\n        if (extracted.text) accumulator.addContent(extracted.text);\n        if (Array.isArray(extracted.functionCalls)) {\n          for (const fc of extracted.functionCalls) accumulator.addToolCall(fc);\n        }\n        if (extracted.finishReason != null) {\n          accumulator.finishReason = String(extracted.finishReason);\n        }\n        accumulator.setUsage({\n          inputTokens: extracted.inputTokens,\n          outputTokens: extracted.outputTokens,\n          totalTokens: extracted.totalTokens,\n        });\n        yield chunk;\n      }\n    } catch (error) {\n      accumulator.setError(\n        error instanceof Error ? error.message : String(error),\n      );\n      throw error;\n    } finally {\n      if (finalResponse != null) {\n        try {\n          const extractedFinal = extractGeminiResponse(await finalResponse);\n          accumulator.setUsage({\n            inputTokens: extractedFinal.inputTokens,\n            outputTokens: extractedFinal.outputTokens,\n            totalTokens: extractedFinal.totalTokens,\n          });\n          if (extractedFinal.finishReason != null) {\n            accumulator.finishReason = String(extractedFinal.finishReason);\n          }\n          if (\n            Array.isArray(extractedFinal.functionCalls) &&\n            accumulator.toolCalls.length === 0\n          ) {\n            for (const fc of extractedFinal.functionCalls) {\n              accumulator.addToolCall(fc);\n            }\n          }\n        } catch {\n          // best-effort final response extraction\n        }\n      }\n\n      const state = accumulator.getState();\n      let costUsd: number | null = null;\n      if (state.inputTokens != null && state.outputTokens != null) {\n        try {\n          costUsd = calculateCost({\n            modelName: model,\n            inputTokens: state.inputTokens,\n            outputTokens: state.outputTokens,\n            defaultProvider: 'google',\n          });\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      this._track({\n        ...contextFields(ctx),\n        modelName: model,\n        provider: 'gemini',\n        responseContent: state.content,\n        latencyMs: accumulator.elapsedMs,\n        inputTokens: state.inputTokens,\n        outputTokens: state.outputTokens,\n        totalTokens: state.totalTokens,\n        totalCostUsd: costUsd,\n        finishReason: state.finishReason,\n        toolCalls: state.toolCalls.length > 0 ? state.toolCalls : undefined,\n        toolDefinitions: extractGeminiToolDefinitions(params),\n        systemPrompt: extractGeminiSystemPrompt(params),\n        temperature: extractGeminiTemperature(params),\n        topP: extractGeminiTopP(params),\n        maxOutputTokens: extractGeminiMaxOutputTokens(params),\n        providerTtfbMs: state.ttfbMs,\n        isStreaming: true,\n        isError: state.isError,\n        errorMessage: state.errorMessage,\n      });\n    }\n  }\n}\n\nexport function extractGeminiResponse(response: unknown): {\n  text: string;\n  inputTokens?: number;\n  outputTokens?: number;\n  totalTokens?: number;\n  finishReason?: string;\n  functionCalls?: Array<Record<string, unknown>>;\n} {\n  const resp = response as GeminiResponse;\n  const respObj = resp.response ?? resp;\n  let text = '';\n  if (typeof respObj.text === 'function') {\n    try {\n      text = String(respObj.text());\n    } catch {\n      // text() throws when the response has no candidates (e.g. safety block)\n    }\n  }\n  const usage = respObj.usageMetadata;\n\n  const candidate = respObj.candidates?.[0];\n  const finishReason = candidate?.finishReason;\n\n  const parts = candidate?.content?.parts;\n  const functionCalls = parts\n    ?.filter((p) => p.functionCall != null)\n    .map((p) => p.functionCall as Record<string, unknown>);\n\n  return {\n    text,\n    inputTokens: usage?.promptTokenCount,\n    outputTokens: usage?.candidatesTokenCount,\n    totalTokens: usage?.totalTokenCount,\n    finishReason,\n    functionCalls: functionCalls?.length ? functionCalls : undefined,\n  };\n}\n\nfunction extractGeminiSystemPrompt(\n  params: Record<string, unknown>,\n): string | undefined {\n  const systemInstruction =\n    (params.systemInstruction as string | undefined) ??\n    ((params.generationConfig as Record<string, unknown> | undefined)\n      ?.systemInstruction as string | undefined);\n  return systemInstruction;\n}\n\nfunction extractGeminiTemperature(\n  params: Record<string, unknown>,\n): number | undefined {\n  return (params.generationConfig as Record<string, unknown> | undefined)\n    ?.temperature as number | undefined;\n}\n\nfunction extractGeminiTopP(\n  params: Record<string, unknown>,\n): number | undefined {\n  return (params.generationConfig as Record<string, unknown> | undefined)\n    ?.topP as number | undefined;\n}\n\nfunction extractGeminiMaxOutputTokens(\n  params: Record<string, unknown>,\n): number | undefined {\n  return (params.generationConfig as Record<string, unknown> | undefined)\n    ?.maxOutputTokens as number | undefined;\n}\n\nfunction extractGeminiToolDefinitions(\n  params: Record<string, unknown>,\n): Array<Record<string, unknown>> | undefined {\n  const tools = params.tools;\n  return Array.isArray(tools) && tools.length > 0\n    ? (tools as Array<Record<string, unknown>>)\n    : undefined;\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"],"mappings":";;;;;;AAWA,MAAM,YAAY,WAAW,wBAAwB;AACrD,MAAa,mBAAmB,aAAa;AAC7C,MAAMA,gBAAgD;AAYtD,IAAa,SAAb,cAA4B,eAAe;CACzC,AAAQ;CAER,YAAY,SAAwB;AAClC,QAAM;GACJ,WAAW,QAAQ;GACnB,eAAe,QAAQ;GACvB,cAAc;GACf,CAAC;EAEF,MAAM,MACH,QAAQ,gBAAmD;AAC9D,MAAI,OAAO,KACT,OAAM,IAAI,MACR,2JACD;EAGH,MAAM,cAAc,IAAI;AAGxB,OAAK,UAAU,IAAI,YAAY,QAAQ,UAAU,GAAG;;CAGtD,MAAM,gBACJ,OACA,QACkB;EAKlB,MAAM,WAJY,KAAK,QACI,mBAGD,KAAK,KAAK,SAAS,EAAE,OAAO,CAAC;EACvD,MAAM,aAAa,SAAS;EAI5B,MAAM,YAAY,YAAY,KAAK;AAEnC,MAAI;GACF,MAAM,WAAW,MAAM,WAAW,KAAK,UAAU,OAAO;GACxD,MAAM,YAAY,YAAY,KAAK,GAAG;GAEtC,MAAM,YAAY,sBAAsB,SAAS;GACjD,IAAIC,UAAyB;AAC7B,OAAI,UAAU,eAAe,QAAQ,UAAU,gBAAgB,KAC7D,KAAI;AACF,cAAU,cAAc;KACtB,WAAW;KACX,aAAa,UAAU;KACvB,cAAc,UAAU;KACxB,iBAAiB;KAClB,CAAC;WACI;GAKV,MAAM,MAAM,qBAAqB;AACjC,QAAK,OAAO;IACV,GAAG,cAAc,IAAI;IACrB,WAAW;IACX,UAAU;IACV,iBAAiB,UAAU;IAC3B;IACA,aAAa,UAAU;IACvB,cAAc,UAAU;IACxB,aAAa,UAAU;IACvB,cAAc;IACd,cAAc,UAAU;IACxB,WAAW,UAAU,eAAe,SAChC,UAAU,gBACV;IACJ,iBAAiB,6BAA6B,OAAO;IACrD,cAAc,0BAA0B,OAAO;IAC/C,aAAa,yBAAyB,OAAO;IAC7C,MAAM,kBAAkB,OAAO;IAC/B,iBAAiB,6BAA6B,OAAO;IACrD,aAAa;IACd,CAAC;AAEF,UAAO;WACA,OAAO;GACd,MAAM,YAAY,YAAY,KAAK,GAAG;GACtC,MAAM,MAAM,qBAAqB;AAEjC,QAAK,OAAO;IACV,GAAG,cAAc,IAAI;IACrB,WAAW;IACX,UAAU;IACV,iBAAiB;IACjB;IACA,SAAS;IACT,cAAc,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IACrE,CAAC;AAEF,SAAM;;;CAIV,MAAM,sBACJ,OACA,QACkB;EAClB,MAAM,MAAM,qBAAqB;EACjC,MAAM,YAAY,YAAY,KAAK;AACnC,MAAI;GAKF,MAAM,WAJY,KAAK,QACI,mBAGD,KAAK,KAAK,SAAS,EAAE,OAAO,CAAC;GACvD,MAAM,WAAW,SAAS;AAG1B,OAAI,OAAO,aAAa,WACtB,OAAM,IAAI,MAAM,mDAAmD;GAIrE,MAAM,iBADW,MAAM,SAAS,KAAK,UAAU,OAAO;GAEtD,MAAM,SAAS,eAAe;GAG9B,MAAM,gBAAgB,eAAe;AAIrC,OAAI,CAAC,iBAAiB,OAAO,CAC3B,OAAM,IAAI,MAAM,8CAA8C;AAGhE,UAAO;IACL,GAAG;IACH,QAAQ,KAAK,YAAY,OAAO,QAAQ,QAAQ,eAAe,IAAI;IACpE;WACM,OAAO;AACd,QAAK,OAAO;IACV,GAAG,cAAc,IAAI;IACrB,WAAW;IACX,UAAU;IACV,iBAAiB;IACjB,WAAW,YAAY,KAAK,GAAG;IAC/B,SAAS;IACT,cAAc,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IACpE,aAAa;IACd,CAAC;AACF,SAAM;;;CAIV,IAAI,SAAkB;AACpB,SAAO,KAAK;;CAGd,OAAe,YACb,OACA,QACA,QACA,eACA,KACyB;EACzB,MAAM,cAAc,IAAI,sBAAsB;AAE9C,MAAI;AACF,cAAW,MAAM,SAAS,QAAQ;IAChC,MAAM,YAAY,sBAAsB,MAAM;AAC9C,QAAI,UAAU,KAAM,aAAY,WAAW,UAAU,KAAK;AAC1D,QAAI,MAAM,QAAQ,UAAU,cAAc,CACxC,MAAK,MAAM,MAAM,UAAU,cAAe,aAAY,YAAY,GAAG;AAEvE,QAAI,UAAU,gBAAgB,KAC5B,aAAY,eAAe,OAAO,UAAU,aAAa;AAE3D,gBAAY,SAAS;KACnB,aAAa,UAAU;KACvB,cAAc,UAAU;KACxB,aAAa,UAAU;KACxB,CAAC;AACF,UAAM;;WAED,OAAO;AACd,eAAY,SACV,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,CACvD;AACD,SAAM;YACE;AACR,OAAI,iBAAiB,KACnB,KAAI;IACF,MAAM,iBAAiB,sBAAsB,MAAM,cAAc;AACjE,gBAAY,SAAS;KACnB,aAAa,eAAe;KAC5B,cAAc,eAAe;KAC7B,aAAa,eAAe;KAC7B,CAAC;AACF,QAAI,eAAe,gBAAgB,KACjC,aAAY,eAAe,OAAO,eAAe,aAAa;AAEhE,QACE,MAAM,QAAQ,eAAe,cAAc,IAC3C,YAAY,UAAU,WAAW,EAEjC,MAAK,MAAM,MAAM,eAAe,cAC9B,aAAY,YAAY,GAAG;WAGzB;GAKV,MAAM,QAAQ,YAAY,UAAU;GACpC,IAAIA,UAAyB;AAC7B,OAAI,MAAM,eAAe,QAAQ,MAAM,gBAAgB,KACrD,KAAI;AACF,cAAU,cAAc;KACtB,WAAW;KACX,aAAa,MAAM;KACnB,cAAc,MAAM;KACpB,iBAAiB;KAClB,CAAC;WACI;AAKV,QAAK,OAAO;IACV,GAAG,cAAc,IAAI;IACrB,WAAW;IACX,UAAU;IACV,iBAAiB,MAAM;IACvB,WAAW,YAAY;IACvB,aAAa,MAAM;IACnB,cAAc,MAAM;IACpB,aAAa,MAAM;IACnB,cAAc;IACd,cAAc,MAAM;IACpB,WAAW,MAAM,UAAU,SAAS,IAAI,MAAM,YAAY;IAC1D,iBAAiB,6BAA6B,OAAO;IACrD,cAAc,0BAA0B,OAAO;IAC/C,aAAa,yBAAyB,OAAO;IAC7C,MAAM,kBAAkB,OAAO;IAC/B,iBAAiB,6BAA6B,OAAO;IACrD,gBAAgB,MAAM;IACtB,aAAa;IACb,SAAS,MAAM;IACf,cAAc,MAAM;IACrB,CAAC;;;;AAKR,SAAgB,sBAAsB,UAOpC;CACA,MAAM,OAAO;CACb,MAAM,UAAU,KAAK,YAAY;CACjC,IAAI,OAAO;AACX,KAAI,OAAO,QAAQ,SAAS,WAC1B,KAAI;AACF,SAAO,OAAO,QAAQ,MAAM,CAAC;SACvB;CAIV,MAAM,QAAQ,QAAQ;CAEtB,MAAM,YAAY,QAAQ,aAAa;CACvC,MAAM,eAAe,WAAW;CAGhC,MAAM,iBADQ,WAAW,SAAS,QAE9B,QAAQ,MAAM,EAAE,gBAAgB,KAAK,CACtC,KAAK,MAAM,EAAE,aAAwC;AAExD,QAAO;EACL;EACA,aAAa,OAAO;EACpB,cAAc,OAAO;EACrB,aAAa,OAAO;EACpB;EACA,eAAe,eAAe,SAAS,gBAAgB;EACxD;;AAGH,SAAS,0BACP,QACoB;AAKpB,QAHG,OAAO,qBACN,OAAO,kBACL;;AAIR,SAAS,yBACP,QACoB;AACpB,QAAQ,OAAO,kBACX;;AAGN,SAAS,kBACP,QACoB;AACpB,QAAQ,OAAO,kBACX;;AAGN,SAAS,6BACP,QACoB;AACpB,QAAQ,OAAO,kBACX;;AAGN,SAAS,6BACP,QAC4C;CAC5C,MAAM,QAAQ,OAAO;AACrB,QAAO,MAAM,QAAQ,MAAM,IAAI,MAAM,SAAS,IACzC,QACD;;AAGN,SAAS,iBAAiB,OAAiD;AACzE,QACE,SAAS,QACT,OAAQ,MAAkC,OAAO,mBAC/C"}

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

@@ -1 +1 @@
-{"version":3,"file":"mistral.d.ts","names":[],"sources":["../../src/providers/mistral.ts"],"sourcesContent":[],"mappings":";;;;;;AAaM,cADO,iBACS,EAAM,OAAA;AAI5B,cAJM,cAIyB,EAJT,MAKT,CAAA,MAAA,EAAA,OAEK,CAAA,GAAA,IAAA;AAOD,UAVA,cAAA,CAUA;EAEM,SAAA,EAXV,aAWU;EAJM,MAAA,CAAA,EAAA,MAAA;EAAc,aAAA,CAAA,EALzB,aAKyB,GAAA,IAAA;EAoC9B;EAI2B,aAAA,CAAA,EAAA,OAAA;;AAQW,cAhDtC,OAAA,SAAgB,cAAA,CAgDsB;EAmF5B,QAAA,OAAA;EAA0B,SAAA,IAAA,EAjIhC,WAiIgC;EAAO,WAAA,CAAA,OAAA,EA/HjC,cA+HiC;;;cA/F3C,WAAA;;;wCAI2B;mBAQf,0BAA0B;iBAmF5B,0BAA0B"}
+{"version":3,"file":"mistral.d.ts","names":[],"sources":["../../src/providers/mistral.ts"],"sourcesContent":[],"mappings":";;;;;;AAaM,cADO,iBACS,EAAM,OAAA;AAI5B,cAJM,cAIyB,EAJT,MAKT,CAAA,MAAA,EAAA,OAEK,CAAA,GAAA,IAAA;AAOD,UAVA,cAAA,CAUA;EAEM,SAAA,EAXV,aAWU;EAJM,MAAA,CAAA,EAAA,MAAA;EAAc,aAAA,CAAA,EALzB,aAKyB,GAAA,IAAA;EAoC9B;EAI2B,aAAA,CAAA,EAAA,OAAA;;AAQW,cAhDtC,OAAA,SAAgB,cAAA,CAgDsB;EAoF5B,QAAA,OAAA;EAA0B,SAAA,IAAA,EAlIhC,WAkIgC;EAAO,WAAA,CAAA,OAAA,EAhIjC,cAgIiC;;;cAhG3C,WAAA;;;wCAI2B;mBAQf,0BAA0B;iBAoF5B,0BAA0B"}

package/dist/providers/mistral.js

@@ -70,6 +70,7 @@ var WrappedChat = class {
 				totalCostUsd: costUsd,
 				finishReason: choice?.finish_reason,
 				toolCalls: toolCalls ?? void 0,
+				toolDefinitions: extractMistralToolDefinitions(params),
 				systemPrompt: extractMistralSystemPrompt(params),
 				temperature: params.temperature,
 				topP: params.top_p,
@@ -179,6 +180,7 @@ var WrappedChat = class {
 				totalCostUsd: costUsd,
 				finishReason: state.finishReason,
 				toolCalls: state.toolCalls.length > 0 ? state.toolCalls : void 0,
+				toolDefinitions: extractMistralToolDefinitions(params),
 				systemPrompt: extractMistralSystemPrompt(params),
 				temperature: params.temperature,
 				topP: params.top_p,
@@ -207,6 +209,10 @@ function extractMistralSystemPrompt(params) {
 	const systemMessage = messages.find((m) => m.role === "system");
 	return typeof systemMessage?.content === "string" ? systemMessage.content : void 0;
 }
+function extractMistralToolDefinitions(params) {
+	const tools = params.tools;
+	return Array.isArray(tools) && tools.length > 0 ? tools : void 0;
+}
 function _isAsyncIterable(value) {
 	return value != null && typeof value[Symbol.asyncIterator] === "function";
 }

package/dist/providers/mistral.js.map

@@ -1 +1 @@
-{"version":3,"file":"mistral.js","names":["_MistralModule: Record<string, unknown> | null","clientOpts: Record<string, unknown>","costUsd: number | null"],"sources":["../../src/providers/mistral.ts"],"sourcesContent":["/**\n * Mistral AI provider wrapper with automatic tracking.\n */\n\nimport type { PrivacyConfig } from '../core/privacy.js';\nimport type { AmplitudeOrAI, MistralChatResponse, TrackFn } from '../types.js';\nimport { calculateCost } from '../utils/costs.js';\nimport { tryRequire } from '../utils/resolve-module.js';\nimport { StreamingAccumulator } from '../utils/streaming.js';\nimport { applySessionContext, BaseAIProvider, contextFields } from './base.js';\n\nconst _resolved = tryRequire('@mistralai/mistralai');\nexport const MISTRAL_AVAILABLE = _resolved != null;\nconst _MistralModule: Record<string, unknown> | null = _resolved;\n\nexport { _MistralModule };\n\nexport interface MistralOptions {\n  amplitude: AmplitudeOrAI;\n  apiKey?: string;\n  privacyConfig?: PrivacyConfig | null;\n  /** Pass the `@mistralai/mistralai` module directly to bypass `tryRequire` (required in bundler environments). */\n  mistralModule?: unknown;\n}\n\nexport class Mistral extends BaseAIProvider {\n  private _client: unknown;\n  readonly chat: WrappedChat;\n\n  constructor(options: MistralOptions) {\n    super({\n      amplitude: options.amplitude,\n      privacyConfig: options.privacyConfig,\n      providerName: 'mistral',\n    });\n\n    const mod =\n      (options.mistralModule as Record<string, unknown> | null) ??\n      _MistralModule;\n    if (mod == null) {\n      throw new Error(\n        '@mistralai/mistralai package is required. Install it with: npm install @mistralai/mistralai — or pass the module directly via the mistralModule option.',\n      );\n    }\n\n    const MistralSDK = (mod.Mistral ??\n      mod.MistralClient ??\n      mod.default) as new (opts: Record<string, unknown>) => unknown;\n\n    const clientOpts: Record<string, unknown> = {};\n    if (options.apiKey) clientOpts.apiKey = options.apiKey;\n\n    this._client = new MistralSDK(clientOpts);\n    this.chat = new WrappedChat(this._client, this.trackFn());\n  }\n\n  get client(): unknown {\n    return this._client;\n  }\n}\n\nexport class WrappedChat {\n  private _client: unknown;\n  private _trackFn: TrackFn;\n\n  constructor(client: unknown, trackFn: TrackFn) {\n    this._client = client;\n    this._trackFn =\n      typeof trackFn === 'function'\n        ? trackFn\n        : (trackFn as unknown as { trackFn(): TrackFn }).trackFn();\n  }\n\n  async complete(params: Record<string, unknown>): Promise<unknown> {\n    const clientObj = this._client as Record<string, unknown>;\n    const chat = clientObj.chat as Record<string, unknown>;\n    const completeFn = chat.complete as (\n      ...args: unknown[]\n    ) => Promise<unknown>;\n\n    const startTime = performance.now();\n    const ctx = applySessionContext();\n\n    try {\n      const response = await completeFn.call(chat, params);\n      if (\n        (params.stream === true || chat.stream != null) &&\n        _isAsyncIterable(response)\n      ) {\n        return this._wrapStream(\n          response as AsyncIterable<unknown>,\n          params,\n          ctx,\n        );\n      }\n      const latencyMs = performance.now() - startTime;\n\n      const resp = response as MistralChatResponse;\n      const choice = resp.choices?.[0];\n      const usage = resp.usage;\n      const modelName = String(resp.model ?? params.model ?? 'unknown');\n      const toolCalls = (\n        choice?.message as { tool_calls?: Array<Record<string, unknown>> }\n      )?.tool_calls;\n\n      let costUsd: number | null = null;\n      if (usage?.prompt_tokens != null && usage?.completion_tokens != null) {\n        try {\n          costUsd = calculateCost({\n            modelName,\n            inputTokens: usage.prompt_tokens,\n            outputTokens: usage.completion_tokens,\n            defaultProvider: 'mistral',\n          });\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      this._trackFn({\n        ...contextFields(ctx),\n        modelName,\n        provider: 'mistral',\n        responseContent: extractMistralContent(choice?.message?.content),\n        latencyMs,\n        inputTokens: usage?.prompt_tokens,\n        outputTokens: usage?.completion_tokens,\n        totalTokens: usage?.total_tokens,\n        totalCostUsd: costUsd,\n        finishReason: choice?.finish_reason,\n        toolCalls: toolCalls ?? undefined,\n        systemPrompt: extractMistralSystemPrompt(params),\n        temperature: params.temperature as number | undefined,\n        topP: params.top_p as number | undefined,\n        maxOutputTokens: params.max_tokens as number | undefined,\n        isStreaming: false,\n      });\n\n      return response;\n    } catch (error) {\n      const latencyMs = performance.now() - startTime;\n\n      this._trackFn({\n        ...contextFields(ctx),\n        modelName: String(params.model ?? 'unknown'),\n        provider: 'mistral',\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  async stream(params: Record<string, unknown>): Promise<unknown> {\n    const ctx = applySessionContext();\n    const startTime = performance.now();\n    try {\n      const clientObj = this._client as Record<string, unknown>;\n      const chat = clientObj.chat as Record<string, unknown>;\n      const streamFn = chat.stream as\n        | ((...args: unknown[]) => Promise<unknown>)\n        | undefined;\n      if (typeof streamFn !== 'function') {\n        throw new Error('Mistral SDK does not expose chat.stream');\n      }\n\n      const response = await streamFn.call(chat, params);\n      if (!_isAsyncIterable(response)) {\n        throw new Error('Mistral stream response is not AsyncIterable');\n      }\n      return this._wrapStream(response as AsyncIterable<unknown>, params, ctx);\n    } catch (error) {\n      this._trackFn({\n        ...contextFields(ctx),\n        modelName: String(params.model ?? 'unknown'),\n        provider: 'mistral',\n        responseContent: '',\n        latencyMs: performance.now() - startTime,\n        isError: true,\n        errorMessage: error instanceof Error ? error.message : String(error),\n        isStreaming: true,\n      });\n      throw error;\n    }\n  }\n\n  private async *_wrapStream(\n    stream: AsyncIterable<unknown>,\n    params: Record<string, unknown>,\n    sessionCtx: ReturnType<typeof applySessionContext>,\n  ): AsyncGenerator<unknown> {\n    const accumulator = new StreamingAccumulator();\n    accumulator.model = String(params.model ?? 'unknown');\n\n    try {\n      for await (const chunk of stream) {\n        const c = chunk as Record<string, unknown>;\n        const choices = c.choices as Array<Record<string, unknown>> | undefined;\n        const delta = choices?.[0]?.delta as\n          | Record<string, unknown>\n          | undefined;\n        const message = choices?.[0]?.message as\n          | Record<string, unknown>\n          | undefined;\n\n        const content =\n          (delta?.content as string | undefined) ??\n          (message?.content as string | undefined);\n        if (typeof content === 'string' && content.length > 0) {\n          accumulator.addContent(content);\n        }\n\n        const toolCalls =\n          (delta?.tool_calls as Array<Record<string, unknown>> | undefined) ??\n          (message?.tool_calls as Array<Record<string, unknown>> | undefined);\n        if (Array.isArray(toolCalls)) {\n          for (const call of toolCalls) {\n            const idx = call.index as number | undefined;\n            const id = call.id as string | undefined;\n            const fn = call.function as Record<string, unknown> | undefined;\n            if (idx != null && id && fn?.name != null) {\n              accumulator.setToolCallAt(idx, {\n                type: 'function',\n                id,\n                function: {\n                  name: fn.name,\n                  arguments: ((fn.arguments as string) ?? ''),\n                },\n              });\n            } else if (idx != null && fn?.arguments) {\n              accumulator.appendToolCallArgs(idx, fn.arguments as string);\n            } else {\n              accumulator.addToolCall(call);\n            }\n          }\n        }\n\n        const finishReason = choices?.[0]?.finish_reason;\n        if (finishReason != null)\n          accumulator.finishReason = String(finishReason);\n\n        const usage = c.usage as Record<string, unknown> | undefined;\n        accumulator.setUsage({\n          inputTokens: usage?.prompt_tokens as number | undefined,\n          outputTokens: usage?.completion_tokens as number | undefined,\n          totalTokens: usage?.total_tokens as number | undefined,\n        });\n\n        yield chunk;\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      let costUsd: number | null = null;\n      if (state.inputTokens != null && state.outputTokens != null) {\n        try {\n          costUsd = calculateCost({\n            modelName,\n            inputTokens: state.inputTokens,\n            outputTokens: state.outputTokens,\n            defaultProvider: 'mistral',\n          });\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      this._trackFn({\n        ...contextFields(sessionCtx),\n        modelName,\n        provider: 'mistral',\n        responseContent: state.content,\n        latencyMs: accumulator.elapsedMs,\n        inputTokens: state.inputTokens,\n        outputTokens: state.outputTokens,\n        totalTokens: state.totalTokens,\n        totalCostUsd: costUsd,\n        finishReason: state.finishReason,\n        toolCalls: state.toolCalls.length > 0 ? state.toolCalls : undefined,\n        systemPrompt: extractMistralSystemPrompt(params),\n        temperature: params.temperature as number | undefined,\n        topP: params.top_p as number | undefined,\n        maxOutputTokens: params.max_tokens as number | undefined,\n        providerTtfbMs: state.ttfbMs,\n        isStreaming: true,\n        isError: state.isError,\n        errorMessage: state.errorMessage,\n      });\n    }\n  }\n}\n\nfunction extractMistralContent(content: unknown): string {\n  if (typeof content === 'string') return content;\n  if (content == null) return '';\n  if (Array.isArray(content)) {\n    return content\n      .map((chunk) => {\n        if (typeof chunk === 'string') return chunk;\n        if (typeof chunk === 'object' && chunk != null) {\n          return String((chunk as Record<string, unknown>).text ?? '');\n        }\n        return '';\n      })\n      .filter(Boolean)\n      .join('');\n  }\n  return String(content);\n}\n\nfunction extractMistralSystemPrompt(\n  params: Record<string, unknown>,\n): string | undefined {\n  const messages = params.messages as\n    | Array<Record<string, unknown>>\n    | undefined;\n  if (!Array.isArray(messages)) return undefined;\n  const systemMessage = messages.find((m) => m.role === 'system');\n  return typeof systemMessage?.content === 'string'\n    ? systemMessage.content\n    : undefined;\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"],"mappings":";;;;;;AAWA,MAAM,YAAY,WAAW,uBAAuB;AACpD,MAAa,oBAAoB,aAAa;AAC9C,MAAMA,iBAAiD;AAYvD,IAAa,UAAb,cAA6B,eAAe;CAC1C,AAAQ;CACR,AAAS;CAET,YAAY,SAAyB;AACnC,QAAM;GACJ,WAAW,QAAQ;GACnB,eAAe,QAAQ;GACvB,cAAc;GACf,CAAC;EAEF,MAAM,MACH,QAAQ,iBACT;AACF,MAAI,OAAO,KACT,OAAM,IAAI,MACR,0JACD;EAGH,MAAM,aAAc,IAAI,WACtB,IAAI,iBACJ,IAAI;EAEN,MAAMC,aAAsC,EAAE;AAC9C,MAAI,QAAQ,OAAQ,YAAW,SAAS,QAAQ;AAEhD,OAAK,UAAU,IAAI,WAAW,WAAW;AACzC,OAAK,OAAO,IAAI,YAAY,KAAK,SAAS,KAAK,SAAS,CAAC;;CAG3D,IAAI,SAAkB;AACpB,SAAO,KAAK;;;AAIhB,IAAa,cAAb,MAAyB;CACvB,AAAQ;CACR,AAAQ;CAER,YAAY,QAAiB,SAAkB;AAC7C,OAAK,UAAU;AACf,OAAK,WACH,OAAO,YAAY,aACf,UACC,QAA8C,SAAS;;CAGhE,MAAM,SAAS,QAAmD;EAEhE,MAAM,OADY,KAAK,QACA;EACvB,MAAM,aAAa,KAAK;EAIxB,MAAM,YAAY,YAAY,KAAK;EACnC,MAAM,MAAM,qBAAqB;AAEjC,MAAI;GACF,MAAM,WAAW,MAAM,WAAW,KAAK,MAAM,OAAO;AACpD,QACG,OAAO,WAAW,QAAQ,KAAK,UAAU,SAC1C,iBAAiB,SAAS,CAE1B,QAAO,KAAK,YACV,UACA,QACA,IACD;GAEH,MAAM,YAAY,YAAY,KAAK,GAAG;GAEtC,MAAM,OAAO;GACb,MAAM,SAAS,KAAK,UAAU;GAC9B,MAAM,QAAQ,KAAK;GACnB,MAAM,YAAY,OAAO,KAAK,SAAS,OAAO,SAAS,UAAU;GACjE,MAAM,aACJ,QAAQ,UACP;GAEH,IAAIC,UAAyB;AAC7B,OAAI,OAAO,iBAAiB,QAAQ,OAAO,qBAAqB,KAC9D,KAAI;AACF,cAAU,cAAc;KACtB;KACA,aAAa,MAAM;KACnB,cAAc,MAAM;KACpB,iBAAiB;KAClB,CAAC;WACI;AAKV,QAAK,SAAS;IACZ,GAAG,cAAc,IAAI;IACrB;IACA,UAAU;IACV,iBAAiB,sBAAsB,QAAQ,SAAS,QAAQ;IAChE;IACA,aAAa,OAAO;IACpB,cAAc,OAAO;IACrB,aAAa,OAAO;IACpB,cAAc;IACd,cAAc,QAAQ;IACtB,WAAW,aAAa;IACxB,cAAc,2BAA2B,OAAO;IAChD,aAAa,OAAO;IACpB,MAAM,OAAO;IACb,iBAAiB,OAAO;IACxB,aAAa;IACd,CAAC;AAEF,UAAO;WACA,OAAO;GACd,MAAM,YAAY,YAAY,KAAK,GAAG;AAEtC,QAAK,SAAS;IACZ,GAAG,cAAc,IAAI;IACrB,WAAW,OAAO,OAAO,SAAS,UAAU;IAC5C,UAAU;IACV,iBAAiB;IACjB;IACA,SAAS;IACT,cAAc,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IACrE,CAAC;AAEF,SAAM;;;CAIV,MAAM,OAAO,QAAmD;EAC9D,MAAM,MAAM,qBAAqB;EACjC,MAAM,YAAY,YAAY,KAAK;AACnC,MAAI;GAEF,MAAM,OADY,KAAK,QACA;GACvB,MAAM,WAAW,KAAK;AAGtB,OAAI,OAAO,aAAa,WACtB,OAAM,IAAI,MAAM,0CAA0C;GAG5D,MAAM,WAAW,MAAM,SAAS,KAAK,MAAM,OAAO;AAClD,OAAI,CAAC,iBAAiB,SAAS,CAC7B,OAAM,IAAI,MAAM,+CAA+C;AAEjE,UAAO,KAAK,YAAY,UAAoC,QAAQ,IAAI;WACjE,OAAO;AACd,QAAK,SAAS;IACZ,GAAG,cAAc,IAAI;IACrB,WAAW,OAAO,OAAO,SAAS,UAAU;IAC5C,UAAU;IACV,iBAAiB;IACjB,WAAW,YAAY,KAAK,GAAG;IAC/B,SAAS;IACT,cAAc,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IACpE,aAAa;IACd,CAAC;AACF,SAAM;;;CAIV,OAAe,YACb,QACA,QACA,YACyB;EACzB,MAAM,cAAc,IAAI,sBAAsB;AAC9C,cAAY,QAAQ,OAAO,OAAO,SAAS,UAAU;AAErD,MAAI;AACF,cAAW,MAAM,SAAS,QAAQ;IAChC,MAAM,IAAI;IACV,MAAM,UAAU,EAAE;IAClB,MAAM,QAAQ,UAAU,IAAI;IAG5B,MAAM,UAAU,UAAU,IAAI;IAI9B,MAAM,UACH,OAAO,WACP,SAAS;AACZ,QAAI,OAAO,YAAY,YAAY,QAAQ,SAAS,EAClD,aAAY,WAAW,QAAQ;IAGjC,MAAM,YACH,OAAO,cACP,SAAS;AACZ,QAAI,MAAM,QAAQ,UAAU,CAC1B,MAAK,MAAM,QAAQ,WAAW;KAC5B,MAAM,MAAM,KAAK;KACjB,MAAM,KAAK,KAAK;KAChB,MAAM,KAAK,KAAK;AAChB,SAAI,OAAO,QAAQ,MAAM,IAAI,QAAQ,KACnC,aAAY,cAAc,KAAK;MAC7B,MAAM;MACN;MACA,UAAU;OACR,MAAM,GAAG;OACT,WAAa,GAAG,aAAwB;OACzC;MACF,CAAC;cACO,OAAO,QAAQ,IAAI,UAC5B,aAAY,mBAAmB,KAAK,GAAG,UAAoB;SAE3D,aAAY,YAAY,KAAK;;IAKnC,MAAM,eAAe,UAAU,IAAI;AACnC,QAAI,gBAAgB,KAClB,aAAY,eAAe,OAAO,aAAa;IAEjD,MAAM,QAAQ,EAAE;AAChB,gBAAY,SAAS;KACnB,aAAa,OAAO;KACpB,cAAc,OAAO;KACrB,aAAa,OAAO;KACrB,CAAC;AAEF,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;GACxE,IAAIA,UAAyB;AAC7B,OAAI,MAAM,eAAe,QAAQ,MAAM,gBAAgB,KACrD,KAAI;AACF,cAAU,cAAc;KACtB;KACA,aAAa,MAAM;KACnB,cAAc,MAAM;KACpB,iBAAiB;KAClB,CAAC;WACI;AAKV,QAAK,SAAS;IACZ,GAAG,cAAc,WAAW;IAC5B;IACA,UAAU;IACV,iBAAiB,MAAM;IACvB,WAAW,YAAY;IACvB,aAAa,MAAM;IACnB,cAAc,MAAM;IACpB,aAAa,MAAM;IACnB,cAAc;IACd,cAAc,MAAM;IACpB,WAAW,MAAM,UAAU,SAAS,IAAI,MAAM,YAAY;IAC1D,cAAc,2BAA2B,OAAO;IAChD,aAAa,OAAO;IACpB,MAAM,OAAO;IACb,iBAAiB,OAAO;IACxB,gBAAgB,MAAM;IACtB,aAAa;IACb,SAAS,MAAM;IACf,cAAc,MAAM;IACrB,CAAC;;;;AAKR,SAAS,sBAAsB,SAA0B;AACvD,KAAI,OAAO,YAAY,SAAU,QAAO;AACxC,KAAI,WAAW,KAAM,QAAO;AAC5B,KAAI,MAAM,QAAQ,QAAQ,CACxB,QAAO,QACJ,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,GAAG;AAEb,QAAO,OAAO,QAAQ;;AAGxB,SAAS,2BACP,QACoB;CACpB,MAAM,WAAW,OAAO;AAGxB,KAAI,CAAC,MAAM,QAAQ,SAAS,CAAE,QAAO;CACrC,MAAM,gBAAgB,SAAS,MAAM,MAAM,EAAE,SAAS,SAAS;AAC/D,QAAO,OAAO,eAAe,YAAY,WACrC,cAAc,UACd;;AAGN,SAAS,iBAAiB,OAAiD;AACzE,QACE,SAAS,QACT,OAAQ,MAAkC,OAAO,mBAC/C"}
+{"version":3,"file":"mistral.js","names":["_MistralModule: Record<string, unknown> | null","clientOpts: Record<string, unknown>","costUsd: number | null"],"sources":["../../src/providers/mistral.ts"],"sourcesContent":["/**\n * Mistral AI provider wrapper with automatic tracking.\n */\n\nimport type { PrivacyConfig } from '../core/privacy.js';\nimport type { AmplitudeOrAI, MistralChatResponse, TrackFn } from '../types.js';\nimport { calculateCost } from '../utils/costs.js';\nimport { tryRequire } from '../utils/resolve-module.js';\nimport { StreamingAccumulator } from '../utils/streaming.js';\nimport { applySessionContext, BaseAIProvider, contextFields } from './base.js';\n\nconst _resolved = tryRequire('@mistralai/mistralai');\nexport const MISTRAL_AVAILABLE = _resolved != null;\nconst _MistralModule: Record<string, unknown> | null = _resolved;\n\nexport { _MistralModule };\n\nexport interface MistralOptions {\n  amplitude: AmplitudeOrAI;\n  apiKey?: string;\n  privacyConfig?: PrivacyConfig | null;\n  /** Pass the `@mistralai/mistralai` module directly to bypass `tryRequire` (required in bundler environments). */\n  mistralModule?: unknown;\n}\n\nexport class Mistral extends BaseAIProvider {\n  private _client: unknown;\n  readonly chat: WrappedChat;\n\n  constructor(options: MistralOptions) {\n    super({\n      amplitude: options.amplitude,\n      privacyConfig: options.privacyConfig,\n      providerName: 'mistral',\n    });\n\n    const mod =\n      (options.mistralModule as Record<string, unknown> | null) ??\n      _MistralModule;\n    if (mod == null) {\n      throw new Error(\n        '@mistralai/mistralai package is required. Install it with: npm install @mistralai/mistralai — or pass the module directly via the mistralModule option.',\n      );\n    }\n\n    const MistralSDK = (mod.Mistral ??\n      mod.MistralClient ??\n      mod.default) as new (opts: Record<string, unknown>) => unknown;\n\n    const clientOpts: Record<string, unknown> = {};\n    if (options.apiKey) clientOpts.apiKey = options.apiKey;\n\n    this._client = new MistralSDK(clientOpts);\n    this.chat = new WrappedChat(this._client, this.trackFn());\n  }\n\n  get client(): unknown {\n    return this._client;\n  }\n}\n\nexport class WrappedChat {\n  private _client: unknown;\n  private _trackFn: TrackFn;\n\n  constructor(client: unknown, trackFn: TrackFn) {\n    this._client = client;\n    this._trackFn =\n      typeof trackFn === 'function'\n        ? trackFn\n        : (trackFn as unknown as { trackFn(): TrackFn }).trackFn();\n  }\n\n  async complete(params: Record<string, unknown>): Promise<unknown> {\n    const clientObj = this._client as Record<string, unknown>;\n    const chat = clientObj.chat as Record<string, unknown>;\n    const completeFn = chat.complete as (\n      ...args: unknown[]\n    ) => Promise<unknown>;\n\n    const startTime = performance.now();\n    const ctx = applySessionContext();\n\n    try {\n      const response = await completeFn.call(chat, params);\n      if (\n        (params.stream === true || chat.stream != null) &&\n        _isAsyncIterable(response)\n      ) {\n        return this._wrapStream(\n          response as AsyncIterable<unknown>,\n          params,\n          ctx,\n        );\n      }\n      const latencyMs = performance.now() - startTime;\n\n      const resp = response as MistralChatResponse;\n      const choice = resp.choices?.[0];\n      const usage = resp.usage;\n      const modelName = String(resp.model ?? params.model ?? 'unknown');\n      const toolCalls = (\n        choice?.message as { tool_calls?: Array<Record<string, unknown>> }\n      )?.tool_calls;\n\n      let costUsd: number | null = null;\n      if (usage?.prompt_tokens != null && usage?.completion_tokens != null) {\n        try {\n          costUsd = calculateCost({\n            modelName,\n            inputTokens: usage.prompt_tokens,\n            outputTokens: usage.completion_tokens,\n            defaultProvider: 'mistral',\n          });\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      this._trackFn({\n        ...contextFields(ctx),\n        modelName,\n        provider: 'mistral',\n        responseContent: extractMistralContent(choice?.message?.content),\n        latencyMs,\n        inputTokens: usage?.prompt_tokens,\n        outputTokens: usage?.completion_tokens,\n        totalTokens: usage?.total_tokens,\n        totalCostUsd: costUsd,\n        finishReason: choice?.finish_reason,\n        toolCalls: toolCalls ?? undefined,\n        toolDefinitions: extractMistralToolDefinitions(params),\n        systemPrompt: extractMistralSystemPrompt(params),\n        temperature: params.temperature as number | undefined,\n        topP: params.top_p as number | undefined,\n        maxOutputTokens: params.max_tokens as number | undefined,\n        isStreaming: false,\n      });\n\n      return response;\n    } catch (error) {\n      const latencyMs = performance.now() - startTime;\n\n      this._trackFn({\n        ...contextFields(ctx),\n        modelName: String(params.model ?? 'unknown'),\n        provider: 'mistral',\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  async stream(params: Record<string, unknown>): Promise<unknown> {\n    const ctx = applySessionContext();\n    const startTime = performance.now();\n    try {\n      const clientObj = this._client as Record<string, unknown>;\n      const chat = clientObj.chat as Record<string, unknown>;\n      const streamFn = chat.stream as\n        | ((...args: unknown[]) => Promise<unknown>)\n        | undefined;\n      if (typeof streamFn !== 'function') {\n        throw new Error('Mistral SDK does not expose chat.stream');\n      }\n\n      const response = await streamFn.call(chat, params);\n      if (!_isAsyncIterable(response)) {\n        throw new Error('Mistral stream response is not AsyncIterable');\n      }\n      return this._wrapStream(response as AsyncIterable<unknown>, params, ctx);\n    } catch (error) {\n      this._trackFn({\n        ...contextFields(ctx),\n        modelName: String(params.model ?? 'unknown'),\n        provider: 'mistral',\n        responseContent: '',\n        latencyMs: performance.now() - startTime,\n        isError: true,\n        errorMessage: error instanceof Error ? error.message : String(error),\n        isStreaming: true,\n      });\n      throw error;\n    }\n  }\n\n  private async *_wrapStream(\n    stream: AsyncIterable<unknown>,\n    params: Record<string, unknown>,\n    sessionCtx: ReturnType<typeof applySessionContext>,\n  ): AsyncGenerator<unknown> {\n    const accumulator = new StreamingAccumulator();\n    accumulator.model = String(params.model ?? 'unknown');\n\n    try {\n      for await (const chunk of stream) {\n        const c = chunk as Record<string, unknown>;\n        const choices = c.choices as Array<Record<string, unknown>> | undefined;\n        const delta = choices?.[0]?.delta as\n          | Record<string, unknown>\n          | undefined;\n        const message = choices?.[0]?.message as\n          | Record<string, unknown>\n          | undefined;\n\n        const content =\n          (delta?.content as string | undefined) ??\n          (message?.content as string | undefined);\n        if (typeof content === 'string' && content.length > 0) {\n          accumulator.addContent(content);\n        }\n\n        const toolCalls =\n          (delta?.tool_calls as Array<Record<string, unknown>> | undefined) ??\n          (message?.tool_calls as Array<Record<string, unknown>> | undefined);\n        if (Array.isArray(toolCalls)) {\n          for (const call of toolCalls) {\n            const idx = call.index as number | undefined;\n            const id = call.id as string | undefined;\n            const fn = call.function as Record<string, unknown> | undefined;\n            if (idx != null && id && fn?.name != null) {\n              accumulator.setToolCallAt(idx, {\n                type: 'function',\n                id,\n                function: {\n                  name: fn.name,\n                  arguments: ((fn.arguments as string) ?? ''),\n                },\n              });\n            } else if (idx != null && fn?.arguments) {\n              accumulator.appendToolCallArgs(idx, fn.arguments as string);\n            } else {\n              accumulator.addToolCall(call);\n            }\n          }\n        }\n\n        const finishReason = choices?.[0]?.finish_reason;\n        if (finishReason != null)\n          accumulator.finishReason = String(finishReason);\n\n        const usage = c.usage as Record<string, unknown> | undefined;\n        accumulator.setUsage({\n          inputTokens: usage?.prompt_tokens as number | undefined,\n          outputTokens: usage?.completion_tokens as number | undefined,\n          totalTokens: usage?.total_tokens as number | undefined,\n        });\n\n        yield chunk;\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      let costUsd: number | null = null;\n      if (state.inputTokens != null && state.outputTokens != null) {\n        try {\n          costUsd = calculateCost({\n            modelName,\n            inputTokens: state.inputTokens,\n            outputTokens: state.outputTokens,\n            defaultProvider: 'mistral',\n          });\n        } catch {\n          // cost calculation is best-effort\n        }\n      }\n\n      this._trackFn({\n        ...contextFields(sessionCtx),\n        modelName,\n        provider: 'mistral',\n        responseContent: state.content,\n        latencyMs: accumulator.elapsedMs,\n        inputTokens: state.inputTokens,\n        outputTokens: state.outputTokens,\n        totalTokens: state.totalTokens,\n        totalCostUsd: costUsd,\n        finishReason: state.finishReason,\n        toolCalls: state.toolCalls.length > 0 ? state.toolCalls : undefined,\n        toolDefinitions: extractMistralToolDefinitions(params),\n        systemPrompt: extractMistralSystemPrompt(params),\n        temperature: params.temperature as number | undefined,\n        topP: params.top_p as number | undefined,\n        maxOutputTokens: params.max_tokens as number | undefined,\n        providerTtfbMs: state.ttfbMs,\n        isStreaming: true,\n        isError: state.isError,\n        errorMessage: state.errorMessage,\n      });\n    }\n  }\n}\n\nfunction extractMistralContent(content: unknown): string {\n  if (typeof content === 'string') return content;\n  if (content == null) return '';\n  if (Array.isArray(content)) {\n    return content\n      .map((chunk) => {\n        if (typeof chunk === 'string') return chunk;\n        if (typeof chunk === 'object' && chunk != null) {\n          return String((chunk as Record<string, unknown>).text ?? '');\n        }\n        return '';\n      })\n      .filter(Boolean)\n      .join('');\n  }\n  return String(content);\n}\n\nfunction extractMistralSystemPrompt(\n  params: Record<string, unknown>,\n): string | undefined {\n  const messages = params.messages as\n    | Array<Record<string, unknown>>\n    | undefined;\n  if (!Array.isArray(messages)) return undefined;\n  const systemMessage = messages.find((m) => m.role === 'system');\n  return typeof systemMessage?.content === 'string'\n    ? systemMessage.content\n    : undefined;\n}\n\nfunction extractMistralToolDefinitions(\n  params: Record<string, unknown>,\n): Array<Record<string, unknown>> | undefined {\n  const tools = params.tools;\n  return Array.isArray(tools) && tools.length > 0\n    ? (tools as Array<Record<string, unknown>>)\n    : undefined;\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"],"mappings":";;;;;;AAWA,MAAM,YAAY,WAAW,uBAAuB;AACpD,MAAa,oBAAoB,aAAa;AAC9C,MAAMA,iBAAiD;AAYvD,IAAa,UAAb,cAA6B,eAAe;CAC1C,AAAQ;CACR,AAAS;CAET,YAAY,SAAyB;AACnC,QAAM;GACJ,WAAW,QAAQ;GACnB,eAAe,QAAQ;GACvB,cAAc;GACf,CAAC;EAEF,MAAM,MACH,QAAQ,iBACT;AACF,MAAI,OAAO,KACT,OAAM,IAAI,MACR,0JACD;EAGH,MAAM,aAAc,IAAI,WACtB,IAAI,iBACJ,IAAI;EAEN,MAAMC,aAAsC,EAAE;AAC9C,MAAI,QAAQ,OAAQ,YAAW,SAAS,QAAQ;AAEhD,OAAK,UAAU,IAAI,WAAW,WAAW;AACzC,OAAK,OAAO,IAAI,YAAY,KAAK,SAAS,KAAK,SAAS,CAAC;;CAG3D,IAAI,SAAkB;AACpB,SAAO,KAAK;;;AAIhB,IAAa,cAAb,MAAyB;CACvB,AAAQ;CACR,AAAQ;CAER,YAAY,QAAiB,SAAkB;AAC7C,OAAK,UAAU;AACf,OAAK,WACH,OAAO,YAAY,aACf,UACC,QAA8C,SAAS;;CAGhE,MAAM,SAAS,QAAmD;EAEhE,MAAM,OADY,KAAK,QACA;EACvB,MAAM,aAAa,KAAK;EAIxB,MAAM,YAAY,YAAY,KAAK;EACnC,MAAM,MAAM,qBAAqB;AAEjC,MAAI;GACF,MAAM,WAAW,MAAM,WAAW,KAAK,MAAM,OAAO;AACpD,QACG,OAAO,WAAW,QAAQ,KAAK,UAAU,SAC1C,iBAAiB,SAAS,CAE1B,QAAO,KAAK,YACV,UACA,QACA,IACD;GAEH,MAAM,YAAY,YAAY,KAAK,GAAG;GAEtC,MAAM,OAAO;GACb,MAAM,SAAS,KAAK,UAAU;GAC9B,MAAM,QAAQ,KAAK;GACnB,MAAM,YAAY,OAAO,KAAK,SAAS,OAAO,SAAS,UAAU;GACjE,MAAM,aACJ,QAAQ,UACP;GAEH,IAAIC,UAAyB;AAC7B,OAAI,OAAO,iBAAiB,QAAQ,OAAO,qBAAqB,KAC9D,KAAI;AACF,cAAU,cAAc;KACtB;KACA,aAAa,MAAM;KACnB,cAAc,MAAM;KACpB,iBAAiB;KAClB,CAAC;WACI;AAKV,QAAK,SAAS;IACZ,GAAG,cAAc,IAAI;IACrB;IACA,UAAU;IACV,iBAAiB,sBAAsB,QAAQ,SAAS,QAAQ;IAChE;IACA,aAAa,OAAO;IACpB,cAAc,OAAO;IACrB,aAAa,OAAO;IACpB,cAAc;IACd,cAAc,QAAQ;IACtB,WAAW,aAAa;IACxB,iBAAiB,8BAA8B,OAAO;IACtD,cAAc,2BAA2B,OAAO;IAChD,aAAa,OAAO;IACpB,MAAM,OAAO;IACb,iBAAiB,OAAO;IACxB,aAAa;IACd,CAAC;AAEF,UAAO;WACA,OAAO;GACd,MAAM,YAAY,YAAY,KAAK,GAAG;AAEtC,QAAK,SAAS;IACZ,GAAG,cAAc,IAAI;IACrB,WAAW,OAAO,OAAO,SAAS,UAAU;IAC5C,UAAU;IACV,iBAAiB;IACjB;IACA,SAAS;IACT,cAAc,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IACrE,CAAC;AAEF,SAAM;;;CAIV,MAAM,OAAO,QAAmD;EAC9D,MAAM,MAAM,qBAAqB;EACjC,MAAM,YAAY,YAAY,KAAK;AACnC,MAAI;GAEF,MAAM,OADY,KAAK,QACA;GACvB,MAAM,WAAW,KAAK;AAGtB,OAAI,OAAO,aAAa,WACtB,OAAM,IAAI,MAAM,0CAA0C;GAG5D,MAAM,WAAW,MAAM,SAAS,KAAK,MAAM,OAAO;AAClD,OAAI,CAAC,iBAAiB,SAAS,CAC7B,OAAM,IAAI,MAAM,+CAA+C;AAEjE,UAAO,KAAK,YAAY,UAAoC,QAAQ,IAAI;WACjE,OAAO;AACd,QAAK,SAAS;IACZ,GAAG,cAAc,IAAI;IACrB,WAAW,OAAO,OAAO,SAAS,UAAU;IAC5C,UAAU;IACV,iBAAiB;IACjB,WAAW,YAAY,KAAK,GAAG;IAC/B,SAAS;IACT,cAAc,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;IACpE,aAAa;IACd,CAAC;AACF,SAAM;;;CAIV,OAAe,YACb,QACA,QACA,YACyB;EACzB,MAAM,cAAc,IAAI,sBAAsB;AAC9C,cAAY,QAAQ,OAAO,OAAO,SAAS,UAAU;AAErD,MAAI;AACF,cAAW,MAAM,SAAS,QAAQ;IAChC,MAAM,IAAI;IACV,MAAM,UAAU,EAAE;IAClB,MAAM,QAAQ,UAAU,IAAI;IAG5B,MAAM,UAAU,UAAU,IAAI;IAI9B,MAAM,UACH,OAAO,WACP,SAAS;AACZ,QAAI,OAAO,YAAY,YAAY,QAAQ,SAAS,EAClD,aAAY,WAAW,QAAQ;IAGjC,MAAM,YACH,OAAO,cACP,SAAS;AACZ,QAAI,MAAM,QAAQ,UAAU,CAC1B,MAAK,MAAM,QAAQ,WAAW;KAC5B,MAAM,MAAM,KAAK;KACjB,MAAM,KAAK,KAAK;KAChB,MAAM,KAAK,KAAK;AAChB,SAAI,OAAO,QAAQ,MAAM,IAAI,QAAQ,KACnC,aAAY,cAAc,KAAK;MAC7B,MAAM;MACN;MACA,UAAU;OACR,MAAM,GAAG;OACT,WAAa,GAAG,aAAwB;OACzC;MACF,CAAC;cACO,OAAO,QAAQ,IAAI,UAC5B,aAAY,mBAAmB,KAAK,GAAG,UAAoB;SAE3D,aAAY,YAAY,KAAK;;IAKnC,MAAM,eAAe,UAAU,IAAI;AACnC,QAAI,gBAAgB,KAClB,aAAY,eAAe,OAAO,aAAa;IAEjD,MAAM,QAAQ,EAAE;AAChB,gBAAY,SAAS;KACnB,aAAa,OAAO;KACpB,cAAc,OAAO;KACrB,aAAa,OAAO;KACrB,CAAC;AAEF,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;GACxE,IAAIA,UAAyB;AAC7B,OAAI,MAAM,eAAe,QAAQ,MAAM,gBAAgB,KACrD,KAAI;AACF,cAAU,cAAc;KACtB;KACA,aAAa,MAAM;KACnB,cAAc,MAAM;KACpB,iBAAiB;KAClB,CAAC;WACI;AAKV,QAAK,SAAS;IACZ,GAAG,cAAc,WAAW;IAC5B;IACA,UAAU;IACV,iBAAiB,MAAM;IACvB,WAAW,YAAY;IACvB,aAAa,MAAM;IACnB,cAAc,MAAM;IACpB,aAAa,MAAM;IACnB,cAAc;IACd,cAAc,MAAM;IACpB,WAAW,MAAM,UAAU,SAAS,IAAI,MAAM,YAAY;IAC1D,iBAAiB,8BAA8B,OAAO;IACtD,cAAc,2BAA2B,OAAO;IAChD,aAAa,OAAO;IACpB,MAAM,OAAO;IACb,iBAAiB,OAAO;IACxB,gBAAgB,MAAM;IACtB,aAAa;IACb,SAAS,MAAM;IACf,cAAc,MAAM;IACrB,CAAC;;;;AAKR,SAAS,sBAAsB,SAA0B;AACvD,KAAI,OAAO,YAAY,SAAU,QAAO;AACxC,KAAI,WAAW,KAAM,QAAO;AAC5B,KAAI,MAAM,QAAQ,QAAQ,CACxB,QAAO,QACJ,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,GAAG;AAEb,QAAO,OAAO,QAAQ;;AAGxB,SAAS,2BACP,QACoB;CACpB,MAAM,WAAW,OAAO;AAGxB,KAAI,CAAC,MAAM,QAAQ,SAAS,CAAE,QAAO;CACrC,MAAM,gBAAgB,SAAS,MAAM,MAAM,EAAE,SAAS,SAAS;AAC/D,QAAO,OAAO,eAAe,YAAY,WACrC,cAAc,UACd;;AAGN,SAAS,8BACP,QAC4C;CAC5C,MAAM,QAAQ,OAAO;AACrB,QAAO,MAAM,QAAQ,MAAM,IAAI,MAAM,SAAS,IACzC,QACD;;AAGN,SAAS,iBAAiB,OAAiD;AACzE,QACE,SAAS,QACT,OAAQ,MAAkC,OAAO,mBAC/C"}

package/dist/providers/openai.d.ts

@@ -56,6 +56,7 @@ declare class WrappedResponses {
   private _trackInputMessages;
 }
 declare function extractSystemPrompt(params: Record<string, unknown>): string | undefined;
+declare function extractToolDefinitions(params: Record<string, unknown>): Array<Record<string, unknown>> | undefined;
 //#endregion
-export { OPENAI_AVAILABLE, OpenAI, OpenAIOptions, WrappedChat, WrappedCompletions, WrappedResponses, _OpenAIModule, extractSystemPrompt };
+export { OPENAI_AVAILABLE, OpenAI, OpenAIOptions, WrappedChat, WrappedCompletions, WrappedResponses, _OpenAIModule, extractSystemPrompt, extractToolDefinitions };
 //# sourceMappingURL=openai.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"openai.d.ts","names":[],"sources":["../../src/providers/openai.ts"],"sourcesContent":[],"mappings":";;;;;;AAoDiB,cAnBJ,gBAmBI,EAAA,OAAA;cAlBX,aAmBgB,EAnBD,MAmBC,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;AA6CN,UA5DC,aAAA,CA4DD;EAhDN,SAAA,EAXG,aAWH;EAAc,MAAA,CAAA,EAAA,MAAA;EAqDX,OAAA,CAAA,EAAA,MAAW;EACA,aAAA,CAAA,EA9DN,aA8DM,GAAA,IAAA;EAIX,gBAAA,CAAA,EAAA,OAAA;EACE;EACI,YAAA,CAAA,EAAA,OAAA;;AAeN,cA7EA,MA6EkB,CAAA,gBA5Eb,MA4Ea,CAAA,MAAA,EAAA,OAAA,CAAA,GA5Ea,MA4Eb,CAAA,MAAA,EAAA,OAAA,CAAA,CAAA,SA3ErB,cAAA,CA2EqB;EAClB,QAAA,OAAA;EAQI,SAAA,IAAA,EAlFA,WAkFA;EACJ,SAAA,SAAA,EAlFS,gBAkFT;EACE,QAAA,iBAAA;EACI,WAAA,CAAA,OAAA,EAjFI,aAiFJ;EAgBP,IAAA,MAAA,CAAA,CAAA,EAvDI,OAuDJ;;AAEC,cApDA,WAAA,CAoDA;EAAyB,SAAA,WAAA,EAnDd,kBAmDc;EAAjC,WAAA,CAAA,MAAA,EAAA,OAAA,EAAA,OAAA,EA/CQ,OA+CR,EAAA,SAAA,EA9CU,aA8CV,EAAA,aAAA,EA7Cc,aA6Cd,GAAA,IAAA,EAAA,gBAAA,EAAA,OAAA;;AA0GoB,cAxIZ,kBAAA,CAwIY;EACZ,SAAA,EAxIA,MAwIA,CAAA,MAAA,EAAA,OAAA,CAAA;EAAyB,QAAA,QAAA;EAAjC,QAAA,UAAA;EAAO,QAAA,cAAA;EAgMC,QAAA,iBAAgB;EAChB,QAAA,aAAA;EASA,WAAA,CAAA,WAAA,EA1UI,MA0UJ,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,OAAA,EAzUA,OAyUA,EAAA,SAAA,EAxUE,aAwUF,EAAA,aAAA,EAvUM,aAuUN,GAAA,IAAA,EAAA,gBAAA,EAAA,OAAA,EAAA,YAAA,CAAA,EAAA,MAAA;EACE,MAAA,CAAA,MAAA,EAxTH,MAwTG,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,kBAAA,CAAA,EAvTU,oBAuTV,CAAA,EAtTV,OAsTU,CAtTF,sBAsTE,GAtTuB,aAsTvB,CAAA,OAAA,CAAA,CAAA;EACI,KAAA,CAAA,MAAA,EA9MP,MA8MO,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,kBAAA,CAAA,EA7MM,oBA6MN,CAAA,EA5Md,OA4Mc,CA5MN,sBA4MM,GA5MmB,aA4MnB,CAAA,OAAA,CAAA,CAAA;EAkBP,QAAA,WAAA;EACa,QAAA,mBAAA;EACZ,QAAA,mBAAA;;AAAR,cAhCQ,gBAAA,CAgCR;EAmFO,SAAA,EAlHC,MAkHD,CAAA,MAAA,EAAA,OAAA,CAAA;EACa,QAAA,QAAA;EACZ,QAAA,UAAA;EAAR,QAAA,cAAA;EAAO,QAAA,iBAAA;EAgLI,QAAA,aAAA;wCA3RH,oBACE,8BACI;iBAkBP,8CACa,uBACpB,QAAQ,iBAAiB;iBAmFlB,8CACa,uBACpB,QAAQ;;;;;iBAgLG,mBAAA,SACN"}
+{"version":3,"file":"openai.d.ts","names":[],"sources":["../../src/providers/openai.ts"],"sourcesContent":[],"mappings":";;;;;;AAoDiB,cAnBJ,gBAmBI,EAAA,OAAA;cAlBX,aAmBgB,EAnBD,MAmBC,CAAA,MAAA,EAAA,OAAA,CAAA,GAAA,IAAA;AA6CN,UA5DC,aAAA,CA4DD;EAhDN,SAAA,EAXG,aAWH;EAAc,MAAA,CAAA,EAAA,MAAA;EAqDX,OAAA,CAAA,EAAA,MAAW;EACA,aAAA,CAAA,EA9DN,aA8DM,GAAA,IAAA;EAIX,gBAAA,CAAA,EAAA,OAAA;EACE;EACI,YAAA,CAAA,EAAA,OAAA;;AAeN,cA7EA,MA6EkB,CAAA,gBA5Eb,MA4Ea,CAAA,MAAA,EAAA,OAAA,CAAA,GA5Ea,MA4Eb,CAAA,MAAA,EAAA,OAAA,CAAA,CAAA,SA3ErB,cAAA,CA2EqB;EAClB,QAAA,OAAA;EAQI,SAAA,IAAA,EAlFA,WAkFA;EACJ,SAAA,SAAA,EAlFS,gBAkFT;EACE,QAAA,iBAAA;EACI,WAAA,CAAA,OAAA,EAjFI,aAiFJ;EAgBP,IAAA,MAAA,CAAA,CAAA,EAvDI,OAuDJ;;AAEC,cApDA,WAAA,CAoDA;EAAyB,SAAA,WAAA,EAnDd,kBAmDc;EAAjC,WAAA,CAAA,MAAA,EAAA,OAAA,EAAA,OAAA,EA/CQ,OA+CR,EAAA,SAAA,EA9CU,aA8CV,EAAA,aAAA,EA7Cc,aA6Cd,GAAA,IAAA,EAAA,gBAAA,EAAA,OAAA;;AA2GoB,cAzIZ,kBAAA,CAyIY;EACZ,SAAA,EAzIA,MAyIA,CAAA,MAAA,EAAA,OAAA,CAAA;EAAyB,QAAA,QAAA;EAAjC,QAAA,UAAA;EAAO,QAAA,cAAA;EA4MC,QAAA,iBAAgB;EAChB,QAAA,aAAA;EASA,WAAA,CAAA,WAAA,EAvVI,MAuVJ,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,OAAA,EAtVA,OAsVA,EAAA,SAAA,EArVE,aAqVF,EAAA,aAAA,EApVM,aAoVN,GAAA,IAAA,EAAA,gBAAA,EAAA,OAAA,EAAA,YAAA,CAAA,EAAA,MAAA;EACE,MAAA,CAAA,MAAA,EArUH,MAqUG,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,kBAAA,CAAA,EApUU,oBAoUV,CAAA,EAnUV,OAmUU,CAnUF,sBAmUE,GAnUuB,aAmUvB,CAAA,OAAA,CAAA,CAAA;EACI,KAAA,CAAA,MAAA,EA1NP,MA0NO,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,kBAAA,CAAA,EAzNM,oBAyNN,CAAA,EAxNd,OAwNc,CAxNN,sBAwNM,GAxNmB,aAwNnB,CAAA,OAAA,CAAA,CAAA;EAkBP,QAAA,WAAA;EACa,QAAA,mBAAA;EACZ,QAAA,mBAAA;;AAAR,cAhCQ,gBAAA,CAgCR;EAoFO,SAAA,EAnHC,MAmHD,CAAA,MAAA,EAAA,OAAA,CAAA;EACa,QAAA,QAAA;EACZ,QAAA,UAAA;EAAR,QAAA,cAAA;EAAO,QAAA,iBAAA;EAiLI,QAAA,aAAA;EAoEA,WAAA,CAAA,MAAA,EAAA,OAAsB,EAAA,OAAA,EAjWzB,OAiWyB,EAAA,SAAA,EAhWvB,aAgWuB,EAAA,aAAA,EA/VnB,aA+VmB,GAAA,IAAA,EAAA,gBAAA,EAAA,OAAA,EAAA,YAAA,CAAA,EAAA,MAAA;EAC5B,MAAA,CAAA,MAAA,EA9UE,MA8UF,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,kBAAA,CAAA,EA7Ue,oBA6Uf,CAAA,EA5UL,OA4UK,CA5UG,cA4UH,GA5UoB,aA4UpB,CAAA,OAAA,CAAA,CAAA;EACD,MAAA,CAAA,MAAA,EAzPG,MAyPH,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,kBAAA,CAAA,EAxPgB,oBAwPhB,CAAA,EAvPJ,OAuPI,CAvPI,aAuPJ,CAAA,OAAA,CAAA,CAAA;EAAN,QAAA,WAAA;EAAK,QAAA,mBAAA;;;iBAtEQ,mBAAA,SACN;iBAmEM,sBAAA,SACN,0BACP,MAAM"}

package/dist/providers/openai.js

@@ -106,6 +106,7 @@ var WrappedCompletions = class {
 				finishReason: choice?.finish_reason,
 				toolCalls: toolCalls ?? void 0,
 				isStreaming: false,
+				toolDefinitions: extractToolDefinitions(requestParams),
 				systemPrompt: extractSystemPrompt(requestParams),
 				temperature: requestParams.temperature,
 				maxOutputTokens: requestParams.max_tokens,
@@ -216,6 +217,7 @@ var WrappedCompletions = class {
 				isError: state.isError,
 				errorMessage: state.errorMessage,
 				reasoningContent: reasoningContent || void 0,
+				toolDefinitions: extractToolDefinitions(params),
 				systemPrompt: extractSystemPrompt(params),
 				temperature: params.temperature,
 				maxOutputTokens: params.max_tokens,
@@ -235,7 +237,10 @@ var WrappedCompletions = class {
 		if (!shouldTrackInputMessages) return;
 		if (ctx.userId == null || ctx.sessionId == null) return;
 		if (!Array.isArray(messages)) return;
-		for (const msg of messages) {
+		const msgs = messages;
+		const lastReplyIdx = msgs.findLastIndex((m) => m?.role === "assistant" || m?.role === "tool");
+		const newMessages = msgs.slice(lastReplyIdx + 1);
+		for (const msg of newMessages) {
 			if (msg?.role !== "user") continue;
 			const content = msg.content;
 			if (typeof content !== "string" || content.length === 0) continue;
@@ -314,6 +319,7 @@ var WrappedResponses = class {
 				finishReason: extractResponsesFinishReason(resp),
 				toolCalls: responseToolCalls.length > 0 ? responseToolCalls : void 0,
 				isStreaming: false,
+				toolDefinitions: extractResponsesToolDefinitions(requestParams),
 				systemPrompt: extractResponsesSystemPrompt(requestParams),
 				temperature: requestParams.temperature,
 				maxOutputTokens: requestParams.max_output_tokens,
@@ -418,6 +424,7 @@ var WrappedResponses = class {
 				isStreaming: true,
 				isError: state.isError,
 				errorMessage: state.errorMessage,
+				toolDefinitions: extractResponsesToolDefinitions(params),
 				systemPrompt: extractResponsesSystemPrompt(params),
 				temperature: params.temperature,
 				maxOutputTokens: params.max_output_tokens,
@@ -500,11 +507,25 @@ function extractOutputItemText(item) {
 	for (const c of item.content) if (typeof c?.text === "string") text += c.text;
 	return text;
 }
+function extractToolDefinitions(params) {
+	const tools = params.tools;
+	return Array.isArray(tools) && tools.length > 0 ? tools : void 0;
+}
+function extractResponsesToolDefinitions(params) {
+	const tools = params.tools;
+	return Array.isArray(tools) && tools.length > 0 ? tools : void 0;
+}
 function extractResponsesUserInputs(input) {
 	if (typeof input === "string") return [input];
 	if (!Array.isArray(input)) return [];
+	const entries = input;
+	const lastReplyIdx = entries.findLastIndex((e) => {
+		if (typeof e === "string") return false;
+		return e.role === "assistant" || e.type === "function_call" || e.type === "function_call_output";
+	});
+	const newEntries = entries.slice(lastReplyIdx + 1);
 	const result = [];
-	for (const entry of input) {
+	for (const entry of newEntries) {
 		if (typeof entry === "string") {
 			result.push(entry);
 			continue;
@@ -520,5 +541,5 @@ function extractResponsesUserInputs(input) {
 }
 
 //#endregion
-export { OPENAI_AVAILABLE, OpenAI, WrappedChat, WrappedCompletions, WrappedResponses, _OpenAIModule, extractSystemPrompt };
+export { OPENAI_AVAILABLE, OpenAI, WrappedChat, WrappedCompletions, WrappedResponses, _OpenAIModule, extractSystemPrompt, extractToolDefinitions };
 //# sourceMappingURL=openai.js.map