File size: 970 Bytes
06e9d12 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
from copy import deepcopy
from typing import Dict, List
from .load_template import get_template_by_name
def generate_prompts(tasks: List[Dict]) -> List[Dict]:
new_tasks = []
for task in tasks:
task["origin_prompt"] = deepcopy(task["prompt"])
# 如果prompt单元值含有模板 {}，或者 没有填写任何值（默认为空模板），则使用原prompt值
if "{" not in task["prompt"] and len(task["prompt"]) != 0:
new_tasks.append(task)
else:
template = get_template_by_name(
template=task["prompt"], name=task.get("template_name", None)
)
prompts = template(task)
if not isinstance(prompts, list) and isinstance(prompts, str):
prompts = [prompts]
for prompt in prompts:
task_cp = deepcopy(task)
task_cp["prompt"] = prompt
new_tasks.append(task_cp)
return new_tasks
|