More small suggestions

- Replace blocks with equals / single expressions

Author: diesalbla

core/src/commonMain/kotlin/com/xebia/functional/xef/auto/CoreAIScope.kt

@@ -99,15 +99,14 @@ class CoreAIScope(
     functions: List<CFunction>,
     serializer: (json: String) -> A,
     promptConfiguration: PromptConfiguration,
-  ): A {
-    return prompt(
+  ): A =
+    prompt(
       prompt = Prompt(prompt),
       context = context,
       functions = functions,
       serializer = serializer,
       promptConfiguration = promptConfiguration,
     )
-  }
 
   @AiDsl
   suspend fun Chat.promptMessage(

core/src/commonMain/kotlin/com/xebia/functional/xef/llm/ChatWithFunctions.kt

@@ -24,16 +24,15 @@ interface ChatWithFunctions : Chat {
     functions: List<CFunction>,
     serializer: (json: String) -> A,
     promptConfiguration: PromptConfiguration,
-  ): A {
-    return tryDeserialize(serializer, promptConfiguration.maxDeserializationAttempts) {
+  ): A =
+    tryDeserialize(serializer, promptConfiguration.maxDeserializationAttempts) {
       promptMessage(
         prompt = Prompt(prompt),
         context = context,
         functions = functions,
         promptConfiguration
       )
     }
-  }
 
   @AiDsl
   suspend fun <A> prompt(
@@ -42,11 +41,10 @@ interface ChatWithFunctions : Chat {
     functions: List<CFunction>,
     serializer: (json: String) -> A,
     promptConfiguration: PromptConfiguration,
-  ): A {
-    return tryDeserialize(serializer, promptConfiguration.maxDeserializationAttempts) {
+  ): A =
+    tryDeserialize(serializer, promptConfiguration.maxDeserializationAttempts) {
       promptMessage(prompt = prompt, context = context, functions = functions, promptConfiguration)
     }
-  }
 
   private suspend fun <A> tryDeserialize(
     serializer: (json: String) -> A,

gpt4all-kotlin/src/jvmMain/kotlin/com/xebia/functional/gpt4all/GPT4All.kt

@@ -46,7 +46,8 @@ interface GPT4All : AutoCloseable, Chat, Completion {
 
       override suspend fun createCompletion(request: CompletionRequest): CompletionResult =
         with(request) {
-          val response: String = generateCompletion(prompt, generationConfig)
+          val response: String =
+            gpt4allModel.prompt(messages.buildPrompt(), llmModelContext(generationConfig))
           return CompletionResult(
             UUID.randomUUID().toString(),
             path.name,
@@ -59,8 +60,8 @@ interface GPT4All : AutoCloseable, Chat, Completion {
 
       override suspend fun createChatCompletion(request: ChatCompletionRequest): ChatCompletionResponse =
         with(request) {
-          val prompt: String = messages.buildPrompt()
-          val response: String = generateCompletion(prompt, generationConfig)
+          val response: String =
+            gpt4allModel.prompt(messages.buildPrompt(), llmModelContext(generationConfig))
           return ChatCompletionResponse(
             UUID.randomUUID().toString(),
             path.name,
@@ -71,9 +72,7 @@ interface GPT4All : AutoCloseable, Chat, Completion {
           )
         }
 
-      override fun tokensFromMessages(messages: List<Message>): Int {
-        return 0
-      }
+      override fun tokensFromMessages(messages: List<Message>): Int = 0
 
       override val name: String = path.name
 
@@ -92,31 +91,25 @@ interface GPT4All : AutoCloseable, Chat, Completion {
         return "$messages\n### Response:"
       }
 
-      private fun generateCompletion(
-        prompt: String,
-        generationConfig: GenerationConfig
-      ): String {
-        val context = LLModelContext(
-          logits_size = LibCAPI.size_t(generationConfig.logitsSize.toLong()),
-          tokens_size = LibCAPI.size_t(generationConfig.tokensSize.toLong()),
-          n_past = generationConfig.nPast,
-          n_ctx = generationConfig.nCtx,
-          n_predict = generationConfig.nPredict,
-          top_k = generationConfig.topK,
-          top_p = generationConfig.topP.toFloat(),
-          temp = generationConfig.temp.toFloat(),
-          n_batch = generationConfig.nBatch,
-          repeat_penalty = generationConfig.repeatPenalty.toFloat(),
-          repeat_last_n = generationConfig.repeatLastN,
-          context_erase = generationConfig.contextErase.toFloat()
-        )
-
-        return gpt4allModel.prompt(prompt, context)
-      }
+      private fun llmModelContext(generationConfig: GenerationConfig): String =
+        with(generationConfig) {
+          LLModelContext(
+            logits_size = LibCAPI.size_t(logitsSize.toLong()),
+            tokens_size = LibCAPI.size_t(tokensSize.toLong()),
+            n_past = nPast,
+            n_ctx = nCtx,
+            n_predict = nPredict,
+            top_k = topK,
+            top_p = topP.toFloat(),
+            temp = temp.toFloat(),
+            n_batch = nBatch,
+            repeat_penalty = repeatPenalty.toFloat(),
+            repeat_last_n = repeatLastN,
+            context_erase = contextErase.toFloat()
+          )
+        }
     }
 
-
   }
 }
 
-