vaspera 2.5.0

package/dist/cost/pricing.d.ts

@@ -0,0 +1,57 @@
+/**
+ * Model Pricing
+ *
+ * Pricing data for supported models and cost calculation utilities.
+ * Prices are in USD per 1 million tokens.
+ *
+ * @module cost/pricing
+ */
+import type { ModelId, ModelProvider, ModelPricing, TokenUsage, CallCost } from "./types.js";
+/**
+ * Model pricing data (USD per 1M tokens)
+ * Updated as of 2024
+ */
+export declare const MODEL_PRICING: Record<ModelId, ModelPricing>;
+/**
+ * Get model provider from model ID
+ */
+export declare function getModelProvider(model: ModelId): ModelProvider;
+/**
+ * Get pricing for a model
+ */
+export declare function getModelPricing(model: ModelId): ModelPricing;
+/**
+ * Calculate cost for token usage
+ */
+export declare function calculateCost(model: ModelId, usage: TokenUsage): CallCost;
+/**
+ * Estimate cost for a given number of tokens
+ * If outputTokens is not provided, assumes 80% input, 20% output ratio
+ */
+export declare function estimateCost(model: ModelId, inputTokens: number, outputTokens?: number): number;
+/**
+ * Format cost as USD string
+ */
+export declare function formatCost(cost: number): string;
+/**
+ * Format tokens with K/M suffix
+ */
+export declare function formatTokens(tokens: number): string;
+/**
+ * Estimate tokens for text content
+ * Uses rough approximation of ~4 characters per token
+ */
+export declare function estimateTokens(text: string): number;
+/**
+ * Get all supported models
+ */
+export declare function getSupportedModels(): ModelId[];
+/**
+ * Get models by provider
+ */
+export declare function getModelsByProvider(provider: ModelProvider): ModelId[];
+/**
+ * Get cheapest model for a provider
+ */
+export declare function getCheapestModel(provider?: ModelProvider): ModelId;
+//# sourceMappingURL=pricing.d.ts.map

package/dist/cost/pricing.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pricing.d.ts","sourceRoot":"","sources":["../../src/cost/pricing.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAE7F;;;GAGG;AACH,eAAO,MAAM,aAAa,EAAE,MAAM,CAAC,OAAO,EAAE,YAAY,CA2EvD,CAAC;AAEF;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,OAAO,GAAG,aAAa,CAW9D;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,KAAK,EAAE,OAAO,GAAG,YAAY,CAE5D;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,GAAG,QAAQ,CAezE;AAED;;;GAGG;AACH,wBAAgB,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,YAAY,CAAC,EAAE,MAAM,GAAG,MAAM,CAW/F;AAED;;GAEG;AACH,wBAAgB,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAK/C;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAQnD;AAED;;;GAGG;AACH,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAInD;AAED;;GAEG;AACH,wBAAgB,kBAAkB,IAAI,OAAO,EAAE,CAE9C;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,QAAQ,EAAE,aAAa,GAAG,OAAO,EAAE,CAEtE;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAAC,QAAQ,CAAC,EAAE,aAAa,GAAG,OAAO,CAUlE"}

package/dist/cost/pricing.js

