Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 4 additions & 5 deletions src/semble/chunking/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,17 +5,16 @@
from logging import getLogger

from tree_sitter import Node, Parser
from tree_sitter_language_pack import SupportedLanguage, get_parser, manifest_languages

logger = getLogger(__name__)
from tree_sitter_language_pack import SupportedLanguage, get_parser

from semble.index.files import ALL_LANGUAGES

_TREE_SITTER_LANGUAGES: frozenset[str] = frozenset(manifest_languages())
logger = getLogger(__name__)


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


@dataclass
Expand Down
6 changes: 3 additions & 3 deletions src/semble/index/files.py
Original file line number Diff line number Diff line change
Expand Up @@ -437,8 +437,8 @@ def _inv_mapping(mapping: dict[str, str]) -> dict[str, list[str]]:
return dict(inv)


_ALL_LANGUAGES = frozenset(_EXTENSION_TO_LANGUAGE.values())
_WITHOUT_DOC = _ALL_LANGUAGES - _DOC_LANGUAGES
ALL_LANGUAGES = frozenset(_EXTENSION_TO_LANGUAGE.values())
_WITHOUT_DOC = ALL_LANGUAGES - _DOC_LANGUAGES
_LANGUAGE_TO_EXTENSION = _inv_mapping(_EXTENSION_TO_LANGUAGE)


Expand All @@ -450,7 +450,7 @@ def detect_language(file_name: Path) -> str | None:
def get_extensions(include_text_files: bool, extensions: Sequence[str] | None) -> list[str]:
"""Returns a list of supported file extensions."""
if include_text_files:
languages = _ALL_LANGUAGES
languages = ALL_LANGUAGES
else:
languages = _WITHOUT_DOC
all_extensions: set[str] = set()
Expand Down
Loading