Function applyHostedKeyCostToResult

apps/sim/tools/index.ts:639–671

* Apply post-execution hosted-key cost tracking to a successful tool result. * Reports custom dimension usage, calculates cost, and merges it into the output.

(
  finalResult: ToolResponse,
  tool: ToolConfig,
  params: Record<string, unknown>,
  executionContext: ExecutionContext | undefined,
  requestId: string,
  envVarName: string | undefined
)

 * Reports custom dimension usage, calculates cost, and merges it into the output.
 */
async function applyHostedKeyCostToResult(
  finalResult: ToolResponse,
  tool: ToolConfig,
  params: Record<string, unknown>,
  executionContext: ExecutionContext | undefined,
  requestId: string,
  envVarName: string | undefined
): Promise<void> {
  await reportCustomDimensionUsage(tool, params, finalResult.output, executionContext, requestId)

  const { cost: hostedKeyCost, metadata } = await processHostedKeyCost(
    tool,
    params,
    finalResult.output,
    executionContext,
    requestId
  )

  const provider = tool.hosting?.byokProviderId || tool.id
  const key = envVarName ?? 'unknown'
  hostedKeyMetrics.recordUsed({ provider, tool: tool.id, key })
  hostedKeyMetrics.recordCostCharged(hostedKeyCost, { provider, tool: tool.id })

  if (hostedKeyCost > 0) {
    finalResult.output = {
      ...finalResult.output,
      cost: {
        ...metadata,
        total: hostedKeyCost,
      },
    }
  }
}

import { normalizeToolId } from '@/tools/normalize'