​
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.