From ef1c24cac6bd635ff2fb8c08802fd75cdbf26697 Mon Sep 17 00:00:00 2001 From: raulraja Date: Thu, 29 Jun 2023 20:11:00 +0200 Subject: [PATCH] Rename enum cases --- .../kotlin/com/xebia/functional/xef/auto/CoreAIScope.kt | 4 ++-- .../com/xebia/functional/xef/llm/models/chat/Message.kt | 2 +- .../xef/llm/models/chat/MessageWithFunctionCall.kt | 2 +- .../com/xebia/functional/xef/llm/models/chat/Role.kt | 8 ++++---- 4 files changed, 8 insertions(+), 8 deletions(-) diff --git a/core/src/commonMain/kotlin/com/xebia/functional/xef/auto/CoreAIScope.kt b/core/src/commonMain/kotlin/com/xebia/functional/xef/auto/CoreAIScope.kt index 114e901b1..e22ba22d0 100644 --- a/core/src/commonMain/kotlin/com/xebia/functional/xef/auto/CoreAIScope.kt +++ b/core/src/commonMain/kotlin/com/xebia/functional/xef/auto/CoreAIScope.kt @@ -256,7 +256,7 @@ class CoreAIScope( } suspend fun buildChatRequest(): ChatCompletionRequest { - val messages: List = listOf(Message(Role.system.name, promptWithContext)) + val messages: List = listOf(Message(Role.SYSTEM.name, promptWithContext)) return ChatCompletionRequest( model = model.name, user = user, @@ -268,7 +268,7 @@ class CoreAIScope( } suspend fun chatWithFunctionsRequest(): ChatCompletionRequestWithFunctions { - val role: String = Role.user.name + val role: String = Role.USER.name val firstFnName: String? = functions.firstOrNull()?.name val messages: List = listOf(Message(role, promptWithContext)) return ChatCompletionRequestWithFunctions( diff --git a/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/models/chat/Message.kt b/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/models/chat/Message.kt index 6b205f7cd..30fc0e8f5 100644 --- a/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/models/chat/Message.kt +++ b/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/models/chat/Message.kt @@ -1,3 +1,3 @@ package com.xebia.functional.xef.llm.models.chat -data class Message(val role: String, val content: String?, val name: String? = Role.assistant.name) +data class Message(val role: String, val content: String?, val name: String? = Role.ASSISTANT.name) diff --git a/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/models/chat/MessageWithFunctionCall.kt b/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/models/chat/MessageWithFunctionCall.kt index 20ad22bb6..eeeb6a3ef 100644 --- a/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/models/chat/MessageWithFunctionCall.kt +++ b/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/models/chat/MessageWithFunctionCall.kt @@ -6,5 +6,5 @@ data class MessageWithFunctionCall( val role: String, val content: String? = null, val functionCall: FunctionCall?, - val name: String? = Role.assistant.name + val name: String? = Role.ASSISTANT.name ) diff --git a/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/models/chat/Role.kt b/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/models/chat/Role.kt index 9f2bc1f4b..7eb3fa41b 100644 --- a/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/models/chat/Role.kt +++ b/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/models/chat/Role.kt @@ -1,8 +1,8 @@ package com.xebia.functional.xef.llm.models.chat enum class Role { - system, - user, - assistant, - function + SYSTEM, + USER, + ASSISTANT, + FUNCTION }