ai-sdk-cost-calculator 0.1.0 → 0.1.1

package/README.md

@@ -37,8 +37,7 @@ const result = await generateText({
 });
 
 const cost = calculateCost({
-  provider: "openai",
-  modelId: "gpt-4o",
+  model: "gpt-4o",
   usage: result.usage,
 });
 
@@ -56,16 +55,17 @@ Calculate the cost for a single API request.
 import { calculateCost } from "ai-sdk-cost-calculator";
 
 const cost = calculateCost({
-  provider: "anthropic",
-  modelId: "claude-sonnet-4-5",
+  model: "claude-sonnet-4-5",
   usage: {
     inputTokens: 1000,
     outputTokens: 500,
     inputTokenDetails: {
+      noCacheTokens: 700,
       cacheReadTokens: 200,
       cacheWriteTokens: 100,
     },
     outputTokenDetails: {
+      textTokens: 500,
       reasoningTokens: 0,
     },
   },
@@ -86,25 +86,24 @@ console.log(cost);
 
 #### Options
 
-| Property | Type | Required | Description |
-|----------|------|----------|-------------|
-| `provider` | `string` | Yes | Provider name: `"openai"`, `"anthropic"`, `"google"`, `"xai"`, `"deepseek"` |
-| `modelId` | `string` | Yes | Model identifier (e.g., `"gpt-4o"`, `"claude-sonnet-4-5"`) |
-| `usage` | `LanguageModelUsage` | Yes | Token usage from AI SDK |
-| `customPricing` | `ModelPricing` | No | Override default pricing |
+| Property        | Type                 | Required | Description                                                |
+| --------------- | -------------------- | -------- | ---------------------------------------------------------- |
+| `model`         | `string`             | Yes      | Model identifier (e.g., `"gpt-4o"`, `"claude-sonnet-4-5"`) |
+| `usage`         | `LanguageModelUsage` | Yes      | Token usage from AI SDK                                    |
+| `customPricing` | `ModelPricing`       | No       | Override default pricing                                   |
 
 #### Return Value: `CostBreakdown`
 
-| Property | Type | Description |
-|----------|------|-------------|
-| `inputCost` | `number` | Cost for input tokens (excluding cache) |
-| `outputCost` | `number` | Cost for output tokens (excluding reasoning) |
-| `cacheReadCost` | `number` | Cost for cached input tokens |
-| `cacheWriteCost` | `number` | Cost for writing to cache |
-| `reasoningCost` | `number` | Cost for reasoning tokens |
-| `totalCost` | `number` | Sum of all costs |
-| `currency` | `"USD"` | Always USD |
-| `isLongContext` | `boolean` | Whether long context pricing was applied |
+| Property         | Type      | Description                                  |
+| ---------------- | --------- | -------------------------------------------- |
+| `inputCost`      | `number`  | Cost for input tokens (excluding cache)      |
+| `outputCost`     | `number`  | Cost for output tokens (excluding reasoning) |
+| `cacheReadCost`  | `number`  | Cost for cached input tokens                 |
+| `cacheWriteCost` | `number`  | Cost for writing to cache                    |
+| `reasoningCost`  | `number`  | Cost for reasoning tokens                    |
+| `totalCost`      | `number`  | Sum of all costs                             |
+| `currency`       | `"USD"`   | Always USD                                   |
+| `isLongContext`  | `boolean` | Whether long context pricing was applied     |
 
 ### `formatCost(cost, decimals?)`
 
@@ -113,8 +112,8 @@ Format a cost value as a currency string.
 ```typescript
 import { formatCost } from "ai-sdk-cost-calculator";
 
-formatCost(0.0015);      // "$0.001500"
-formatCost(0.0015, 4);   // "$0.0015"
+formatCost(0.0015); // "$0.001500"
+formatCost(0.0015, 4); // "$0.0015"
 ```
 
 ### `formatCostBreakdown(breakdown, decimals?)`
@@ -140,8 +139,7 @@ Track cumulative costs across multiple requests.
 import { createCostTracker } from "ai-sdk-cost-calculator";
 
 const tracker = createCostTracker({
-  provider: "openai",
-  modelId: "gpt-4o",
+  model: "gpt-4o",
 });
 
 // Add usage from multiple requests
@@ -171,8 +169,7 @@ const stream = await streamText({
 });
 
 const cost = await calculateStreamCost(stream, {
-  provider: "openai",
-  modelId: "gpt-4o",
+  model: "gpt-4o",
   onCost: (cost, usage) => {
     console.log(`Stream completed: ${formatCost(cost.totalCost)}`);
   },
@@ -220,6 +217,7 @@ const models = getAllSupportedModels();
 ## Supported Models
 
 ### OpenAI (23 models)
+
 - GPT-4o, GPT-4o-mini
 - GPT-4.1, GPT-4.1-mini, GPT-4.1-nano
 - GPT-4 Turbo, GPT-4, GPT-4-32k
@@ -229,6 +227,7 @@ const models = getAllSupportedModels();
 - o4-mini
 
 ### Anthropic (23 models)
+
 - Claude Opus 4.5, Sonnet 4.5, Haiku 4.5
 - Claude Opus 4.1, Sonnet 4, Opus 4
 - Claude 3.5 Sonnet, 3.5 Haiku
@@ -236,12 +235,14 @@ const models = getAllSupportedModels();
 - All dated variants (e.g., `claude-3-5-sonnet-20241022`)
 
 ### Google Gemini (17 models)
+
 - Gemini 3 Pro, 3 Flash
 - Gemini 2.5 Pro, 2.5 Flash, 2.5 Flash-Lite
 - Gemini 2.0 Flash, 2.0 Flash-Lite
 - Gemini 1.5 Pro, 1.5 Flash, 1.5 Flash-8B
 
 ### xAI Grok (12 models)
+
 - Grok 4, Grok 4-0709
 - Grok 4 Fast (reasoning/non-reasoning)
 - Grok 4.1 Fast (reasoning/non-reasoning)
@@ -250,6 +251,7 @@ const models = getAllSupportedModels();
 - Grok 2, Grok 2 Vision
 
 ### DeepSeek (10 models)
+
 - DeepSeek Chat (V3, V3.2)
 - DeepSeek Reasoner (R1)
 - DeepSeek R1 Distill variants
@@ -259,12 +261,12 @@ const models = getAllSupportedModels();
 
 Some models have tiered pricing based on input token count:
 
-| Model | Threshold | Standard | Long Context |
-|-------|-----------|----------|--------------|
-| Claude Sonnet 4.5 | 200K | $3/$15 | $6/$22.50 |
-| Gemini 2.5 Pro | 200K | $1.25/$10 | $2.50/$15 |
-| Gemini 1.5 Pro | 128K | $1.25/$5 | $2.50/$10 |
-| Grok 4 Fast | 128K | $0.20/$0.50 | $0.40/$1.00 |
+| Model             | Threshold | Standard    | Long Context |
+| ----------------- | --------- | ----------- | ------------ |
+| Claude Sonnet 4.5 | 200K      | $3/$15      | $6/$22.50    |
+| Gemini 2.5 Pro    | 200K      | $1.25/$10   | $2.50/$15    |
+| Gemini 1.5 Pro    | 128K      | $1.25/$5    | $2.50/$10    |
+| Grok 4 Fast       | 128K      | $0.20/$0.50 | $0.40/$1.00  |
 
 The calculator automatically applies the correct pricing tier based on input token count.
 
@@ -274,11 +276,10 @@ Override default pricing for any model:
 
 ```typescript
 const cost = calculateCost({
-  provider: "openai",
-  modelId: "gpt-4o",
+  model: "gpt-4o",
   usage: result.usage,
   customPricing: {
-    inputPer1MTokens: 2.0,  // Custom rate
+    inputPer1MTokens: 2.0, // Custom rate
     outputPer1MTokens: 8.0,
     cacheReadPer1MTokens: 1.0,
   },

package/dist/calculator.d.ts

@@ -11,8 +11,7 @@ export interface CostBreakdown {
     isLongContext: boolean;
 }
 export interface CalculateCostOptions {
-    provider: string;
-    modelId: string;
+    model: string;
     usage: LanguageModelUsage;
     customPricing?: ModelPricing;
 }

package/dist/calculator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"calculator.d.ts","sourceRoot":"","sources":["../src/calculator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,EAAmB,KAAK,YAAY,EAAE,MAAM,UAAU,CAAC;AAE9D,MAAM,WAAW,aAAa;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;IACtB,cAAc,EAAE,MAAM,CAAC;IACvB,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,KAAK,CAAC;IAChB,aAAa,EAAE,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,oBAAoB;IACnC,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,kBAAkB,CAAC;IAC1B,aAAa,CAAC,EAAE,YAAY,CAAC;CAC9B;AAqDD,wBAAgB,aAAa,CAAC,OAAO,EAAE,oBAAoB,GAAG,aAAa,CAwD1E;AAMD,wBAAgB,UAAU,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,GAAE,MAAU,GAAG,MAAM,CAErE;AAED,wBAAgB,mBAAmB,CACjC,MAAM,EAAE,aAAa,EACrB,QAAQ,GAAE,MAAU,GACnB,MAAM,CAwBR"}
+{"version":3,"file":"calculator.d.ts","sourceRoot":"","sources":["../src/calculator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,EAA4B,KAAK,YAAY,EAAE,MAAM,UAAU,CAAC;AAIvE,MAAM,WAAW,aAAa;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;IACtB,cAAc,EAAE,MAAM,CAAC;IACvB,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,KAAK,CAAC;IAChB,aAAa,EAAE,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,oBAAoB;IACnC,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,kBAAkB,CAAC;IAC1B,aAAa,CAAC,EAAE,YAAY,CAAC;CAC9B;AAkGD,wBAAgB,aAAa,CAAC,OAAO,EAAE,oBAAoB,GAAG,aAAa,CAyD1E;AAMD,wBAAgB,UAAU,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,GAAE,MAAU,GAAG,MAAM,CAErE;AAED,wBAAgB,mBAAmB,CACjC,MAAM,EAAE,aAAa,EACrB,QAAQ,GAAE,MAAU,GACnB,MAAM,CAwBR"}

package/dist/calculator.js

@@ -1,4 +1,5 @@
-import { getModelPricing } from "./prices";
+import { getModelPricingByModelId } from "./prices";
+const TOKENS_PER_MILLION = 1_000_000;
 function getEffectivePricing(pricing, inputTokens) {
     const threshold = pricing.longContextThreshold ?? 200000;
     const isLongContext = inputTokens > threshold &&
@@ -28,32 +29,48 @@ function getEffectivePricing(pricing, inputTokens) {
         isLongContext: false,
     };
 }
+function getUsageTokenDetails(usage) {
+    const inputDetails = usage.inputTokenDetails;
+    const outputDetails = usage.outputTokenDetails;
+    let noCacheTokens = inputDetails?.noCacheTokens ?? 0;
+    let cacheReadTokens = inputDetails?.cacheReadTokens ?? 0;
+    let cacheWriteTokens = inputDetails?.cacheWriteTokens ?? 0;
+    let textTokens = outputDetails?.textTokens ?? 0;
+    let reasoningTokens = outputDetails?.reasoningTokens ?? 0;
+    if (!inputDetails && usage.inputTokens !== undefined) {
+        noCacheTokens = usage.inputTokens;
+    }
+    if (!outputDetails && usage.outputTokens !== undefined) {
+        textTokens = usage.outputTokens;
+    }
+    const totalInputTokens = usage.inputTokens ?? noCacheTokens + cacheReadTokens + cacheWriteTokens;
+    return {
+        noCacheTokens,
+        cacheReadTokens,
+        cacheWriteTokens,
+        textTokens,
+        reasoningTokens,
+        totalInputTokens,
+    };
+}
+function costFromTokens(tokens, per1MTokens) {
+    return (tokens / TOKENS_PER_MILLION) * per1MTokens;
+}
 export function calculateCost(options) {
-    const { provider, modelId, usage, customPricing } = options;
-    const pricing = customPricing ?? getModelPricing(provider, modelId);
+    const { model, usage, customPricing } = options;
+    const pricing = customPricing ?? getModelPricingByModelId(model);
     if (!pricing) {
-        throw new Error(`Unknown model: ${provider}/${modelId}. Use customPricing option or add the model to prices.ts`);
+        throw new Error(`Unknown model: ${model}. Use customPricing option or add the model to prices.ts`);
     }
-    // Extract token counts from details
-    const noCacheTokens = usage.inputTokenDetails?.noCacheTokens ?? 0;
-    const cacheReadTokens = usage.inputTokenDetails?.cacheReadTokens ?? 0;
-    const cacheWriteTokens = usage.inputTokenDetails?.cacheWriteTokens ?? 0;
-    const textTokens = usage.outputTokenDetails?.textTokens ?? 0;
-    const reasoningTokens = usage.outputTokenDetails?.reasoningTokens ?? 0;
-    // Total input tokens for long context threshold check
-    const totalInputTokens = usage.inputTokens ?? (noCacheTokens + cacheReadTokens);
+    const usageDetails = getUsageTokenDetails(usage);
     // Get effective pricing based on input token count
-    const effectivePricing = getEffectivePricing(pricing, totalInputTokens);
-    const inputCost = (noCacheTokens / 1_000_000) * effectivePricing.inputPer1MTokens;
-    const outputCost = (textTokens / 1_000_000) * effectivePricing.outputPer1MTokens;
-    const cacheReadCost = (cacheReadTokens / 1_000_000) * effectivePricing.cacheReadPer1MTokens;
-    const cacheWriteCost = (cacheWriteTokens / 1_000_000) * effectivePricing.cacheWritePer1MTokens;
-    const reasoningCost = (reasoningTokens / 1_000_000) * effectivePricing.reasoningPer1MTokens;
-    const totalCost = inputCost +
-        outputCost +
-        cacheReadCost +
-        cacheWriteCost +
-        reasoningCost;
+    const effectivePricing = getEffectivePricing(pricing, usageDetails.totalInputTokens);
+    const inputCost = costFromTokens(usageDetails.noCacheTokens, effectivePricing.inputPer1MTokens);
+    const outputCost = costFromTokens(usageDetails.textTokens, effectivePricing.outputPer1MTokens);
+    const cacheReadCost = costFromTokens(usageDetails.cacheReadTokens, effectivePricing.cacheReadPer1MTokens);
+    const cacheWriteCost = costFromTokens(usageDetails.cacheWriteTokens, effectivePricing.cacheWritePer1MTokens);
+    const reasoningCost = costFromTokens(usageDetails.reasoningTokens, effectivePricing.reasoningPer1MTokens);
+    const totalCost = inputCost + outputCost + cacheReadCost + cacheWriteCost + reasoningCost;
     return {
         inputCost: roundToMicroDollars(inputCost),
         outputCost: roundToMicroDollars(outputCost),

package/dist/index.d.mts

@@ -47,8 +47,7 @@ interface CostBreakdown {
     isLongContext: boolean;
 }
 interface CalculateCostOptions {
-    provider: string;
-    modelId: string;
+    model: string;
     usage: LanguageModelUsage;
     customPricing?: ModelPricing;
 }
@@ -64,14 +63,12 @@ interface CostTracker {
     reset(): void;
 }
 interface CreateCostTrackerOptions {
-    provider: string;
-    modelId: string;
+    model: string;
     customPricing?: ModelPricing;
 }
 declare function createCostTracker(options: CreateCostTrackerOptions): CostTracker;
 interface StreamCostOptions {
-    provider: string;
-    modelId: string;
+    model: string;
     customPricing?: ModelPricing;
     onCost?: (cost: CostBreakdown, usage: LanguageModelUsage) => void;
 }

package/dist/index.d.ts

@@ -47,8 +47,7 @@ interface CostBreakdown {
     isLongContext: boolean;
 }
 interface CalculateCostOptions {
-    provider: string;
-    modelId: string;
+    model: string;
     usage: LanguageModelUsage;
     customPricing?: ModelPricing;
 }
@@ -64,14 +63,12 @@ interface CostTracker {
     reset(): void;
 }
 interface CreateCostTrackerOptions {
-    provider: string;
-    modelId: string;
+    model: string;
     customPricing?: ModelPricing;
 }
 declare function createCostTracker(options: CreateCostTrackerOptions): CostTracker;
 interface StreamCostOptions {
-    provider: string;
-    modelId: string;
+    model: string;
     customPricing?: ModelPricing;
     onCost?: (cost: CostBreakdown, usage: LanguageModelUsage) => void;
 }

package/dist/index.js

@@ -629,6 +629,7 @@ function getAllSupportedModels() {
 }
 
 // src/calculator.ts
+var TOKENS_PER_MILLION = 1e6;
 function getEffectivePricing(pricing, inputTokens) {
   const threshold = pricing.longContextThreshold ?? 2e5;
   const isLongContext = inputTokens > threshold && pricing.longContextInputPer1MTokens !== void 0;
@@ -651,26 +652,66 @@ function getEffectivePricing(pricing, inputTokens) {
     isLongContext: false
   };
 }
+function getUsageTokenDetails(usage) {
+  const inputDetails = usage.inputTokenDetails;
+  const outputDetails = usage.outputTokenDetails;
+  let noCacheTokens = inputDetails?.noCacheTokens ?? 0;
+  let cacheReadTokens = inputDetails?.cacheReadTokens ?? 0;
+  let cacheWriteTokens = inputDetails?.cacheWriteTokens ?? 0;
+  let textTokens = outputDetails?.textTokens ?? 0;
+  let reasoningTokens = outputDetails?.reasoningTokens ?? 0;
+  if (!inputDetails && usage.inputTokens !== void 0) {
+    noCacheTokens = usage.inputTokens;
+  }
+  if (!outputDetails && usage.outputTokens !== void 0) {
+    textTokens = usage.outputTokens;
+  }
+  const totalInputTokens = usage.inputTokens ?? noCacheTokens + cacheReadTokens + cacheWriteTokens;
+  return {
+    noCacheTokens,
+    cacheReadTokens,
+    cacheWriteTokens,
+    textTokens,
+    reasoningTokens,
+    totalInputTokens
+  };
+}
+function costFromTokens(tokens, per1MTokens) {
+  return tokens / TOKENS_PER_MILLION * per1MTokens;
+}
 function calculateCost(options) {
-  const { provider, modelId, usage, customPricing } = options;
-  const pricing = customPricing ?? getModelPricing(provider, modelId);
+  const { model, usage, customPricing } = options;
+  const pricing = customPricing ?? getModelPricingByModelId(model);
   if (!pricing) {
     throw new Error(
-      `Unknown model: ${provider}/${modelId}. Use customPricing option or add the model to prices.ts`
+      `Unknown model: ${model}. Use customPricing option or add the model to prices.ts`
     );
   }
-  const noCacheTokens = usage.inputTokenDetails?.noCacheTokens ?? 0;
-  const cacheReadTokens = usage.inputTokenDetails?.cacheReadTokens ?? 0;
-  const cacheWriteTokens = usage.inputTokenDetails?.cacheWriteTokens ?? 0;
-  const textTokens = usage.outputTokenDetails?.textTokens ?? 0;
-  const reasoningTokens = usage.outputTokenDetails?.reasoningTokens ?? 0;
-  const totalInputTokens = usage.inputTokens ?? noCacheTokens + cacheReadTokens;
-  const effectivePricing = getEffectivePricing(pricing, totalInputTokens);
-  const inputCost = noCacheTokens / 1e6 * effectivePricing.inputPer1MTokens;
-  const outputCost = textTokens / 1e6 * effectivePricing.outputPer1MTokens;
-  const cacheReadCost = cacheReadTokens / 1e6 * effectivePricing.cacheReadPer1MTokens;
-  const cacheWriteCost = cacheWriteTokens / 1e6 * effectivePricing.cacheWritePer1MTokens;
-  const reasoningCost = reasoningTokens / 1e6 * effectivePricing.reasoningPer1MTokens;
+  const usageDetails = getUsageTokenDetails(usage);
+  const effectivePricing = getEffectivePricing(
+    pricing,
+    usageDetails.totalInputTokens
+  );
+  const inputCost = costFromTokens(
+    usageDetails.noCacheTokens,
+    effectivePricing.inputPer1MTokens
+  );
+  const outputCost = costFromTokens(
+    usageDetails.textTokens,
+    effectivePricing.outputPer1MTokens
+  );
+  const cacheReadCost = costFromTokens(
+    usageDetails.cacheReadTokens,
+    effectivePricing.cacheReadPer1MTokens
+  );
+  const cacheWriteCost = costFromTokens(
+    usageDetails.cacheWriteTokens,
+    effectivePricing.cacheWritePer1MTokens
+  );
+  const reasoningCost = costFromTokens(
+    usageDetails.reasoningTokens,
+    effectivePricing.reasoningPer1MTokens
+  );
   const totalCost = inputCost + outputCost + cacheReadCost + cacheWriteCost + reasoningCost;
   return {
     inputCost: roundToMicroDollars(inputCost),
@@ -715,7 +756,7 @@ function formatCostBreakdown(result, decimals = 6) {
 
 // src/streaming.ts
 function createCostTracker(options) {
-  const { provider, modelId, customPricing } = options;
+  const { model, customPricing } = options;
   let totalNoCacheTokens = 0;
   let totalCacheReadTokens = 0;
   let totalCacheWriteTokens = 0;
@@ -724,15 +765,25 @@ function createCostTracker(options) {
   let requestCount = 0;
   return {
     addUsage(usage) {
-      totalNoCacheTokens += usage.inputTokenDetails?.noCacheTokens ?? 0;
-      totalCacheReadTokens += usage.inputTokenDetails?.cacheReadTokens ?? 0;
-      totalCacheWriteTokens += usage.inputTokenDetails?.cacheWriteTokens ?? 0;
-      totalTextTokens += usage.outputTokenDetails?.textTokens ?? 0;
-      totalReasoningTokens += usage.outputTokenDetails?.reasoningTokens ?? 0;
+      const inputDetails = usage.inputTokenDetails;
+      if (inputDetails) {
+        totalNoCacheTokens += inputDetails.noCacheTokens ?? 0;
+        totalCacheReadTokens += inputDetails.cacheReadTokens ?? 0;
+        totalCacheWriteTokens += inputDetails.cacheWriteTokens ?? 0;
+      } else if (usage.inputTokens !== void 0) {
+        totalNoCacheTokens += usage.inputTokens;
+      }
+      const outputDetails = usage.outputTokenDetails;
+      if (outputDetails) {
+        totalTextTokens += outputDetails.textTokens ?? 0;
+        totalReasoningTokens += outputDetails.reasoningTokens ?? 0;
+      } else if (usage.outputTokens !== void 0) {
+        totalTextTokens += usage.outputTokens;
+      }
       requestCount++;
     },
     getTotalUsage() {
-      const inputTokens = totalNoCacheTokens + totalCacheReadTokens;
+      const inputTokens = totalNoCacheTokens + totalCacheReadTokens + totalCacheWriteTokens;
       const outputTokens = totalTextTokens + totalReasoningTokens;
       return {
         inputTokens,
@@ -751,8 +802,7 @@ function createCostTracker(options) {
     },
     getTotalCost() {
       return calculateCost({
-        provider,
-        modelId,
+        model,
         usage: this.getTotalUsage(),
         customPricing
       });
@@ -774,8 +824,7 @@ async function calculateStreamCost(streamResult, options) {
   const result = await streamResult;
   const usage = result.usage;
   const cost = calculateCost({
-    provider: options.provider,
-    modelId: options.modelId,
+    model: options.model,
     usage,
     customPricing: options.customPricing
   });

package/dist/index.mjs

@@ -589,6 +589,7 @@ function getAllSupportedModels() {
 }
 
 // src/calculator.ts
+var TOKENS_PER_MILLION = 1e6;
 function getEffectivePricing(pricing, inputTokens) {
   const threshold = pricing.longContextThreshold ?? 2e5;
   const isLongContext = inputTokens > threshold && pricing.longContextInputPer1MTokens !== void 0;
@@ -611,26 +612,66 @@ function getEffectivePricing(pricing, inputTokens) {
     isLongContext: false
   };
 }
+function getUsageTokenDetails(usage) {
+  const inputDetails = usage.inputTokenDetails;
+  const outputDetails = usage.outputTokenDetails;
+  let noCacheTokens = inputDetails?.noCacheTokens ?? 0;
+  let cacheReadTokens = inputDetails?.cacheReadTokens ?? 0;
+  let cacheWriteTokens = inputDetails?.cacheWriteTokens ?? 0;
+  let textTokens = outputDetails?.textTokens ?? 0;
+  let reasoningTokens = outputDetails?.reasoningTokens ?? 0;
+  if (!inputDetails && usage.inputTokens !== void 0) {
+    noCacheTokens = usage.inputTokens;
+  }
+  if (!outputDetails && usage.outputTokens !== void 0) {
+    textTokens = usage.outputTokens;
+  }
+  const totalInputTokens = usage.inputTokens ?? noCacheTokens + cacheReadTokens + cacheWriteTokens;
+  return {
+    noCacheTokens,
+    cacheReadTokens,
+    cacheWriteTokens,
+    textTokens,
+    reasoningTokens,
+    totalInputTokens
+  };
+}
+function costFromTokens(tokens, per1MTokens) {
+  return tokens / TOKENS_PER_MILLION * per1MTokens;
+}
 function calculateCost(options) {
-  const { provider, modelId, usage, customPricing } = options;
-  const pricing = customPricing ?? getModelPricing(provider, modelId);
+  const { model, usage, customPricing } = options;
+  const pricing = customPricing ?? getModelPricingByModelId(model);
   if (!pricing) {
     throw new Error(
-      `Unknown model: ${provider}/${modelId}. Use customPricing option or add the model to prices.ts`
+      `Unknown model: ${model}. Use customPricing option or add the model to prices.ts`
     );
   }
-  const noCacheTokens = usage.inputTokenDetails?.noCacheTokens ?? 0;
-  const cacheReadTokens = usage.inputTokenDetails?.cacheReadTokens ?? 0;
-  const cacheWriteTokens = usage.inputTokenDetails?.cacheWriteTokens ?? 0;
-  const textTokens = usage.outputTokenDetails?.textTokens ?? 0;
-  const reasoningTokens = usage.outputTokenDetails?.reasoningTokens ?? 0;
-  const totalInputTokens = usage.inputTokens ?? noCacheTokens + cacheReadTokens;
-  const effectivePricing = getEffectivePricing(pricing, totalInputTokens);
-  const inputCost = noCacheTokens / 1e6 * effectivePricing.inputPer1MTokens;
-  const outputCost = textTokens / 1e6 * effectivePricing.outputPer1MTokens;
-  const cacheReadCost = cacheReadTokens / 1e6 * effectivePricing.cacheReadPer1MTokens;
-  const cacheWriteCost = cacheWriteTokens / 1e6 * effectivePricing.cacheWritePer1MTokens;
-  const reasoningCost = reasoningTokens / 1e6 * effectivePricing.reasoningPer1MTokens;
+  const usageDetails = getUsageTokenDetails(usage);
+  const effectivePricing = getEffectivePricing(
+    pricing,
+    usageDetails.totalInputTokens
+  );
+  const inputCost = costFromTokens(
+    usageDetails.noCacheTokens,
+    effectivePricing.inputPer1MTokens
+  );
+  const outputCost = costFromTokens(
+    usageDetails.textTokens,
+    effectivePricing.outputPer1MTokens
+  );
+  const cacheReadCost = costFromTokens(
+    usageDetails.cacheReadTokens,
+    effectivePricing.cacheReadPer1MTokens
+  );
+  const cacheWriteCost = costFromTokens(
+    usageDetails.cacheWriteTokens,
+    effectivePricing.cacheWritePer1MTokens
+  );
+  const reasoningCost = costFromTokens(
+    usageDetails.reasoningTokens,
+    effectivePricing.reasoningPer1MTokens
+  );
   const totalCost = inputCost + outputCost + cacheReadCost + cacheWriteCost + reasoningCost;
   return {
     inputCost: roundToMicroDollars(inputCost),
@@ -675,7 +716,7 @@ function formatCostBreakdown(result, decimals = 6) {
 
 // src/streaming.ts
 function createCostTracker(options) {
-  const { provider, modelId, customPricing } = options;
+  const { model, customPricing } = options;
   let totalNoCacheTokens = 0;
   let totalCacheReadTokens = 0;
   let totalCacheWriteTokens = 0;
@@ -684,15 +725,25 @@ function createCostTracker(options) {
   let requestCount = 0;
   return {
     addUsage(usage) {
-      totalNoCacheTokens += usage.inputTokenDetails?.noCacheTokens ?? 0;
-      totalCacheReadTokens += usage.inputTokenDetails?.cacheReadTokens ?? 0;
-      totalCacheWriteTokens += usage.inputTokenDetails?.cacheWriteTokens ?? 0;
-      totalTextTokens += usage.outputTokenDetails?.textTokens ?? 0;
-      totalReasoningTokens += usage.outputTokenDetails?.reasoningTokens ?? 0;
+      const inputDetails = usage.inputTokenDetails;
+      if (inputDetails) {
+        totalNoCacheTokens += inputDetails.noCacheTokens ?? 0;
+        totalCacheReadTokens += inputDetails.cacheReadTokens ?? 0;
+        totalCacheWriteTokens += inputDetails.cacheWriteTokens ?? 0;
+      } else if (usage.inputTokens !== void 0) {
+        totalNoCacheTokens += usage.inputTokens;
+      }
+      const outputDetails = usage.outputTokenDetails;
+      if (outputDetails) {
+        totalTextTokens += outputDetails.textTokens ?? 0;
+        totalReasoningTokens += outputDetails.reasoningTokens ?? 0;
+      } else if (usage.outputTokens !== void 0) {
+        totalTextTokens += usage.outputTokens;
+      }
       requestCount++;
     },
     getTotalUsage() {
-      const inputTokens = totalNoCacheTokens + totalCacheReadTokens;
+      const inputTokens = totalNoCacheTokens + totalCacheReadTokens + totalCacheWriteTokens;
       const outputTokens = totalTextTokens + totalReasoningTokens;
       return {
         inputTokens,
@@ -711,8 +762,7 @@ function createCostTracker(options) {
     },
     getTotalCost() {
       return calculateCost({
-        provider,
-        modelId,
+        model,
         usage: this.getTotalUsage(),
         customPricing
       });
@@ -734,8 +784,7 @@ async function calculateStreamCost(streamResult, options) {
   const result = await streamResult;
   const usage = result.usage;
   const cost = calculateCost({
-    provider: options.provider,
-    modelId: options.modelId,
+    model: options.model,
     usage,
     customPricing: options.customPricing
   });

package/dist/prices/index.d.ts

@@ -1,17 +1,2 @@
-export type { ModelPricing, ProviderPricing } from "./types";
-export { openaiPricing } from "./openai";
-export { anthropicPricing } from "./anthropic";
-export { googlePricing } from "./google";
-export { xaiPricing } from "./xai";
-export { deepseekPricing } from "./deepseek";
-import type { ProviderPricing } from "./types";
-export declare const allPricing: ProviderPricing;
-export declare const pricingByProvider: {
-    readonly openai: ProviderPricing;
-    readonly anthropic: ProviderPricing;
-    readonly google: ProviderPricing;
-    readonly xai: ProviderPricing;
-    readonly deepseek: ProviderPricing;
-};
-export type Provider = keyof typeof pricingByProvider;
+export * from "../prices";
 //# sourceMappingURL=index.d.ts.map

package/dist/prices/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/prices/index.ts"],"names":[],"mappings":"AAAA,YAAY,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAE7D,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,UAAU,EAAE,MAAM,OAAO,CAAC;AACnC,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAO7C,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAG/C,eAAO,MAAM,UAAU,EAAE,eAMxB,CAAC;AAGF,eAAO,MAAM,iBAAiB;;;;;;CAMpB,CAAC;AAEX,MAAM,MAAM,QAAQ,GAAG,MAAM,OAAO,iBAAiB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/prices/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAC"}

package/dist/prices/index.js

@@ -1,26 +1 @@
-export { openaiPricing } from "./openai";
-export { anthropicPricing } from "./anthropic";
-export { googlePricing } from "./google";
-export { xaiPricing } from "./xai";
-export { deepseekPricing } from "./deepseek";
-import { openaiPricing } from "./openai";
-import { anthropicPricing } from "./anthropic";
-import { googlePricing } from "./google";
-import { xaiPricing } from "./xai";
-import { deepseekPricing } from "./deepseek";
-// Combined pricing for all providers
-export const allPricing = {
-    ...openaiPricing,
-    ...anthropicPricing,
-    ...googlePricing,
-    ...xaiPricing,
-    ...deepseekPricing,
-};
-// Provider name to pricing map
-export const pricingByProvider = {
-    openai: openaiPricing,
-    anthropic: anthropicPricing,
-    google: googlePricing,
-    xai: xaiPricing,
-    deepseek: deepseekPricing,
-};
+export * from "../prices";

package/dist/streaming.d.ts

@@ -9,14 +9,12 @@ export interface CostTracker {
     reset(): void;
 }
 export interface CreateCostTrackerOptions {
-    provider: string;
-    modelId: string;
+    model: string;
     customPricing?: ModelPricing;
 }
 export declare function createCostTracker(options: CreateCostTrackerOptions): CostTracker;
 export interface StreamCostOptions {
-    provider: string;
-    modelId: string;
+    model: string;
     customPricing?: ModelPricing;
     onCost?: (cost: CostBreakdown, usage: LanguageModelUsage) => void;
 }

package/dist/streaming.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"streaming.d.ts","sourceRoot":"","sources":["../src/streaming.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,EAAiB,KAAK,aAAa,EAAE,MAAM,cAAc,CAAC;AACjE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAE7C,MAAM,WAAW,WAAW;IAC1B,QAAQ,CAAC,KAAK,EAAE,kBAAkB,GAAG,IAAI,CAAC;IAC1C,aAAa,IAAI,kBAAkB,CAAC;IACpC,YAAY,IAAI,aAAa,CAAC;IAC9B,eAAe,IAAI,MAAM,CAAC;IAC1B,KAAK,IAAI,IAAI,CAAC;CACf;AAED,MAAM,WAAW,wBAAwB;IACvC,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,CAAC,EAAE,YAAY,CAAC;CAC9B;AAED,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,wBAAwB,GAAG,WAAW,CAsEhF;AAED,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,aAAa,CAAC,EAAE,YAAY,CAAC;IAC7B,MAAM,CAAC,EAAE,CAAC,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,kBAAkB,KAAK,IAAI,CAAC;CACnE;AAED,wBAAsB,mBAAmB,CAAC,CAAC,SAAS;IAAE,KAAK,EAAE,kBAAkB,CAAA;CAAE,EAC/E,YAAY,EAAE,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,EAC5B,OAAO,EAAE,iBAAiB,GACzB,OAAO,CAAC,aAAa,GAAG;IAAE,KAAK,EAAE,kBAAkB,CAAA;CAAE,CAAC,CAmBxD"}
+{"version":3,"file":"streaming.d.ts","sourceRoot":"","sources":["../src/streaming.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,EAAiB,KAAK,aAAa,EAAE,MAAM,cAAc,CAAC;AACjE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAE7C,MAAM,WAAW,WAAW;IAC1B,QAAQ,CAAC,KAAK,EAAE,kBAAkB,GAAG,IAAI,CAAC;IAC1C,aAAa,IAAI,kBAAkB,CAAC;IACpC,YAAY,IAAI,aAAa,CAAC;IAC9B,eAAe,IAAI,MAAM,CAAC;IAC1B,KAAK,IAAI,IAAI,CAAC;CACf;AAED,MAAM,WAAW,wBAAwB;IACvC,KAAK,EAAE,MAAM,CAAC;IACd,aAAa,CAAC,EAAE,YAAY,CAAC;CAC9B;AAED,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,wBAAwB,GAAG,WAAW,CA2EhF;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,MAAM,CAAC;IACd,aAAa,CAAC,EAAE,YAAY,CAAC;IAC7B,MAAM,CAAC,EAAE,CAAC,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,kBAAkB,KAAK,IAAI,CAAC;CACnE;AAED,wBAAsB,mBAAmB,CAAC,CAAC,SAAS;IAAE,KAAK,EAAE,kBAAkB,CAAA;CAAE,EAC/E,YAAY,EAAE,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,EAC5B,OAAO,EAAE,iBAAiB,GACzB,OAAO,CAAC,aAAa,GAAG;IAAE,KAAK,EAAE,kBAAkB,CAAA;CAAE,CAAC,CAkBxD"}

package/dist/streaming.js

@@ -1,51 +1,55 @@
 import { calculateCost } from "./calculator";
 export function createCostTracker(options) {
-    const { provider, modelId, customPricing } = options;
-    let totalInputTokens = 0;
-    let totalOutputTokens = 0;
+    const { model, customPricing } = options;
+    // Track detail fields directly
+    let totalNoCacheTokens = 0;
     let totalCacheReadTokens = 0;
     let totalCacheWriteTokens = 0;
+    let totalTextTokens = 0;
     let totalReasoningTokens = 0;
     let requestCount = 0;
     return {
         addUsage(usage) {
-            totalInputTokens += usage.inputTokens ?? 0;
-            totalOutputTokens += usage.outputTokens ?? 0;
-            if (usage.inputTokenDetails) {
-                totalCacheReadTokens += usage.inputTokenDetails.cacheReadTokens ?? 0;
-                totalCacheWriteTokens += usage.inputTokenDetails.cacheWriteTokens ?? 0;
+            const inputDetails = usage.inputTokenDetails;
+            if (inputDetails) {
+                totalNoCacheTokens += inputDetails.noCacheTokens ?? 0;
+                totalCacheReadTokens += inputDetails.cacheReadTokens ?? 0;
+                totalCacheWriteTokens += inputDetails.cacheWriteTokens ?? 0;
             }
-            else if (usage.cachedInputTokens) {
-                totalCacheReadTokens += usage.cachedInputTokens;
+            else if (usage.inputTokens !== undefined) {
+                totalNoCacheTokens += usage.inputTokens;
             }
-            if (usage.outputTokenDetails?.reasoningTokens) {
-                totalReasoningTokens += usage.outputTokenDetails.reasoningTokens;
+            const outputDetails = usage.outputTokenDetails;
+            if (outputDetails) {
+                totalTextTokens += outputDetails.textTokens ?? 0;
+                totalReasoningTokens += outputDetails.reasoningTokens ?? 0;
             }
-            else if (usage.reasoningTokens) {
-                totalReasoningTokens += usage.reasoningTokens;
+            else if (usage.outputTokens !== undefined) {
+                totalTextTokens += usage.outputTokens;
             }
             requestCount++;
         },
         getTotalUsage() {
+            const inputTokens = totalNoCacheTokens + totalCacheReadTokens + totalCacheWriteTokens;
+            const outputTokens = totalTextTokens + totalReasoningTokens;
             return {
-                inputTokens: totalInputTokens,
-                outputTokens: totalOutputTokens,
-                totalTokens: totalInputTokens + totalOutputTokens,
+                inputTokens,
+                outputTokens,
+                totalTokens: inputTokens + outputTokens,
                 inputTokenDetails: {
+                    noCacheTokens: totalNoCacheTokens,
                     cacheReadTokens: totalCacheReadTokens,
                     cacheWriteTokens: totalCacheWriteTokens,
-                    noCacheTokens: totalInputTokens - totalCacheReadTokens,
                 },
                 outputTokenDetails: {
+                    textTokens: totalTextTokens,
                     reasoningTokens: totalReasoningTokens,
-                    textTokens: totalOutputTokens - totalReasoningTokens,
                 },
             };
         },
         getTotalCost() {
             return calculateCost({
-                provider,
-                modelId,
+                model,
                 usage: this.getTotalUsage(),
                 customPricing,
             });
@@ -54,10 +58,10 @@ export function createCostTracker(options) {
             return requestCount;
         },
         reset() {
-            totalInputTokens = 0;
-            totalOutputTokens = 0;
+            totalNoCacheTokens = 0;
             totalCacheReadTokens = 0;
             totalCacheWriteTokens = 0;
+            totalTextTokens = 0;
             totalReasoningTokens = 0;
             requestCount = 0;
         },
@@ -67,8 +71,7 @@ export async function calculateStreamCost(streamResult, options) {
     const result = await streamResult;
     const usage = result.usage;
     const cost = calculateCost({
-        provider: options.provider,
-        modelId: options.modelId,
+        model: options.model,
         usage,
         customPricing: options.customPricing,
     });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-sdk-cost-calculator",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Cost calculator for Vercel AI SDK token usage. Supports OpenAI, Anthropic, Google, xAI (Grok), and DeepSeek with long context pricing, prompt caching, and reasoning tokens.",
   "main": "dist/index.js",
   "module": "dist/index.mjs",