@@ -0,0 +1,196 @@
+/**
+ * Model Pricing
+ *
+ * Pricing data for supported models and cost calculation utilities.
+ * Prices are in USD per 1 million tokens.
+ *
+ * @module cost/pricing
+ */
+/**
+ * Model pricing data (USD per 1M tokens)
+ * Updated as of 2024
+ */
+export const MODEL_PRICING = {
+    // Anthropic Claude 3.x
+    "claude-3-opus": {
+        provider: "anthropic",
+        inputPer1M: 15.0,
+        outputPer1M: 75.0,
+        cachedInputPer1M: 1.5,
+    },
+    "claude-3-sonnet": {
+        provider: "anthropic",
+        inputPer1M: 3.0,
+        outputPer1M: 15.0,
+        cachedInputPer1M: 0.3,
+    },
+    "claude-3-haiku": {
+        provider: "anthropic",
+        inputPer1M: 0.25,
+        outputPer1M: 1.25,
+        cachedInputPer1M: 0.03,
+    },
+    // Anthropic Claude 3.5
+    "claude-3.5-sonnet": {
+        provider: "anthropic",
+        inputPer1M: 3.0,
+        outputPer1M: 15.0,
+        cachedInputPer1M: 0.3,
+    },
+    "claude-3.5-haiku": {
+        provider: "anthropic",
+        inputPer1M: 1.0,
+        outputPer1M: 5.0,
+        cachedInputPer1M: 0.1,
+    },
+    // OpenAI GPT-4
+    "gpt-4": {
+        provider: "openai",
+        inputPer1M: 30.0,
+        outputPer1M: 60.0,
+    },
+    "gpt-4-turbo": {
+        provider: "openai",
+        inputPer1M: 10.0,
+        outputPer1M: 30.0,
+    },
+    "gpt-4o": {
+        provider: "openai",
+        inputPer1M: 2.5,
+        outputPer1M: 10.0,
+    },
+    "gpt-4o-mini": {
+        provider: "openai",
+        inputPer1M: 0.15,
+        outputPer1M: 0.6,
+    },
+    "gpt-3.5-turbo": {
+        provider: "openai",
+        inputPer1M: 0.5,
+        outputPer1M: 1.5,
+    },
+    // Google Gemini
+    "gemini-1.5-pro": {
+        provider: "google",
+        inputPer1M: 1.25,
+        outputPer1M: 5.0,
+    },
+    "gemini-1.5-flash": {
+        provider: "google",
+        inputPer1M: 0.075,
+        outputPer1M: 0.3,
+    },
+    "gemini-1.0-pro": {
+        provider: "google",
+        inputPer1M: 0.5,
+        outputPer1M: 1.5,
+    },
+};
+/**
+ * Get model provider from model ID
+ */
+export function getModelProvider(model) {
+    if (model.startsWith("claude")) {
+        return "anthropic";
+    }
+    if (model.startsWith("gpt")) {
+        return "openai";
+    }
+    if (model.startsWith("gemini")) {
+        return "google";
+    }
+    return "anthropic"; // Default fallback
+}
+/**
+ * Get pricing for a model
+ */
+export function getModelPricing(model) {
+    return MODEL_PRICING[model] || MODEL_PRICING["claude-3.5-sonnet"];
+}
+/**
+ * Calculate cost for token usage
+ */
+export function calculateCost(model, usage) {
+    const pricing = getModelPricing(model);
+    const inputCost = (usage.inputTokens / 1_000_000) * pricing.inputPer1M;
+    const outputCost = (usage.outputTokens / 1_000_000) * pricing.outputPer1M;
+    const cachedInputCost = usage.cachedInputTokens
+        ? (usage.cachedInputTokens / 1_000_000) * (pricing.cachedInputPer1M || pricing.inputPer1M)
+        : 0;
+    return {
+        inputCost,
+        outputCost,
+        cachedInputCost,
+        totalCost: inputCost + outputCost + cachedInputCost,
+    };
+}
+/**
+ * Estimate cost for a given number of tokens
+ * If outputTokens is not provided, assumes 80% input, 20% output ratio
+ */
+export function estimateCost(model, inputTokens, outputTokens) {
+    const actualOutputTokens = outputTokens ?? Math.floor(inputTokens * 0.25);
+    const totalTokens = inputTokens + actualOutputTokens;
+    const cost = calculateCost(model, {
+        inputTokens,
+        outputTokens: actualOutputTokens,
+        totalTokens,
+    });
+    return cost.totalCost;
+}
+/**
+ * Format cost as USD string
+ */
+export function formatCost(cost) {
+    if (cost < 0.01) {
+        return `$${(cost * 100).toFixed(3)}¢`;
+    }
+    return `$${cost.toFixed(4)}`;
+}
+/**
+ * Format tokens with K/M suffix
+ */
+export function formatTokens(tokens) {
+    if (tokens >= 1_000_000) {
+        return `${(tokens / 1_000_000).toFixed(2)}M`;
+    }
+    if (tokens >= 1_000) {
+        return `${(tokens / 1_000).toFixed(1)}K`;
+    }
+    return tokens.toString();
+}
+/**
+ * Estimate tokens for text content
+ * Uses rough approximation of ~4 characters per token
+ */
+export function estimateTokens(text) {
+    // Rough approximation: 1 token ≈ 4 characters for English text
+    // This is a simplification - actual tokenization varies by model
+    return Math.ceil(text.length / 4);
+}
+/**
+ * Get all supported models
+ */
+export function getSupportedModels() {
+    return Object.keys(MODEL_PRICING);
+}
+/**
+ * Get models by provider
+ */
+export function getModelsByProvider(provider) {
+    return getSupportedModels().filter((model) => getModelProvider(model) === provider);
+}
+/**
+ * Get cheapest model for a provider
+ */
+export function getCheapestModel(provider) {
+    const models = provider ? getModelsByProvider(provider) : getSupportedModels();
+    return models.reduce((cheapest, model) => {
+        const cheapestPricing = getModelPricing(cheapest);
+        const modelPricing = getModelPricing(model);
+        const cheapestAvg = (cheapestPricing.inputPer1M + cheapestPricing.outputPer1M) / 2;
+        const modelAvg = (modelPricing.inputPer1M + modelPricing.outputPer1M) / 2;
+        return modelAvg < cheapestAvg ? model : cheapest;
+    }, models[0]);
+}
+//# sourceMappingURL=pricing.js.map

