use vyper cli to get search paths

vyperlang · Aug 10, 2024 · 100783f · 100783f
1 parent facadfe
commit 100783f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/vyper_lsp/ast.py b/vyper_lsp/ast.py
@@ -8,6 +8,7 @@
 from vyper.compiler import CompilerData
 from vyper.compiler.input_bundle import FilesystemInputBundle
 from vyper.utils import VyperException
+from vyper.cli.vyper_compile import get_search_paths
 import warnings
 import re
 
@@ -41,9 +42,9 @@ def build_ast(self, doc: Document) -> List[Diagnostic]:
         src = doc.source
         uri = doc.uri
         processed_uri = uri.replace("file://", "")
-        uri_path = Path(processed_uri)
-        uri_parent_path = uri_path.parent
-        compiler_data = CompilerData(src, input_bundle=FilesystemInputBundle([uri_parent_path]))
+        uri_parent_path = Path(processed_uri).parent
+        search_paths = get_search_paths([str(uri_parent_path)])
+        compiler_data = CompilerData(src, input_bundle=FilesystemInputBundle(search_paths))
         diagnostics = []
         replacements = {}
         warnings.simplefilter("always")