PromptTemplateGenerator

class PromptTemplateGenerator:
A class for generating prompt templates for tasks. Parameters:
  • task_prompt_template_dict (TaskPromptTemplateDict, optional): A dictionary of task prompt templates for each task type. If not provided, an empty dictionary is used as default.

init

def __init__(
    self,
    task_prompt_template_dict: Optional[TaskPromptTemplateDict] = None
):

get_prompt_from_key

def get_prompt_from_key(self, task_type: TaskType, key: Any):
Generates a text prompt using the specified :obj:task_type and :obj:key. Parameters:
  • task_type (TaskType): The type of task.
  • key (Any): The key used to generate the prompt.
Returns: TextPrompt: The generated text prompt.

get_system_prompt

def get_system_prompt(self, task_type: TaskType, role_type: RoleType):
Generates a text prompt for the system role, using the specified :obj:task_type and :obj:role_type. Parameters:
  • task_type (TaskType): The type of task.
  • role_type (RoleType): The type of role, either “USER” or “ASSISTANT”.
Returns: TextPrompt: The generated text prompt.

get_generate_tasks_prompt

def get_generate_tasks_prompt(self, task_type: TaskType):
Gets the prompt for generating tasks for a given task type. Parameters:
  • task_type (TaskType): The type of the task.
Returns: TextPrompt: The generated prompt for generating tasks.

get_task_specify_prompt

def get_task_specify_prompt(self, task_type: TaskType):
Gets the prompt for specifying a task for a given task type. Parameters:
  • task_type (TaskType): The type of the task.
Returns: TextPrompt: The generated prompt for specifying a task.