



'''simple docstring'''







from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_tf_available,
    is_tokenizers_available,
    is_torch_available,
)


lowercase__     :  Union[str, Any]    					=			{
    'configuration_dpr': ['DPR_PRETRAINED_CONFIG_ARCHIVE_MAP', 'DPRConfig'],
    'tokenization_dpr': [
        'DPRContextEncoderTokenizer',
        'DPRQuestionEncoderTokenizer',
        'DPRReaderOutput',
        'DPRReaderTokenizer',
    ],
}


try:
      if not is_tokenizers_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      lowercase__     :  Tuple    					=			[
          'DPRContextEncoderTokenizerFast',
          'DPRQuestionEncoderTokenizerFast',
          'DPRReaderTokenizerFast',
      ]

try:
      if not is_torch_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      lowercase__     :  Optional[int]    					=			[
          'DPR_CONTEXT_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST',
          'DPR_QUESTION_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST',
          'DPR_READER_PRETRAINED_MODEL_ARCHIVE_LIST',
          'DPRContextEncoder',
          'DPRPretrainedContextEncoder',
          'DPRPreTrainedModel',
          'DPRPretrainedQuestionEncoder',
          'DPRPretrainedReader',
          'DPRQuestionEncoder',
          'DPRReader',
      ]

try:
      if not is_tf_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      lowercase__     :  Optional[Any]    					=			[
          'TF_DPR_CONTEXT_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST',
          'TF_DPR_QUESTION_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST',
          'TF_DPR_READER_PRETRAINED_MODEL_ARCHIVE_LIST',
          'TFDPRContextEncoder',
          'TFDPRPretrainedContextEncoder',
          'TFDPRPretrainedQuestionEncoder',
          'TFDPRPretrainedReader',
          'TFDPRQuestionEncoder',
          'TFDPRReader',
      ]


if TYPE_CHECKING:
      from .configuration_dpr import DPR_PRETRAINED_CONFIG_ARCHIVE_MAP, DPRConfig
      from .tokenization_dpr import (
          DPRContextEncoderTokenizer,
          DPRQuestionEncoderTokenizer,
          DPRReaderOutput,
          DPRReaderTokenizer,
      )

      try:
            if not is_tokenizers_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .tokenization_dpr_fast import (
                DPRContextEncoderTokenizerFast,
                DPRQuestionEncoderTokenizerFast,
                DPRReaderTokenizerFast,
            )

      try:
            if not is_torch_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .modeling_dpr import (
                DPR_CONTEXT_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST,
                DPR_QUESTION_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST,
                DPR_READER_PRETRAINED_MODEL_ARCHIVE_LIST,
                DPRContextEncoder,
                DPRPretrainedContextEncoder,
                DPRPreTrainedModel,
                DPRPretrainedQuestionEncoder,
                DPRPretrainedReader,
                DPRQuestionEncoder,
                DPRReader,
            )

      try:
            if not is_tf_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .modeling_tf_dpr import (
                TF_DPR_CONTEXT_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST,
                TF_DPR_QUESTION_ENCODER_PRETRAINED_MODEL_ARCHIVE_LIST,
                TF_DPR_READER_PRETRAINED_MODEL_ARCHIVE_LIST,
                TFDPRContextEncoder,
                TFDPRPretrainedContextEncoder,
                TFDPRPretrainedQuestionEncoder,
                TFDPRPretrainedReader,
                TFDPRQuestionEncoder,
                TFDPRReader,
            )

else:
      import sys

      lowercase__     :  Tuple    					=			_LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)









'''simple docstring'''







import unittest

from transformers import load_tool
from transformers.tools.agent_types import AGENT_TYPE_MAPPING

from .test_tools_common import ToolTesterMixin, output_types

class       lowerCamelCase_							(	unittest.TestCase					,    lowerCAmelCase_  ):






   '''simple docstring'''

   def 	_UpperCamelCase       (  self ):
       UpperCamelCase__      =			load_tool("translation" )
       self.tool.setup()
       UpperCamelCase__      =			load_tool("translation"     ,						remote=__UpperCamelCase )


   def 	_UpperCamelCase       (  self ):
       UpperCamelCase__      =			self.tool("Hey, what's up?"     ,						src_lang="English"     ,						tgt_lang="French" )
       self.assertEqual(__UpperCamelCase     ,						"- Hé, comment ça va?" )


   def 	_UpperCamelCase       (  self ):
       UpperCamelCase__      =			self.remote_tool("Hey, what's up?"     ,						src_lang="English"     ,						tgt_lang="French" )
       self.assertEqual(__UpperCamelCase     ,						"- Hé, comment ça va?" )


   def 	_UpperCamelCase       (  self ):
       UpperCamelCase__      =			self.tool(text="Hey, what's up?"     ,						src_lang="English"     ,						tgt_lang="French" )
       self.assertEqual(__UpperCamelCase     ,						"- Hé, comment ça va?" )


   def 	_UpperCamelCase       (  self ):
       UpperCamelCase__      =			self.remote_tool(text="Hey, what's up?"     ,						src_lang="English"     ,						tgt_lang="French" )
       self.assertEqual(__UpperCamelCase     ,						"- Hé, comment ça va?" )


   def 	_UpperCamelCase       (  self ):
       UpperCamelCase__      =			["Hey, what's up?", "English", "Spanish"]
       UpperCamelCase__      =			self.tool(*__UpperCamelCase )

       # There is a single output
       if len(self.tool.outputs ) == 1:
           UpperCamelCase__      =			[outputs]

       self.assertListEqual(output_types(__UpperCamelCase )     ,						self.tool.outputs )


   def 	_UpperCamelCase       (  self ):
       UpperCamelCase__      =			["Hey, what's up?", "English", "Spanish"]
       UpperCamelCase__      =			self.tool(*__UpperCamelCase )

       if not isinstance(__UpperCamelCase     ,						__UpperCamelCase ):
           UpperCamelCase__      =			[outputs]

       self.assertEqual(len(__UpperCamelCase )     ,						len(self.tool.outputs ) )

       for output, output_type in zip(__UpperCamelCase     ,						self.tool.outputs ):
           UpperCamelCase__      =			AGENT_TYPE_MAPPING[output_type]
           self.assertTrue(isinstance(__UpperCamelCase     ,						__UpperCamelCase ) )







   def 	_UpperCamelCase       (  self ):
       UpperCamelCase__      =			["Hey, what's up?", "English", "Spanish"]

       UpperCamelCase__      =			[]

       for _input, input_type in zip(__UpperCamelCase     ,						self.tool.inputs ):
           if isinstance(__UpperCamelCase     ,						__UpperCamelCase ):
               _inputs.append([AGENT_TYPE_MAPPING[_input_type](_input ) for _input_type in input_type] )
           else:
               _inputs.append(AGENT_TYPE_MAPPING[input_type](_input ) )

        # Should not raise an error
       UpperCamelCase__      =			self.tool(*__UpperCamelCase )

       if not isinstance(__UpperCamelCase     ,						__UpperCamelCase ):
           UpperCamelCase__      =			[outputs]

       self.assertEqual(len(__UpperCamelCase )     ,						len(self.tool.outputs ) )




