Source code for langchain_community.document_loaders.parsers.language.rust

from typing import TYPE_CHECKING

from langchain_community.document_loaders.parsers.language.tree_sitter_segmenter import (  # noqa: E501
    TreeSitterSegmenter,
)

if TYPE_CHECKING:
    from tree_sitter import Language


CHUNK_QUERY = """
    [
        (function_item
            name: (identifier)
            body: (block)) @function
        (struct_item) @struct
        (trait_item) @trait
    ]
""".strip()


[docs]class RustSegmenter(TreeSitterSegmenter):
    """Code segmenter for Rust."""

[docs]    def get_language(self) -> "Language":
        from tree_sitter_languages import get_language

        return get_language("rust")

[docs]    def get_chunk_query(self) -> str:
        return CHUNK_QUERY

[docs]    def make_line_comment(self, text: str) -> str:
        return f"// {text}"