Cost tracking

export interface CostBreakdown {
    input: number;
    output: number;
    reasoning: number;
    cached: number;
    audioInput: number;
    audioOutput: number;
    total: number;
}

export interface CostBudget {
    maxCostPerRun?: number;
    maxCostPerSession?: number;
    maxCostPerUser?: number;
    maxTokensPerRun?: number;
    onBudgetExceeded?: "throw" | "warn";
}

export interface CostEntry {
    runId: string;
    agentName: string;
    modelId: string;
    usage: TokenUsage;
    cost: number;
    breakdown: CostBreakdown;
    timestamp: Date;
    sessionId?: string;
    userId?: string;
}

export interface CostSummary {
    totalCost: number;
    totalTokens: TokenUsage;
    totalBreakdown: CostBreakdown;
    entries: number;
    byAgent: Record<string, {
        cost: number;
        breakdown: CostBreakdown;
        tokens: TokenUsage;
        runs: number;
    }>;
    byModel: Record<string, {
        cost: number;
        breakdown: CostBreakdown;
        tokens: TokenUsage;
    }>;
    byUser: Record<string, {
        cost: number;
        breakdown: CostBreakdown;
        tokens: TokenUsage;
    }>;
}

export declare class CostTracker {
    constructor(config?: CostTrackerConfig);
    track(entry: Omit<CostEntry, "cost" | "breakdown" | "timestamp">): CostEntry;
    checkBudget(runId: string, sessionId?: string, userId?: string): void;
    getSummary(filter?: {
        agentName?: string;
        userId?: string;
        since?: Date;
    }): CostSummary;
    /**
     * Check budget using in-progress (cumulative) usage without persisting an entry.
     * Called during multi-roundtrip runs (e.g. tool calling) to enforce limits mid-run.
     */
    checkInProgressBudget(modelId: string, cumulativeUsage: TokenUsage, runId?: string, sessionId?: string, userId?: string): boolean;
    /** @deprecated Use checkInProgressBudget for mid-run checks. This now delegates to it. */
    isBudgetExceeded(runId: string, sessionId?: string, userId?: string): boolean;
    estimateRemaining(runId: string): {
        costRemaining: number | null;
        tokensRemaining: number | null;
    };
    getEntries(): readonly CostEntry[];
    reset(): void;
}

export interface CostTrackerConfig {
    pricing?: Record<string, ModelPricing>;
    budget?: CostBudget;
}

export declare const DEFAULT_PRICING: Record<string, ModelPricing>;

export declare function lookupPricing(modelId: string, customPricing?: Record<string, ModelPricing>): ModelPricing | undefined;

export interface ModelPricing {
    promptPer1k: number;
    completionPer1k: number;
    reasoningPer1k?: number;
    cachedPromptPer1k?: number;
    audioInputPer1k?: number;
    audioOutputPer1k?: number;
}