Update tree-sitter

openhands_aci/linter/impl/treesitter.py

@@ -2,9 +2,9 @@
 
 from grep_ast import TreeContext, filename_to_lang
 from grep_ast.parsers import PARSERS
-from tree_sitter_languages import get_parser
 
 from ..base import BaseLinter, LintResult
+from .treesitter_compat import get_parser
 
 # tree_sitter is throwing a FutureWarning
 warnings.simplefilter('ignore', category=FutureWarning)

openhands_aci/linter/impl/treesitter_compat.py

@@ -0,0 +1,23 @@
+"""Compatibility layer for tree-sitter 0.24.0."""
+
+import importlib
+from tree_sitter import Language, Parser
+
+# Cache of loaded languages
+_language_cache = {}
+
+
+def get_parser(language):
+    """Get a Parser object for the given language name."""
+    if language not in _language_cache:
+        # Try to import the language module
+        module_name = f"tree_sitter_{language}"
+        try:
+            module = importlib.import_module(module_name)
+            _language_cache[language] = Language(module.language())
+        except ImportError:
+            raise ValueError(
+                f"Language {language} is not supported. Please install {module_name} package."
+            )
+
+    return Parser(_language_cache[language])

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "openhands-aci"
-version = "0.2.2"
+version = "0.2.3"
 description = "An Agent-Computer Interface (ACI) designed for software development agents OpenHands."
 authors = ["OpenHands"]
 license = "MIT"
@@ -17,7 +17,11 @@ scipy = "*"
 networkx = "*"
 litellm = "*"
 gitpython = "*"
-tree-sitter = "0.21.3"
+tree-sitter = "^0.24.0"
+tree-sitter-python = "^0.23.6"
+tree-sitter-javascript = "^0.23.1"
+tree-sitter-typescript = "^0.23.2"
+tree-sitter-ruby = "^0.23.1"
 grep-ast = "0.3.3"
 diskcache = "^5.6.3"
 flake8 = "*"