




# 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.
import re

from ..utils import cached_file


# docstyle-ignore
__magic_name__  :  List[str]					=       "\nHuman: <<task>>\n\nAssistant: "


__magic_name__  :  Tuple					=       "huggingface-tools/default-prompts"
__magic_name__  :  Optional[Any]					=       {"chat": "chat_prompt_template.txt", "run": "run_prompt_template.txt"}
def 		a_							( __lowerCAmelCase   , __lowerCAmelCase   , __lowerCAmelCase="run"				):
  if prompt_or_repo_id is None:
    lowerCAmelCase__										=      DEFAULT_PROMPTS_REPO

  # prompt is considered a repo ID when it does not contain any kind of space
  if re.search('''\\s'''   , A__				) is not None:
    return prompt_or_repo_id

  lowerCAmelCase__										=      cached_file(
      A__   , PROMPT_FILES[mode]   , repo_type='''dataset'''   , user_agent={'''agent''': agent_name}				)
  with open(A__   , '''r'''   , encoding='''utf-8'''				) as f:
    return f.read()






import pytest
import requests

from datasets.utils.file_utils import http_head

from .utils import OfflineSimulationMode, RequestWouldHangIndefinitelyError, offline

@pytest.mark.integration
def 	_lowerCAmelCase (    ):
      with offline(OfflineSimulationMode.CONNECTION_TIMES_OUT						):
            with pytest.raises(A__						):
                  requests.request('GET'    ,	'https://huggingface.co'						)
            with pytest.raises(requests.exceptions.ConnectTimeout						):
                  requests.request('GET'    ,	'https://huggingface.co'    ,	timeout=1.0						)

@pytest.mark.integration
def 	_lowerCAmelCase (    ):
      with offline(OfflineSimulationMode.CONNECTION_FAILS						):
            with pytest.raises(requests.exceptions.ConnectionError						):
                  requests.request('GET'    ,	'https://huggingface.co'						)

def 	_lowerCAmelCase (    ):
      with offline(OfflineSimulationMode.HF_DATASETS_OFFLINE_SET_TO_1						):
            with pytest.raises(A__						):
                  http_head('https://huggingface.co'						)
