Method apply_template

src/prompt.py:79–102

(self, name, rd, kwargs, condition = None)

    def apply_template(self, name, rd, kwargs, condition = None):
        for t in self.templates[name]:
            if t["rd"] == rd and t["condition"] == condition:
                template = t
                break
        prompt = ""
        if "demo" in template and template["demo"] != "":
            index = 0
            while True:
                prompt += template["demo"]
                for parameter in template["demo_parameters"]:
                    if isinstance(kwargs[parameter], str):
                        prompt = prompt.replace("{" + parameter + "}", kwargs[parameter])
                        break
                    elif isinstance(kwargs[parameter], list):
                        prompt = prompt.replace("{" + parameter + "}", kwargs[parameter][index])
                index += 1
                if index >= len(kwargs[list(kwargs.keys())[0]]):
                    break
        prompt += template["instruction"]
        for parameter in template["instruction_parameters"]:
            prompt = prompt.replace("{" + parameter + "}", kwargs[parameter])
        prompt = prompt.replace("[trigger]", self.templates["trigger"]).replace("[regular_requirement]", self.templates["regular_requirement"])
        return prompt

    def build_chat_message(self, message, history):
        return history + [{"role": "user", "content": message}]