From c640921fa52d49a1a369a8126241a0acb274320c Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Wed, 15 Apr 2026 03:16:34 +0000
Subject: [PATCH] Extract shared model validation helper in model selection
 panel

Agent-Logs-Url: https://github.com/AkhilKonduru1/Atoll/sessions/d40c666d-f589-48ca-9539-1fd33c799962

Co-authored-by: AkhilKonduru1 <210101196+AkhilKonduru1@users.noreply.github.com>
---
 .../ScreenAssistant/ModelSelectionPanel.swift    | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/DynamicIsland/components/ScreenAssistant/ModelSelectionPanel.swift b/DynamicIsland/components/ScreenAssistant/ModelSelectionPanel.swift
index 8550708..4786607 100644
--- a/DynamicIsland/components/ScreenAssistant/ModelSelectionPanel.swift
+++ b/DynamicIsland/components/ScreenAssistant/ModelSelectionPanel.swift
@@ -309,9 +309,7 @@ struct ModelSelectionView: View {
     private func loadCurrentConfiguration() {
         selectedProvider = Defaults[.selectedAIProvider]
         selectedModel = Defaults[.selectedAIModel]
-        if selectedModel == nil || !selectedProvider.supportedModels.contains(where: { $0.id == selectedModel?.id }) {
-            selectedModel = selectedProvider.supportedModels.first
-        }
+        ensureValidModelSelection()
         enableThinking = Defaults[.enableThinkingMode]
         
         geminiApiKey = Defaults[.geminiApiKey]
@@ -322,9 +320,7 @@ struct ModelSelectionView: View {
     }
     
     private func saveConfiguration() {
-        if selectedModel == nil || !selectedProvider.supportedModels.contains(where: { $0.id == selectedModel?.id }) {
-            selectedModel = selectedProvider.supportedModels.first
-        }
+        ensureValidModelSelection()
 
         Defaults[.selectedAIProvider] = selectedProvider
         Defaults[.selectedAIModel] = selectedModel
@@ -344,8 +340,12 @@ struct ModelSelectionView: View {
 
     private func selectProvider(_ provider: AIModelProvider) {
         selectedProvider = provider
-        if !provider.supportedModels.contains(where: { $0.id == selectedModel?.id }) {
-            selectedModel = provider.supportedModels.first
+        ensureValidModelSelection()
+    }
+
+    private func ensureValidModelSelection() {
+        if selectedModel == nil || !selectedProvider.supportedModels.contains(where: { $0.id == selectedModel?.id }) {
+            selectedModel = selectedProvider.supportedModels.first
         }
     }