fixing tree-sitter bug

Rangeet Pan · Rangeet Pan · commit 00b486620d8d · 2024-10-17T17:33:15.000-04:00
diff --git a/cldk/models/treesitter/models.py b/cldk/models/treesitter/models.py
@@ -28,7 +28,9 @@ class Capture:
         name: str
 
     def __init__(self, captures: List[Tuple[Node, str]]):
-        self.captures = [self.Capture(node=node, name=text) for node, text in captures]
+        self.captures = []
+        for capture_name, captures in captures.items():
+            self.captures = [self.Capture(node=node, name=capture_name) for node in captures]
 
     def __getitem__(self, index: int) -> Capture:
         """Get the capture at the specified index.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,7 +15,7 @@ pandas = "^2.2.0"
 networkx = "^3.2.1"
 pyarrow = "^15.0.0"
 tree-sitter-languages = "^1.10.2"
-tree-sitter = "^0.22.3"
+tree-sitter = "0.23.0"
 rich = "^13.7.1"
 wget = "^3.2"
 requests = "^2.31.0"
