Initial commit
This commit is contained in:
56
oAI/Models/ModelInfo.swift
Normal file
56
oAI/Models/ModelInfo.swift
Normal file
@@ -0,0 +1,56 @@
|
||||
//
|
||||
// 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)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user