fix: fix related code format to instructions issues

unit-mesh · Dec 19, 2023 · a66d8b1 · a66d8b1
1 parent 59b194e
commit a66d8b1
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/prompt/strategy/RelatedCodeStrategyBuilder.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/prompt/strategy/RelatedCodeStrategyBuilder.kt
@@ -87,18 +87,15 @@ class RelatedCodeStrategyBuilder(private val context: JobContext) :
 
     override fun unique(list: List<RelatedCodeCompletionIns>): List<Instruction> {
         return list.map {
+            val relatedCode = if (it.relatedCode.isNotBlank() || it.relatedCode.isNotEmpty()) {
+                "\n// Compare this snippets: \n ```${it.language}\n${it.relatedCode}\n```"
+            } else {
+                ""
+            }
             Instruction(
                 instruction = "Complete ${it.language} code, return rest code, no explaining",
                 output = it.output,
-                input = """
-                |```${it.language}
-                |${it.relatedCode}
-                |```
-                |
-                |Code:
-                |```${it.language}
-                |${it.beforeCursor}
-                |```""".trimMargin()
+                input = "$relatedCode\n\nCode:\n```${it.language}\n${it.beforeCursor}\n```"
             )
         }
     }