
    +h                    N    S SK Jr  S SKJrJrJr  S SKJrJrJ	r	   " S S\5      r
g)    )annotations)AnyOptionalcast)TextSplitter	Tokenizersplit_text_on_tokensc                     ^  \ rS rSr% Sr   S         SU 4S jjjr    SS jrSS jrSS jrSr	S\
S	'   SS
 jrSrU =r$ )%SentenceTransformersTokenTextSplitter   z8Splitting text to tokens using sentence model tokenizer.intc                
  > [         TU ]  " S0 UDSU0D6   SSKJn  X l        U" U R
                  5      U l        U R                  R                  U l        U R                  US9  g! [         a  nSn[	        U5      UeSnAff = f)zCreate a new TextSplitter.chunk_overlapr   )SentenceTransformerzCould not import sentence_transformers python package. This is needed in order to for SentenceTransformersTokenTextSplitter. Please install it with `pip install sentence-transformers`.N)tokens_per_chunk )	super__init__sentence_transformersr   ImportError
model_name_model	tokenizer_initialize_chunk_configuration)	selfr   r   r   kwargsr   errmsg	__class__s	           ^/root/34ku/venv/lib/python3.13/site-packages/langchain_text_splitters/sentence_transformers.pyr   .SentenceTransformersTokenTextSplitter.__init__   s     	?6??	,A %)$//:..,,>N,O  	,N 
 c"+	,s   A% %
B/A==Bc                  U R                   R                  U l        Uc  U R                  U l        OXl        U R                  U R                  :  a5  SU R                   SU R                   SU R                   S3n[        U5      eg )NzThe token limit of the models 'z' is: z. Argument tokens_per_chunk=z > maximum token limit.)r   max_seq_lengthmaximum_tokens_per_chunkr   r   
ValueError)r   r   r   s      r    r   ESentenceTransformersTokenTextSplitter._initialize_chunk_configuration$   s     )-(B(B%#$($A$AD!$4!  4#@#@@1$//1B C556 7..2.C.C-D)+  S/! A    c                   ^  SU 4S jjn[        T R                  T R                  T R                  R                  US9n[        XS9$ )a  Splits the input text into smaller components by splitting text on tokens.

This method encodes the input text using a private `_encode` method, then
strips the start and stop token IDs from the encoded result. It returns the
processed segments as a list of strings.

Args:
    text (str): The input text to be split.

Returns:
    List[str]: A list of string components derived from the input text after
    encoding and processing.
c                ,   > TR                  U 5      SS $ )N   )_encode)textr   s    r    %encode_strip_start_and_stop_token_ids_SentenceTransformersTokenTextSplitter.split_text.<locals>.encode_strip_start_and_stop_token_idsF   s    <<%a++r'   )r   r   decodeencode)r-   r   r-   strreturn	list[int])r   _chunk_overlapr   r   r0   r	   )r   r-   r.   r   s   `   r    
split_text0SentenceTransformersTokenTextSplitter.split_text7   sC    	, --!22>>((8	
	 $CCr'   c               6    [        U R                  U5      5      $ )aA  Counts the number of tokens in the given text.

This method encodes the input text using a private `_encode` method and
calculates the total number of tokens in the encoded result.

Args:
    text (str): The input text for which the token count is calculated.

Returns:
    int: The number of tokens in the encoded text.
)lenr,   )r   r-   s     r    count_tokens2SentenceTransformersTokenTextSplitter.count_tokensR   s     4<<%&&r'   l         _max_length_equal_32_bit_integerc                d    U R                   R                  UU R                  SS9n[        SU5      $ )Ndo_not_truncate)
max_length
truncationr5   )r   r1   r=   r   )r   r-   &token_ids_with_start_and_end_token_idss      r    r,   -SentenceTransformersTokenTextSplitter._encodeb   s<    151F1F<<( 2G 2
.
 K!GHHr'   )r   r$   r   r   r   )2   z'sentence-transformers/all-mpnet-base-v2N)
r   r   r   r3   r   Optional[int]r   r   r4   None)r   rE   r4   rF   )r-   r3   r4   z	list[str])r-   r3   r4   r   r2   )__name__
__module____qualname____firstlineno____doc__r   r   r7   r;   r=   __annotations__r,   __static_attributes____classcell__)r   s   @r    r   r      s    B  C*.	PP P (	P
 P 
P P2"#0"	"&D6' -2$c1I Ir'   r   N)
__future__r   typingr   r   r   langchain_text_splitters.baser   r   r	   r   r   r'   r    <module>rR      s$    " & & W W`IL `Ir'   