@pydantic/genai-prices 0.0.19 → 0.0.21

package/dist/matcher.d.ts

@@ -1,3 +0,0 @@
-import { Provider, ModelInfo } from './types.js';
-export declare function matchProvider(providers: Provider[], modelRef: string, providerId?: string, providerApiUrl?: string): Provider | undefined;
-export declare function matchModel(models: ModelInfo[], modelRef: string): ModelInfo | undefined;

package/dist/matcher.js

@@ -1,43 +0,0 @@
-function matchLogic(logic, text) {
-    if ('or' in logic) {
-        return logic.or.some((clause) => matchLogic(clause, text));
-    }
-    if ('and' in logic) {
-        return logic.and.every((clause) => matchLogic(clause, text));
-    }
-    if ('equals' in logic) {
-        return text === logic.equals;
-    }
-    if ('starts_with' in logic) {
-        return text.startsWith(logic.starts_with);
-    }
-    if ('ends_with' in logic) {
-        return text.endsWith(logic.ends_with);
-    }
-    if ('contains' in logic) {
-        return text.includes(logic.contains);
-    }
-    if ('regex' in logic) {
-        return new RegExp(logic.regex).test(text);
-    }
-    return false;
-}
-function findProviderById(providers, providerId) {
-    const normalizedProviderId = providerId.toLowerCase().trim();
-    const exactMatch = providers.find((p) => p.id === normalizedProviderId);
-    if (exactMatch)
-        return exactMatch;
-    return providers.find((p) => p.provider_match && matchLogic(p.provider_match, normalizedProviderId));
-}
-export function matchProvider(providers, modelRef, providerId, providerApiUrl) {
-    if (providerId) {
-        return findProviderById(providers, providerId);
-    }
-    if (providerApiUrl) {
-        return providers.find((p) => new RegExp(p.api_pattern).test(providerApiUrl));
-    }
-    return providers.find((p) => p.model_match && matchLogic(p.model_match, modelRef));
-}
-export function matchModel(models, modelRef) {
-    return models.find((m) => matchLogic(m.match, modelRef));
-}

package/dist/priceCalc.d.ts

@@ -1,3 +0,0 @@
-import { Usage, ModelPrice, ModelInfo, CalcPrice } from './types.js';
-export declare function calcPrice(usage: Usage, modelPrice: ModelPrice): CalcPrice;
-export declare function getActiveModelPrice(model: ModelInfo, timestamp: Date): ModelPrice;

package/dist/priceCalc.js

@@ -1,90 +0,0 @@
-function calcTieredPrice(tiered, tokens) {
-    if (tokens <= 0)
-        return 0;
-    let price = 0;
-    // Sort tiers by start ascending
-    const tiers = [...tiered.tiers].sort((a, b) => a.start - b.start);
-    // Base price for tokens up to the first tier start
-    const firstTierStart = tiers[0]?.start ?? tokens;
-    const baseTokens = Math.min(tokens, firstTierStart);
-    price += (baseTokens * tiered.base) / 1_000_000;
-    // Price for each tier
-    for (let i = 0; i < tiers.length; i++) {
-        const tier = tiers[i];
-        const nextTierStart = tiers[i + 1]?.start ?? Infinity;
-        // Tokens in this tier: from tier.start up to nextTierStart or tokens
-        const tierTokenCount = Math.max(0, Math.min(tokens, nextTierStart) - tier.start);
-        if (tierTokenCount > 0) {
-            price += (tierTokenCount * tier.price) / 1_000_000;
-        }
-    }
-    return price;
-}
-function calcMtokPrice(price, tokens, _field) {
-    if (price === undefined || tokens === undefined)
-        return 0;
-    if (typeof price === 'number') {
-        return (price * tokens) / 1_000_000;
-    }
-    return calcTieredPrice(price, tokens);
-}
-export function calcPrice(usage, modelPrice) {
-    let input_price = 0;
-    let output_price = 0;
-    // Input-related prices
-    input_price += calcMtokPrice(modelPrice.input_mtok, usage.input_tokens, 'input_mtok');
-    input_price += calcMtokPrice(modelPrice.cache_write_mtok, usage.cache_write_tokens, 'cache_write_mtok');
-    input_price += calcMtokPrice(modelPrice.cache_read_mtok, usage.cache_read_tokens, 'cache_read_mtok');
-    input_price += calcMtokPrice(modelPrice.input_audio_mtok, usage.input_audio_tokens, 'input_audio_mtok');
-    input_price += calcMtokPrice(modelPrice.cache_audio_read_mtok, usage.cache_audio_read_tokens, 'cache_audio_read_mtok');
-    // Output-related prices
-    output_price += calcMtokPrice(modelPrice.output_mtok, usage.output_tokens, 'output_mtok');
-    output_price += calcMtokPrice(modelPrice.output_audio_mtok, usage.output_audio_tokens, 'output_audio_mtok');
-    // Requests price (counted as input cost)
-    if (modelPrice.requests_kcount !== undefined) {
-        input_price += modelPrice.requests_kcount / 1000;
-    }
-    const total_price = input_price + output_price;
-    return {
-        input_price,
-        output_price,
-        total_price,
-    };
-}
-export function getActiveModelPrice(model, timestamp) {
-    if (!Array.isArray(model.prices)) {
-        return model.prices;
-    }
-    // Conditional prices: last active wins
-    for (let i = model.prices.length - 1; i >= 0; i--) {
-        const cond = model.prices[i];
-        if (!cond.constraint)
-            return cond.prices;
-        if (cond.constraint.type === 'start_date') {
-            if (timestamp >= new Date(cond.constraint.start_date)) {
-                return cond.prices;
-            }
-        }
-        else if (cond.constraint.type === 'time_of_date') {
-            // Extract UTC time to match constraint times which are in UTC (with 'Z' suffix)
-            const t = timestamp.toISOString().slice(11, 19); // Get "HH:MM:SS" from ISO string
-            const startTime = cond.constraint.start_time;
-            const endTime = cond.constraint.end_time;
-            // Handle time ranges that span midnight (end time < start time)
-            if (endTime < startTime) {
-                // Time is in range if it's >= start OR < end
-                if (t >= startTime || t < endTime) {
-                    return cond.prices;
-                }
-            }
-            else {
-                // Normal time range (start <= time < end)
-                if (t >= startTime && t < endTime) {
-                    return cond.prices;
-                }
-            }
-        }
-    }
-    // Fallback to first
-    return model.prices[0].prices;
-}

