'''simple docstring'''




# Copyright 2023 The HuggingFace Inc. team. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from ..models.auto import AutoModelForSeqaSeqLM, AutoTokenizer
from .base import PipelineTool


__lowercase				   =  '''Here is a text containing a lot of information: \'\'\'{text}\'\'\'.

Can you answer this question about the text: \'{question}\''''






class     a__( lowerCAmelCase__     ):


			'''simple docstring'''






			UpperCAmelCase_      :					str								=	'''google/flan-t5-base'''
			UpperCAmelCase_      :					Dict								=	(
			    '''This is a tool that answers questions related to a text. It takes two arguments named `text`, which is the '''
			    '''text where to find the answer, and `question`, which is the question, and returns the answer to the question.'''
			)
			UpperCAmelCase_      :					List[str]								=	'''text_qa'''
			UpperCAmelCase_      :					str								=	AutoTokenizer
			UpperCAmelCase_      :					Dict								=	AutoModelForSeqaSeqLM

			UpperCAmelCase_      :					Dict								=	['''text''', '''text''']
			UpperCAmelCase_      :					Tuple								=	['''text''']



			def 							a_	(       self	,   __lowerCAmelCase	,   __lowerCAmelCase):




					"""simple docstring"""


					lowerCAmelCase        =  QA_PROMPT.format(text=__lowerCAmelCase	,   question=__lowerCAmelCase)
					return self.pre_processor(__lowerCAmelCase	,   return_tensors="""pt""")



			def 							a_	(       self	,   __lowerCAmelCase):




					"""simple docstring"""


					lowerCAmelCase        =  self.model.generate(**__lowerCAmelCase)

					lowerCAmelCase ,					lowerCAmelCase        =  inputs["""input_ids"""].shape
					lowerCAmelCase        =  output_ids.shape[0]

					return output_ids.reshape(__lowerCAmelCase	,   out_b // in_b	,   *output_ids.shape[1:])[0][0]







			def 							a_	(       self	,   __lowerCAmelCase):




					"""simple docstring"""


					return self.pre_processor.decode(__lowerCAmelCase	,   skip_special_tokens=__lowerCAmelCase	,   clean_up_tokenization_spaces=__lowerCAmelCase)





'''simple docstring'''




from collections import OrderedDict
from typing import Mapping

from ...configuration_utils import PretrainedConfig
from ...onnx import OnnxConfig
from ...utils import logging


__lowercase				   =  logging.get_logger(__name__)

__lowercase				   =  {
    '''flaubert/flaubert_small_cased''': '''https://huggingface.co/flaubert/flaubert_small_cased/resolve/main/config.json''',
    '''flaubert/flaubert_base_uncased''': '''https://huggingface.co/flaubert/flaubert_base_uncased/resolve/main/config.json''',
    '''flaubert/flaubert_base_cased''': '''https://huggingface.co/flaubert/flaubert_base_cased/resolve/main/config.json''',
    '''flaubert/flaubert_large_cased''': '''https://huggingface.co/flaubert/flaubert_large_cased/resolve/main/config.json''',
}






class     a__( lowerCAmelCase__     ):


			'''simple docstring'''






			UpperCAmelCase_      :					int								=	'''flaubert'''
			UpperCAmelCase_      :					Optional[Any]								=	{
			    '''hidden_size''': '''emb_dim''',
			    '''num_attention_heads''': '''n_heads''',
			    '''num_hidden_layers''': '''n_layers''',
			    '''n_words''': '''vocab_size''',  # For backward compatibility
			}



			def __init__(       self	,   __lowerCAmelCase=False	,   __lowerCAmelCase=0.0	,   __lowerCAmelCase=30145	,   __lowerCAmelCase=2048	,   __lowerCAmelCase=12	,   __lowerCAmelCase=16	,   __lowerCAmelCase=0.1	,   __lowerCAmelCase=0.1	,   __lowerCAmelCase=True	,   __lowerCAmelCase=False	,   __lowerCAmelCase=False	,   __lowerCAmelCase=False	,   __lowerCAmelCase=1	,   __lowerCAmelCase=True	,   __lowerCAmelCase=512	,   __lowerCAmelCase=2048**-0.5	,   __lowerCAmelCase=1E-1_2	,   __lowerCAmelCase=0.02	,   __lowerCAmelCase=0	,   __lowerCAmelCase=1	,   __lowerCAmelCase=2	,   __lowerCAmelCase=3	,   __lowerCAmelCase=5	,   __lowerCAmelCase=True	,   __lowerCAmelCase="first"	,   __lowerCAmelCase=True	,   __lowerCAmelCase=None	,   __lowerCAmelCase=True	,   __lowerCAmelCase=0.1	,   __lowerCAmelCase=5	,   __lowerCAmelCase=5	,   __lowerCAmelCase=0	,   __lowerCAmelCase=0	,   __lowerCAmelCase=2	,   __lowerCAmelCase=0	,   **__lowerCAmelCase	,   ):




					"""simple docstring"""


					lowerCAmelCase        =  pre_norm
					lowerCAmelCase        =  layerdrop
					lowerCAmelCase        =  vocab_size
					lowerCAmelCase        =  emb_dim
					lowerCAmelCase        =  n_layers
					lowerCAmelCase        =  n_heads
					lowerCAmelCase        =  dropout
					lowerCAmelCase        =  attention_dropout
					lowerCAmelCase        =  gelu_activation
					lowerCAmelCase        =  sinusoidal_embeddings
					lowerCAmelCase        =  causal
					lowerCAmelCase        =  asm
					lowerCAmelCase        =  n_langs
					lowerCAmelCase        =  use_lang_emb
					lowerCAmelCase        =  layer_norm_eps
					lowerCAmelCase        =  bos_index
					lowerCAmelCase        =  eos_index
					lowerCAmelCase        =  pad_index
					lowerCAmelCase        =  unk_index
					lowerCAmelCase        =  mask_index
					lowerCAmelCase        =  is_encoder
					lowerCAmelCase        =  max_position_embeddings
					lowerCAmelCase        =  embed_init_std
					lowerCAmelCase        =  init_std
					lowerCAmelCase        =  summary_type
					lowerCAmelCase        =  summary_use_proj
					lowerCAmelCase        =  summary_activation
					lowerCAmelCase        =  summary_proj_to_labels
					lowerCAmelCase        =  summary_first_dropout
					lowerCAmelCase        =  start_n_top
					lowerCAmelCase        =  end_n_top
					lowerCAmelCase        =  mask_token_id
					lowerCAmelCase        =  lang_id

					if "n_words" in kwargs:
							lowerCAmelCase        =  kwargs["""n_words"""]

					super().__init__(pad_token_id=__lowerCAmelCase	,   bos_token_id=__lowerCAmelCase	,   **__lowerCAmelCase)







class     a__( lowerCAmelCase__     ):


			'''simple docstring'''






			@property
			def 							a_	(       self):




					"""simple docstring"""


					if self.task == "multiple-choice":
							lowerCAmelCase        =  {0: """batch""", 1: """choice""", 2: """sequence"""}
					else:
							lowerCAmelCase        =  {0: """batch""", 1: """sequence"""}
					return OrderedDict(
					    [
					        ("""input_ids""", dynamic_axis),
					        ("""attention_mask""", dynamic_axis),
					    ])




