fix: fix import issue

unit-mesh · Dec 7, 2023 · 68e0ba3 · 68e0ba3
1 parent b71938e
commit 68e0ba3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/unit-picker/src/main/kotlin/cc/unitmesh/pick/prompt/builder/RelatedCodeCompletionBuilder.kt b/unit-picker/src/main/kotlin/cc/unitmesh/pick/prompt/builder/RelatedCodeCompletionBuilder.kt
@@ -4,6 +4,7 @@ import cc.unitmesh.pick.prompt.Instruction
 import cc.unitmesh.pick.prompt.InstructionBuilder
 import cc.unitmesh.pick.prompt.InstructionContext
 import chapi.domain.core.CodeDataStruct
+import chapi.domain.core.CodePosition
 
 class RelatedCodeCompletionBuilder(private val context: InstructionContext) : InstructionBuilder {
     private val instruction: String = "";
@@ -16,7 +17,7 @@ class RelatedCodeCompletionBuilder(private val context: InstructionContext) : In
         val container = context.job.container ?: return emptyList()
 
         val relatedDataStructure = container.Imports.mapNotNull {
-            context.fileTree[it.AsName]?.container?.DataStructures
+            context.fileTree[it.Source]?.container?.DataStructures
         }.flatten()
 
         val relatedCode = relatedDataStructure.joinToString("\n") {