Handle new chat model format case

Author: intitni

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

@@ -22,6 +22,8 @@ struct ChatModelEditView: View {
                             azureOpenAI
                         case .openAICompatible:
                             openAICompatible
+                        case .googleAI:
+                            googleAI
                         }
                     }
                 }
@@ -88,6 +90,8 @@ struct ChatModelEditView: View {
                             Text("Azure OpenAI").tag(format)
                         case .openAICompatible:
                             Text("OpenAI Compatible").tag(format)
+                        case .googleAI:
+                            Text("Google Generative AI").tag(format)
                         }
                     }
                 },
@@ -269,6 +273,20 @@ struct ChatModelEditView: View {
         maxTokensTextField
         supportsFunctionCallingToggle
     }
+    
+    @ViewBuilder
+    var googleAI: some View {
+        apiKeyNamePicker
+
+        WithViewStore(
+            store,
+            removeDuplicates: { $0.modelName == $1.modelName }
+        ) { viewStore in
+            TextField("Model Name", text: viewStore.$modelName)
+        }
+
+        maxTokensTextField
+    }
 }
 
 #Preview("OpenAI") {

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

@@ -10,6 +10,7 @@ extension ChatModel: ManageableAIModel {
         case .openAI: return "OpenAI"
         case .azureOpenAI: return "Azure OpenAI"
         case .openAICompatible: return "OpenAI Compatible"
+        case .googleAI: return "Google Generative AI"
         }
     }
 

Tool/Sources/AIModel/ChatModel.swift

@@ -20,6 +20,7 @@ public struct ChatModel: Codable, Equatable, Identifiable {
         case openAI
         case azureOpenAI
         case openAICompatible
+        case googleAI
     }
 
     public struct Info: Codable, Equatable {
@@ -69,6 +70,10 @@ public struct ChatModel: Codable, Equatable, Identifiable {
             let version = "2023-07-01-preview"
             if baseURL.isEmpty { return "" }
             return "\(baseURL)/openai/deployments/\(deployment)/chat/completions?api-version=\(version)"
+        case .googleAI:
+            let baseURL = info.baseURL
+            if baseURL.isEmpty { return "https://generativelanguage.googleapis.com/v1" }
+            return "\(baseURL)/v1/chat/completions"
         }
     }
 }

Tool/Sources/OpenAIService/APIs/OpenAICompletionAPI.swift

@@ -94,6 +94,8 @@ struct OpenAICompletionAPI: CompletionAPI {
                 request.setValue("Bearer \(apiKey)", forHTTPHeaderField: "Authorization")
             case .azureOpenAI:
                 request.setValue(apiKey, forHTTPHeaderField: "api-key")
+            case .googleAI:
+                assert(false, "Unsupported")
             }
         }
 

Tool/Sources/OpenAIService/APIs/OpenAICompletionStreamAPI.swift

@@ -186,6 +186,8 @@ struct OpenAICompletionStreamAPI: CompletionStreamAPI {
                 request.setValue("Bearer \(apiKey)", forHTTPHeaderField: "Authorization")
             case .azureOpenAI:
                 request.setValue(apiKey, forHTTPHeaderField: "api-key")
+            case .googleAI:
+                assert(false, "Unsupported")
             }
         }