U
    h                     @  s<   d dl mZ d dlmZmZ d dlmZ G dd deZdS )    )annotations)AnyList)TextSplitterc                      s>   e Zd ZdZdddddd fdd	Zdd
dddZ  ZS )NLTKTextSplitterz"Splitting text using NLTK package.

englishstrr   None)	separatorlanguagekwargsreturnc                   sR   t  jf | zddlm} || _W n tk
r@   tdY nX || _|| _dS )zInitialize the NLTK splitter.r   )sent_tokenizezANLTK is not installed, please install it with `pip install nltk`.N)super__init__Znltk.tokenizer   
_tokenizerImportError
_separator	_language)selfr   r   r   r   	__class__ A/tmp/pip-unpacked-wheel-a648t6hw/langchain_text_splitters/nltk.pyr      s    

zNLTKTextSplitter.__init__z	List[str])textr   c                 C  s   | j || jd}| || jS )z&Split incoming text and return chunks.)r   )r   r   Z_merge_splitsr   )r   r   splitsr   r   r   
split_text   s    zNLTKTextSplitter.split_text)r   r   )__name__
__module____qualname____doc__r   r   __classcell__r   r   r   r   r      s
      r   N)
__future__r   typingr   r   Zlangchain_text_splitters.baser   r   r   r   r   r   <module>   s   