Function reacquireHostedKey

apps/sim/tools/index.ts:330–362

* Re-acquire a hosted key after upstream-429 retries have been exhausted. Calls * `acquireKey` (which now blocks on the per-workspace bucket) and re-injects the * fresh key into `params`. Returns false if no key could be obtained — caller * should re-throw the original upstream 429. * * Does no

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

 * branch of `executeTool`, so BYOK has already been ruled out for this call.
 */
async function reacquireHostedKey(
  tool: ToolConfig,
  params: Record<string, unknown>,
  executionContext: ExecutionContext | undefined,
  requestId: string
): Promise<string | null> {
  if (!tool.hosting) return null
  const { envKeyPrefix, apiKeyParam, byokProviderId, rateLimit } = tool.hosting
  const { workspaceId } = resolveToolScope(params, executionContext)
  if (!workspaceId) return null

  const provider = byokProviderId || tool.id
  const acquireResult = await getHostedKeyRateLimiter().acquireKey(
    provider,
    envKeyPrefix,
    rateLimit,
    workspaceId,
    executionContext?.abortSignal
  )

  if (!acquireResult.success || !acquireResult.key) {
    logger.warn(
      `[${requestId}] Re-acquire of hosted key for ${tool.id} failed: ${acquireResult.error ?? 'unknown'}`
    )
    return null
  }

  params[apiKeyParam] = acquireResult.key
  logger.info(
    `[${requestId}] Re-acquired hosted key for ${tool.id} (${acquireResult.envVarName}) after upstream throttling`
  )
  return acquireResult.envVarName ?? 'unknown'
}

/**
 * Check if an error is a rate limit (throttling) or quota exhaustion error.