package/dist/cost/pricing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pricing.js","sourceRoot":"","sources":["../../src/cost/pricing.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAIH;;;GAGG;AACH,MAAM,CAAC,MAAM,aAAa,GAAkC;IAC1D,uBAAuB;IACvB,eAAe,EAAE;QACf,QAAQ,EAAE,WAAW;QACrB,UAAU,EAAE,IAAI;QAChB,WAAW,EAAE,IAAI;QACjB,gBAAgB,EAAE,GAAG;KACtB;IACD,iBAAiB,EAAE;QACjB,QAAQ,EAAE,WAAW;QACrB,UAAU,EAAE,GAAG;QACf,WAAW,EAAE,IAAI;QACjB,gBAAgB,EAAE,GAAG;KACtB;IACD,gBAAgB,EAAE;QAChB,QAAQ,EAAE,WAAW;QACrB,UAAU,EAAE,IAAI;QAChB,WAAW,EAAE,IAAI;QACjB,gBAAgB,EAAE,IAAI;KACvB;IACD,uBAAuB;IACvB,mBAAmB,EAAE;QACnB,QAAQ,EAAE,WAAW;QACrB,UAAU,EAAE,GAAG;QACf,WAAW,EAAE,IAAI;QACjB,gBAAgB,EAAE,GAAG;KACtB;IACD,kBAAkB,EAAE;QAClB,QAAQ,EAAE,WAAW;QACrB,UAAU,EAAE,GAAG;QACf,WAAW,EAAE,GAAG;QAChB,gBAAgB,EAAE,GAAG;KACtB;IACD,eAAe;IACf,OAAO,EAAE;QACP,QAAQ,EAAE,QAAQ;QAClB,UAAU,EAAE,IAAI;QAChB,WAAW,EAAE,IAAI;KAClB;IACD,aAAa,EAAE;QACb,QAAQ,EAAE,QAAQ;QAClB,UAAU,EAAE,IAAI;QAChB,WAAW,EAAE,IAAI;KAClB;IACD,QAAQ,EAAE;QACR,QAAQ,EAAE,QAAQ;QAClB,UAAU,EAAE,GAAG;QACf,WAAW,EAAE,IAAI;KAClB;IACD,aAAa,EAAE;QACb,QAAQ,EAAE,QAAQ;QAClB,UAAU,EAAE,IAAI;QAChB,WAAW,EAAE,GAAG;KACjB;IACD,eAAe,EAAE;QACf,QAAQ,EAAE,QAAQ;QAClB,UAAU,EAAE,GAAG;QACf,WAAW,EAAE,GAAG;KACjB;IACD,gBAAgB;IAChB,gBAAgB,EAAE;QAChB,QAAQ,EAAE,QAAQ;QAClB,UAAU,EAAE,IAAI;QAChB,WAAW,EAAE,GAAG;KACjB;IACD,kBAAkB,EAAE;QAClB,QAAQ,EAAE,QAAQ;QAClB,UAAU,EAAE,KAAK;QACjB,WAAW,EAAE,GAAG;KACjB;IACD,gBAAgB,EAAE;QAChB,QAAQ,EAAE,QAAQ;QAClB,UAAU,EAAE,GAAG;QACf,WAAW,EAAE,GAAG;KACjB;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,KAAc;IAC7C,IAAI,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC/B,OAAO,WAAW,CAAC;IACrB,CAAC;IACD,IAAI,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;QAC5B,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,IAAI,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC/B,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,OAAO,WAAW,CAAC,CAAC,mBAAmB;AACzC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,eAAe,CAAC,KAAc;IAC5C,OAAO,aAAa,CAAC,KAAK,CAAC,IAAI,aAAa,CAAC,mBAAmB,CAAC,CAAC;AACpE,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,aAAa,CAAC,KAAc,EAAE,KAAiB;IAC7D,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IAEvC,MAAM,SAAS,GAAG,CAAC,KAAK,CAAC,WAAW,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,UAAU,CAAC;IACvE,MAAM,UAAU,GAAG,CAAC,KAAK,CAAC,YAAY,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,WAAW,CAAC;IAC1E,MAAM,eAAe,GAAG,KAAK,CAAC,iBAAiB;QAC7C,CAAC,CAAC,CAAC,KAAK,CAAC,iBAAiB,GAAG,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,gBAAgB,IAAI,OAAO,CAAC,UAAU,CAAC;QAC1F,CAAC,CAAC,CAAC,CAAC;IAEN,OAAO;QACL,SAAS;QACT,UAAU;QACV,eAAe;QACf,SAAS,EAAE,SAAS,GAAG,UAAU,GAAG,eAAe;KACpD,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,YAAY,CAAC,KAAc,EAAE,WAAmB,EAAE,YAAqB;IACrF,MAAM,kBAAkB,GAAG,YAAY,IAAI,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,CAAC;IAC1E,MAAM,WAAW,GAAG,WAAW,GAAG,kBAAkB,CAAC;IAErD,MAAM,IAAI,GAAG,aAAa,CAAC,KAAK,EAAE;QAChC,WAAW;QACX,YAAY,EAAE,kBAAkB;QAChC,WAAW;KACZ,CAAC,CAAC;IAEH,OAAO,IAAI,CAAC,SAAS,CAAC;AACxB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,UAAU,CAAC,IAAY;IACrC,IAAI,IAAI,GAAG,IAAI,EAAE,CAAC;QAChB,OAAO,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IACxC,CAAC;IACD,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;AAC/B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,YAAY,CAAC,MAAc;IACzC,IAAI,MAAM,IAAI,SAAS,EAAE,CAAC;QACxB,OAAO,GAAG,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IAC/C,CAAC;IACD,IAAI,MAAM,IAAI,KAAK,EAAE,CAAC;QACpB,OAAO,GAAG,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IAC3C,CAAC;IACD,OAAO,MAAM,CAAC,QAAQ,EAAE,CAAC;AAC3B,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,cAAc,CAAC,IAAY;IACzC,+DAA+D;IAC/D,iEAAiE;IACjE,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AACpC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB;IAChC,OAAO,MAAM,CAAC,IAAI,CAAC,aAAa,CAAc,CAAC;AACjD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,QAAuB;IACzD,OAAO,kBAAkB,EAAE,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,gBAAgB,CAAC,KAAK,CAAC,KAAK,QAAQ,CAAC,CAAC;AACtF,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,QAAwB;IACvD,MAAM,MAAM,GAAG,QAAQ,CAAC,CAAC,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,kBAAkB,EAAE,CAAC;IAE/E,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,KAAK,EAAE,EAAE;QACvC,MAAM,eAAe,GAAG,eAAe,CAAC,QAAQ,CAAC,CAAC;QAClD,MAAM,YAAY,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;QAC5C,MAAM,WAAW,GAAG,CAAC,eAAe,CAAC,UAAU,GAAG,eAAe,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QACnF,MAAM,QAAQ,GAAG,CAAC,YAAY,CAAC,UAAU,GAAG,YAAY,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAC1E,OAAO,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC;IACnD,CAAC,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;AAChB,CAAC"}

