From 6f8b0725ff8a713bca4c5c0a8a9abe608fea6de1 Mon Sep 17 00:00:00 2001
From: Vladislav Artiukhov <vladislav.artiukhov@jetbrains.com>
Date: Mon, 7 Oct 2024 14:57:28 +0200
Subject: [PATCH] refactor: apply ktlint

---
 .../testspark/core/generation/llm/prompt/PromptGenerator.kt | 2 +-
 .../testspark/tools/llm/generation/PromptManager.kt         | 6 ++----
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/core/src/main/kotlin/org/jetbrains/research/testspark/core/generation/llm/prompt/PromptGenerator.kt b/core/src/main/kotlin/org/jetbrains/research/testspark/core/generation/llm/prompt/PromptGenerator.kt
index ab767b07e..83c9c87f7 100644
--- a/core/src/main/kotlin/org/jetbrains/research/testspark/core/generation/llm/prompt/PromptGenerator.kt
+++ b/core/src/main/kotlin/org/jetbrains/research/testspark/core/generation/llm/prompt/PromptGenerator.kt
@@ -103,7 +103,7 @@ class PromptGenerator(
     fun generatePromptForLine(
         lineUnderTest: String,
         interestingClasses: List<ClassRepresentation>,
-        testSamplesCode: String
+        testSamplesCode: String,
     ): String {
         val prompt = PromptBuilder(promptTemplates.linePrompt)
             .insertLanguage(context.promptConfiguration.desiredLanguage)
diff --git a/src/main/kotlin/org/jetbrains/research/testspark/tools/llm/generation/PromptManager.kt b/src/main/kotlin/org/jetbrains/research/testspark/tools/llm/generation/PromptManager.kt
index 59eca489f..a020a09a3 100644
--- a/src/main/kotlin/org/jetbrains/research/testspark/tools/llm/generation/PromptManager.kt
+++ b/src/main/kotlin/org/jetbrains/research/testspark/tools/llm/generation/PromptManager.kt
@@ -153,8 +153,7 @@ class PromptManager(
                                 interestingClassesFromMethod,
                                 testSamplesCode,
                             )
-                        }
-                        else {
+                        } else {
                             return@Computable promptGenerator.generatePromptForLine(
                                 lineUnderTest,
                                 interestingClasses,
@@ -315,8 +314,7 @@ class PromptManager(
              * we already know that the current language is supported
              */
             return psiHelper!!.generateMethodDescriptor(containingPsiMethod)
-        }
-        else {
+        } else {
             /**
              * When no PSI class provided we are dealing with a top-level function.
              * Processing function outside the class