
    @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    Sn[	        U5      e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`.)tokens_per_chunkN )	super__init__sentence_transformersr   ImportError
model_name_model	tokenizer_initialize_chunk_configuration)selfr   r   r   kwargsr   msg	__class__s          f/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_text_splitters/sentence_transformers.pyr   .SentenceTransformersTokenTextSplitter.__init__   s     	?6??	#A %)$//:..,,>N,O  	#N 
 c""	#s   A% %A=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   r/   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
truncationr4   )r   r0   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   r2   r   Optional[int]r   r   r3   None)r   rD   r3   rE   )r,   r2   r3   z	list[str])r,   r2   r3   r   r1   )__name__
__module____qualname____firstlineno____doc__r   r   r6   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>rQ      s$    " & & W W`IL `Ir&   