Add gemini support in UI

Author: intitni

Core/Sources/HostApp/AccountSettings/ChatModelManagement/ChatModelEdit.swift

@@ -114,15 +114,26 @@ struct ChatModelEdit: ReducerProtocol {
                 return .none
 
             case .checkSuggestedMaxTokens:
-                guard state.format == .openAI,
-                      let knownModel = ChatGPTModel(rawValue: state.modelName)
-                else {
+                switch state.format {
+                case .openAI:
+                    if let knownModel = ChatGPTModel(rawValue: state.modelName) {
+                        state.suggestedMaxTokens = knownModel.maxToken
+                    } else {
+                        state.suggestedMaxTokens = nil
+                    }
+                    return .none
+                case .googleAI:
+                    if let knownModel = GoogleGenerativeAIModel(rawValue: state.modelName) {
+                        state.suggestedMaxTokens = knownModel.maxToken
+                    } else {
+                        state.suggestedMaxTokens = nil
+                    }
+                    return .none
+                default:
                     state.suggestedMaxTokens = nil
                     return .none
                 }
-                state.suggestedMaxTokens = knownModel.maxToken
-                return .none
-
+                
             case .apiKeySelection:
                 return .none
 

Core/Sources/HostApp/AccountSettings/ChatModelManagement/ChatModelEditView.swift

@@ -277,12 +277,27 @@ struct ChatModelEditView: View {
     @ViewBuilder
     var googleAI: some View {
         apiKeyNamePicker
-
+        
         WithViewStore(
             store,
             removeDuplicates: { $0.modelName == $1.modelName }
         ) { viewStore in
             TextField("Model Name", text: viewStore.$modelName)
+                .overlay(alignment: .trailing) {
+                    Picker(
+                        "",
+                        selection: viewStore.$modelName,
+                        content: {
+                            if GoogleGenerativeAIModel(rawValue: viewStore.state.modelName) == nil {
+                                Text("Custom Model").tag(viewStore.state.modelName)
+                            }
+                            ForEach(GoogleGenerativeAIModel.allCases, id: \.self) { model in
+                                Text(model.rawValue).tag(model.rawValue)
+                            }
+                        }
+                    )
+                    .frame(width: 20)
+                }
         }
 
         maxTokensTextField

Pro

@@ -0,0 +1,17 @@
+import Foundation
+
+public enum GoogleGenerativeAIModel: String {
+    case geminiPro = "gemini-pro"
+}
+
+public extension GoogleGenerativeAIModel {
+    var maxToken: Int {
+        switch self {
+        case .geminiPro:
+            return 32768
+        }
+    }
+}
+
+extension GoogleGenerativeAIModel: CaseIterable {}
+