Add PaLMChat & Multi LLM Providers support

Author: alfianlosari

README.MD

@@ -1,8 +1,8 @@
-# ChatGPT SwiftUI iOS, macOS, watchOS, tvOS App
+# PaLMChat & ChatGPT SwiftUI iOS, macOS, watchOS, tvOS App
 
-![Alt text](https://imagizer.imageshack.com/v2/640x480q90/922/hmlopw.png "image")
+![Alt text](https://imagizer.imageshack.com/v2/640x480q70/924/4Qgrta.jpg "image")
 
-This is a native iOS, macOS, watchOS, tvOS App for interacting with ChatGPT built using SwiftUI and OpenAPI API. It's using Official ChatGPT endpoint with `gpt-3.5-turbo` model.
+This is a native iOS, macOS, watchOS, tvOS App for interacting with PaLM API & ChatGPT LLM Chatbots built using SwiftUI, OpenAPI Official ChatGPT API, & Google Generative AI SDK SPM.
 
 It is also able to render response with markdown and code syntax highlighting.
 
@@ -16,8 +16,7 @@ It is also able to render response with markdown and code syntax highlighting.
 ## Requirements
 - Xcode 14 
 - Register at openai.com/api
-- Create API Key
-- Paste API key in ContentView where the ChatGPTAPI instance is declared
+- Create API Key from either OpenAI or PaLM API MakerSuite
 
 ## ChatGPTSwift API Lib
 You can use this standalone api client to access ChatGPT API, you can add dependency for [ChatGPTSwift](https://github.com/alfianlosari/ChatGPTSwift) to access the API only if you want to integrate into your own app.

Shared/ChatGPTAPI.swift

@@ -7,7 +7,9 @@
 
 import Foundation
 
-class ChatGPTAPI: @unchecked Sendable {
+class ChatGPTAPI: LLMClient, @unchecked Sendable {
+    
+    var provider: LLMProvider { .chatGPT }
 
     private let systemMessage: Message
     private let temperature: Double
@@ -158,3 +160,4 @@ extension String: CustomNSError {
 }
 
 
+

Shared/ChatGPTAPIModels.swift

@@ -7,6 +7,23 @@
 
 import Foundation
 
+enum ChatGPTModel: String, Identifiable, CaseIterable {
+    
+    var id: Self { self }
+    
+    case gpt3Turbo = "gpt-3.5-turbo"
+    case gpt4 = "gpt-4"
+    
+    var text: String {
+        switch self {
+        case .gpt3Turbo:
+            return "GPT-3.5"
+        case .gpt4:
+            return "GPT-4"
+        }
+    }
+}
+
 struct Message: Codable {
     let role: String
     let content: String

Shared/ContentView.swift

@@ -16,7 +16,7 @@ struct ContentView: View {
 
     var body: some View {
         chatListView
-            .navigationTitle("XCA ChatGPT")
+            .navigationTitle(vm.navigationTitle)
     }
 
     var chatListView: some View {
@@ -66,13 +66,14 @@ struct ContentView: View {
             }
 
             TextField("Send message", text: $vm.inputMessage, axis: .vertical)
+                .autocorrectionDisabled()
                 #if os(iOS) || os(macOS)
                 .textFieldStyle(.roundedBorder)
                 #endif
                 .focused($isTextFieldFocused)
-                .disabled(vm.isInteractingWithChatGPT)
+                .disabled(vm.isInteracting)
 
-            if vm.isInteractingWithChatGPT {
+            if vm.isInteracting {
                 #if os(iOS)
                 Button {
                     vm.cancelStreamingResponse()
@@ -122,3 +123,4 @@ struct ContentView_Previews: PreviewProvider {
         }
     }
 }
+

Shared/LLM Provider/LLMClient.swift

@@ -0,0 +1,18 @@
+//
+//  LLMClient.swift
+//  XCAChatGPT
+//
+//  Created by Alfian Losari on 03/06/23.
+//
+
+import Foundation
+
+protocol LLMClient {
+    
+    var provider: LLMProvider { get }
+    
+    func sendMessageStream(text: String) async throws -> AsyncThrowingStream<String, Error>
+    func sendMessage(_ text: String) async throws -> String
+    func deleteHistoryList()
+    
+}

Shared/LLM Provider/LLMConfig.swift

@@ -0,0 +1,31 @@
+//
+//  LLMConfig.swift
+//  XCAChatGPT
+//
+//  Created by Alfian Losari on 03/06/23.
+//
+
+import Foundation
+
+struct LLMConfig: Identifiable, Hashable {
+    
+    var id: String { apiKey }
+    
+    let apiKey: String
+    let type: ConfigType
+    
+    enum ConfigType: Hashable {
+        case chatGPT(ChatGPTModel)
+        case palm
+    }
+    
+    func createClient() -> LLMClient {
+        switch self.type {
+        case .chatGPT(let model):
+            return ChatGPTAPI(apiKey: apiKey, model: model.rawValue)
+        case .palm:
+            return PaLMChatAPI(apiKey: apiKey)
+        }
+    }
+    
+}

Shared/LLM Provider/LLMProvider.swift

@@ -0,0 +1,59 @@
+//
+//  LLMProvider.swift
+//  XCAChatGPT
+//
+//  Created by Alfian Losari on 03/06/23.
+//
+
+import Foundation
+
+enum LLMProvider: Identifiable, CaseIterable {
+    
+    case chatGPT
+    case palm
+    
+    var id: Self { self }
+    
+    var text: String {
+        switch self {
+        case .chatGPT:
+            return "OpenAI ChatGPT"
+        case .palm:
+            return "Google PaLM"
+        }
+    }
+    
+    var footerInfo: String {
+        switch self {
+        case .chatGPT:
+            return """
+ChatGPT is an artificial intelligence (AI) chatbot developed by OpenAI and released in November 2022. The name "ChatGPT" combines "Chat", referring to its chatbot functionality, and "GPT", which stands for Generative Pre-trained Transformer, a type of large language model (LLM). ChatGPT is built upon OpenAI's foundational GPT models, specifically GPT-3.5 and GPT-4, and has been fine-tuned (an approach to transfer learning) for conversational applications using a combination of supervised and reinforcement learning techniques.
+"""
+        case .palm:
+            return """
+PaLM (Pathways Language Model) is a 540 billion parameter transformer-based large language model developed by Google AI.Researchers also trained smaller versions of PaLM, 8 and 62 billion parameter models, to test the effects of model scale.
+            
+PaLM is capable of a wide range of tasks, including commonsense reasoning, arithmetic reasoning, joke explanation, code generation, and translation. When combined with chain-of-thought prompting, PaLM achieved significantly better performance on datasets requiring reasoning of multiple steps, such as word problems and logic-based questions.
+"""
+        }
+    }
+    
+    var navigationTitle: String {
+        switch self {
+        case .chatGPT:
+            return "XCA ChatGPT"
+            
+        case .palm:
+            return "XCA PaLMChat"
+        }
+    }
+    
+    var imageName: String {
+        switch self {
+        case .chatGPT:
+            return "openai"
+        case .palm:
+            return "palm"
+        }
+    }
+}

Shared/LLM Provider/PaLMChatAPI.swift

@@ -0,0 +1,42 @@
+//
+//  PaLMChatAPI.swift
+//  XCAChatGPT
+//
+//  Created by Alfian Losari on 03/06/23.
+//
+
+import Foundation
+import GoogleGenerativeAI
+
+class PaLMChatAPI: LLMClient {
+    
+    var provider: LLMProvider { .palm }
+    private let palmClient: GenerativeLanguage
+    private var history = [GoogleGenerativeAI.Message]()
+    
+    init(apiKey: String) {
+        self.palmClient = .init(apiKey: apiKey)
+    }
+    
+    func sendMessage(_ text: String) async throws -> String {
+        let response = try await palmClient.chat(message: text, history: history)
+        if let candidate = response.candidates?.first, let responseText = candidate.content {
+            if let historicMessages = response.messages {
+                self.history = historicMessages
+                self.history.append(candidate)
+            }
+            return responseText
+        } else {
+            throw "No response"
+        }
+    }
+    
+    func sendMessageStream(text: String) async throws -> AsyncThrowingStream<String, Error> {
+        fatalError("Not supported")
+    }
+    
+    func deleteHistoryList() {
+        self.history = []
+    }
+    
+}

Shared/LLMConfigView.swift

@@ -0,0 +1,95 @@
+//
+//  LLMConfigView.swift
+//  XCAChatGPT
+//
+//  Created by Alfian Losari on 03/06/23.
+//
+
+import SwiftUI
+
+struct LLMConfigView: View {
+    
+    let onStartChatTapped: (_ result: LLMConfig) -> Void
+    @State var apiKey = ""
+    @State var llmProvider = LLMProvider.chatGPT
+    @State var chatGPTModel = ChatGPTModel.gpt3Turbo
+    
+    var body: some View {
+        #if os(macOS)
+        ScrollView { sectionsView }
+            .padding(.horizontal)
+        #else
+        List { sectionsView }
+        #endif
+    }
+    
+    @ViewBuilder
+    var sectionsView: some View {
+        Section("LLM Provider") {
+            Picker("Provider", selection: $llmProvider) {
+                ForEach(LLMProvider.allCases) {
+                    Text($0.text).id($0)
+                }
+            }
+            #if !os(watchOS)
+            .pickerStyle(.segmented)
+            #endif
+        }
+        
+        Section("Configuration") {
+            TextField("Enter API Key", text: $apiKey)
+                .autocorrectionDisabled()
+                #if os(macOS)
+                .textCase(.none)
+                .textFieldStyle(.roundedBorder)
+                #else
+                .textInputAutocapitalization(.never)
+                #endif
+            
+            if llmProvider == .chatGPT {
+                Picker("Model", selection: $chatGPTModel) {
+                    ForEach(ChatGPTModel.allCases) {
+                        Text($0.text).id($0)
+                    }
+                }
+                #if !os(watchOS)
+                .pickerStyle(.segmented)
+                #endif
+            }
+        }
+        
+        Section {
+            Button("Start Chat") {
+                let type: LLMConfig.ConfigType
+                switch llmProvider {
+                case .chatGPT:
+                    type = .chatGPT(chatGPTModel)
+                case .palm:
+                    type = .palm
+                }
+                
+                self.onStartChatTapped(.init(apiKey: apiKey, type: type))
+            }
+            #if os(macOS)
+            .buttonStyle(.borderedProminent)
+            #endif
+            .frame(maxWidth: .infinity)
+            .disabled(apiKey.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty)
+            
+        } footer: {
+            Text(llmProvider.footerInfo)
+                .padding(.vertical)
+        }
+    }
+}
+
+struct LLMConfigView_Previews: PreviewProvider {
+    static var previews: some View {
+        NavigationStack {
+            LLMConfigView { result in
+                
+            }
+        }
+    }
+}
+

Shared/MessageRow.swift

@@ -30,7 +30,7 @@ struct MessageRow: Identifiable {
 
     let id = UUID()
 
-    var isInteractingWithChatGPT: Bool
+    var isInteracting: Bool
 
     let sendImage: String
     var send: MessageRowType