package/dist/cost/pricing.test.d.ts

@@ -0,0 +1,5 @@
+/**
+ * Tests for cost pricing utilities
+ */
+export {};
+//# sourceMappingURL=pricing.test.d.ts.map

package/dist/cost/pricing.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pricing.test.d.ts","sourceRoot":"","sources":["../../src/cost/pricing.test.ts"],"names":[],"mappings":"AAAA;;GAEG"}

package/dist/cost/pricing.test.js

@@ -0,0 +1,195 @@
+/**
+ * Tests for cost pricing utilities
+ */
+import { describe, it, expect } from "vitest";
+import { MODEL_PRICING, getModelProvider, getModelPricing, calculateCost, estimateCost, formatCost, formatTokens, estimateTokens, getSupportedModels, getModelsByProvider, getCheapestModel, } from "./pricing.js";
+describe("Cost Pricing", () => {
+    describe("MODEL_PRICING", () => {
+        it("has pricing for claude-3.5-sonnet", () => {
+            expect(MODEL_PRICING["claude-3.5-sonnet"]).toBeDefined();
+            expect(MODEL_PRICING["claude-3.5-sonnet"].provider).toBe("anthropic");
+        });
+        it("has pricing for gpt-4-turbo", () => {
+            expect(MODEL_PRICING["gpt-4-turbo"]).toBeDefined();
+            expect(MODEL_PRICING["gpt-4-turbo"].provider).toBe("openai");
+        });
+        it("has pricing for gemini-1.5-pro", () => {
+            expect(MODEL_PRICING["gemini-1.5-pro"]).toBeDefined();
+            expect(MODEL_PRICING["gemini-1.5-pro"].provider).toBe("google");
+        });
+    });
+    describe("getModelProvider", () => {
+        it("returns anthropic for claude models", () => {
+            expect(getModelProvider("claude-3.5-sonnet")).toBe("anthropic");
+            expect(getModelProvider("claude-3-opus")).toBe("anthropic");
+            expect(getModelProvider("claude-3-haiku")).toBe("anthropic");
+        });
+        it("returns openai for gpt models", () => {
+            expect(getModelProvider("gpt-4-turbo")).toBe("openai");
+            expect(getModelProvider("gpt-4o")).toBe("openai");
+            expect(getModelProvider("gpt-4o-mini")).toBe("openai");
+        });
+        it("returns google for gemini models", () => {
+            expect(getModelProvider("gemini-1.5-pro")).toBe("google");
+            expect(getModelProvider("gemini-1.5-flash")).toBe("google");
+        });
+        it("returns anthropic as fallback for unrecognized models", () => {
+            expect(getModelProvider("unknown-model")).toBe("anthropic");
+        });
+    });
+    describe("getModelPricing", () => {
+        it("returns pricing for known models", () => {
+            const pricing = getModelPricing("claude-3.5-sonnet");
+            expect(pricing).toBeDefined();
+            expect(pricing.inputPer1M).toBeGreaterThan(0);
+            expect(pricing.outputPer1M).toBeGreaterThan(0);
+        });
+        it("returns fallback pricing for unknown models", () => {
+            const pricing = getModelPricing("unknown-model");
+            expect(pricing).toBeDefined();
+            // Falls back to claude-3.5-sonnet pricing
+            expect(pricing.inputPer1M).toBe(3.0);
+        });
+    });
+    describe("calculateCost", () => {
+        it("calculates basic cost correctly", () => {
+            const usage = {
+                inputTokens: 1000,
+                outputTokens: 500,
+                totalTokens: 1500,
+            };
+            const cost = calculateCost("claude-3.5-sonnet", usage);
+            // claude-3.5-sonnet: $3/1M input, $15/1M output
+            // Input: 1000 tokens * $3/1M = $0.003
+            // Output: 500 tokens * $15/1M = $0.0075
+            expect(cost.inputCost).toBeCloseTo(0.003, 6);
+            expect(cost.outputCost).toBeCloseTo(0.0075, 6);
+            expect(cost.totalCost).toBeCloseTo(0.0105, 6);
+        });
+        it("applies cached token discount", () => {
+            const usage = {
+                inputTokens: 500,
+                outputTokens: 500,
+                cachedInputTokens: 500,
+                totalTokens: 1500,
+            };
+            const cost = calculateCost("claude-3.5-sonnet", usage);
+            // claude-3.5-sonnet: $3/1M input, $15/1M output, $0.30/1M cached
+            // Input: 500 tokens * $3/1M = $0.0015
+            // Cached: 500 tokens * $0.30/1M = $0.00015
+            // Output: 500 tokens * $15/1M = $0.0075
+            expect(cost.inputCost).toBeCloseTo(0.0015, 6);
+            expect(cost.cachedInputCost).toBeCloseTo(0.00015, 6);
+            expect(cost.outputCost).toBeCloseTo(0.0075, 6);
+        });
+        it("uses fallback pricing for unknown models", () => {
+            const usage = {
+                inputTokens: 1000,
+                outputTokens: 500,
+                totalTokens: 1500,
+            };
+            const cost = calculateCost("unknown-model", usage);
+            // Uses claude-3.5-sonnet pricing as fallback
+            expect(cost.totalCost).toBeGreaterThan(0);
+        });
+    });
+    describe("estimateCost", () => {
+        it("estimates cost from token counts", () => {
+            const estimate = estimateCost("claude-3.5-sonnet", 1000, 500);
+            // Same as calculateCost but without cached tokens
+            expect(estimate).toBeCloseTo(0.0105, 6);
+        });
+        it("estimates with default output ratio when outputTokens not provided", () => {
+            const estimate = estimateCost("claude-3.5-sonnet", 1000);
+            // With 1000 input tokens, default output is 25% = 250 tokens
+            // Input: 1000 * $3/1M = $0.003
+            // Output: 250 * $15/1M = $0.00375
+            expect(estimate).toBeCloseTo(0.00675, 6);
+        });
+        it("uses fallback pricing for unknown models", () => {
+            const estimate = estimateCost("unknown-model", 1000, 500);
+            expect(estimate).toBeGreaterThan(0);
+        });
+    });
+    describe("formatCost", () => {
+        it("formats small costs in cents", () => {
+            // Values < $0.01 use cents notation
+            expect(formatCost(0.001)).toBe("$0.100¢");
+            expect(formatCost(0.0001)).toBe("$0.010¢");
+        });
+        it("formats larger costs with 4 decimals", () => {
+            expect(formatCost(0.0105)).toBe("$0.0105");
+            expect(formatCost(1.50)).toBe("$1.5000");
+            expect(formatCost(10.00)).toBe("$10.0000");
+        });
+        it("handles zero", () => {
+            expect(formatCost(0)).toBe("$0.000¢");
+        });
+    });
+    describe("formatTokens", () => {
+        it("formats small token counts without suffix", () => {
+            expect(formatTokens(500)).toBe("500");
+            expect(formatTokens(999)).toBe("999");
+        });
+        it("formats thousands with K suffix", () => {
+            expect(formatTokens(1000)).toBe("1.0K");
+            expect(formatTokens(5500)).toBe("5.5K");
+            expect(formatTokens(10000)).toBe("10.0K");
+        });
+        it("formats millions with M suffix", () => {
+            expect(formatTokens(1000000)).toBe("1.00M");
+            expect(formatTokens(2500000)).toBe("2.50M");
+        });
+    });
+    describe("estimateTokens", () => {
+        it("estimates tokens from text", () => {
+            // Roughly 4 characters per token
+            const tokens = estimateTokens("Hello, world!"); // 13 chars
+            expect(tokens).toBe(4); // ceil(13/4)
+        });
+        it("handles empty string", () => {
+            expect(estimateTokens("")).toBe(0);
+        });
+        it("handles long text", () => {
+            const longText = "a".repeat(1000);
+            expect(estimateTokens(longText)).toBe(250);
+        });
+    });
+    describe("getSupportedModels", () => {
+        it("returns array of model IDs", () => {
+            const models = getSupportedModels();
+            expect(Array.isArray(models)).toBe(true);
+            expect(models.length).toBeGreaterThan(0);
+            expect(models).toContain("claude-3.5-sonnet");
+            expect(models).toContain("gpt-4-turbo");
+            expect(models).toContain("gemini-1.5-pro");
+        });
+    });
+    describe("getModelsByProvider", () => {
+        it("returns only anthropic models", () => {
+            const models = getModelsByProvider("anthropic");
+            expect(models.every((m) => m.startsWith("claude"))).toBe(true);
+        });
+        it("returns only openai models", () => {
+            const models = getModelsByProvider("openai");
+            expect(models.every((m) => m.startsWith("gpt") || m.startsWith("o"))).toBe(true);
+        });
+        it("returns only google models", () => {
+            const models = getModelsByProvider("google");
+            expect(models.every((m) => m.startsWith("gemini"))).toBe(true);
+        });
+    });
+    describe("getCheapestModel", () => {
+        it("returns cheapest model for provider", () => {
+            const cheapest = getCheapestModel("anthropic");
+            expect(cheapest).toBeDefined();
+            // Haiku should be cheapest Claude model
+            expect(cheapest).toBe("claude-3-haiku");
+        });
+        it("returns overall cheapest when no provider specified", () => {
+            const cheapest = getCheapestModel();
+            expect(cheapest).toBeDefined();
+        });
+    });
+});
+//# sourceMappingURL=pricing.test.js.map

