utils.py

import pandas as pd
from typing import List


def generate_prompts(prompt_template: str, data_df: pd.DataFrame) -> List[str]:
    """
    Generates prompts for the rows in data_df using the template prompt_template.

    Args:
        prompt_template: a prompt template
        data_df: pandas dataframe of samples to generate prompts for
    
    Returns:
        prompts: a list of prompts corresponding to the rows of data_df
    """
    assert (
        "{{" in prompt_template
    ), f"Prompt template has no fields to fill, {prompt_template}"

    prompts = []
    dicts = data_df.to_dict(orient="records")
    for dd in dicts:
        prompt = str(prompt_template)
        for k, v in dd.items():
            prompt = prompt.replace("{{" + k + "}}", str(v))
        assert not "{{" in prompt, print(prompt)
        prompts.append(prompt)
    assert len(set(prompts)) == len(prompts), "Duplicated prompts detected"
    return prompts