Source code for langchain.document_loaders.parsers.language.code_segmenter

from abc import ABC, abstractmethod
from typing import List



[docs]class CodeSegmenter(ABC):
    """The abstract class for the code segmenter."""

    def __init__(self, code: str):
        self.code = code


[docs]    def is_valid(self) -> bool:
        return True



[docs]    @abstractmethod
    def simplify_code(self) -> str:
        raise NotImplementedError()  # pragma: no cover



[docs]    @abstractmethod
    def extract_functions_classes(self) -> List[str]:
        raise NotImplementedError()  # pragma: no cover