package/dist/cost/pricing.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pricing.test.js","sourceRoot":"","sources":["../../src/cost/pricing.test.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,aAAa,EACb,gBAAgB,EAChB,eAAe,EACf,aAAa,EACb,YAAY,EACZ,UAAU,EACV,YAAY,EACZ,cAAc,EACd,kBAAkB,EAClB,mBAAmB,EACnB,gBAAgB,GACjB,MAAM,cAAc,CAAC;AAGtB,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YACzD,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YACnD,MAAM,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;YACtD,MAAM,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAChE,MAAM,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC5D,MAAM,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvD,MAAM,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAClD,MAAM,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC1D,MAAM,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,MAAM,CAAC,gBAAgB,CAAC,eAA0B,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,OAAO,GAAG,eAAe,CAAC,mBAAmB,CAAC,CAAC;YACrD,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;YAC9B,MAAM,CAAC,OAAQ,CAAC,UAAU,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAC/C,MAAM,CAAC,OAAQ,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,OAAO,GAAG,eAAe,CAAC,eAA0B,CAAC,CAAC;YAC5D,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;YAC9B,0CAA0C;YAC1C,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,KAAK,GAAe;gBACxB,WAAW,EAAE,IAAI;gBACjB,YAAY,EAAE,GAAG;gBACjB,WAAW,EAAE,IAAI;aAClB,CAAC;YAEF,MAAM,IAAI,GAAG,aAAa,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAC;YAEvD,gDAAgD;YAChD,sCAAsC;YACtC,wCAAwC;YACxC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YAC7C,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAC/C,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,KAAK,GAAe;gBACxB,WAAW,EAAE,GAAG;gBAChB,YAAY,EAAE,GAAG;gBACjB,iBAAiB,EAAE,GAAG;gBACtB,WAAW,EAAE,IAAI;aAClB,CAAC;YAEF,MAAM,IAAI,GAAG,aAAa,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAC;YAEvD,iEAAiE;YACjE,sCAAsC;YACtC,2CAA2C;YAC3C,wCAAwC;YACxC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAC9C,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YACrD,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAe;gBACxB,WAAW,EAAE,IAAI;gBACjB,YAAY,EAAE,GAAG;gBACjB,WAAW,EAAE,IAAI;aAClB,CAAC;YAEF,MAAM,IAAI,GAAG,aAAa,CAAC,eAA0B,EAAE,KAAK,CAAC,CAAC;YAE9D,6CAA6C;YAC7C,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,QAAQ,GAAG,YAAY,CAAC,mBAAmB,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;YAE9D,kDAAkD;YAClD,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;YAC5E,MAAM,QAAQ,GAAG,YAAY,CAAC,mBAAmB,EAAE,IAAI,CAAC,CAAC;YAEzD,6DAA6D;YAC7D,+BAA+B;YAC/B,kCAAkC;YAClC,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,QAAQ,GAAG,YAAY,CAAC,eAA0B,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;YACrE,MAAM,CAAC,QAAQ,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,oCAAoC;YACpC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC1C,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC3C,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACzC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,cAAc,EAAE,GAAG,EAAE;YACtB,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACtC,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACxC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACxC,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC5C,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,iCAAiC;YACjC,MAAM,MAAM,GAAG,cAAc,CAAC,eAAe,CAAC,CAAC,CAAC,WAAW;YAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;YAC9B,MAAM,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mBAAmB,EAAE,GAAG,EAAE;YAC3B,MAAM,QAAQ,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,MAAM,GAAG,kBAAkB,EAAE,CAAC;YACpC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,MAAM,GAAG,mBAAmB,CAAC,WAAW,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,QAAQ,GAAG,gBAAgB,CAAC,WAAW,CAAC,CAAC;YAC/C,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;YAC/B,wCAAwC;YACxC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,QAAQ,GAAG,gBAAgB,EAAE,CAAC;YACpC,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/cost/tracker.d.ts

