Update AI enhancement models

This commit is contained in:
Beingpax 2025-12-04 20:53:15 +05:45
parent b754f39d89
commit ea64ff3109
2 changed files with 13 additions and 8 deletions

View File

@ -49,13 +49,13 @@ enum AIProvider: String, CaseIterable {
case .cerebras: case .cerebras:
return "gpt-oss-120b" return "gpt-oss-120b"
case .groq: case .groq:
return "qwen/qwen3-32b" return "openai/gpt-oss-120b"
case .gemini: case .gemini:
return "gemini-2.5-flash-lite" return "gemini-2.5-flash-lite"
case .anthropic: case .anthropic:
return "claude-haiku-4-5" return "claude-sonnet-4-5"
case .openAI: case .openAI:
return "gpt-5-mini" return "gpt-5.1"
case .mistral: case .mistral:
return "mistral-large-latest" return "mistral-large-latest"
case .elevenLabs: case .elevenLabs:
@ -77,9 +77,9 @@ enum AIProvider: String, CaseIterable {
switch self { switch self {
case .cerebras: case .cerebras:
return [ return [
"gpt-oss-120b",
"llama-4-scout-17b-16e-instruct", "llama-4-scout-17b-16e-instruct",
"llama-3.3-70b", "llama-3.3-70b",
"gpt-oss-120b",
"qwen-3-32b", "qwen-3-32b",
"qwen-3-235b-a22b-instruct-2507" "qwen-3-235b-a22b-instruct-2507"
] ]
@ -95,6 +95,7 @@ enum AIProvider: String, CaseIterable {
] ]
case .gemini: case .gemini:
return [ return [
"gemini-3-pro-preview",
"gemini-2.5-pro", "gemini-2.5-pro",
"gemini-2.5-flash", "gemini-2.5-flash",
"gemini-2.5-flash-lite", "gemini-2.5-flash-lite",
@ -102,14 +103,13 @@ enum AIProvider: String, CaseIterable {
] ]
case .anthropic: case .anthropic:
return [ return [
"claude-opus-4-0", "claude-opus-4-5",
"claude-sonnet-4-0",
"claude-sonnet-4-5", "claude-sonnet-4-5",
"claude-haiku-4-5" "claude-haiku-4-5"
] ]
case .openAI: case .openAI:
return [ return [
"gpt-5", "gpt-5.1",
"gpt-5-mini", "gpt-5-mini",
"gpt-5-nano", "gpt-5-nano",
"gpt-4.1", "gpt-4.1",

View File

@ -7,16 +7,21 @@ struct ReasoningConfig {
] ]
static let openAIReasoningModels: Set<String> = [ static let openAIReasoningModels: Set<String> = [
"gpt-5",
"gpt-5-mini", "gpt-5-mini",
"gpt-5-nano" "gpt-5-nano"
] ]
static let cerebrasReasoningModels: Set<String> = [
"gpt-oss-120b"
]
static func getReasoningParameter(for modelName: String) -> String? { static func getReasoningParameter(for modelName: String) -> String? {
if geminiReasoningModels.contains(modelName) { if geminiReasoningModels.contains(modelName) {
return "low" return "low"
} else if openAIReasoningModels.contains(modelName) { } else if openAIReasoningModels.contains(modelName) {
return "minimal" return "minimal"
} else if cerebrasReasoningModels.contains(modelName) {
return "low"
} }
return nil return nil
} }