UNPKG

@ai-sdk/amazon-bedrock

Version:

The **[Amazon Bedrock provider](https://ai-sdk.dev/providers/ai-sdk-providers/amazon-bedrock)** for the [AI SDK](https://ai-sdk.dev/docs) contains language model support for the Amazon Bedrock [converse API](https://docs.aws.amazon.com/bedrock/latest/APIR

51 lines (46 loc) 1.23 kB
import type { LanguageModelV4Usage } from '@ai-sdk/provider'; export type AmazonBedrockUsage = { inputTokens: number; outputTokens: number; totalTokens?: number; cacheReadInputTokens?: number | null; cacheWriteInputTokens?: number | null; }; export function convertAmazonBedrockUsage( usage: AmazonBedrockUsage | undefined | null, ): LanguageModelV4Usage { if (usage == null) { return { inputTokens: { total: undefined, noCache: undefined, cacheRead: undefined, cacheWrite: undefined, }, outputTokens: { total: undefined, text: undefined, reasoning: undefined, }, raw: undefined, }; } const inputTokens = usage.inputTokens; const outputTokens = usage.outputTokens; const cacheReadTokens = usage.cacheReadInputTokens ?? 0; const cacheWriteTokens = usage.cacheWriteInputTokens ?? 0; return { inputTokens: { total: inputTokens + cacheReadTokens + cacheWriteTokens, noCache: inputTokens, cacheRead: cacheReadTokens, cacheWrite: cacheWriteTokens, }, outputTokens: { total: outputTokens, text: outputTokens, reasoning: undefined, }, raw: usage, }; }