Source code for langchain_text_splitters.konlpy

from __future__ import annotations

from typing import Any, List

from langchain_text_splitters.base import TextSplitter


class KonlpyTextSplitter(TextSplitter):
    """使用Konlpy包来分割文本。

适用于分割韩文文本。
"""

    def __init__(
        self,
        separator: str = "\n\n",
        **kwargs: Any,
    ) -> None:
        """初始化Konlpy文本分割器。"""
        super().__init__(**kwargs)
        self._separator = separator
        try:
            from konlpy.tag import Kkma
        except ImportError:
            raise ImportError(
                """
                Konlpy is not installed, please install it with 
                `pip install konlpy`
                """
            )
        self.kkma = Kkma()

    def split_text(self, text: str) -> List[str]:
        """分割传入的文本并返回分块。"""
        splits = self.kkma.sentences(text)
        return self._merge_splits(splits, self._separator)