musev/auto_prompt/util.py

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