@ccusage/amp 18.0.9 → 18.0.10

package/dist/index.js

@@ -1143,7 +1143,7 @@ async function executeCommand(cmd, ctx, name$1) {
 //#endregion
 //#region package.json
 var name = "@ccusage/amp";
-var version = "18.0.9";
+var version = "18.0.10";
 var description = "Usage analysis tool for Amp CLI sessions";
 
 //#endregion
@@ -8183,7 +8183,8 @@ const liteLLMModelPricingSchema = object({
 	cache_creation_input_token_cost_above_200k_tokens: optional(number()),
 	cache_read_input_token_cost_above_200k_tokens: optional(number()),
 	input_cost_per_token_above_128k_tokens: optional(number()),
-	output_cost_per_token_above_128k_tokens: optional(number())
+	output_cost_per_token_above_128k_tokens: optional(number()),
+	provider_specific_entry: optional(object({ fast: optional(number()) }))
 });
 const DEFAULT_PROVIDER_PREFIXES = [
 	"anthropic/",
@@ -8343,11 +8344,13 @@ var LiteLLMPricingFetcher = class {
 		const cacheReadCost = calculateTieredCost(tokens.cache_read_input_tokens, pricing.cache_read_input_token_cost, pricing.cache_read_input_token_cost_above_200k_tokens);
 		return inputCost + outputCost + cacheCreationCost + cacheReadCost;
 	}
-	async calculateCostFromTokens(tokens, modelName) {
+	async calculateCostFromTokens(tokens, modelName, options) {
 		if (modelName == null || modelName === "") return succeed(0);
 		return pipe(this.getModelPricing(modelName), andThen((pricing) => {
 			if (pricing == null) return fail(/* @__PURE__ */ new Error(`Model pricing not found for ${modelName}`));
-			return succeed(this.calculateCostFromPricing(tokens, pricing));
+			const baseCost = this.calculateCostFromPricing(tokens, pricing);
+			const multiplier = options?.speed === "fast" ? pricing.provider_specific_entry?.fast ?? 1 : 1;
+			return succeed(baseCost * multiplier);
 		}));
 	}
 };
@@ -8501,6 +8504,79 @@ if (import.meta.vitest != null) describe("LiteLLMPricingFetcher", () => {
 			_usingCtx7.d();
 		}
 	});
+	it("applies fast speed multiplier from provider_specific_entry", async () => {
+		try {
+			var _usingCtx8 = _usingCtx();
+			const fetcher = _usingCtx8.u(new LiteLLMPricingFetcher({
+				offline: true,
+				offlineLoader: async () => ({ "claude-opus-4-6": {
+					input_cost_per_token: 5e-6,
+					output_cost_per_token: 25e-6,
+					provider_specific_entry: { fast: 6 }
+				} })
+			}));
+			const tokens = {
+				input_tokens: 1e3,
+				output_tokens: 500
+			};
+			const standardCost = await unwrap(fetcher.calculateCostFromTokens(tokens, "claude-opus-4-6"));
+			const fastCost = await unwrap(fetcher.calculateCostFromTokens(tokens, "claude-opus-4-6", { speed: "fast" }));
+			const expectedStandard = .0175;
+			expect(standardCost).toBeCloseTo(expectedStandard);
+			expect(fastCost).toBeCloseTo(expectedStandard * 6);
+		} catch (_$1) {
+			_usingCtx8.e = _$1;
+		} finally {
+			_usingCtx8.d();
+		}
+	});
+	it("defaults to 1x multiplier when provider_specific_entry has no fast field", async () => {
+		try {
+			var _usingCtx9 = _usingCtx();
+			const fetcher = _usingCtx9.u(new LiteLLMPricingFetcher({
+				offline: true,
+				offlineLoader: async () => ({ "claude-sonnet-4-6": {
+					input_cost_per_token: 3e-6,
+					output_cost_per_token: 15e-6
+				} })
+			}));
+			const tokens = {
+				input_tokens: 1e3,
+				output_tokens: 500
+			};
+			const standardCost = await unwrap(fetcher.calculateCostFromTokens(tokens, "claude-sonnet-4-6"));
+			const fastCost = await unwrap(fetcher.calculateCostFromTokens(tokens, "claude-sonnet-4-6", { speed: "fast" }));
+			expect(fastCost).toBeCloseTo(standardCost);
+		} catch (_$1) {
+			_usingCtx9.e = _$1;
+		} finally {
+			_usingCtx9.d();
+		}
+	});
+	it("does not apply multiplier when speed is standard", async () => {
+		try {
+			var _usingCtx10 = _usingCtx();
+			const fetcher = _usingCtx10.u(new LiteLLMPricingFetcher({
+				offline: true,
+				offlineLoader: async () => ({ "claude-opus-4-6": {
+					input_cost_per_token: 5e-6,
+					output_cost_per_token: 25e-6,
+					provider_specific_entry: { fast: 6 }
+				} })
+			}));
+			const tokens = {
+				input_tokens: 1e3,
+				output_tokens: 500
+			};
+			const noSpeedCost = await unwrap(fetcher.calculateCostFromTokens(tokens, "claude-opus-4-6"));
+			const standardCost = await unwrap(fetcher.calculateCostFromTokens(tokens, "claude-opus-4-6", { speed: "standard" }));
+			expect(standardCost).toBeCloseTo(noSpeedCost);
+		} catch (_$1) {
+			_usingCtx10.e = _$1;
+		} finally {
+			_usingCtx10.d();
+		}
+	});
 });
 
 //#endregion

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@ccusage/amp",
   "type": "module",
-  "version": "18.0.9",
+  "version": "18.0.10",
   "description": "Usage analysis tool for Amp CLI sessions",
   "author": "ryoppippi",
   "license": "MIT",