refactor: apply ktlint

JetBrains-Research · Oct 7, 2024 · 6f8b072 · 6f8b072
1 parent 4f7c5a3
commit 6f8b072
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/...ain/kotlin/org/jetbrains/research/testspark/core/generation/llm/prompt/PromptGenerator.kt b/...ain/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
@@ -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