diff --git a/code_extractor/languages.py b/code_extractor/languages.py index 5922562..7113785 100644 --- a/code_extractor/languages.py +++ b/code_extractor/languages.py @@ -5,7 +5,7 @@ import os from typing import Dict, Optional from tree_sitter import Language, Parser -from tree_sitter_languages import get_language, get_parser +from tree_sitter_language_pack import get_language, get_parser # Supported languages mapping @@ -123,4 +123,4 @@ def is_language_supported(language: str) -> bool: Returns: True if supported, False otherwise """ - return get_tree_sitter_parser(language) is not None \ No newline at end of file + return get_tree_sitter_parser(language) is not None diff --git a/code_extractor/search_engine.py b/code_extractor/search_engine.py index 00ba72b..61b24d6 100644 --- a/code_extractor/search_engine.py +++ b/code_extractor/search_engine.py @@ -10,7 +10,7 @@ import os import fnmatch from tree_sitter import Node, Query -from tree_sitter_languages import get_parser, get_language +from tree_sitter_language_pack import get_parser, get_language from .models import SearchResult, SearchParameters from .file_reader import get_file_content @@ -401,4 +401,4 @@ def _deduplicate_results(self, results: List[SearchResult]) -> List[SearchResult # Sort by file path, then by line number unique_results.sort(key=lambda r: (r.file_path, r.start_line)) - return unique_results \ No newline at end of file + return unique_results diff --git a/code_extractor/server.py b/code_extractor/server.py index a57abc2..b569f10 100644 --- a/code_extractor/server.py +++ b/code_extractor/server.py @@ -17,7 +17,7 @@ sys.exit(1) try: - from tree_sitter_languages import get_parser + from tree_sitter_language_pack import get_parser except ImportError: print("Error: tree-sitter-languages not installed. Install with: pip install tree-sitter-languages", file=sys.stderr) sys.exit(1) @@ -630,4 +630,4 @@ def search_code_tool( if __name__ == "__main__": - main() \ No newline at end of file + main() diff --git a/pyproject.toml b/pyproject.toml index 5f7c84d..0ad9b43 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -25,8 +25,8 @@ classifiers = [ ] dependencies = [ "mcp>=1.11.0", - "tree-sitter-languages>=1.10.2", - "tree-sitter==0.21.3", + "tree-sitter-language-pack>=0.9.0", + "tree-sitter==0.23.2", "requests>=2.31.0", "cachetools>=5.3.0", ]