Skip to content

Commit f02622f

Browse files
authored
feat: remove tree sitter language manifest (#102)
1 parent ae39ec5 commit f02622f

2 files changed

Lines changed: 7 additions & 8 deletions

File tree

src/semble/chunking/core.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5,17 +5,16 @@
55
from logging import getLogger
66

77
from tree_sitter import Node, Parser
8-
from tree_sitter_language_pack import SupportedLanguage, get_parser, manifest_languages
9-
10-
logger = getLogger(__name__)
8+
from tree_sitter_language_pack import SupportedLanguage, get_parser
119

10+
from semble.index.files import ALL_LANGUAGES
1211

13-
_TREE_SITTER_LANGUAGES: frozenset[str] = frozenset(manifest_languages())
12+
logger = getLogger(__name__)
1413

1514

1615
def is_supported_language(language: str) -> bool:
1716
"""Check if the language is supported by tree-sitter."""
18-
return language in _TREE_SITTER_LANGUAGES
17+
return language in ALL_LANGUAGES
1918

2019

2120
@dataclass

src/semble/index/files.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -437,8 +437,8 @@ def _inv_mapping(mapping: dict[str, str]) -> dict[str, list[str]]:
437437
return dict(inv)
438438

439439

440-
_ALL_LANGUAGES = frozenset(_EXTENSION_TO_LANGUAGE.values())
441-
_WITHOUT_DOC = _ALL_LANGUAGES - _DOC_LANGUAGES
440+
ALL_LANGUAGES = frozenset(_EXTENSION_TO_LANGUAGE.values())
441+
_WITHOUT_DOC = ALL_LANGUAGES - _DOC_LANGUAGES
442442
_LANGUAGE_TO_EXTENSION = _inv_mapping(_EXTENSION_TO_LANGUAGE)
443443

444444

@@ -450,7 +450,7 @@ def detect_language(file_name: Path) -> str | None:
450450
def get_extensions(include_text_files: bool, extensions: Sequence[str] | None) -> list[str]:
451451
"""Returns a list of supported file extensions."""
452452
if include_text_files:
453-
languages = _ALL_LANGUAGES
453+
languages = ALL_LANGUAGES
454454
else:
455455
languages = _WITHOUT_DOC
456456
all_extensions: set[str] = set()

0 commit comments

Comments
 (0)