diff --git a/bindings/python/Cargo.toml b/bindings/python/Cargo.toml index c644128..52473c5 100644 --- a/bindings/python/Cargo.toml +++ b/bindings/python/Cargo.toml @@ -22,6 +22,7 @@ tiktoken-rs = "0.5.9" tokenizers = { version = "0.19.1", features = ["http"] } tree-sitter = "0.22" tree-sitter-go = "0.21" +tree-sitter-java = "0.21" tree-sitter-md = "0.2" tree-sitter-python = "0.21" tree-sitter-rust = "0.21" diff --git a/bindings/python/src/language.rs b/bindings/python/src/language.rs index 327492e..b4e4ea8 100644 --- a/bindings/python/src/language.rs +++ b/bindings/python/src/language.rs @@ -6,6 +6,7 @@ use tree_sitter::Language as TreeSitterLanguage; #[non_exhaustive] pub enum Language { Golang, + Java, Markdown, Python, Rust, @@ -15,6 +16,7 @@ impl Language { pub fn as_tree_sitter_language(&self) -> TreeSitterLanguage { match self { Language::Golang => tree_sitter_go::language(), + Language::Java => tree_sitter_java::language(), Language::Markdown => tree_sitter_md::language(), Language::Python => tree_sitter_python::language(), Language::Rust => tree_sitter_rust::language(),