feat: remove before multiple \n

unit-mesh · Dec 18, 2023 · cc62f95 · cc62f95
1 parent 54c5f47
commit cc62f95
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/similar/JavaSimilarChunker.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/similar/JavaSimilarChunker.kt
@@ -54,6 +54,9 @@ class JavaSimilarChunker(private val fileTree: HashMap<String, InstructionFileJo
             similarChunks.take(3)
         } else {
             similarChunks
+        }.map {
+            // clean up the similar chunks, if start with multiple \n, remove to one
+            it.replace(Regex("^\\n+"), "\n")
         }
 
         return SimilarChunkContext("java", relatedCodePath, similarChunksText)

diff --git a/...cker/src/test/kotlin/cc/unitmesh/pick/prompt/strategy/SimilarChunksStrategyBuilderTest.kt b/...cker/src/test/kotlin/cc/unitmesh/pick/prompt/strategy/SimilarChunksStrategyBuilderTest.kt
@@ -99,7 +99,6 @@ public class HelloController {
 
         Assertions.assertEquals(
             first.similarChunks.joinToString("\n"), """
-
 @Repository
 public interface BlogRepository extends CrudRepository<BlogPost, Long> {