@ccusage/amp 18.0.8 → 18.0.10

package/dist/index.js

@@ -4,7 +4,7 @@ import process$1 from "node:process";
 import a, { readFile } from "node:fs/promises";
 import path, { sep } from "node:path";
 import b from "node:fs";
-import F from "node:os";
+import F, { homedir } from "node:os";
 import * as nativeFs$1 from "fs";
 import nativeFs from "fs";
 import path$1, { basename, dirname, normalize, posix, relative, resolve, sep as sep$1 } from "path";
@@ -1143,7 +1143,7 @@ async function executeCommand(cmd, ctx, name$1) {
 //#endregion
 //#region package.json
 var name = "@ccusage/amp";
-var version = "18.0.8";
+var version = "18.0.10";
 var description = "Usage analysis tool for Amp CLI sessions";
 
 //#endregion
@@ -6840,7 +6840,7 @@ const DEFAULT_AMP_PATH = ".local/share/amp";
 /**
 * User home directory
 */
-const USER_HOME_DIR = process$1.env.HOME ?? process$1.env.USERPROFILE ?? process$1.cwd();
+const USER_HOME_DIR = homedir();
 /**
 * Default Amp data directory (absolute path)
 */
@@ -8129,7 +8129,7 @@ if (import.meta.vitest != null) describe("loadAmpUsageEvents", () => {
 	it("handles missing directories gracefully", async () => {
 		const { events, missingDirectories } = await loadAmpUsageEvents({ threadDirs: ["/nonexistent/path"] });
 		expect(events).toEqual([]);
-		expect(missingDirectories).toContain("/nonexistent/path");
+		expect(missingDirectories).toContain(path.resolve("/nonexistent/path"));
 	});
 	it("handles malformed JSON gracefully", async () => {
 		try {
@@ -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.8",
+  "version": "18.0.10",
   "description": "Usage analysis tool for Amp CLI sessions",
   "author": "ryoppippi",
   "license": "MIT",