49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
#!/usr/bin/env python
|
|
# coding=utf-8
|
|
|
|
# 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
|
|
CHAT_MESSAGE_PROMPT = """
|
|
Human: <<task>>
|
|
|
|
Assistant: """
|
|
|
|
|
|
DEFAULT_PROMPTS_REPO = "huggingface-tools/default-prompts"
|
|
PROMPT_FILES = {"chat": "chat_prompt_template.txt", "run": "run_prompt_template.txt"}
|
|
|
|
|
|
def download_prompt(prompt_or_repo_id, agent_name, mode="run"):
|
|
"""
|
|
Downloads and caches the prompt from a repo and returns it contents (if necessary)
|
|
"""
|
|
if prompt_or_repo_id is None:
|
|
prompt_or_repo_id = DEFAULT_PROMPTS_REPO
|
|
|
|
# prompt is considered a repo ID when it does not contain any kind of space
|
|
if re.search("\\s", prompt_or_repo_id) is not None:
|
|
return prompt_or_repo_id
|
|
|
|
prompt_file = cached_file(
|
|
prompt_or_repo_id, PROMPT_FILES[mode], repo_type="dataset", user_agent={"agent": agent_name}
|
|
)
|
|
with open(prompt_file, "r", encoding="utf-8") as f:
|
|
return f.read()
|