package/dist/sync/calcPriceSync.d.ts

@@ -1,8 +0,0 @@
-import { Usage, PriceCalculationResult } from '../types.js';
-export type { Usage, PriceCalculation, PriceCalculationResult } from '../types.js';
-export interface CalcPriceOptions {
-    providerId?: string;
-    providerApiUrl?: string;
-    timestamp?: Date;
-}
-export declare function calcPriceSync(usage: Usage, modelRef: string, options?: CalcPriceOptions): PriceCalculationResult;

package/dist/sync/calcPriceSync.js

@@ -1,24 +0,0 @@
-import { getProvidersSync } from '../dataLoader.js';
-import { matchProvider, matchModel } from '../matcher.js';
-import { calcPrice as calcModelPrice, getActiveModelPrice } from '../priceCalc.js';
-export function calcPriceSync(usage, modelRef, options = {}) {
-    const providers = getProvidersSync();
-    const provider = matchProvider(providers, modelRef, options.providerId, options.providerApiUrl);
-    if (!provider)
-        return null;
-    const model = matchModel(provider.models, modelRef);
-    if (!model)
-        return null;
-    const timestamp = options.timestamp || new Date();
-    const model_price = getActiveModelPrice(model, timestamp);
-    const priceResult = calcModelPrice(usage, model_price);
-    return {
-        input_price: priceResult.input_price,
-        output_price: priceResult.output_price,
-        total_price: priceResult.total_price,
-        provider,
-        model,
-        model_price,
-        auto_update_timestamp: undefined,
-    };
-}

package/dist/types.d.ts

@@ -1,95 +0,0 @@
-export interface Usage {
-    input_tokens?: number;
-    cache_write_tokens?: number;
-    cache_read_tokens?: number;
-    output_tokens?: number;
-    input_audio_tokens?: number;
-    cache_audio_read_tokens?: number;
-    output_audio_tokens?: number;
-}
-export interface Tier {
-    start: number;
-    price: number;
-}
-export interface TieredPrices {
-    base: number;
-    tiers: Tier[];
-}
-export interface ModelPrice {
-    input_mtok?: number | TieredPrices;
-    cache_write_mtok?: number | TieredPrices;
-    cache_read_mtok?: number | TieredPrices;
-    output_mtok?: number | TieredPrices;
-    input_audio_mtok?: number | TieredPrices;
-    cache_audio_read_mtok?: number | TieredPrices;
-    output_audio_mtok?: number | TieredPrices;
-    requests_kcount?: number;
-}
-export interface ConditionalPrice {
-    constraint?: StartDateConstraint | TimeOfDateConstraint;
-    prices: ModelPrice;
-}
-export interface StartDateConstraint {
-    start_date: string;
-    type: 'start_date';
-}
-export interface TimeOfDateConstraint {
-    start_time: string;
-    end_time: string;
-    type: 'time_of_date';
-}
-export type MatchLogic = {
-    starts_with: string;
-} | {
-    ends_with: string;
-} | {
-    contains: string;
-} | {
-    equals: string;
-} | {
-    regex: string;
-} | {
-    or: MatchLogic[];
-} | {
-    and: MatchLogic[];
-};
-export interface ModelInfo {
-    id: string;
-    match: MatchLogic;
-    name?: string;
-    description?: string;
-    context_window?: number;
-    price_comments?: string;
-    prices: ModelPrice | ConditionalPrice[];
-}
-export interface Provider {
-    id: string;
-    name: string;
-    api_pattern: string;
-    pricing_urls?: string[];
-    description?: string;
-    price_comments?: string;
-    model_match?: MatchLogic;
-    provider_match?: MatchLogic;
-    models: ModelInfo[];
-}
-export interface CalcPrice {
-    input_price: number;
-    output_price: number;
-    total_price: number;
-}
-export interface PriceCalculation {
-    input_price: number;
-    output_price: number;
-    total_price: number;
-    provider: Provider;
-    model: ModelInfo;
-    model_price: ModelPrice;
-    auto_update_timestamp?: string;
-}
-export type PriceCalculationResult = PriceCalculation | null;
-export type PriceDataStorage = {
-    get: () => Promise<string | null>;
-    set: (data: string) => Promise<void>;
-    get_last_modified?: () => Promise<number | null>;
-};

package/dist/types.js

@@ -1 +0,0 @@
-export {};