From ea64ff3109c0f1c7f2941bc4d7979ff2ec47b870 Mon Sep 17 00:00:00 2001
From: Beingpax <prakashjoshipaxtek@gmail.com>
Date: Thu, 4 Dec 2025 20:53:15 +0545
Subject: [PATCH] Update AI enhancement models

---
 VoiceInk/Services/AIEnhancement/AIService.swift    | 14 +++++++-------
 .../Services/AIEnhancement/ReasoningConfig.swift   |  7 ++++++-
 2 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/VoiceInk/Services/AIEnhancement/AIService.swift b/VoiceInk/Services/AIEnhancement/AIService.swift
index 41dca21..1c6b663 100644
--- a/VoiceInk/Services/AIEnhancement/AIService.swift
+++ b/VoiceInk/Services/AIEnhancement/AIService.swift
@@ -49,13 +49,13 @@ enum AIProvider: String, CaseIterable {
         case .cerebras:
             return "gpt-oss-120b"
         case .groq:
-            return "qwen/qwen3-32b"
+            return "openai/gpt-oss-120b"
         case .gemini:
             return "gemini-2.5-flash-lite"
         case .anthropic:
-            return "claude-haiku-4-5"
+            return "claude-sonnet-4-5"
         case .openAI:
-            return "gpt-5-mini"
+            return "gpt-5.1"
         case .mistral:
             return "mistral-large-latest"
         case .elevenLabs:
@@ -77,9 +77,9 @@ enum AIProvider: String, CaseIterable {
         switch self {
         case .cerebras:
             return [
+                "gpt-oss-120b",
                 "llama-4-scout-17b-16e-instruct",
                 "llama-3.3-70b",
-                "gpt-oss-120b",
                 "qwen-3-32b",
                 "qwen-3-235b-a22b-instruct-2507"
             ]
@@ -95,6 +95,7 @@ enum AIProvider: String, CaseIterable {
             ]
         case .gemini:
             return [
+                "gemini-3-pro-preview",
                 "gemini-2.5-pro",
                 "gemini-2.5-flash",
                 "gemini-2.5-flash-lite",
@@ -102,14 +103,13 @@ enum AIProvider: String, CaseIterable {
             ]
         case .anthropic:
             return [
-                "claude-opus-4-0",
-                "claude-sonnet-4-0",
+                "claude-opus-4-5",
                 "claude-sonnet-4-5",
                 "claude-haiku-4-5"
             ]
         case .openAI:
             return [
-                "gpt-5",
+                "gpt-5.1",
                 "gpt-5-mini",
                 "gpt-5-nano",
                 "gpt-4.1",
diff --git a/VoiceInk/Services/AIEnhancement/ReasoningConfig.swift b/VoiceInk/Services/AIEnhancement/ReasoningConfig.swift
index 8a0b1be..72abf2b 100644
--- a/VoiceInk/Services/AIEnhancement/ReasoningConfig.swift
+++ b/VoiceInk/Services/AIEnhancement/ReasoningConfig.swift
@@ -7,16 +7,21 @@ struct ReasoningConfig {
     ]
 
     static let openAIReasoningModels: Set<String> = [
-        "gpt-5",
         "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
     }