fix: fix tests

unit-mesh · Dec 16, 2023 · 15683a0 · 15683a0
1 parent 4db437c
commit 15683a0
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 6 deletions.
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/prompt/CodeContextStrategy.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/prompt/CodeContextStrategy.kt
@@ -47,11 +47,15 @@ enum class CompletionType {
     // TODO: support in future for this
     AFTER_BLOCK_COMPLETION;
 
-    fun builder(context: JobContext): CodeContextBuilder<out Any> {
+    fun builder(context: JobContext): InstructionBuilder {
         return mapOf(
             INLINE_COMPLETION to InlineCodeCompletionBuilder(context),
             IN_BLOCK_COMPLETION to InBlockCodeCompletionBuilder(context),
             AFTER_BLOCK_COMPLETION to AfterBlockCodeCompletionBuilder(context),
         )[this] ?: throw SerializationException("Unknown message type: $this")
     }
+
+    fun builders(types: List<CompletionType>, context: JobContext) : List<InstructionBuilder> {
+        return types.map { it.builder(context) }
+    }
 }
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/prompt/InstructionBuilder.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/prompt/InstructionBuilder.kt
@@ -17,11 +17,11 @@ data class CodeCompletionIns(
 }
 
 interface InstructionBuilder {
-    fun buildDataStruct(dataStruct: CodeDataStruct): List<CodeCompletionIns> {
+    fun build(dataStruct: CodeDataStruct): List<CodeCompletionIns> {
         return listOf()
     }
 
-    fun buildFun(function: CodeFunction): List<CodeCompletionIns>
+    fun build(function: CodeFunction): List<CodeCompletionIns>
 }
 
 
diff --git a/...picker/src/main/kotlin/cc/unitmesh/pick/prompt/builder/AfterBlockCodeCompletionBuilder.kt b/...picker/src/main/kotlin/cc/unitmesh/pick/prompt/builder/AfterBlockCodeCompletionBuilder.kt
@@ -7,7 +7,7 @@ import chapi.domain.core.CodeFunction
 
 
 class AfterBlockCodeCompletionBuilder(val context: JobContext) : InstructionBuilder {
-    override fun buildFun(function: CodeFunction): List<CodeCompletionIns> {
+    override fun build(function: CodeFunction): List<CodeCompletionIns> {
         TODO("Not yet implemented")
     }
 }
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/prompt/builder/InBlockCodeCompletionBuilder.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/prompt/builder/InBlockCodeCompletionBuilder.kt
@@ -6,7 +6,7 @@ import cc.unitmesh.pick.prompt.JobContext
 import chapi.domain.core.CodeFunction
 
 class InBlockCodeCompletionBuilder(val context: JobContext) : InstructionBuilder {
-    override fun buildFun(function: CodeFunction): List<CodeCompletionIns> {
+    override fun build(function: CodeFunction): List<CodeCompletionIns> {
         TODO("Not yet implemented")
     }
 

diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/prompt/builder/InlineCodeCompletionBuilder.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/prompt/builder/InlineCodeCompletionBuilder.kt
@@ -6,7 +6,7 @@ import cc.unitmesh.pick.prompt.JobContext
 import chapi.domain.core.CodeFunction
 
 class InlineCodeCompletionBuilder(val context: JobContext) : InstructionBuilder {
-    override fun buildFun(function: CodeFunction): List<CodeCompletionIns> {
+    override fun build(function: CodeFunction): List<CodeCompletionIns> {
         TODO("Not yet implemented")
     }