@@ -0,0 +1,100 @@
+/**
+ * Cost Tracker
+ *
+ * Tracks API costs during certifications with budget limits.
+ *
+ * @module cost/tracker
+ */
+import type { ModelId, TokenUsage, ApiCallRecord, CostSummary, BudgetConfig, BudgetStatus, CostTrackerState, CostTrackingOptions } from "./types.js";
+/**
+ * Cost tracker for a certification
+ */
+export declare class CostTracker {
+    private state;
+    private options;
+    private persistPath?;
+    constructor(certificationId: string, projectPath: string, options?: CostTrackingOptions);
+    /**
+     * Record an API call
+     */
+    recordCall(model: ModelId, usage: TokenUsage, options?: {
+        agent?: string;
+        operation?: string;
+        cached?: boolean;
+    }): ApiCallRecord;
+    /**
+     * Check if within budget
+     */
+    isWithinBudget(): boolean;
+    /**
+     * Check budget and throw if exceeded and abortOnExceeded is true
+     */
+    checkBudget(): void;
+    /**
+     * Get current summary
+     */
+    getSummary(): CostSummary;
+    /**
+     * Get current status
+     */
+    getStatus(): BudgetStatus;
+    /**
+     * Get budget configuration
+     */
+    getBudget(): BudgetConfig;
+    /**
+     * Update budget configuration
+     */
+    setBudget(budget: Partial<BudgetConfig>): void;
+    /**
+     * Get all call records
+     */
+    getCalls(): ApiCallRecord[];
+    /**
+     * Get full state
+     */
+    getState(): CostTrackerState;
+    /**
+     * Save state to file
+     */
+    persist(): Promise<void>;
+    /**
+     * Load state from file
+     */
+    load(): Promise<boolean>;
+    /**
+     * Generate cost report
+     */
+    generateReport(): string;
+    /**
+     * Generate compact summary string
+     */
+    generateCompactSummary(): string;
+    private createEmptySummary;
+    private createInitialStatus;
+    private createDummyRecord;
+    private updateSummary;
+    private updateStatus;
+    private generateProgressBar;
+}
+/**
+ * Error thrown when budget is exceeded
+ */
+export declare class BudgetExceededError extends Error {
+    readonly summary: CostSummary;
+    readonly budget: BudgetConfig;
+    constructor(message: string, summary: CostSummary, budget: BudgetConfig);
+}
+/**
+ * Get or create tracker for a certification
+ */
+export declare function getTracker(certificationId: string, projectPath: string, options?: CostTrackingOptions): CostTracker;
+/**
+ * Remove tracker from registry
+ */
+export declare function removeTracker(certificationId: string, projectPath: string): void;
+/**
+ * Get all active trackers
+ */
+export declare function getActiveTrackers(): Map<string, CostTracker>;
+//# sourceMappingURL=tracker.d.ts.map

