You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

prompt_builder.py 387B

2 years ago
12345678910
  1. from core.prompt.prompt_template import PromptTemplateParser
  2. class PromptBuilder:
  3. @classmethod
  4. def parse_prompt(cls, prompt: str, inputs: dict) -> str:
  5. prompt_template = PromptTemplateParser(prompt)
  6. prompt_inputs = {k: inputs[k] for k in prompt_template.variable_keys if k in inputs}
  7. prompt = prompt_template.format(prompt_inputs)
  8. return prompt