





"""simple docstring"""





import json
from typing import TYPE_CHECKING, List, Optional, Tuple

from tokenizers import pre_tokenizers, processors

from ...tokenization_utils_base import AddedToken, BatchEncoding
from ...tokenization_utils_fast import PreTrainedTokenizerFast
from ...utils import logging
from .tokenization_blenderbot import BlenderbotTokenizer


if TYPE_CHECKING:
  from transformers.pipelines.conversational import Conversation

__A											=						logging.get_logger(__name__)


__A											=						{
    """vocab_file""": """vocab.json""",
    """merges_file""": """merges.txt""",
    """tokenizer_config_file""": """tokenizer_config.json""",
}

__A											=						{
    """vocab_file""": {"""facebook/blenderbot-3B""": """https://huggingface.co/facebook/blenderbot-3B/resolve/main/vocab.json"""},
    """merges_file""": {"""facebook/blenderbot-3B""": """https://huggingface.co/facebook/blenderbot-3B/resolve/main/merges.txt"""},
    """tokenizer_config_file""": {
        """facebook/blenderbot-3B""": """https://huggingface.co/facebook/blenderbot-3B/resolve/main/tokenizer_config.json"""
    },
}

__A											=						{"""facebook/blenderbot-3B""": 128}






class    _lowerCAmelCase  (	a			):
       """simple docstring"""




       __magic_name__	:List[Any]						    =   VOCAB_FILES_NAMES
       __magic_name__	:int						    =   PRETRAINED_VOCAB_FILES_MAP
       __magic_name__	:List[str]						    =   PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
       __magic_name__	:Optional[int]						    =   ["""input_ids""", """attention_mask"""]
       __magic_name__	:str						    =   BlenderbotTokenizer






       def __init__(       self       ,					__UpperCAmelCase=None       ,					__UpperCAmelCase=None       ,					__UpperCAmelCase=None       ,					__UpperCAmelCase="replace"       ,					__UpperCAmelCase="<s>"       ,					__UpperCAmelCase="</s>"       ,					__UpperCAmelCase="</s>"       ,					__UpperCAmelCase="<s>"       ,					__UpperCAmelCase="<unk>"       ,					__UpperCAmelCase="<pad>"       ,					__UpperCAmelCase="<mask>"       ,					__UpperCAmelCase=False       ,					__UpperCAmelCase=True       ,					**__UpperCAmelCase       ,					):



         '''simple docstring'''

         super().__init__(
             __UpperCAmelCase       ,					__UpperCAmelCase       ,					tokenizer_file=__UpperCAmelCase       ,					errors=__UpperCAmelCase       ,					bos_token=__UpperCAmelCase       ,					eos_token=__UpperCAmelCase       ,					sep_token=__UpperCAmelCase       ,					cls_token=__UpperCAmelCase       ,					unk_token=__UpperCAmelCase       ,					pad_token=__UpperCAmelCase       ,					mask_token=__UpperCAmelCase       ,					add_prefix_space=__UpperCAmelCase       ,					trim_offsets=__UpperCAmelCase       ,					**__UpperCAmelCase       ,					)

         lowerCAmelCase__     :List[str]											=				json.loads(self.backend_tokenizer.pre_tokenizer.__getstate__() )
         if pre_tok_state.get('add_prefix_space'       ,					__UpperCAmelCase ) != add_prefix_space:
           lowerCAmelCase__     :List[str]											=				getattr(__UpperCAmelCase       ,					pre_tok_state.pop('type' ) )
           lowerCAmelCase__     :int											=				add_prefix_space
           lowerCAmelCase__     :str											=				pre_tok_class(**__UpperCAmelCase )

         lowerCAmelCase__     :Dict											=				add_prefix_space

         lowerCAmelCase__     :Optional[Any]											=				'post_processor'
         lowerCAmelCase__     :List[Any]											=				getattr(self.backend_tokenizer       ,					__UpperCAmelCase       ,					__UpperCAmelCase )
         if tokenizer_component_instance:
           lowerCAmelCase__     :List[str]											=				json.loads(tokenizer_component_instance.__getstate__() )

           # The lists 'sep' and 'cls' must be cased in tuples for the object `post_processor_class`
           if "sep" in state:
             lowerCAmelCase__     :Union[str, Any]											=				tuple(state['sep'] )
           if "cls" in state:
             lowerCAmelCase__     :Optional[int]											=				tuple(state['cls'] )

           lowerCAmelCase__     :Optional[Any]											=				False

           if state.get('add_prefix_space'       ,					__UpperCAmelCase ) != add_prefix_space:
             lowerCAmelCase__     :Dict											=				add_prefix_space
             lowerCAmelCase__     :List[Any]											=				True

           if state.get('trim_offsets'       ,					__UpperCAmelCase ) != trim_offsets:
             lowerCAmelCase__     :Tuple											=				trim_offsets
             lowerCAmelCase__     :Any											=				True

           if changes_to_apply:
             lowerCAmelCase__     :int											=				getattr(__UpperCAmelCase       ,					state.pop('type' ) )
             lowerCAmelCase__     :Any											=				component_class(**__UpperCAmelCase )
             setattr(self.backend_tokenizer       ,					__UpperCAmelCase       ,					__UpperCAmelCase )






       @property
       # Copied from transformers.models.roberta.tokenization_roberta_fast.RobertaTokenizerFast.mask_token with Roberta->Blenderbot, RoBERTa->Blenderbot
       def 	snake_case		(       self ):



         '''simple docstring'''

         if self._mask_token is None:
           if self.verbose:
             logger.error('Using mask_token, but it is not set yet.' )
           return None
         return str(self._mask_token )






       @mask_token.setter
       def 	snake_case		(       self       ,					__UpperCAmelCase ):



         '''simple docstring'''

         lowerCAmelCase__     :List[Any]											=				AddedToken(__UpperCAmelCase       ,					lstrip=__UpperCAmelCase       ,					rstrip=__UpperCAmelCase ) if isinstance(__UpperCAmelCase       ,					__UpperCAmelCase ) else value
         lowerCAmelCase__     :List[str]											=				value






       def 	snake_case		(       self       ,					*__UpperCAmelCase       ,					**__UpperCAmelCase ):



         '''simple docstring'''

         lowerCAmelCase__     :str											=				kwargs.get('is_split_into_words'       ,					__UpperCAmelCase )
         assert self.add_prefix_space or not is_split_into_words, (
             F"You need to instantiate {self.__class__.__name__} with add_prefix_space=True "
             "to use it with pretokenized inputs."
         )

         return super()._batch_encode_plus(*__UpperCAmelCase       ,					**__UpperCAmelCase )






       def 	snake_case		(       self       ,					*__UpperCAmelCase       ,					**__UpperCAmelCase ):



         '''simple docstring'''

         lowerCAmelCase__     :List[str]											=				kwargs.get('is_split_into_words'       ,					__UpperCAmelCase )

         assert self.add_prefix_space or not is_split_into_words, (
             F"You need to instantiate {self.__class__.__name__} with add_prefix_space=True "
             "to use it with pretokenized inputs."
         )

         return super()._encode_plus(*__UpperCAmelCase       ,					**__UpperCAmelCase )






       def 	snake_case		(       self       ,					__UpperCAmelCase       ,					__UpperCAmelCase = None ):



         '''simple docstring'''

         lowerCAmelCase__     :List[str]											=				self._tokenizer.model.save(__UpperCAmelCase       ,					name=__UpperCAmelCase )
         return tuple(__UpperCAmelCase )






       def 	snake_case		(       self       ,					__UpperCAmelCase       ,					__UpperCAmelCase = None ):



         '''simple docstring'''

         lowerCAmelCase__     :Any											=				[self.sep_token_id]
         lowerCAmelCase__     :int											=				[self.cls_token_id]

         if token_ids_a is None:
           return len(cls + token_ids_a + sep ) * [0]
         return len(cls + token_ids_a + sep + sep + token_ids_a + sep ) * [0]






       def 	snake_case		(       self       ,					__UpperCAmelCase       ,					__UpperCAmelCase = None ):



         '''simple docstring'''

         return token_ids_a + [self.eos_token_id]






       def 	snake_case		(       self       ,					__UpperCAmelCase ):



         '''simple docstring'''

         lowerCAmelCase__     :str											=				[]
         for is_user, text in conversation.iter_texts():
           if is_user:
             # We need to space prefix as it's being done within blenderbot
             inputs.append(' ' + text )
           else:
             # Generated responses should contain them already.
             inputs.append(__UpperCAmelCase )

         lowerCAmelCase__     :Dict											=				'  '.join(__UpperCAmelCase )
         lowerCAmelCase__     :Tuple											=				self.encode(__UpperCAmelCase )
         if len(__UpperCAmelCase ) > self.model_max_length:
           lowerCAmelCase__     :Optional[Any]											=				input_ids[-self.model_max_length :]
           logger.warning(F"Trimmed input from conversation as it was longer than {self.model_max_length} tokens." )
         return input_ids







