refactor: Adjust android and iOS sample to use ChatCompletion entrypoint

Author: osugikoji

sample/android/src/main/java/co/yml/ychat/android/MainViewModel.kt

@@ -11,6 +11,15 @@ import kotlinx.coroutines.launch
 
 class MainViewModel(private val chatGpt: YChat) : ViewModel() {
 
+    private val chatCompletions by lazy {
+        chatGpt.chatCompletions()
+            .setMaxTokens(MAX_TOKENS)
+            .addMessage(
+                role = "assistant",
+                content = "You are a helpful assistant."
+            )
+    }
+
     private val _items = mutableStateListOf<MessageItem>()
     val items = _items
 
@@ -47,11 +56,7 @@ class MainViewModel(private val chatGpt: YChat) : ViewModel() {
 
     private suspend fun requestCompletion(message: String): String {
         return try {
-            chatGpt.completion()
-                .setInput(message)
-                .saveHistory(false)
-                .setMaxTokens(MAX_TOKENS)
-                .execute()
+            chatCompletions.execute(message).last().content
         } catch (e: Exception) {
             e.message ?: ERROR
         }

sample/ios/YChatApp/Features/Completion/ViewModel/CompletionViewModel.swift

@@ -10,9 +10,14 @@ import YChat
 import Foundation
 
 internal final class CompletionViewModel: ObservableObject {
-    private var chatGpt: YChat {
+    private var chatCompletions: ChatCompletions =
         YChatCompanion.shared.create(apiKey: Config.apiKey)
-    }
+            .chatCompletions()
+            .setMaxTokens(tokens: 1024)
+            .addMessage(
+                role: "assistant",
+                content: "You are a helpful assistant."
+            )
 
     @Published
     var message: String = ""
@@ -32,11 +37,7 @@ internal final class CompletionViewModel: ObservableObject {
             cleanLastMessage()
             addLoading()
             do {
-                let result = try await chatGpt.completion()
-                    .setInput(input: input)
-                    .setMaxTokens(tokens: 1024)
-                    .saveHistory(isSaveHistory: false)
-                    .execute()
+                let result = try await chatCompletions.execute(content: input)[0].content
                 removeLoading()
                 addAIMessage(message: result)
             } catch {

ychat/src/commonMain/kotlin/co/yml/ychat/data/dto/ChatCompletionsChoiceDto.kt

@@ -10,5 +10,5 @@ internal data class ChatCompletionsChoiceDto(
     @SerialName("message")
     val message: ChatMessageDto,
     @SerialName("finish_reason")
-    val finishReason: String,
+    val finishReason: String?,
 )

ychat/src/commonMain/kotlin/co/yml/ychat/entrypoint/impl/ChatCompletionsImpl.kt

@@ -50,7 +50,7 @@ internal class ChatCompletionsImpl(
     }
 
     override suspend fun execute(content: String): List<ChatMessage> {
-        params.messages.add(ChatMessage("user", content))
+        addMessage("user", content)
         return chatCompletionsUseCase.requestChatCompletions(params)
             .also { params.messages.addAll(it) }
     }