Remove last message in chat history

LikeLionHGU · Aug 6, 2024 · cd68633 · cd68633
1 parent 92ca6aa
commit cd68633
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/main/kotlin/com/likelionhgu/stepper/chat/ChatService.kt b/src/main/kotlin/com/likelionhgu/stepper/chat/ChatService.kt
@@ -124,7 +124,7 @@ class ChatService(
     }
 
     fun generateSummaryOf(chatId: String): ChatSummaryResponse {
-        val chatHistory = chatHistoryOf(chatId).toSimpleMessage()
+        val chatHistory = chatHistoryOf(chatId).removeLast().toSimpleMessage()
 
         with(openAiProperties.completion) {
             val requestBody = CompletionRequest.of(modelType.id, instructions, chatHistory)

diff --git a/src/main/kotlin/com/likelionhgu/stepper/openai/assistant/message/MessageResponseWrapper.kt b/src/main/kotlin/com/likelionhgu/stepper/openai/assistant/message/MessageResponseWrapper.kt
@@ -18,6 +18,10 @@ data class MessageResponseWrapper(val data: List<MessageResponse>) {
         }
     }
 
+    fun removeLast(): MessageResponseWrapper {
+        return MessageResponseWrapper(data.dropLast(1))
+    }
+
     companion object {
         fun reverseOf(response: MessageResponseWrapper): MessageResponseWrapper {
             return MessageResponseWrapper(response.data.reversed())