30 lines
727 B
Swift
30 lines
727 B
Swift
import Foundation
|
|
|
|
struct ReasoningConfig {
|
|
static let geminiReasoningModels: Set<String> = [
|
|
"gemini-2.5-flash",
|
|
"gemini-2.5-flash-lite"
|
|
]
|
|
|
|
static let openAIReasoningModels: Set<String> = [
|
|
"gpt-5-mini",
|
|
"gpt-5-nano"
|
|
]
|
|
|
|
static let cerebrasReasoningModels: Set<String> = [
|
|
"gpt-oss-120b"
|
|
]
|
|
|
|
static func getReasoningParameter(for modelName: String) -> String? {
|
|
if geminiReasoningModels.contains(modelName) {
|
|
return "low"
|
|
} else if openAIReasoningModels.contains(modelName) {
|
|
return "minimal"
|
|
} else if cerebrasReasoningModels.contains(modelName) {
|
|
return "low"
|
|
}
|
|
return nil
|
|
}
|
|
}
|
|
|