//
//  ModelInfo.swift
//  oAI
//
//  Model information and capabilities
//

import Foundation

struct ModelInfo: Identifiable, Codable, Hashable {
    let id: String
    let name: String
    let description: String?
    let contextLength: Int
    let pricing: Pricing
    let capabilities: ModelCapabilities
    var architecture: Architecture? = nil
    var topProvider: String? = nil

    struct Pricing: Codable, Hashable {
        let prompt: Double    // per 1M tokens
        let completion: Double
    }

    struct ModelCapabilities: Codable, Hashable {
        let vision: Bool           // Images/PDFs
        let tools: Bool            // Function calling
        let online: Bool           // Web search
        var imageGeneration: Bool = false  // Image output
    }

    struct Architecture: Codable, Hashable {
        let tokenizer: String?
        let instructType: String?
        let modality: String?
    }

    // Computed properties
    var contextLengthDisplay: String {
        if contextLength >= 1_000_000 {
            return "\(contextLength / 1_000_000)M"
        } else if contextLength >= 1000 {
            return "\(contextLength / 1000)K"
        } else {
            return "\(contextLength)"
        }
    }
    
    var promptPriceDisplay: String {
        String(format: "$%.2f", pricing.prompt)
    }
    
    var completionPriceDisplay: String {
        String(format: "$%.2f", pricing.completion)
    }
}