"""simple docstring"""





import unittest
from pathlib import Path
from tempfile import TemporaryDirectory

from transformers import AutoConfig, TFGPTaLMHeadModel, is_keras_nlp_available, is_tf_available
from transformers.models.gpta.tokenization_gpta import GPTaTokenizer
from transformers.testing_utils import require_keras_nlp, require_tf, slow


if is_tf_available():
  import tensorflow as tf

if is_keras_nlp_available():
  from transformers.models.gpta import TFGPTaTokenizer


__A											=						["""gpt2"""]
__A											=						"""gpt2"""

if is_tf_available():






  class    _lowerCAmelCase  (	tf.Module			):
         """simple docstring"""




         def __init__(       self       ,					__UpperCAmelCase ):



           '''simple docstring'''

           super().__init__()
           lowerCAmelCase__     :List[str]											=				tokenizer
           lowerCAmelCase__     :Optional[int]											=				AutoConfig.from_pretrained(__UpperCAmelCase )
           lowerCAmelCase__     :Optional[Any]											=				TFGPTaLMHeadModel.from_config(__UpperCAmelCase )






         @tf.function(input_signature=(tf.TensorSpec((None,)       ,					tf.string       ,					name='text' ),) )
         def 	snake_case		(       self       ,					__UpperCAmelCase ):



           '''simple docstring'''

           lowerCAmelCase__     :str											=				self.tokenizer(__UpperCAmelCase )
           lowerCAmelCase__     :int											=				tokenized['input_ids'].to_tensor()

           lowerCAmelCase__     :Optional[Any]											=				tf.cast(input_ids_dense > 0       ,					tf.intaa )
           # input_mask = tf.reshape(input_mask, [-1, MAX_SEQ_LEN])

           lowerCAmelCase__     :int											=				self.model(input_ids=__UpperCAmelCase       ,					attention_mask=__UpperCAmelCase )['logits']

           return outputs