package/dist/cost/tracker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tracker.d.ts","sourceRoot":"","sources":["../../src/cost/tracker.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAKH,OAAO,KAAK,EACV,OAAO,EACP,UAAU,EACV,aAAa,EACb,WAAW,EACX,YAAY,EACZ,YAAY,EACZ,gBAAgB,EAChB,mBAAmB,EACpB,MAAM,YAAY,CAAC;AAYpB;;GAEG;AACH,qBAAa,WAAW;IACtB,OAAO,CAAC,KAAK,CAAmB;IAChC,OAAO,CAAC,OAAO,CAAsB;IACrC,OAAO,CAAC,WAAW,CAAC,CAAS;gBAG3B,eAAe,EAAE,MAAM,EACvB,WAAW,EAAE,MAAM,EACnB,OAAO,CAAC,EAAE,mBAAmB;IAsB/B;;OAEG;IACH,UAAU,CACR,KAAK,EAAE,OAAO,EACd,KAAK,EAAE,UAAU,EACjB,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,OAAO,CAAA;KAAE,GACjE,aAAa;IA+BhB;;OAEG;IACH,cAAc,IAAI,OAAO;IAIzB;;OAEG;IACH,WAAW,IAAI,IAAI;IAWnB;;OAEG;IACH,UAAU,IAAI,WAAW;IAIzB;;OAEG;IACH,SAAS,IAAI,YAAY;IAIzB;;OAEG;IACH,SAAS,IAAI,YAAY;IAIzB;;OAEG;IACH,SAAS,CAAC,MAAM,EAAE,OAAO,CAAC,YAAY,CAAC,GAAG,IAAI;IAK9C;;OAEG;IACH,QAAQ,IAAI,aAAa,EAAE;IAI3B;;OAEG;IACH,QAAQ,IAAI,gBAAgB;IAU5B;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAO9B;;OAEG;IACG,IAAI,IAAI,OAAO,CAAC,OAAO,CAAC;IAa9B;;OAEG;IACH,cAAc,IAAI,MAAM;IAkFxB;;OAEG;IACH,sBAAsB,IAAI,MAAM;IAQhC,OAAO,CAAC,kBAAkB;IAc1B,OAAO,CAAC,mBAAmB;IAW3B,OAAO,CAAC,iBAAiB;IAWzB,OAAO,CAAC,aAAa;IA8BrB,OAAO,CAAC,YAAY;IAkDpB,OAAO,CAAC,mBAAmB;CAO5B;AAED;;GAEG;AACH,qBAAa,mBAAoB,SAAQ,KAAK;aAG1B,OAAO,EAAE,WAAW;aACpB,MAAM,EAAE,YAAY;gBAFpC,OAAO,EAAE,MAAM,EACC,OAAO,EAAE,WAAW,EACpB,MAAM,EAAE,YAAY;CAKvC;AAOD;;GAEG;AACH,wBAAgB,UAAU,CACxB,eAAe,EAAE,MAAM,EACvB,WAAW,EAAE,MAAM,EACnB,OAAO,CAAC,EAAE,mBAAmB,GAC5B,WAAW,CAQb;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,eAAe,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,GAAG,IAAI,CAGhF;AAED;;GAEG;AACH,wBAAgB,iBAAiB,IAAI,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,CAE5D"}