@pydantic/genai-prices 0.0.20 → 0.0.21

package/dist/index.d.ts

@@ -1,5 +1,146 @@
-export { calcPriceSync } from './sync/calcPriceSync.js';
-export { calcPriceAsync } from './async/calcPriceAsync.js';
-export { getProvidersSync, getProvidersAsync, enableAutoUpdate, isLocalDataOutdated, prefetchAsync, getEnvironmentInfo, } from './dataLoader.js';
-export { matchProvider, matchModel } from './matcher.js';
-export type { Usage, PriceCalculation, PriceCalculationResult, Provider, ModelInfo, CalcPrice } from './types.js';
+export declare function calcPrice(usage: Usage, modelId: string, options?: PriceOptions): PriceCalculationResult;
+
+export declare interface ConditionalPrice {
+    constraint?: StartDateConstraint | TimeOfDateConstraint;
+    prices: ModelPrice;
+}
+
+export declare function extractUsage(provider: Provider, responseData: unknown, apiFlavor?: string): [string, Usage];
+
+export declare type MatchLogic = {
+    and: MatchLogic[];
+} | {
+    contains: string;
+} | {
+    ends_with: string;
+} | {
+    equals: string;
+} | {
+    or: MatchLogic[];
+} | {
+    regex: string;
+} | {
+    starts_with: string;
+};
+
+export declare interface ModelInfo {
+    context_window?: number;
+    description?: string;
+    id: string;
+    match: MatchLogic;
+    name?: string;
+    price_comments?: string;
+    prices: ConditionalPrice[] | ModelPrice;
+}
+
+export declare interface ModelPrice {
+    cache_audio_read_mtok?: number | TieredPrices;
+    cache_read_mtok?: number | TieredPrices;
+    cache_write_mtok?: number | TieredPrices;
+    input_audio_mtok?: number | TieredPrices;
+    input_mtok?: number | TieredPrices;
+    output_audio_mtok?: number | TieredPrices;
+    output_mtok?: number | TieredPrices;
+    requests_kcount?: number;
+}
+
+export declare interface ModelPriceCalculationResult {
+    input_price: number;
+    output_price: number;
+    total_price: number;
+}
+
+export declare interface PriceCalculation {
+    auto_update_timestamp?: string;
+    input_price: number;
+    model: ModelInfo;
+    model_price: ModelPrice;
+    output_price: number;
+    provider: Provider;
+    total_price: number;
+}
+
+export declare type PriceCalculationResult = null | PriceCalculation;
+
+export declare interface PriceDataStorage {
+    get: () => Promise<null | string>;
+    get_last_modified?: () => Promise<null | number>;
+    set: (data: string) => Promise<void>;
+}
+
+export declare interface PriceOptions {
+    providerApiUrl?: string;
+    providerId?: string;
+    timestamp?: Date;
+}
+
+export declare interface Provider {
+    api_pattern: string;
+    description?: string;
+    extractors?: UsageExtractor[];
+    id: string;
+    model_match?: MatchLogic;
+    models: ModelInfo[];
+    name: string;
+    price_comments?: string;
+    pricing_urls?: string[];
+    provider_match?: MatchLogic;
+}
+
+export declare const REMOTE_DATA_JSON_URL = "https://raw.githubusercontent.com/pydantic/genai-prices/main/prices/data.json";
+
+export declare interface StartDateConstraint {
+    start_date: string;
+    type: 'start_date';
+}
+
+export declare interface StorageFactoryParams {
+    onCalc: (cb: () => void) => void;
+    remoteDataUrl: string;
+    setProviderData: (data: Promise<Provider[]> | Provider[]) => void;
+}
+
+export declare interface Tier {
+    price: number;
+    start: number;
+}
+
+export declare interface TieredPrices {
+    base: number;
+    tiers: Tier[];
+}
+
+export declare interface TimeOfDateConstraint {
+    end_time: string;
+    start_time: string;
+    type: 'time_of_date';
+}
+
+export declare function updatePrices(factory: (options: StorageFactoryParams) => any): void;
+
+export declare interface Usage {
+    cache_audio_read_tokens?: number;
+    cache_read_tokens?: number;
+    cache_write_tokens?: number;
+    input_audio_tokens?: number;
+    input_tokens?: number;
+    output_audio_tokens?: number;
+    output_tokens?: number;
+}
+
+export declare interface UsageExtractor {
+    api_flavor: string;
+    mappings: UsageExtractorMapping[];
+    model_path: string | string[];
+    root: string | string[];
+}
+
+export declare interface UsageExtractorMapping {
+    dest: 'cache_audio_read_tokens' | 'cache_read_tokens' | 'cache_write_tokens' | 'input_audio_tokens' | 'input_tokens' | 'output_audio_tokens' | 'output_tokens';
+    path: string | string[];
+    required: boolean;
+}
+
+export declare function waitForUpdate(): Promise<Provider[]>;
+
+export { }