@require_tf
@require_keras_nlp
class    _lowerCAmelCase  (	unittest.TestCase			):
       """simple docstring"""




       def 	snake_case		(       self ):



         '''simple docstring'''

         super().setUp()

         lowerCAmelCase__     :List[str]											=				[GPTaTokenizer.from_pretrained(__UpperCAmelCase ) for checkpoint in (TOKENIZER_CHECKPOINTS)]
         lowerCAmelCase__     :Union[str, Any]											=				[TFGPTaTokenizer.from_pretrained(__UpperCAmelCase ) for checkpoint in TOKENIZER_CHECKPOINTS]
         assert len(self.tokenizers ) == len(self.tf_tokenizers )

         lowerCAmelCase__     :int											=				[
             'This is a straightforward English test sentence.',
             'This one has some weird characters\rto\nsee\r\nif  those\u00E9break things.',
             'Now we\'re going to add some Chinese: 一 二 三 一二三',
             'And some much more rare Chinese: 齉 堃 齉堃',
             'Je vais aussi écrire en français pour tester les accents',
             'Classical Irish also has some unusual characters, so in they go: Gaelaċ, ꝼ',
         ]
         lowerCAmelCase__     :str											=				list(zip(self.test_sentences       ,					self.test_sentences[::-1] ) )






       def 	snake_case		(       self ):



         '''simple docstring'''

         for tokenizer, tf_tokenizer in zip(self.tokenizers       ,					self.tf_tokenizers ):
           for test_inputs in self.test_sentences:
             lowerCAmelCase__     :int											=				tokenizer([test_inputs]       ,					return_tensors='tf' )
             lowerCAmelCase__     :Optional[int]											=				tf_tokenizer([test_inputs] )

             for key in python_outputs.keys():
               # convert them to numpy to avoid messing with ragged tensors
               lowerCAmelCase__     :Optional[int]											=				python_outputs[key].numpy()
               lowerCAmelCase__     :List[str]											=				tf_outputs[key].numpy()

               self.assertTrue(tf.reduce_all(python_outputs_values.shape == tf_outputs_values.shape ) )
               self.assertTrue(tf.reduce_all(tf.cast(__UpperCAmelCase       ,					tf.intaa ) == tf_outputs_values ) )






       @slow
       def 	snake_case		(       self ):



         '''simple docstring'''

         for tf_tokenizer in self.tf_tokenizers:
           lowerCAmelCase__     :Optional[int]											=				tf.function(__UpperCAmelCase )
           for test_inputs in self.test_sentences:
             lowerCAmelCase__     :Any											=				tf.constant(__UpperCAmelCase )
             lowerCAmelCase__     :int											=				compiled_tokenizer(__UpperCAmelCase )
             lowerCAmelCase__     :Any											=				tf_tokenizer(__UpperCAmelCase )

             for key in eager_outputs.keys():
               self.assertTrue(tf.reduce_all(eager_outputs[key] == compiled_outputs[key] ) )






       @slow
       def 	snake_case		(       self ):



         '''simple docstring'''

         for tf_tokenizer in self.tf_tokenizers:
           lowerCAmelCase__     :List[str]											=				ModelToSave(tokenizer=__UpperCAmelCase )
           lowerCAmelCase__     :Union[str, Any]											=				tf.convert_to_tensor([self.test_sentences[0]] )
           lowerCAmelCase__     :List[str]											=				model.serving(__UpperCAmelCase )  # Build model with some sample inputs
           with TemporaryDirectory() as tempdir:
             lowerCAmelCase__     :Union[str, Any]											=				Path(__UpperCAmelCase ) / 'saved.model'
             tf.saved_model.save(__UpperCAmelCase       ,					__UpperCAmelCase       ,					signatures={'serving_default': model.serving} )
             lowerCAmelCase__     :str											=				tf.saved_model.load(__UpperCAmelCase )
           lowerCAmelCase__     :Dict											=				loaded_model.signatures['serving_default'](__UpperCAmelCase )['output_0']
           # We may see small differences because the loaded model is compiled, so we need an epsilon for the test
           self.assertTrue(tf.reduce_all(out == loaded_output ) )






       @slow
       def 	snake_case		(       self ):



         '''simple docstring'''

         for tf_tokenizer in self.tf_tokenizers:
           lowerCAmelCase__     :List[str]											=				tf.convert_to_tensor([self.test_sentences[0]] )
           lowerCAmelCase__     :List[str]											=				tf_tokenizer(__UpperCAmelCase )  # Build model with some sample inputs

           lowerCAmelCase__     :Union[str, Any]											=				tf_tokenizer.get_config()
           lowerCAmelCase__     :Tuple											=				TFGPTaTokenizer.from_config(__UpperCAmelCase )
           lowerCAmelCase__     :Tuple											=				model_from_config(__UpperCAmelCase )

           for key in from_config_output.keys():
             self.assertTrue(tf.reduce_all(from_config_output[key] == out[key] ) )






       @slow
       def 	snake_case		(       self ):



         '''simple docstring'''

         for tf_tokenizer in self.tf_tokenizers:
           # for the test to run
           lowerCAmelCase__     :int											=				1_2_3_1_2_3

           for max_length in [3, 5, 1_0_2_4]:
             lowerCAmelCase__     :Tuple											=				tf.convert_to_tensor([self.test_sentences[0]] )
             lowerCAmelCase__     :List[str]											=				tf_tokenizer(__UpperCAmelCase       ,					max_length=__UpperCAmelCase )

             lowerCAmelCase__     :Union[str, Any]											=				out['input_ids'].numpy().shape[1]

             assert out_length == max_length
