57 lines
1.4 KiB
Swift
57 lines
1.4 KiB
Swift
//
|
|
// 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)
|
|
}
|
|
}
|